refactor(PointNetBackbone): remove unnecessary clone

adosar · Jan 4, 2025 · 59f26d0 · 59f26d0
1 parent 8cdf4c6
commit 59f26d0
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/aidsorb/modules.py b/src/aidsorb/modules.py
@@ -317,10 +317,7 @@ def forward(self, x):
         n_points = x.shape[2]
 
         x = self.shared_mlp_1(x)
-
-        if self.local_feats:
-            point_feats = x.clone()
-
+        point_feats = x
         x = self.shared_mlp_2(x)
 
         # Shape (B, n_global_feats).