定義
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
)
給定優(yōu)化器的損失和參數(shù),返回 training op。
參數(shù)定義
- loss:損失函數(shù)
- global_step:獲取訓練步數(shù)并在訓練時更新
- learning_rate:學習率
- optimizer:定義優(yōu)化器
- gradient_noise_scale:float或None,添加由此值縮放的0均值正常噪聲。
-
gradient_multipliers:浮點數(shù)的變量或變量名稱。 如果存在,指定變量的梯度將乘以給定常量。
……
……
……