@@ -43,7 +43,7 @@ def prune_weights(model_path, pruned_model, pruning_percentage=0.05):
)
# 创建新的模型
- new_model = onnx.helper.make_model(new_graph, producer_name='onnx-example')
+ new_model = onnx.helper.make_model(new_graph, producer_name='onnx-example', opset_imports=model.opset_import)
# 保存剪枝后的模型
onnx.save(new_model, pruned_model)