admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 12:09:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

print batch size

Browse Source
This commit is contained in:
Pepijn
2025-11-25 13:26:30 +01:00
parent 4367348327
commit 5245332e36
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+1 -1
View File
@@ -378,7 +378,7 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
) )
if is_main_process: 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): for _ in range(step, cfg.steps):
start_time = time.perf_counter() start_time = time.perf_counter()