admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 01:41:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initialize logging in training script for both main and non-main processes

Browse Source 
- 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.
This commit is contained in:
AdilZouitine
2025-10-03 16:43:05 +02:00
parent f30da2dec1
commit dbce707db5
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+3
View File
@@ -439,6 +439,9 @@ if __name__ == "__main__":
# We set step_scheduler_with_optimizer False to prevent accelerate from # We set step_scheduler_with_optimizer False to prevent accelerate from
# adjusting the lr_scheduler steps based on the num_processes # adjusting the lr_scheduler steps based on the num_processes
accelerator = accelerate.Accelerator(step_scheduler_with_optimizer=False) accelerator = accelerate.Accelerator(step_scheduler_with_optimizer=False)
init_logging(accelerator=accelerator)
train(accelerator=accelerator) train(accelerator=accelerator)
else: else:
init_logging()
train() train()