是否有任何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_step和frequency？还是应该使用ConstantSparsity代替PolynomialDecay？