diff --git a/TTS/tts/models/base_tts.py b/TTS/tts/models/base_tts.py index ca3c4a28..becf22f9 100644 --- a/TTS/tts/models/base_tts.py +++ b/TTS/tts/models/base_tts.py @@ -326,7 +326,7 @@ class BaseTTS(BaseModel): batch_size=config.eval_batch_size if is_eval else config.batch_size, shuffle=False, # shuffle is done in the dataset. collate_fn=dataset.collate_fn, - drop_last=True, # setting this False might cause issues in AMP training. + drop_last=False, # setting this False might cause issues in AMP training. sampler=sampler, num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers, pin_memory=False,