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

时间:2017-04-28 13:22:30

标签: tensorflow

我想在训练期间开始或停止学习变量。虽然Variable的构造函数具有trainable参数,但在构造之后无法更改此参数。

为了达到这个效果,我目前的解决方案是将模型保存在我想要更改变量的可训练状态的点上,使用设置为所需训练状态的变量重建网络,然后重新加载权重。

是否有一种不那么繁琐的方法来以编程方式改变训练对变量的影响?

1 个答案:

答案 0 :(得分:0)

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

有关自定义学习费率乘数here的更多信息。