search

NVIDIA-AI-IOT / Lidar_AI_Solution

A project demonstrating Lidar related AI solutions, including three GPU accelerated Lidar/camera DL networks (PointPillars, CenterPoint, BEVFusion) and the related libs (cuPCL, 3D SparseConvolution, YUV2RGB, cuOSD,).
Other
1.27k stars 224 forks source link

export ONNX error #272

Open slayerlpj opened 2 months ago

slayerlpj commented 2 months ago

Thanks for you repo! I'm tring to convert pth to onnx. But When I use "python qat/export-camera.py --ckpt=model/resnet50/bevfusion-det.pth --fp16". I get a error "model = torch.load(args.ckpt).module AttributeError: 'dict' object has no attribute 'module' ". How can I fix the problem?

jorgemn9 commented 2 weeks ago

Hi! Did you solve this? Thanks in advance.