fix error

WangYueFt · Jun 28, 2019 · 72375d3 · 72375d3
1 parent 2355c94
commit 72375d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch/model.py b/pytorch/model.py
@@ -48,7 +48,7 @@ def get_graph_feature(x, k=20, idx=None):
     feature = feature.view(batch_size, num_points, k, num_dims) 
     x = x.view(batch_size, num_points, 1, num_dims).repeat(1, 1, k, 1)
 
-    feature = torch.cat((feature, x), dim=3).permute(0, 3, 1, 2)
+    feature = torch.cat((feature-x, x), dim=3).permute(0, 3, 1, 2)
 
     return feature