Hugging Face's logo

Spaces:
XUSHUYA
/
YOLO_modol
Sleeping

App Files Community
henry000 commited on
Commit
b0c4448
·
1 Parent(s): e8991f8

✏️ [Revert] typo in cache files

Browse files
Files changed (1) hide show
  1. yolo/tools/data_loader.py +1 -1
yolo/tools/data_loader.py CHANGED
@@ -46,7 +46,7 @@ class YoloDataset(Dataset):
46
  Returns:
47
  dict: The loaded data from the cache for the specified phase.
48
  """
49
- cache_path = dataset_path / f"{phase_name}.cache1"
50
 
51
  if not cache_path.exists():
52
  logger.info(f":factory: Generating {phase_name} cache")
 
46
  Returns:
47
  dict: The loaded data from the cache for the specified phase.
48
  """
49
+ cache_path = dataset_path / f"{phase_name}.cache"
50
 
51
  if not cache_path.exists():
52
  logger.info(f":factory: Generating {phase_name} cache")