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

smaller data in specular test

Browse Source 
Summary: Reduce the size of the data in this test, so that on circleci it doesn't run out of memory when pytorch (1.6) is used.

Reviewed By: gkioxari

Differential Revision: D22801490

fbshipit-source-id: 9591253c3d47430facd769a2c51a0b1722e0a305
This commit is contained in:
Jeremy Reizenstein 2020-07-29 04:59:47 -07:00 committed by Facebook GitHub Bot
parent 42b5b96595
commit b73d3d6ed9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_lighting.py
View File

@ -469,7 +469,7 @@ class TestSpecularLighting(TestCaseMixin, unittest.TestCase):
top K faces per pixel.
"""
device = torch.device("cuda:0")
N, H, W, K = 16, 256, 256, 100
N, H, W, K = 8, 128, 128, 100
color = torch.tensor([1, 0, 1], dtype=torch.float32, device=device)
direction = torch.tensor(
[-1 / np.sqrt(2), 1 / np.sqrt(2), 0], dtype=torch.float32