Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "9bdaefc8a685389c26b9775e219a785ba5d7e1cc"
Unverified Commit 1feb8917 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

mark tensor not differentiable (#377)

parent 22c47ac7
Hide 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