tensorflow 梯度裁剪
Posted callyblog
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow 梯度裁剪相关的知识,希望对你有一定的参考价值。
gvs = optimizer.compute_gradients(loss) # 计算出梯度和变量值 capped_gvs = [(tf.clip_by_value(grad, -5e+10, 5e+10), var) for grad, var in gvs] # 梯度裁剪 train_op = optimizer.apply_gradients(capped_gvs, global_step=global_step) # 梯度下降
以上是关于tensorflow 梯度裁剪的主要内容,如果未能解决你的问题,请参考以下文章