执行pytorch2onnx.py时报错

[ZZHHogan]

您好，我在执行pytorch2onnx.py这个文件的时候报错： ====> check onnx model... ====> Simplifying... Traceback (most recent call last): File "pytorch2onnx.py", line 46, in onnx.save(model_opt, args.onnx_model_sim) File "C:\Users\Administrator\anaconda3\envs\py37\lib\site-packages\onnx_init_.py", line 184, in save_model proto = write_external_data_tensors(proto, basepath) File "C:\Users\Administrator\anaconda3\envs\py37\lib\site-packages\onnx\external_data_helper.py", line 225, in write_external_data_tensors for tensor in _get_all_tensors(model): File "C:\Users\Administrator\anaconda3\envs\py37\lib\site-packages\onnx\external_data_helper.py", line 170, in _get_initializer_tensors for initializer in onnx_model_proto.graph.initializer: AttributeError: 'tuple' object has no attribute 'graph' 问题应该是Simplifying的时候出错了，应该怎么解决呢？

[chenchangquan5]

代码： model_opt = onnxsim.simplify(args.onnx_model) 修改成如下： model_opt, check = onnxsim.simplify(args.onnx_model) 即可解决