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

export ONNX error

Open slayerlpj opened this issue 6 months ago • 1 comments

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?

slayerlpj avatar Jul 30 '24 02:07 slayerlpj