https://tvm.apache.org/docs/install/from_source.html
https://www.tensorflow.org/install/
- Dowload and extract existing model
http://download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224.tgz
import tensorflow as tf
cat 'models/mobilenet_v1_1.0_224/mobilenet_v1_1.0_224_info.txt'
Model: mobilenet_v1_1.0_224 Input: input Output: MobilenetV1/Predictions/Reshape_1
There should be a .pb file that is to be used for compilation.
- Create own toy model
Run cnn.py
python3 cnn.py
The trained model will be saved to a cnn.pb file
Run compiler.py
python3 tvm_compiler.py
This generates .json and .params