From 96da10415e53014339039093aebe4149f20f6b1a Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 30 May 2022 16:48:18 -0600 Subject: [PATCH] facepalm --- codes/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/train.py b/codes/train.py index 5acf30fe..4717dc68 100644 --- a/codes/train.py +++ b/codes/train.py @@ -245,9 +245,9 @@ class Trainer: else: wandb_logs[k] = v if opt_get(opt, ['wandb_progress_use_raw_steps'], False): - wandb.log(logs, step=self.current_step) + wandb.log(wandb_logs, step=self.current_step) else: - wandb.log(logs, step=self.total_training_data_encountered) + wandb.log(wandb_logs, step=self.total_training_data_encountered) self.logger.info(message) #### save models and training states