423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 20:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Looser gradient check in test_rasterize_meshes

Browse Source 
Summary: This has been failing intermittently

Reviewed By: nikhilaravi

Differential Revision: D21403157

fbshipit-source-id: 51b74d6c813b52effe72d14b565e250fcabbb463
This commit is contained in:
Jeremy Reizenstein 2020-05-05 09:24:24 -07:00 committed by Facebook GitHub Bot
parent 06ad1fb6c4
commit 8fc28baa27
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch3d/csrc/knn/knn_cpu.cpp
View File

@ -13,7 +13,6 @@ std::tuple<at::Tensor, at::Tensor> KNearestNeighborIdxCpu(
const int N = p1.size(0); const int N = p1.size(0);
const int P1 = p1.size(1); const int P1 = p1.size(1);
const int D = p1.size(2); const int D = p1.size(2);
const int P2 = p2.size(1);
auto long_opts = p1.options().dtype(torch::kInt64); auto long_opts = p1.options().dtype(torch::kInt64);
torch::Tensor idxs = torch::full({N, P1, K}, 0, long_opts); torch::Tensor idxs = torch::full({N, P1, K}, 0, long_opts);

2
tests/test_rasterize_meshes.py
View File

@ -507,7 +507,7 @@ class TestRasterizeMeshes(TestCaseMixin, unittest.TestCase):
grad_var1.grad.data.zero_() grad_var1.grad.data.zero_()
loss2.backward() loss2.backward()
grad_verts2 = grad_var2.grad.data.clone().cpu() grad_verts2 = grad_var2.grad.data.clone().cpu()
self.assertClose(grad_verts1, grad_verts2, rtol=1e-3) self.assertClose(grad_verts1, grad_verts2, rtol=2e-3)
def _test_perspective_correct(self, rasterize_meshes_fn, device, bin_size=None): def _test_perspective_correct(self, rasterize_meshes_fn, device, bin_size=None):
# fmt: off # fmt: off