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