tensorflow - 是否有任何 keras 代码可以重现 MobileNet 的权重修剪？

Question

我想使用https://www.tensorflow.org/model_optimization/guide/pruning中的权重修剪方法修剪 MobileNetV1 模型 但是，我不知道如何设置pruning_schedule.

pruning_params = {
      'pruning_schedule': sparsity.PolynomialDecay(initial_sparsity=0.50,
                                                   final_sparsity=0.90,
                                                   begin_step=2000,
                                                   end_step=end_step,
                                                   frequency=100)
}

这是来自 mnist 示例的代码。如果我希望 final_sparsity 为 0.75 并从 keras MobileNet 进行微调，我应该如何设置initial_sparsity、begin_step和？或者我应该使用而不是？end_stepfrequencyConstantSparsityPolynomialDecay

标签： tensorflowkeraspruningmobilenet