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,