guided attn #11

2018-04-24 11:43:24 -07:00 · 2018-04-24 11:43:24 -07:00 · a42669327e
parent 3161a16101
commit a42669327e
1 changed files with 0 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -128,8 +128,6 @@ def train(model, criterion, data_loader, optimizer, epoch):
            + 0.5 * criterion(linear_output[:, :, :n_priority_freq],
                              linear_spec_var[:, :, :n_priority_freq],
                              mel_lengths_var)
-        print(M.shape)
-        print(alignments.shape)
        attention_loss = criterion(alignments, M, mel_lengths_var)
        loss = mel_loss + linear_loss + 0.2 * attention_loss