|
@@ -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)
|