其他分享
首页 > 其他分享> > TensorFLow: Gradient Clipping

TensorFLow: Gradient Clipping

作者:互联网

The parameters clipnorm and clipvalue can be used with all optimizers to control gradient clipping。
Keras的所有optimizer都可以使用clipnormclipvalue来防止梯度过大。

from keras import optimizers

# All parameter gradients will be clipped to
# a maximum norm of 1.
sgd = optimizers.SGD(lr=0.01, clipnorm=1.)
from keras import optimizers

# All parameter gradients will be clipped to
# a maximum value of 0.5 and
# a minimum value of -0.5.
sgd = optimizers.SGD(lr=0.01, clipvalue=0.5)

标签:clipnorm,Clipping,Gradient,0.5,clipvalue,TensorFLow,gradients,sgd,optimizers
来源: https://www.cnblogs.com/yaos/p/12770046.html