标签: tensorflow keras deep-learning
所以,我只需要冻结角膜完全连接的致密层中的一两个神经元,以使其在训练过程中不会改变其重量。在喀拉拉邦有办法吗?
答案 0 :(得分:0)
您可能需要定义一个蒙版并将与要冻结的神经元相对应的蒙版元素设置为0。这样梯度就不会传播,也就不会训练神经元。
类似于以下内容: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/model_pruning