Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert drop_last

This commit is contained in:
Eren Gölge 2022-02-05 20:34:39 +01:00
parent 7dfd753d91
commit 1e219fef0a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/models/base_tts.py
View File

@ -326,7 +326,7 @@ class BaseTTS(BaseModel):
batch_size=config.eval_batch_size if is_eval else config.batch_size, batch_size=config.eval_batch_size if is_eval else config.batch_size,
shuffle=False, # shuffle is done in the dataset. shuffle=False, # shuffle is done in the dataset.
collate_fn=dataset.collate_fn, 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, sampler=sampler,
num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers, num_workers=config.num_eval_loader_workers if is_eval else config.num_loader_workers,
pin_memory=False, pin_memory=False,