mirror of https://github.com/coqui-ai/TTS.git
small logging bug fix
This commit is contained in:
parent
64f2b95c31
commit
446cd6fa06
2
train.py
2
train.py
|
@ -258,7 +258,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
|
|||
" | > EPOCH END -- GlobalStep:{} AvgTotalLoss:{:.5f} "
|
||||
"AvgPostnetLoss:{:.5f} AvgDecoderLoss:{:.5f} "
|
||||
"AvgStopLoss:{:.5f} EpochTime:{:.2f} "
|
||||
"AvgStepTime:{:.2f}".format(global_step, avg_total_loss,
|
||||
"AvgStepTime:{:.2f} AvgLoaderTime:{:.2f}".format(global_step, avg_total_loss,
|
||||
avg_postnet_loss, avg_decoder_loss,
|
||||
avg_stop_loss, epoch_time, avg_step_time,
|
||||
avg_loader_time),
|
||||
|
|
Loading…
Reference in New Issue