Tensorflow no_grad概念

时间:2019-06-07 21:11:34

标签: python tensorflow pytorch

我知道使用pytorch可以通过将模型设置为eval()来关闭给定元素的训练。您也可以将require_grad设置为false。如何确保在训练过程中不会修改张量流元素?

1 个答案:

答案 0 :(得分:0)

如果您不想在TensorFlow中训练某些变量,可以通过在变量中添加trainable=False来实现此行为。