wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix wandb logging (#730)

This commit is contained in:
Simon Alibert 2025-02-14 18:00:12 +01:00 committed by GitHub
parent c574eb4984
commit b86a2c0b47
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/train.py
View File

@ -232,7 +232,9 @@ def train(cfg: TrainPipelineConfig):
if is_log_step: if is_log_step:
logging.info(train_tracker) logging.info(train_tracker)
if wandb_logger: if wandb_logger:
wandb_log_dict = {**train_tracker.to_dict(), **output_dict} wandb_log_dict = train_tracker.to_dict()
if output_dict:
wandb_log_dict.update(output_dict)
wandb_logger.log_dict(wandb_log_dict, step) wandb_logger.log_dict(wandb_log_dict, step)
train_tracker.reset_averages() train_tracker.reset_averages()