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

Question

我将从tensorflow_model_optimization、tfmot.sparsity.keras.prune_low_magnitude()到MobileNetV2应用修剪函数。你知道吗

有没有办法只将模型的某些层设置为可修剪的？对于训练，有一个方法“set_trainable”，但我还没有找到任何等效的修剪方法。你知道吗

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

Answer 1

最后，我发现你也可以在每层应用prune\u low\u magnitude（）。你知道吗

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