Skip to content

Export model to onnx #20

@jorgemn9

Description

@jorgemn9

Hi! First of all, thanks for your great work. I have successfully tested BEVFusion with the original Nuscenes dataset and trained the network with a CARLA dataset using swint_v0p075. I have obtained the .pth file but I'm in trouble exporting the pth model of swint to the .onnx files required by BEVFusion. I have tried to use the scripts "export-camera.py", "export-scn.py" and "export-transfuser.py" to obtain the .onnx files, but I find the following error: AttributeError: 'dict' object has no attribute 'module'.

Do you have any clue on how to solve this?

Thanks in advance.

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