storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-29 09:18:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding loss values

Browse Source
This commit is contained in:
arensc
2024-03-08 19:50:21 -05:00
parent 10749503d5
commit 42a881a3f7
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/StyleTTS2Train/train_finetune_accelerate.py
+4 -1
View File
@@ -613,6 +613,9 @@ def main(config_path):
'train/d_loss_slm': float(d_loss_slm),
'train/gen_loss_slm': float(loss_gen_lm),
'epoch': int(epoch) + 1}, step=iters)
running_loss = 0
@@ -738,7 +741,7 @@ def main(config_path):
# writer.add_scalar('eval/mel_loss', loss_test / iters_test, epoch + 1)
# writer.add_scalar('eval/dur_loss', loss_test / iters_test, epoch + 1)
# writer.add_scalar('eval/F0_loss', loss_f / iters_test, epoch + 1)
harness.log_eval(mel_loss=float(loss_test / iters_test),dur_loss=float(loss_test / iters_test),F0_loss=float(loss_f / iters_test),epoch)
accelerator.log({'eval/mel_loss': float(loss_test / iters_test),
'eval/dur_loss': float(loss_test / iters_test),
'eval/F0_loss': float(loss_f / iters_test)}, step=epoch + 1)