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

Author: bottler

tests/test_lighting.py

@@ -469,7 +469,7 @@ def test_specular_batched_arbitrary_input_dims(self):
         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