Add keepdim=True for PyTorch version upgrade to 0.2.0

Author: pplantinga

model.py

@@ -48,7 +48,7 @@ def __init__(self, embedding_tokens):
     def forward(self, v, q, q_len):
         q = self.text(q, list(q_len.data))
 
-        v = v / (v.norm(p=2, dim=1).expand_as(v) + 1e-8)
+        v = v / (v.norm(p=2, dim=1, keepdim=True).expand_as(v) + 1e-8)
         a = self.attention(v, q)
         v = apply_attention(v, a)
 

utils.py

@@ -10,7 +10,7 @@
 
 def batch_accuracy(predicted, true):
     """ Compute the accuracies for a batch of predictions and answers """
-    _, predicted_index = predicted.max(dim=1)
+    _, predicted_index = predicted.max(dim=1, keepdim=True)
     agreeing = true.gather(dim=1, index=predicted_index)
     '''
     Acc needs to be averaged over all 10 choose 9 subsets of human answers.