我想在训练期间开始或停止变量的学习。虽然Variable
的构造函数有一个trainable
参数,该参数构建后不可更改。
为了实现这种效果,我当前的解决方案是在我想要更改变量的可训练状态时保存模型,将变量设置为所需的可训练状态来重建网络,然后重新加载权重。
有没有一种不太麻烦的方法来以编程方式改变训练对变量的影响?
从 TF 1.8 开始,现在可以通过为可动态设置为 0 或 1 的变量设置自定义学习率乘数来轻松完成此操作,从而有效地打开或关闭变量的训练。
有关自定义学习率乘数的更多信息here https://stackoverflow.com/a/50388264/1735003.
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)