Commit b73d3d6e authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

smaller data in specular test

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
parent 42b5b965
......@@ -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
......
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