Merge 9fb6d948d0 into 77d1d5a69c

small fix
2024-04-29 16:08:19 +02:00 · 2024-04-29 08:47:32 -05:00
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
Author	SHA1	Message	Date
Pranay Gosar	49524200f6	Merge `9fb6d948d0` into `77d1d5a69c`	2024-04-29 16:08:19 +02:00
pyp_l40	77d1d5a69c	small fix	2024-04-29 08:47:32 -05:00