Update models.py

Mamengru6688 · Aug 2, 2018 · 8e42cae · 8e42cae
1 parent 8d8e0bd
commit 8e42cae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygcn/models.py b/pygcn/models.py
@@ -15,4 +15,4 @@ def forward(self, x, adj):
         x = F.relu(self.gc1(x, adj))
         x = F.dropout(x, self.dropout, training=self.training)
         x = self.gc2(x, adj)
-        return F.log_softmax(x)
+        return F.log_softmax(x, dim =1)