@?

Jodiechou · Dec 17, 2017 · 0774ead · 0774ead
1 parent a28dd92
commit 0774ead
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/memn2n/model.py b/memn2n/model.py
@@ -81,6 +81,9 @@ def forward(self, story, query):
 
             u_k = u[-1] + o_k
             u.append(u_k)
-
+
+        # print "transpose",self.C[self.max_hops].weight.transpose(0, 1).size()
+        # print "u-1",u[-1].size()
+        # a_hat = torch.matmul(u[-1],self.C[self.max_hops].weight.transpose(0, 1))
         a_hat = u[-1]@self.C[self.max_hops].weight.transpose(0, 1)
         return a_hat, self.softmax(a_hat)