@@ -686,7 +686,7 @@ def test_classification_model(model_fn, dev):
686
686
model .eval ().to (device = dev )
687
687
x = _get_image (input_shape = input_shape , real_image = real_image , device = dev )
688
688
out = model (x )
689
- _assert_expected (out .cpu (), model_name , prec = 1e-3 )
689
+ _assert_expected (out .cpu (), model_name , prec = 0.1 )
690
690
assert out .shape [- 1 ] == num_classes
691
691
_check_jit_scriptable (model , (x ,), unwrapper = script_model_unwrapper .get (model_name , None ), eager_out = out )
692
692
_check_fx_compatible (model , x , eager_out = out )
@@ -917,7 +917,7 @@ def test_video_model(model_fn, dev):
917
917
# RNG always on CPU, to ensure x in cuda tests is bitwise identical to x in cpu tests
918
918
x = torch .rand (input_shape ).to (device = dev )
919
919
out = model (x )
920
- _assert_expected (out .cpu (), model_name , prec = 1e-5 )
920
+ _assert_expected (out .cpu (), model_name , prec = 0.1 )
921
921
assert out .shape [- 1 ] == num_classes
922
922
_check_jit_scriptable (model , (x ,), unwrapper = script_model_unwrapper .get (model_name , None ), eager_out = out )
923
923
_check_fx_compatible (model , x , eager_out = out )
0 commit comments