mirror of https://github.com/coqui-ai/TTS.git
make attn guiding optional #4
This commit is contained in:
parent
3c8ded5a18
commit
9cf1df9219
2
train.py
2
train.py
|
@ -148,7 +148,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
|
||||||
progbar_display['grad_norm'] = grad_norm
|
progbar_display['grad_norm'] = grad_norm
|
||||||
|
|
||||||
# update
|
# update
|
||||||
progbar.update(num_iter+1, values=[tuple(progbar_display.iteritems())])
|
progbar.update(num_iter+1, values=[progbar_display.items()])
|
||||||
avg_linear_loss += linear_loss.data[0]
|
avg_linear_loss += linear_loss.data[0]
|
||||||
avg_mel_loss += mel_loss.data[0]
|
avg_mel_loss += mel_loss.data[0]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue