Adding loss values

2026-10-28 09:08:54 +00:00 · 2024-03-08 19:50:21 -05:00
parent 10749503d5
commit 42a881a3f7
2 changed files with 13 additions and 2 deletions
@@ -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)})