Add check_gradgrad=False for new NLLLoss2d test.

zhouyonglong · Jul 25, 2017 · 506d52d · 506d52d
1 parent 7687c26
commit 506d52d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/common_nn.py b/test/common_nn.py
@@ -311,7 +311,8 @@
         constructor_args=(None, True, 3),
         input_size=(2, 3, 5, 5),
         target=torch.rand(2, 5, 5).mul(4).floor().long(),
-        desc='ignore_index'
+        desc='ignore_index',
+        check_gradgrad=False,
     ),
     dict(
         module_name='HingeEmbeddingLoss',