Bug fixe on unit tests

2023-12-26 09:30:27 -03:00 · 2023-12-26 09:30:27 -03:00 · 62b7547033
parent e741cab78e
commit 62b7547033
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,