Как установить сокращаемые слои для tfmot.sparsity.keras.prune_low_magnitude? - PullRequest
0 голосов
/ 05 ноября 2019

Я применяю функцию обрезки с tensorflow_model_optimization, tfmot.sparsity.keras.prune_low_magnitude() к MobileNetV2.

Можно ли настроить обрезку только для некоторых слоев модели? Для обучения есть метод "set_trainable", но я не нашел никакого эквивалента для обрезки.

Любые идеи или комментарии будут оценены по достоинству! :)

...