From 83d775685638ba56960682e9a2a28301e4ac38ee Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 25 Apr 2018 08:25:55 -0700 Subject: [PATCH] against blocking stduot --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 1e1613c4..0c069607 100644 --- 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()