From fc099218df2f325490e5779b3b60b1126427afd8 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Wed, 6 Dec 2023 17:47:33 -0300 Subject: [PATCH 1/3] Fix aux 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 448fefc712..29e087ef06 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, From 9f325b1f6c6dee00cbaab1bfaaea22efdbde25ce Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Fri, 8 Dec 2023 08:37:28 +0100 Subject: [PATCH 2/3] fixup! Fix aux 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 29e087ef06..a32ad00f56 100644 --- 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): From 99fee6f5ad9cdb990a6dab864662800ddd7838a3 Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Fri, 8 Dec 2023 08:37:53 +0100 Subject: [PATCH 3/3] build: use Trainer>=0.0.36 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 1f7a44f6d8..23e8d2d013 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,7 +27,7 @@ pandas>=1.4,<2.0 # deps for training matplotlib>=3.7.0 # coqui stack -trainer>=0.0.32 +trainer>=0.0.36 # config management coqpit>=0.0.16 # chinese g2p deps