Fix aux unit tests

2023-12-11 17:27:59 -03:00 · 2023-12-11 17:27:59 -03:00 · e741cab78e
parent 90cb358424
commit e741cab78e
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(
-                {"train_loss": None, "eval_loss": eval_loss},
+                eval_loss,
                best_loss,
                c,
                model,