说明:pytorch的随机深度学习
Storchastic是一个用于深度学习中的随机梯度估计的PyTorch库[1]。 许多最先进的深度学习模型都使用梯度估计,尤其是在变分推理和强化学习领域。 虽然PyTorch自动计算确定性计算图的梯度,但它不会估计随机计算图上的梯度[2]。
借助Storchastic,您可以轻松定义任何随机深度学习模型,并让其为您估计梯度。 Storchastic提供了多种即插即用的梯度估计方法,以确定哪种方法最适合您的问题。 Storchastic可自动广播采样批次的尺寸
<weixin_42107374> 上传 | 大小:1mb