Merge pull request #55 from hqucms/patch-1

Call contiguous() after permute() in the pytorch model
WangYueFt · Apr 13, 2020 · 20fdb45 · 20fdb45
2 parents 2f31857 + 98b744a
commit 20fdb45
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, x), dim=3).permute(0, 3, 1, 2)
+    feature = torch.cat((feature-x, x), dim=3).permute(0, 3, 1, 2).contiguous()
 
     return feature