mirror of
https://github.com/wassname/TTS.git
synced 2026-10-30 13:00:27 +08:00
Revert drop_last
This commit is contained in:
1 parent
7dfd753d91
commit
1e219fef0a
1 file changed
+1
-1
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user