在训练期间启用和禁用变量学习

2024-04-19

我想在训练期间开始或停止变量的学习。虽然Variable的构造函数有一个trainable参数,该参数构建后不可更改。

为了实现这种效果,我当前的解决方案是在我想要更改变量的可训练状态时保存模型,将变量设置为所需的可训练状态来重建网络,然后重新加载权重。

有没有一种不太麻烦的方法来以编程方式改变训练对变量的影响?


从 TF 1.8 开始,现在可以通过为可动态设置为 0 或 1 的变量设置自定义学习率乘数来轻松完成此操作,从而有效地打开或关闭变量的训练。

有关自定义学习率乘数的更多信息here https://stackoverflow.com/a/50388264/1735003.

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在训练期间启用和禁用变量学习 的相关文章

随机推荐