-
Notifications
You must be signed in to change notification settings - Fork 12
Description
I replaced deform_conv.py as described but keep getting ;
RuntimeError: No Op registered for MMCVDeformConv2d with domain_version of 11
==> Context: Bad node spec for node. Name: MMCVDeformConv2d_133 OpType: MMCVDeformConv2d
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "ganet_pth2onnx.py", line 59, in
export_onnx('./ganet.onnx')
File "ganet_pth2onnx.py", line 54, in export_onnx
torch.onnx.export(ganet_model, dummy_img, onnx_file_path, opset_version=11, enable_onnx_checker=False)
File "/home/aptiv/anaconda3/envs/ganet/lib/python3.8/site-packages/torch/onnx/init.py", line 316, in export
return utils.export(model, args, f, export_params, verbose, training,
File "/home/aptiv/anaconda3/envs/ganet/lib/python3.8/site-packages/torch/onnx/utils.py", line 107, in export
_export(model, args, f, export_params, verbose, training, input_names, output_names,
File "/home/aptiv/anaconda3/envs/ganet/lib/python3.8/site-packages/torch/onnx/utils.py", line 787, in _export
raise ONNXCheckerError(e)
torch.onnx.utils.ONNXCheckerError: No Op registered for MMCVDeformConv2d with domain_version of 11
==> Context: Bad node spec for node. Name: MMCVDeformConv2d_133 OpType: MMCVDeformConv2d
when running python ganet_pth2onnx.py
And idea what is causing this problem?