bug fix

2018-04-25 05:39:09 -07:00 · 2018-04-25 05:39:09 -07:00 · c82a17cfb2
parent 6426c7b818
commit c82a17cfb2
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -124,6 +124,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
                              linear_spec_var[:, :, :n_priority_freq],
                              mel_lengths_var)
        attention_loss = criterion(alignments, M, mel_lengths_var)
+        print(mk)
        loss = mel_loss + linear_loss + mk * attention_loss

        # backpass and check the grad norm