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()