我们可以将选定的神经元冻结在角膜的完全连接层中吗?

时间:2019-05-30 17:03:38

标签: tensorflow keras deep-learning

所以,我只需要冻结角膜完全连接的致密层中的一两个神经元,以使其在训练过程中不会改变其重量。在喀拉拉邦有办法吗?

1 个答案:

答案 0 :(得分:0)

您可能需要定义一个蒙版并将与要冻结的神经元相对应的蒙版元素设置为0。这样梯度就不会传播,也就不会训练神经元。

类似于以下内容: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/model_pruning