Make 3D gradcheck test use less memory

ahendriksen · May 26, 2021 · 081fd6d · 081fd6d
1 parent ee43f4f
commit 081fd6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msd_pytorch/tests/test_conv3d.py b/msd_pytorch/tests/test_conv3d.py
@@ -388,7 +388,7 @@ def test_grad_check():
     C_OUT = 2               # Output channels
     D = 11                  # Depth
     H = 13                  # Height
-    W = 21                  # Width
+    W = 7                   # Width
     dilation = 3            # Dilation
 
     x = torch.randn(B, C_IN, D, H, W, dtype=dtype, requires_grad=True).cuda()