bug3 (deepNN) agxxavier@agxxavier-desktop:~/yolov5$ python gen_wts.py -w yolov5s.pt -o yolov5s.wts Traceback (most recent call last): File "gen_wts.py", line 30, in <module> model = torch.load(pt_file, map_location=device)['model'].float() # load to FP32 File "/home...