From b73d3d6ed96f8459776b23d4fba1af746e643fa4 Mon Sep 17 00:00:00 2001
From: Jeremy Reizenstein <reizenstein@fb.com>
Date: Wed, 29 Jul 2020 04:59:47 -0700
Subject: [PATCH] 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
---
 tests/test_lighting.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_lighting.py b/tests/test_lighting.py
index 37d8653a..6dcc1ba6 100644
--- a/tests/test_lighting.py
+++ b/tests/test_lighting.py
@@ -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