-
Notifications
You must be signed in to change notification settings - Fork 38
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 script to convert models to tflite #519
base: master
Are you sure you want to change the base?
Conversation
@FenixFly, а есть ли смысл выкладывать этот скрипт? Насколько я понимаю, из всего множества моделей, перечисленных в этом скрипте, реально удалось сконвертировать меньше половины. |
Этим скриптом получены onnx модели, а из них 5 tflite моделей, на которых выложены тесты tflite на riscv, наверное всё-таки его нужно оставить. Если позже появится скрипт, которых "лечит" батч для onnx моделей, то в tflite сконвертится больше моделей. |
Тогда я бы предложила разместить его рядом с dockerfile для tflite по аналогии с TVM (https://github.com/itlab-vision/dl-benchmark/tree/master/docker/TVM). Единственное - тогда имеет смысл положить какой-то минимальный ридми с описанием скриптов, которые лежат рядом с этим dockerfile, чтобы понимать, что и зачем. |
pip install onnx-tf==1.10.0 | ||
pip install tensorflow-addons==0.22.0 | ||
pip install tensorflow-probability==0.22.0 | ||
pip install torch==1.13.1 torchvisio==0.14.1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
pip install torch==1.13.1 torchvisio==0.14.1 | |
pip install torch==1.13.1 torchvision==0.14.1 |
@FenixFly, поправь, пожалуйста, мелкие замечания, чтобы залить. Спасибо! |
No description provided.