small gst config change

This commit is contained in:
SanjaESC
2020-08-05 19:19:07 +02:00
committed by erogol
parent 1436206224
commit c4828b2b9e
2 changed files with 2 additions and 2 deletions
+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)