From 9027c7866f5db27afec59691eab2949f754f48c5 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Mon, 1 Sep 2025 12:12:36 +0200 Subject: [PATCH] less prefetching --- src/lerobot/scripts/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)