We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2be018c commit 6a41c09Copy full SHA for 6a41c09
_unittests/ut_torch_models/test_validate_whole_models.py
@@ -227,7 +227,7 @@ def test_m_validate_model_vit_model(self):
227
self.assertIsInstance(summary, dict)
228
self.assertIsInstance(data, dict)
229
self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-3)
230
- self.assertLess(summary["disc_onnx_ort_run2_abs"], 1e-3)
+ self.assertLess(summary["disc_onnx_ort_run22_abs"], 1e-3)
231
self.assertEqual("dict(pixel_values:A1s2x3x30x30)", summary["run_feeds_inputs"])
232
self.assertEqual("dict(pixel_values:A1s3x3x31x31)", summary["run_feeds_inputs2"])
233
self.assertEqual("#1[A1s2x2]", summary["run_output_inputs"])
0 commit comments