From 90363d41891739d032bfd40d01d1df708b7061bc Mon Sep 17 00:00:00 2001 From: Bruce Wayne Date: Fri, 18 Sep 2026 13:39:16 -0700 Subject: [PATCH] worldmodel: default shuffle buffer to 25,000 --- torchtitan/experiments/worldmodel/dataset_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchtitan/experiments/worldmodel/dataset_config.py b/torchtitan/experiments/worldmodel/dataset_config.py index d77594c159a..76c4f926529 100644 --- a/torchtitan/experiments/worldmodel/dataset_config.py +++ b/torchtitan/experiments/worldmodel/dataset_config.py @@ -19,7 +19,7 @@ def _dataloader_config( split: str, dataset: str = DEFAULT_TRAIN_LIST, dataset_path: str | None = None, - shuffle_size: int = 50_000, + shuffle_size: int = 25_000, min_mixing: float = 0.5, num_writers: int = 2, num_readers: int = 4,