From 59f26d0f6da351372735349362a11505b8ade9af Mon Sep 17 00:00:00 2001
From: Antonios Sarikas <antonios.sarikas@gmail.com>
Date: Sat, 4 Jan 2025 03:13:47 +0200
Subject: [PATCH] refactor(PointNetBackbone): remove unnecessary `clone`

---
 src/aidsorb/modules.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/aidsorb/modules.py b/src/aidsorb/modules.py
index db19064..9f78007 100644
--- 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).