henry000 commited on
Commit
97211aa
Β·
2 Parent(s): da1069f b0c4448

πŸ”€ [Merge] branch 'TEST'

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