fixup! Fix aux unit tests

2023-12-08 08:37:28 +01:00 · 2023-12-08 08:37:28 +01:00 · 9f325b1f6c
parent fc099218df
commit 9f325b1f6c
1 changed files with 1 additions and 1 deletions
--- a/TTS/bin/train_encoder.py
+++ b/TTS/bin/train_encoder.py
@ -125,7 +125,7 @@ def evaluation(model, criterion, data_loader, global_step):

 def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader, global_step):
    model.train()
-    best_loss = float("inf")
+    best_loss = {"train_loss": None, "eval_loss": float("inf")}
    avg_loader_time = 0
    end_time = time.time()
    for epoch in range(c.epochs):