CloudAnts commited on
Commit
9988247
·
verified ·
1 Parent(s): b967d8f

Update ultralytics/nn/tasks.py

Browse files
Files changed (1) hide show
  1. ultralytics/nn/tasks.py +1 -1
ultralytics/nn/tasks.py CHANGED
@@ -730,7 +730,7 @@ def torch_safe_load(weight):
730
  "ultralytics.yolo.data": "ultralytics.data",
731
  }
732
  ): # for legacy 8.0 Classify and Pose models
733
- ckpt = torch.load(file, map_location="cpu")
734
 
735
  except ModuleNotFoundError as e: # e.name is missing module name
736
  if e.name == "models":
 
730
  "ultralytics.yolo.data": "ultralytics.data",
731
  }
732
  ): # for legacy 8.0 Classify and Pose models
733
+ ckpt = torch.load(file, map_location="cpu",weights_only=False)
734
 
735
  except ModuleNotFoundError as e: # e.name is missing module name
736
  if e.name == "models":