reduce TB load for step stats

2019-08-22 15:28:11 +02:00 · 2019-08-22 15:28:11 +02:00 · 0a717faf1c
parent 549dbba128
commit 0a717faf1c
1 changed files with 9 additions and 7 deletions
--- a/train.py
+++ b/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,