From c4828b2b9e94fa70e5f8097aae405077fbb24aca Mon Sep 17 00:00:00 2001 From: SanjaESC Date: Mon, 13 Jul 2020 08:51:37 +0200 Subject: [PATCH] small gst config change --- tests/test_tacotron2_model.py | 2 +- utils/generic_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_tacotron2_model.py b/tests/test_tacotron2_model.py index c6d08160..92ffb9aa 100644 --- a/tests/test_tacotron2_model.py +++ b/tests/test_tacotron2_model.py @@ -239,4 +239,4 @@ class TacotronGSTTrainTest(unittest.TestCase): assert (param != param_ref).any( ), "param {} {} with shape {} not updated!! \n{}\n{}".format( name, count, param.shape, param, param_ref) - count += 1 \ No newline at end of file + count += 1 diff --git a/utils/generic_utils.py b/utils/generic_utils.py index 8b4b1f12..3bb99e08 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -359,8 +359,8 @@ def check_config(c): # GST _check_argument('use_gst', c, restricted=True, val_type=bool) - _check_argument('gst_style_input', c, restricted=True, val_type=str) _check_argument('gst', c, restricted=True, val_type=dict) + _check_argument('gst_style_input', c['gst'], restricted=True, val_type=str) _check_argument('gst_embedding_dim', c['gst'], restricted=True, val_type=int, min_val=1) _check_argument('gst_num_heads', c['gst'], restricted=True, val_type=int, min_val=1) _check_argument('gst_style_tokens', c['gst'], restricted=True, val_type=int, min_val=1)