Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add edgetpu_compiler checks #6218

Merged
merged 7 commits into from
Jan 5, 2022
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update export.py
  • Loading branch information
glenn-jocher authored Jan 5, 2022
commit f6485ae2324528b2a6f2ae7b87fe269c23e452fa
2 changes: 1 addition & 1 deletion export.py
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,7 @@ def export_edgetpu(keras_model, im, file, prefix=colorstr('Edge TPU:')):
try:
cmd = 'edgetpu_compiler --version'
help_url = 'https://coral.ai/docs/edgetpu/compiler/'
assert platform.system() == 'Linux', 'export only supported on Linux'
assert platform.system() == 'Linux', f'export only supported on Linux. See {help_url}'
assert subprocess.run(cmd, shell=True, check=True), f'export requires edgetpu-compiler. See {help_url}'
out = subprocess.run(cmd, shell=True, capture_output=True, check=True)
ver = out.stdout.decode().split()[-1]
Expand Down