From da49089a7246bf75c82b05afa8038d672268997e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 12 May 2021 10:12:11 +0200 Subject: [PATCH] update melgan training test batch size --- TTS/speaker_encoder/losses.py | 4 ++-- tests/vocoder_tests/test_melgan_train.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/TTS/speaker_encoder/losses.py b/TTS/speaker_encoder/losses.py index 69264ab4..1b6b6e43 100644 --- a/TTS/speaker_encoder/losses.py +++ b/TTS/speaker_encoder/losses.py @@ -23,7 +23,7 @@ class GE2ELoss(nn.Module): self.b = nn.Parameter(torch.tensor(init_b)) self.loss_method = loss_method - print(" > Initialised Generalized End-to-End loss") + print(" > Initialized Generalized End-to-End loss") assert self.loss_method in ["softmax", "contrast"] @@ -136,7 +136,7 @@ class AngleProtoLoss(nn.Module): self.b = nn.Parameter(torch.tensor(init_b)) self.criterion = torch.nn.CrossEntropyLoss() - print(" > Initialised Angular Prototypical loss") + print(" > Initialized Angular Prototypical loss") def forward(self, x): """ diff --git a/tests/vocoder_tests/test_melgan_train.py b/tests/vocoder_tests/test_melgan_train.py index b9e3be7f..de48ca24 100644 --- a/tests/vocoder_tests/test_melgan_train.py +++ b/tests/vocoder_tests/test_melgan_train.py @@ -9,14 +9,14 @@ config_path = os.path.join(get_tests_output_path(), "test_vocoder_config.json") output_path = os.path.join(get_tests_output_path(), "train_outputs") config = MelganConfig( - batch_size=8, - eval_batch_size=8, + batch_size=4, + eval_batch_size=4, num_loader_workers=0, num_val_loader_workers=0, run_eval=True, test_delay_epochs=-1, epochs=1, - seq_len=8192, + seq_len=2048, eval_split_size=1, print_step=1, print_eval=True,