mirror of https://github.com/coqui-ai/TTS.git
reduce TB load for step stats
This commit is contained in:
parent
549dbba128
commit
0a717faf1c
2
train.py
2
train.py
|
@ -212,6 +212,8 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
|
|||
avg_loader_time += loader_time
|
||||
|
||||
# Plot Training Iter Stats
|
||||
# reduce TB load
|
||||
if global_step % 10 == 0:
|
||||
iter_stats = {"loss_posnet": postnet_loss.item(),
|
||||
"loss_decoder": decoder_loss.item(),
|
||||
"lr": current_lr,
|
||||
|
|
Loading…
Reference in New Issue