From 5245332e3616330e1d3a5683eb214695053cbf67 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 25 Nov 2025 13:26:30 +0100 Subject: [PATCH] print batch size --- src/lerobot/scripts/lerobot_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 435cc1256..a4fede3d3 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -378,7 +378,7 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None): ) if is_main_process: - logging.info("Start offline training on a fixed dataset") + logging.info(f"Start offline training on a fixed dataset, with effective batch size: {effective_batch_size}") for _ in range(step, cfg.steps): start_time = time.perf_counter()