⬆️ [Update] dependency and the load cache function
Browse files- requirements-dev.txt +1 -0
- yolo/tools/data_loader.py +1 -1
requirements-dev.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
-r requirements.txt
|
|
|
2 |
pytest
|
3 |
pytest-cov
|
4 |
pre-commit
|
|
|
1 |
-r requirements.txt
|
2 |
+
gradio
|
3 |
pytest
|
4 |
pytest-cov
|
5 |
pre-commit
|
yolo/tools/data_loader.py
CHANGED
@@ -52,7 +52,7 @@ class YoloDataset(Dataset):
|
|
52 |
data = self.filter_data(dataset_path, phase_name)
|
53 |
torch.save(data, cache_path)
|
54 |
else:
|
55 |
-
data = torch.load(cache_path)
|
56 |
logger.info("📦 Loaded {} cache", phase_name)
|
57 |
return data
|
58 |
|
|
|
52 |
data = self.filter_data(dataset_path, phase_name)
|
53 |
torch.save(data, cache_path)
|
54 |
else:
|
55 |
+
data = torch.load(cache_path, weights_only=False)
|
56 |
logger.info("📦 Loaded {} cache", phase_name)
|
57 |
return data
|
58 |
|