diff --git a/train.py b/train.py index 458845b2..cb792700 100644 --- a/train.py +++ b/train.py @@ -64,8 +64,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,