RuntimeError: The size of tensor a (32000) must match the size of tensor b (17444) at non-singleton dimension 0

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:
RuntimeError: The size of tensor a (32000) must match the size of tensor b (17444) at non-singleton dimension 0

Please tell me how to solve this problem?