소스 검색

add prune configs and prompt

sunyanfang01 5 년 전
부모
커밋
b3ab0adfe7
1개의 변경된 파일0개의 추가작업 그리고 1개의 파일을 삭제
  1. 0 1
      paddlex/cv/models/slim/prune.py

+ 0 - 1
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)