From 62b7547033b678fb4624b6f9719fc46f2d5d0cbe Mon Sep 17 00:00:00 2001
From: Edresson Casanova <edresson1@gmail.com>
Date: Tue, 26 Dec 2023 09:30:27 -0300
Subject: [PATCH] Bug fixe on unit tests

---
 TTS/bin/train_encoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TTS/bin/train_encoder.py b/TTS/bin/train_encoder.py
index 1340a02a..a32ad00f 100644
--- 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,