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