Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "7d8522a29f3ea07d921fb54e4f4b723fc108ac6d"
Unverified Commit 1feb8917 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

mark tensor not differentiable (#377)

parent 22c47ac7
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
mmdet3d/ops/interpolate/three_nn.py
View file @ 1feb8917
...@@ -32,6 +32,9 @@ class ThreeNN(Function): ...@@ -32,6 +32,9 @@ class ThreeNN(Function):
idx = torch.cuda.IntTensor(B, N, 3) idx = torch.cuda.IntTensor(B, N, 3)
interpolate_ext.three_nn_wrapper(B, N, m, target, source, dist2, idx) interpolate_ext.three_nn_wrapper(B, N, m, target, source, dist2, idx)
ctx.mark_non_differentiable(idx)
return torch.sqrt(dist2), idx return torch.sqrt(dist2), idx
@staticmethod @staticmethod
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment