-
Notifications
You must be signed in to change notification settings - Fork 1
Description
when I convert .pt to .onnx, occur some quetion:
Traceback (most recent call last):
File "export_onnx.py", line 171, in
convert_onnx()
File "export_onnx.py", line 69, in convert_onnx
fuse_modules(deblock, [['0', '1']], inplace=True)
File "/home/maomao/anaconda3/envs/centerpoint/lib/python3.8/site-packages/torch/ao/quantization/fuse_modules.py", line 146, in fuse_modules
_fuse_modules(model, module_list, fuser_func, fuse_custom_config_dict)
File "/home/maomao/anaconda3/envs/centerpoint/lib/python3.8/site-packages/torch/ao/quantization/fuse_modules.py", line 77, in _fuse_modules
new_mod_list = fuser_func(mod_list, additional_fuser_method_mapping)
File "/home/maomao/anaconda3/envs/centerpoint/lib/python3.8/site-packages/torch/ao/quantization/fuse_modules.py", line 45, in fuse_known_modules
fuser_method = get_fuser_method(types, additional_fuser_method_mapping)
File "/home/maomao/anaconda3/envs/centerpoint/lib/python3.8/site-packages/torch/ao/quantization/fuser_method_mappings.py", line 132, in get_fuser_method
assert fuser_method is not None, "did not find fuser method for: {} ".format(op_list)
AssertionError: did not find fuser method for: (<class 'torch.nn.modules.conv.ConvTranspose2d'>, <class 'torch.nn.modules.batchnorm.BatchNorm2d'>)
How can i do? Thanks for your reply