From cb1e42476e15d0bbc6e8e26e038940174acb57a1 Mon Sep 17 00:00:00 2001 From: Konstantin Date: Fri, 23 Jun 2023 14:12:54 +0000 Subject: [PATCH 1/2] Remove dev print statements --- src/sparsify/auto/tasks/image_classification/runner.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/sparsify/auto/tasks/image_classification/runner.py b/src/sparsify/auto/tasks/image_classification/runner.py index da7ddb45..ba26fe74 100644 --- a/src/sparsify/auto/tasks/image_classification/runner.py +++ b/src/sparsify/auto/tasks/image_classification/runner.py @@ -14,6 +14,7 @@ import json +import logging import os from typing import Tuple @@ -42,6 +43,8 @@ __all__ = ["ImageClassificationRunner"] +_LOGGER = logging.getLogger(__name__) + @TaskRunner.register_task(task=TASK_REGISTRY["image_classification"]) class ImageClassificationRunner(TaskRunner): @@ -169,14 +172,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 From 217a7f573783cc8e6f346addb3e6c6dfe70697da Mon Sep 17 00:00:00 2001 From: Konstantin Date: Fri, 23 Jun 2023 14:13:58 +0000 Subject: [PATCH 2/2] Remove logger --- src/sparsify/auto/tasks/image_classification/runner.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/sparsify/auto/tasks/image_classification/runner.py b/src/sparsify/auto/tasks/image_classification/runner.py index ba26fe74..b7bd486e 100644 --- a/src/sparsify/auto/tasks/image_classification/runner.py +++ b/src/sparsify/auto/tasks/image_classification/runner.py @@ -14,7 +14,6 @@ import json -import logging import os from typing import Tuple @@ -43,8 +42,6 @@ __all__ = ["ImageClassificationRunner"] -_LOGGER = logging.getLogger(__name__) - @TaskRunner.register_task(task=TASK_REGISTRY["image_classification"]) class ImageClassificationRunner(TaskRunner):