From e58260ef4f1ee446a0f8413e3c0db7f0f65ae240 Mon Sep 17 00:00:00 2001 From: Konstantin Gulin <66528950+KSGulin@users.noreply.github.com> Date: Fri, 23 Jun 2023 16:22:54 +0100 Subject: [PATCH] Remove dev print statements from IC runner (#231) * Remove dev print statements * Remove logger --- src/sparsify/auto/tasks/image_classification/runner.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/sparsify/auto/tasks/image_classification/runner.py b/src/sparsify/auto/tasks/image_classification/runner.py index da7ddb45..b7bd486e 100644 --- a/src/sparsify/auto/tasks/image_classification/runner.py +++ b/src/sparsify/auto/tasks/image_classification/runner.py @@ -169,14 +169,12 @@ def _export_completion_check(self) -> bool: # Check mode file exists if not os.path.isfile(onnx_file): - print(f"ONNX FILE PATH: {onnx_file}") return False # Check onnx graph try: model = onnx.load(onnx_file) onnx.checker.check_model(model) - print("FAILED ONNX CHECKER") except Exception: return False