From ee00b3db5c75a1f7cd79f94f182699a75080cd81 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 24 Apr 2018 11:36:56 -0700 Subject: [PATCH] guided attn #9 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 08a1f5a2..fcd4d92f 100644 --- a/train.py +++ b/train.py @@ -106,7 +106,7 @@ def train(model, criterion, data_loader, optimizer, epoch): # create attention mask # TODO: vectorize - print(tet_input_var.shape) + print(text_input_var.shape) print(mel_spec_var.shape) N = text_input_var.shape[1] T = mel_spec_var.shape[1]