fix bug in gradient computation

This commit is contained in:
Mohamed Hassan 2019-12-10 18:21:36 +01:00
parent 9b6ab62310
commit 3b3552e373

View File

@ -254,7 +254,7 @@ class DistanceFieldPenetrationLoss(nn.Module):
recv_triangles = triangles[batch_idxs, receiver_faces] recv_triangles = triangles[batch_idxs, receiver_faces]
intr_triangles = triangles[batch_idxs, intruder_faces] intr_triangles = triangles[batch_idxs, intruder_faces]
with torch.no_grad():
recv_normals = normals[batch_idxs, receiver_faces] recv_normals = normals[batch_idxs, receiver_faces]
recv_circumradius = circumradius[batch_idxs, receiver_faces] recv_circumradius = circumradius[batch_idxs, receiver_faces]
recv_circumcenter = circumcenter[batch_idxs, receiver_faces] recv_circumcenter = circumcenter[batch_idxs, receiver_faces]