pytorch模型转onnx模型
kiwins:
Traceback (most recent call last):
File "torch2onnx_2.py", line 5, in
model = torch.load('FCN.pth')
File "/home/umhos/.local/lib/python3.8/site-packages/torch/serialization.py", line 713, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/umhos/.local/lib/python3.8/site-packages/torch/serialization.py", line 930, in _legacy_load
result = unpickler.load()
File "/home/umhos/.local/lib/python3.8/site-packages/torch/serialization.py", line 746, in find_class
return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'FCN8s' on
请问执行转换的时候总是报这个错误,是什么原因呢?
|