From 12cd959ff823854b8d5426508a824bc872733ad6 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 26 Apr 2018 05:13:51 -0700 Subject: [PATCH] manual seeding --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index f63783fb..fb0c47fc 100644 --- a/train.py +++ b/train.py @@ -25,6 +25,7 @@ from utils.visual import plot_alignment, plot_spectrogram from models.tacotron import Tacotron from layers.losses import L1LossMasked +torch.manual_seed(1) use_cuda = torch.cuda.is_available() @@ -217,7 +218,7 @@ def evaluate(model, criterion, data_loader, current_step): avg_linear_loss = 0 avg_mel_loss = 0 - print(" | > Validation") + print("\n | > Validation") progbar = Progbar(len(data_loader.dataset) / c.batch_size) n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq) with torch.no_grad():