rixty/VoiceCraft
1
0
Fork 0
mirror of https://github.com/jasonppy/VoiceCraft.git synced 2025-06-05 21:49:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
pyp_l40
2024-04-29 08:47:32 -05:00
parent da6d34e26e
commit 77d1d5a69c
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
steps/trainer.py
View File

@@ -78,7 +78,6 @@ class Trainer:
if self.rank == 0 and self.progress['step'] % self.args.tb_write_every_n_steps == 0: 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.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']))) all_inds = list(range(len(batch['y'])))
sum_losses = 0 sum_losses = 0