fix bug in gradient computation

sssphil · Dec 10, 2019 · 3b3552e · 3b3552e
1 parent 9b6ab62
commit 3b3552e
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/mesh_intersection/loss.py b/mesh_intersection/loss.py
@@ -254,14 +254,14 @@ def forward(self, triangles, collision_idxs):
 
         recv_triangles = triangles[batch_idxs, receiver_faces]
         intr_triangles = triangles[batch_idxs, intruder_faces]
-        with torch.no_grad():
-            recv_normals = normals[batch_idxs, receiver_faces]
-            recv_circumradius = circumradius[batch_idxs, receiver_faces]
-            recv_circumcenter = circumcenter[batch_idxs, receiver_faces]
-
-            intr_normals = normals[batch_idxs, intruder_faces]
-            intr_circumradius = circumradius[batch_idxs, intruder_faces]
-            intr_circumcenter = circumcenter[batch_idxs, intruder_faces]
+
+        recv_normals = normals[batch_idxs, receiver_faces]
+        recv_circumradius = circumradius[batch_idxs, receiver_faces]
+        recv_circumcenter = circumcenter[batch_idxs, receiver_faces]
+
+        intr_normals = normals[batch_idxs, intruder_faces]
+        intr_circumradius = circumradius[batch_idxs, intruder_faces]
+        intr_circumcenter = circumcenter[batch_idxs, intruder_faces]
 
         # Compute the distance field for the intruding triangles
         # B x NUM_COLLISIONS x 3