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":
|