add comment about gross code

2025-11-29 08:30:03 +09:00 · 2024-02-02 22:18:14 -05:00
parent b374052de2
commit 1ed748e7f3
1 changed files with 1 additions and 0 deletions
--- 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