Run the command:
python3 export_onnx.py
--config waymo_centerpoint_pp_two_pfn_stride1_3x.py
--ckpt your_model.pth
--pfe_save_path pfe.onnx
--rpn_save_path rpn.onnx,
The terminal outputs the following:
File "/usr/local/lib/python3.8/dist-packages/torch/nn/functional.py", line 1756, in linear
return torch._C._nn.linear(input, weight, bias)
RuntimeError: mat1 dim 1 must match mat2 dim 0
Please tell me how to solve this problem?
Run the command:
python3 export_onnx.py
--config waymo_centerpoint_pp_two_pfn_stride1_3x.py
--ckpt your_model.pth
--pfe_save_path pfe.onnx
--rpn_save_path rpn.onnx,
The terminal outputs the following:
File "/usr/local/lib/python3.8/dist-packages/torch/nn/functional.py", line 1756, in linear
return torch._C._nn.linear(input, weight, bias)
RuntimeError: mat1 dim 1 must match mat2 dim 0
Please tell me how to solve this problem?