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

eval batch size added

This commit is contained in:
Eren Golge 2018-03-09 09:55:15 -08:00
parent cf3188d7da
commit 66bf60800a
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config.json
View File

@ -15,6 +15,7 @@
"lr": 0.00001875, "lr": 0.00001875,
"warmup_steps": 4000, "warmup_steps": 4000,
"batch_size": 1, "batch_size": 1,
"eval_batch_size": 32,
"r": 5, "r": 5,
"griffin_lim_iters": 60, "griffin_lim_iters": 60,

2
train.py
View File

@ -326,7 +326,7 @@ def main(args):
c.power c.power
) )
val_loader = DataLoader(val_dataset, batch_size=c.batch_size, val_loader = DataLoader(val_dataset, batch_size=c.eval_batch_size,
shuffle=False, collate_fn=val_dataset.collate_fn, shuffle=False, collate_fn=val_dataset.collate_fn,
drop_last=False, num_workers= 4, drop_last=False, num_workers= 4,
pin_memory=True) pin_memory=True)