From 1ed748e7f303bda8d917aa2b501fe49c9a87c87b Mon Sep 17 00:00:00 2001 From: mertalev <101130780+mertalev@users.noreply.github.com> Date: Fri, 2 Feb 2024 22:18:14 -0500 Subject: [PATCH] add comment about gross code --- machine-learning/app/models/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py index c25bd370c2..fd1be17e5b 100644 --- a/machine-learning/app/models/base.py +++ b/machine-learning/app/models/base.py @@ -143,6 +143,7 @@ class InferenceModel(ABC): inputs = self._get_static_dims(inferred.graph.input) outputs = self._get_static_dims(inferred.graph.output) + # check_model gets called in update_inputs_outputs_dims and doesn't work for large models check_model = onnx.checker.check_model try: onnx.checker.check_model = lambda _: None