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"

相关文章:

  • 2021-10-17
  • 2022-01-08
  • 2022-02-04
  • 2021-08-23
  • 2021-07-30
  • 2022-02-05
  • 2022-12-23
  • 2022-12-23
猜你喜欢
  • 2021-10-08
  • 2021-08-24
  • 2021-08-15
  • 2021-12-19
  • 2021-06-17
相关资源
相似解决方案