gpt4 book ai didi

python - tensorflow no_grad 概念

转载 作者:太空宇宙 更新时间:2023-11-03 20:47:15 36 4
gpt4 key购买 nike

我知道使用 pytorch,您可以通过在模型上调用 eval() 来关闭训练。您还可以设置requires_grad=False

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

最佳答案

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

关于python - tensorflow no_grad 概念,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56501260/

36 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com