Skip to content

RuntimeError: mat1 dim 1 must match mat2 dim 0 #44

@dongdongzi

Description

@dongdongzi

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?

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