From 721345f1f24890d1aa5264759b65f09025c558e0 Mon Sep 17 00:00:00 2001 From: Thomas Werkmeister Date: Wed, 17 Jul 2019 12:11:16 +0200 Subject: [PATCH] use min and max seq length during eval --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index ca40986e..56b81d8b 100644 --- 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,