我正在将来自 tensorflow_model_optimization 的修剪函数 tfmot.sparsity.keras.prune_low_magnitude() 应用于 MobileNetV2。

有没有办法只设置模型的某些层为可修剪的？对于训练，有一个方法 “set_trainable“，但我没有找到修剪的等效方法。

任何想法或评论都将不胜感激！ 🙂

回答：

最后我发现你也可以对每一层应用 prune_low_magnitude()。

因此，解决方法是定义一个列表，包含要修剪的层的名称或类型，并在这个列表中对所有层进行逐层修剪。