From a3b1db98e922f99146d757e9e179a83824cbcea7 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 30 Apr 2018 05:07:41 -0700 Subject: [PATCH] stop targets --- train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train.py b/train.py index d98fc26d..151cbbeb 100644 --- a/train.py +++ b/train.py @@ -81,6 +81,7 @@ def train(model, criterion, data_loader, optimizer, epoch): linear_spec = data[2] mel_spec = data[3] mel_lengths = data[4] + stop_target = data[5] current_step = num_iter + args.restore_step + \ epoch * len(data_loader) + 1 @@ -225,6 +226,7 @@ def evaluate(model, criterion, data_loader, current_step): linear_spec = data[2] mel_spec = data[3] mel_lengths = data[4] + stop_target = data[5] # dispatch data to GPU if use_cuda: