Tensorflow——tf.contrib.layers.optimize_loss

2023年9月6日 818次阅读 来源: SpareNoEfforts

定义

tf.contrib.layers.optimize_loss(
    loss,
    global_step,
    learning_rate,
    optimizer,
    gradient_noise_scale=None,
    gradient_multipliers=None,
    clip_gradients=None,
    learning_rate_decay_fn=None,
    update_ops=None,
    variables=None,
    name=None,
    summaries=None,
    colocate_gradients_with_ops=False,
    increment_global_step=True
)

给定优化器的损失和参数，返回 training op。

参数定义

  • loss：损失函数
  • global_step：获取训练步数并在训练时更新
  • learning_rate：学习率
  • optimizer：定义优化器
  • gradient_noise_scale：float或None，添加由此值缩放的0均值正常噪声。
  • gradient_multipliers：浮点数的变量或变量名称。 如果存在，指定变量的梯度将乘以给定常量。
    ……
    ……
    ……
    原文作者：SpareNoEfforts
    原文地址: https://www.jianshu.com/p/36d8ac108f6c
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞