shubham-mishra-verificient

Hi, I've tried your code on your provided weights its working fine, but when I tried my custom weight exported using ultralytics repo, even using older tags is not working....

getting following error while loading snapshot Traceback (most recent call last): File "code/test_on_image.py", line 46, in saved_state_dict = torch.load(snapshot_path) File "/usr/local/lib/python3.7/site-packages/torch/serialization.py", line 593, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)...