storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-09 00:09:55 +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/test_harness.py
+9 -1
View File
@@ -48,7 +48,15 @@ class TestHarness:
'step':step,
'total_steps':train_list_len//batch_size
})
def log_eval(self,mel_loss,dur_loss,F0_loss,epoch):
wandb.log({
'eval/mel_loss':mel_loss,
'eval/dur_loss':dur_loss,
'eval/F0_loss':F0_loss,
'epoch':epoch+1
})
def sample_ood_voice(self,file:str="testing.wav",sample_rate=24000,character:str="character_name",caption:str="What is supposed to be said"):
wandb.log({character: wandb.Audio(file, sample_rate=sample_rate, caption=caption)})
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)