Python 如何为tfmot.sparsity.keras.prune\u low\u震级设置可修剪层？

我正在将

tensorflow\u model\u optimization

，

tfmot.sparsity.keras.prune\u low\u magnity（）

中的修剪函数应用于MobileNetV2

有没有办法只将模型的某些层设置为可修剪的？对于培训，有一种方法“

set\u trainable

”，但我还没有找到任何等效的修剪方法

---

如有任何想法或意见，将不胜感激！：）

最后，我发现你也可以在每层应用prune\u low\u magnity（）

---

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

对于任何试图进行类似操作的人，都可以使用相同的文档，感谢链接，此文档在当时不可用：）这非常有用，的确