Skip to content

GitLab

Unverified Commit 539c6e29 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Put back previous tolerance for test_classification and test_video (#7202)

parent 86f551d3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/test_models.py
View file @ 539c6e29
...@@ -686,7 +686,7 @@ def test_classification_model(model_fn, dev): ...@@ -686,7 +686,7 @@ def test_classification_model(model_fn, dev):
model.eval().to(device=dev) model.eval().to(device=dev)
x = _get_image(input_shape=input_shape, real_image=real_image, device=dev) x = _get_image(input_shape=input_shape, real_image=real_image, device=dev)
out = model(x) out = model(x)
_assert_expected(out.cpu(), model_name, prec=1e-3) _assert_expected(out.cpu(), model_name, prec=0.1)
assert out.shape[-1] == num_classes assert out.shape[-1] == num_classes
_check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out) _check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out)
_check_fx_compatible(model, x, eager_out=out) _check_fx_compatible(model, x, eager_out=out)
...@@ -917,7 +917,7 @@ def test_video_model(model_fn, dev): ...@@ -917,7 +917,7 @@ def test_video_model(model_fn, dev):
# RNG always on CPU, to ensure x in cuda tests is bitwise identical to x in cpu tests # RNG always on CPU, to ensure x in cuda tests is bitwise identical to x in cpu tests
x = torch.rand(input_shape).to(device=dev) x = torch.rand(input_shape).to(device=dev)
out = model(x) out = model(x)
_assert_expected(out.cpu(), model_name, prec=1e-5) _assert_expected(out.cpu(), model_name, prec=0.1)
assert out.shape[-1] == num_classes assert out.shape[-1] == num_classes
_check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out) _check_jit_scriptable(model, (x,), unwrapper=script_model_unwrapper.get(model_name, None), eager_out=out)
_check_fx_compatible(model, x, eager_out=out) _check_fx_compatible(model, x, eager_out=out)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment