Skip to content

convert .pt to .onnx? #4

@maomao0220

Description

@maomao0220

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions