Fix wandb logging (#730)

2025-02-14 18:00:12 +01:00
parent c574eb4984
commit b86a2c0b47
1 changed files with 3 additions and 1 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -232,7 +232,9 @@ def train(cfg: TrainPipelineConfig):
        if is_log_step:
            logging.info(train_tracker)
            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)
            train_tracker.reset_averages()