From ca28e05ed71cea7462d9a4517a121edabf900239 Mon Sep 17 00:00:00 2001 From: Alexander Korolev Date: Wed, 27 Jan 2021 16:33:25 +0100 Subject: [PATCH] update fixed stopnet_pos_weight parameter config parameter c.stopnet_pos_weight has currently no effect as it is not used. --- TTS/bin/train_tacotron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/bin/train_tacotron.py b/TTS/bin/train_tacotron.py index ccb35a7c..be609905 100644 --- a/TTS/bin/train_tacotron.py +++ b/TTS/bin/train_tacotron.py @@ -534,7 +534,7 @@ def main(args): # pylint: disable=redefined-outer-name optimizer_st = None # setup criterion - criterion = TacotronLoss(c, stopnet_pos_weight=10.0, ga_sigma=0.4) + criterion = TacotronLoss(c, stopnet_pos_weight=c.stopnet_pos_weight, ga_sigma=0.4) if args.restore_path: checkpoint = torch.load(args.restore_path, map_location='cpu')