[v0.2.0] Fixed bugs in vocoder validation

dunky11 · Jun 11, 2022 · 237bd26 · 237bd26
1 parent 31cb02b
commit 237bd26
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/backend/voice_smith/vocoder_training.py b/backend/voice_smith/vocoder_training.py
@@ -196,7 +196,12 @@ def evaluate(
     preprocess_config: PreprocessingConfig,
 ):
     generator.eval()
-    loss_means = {"mel_loss": 0, "pesq": 0, "estoi": 0, "rmse": 0}
+    loss_means = {
+        "mel_loss": torch.FloatTensor([0.0]).to(device),
+        "pesq": torch.FloatTensor([0.0]).to(device),
+        "estoi": torch.FloatTensor([0.0]).to(device),
+        "rmse": torch.FloatTensor([0.0]).to(device),
+    }
     len_group = 0
     len_group_pesq = 0
     batch_size = 0