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 authored and bfineran committed Jun 30, 2023
1 parent 394d942 commit e58260e
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 e58260e

Please sign in to comment.