Fix aux unit tests

2023-12-06 17:47:33 -03:00 · 2023-12-06 17:47:33 -03:00 · 32e777ae7a
parent 75feba06fc
commit 32e777ae7a
1 changed files with 1 additions and 1 deletions
--- 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,