diff --git a/src/lerobot/scripts/train.py b/src/lerobot/scripts/train.py index e5b6c33de..d0becf380 100644 --- a/src/lerobot/scripts/train.py +++ b/src/lerobot/scripts/train.py @@ -270,7 +270,7 @@ def train(cfg: TrainPipelineConfig): pin_memory=device.type == "cuda", drop_last=False, persistent_workers=cfg.num_workers > 0, # Keep workers alive between epochs - prefetch_factor=6, # Maximum prefetch for video pipeline + prefetch_factor=3, # Prefetch for video pipeline timeout=30, # Prevent hanging on video decode errors ) dl_iter = cycle(dataloader)