Bug fixe on unit tests

coqui-ai · Dec 26, 2023 · 62b7547 · 62b7547
1 parent e741cab
commit 62b7547
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/bin/train_encoder.py b/TTS/bin/train_encoder.py
@@ -248,7 +248,7 @@ def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader,
             )
             # save the best checkpoint
             best_loss = save_best_model(
-                eval_loss,
+                {"train_loss": None, "eval_loss": eval_loss},
                 best_loss,
                 c,
                 model,