目录

定义

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:浮点数的变量或变量名称。 如果存在,指定变量的梯度将乘以给定常量。