소스 검색

增加日志输出

zhy 5 일 전
부모
커밋
12e6814f87
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      tests/prune_tool.py

+ 1 - 0
tests/prune_tool.py

@@ -75,4 +75,5 @@ if __name__ == '__main__':
             pruned_file = args.pruned_saved_dir + '/' + os.path.basename(onnx_file).replace('.onnx', '_pruned.onnx')
         else:
             pruned_file = onnx_file.replace('.onnx', '_pruned.onnx')
+        print(f"正在剪枝模型文件: {onnx_file} -> {pruned_file}")
         prune_weights(model_path=onnx_file, pruned_model=pruned_file, pruning_percentage=args.percent)