423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 22:30:35 +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
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_raymarching.py
View File

@@ -163,7 +163,7 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
loss = features_render.mean()
loss.backward()
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):
"""
@@ -193,4 +193,4 @@ class TestRaymarching(TestCaseMixin, unittest.TestCase):
# check differentiability
loss = opacities_render.mean()
loss.backward()
self.assertTrue(rays_densities.grad.data.isfinite().all())
self.assertTrue(torch.isfinite(rays_densities.grad.data).all())