less prefetching

2026-05-16 09:09:48 +00:00 · 2025-09-01 12:12:36 +02:00
parent c2bf226082
commit 9027c7866f
1 changed files with 1 additions and 1 deletions
@@ -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)