Add num_workers and batch_size to default.yaml

lerobot/configs/default.yaml

@@ -26,6 +26,8 @@ training:
   save_freq: ???
   log_freq: 250
   save_model: true
+  num_workers: 4
+  batch_size: ???
 
 eval:
   n_episodes: 1

lerobot/scripts/train.py

@@ -386,7 +386,7 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No
     # create dataloader for offline training
     dataloader = torch.utils.data.DataLoader(
         offline_dataset,
-        num_workers=4,
+        num_workers=cfg.training.num_workers,
         batch_size=cfg.training.batch_size,
         shuffle=True,
         pin_memory=cfg.device != "cpu",