From b3ab0adfe7958c3c06a90ab4af695400a5f94c03 Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Tue, 16 Jun 2020 10:47:31 +0800 Subject: [PATCH] add prune configs and prompt --- paddlex/cv/models/slim/prune.py | 1 - 1 file changed, 1 deletion(-) diff --git a/paddlex/cv/models/slim/prune.py b/paddlex/cv/models/slim/prune.py index f60f6b8..bd6c1aa 100644 --- a/paddlex/cv/models/slim/prune.py +++ b/paddlex/cv/models/slim/prune.py @@ -236,7 +236,6 @@ def cal_params_sensitivities(model, save_file, eval_dataset, batch_size=8): 其中``weight_0``是卷积Kernel名;``sensitivities['weight_0']``是一个字典,key是裁剪率,value是敏感度。 """ - print('-----------', model.status) assert model.status == 'Normal', 'Only the model after training can calculate sensitivities data!' if os.path.exists(save_file): os.remove(save_file) -- GitLab