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

Question

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

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

任何想法或 cmets 将不胜感激！ :)

Answer 1

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

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