Lidar_AI_Solution icon indicating copy to clipboard operation
Lidar_AI_Solution copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

How to export Swint model to onnx?

Open jorgemn9 opened this issue 4 months ago • 0 comments

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. 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 export swint model to onnx?

Thanks in advance.

jorgemn9 avatar Sep 29 '24 21:09 jorgemn9