Существует ли какой-либо код keras для воспроизведения сокращения веса MobileNet? - PullRequest
Новая
       37

Существует ли какой-либо код keras для воспроизведения сокращения веса MobileNet?

0 голосов
/ 06 июня 2019

Я хочу сократить модель MobileNetV1, используя метод сокращения веса от https://www.tensorflow.org/model_optimization/guide/pruning Тем не менее, я не знаю, как установить 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?

...