wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-05 16:20:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

larger lr for stop token

Browse Source
This commit is contained in:
Eren Golge
2018-05-14 19:04:29 -07:00
parent 02d72ccbfe
commit b2e09f9f58
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -1
View File
@@ -88,12 +88,13 @@ def train(model, criterion, criterion_st, data_loader, optimizer, optimizer_st,
# setup lr
current_lr = lr_decay(c.lr, current_step, c.warmup_steps)
current_lr_st = lr_decay(0.01, current_step, c.warmup_steps)
for params_group in optimizer.param_groups:
params_group['lr'] = current_lr
for params_group in optimizer_st.param_groups:
params_group['lr'] = current_lr
params_group['lr'] = current_lr_st
optimizer.zero_grad()
optimizer_st.zero_grad()