tf.train.MomentumOptimizer 优化器

tf.train.MomentumOptimizer(
    learning_rate,
    momentum,
    use_locking=False,
    use_nesterov=False,
    name='Momentum'
)
参数:
learning_rate: (学习率)张量或者浮点数
momentum: (动量)张量或者浮点数
use_locking: 为True时锁定更新
use_nesterov:  为True时,使用 Nesterov Momentum
name:  梯度下降名称,默认为 "Momentum"
posted @ 2019-07-17 20:02  JohnRed  阅读(6836)  评论(0)    收藏  举报