From 77d1d5a69c0e81ba3fb7585599061aa913b11a8e Mon Sep 17 00:00:00 2001 From: pyp_l40 Date: Mon, 29 Apr 2024 08:47:32 -0500 Subject: [PATCH] small fix --- steps/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/steps/trainer.py b/steps/trainer.py index b312258..7c2513f 100644 --- a/steps/trainer.py +++ b/steps/trainer.py @@ -78,7 +78,6 @@ class Trainer: if self.rank == 0 and self.progress['step'] % self.args.tb_write_every_n_steps == 0: self.writer.add_scalar("train/lr", cur_lr, self.progress['step']) - self.wandb.log({"train/lr": cur_lr}, step=self.progress['step']) all_inds = list(range(len(batch['y']))) sum_losses = 0