mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
73c3ab546d
commit
03e1c06f5b
2
train.py
2
train.py
|
@ -116,7 +116,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, optimizer_st,
|
||||||
"GradNormST:{:.5f} StepTime:{:.2f}".format(
|
"GradNormST:{:.5f} StepTime:{:.2f}".format(
|
||||||
num_iter, batch_n_iter, current_step, loss.item(),
|
num_iter, batch_n_iter, current_step, loss.item(),
|
||||||
linear_loss.item(), mel_loss.item(), stop_loss.item(),
|
linear_loss.item(), mel_loss.item(), stop_loss.item(),
|
||||||
grad_norm.item(), grad_norm_st.item(), step_time),
|
grad_norm, grad_norm_st, step_time),
|
||||||
flush=True)
|
flush=True)
|
||||||
|
|
||||||
avg_linear_loss += linear_loss.item()
|
avg_linear_loss += linear_loss.item()
|
||||||
|
|
Loading…
Reference in New Issue