当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.optimizer.Adamax用法及代码示例


用法:

class mxnet.optimizer.Adamax(learning_rate=0.002, beta1=0.9, beta2=0.999, **kwargs)

参数

  • beta1(float, optional) - 一阶矩估计的 index 衰减率。
  • beta2(float, optional) - 二阶矩估计的 index 衰减率。

基础:mxnet.optimizer.optimizer.Optimizer

AdaMax 优化器。

它是基于 http://arxiv.org/abs/1412.6980 第 7 节中可用的无穷范数的 Adam 的变体。

优化器通过以下方式更新权重:

grad = clip(grad * rescale_grad + wd * weight, clip_gradient)
m = beta1 * m_t + (1 - beta1) * grad
u = maximum(beta2 * u, abs(grad))
weight -= lr / (1 - beta1**t) * m / u

除了 Optimizer 接受的参数之外,此优化器还接受以下参数。

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.optimizer.Adamax。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。