如何为 tfmot.sparsity.keras.prune_low_magnitude 设置可修剪层？

Question

我正在将 tensorflow_model_optimization、tfmot.sparsity.keras.prune_low_magnitude() 的剪枝功能应用到 MobileNetV2。

有没有办法只设置模型的某些层可修剪？对于训练，有一种方法“set_trainable”，但我还没有找到任何等价的修剪方法。

如有任何想法或意见，我们将不胜感激！ :)

Answer 1

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

因此解决方法是定义一个列表，其中包含应被修剪的层的名称或类型，并在该列表中的所有层上迭代逐层修剪。