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

Use old style isfinite

Browse Source 
Summary: Avoid using the newish member function isfinite. We use torch.isfinite instead for torch 1.4.0 compatibility.

Reviewed By: nikhilaravi

Differential Revision: D26946672

fbshipit-source-id: 853c3716f40061152f1ea54a39eb60b565de7c2c
This commit is contained in:
Jeremy Reizenstein 2021-03-11 03:11:41 -08:00 committed by Facebook GitHub Bot
parent 4bb3fff52b
commit ff9c6612b4
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_points_to_volumes.py
View File

@ -354,7 +354,7 @@ class TestPointsToVolumes(TestCaseMixin, unittest.TestCase):
# this does not produce grads w.r.t. xyz # this does not produce grads w.r.t. xyz
self.assertIsNone(field.grad) self.assertIsNone(field.grad)
else: else:
self.assertTrue(field.grad.data.isfinite().all()) self.assertTrue(torch.isfinite(field.grad.data).all())
def _check_volume_slice_color_density( def _check_volume_slice_color_density(
self, V, split_dim, interp_mode, clr_gt, slice_type, border=3 self, V, split_dim, interp_mode, clr_gt, slice_type, border=3

4
tests/test_raymarching.py
View File

@ -163,7 +163,7 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
loss = features_render.mean() loss = features_render.mean()
loss.backward() loss.backward()
for field in (rays_densities, rays_features): for field in (rays_densities, rays_features):
self.assertTrue(field.grad.data.isfinite().all()) self.assertTrue(torch.isfinite(field.grad.data).all())
def test_absorption_only(self): def test_absorption_only(self):
""" """
@ -193,4 +193,4 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
# check differentiability # check differentiability
loss = opacities_render.mean() loss = opacities_render.mean()
loss.backward() loss.backward()
self.assertTrue(rays_densities.grad.data.isfinite().all()) self.assertTrue(torch.isfinite(rays_densities.grad.data).all())