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

change lr_decay config field to noam_schedule

This commit is contained in:
root 2020-01-03 00:38:08 +01:00
parent 9d666af715
commit 6e09224697
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -602,7 +602,7 @@ def main(args): # pylint: disable=redefined-outer-name
if num_gpus > 1: if num_gpus > 1:
model = apply_gradient_allreduce(model) model = apply_gradient_allreduce(model)
if c.lr_decay: if c.noam_schedule:
scheduler = NoamLR(optimizer, scheduler = NoamLR(optimizer,
warmup_steps=c.warmup_steps, warmup_steps=c.warmup_steps,
last_epoch=args.restore_step - 1) last_epoch=args.restore_step - 1)