Skip to content

Commit

Permalink
Remove dev print statements from IC runner (#231)
Browse files Browse the repository at this point in the history
* Remove dev print statements

* Remove logger
  • Loading branch information
KSGulin committed Jun 23, 2023
1 parent 6ddd6d4 commit 9eb5cc9
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/sparsify/auto/tasks/image_classification/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 9eb5cc9

Please sign in to comment.