python - TensorFlow no_grad 概念

Question

eval()我知道使用 pytorch 你可以通过调用你的模型来关闭训练。您也可以设置requires_grad=False.

如何确保在训练期间不修改 TensorFlow 元素？

标签： pythontensorflowpytorch

Answer 1

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