有什么方法可以验证我是否已成功冻结Tensorflow模型中的某些变量/参数？

Question

我试图通过仅传递我想通过

进行微调的变量来冻结某些变量。

 tf.train.AdamOptimizer.compute_gradients(var_list=[])

方法。

但是，当我继续微调模型时，损失在增加，而准确性在下降，因此我怀疑我是否已经成功冻结了最初的几层。

有什么方法可以检查我是否正确冻结了变量？谢谢！

Answer 1

尝试tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)。