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