Set drop_last

2026-09-09 11:16:00 +08:00 · 2022-02-25 11:26:59 +01:00
parent 846e0e4284
commit 38314194e7
1 changed files with 2 additions and 2 deletions
@@ -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,