我知道使用 pytorch,您可以通过在模型上调用 eval()
来关闭训练。
您还可以设置requires_grad=False
。
如何确保 TensorFlow 元素在训练期间不被修改?
最佳答案
如果您不想在 TensorFlow 中训练某些变量,可以通过向变量添加 trainable=False
来实现此行为。
关于python - tensorflow no_grad 概念,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56501260/