Skip to content

Commit 539c6e2

Browse files
authored
Put back previous tolerance for test_classification and test_video (#7202)
1 parent 86f551d commit 539c6e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -686,7 +686,7 @@ def test_classification_model(model_fn, dev):
686686
model.eval().to(device=dev)
687687
x = _get_image(input_shape=input_shape, real_image=real_image, device=dev)
688688
out = model(x)
689-
_assert_expected(out.cpu(), model_name, prec=1e-3)
689+
_assert_expected(out.cpu(), model_name, prec=0.1)
690690
assert out.shape[-1] == num_classes
691691
_check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out)
692692
_check_fx_compatible(model, x, eager_out=out)
@@ -917,7 +917,7 @@ def test_video_model(model_fn, dev):
917917
# RNG always on CPU, to ensure x in cuda tests is bitwise identical to x in cpu tests
918918
x = torch.rand(input_shape).to(device=dev)
919919
out = model(x)
920-
_assert_expected(out.cpu(), model_name, prec=1e-5)
920+
_assert_expected(out.cpu(), model_name, prec=0.1)
921921
assert out.shape[-1] == num_classes
922922
_check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out)
923923
_check_fx_compatible(model, x, eager_out=out)

0 commit comments

Comments
 (0)