against blocking stduot

2018-04-25 08:25:55 -07:00 · 2018-04-25 08:25:55 -07:00 · 83d7756856
parent c562fac7f4
commit 83d7756856
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -258,6 +258,7 @@ def evaluate(model, criterion, data_loader, current_step):
                                               ('linear_loss',
                                                linear_loss.item()),
                                               ('mel_loss', mel_loss.item())])
+            sys.stdout.flush()

            avg_linear_loss += linear_loss.item()
            avg_mel_loss += mel_loss.item()