From dbce707db5adbb8a973f7c46d0bde19d30349614 Mon Sep 17 00:00:00 2001 From: AdilZouitine Date: Fri, 3 Oct 2025 16:43:05 +0200 Subject: [PATCH] Initialize logging in training script for both main and non-main processes - Added `init_logging` calls to ensure proper logging setup when using the accelerator and in standard training mode. - This change enhances the clarity and consistency of logging during training sessions. --- src/lerobot/scripts/lerobot_train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 4493b1167..7d8b2853a 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -439,6 +439,9 @@ if __name__ == "__main__": # We set step_scheduler_with_optimizer False to prevent accelerate from # adjusting the lr_scheduler steps based on the num_processes accelerator = accelerate.Accelerator(step_scheduler_with_optimizer=False) + + init_logging(accelerator=accelerator) train(accelerator=accelerator) else: + init_logging() train()