Skip to content

Commit 8b48ab5

Browse files
[PT] less atol on windows for test_same_outputs_in_torch_and_exported_onnx (#3311)
### Changes Use atol=1e-5 in test_same_outputs_in_torch_and_exported_onnx ### Reason for changes https://github.com/openvinotoolkit/nncf/actions/runs/13510568468/job/37749802966 ### Tests https://github.com/openvinotoolkit/nncf/actions/runs/13533343712/job/37820218205?pr=3311
1 parent 3e087cd commit 8b48ab5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/torch/sparsity/movement/test_model_saving.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def test_same_outputs_in_torch_and_exported_onnx(self, tmp_path: Path, recipe: B
129129
compression_ctrl.export_model(onnx_model_path)
130130
onnx_output_dict = self._get_onnx_model_inference_outputs(onnx_model_path, dataset, recipe)
131131
onnx_outputs = next(iter(onnx_output_dict.values()))
132-
assert np.allclose(softmax(onnx_outputs, axis=-1), softmax(torch_outputs, axis=-1), atol=1e-6)
132+
assert np.allclose(softmax(onnx_outputs, axis=-1), softmax(torch_outputs, axis=-1), atol=1e-5)
133133

134134
@pytest.mark.skipif(
135135
version.parse(torch.__version__) < version.parse("1.12"),

0 commit comments

Comments
 (0)