标签: tensorflow keras deep-learning pytorch
我正在尝试与Keras进行深度模型修剪。这些连接位于不同的位置和不同的完全连接层,因此设置layer.trainable=False对我不起作用。
layer.trainable=False
那么如何在Keras中修剪这些连接(在训练过程中将其权重冻结为0)?我需要编写自己的图层以及如何编写吗?
诸如PyTorch之类的其他工具是否容易支持这种修剪(或冻结)操作(无需编写自己的图层)?
答案 0 :(得分:0)
您会感兴趣的这个工具:https://www.tensorflow.org/model_optimization/guide/pruning