我是靠谱客的博主 天真冰淇淋,这篇文章主要介绍tf.train.AdamOptimizertf.train.AdamOptimizer,现在分享给大家,希望可以做个参考。

tf.train.AdamOptimizer

作用:

  • 实现Adam算法的优化器

函数原型:

复制代码
1
2
3
4
5
6
7
8
__init__( learning_rate=0.001, beta1=0.9, beta2=0.999, epsilon=1e-08, use_locking=False, name='Adam')

方法:

复制代码
1
2
3
4
5
6
7
8
compute_gradients( loss, var_list=None, gate_gradients=GATE_OP, aggregation_method=None, colocate_gradients_with_ops=False, grad_loss=None)
  • 作用:计算loss相对变量var_list的梯度,返回一个(gradient, variable)的list。
复制代码
1
2
3
4
5
apply_gradients( grads_and_vars, global_step=None, name=None)
  • 作用:应用梯度到相应的变量grads_and_vars是上一个函数的返回值
  • global_step:每当变量更新一次就加一

最后

以上就是天真冰淇淋最近收集整理的关于tf.train.AdamOptimizertf.train.AdamOptimizer的全部内容,更多相关tf内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(78)

评论列表共有 0 条评论

立即
投稿
返回
顶部