henry000 commited on
Commit
564b976
·
1 Parent(s): 5e0d5e2

🩹 [Fix] config for auto download dataset

Browse files
Files changed (1) hide show
  1. yolo/tools/data_loader.py +1 -1
yolo/tools/data_loader.py CHANGED
@@ -213,7 +213,7 @@ def create_dataloader(data_cfg: DataConfig, dataset_cfg: DatasetConfig, task: st
213
  if task == "inference":
214
  return StreamDataLoader(data_cfg)
215
 
216
- if dataset_cfg.auto_download:
217
  prepare_dataset(dataset_cfg, task)
218
  dataset = YoloDataset(data_cfg, dataset_cfg, task)
219
 
 
213
  if task == "inference":
214
  return StreamDataLoader(data_cfg)
215
 
216
+ if getattr(dataset_cfg, "auto_download", False):
217
  prepare_dataset(dataset_cfg, task)
218
  dataset = YoloDataset(data_cfg, dataset_cfg, task)
219