File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 1919 TEST_WITH_ROCM , _assertGradAndGradgradChecks
2020from torch .testing ._internal .common_cuda import TEST_CUDA
2121from torch .testing ._internal .common_device_type import expectedAlertNondeterministic
22- from torch .autograd .gradcheck import get_numerical_jacobian , iter_tensors
22+ from torch .autograd .gradcheck import get_numerical_jacobian , iter_tensors , \
23+ gradcheck , gradgradcheck
2324from torch .autograd import Variable
2425import torch .backends .cudnn
2526
@@ -4861,7 +4862,7 @@ class CriterionTest(TestBase):
48614862 def __init__ (self , * args , ** kwargs ):
48624863 super ().__init__ (* args , ** kwargs )
48634864 self .should_test_cuda = kwargs .get ('test_cuda' , True )
4864- self .check_forward_only = kwargs .get ('check_forward_only' , True )
4865+ self .check_forward_only = kwargs .get ('check_forward_only' , False )
48654866
48664867 def _get_target (self ):
48674868 return self ._get_arg ('target' , True )
You can’t perform that action at this time.
0 commit comments