Compare commits

..

2 Commits

Author SHA1 Message Date
Forkoz acecbe7ceb
Merge 6dda1a4f32 into 77d1d5a69c 2024-04-30 14:00:34 +08:00
pyp_l40 77d1d5a69c small fix 2024-04-29 08:47:32 -05:00
1 changed files with 0 additions and 1 deletions

View File

@ -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