Update ultralytics/nn/tasks.py
Browse files- 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":
|