如何用Keras修剪某些重量(而不是冻结一层)?

时间:2019-03-12 07:37:05

标签: tensorflow keras deep-learning pytorch

我正在尝试与Keras进行深度模型修剪。这些连接位于不同的位置和不同的完全连接层,因此设置layer.trainable=False对我不起作用。

  1. 那么如何在Keras中修剪这些连接(在训练过程中将其权重冻结为0)?我需要编写自己的图层以及如何编写吗?

  2. 诸如PyTorch之类的其他工具是否容易支持这种修剪(或冻结)操作(无需编写自己的图层)?

1 个答案:

答案 0 :(得分:0)