Skip to content

Commit 6a41c09

Browse files
committed
fix
1 parent 2be018c commit 6a41c09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

_unittests/ut_torch_models/test_validate_whole_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def test_m_validate_model_vit_model(self):
227227
self.assertIsInstance(summary, dict)
228228
self.assertIsInstance(data, dict)
229229
self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-3)
230-
self.assertLess(summary["disc_onnx_ort_run2_abs"], 1e-3)
230+
self.assertLess(summary["disc_onnx_ort_run22_abs"], 1e-3)
231231
self.assertEqual("dict(pixel_values:A1s2x3x30x30)", summary["run_feeds_inputs"])
232232
self.assertEqual("dict(pixel_values:A1s3x3x31x31)", summary["run_feeds_inputs2"])
233233
self.assertEqual("#1[A1s2x2]", summary["run_output_inputs"])

0 commit comments

Comments
 (0)