Apply bugfix from B1ueber2y#4

rakeshshrestha31 · Feb 18, 2022 · 708d545 · 708d545
1 parent 57d8424
commit 708d545
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/utils/decoder_utils.py b/core/utils/decoder_utils.py
@@ -81,7 +81,8 @@ def decode_sdf_gradient(decoder, latent_vector, points, clamp_dist=0.1, MAX_POIN
         points_batch = points[start:end]
         sdf = decode_sdf(decoder, latent_vector, points_batch, clamp_dist=clamp_dist)
         start = end
-        grad_tensor = torch.autograd.grad(outputs=sdf, inputs=points_batch, grad_outputs=torch.ones_like(points_batch), create_graph=True, retain_graph=True)
+        # grad_tensor = torch.autograd.grad(outputs=sdf, inputs=points_batch, grad_outputs=torch.ones_like(points_batch), create_graph=True, retain_graph=True)
+        grad_tensor = torch.autograd.grad(outputs=sdf, inputs=points_batch, grad_outputs=torch.ones_like(sdf), create_graph=True, retain_graph=True)
         grad_tensor = grad_tensor[0]
         if no_grad:
             grad_tensor = grad_tensor.detach()