Merge pull request #228 from twerkmeister/patch-1

use min and max seq length during eval
2019-07-17 12:22:04 +02:00 · 2019-07-17 12:22:04 +02:00 · 1c15378939
parent b7036e458d 721345f1f2
commit 1c15378939
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -63,8 +63,8 @@ def setup_loader(is_val=False, verbose=False):
            meta_data=meta_data_eval if is_val else meta_data_train,
            ap=ap,
            batch_group_size=0 if is_val else c.batch_group_size * c.batch_size,
-            min_seq_len=0 if is_val else c.min_seq_len,
-            max_seq_len=float("inf") if is_val else c.max_seq_len,
+            min_seq_len=c.min_seq_len,
+            max_seq_len=c.max_seq_len,
            phoneme_cache_path=c.phoneme_cache_path,
            use_phonemes=c.use_phonemes,
            phoneme_language=c.phoneme_language,