From aa505d4192cc496145250f929bec49ec7044e7ec Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Sat, 30 Aug 2025 16:40:18 +0200
Subject: [PATCH] more video prefetch

---
 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 9e9d9ff77..6594e81b5 100644
--- a/src/lerobot/scripts/train.py
+++ b/src/lerobot/scripts/train.py
@@ -258,7 +258,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,  # More aggressive prefetching for video
+        prefetch_factor=12,  # Maximum prefetch for video pipeline
         timeout=30,  # Prevent hanging on video decode errors
     )
     dl_iter = cycle(dataloader)