wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-27 12:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

small gst config change

Browse Source
This commit is contained in:
SanjaESC authored and erogol committed 2020-08-05 19:19:07 +02:00
1 parent 1436206224
commit c4828b2b9e
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/test_tacotron2_model.py
+1 -1
View File
@@ -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
count += 1
utils/generic_utils.py
+1 -1
View File
@@ -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)