guided attn #9

2026-09-11 12:00:24 +08:00 · 2018-04-24 11:36:56 -07:00
parent 80c18684c6
commit 8c4e03cebf
1 changed files with 1 additions and 1 deletions
@@ -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]