Set `drop_last`

2022-01-28 13:50:58 +01:00 · 2022-01-28 13:50:58 +01:00 · 38314194e7
parent 846e0e4284
commit 38314194e7
1 changed files with 2 additions and 2 deletions
--- a/TTS/tts/models/base_tts.py
+++ b/TTS/tts/models/base_tts.py
@ -324,9 +324,9 @@ class BaseTTS(BaseModel):
            loader = DataLoader(
                dataset,
                batch_size=config.eval_batch_size if is_eval else config.batch_size,
-                shuffle=False,
+                shuffle=False,      # shuffle is done in the dataset.
                collate_fn=dataset.collate_fn,
-                drop_last=False,
+                drop_last=True,     # 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,