Merge pull request #33 from gsp-27/patch-1

Update pointnet.py
fxia22 · Feb 28, 2019 · 610f826 · 610f826
2 parents 2ec315f + 1551005
commit 610f826
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pointnet.py b/pointnet.py
@@ -101,7 +101,7 @@ def forward(self, x):
         x = F.relu(self.bn1(self.fc1(x)))
         x = F.relu(self.bn2(self.fc2(x)))
         x = self.fc3(x)
-        return F.log_softmax(x, dim=0), trans
+        return F.log_softmax(x, dim=1), trans
 
 class PointNetDenseCls(nn.Module):
     def __init__(self, k = 2):