Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

reduce TB load for step stats

This commit is contained in:
Eren Golge 2019-08-22 15:28:11 +02:00
parent 549dbba128
commit 0a717faf1c
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -212,6 +212,8 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
avg_loader_time += loader_time avg_loader_time += loader_time
# Plot Training Iter Stats # Plot Training Iter Stats
# reduce TB load
if global_step % 10 == 0:
iter_stats = {"loss_posnet": postnet_loss.item(), iter_stats = {"loss_posnet": postnet_loss.item(),
"loss_decoder": decoder_loss.item(), "loss_decoder": decoder_loss.item(),
"lr": current_lr, "lr": current_lr,