开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

通过张量融合实现的高性能Adam优化器,核心功能和[object Object]兼容。

Adam的功能和原理可参考

[object Object]
[object Object]
[object Object]
  • params ([object Object]):必选参数,模型参数或模型参数组,[object Object]为参数的可迭代对象或参数组的dict类型。
  • lr ([object Object]):可选参数,学习率,默认值为1e-3。[object Object]的值小于0时,打印“ValueError”异常信息。
  • betas ([object Object]):可选参数,用于计算梯度及其平方的运行平均值的系数,[object Object]为包含两个值的tuple类型,默认值为(0.9,0.999)。[object Object]的值小于0或者[object Object]的值大于1时,打印“ValueError”异常信息。
  • eps ([object Object]):可选参数,防止除0,提高数值稳定性,默认值为1e-8。[object Object]小于0时,打印“ValueError”异常信息。
  • weight_decay ([object Object]):可选参数,权重衰减,默认值为0。[object Object]小于0时,打印“ValueError”异常信息。
  • amsgrad ([object Object]):可选参数,是否使用AMSGrad,默认值为False。
[object Object]

类型为[object Object]的对象。

[object Object]

[object Object]的实现机制要求[object Object]中的每一个模型参数对象在使用过程中不能被重新申请,否则将导致无法预料的结果。引起模型参数对象被重新申请的操作包括但不限于:

  • 将模型或其子Module进行.cpu操作
  • 将模型参数对象指向新的对象
  • 将模型参数对象置为None

对模型参数对象进行inplace计算,或者读取参数的值,[object Object]可正常工作。

[object Object]
[object Object]