如何只训练网络的第一层？

Question

我已经训练了一个具有两个卷积层和两个密集层的网络。
然后由于某种原因，我需要添加另一个卷积层作为网络的第一层，所以我添加了一个层并冻结其他层，但是正如您在下图中看到的那样，该层的权重没有显着变化。

有没有人知道为什么会发生这种情况？这是消失梯度的影响还是其他什么？

Answer 1

tf.optimizer.minimize()s 接受var_list 参数，把你要更新的变量放进去就可以了。