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

Question

我想在训练期间开始或停止学习变量。虽然Variable的构造函数有trainable参数，但是构造后这个参数是不能改变的。

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

有没有更简单的方法来以编程方式改变训练对变量的影响？

Answer 1

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

更多关于自定义学习率乘数here。