This repository was archived by the owner on Nov 25, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +37
-1
lines changed Expand file tree Collapse file tree 2 files changed +37
-1
lines changed Original file line number Diff line number Diff line change @@ -26,20 +26,22 @@ jobs:
2626 - name : Run a one-line script
2727 run : echo Hello, world!
2828
29+
2930 # Runs a set of commands using the runners shell
3031 - name : Run a multi-line script
3132 run : |
3233 echo Add other actions to build,
3334 echo test, and deploy your project.
3435
36+
3537 - name : pre-commit check
3638 run :
3739 cd .github/workflows && ./pre-commit-job.sh
3840
41+
3942 - name : python unit test
4043 run :
4144 .github/workflows/unit-test-job.sh
42-
4345
4446
4547 tnn :
4951 - name : Run tnn build
5052 run : cd tnn && ./build.sh
5153
54+
55+ tnn-model-convertor :
56+ runs-on : ubuntu-latest
57+ steps :
58+ - uses : actions/checkout@v2
59+ - name : Run tnn model convertor build
60+ run : cd tnn && ./build_tnn_model_convertor.sh
61+
62+
5263 models :
5364 runs-on : ubuntu-latest
5465 steps :
Original file line number Diff line number Diff line change 1+ #! /usr/bin/env bash
2+ set -ex
3+
4+ function prepare_env() {
5+ # default
6+ pip3 install onnx==1.6.0 onnxruntime numpy onnx-simplifier
7+
8+ # optional: tensorflow
9+ pip3 install tensorflow==1.15.0
10+ pip3 install tf2onnx
11+ pip3 install onnxruntime
12+
13+ # optional: caffe
14+ sudo apt-get install libprotobuf-dev protobuf-compiler
15+ }
16+
17+ function main() {
18+ prepare_env
19+
20+ # build
21+ cd tnn/tools/convert2tnn
22+ ./build.sh
23+ }
24+
25+ main
You can’t perform that action at this time.
0 commit comments