henry000 commited on
Commit
699f2d2
·
1 Parent(s): 0efc848

⬆️ [Update] dependency and the load cache function

Browse files
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