gradients = optimizer.compute_gradients(loss, var_list)
capped_gradients = [(tf.clip_by_value(grad, -5., 5.), var) for grad, var in gradients if grad is not None]
train_op = optimizer.apply_gradients(capped_gradients)
梯度截断的tensorflow实现
最新推荐文章于 2022-05-22 11:17:32 发布