guided attn #4

2018-04-24 11:30:58 -07:00 · 2018-04-24 11:30:58 -07:00 · 973e18a86c
parent 1458f1517c
commit 973e18a86c
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -115,7 +115,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
                M[n, t] = val
        e_x = np.exp(M - np.max(M))
        M = e_x / e_x.sum(axis=0) # only difference
-        M = Variable(M)
+        M = Variable(torch.FloatTensor(M))
        M = torch.stack([M]*32)

        # forward pass