small fix

2024-04-29 08:47:32 -05:00 · 2024-04-29 08:47:32 -05:00 · 77d1d5a69c
parent da6d34e26e
commit 77d1d5a69c
1 changed files with 0 additions and 1 deletions
--- 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