Skip to content
Snippets Groups Projects
Commit fbbd52c6 authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Adding CNN model description in developer docs

parent 8a3bf97b
No related branches found
No related tags found
No related merge requests found
CNN Model Weights
===================
The CNN weights (and input) files can be downloaded from here: https://databank.illinois.edu/datasets/IDB-6565690
The extracted `model_params` directory is to be copied to `hpvm/hpvm/test/dnn_benchmarks/model_params` - the CNN benchmark expect the model weights at this specific location. The automatic HPVM install (`install.sh`) does the data download, extraction, and copying automatically.
We support CNN weights in 3 formats:
* `.h5` file format: The entire CNN model is stored as a single `.h5` file. The Keras models are shipped as `.h5` files. These can be found under `model_params/keras`
* `.pth.tar` file format: The PyTorch models are shipped as `pth.tar` files and can be found under `model_params/pytorch`
* `.bin` serialized binary file format: This format is used by the HPVM binaries. Convolution, Dense, and BatchNormalization parameters for each layer are stored as individual files. The weights are serialized FP32 values layed out serially in `NCHW` format. Our frontends (Keras and PyTorch) convert `.h5` and `pth.tar` files into `.bin` files in the frontend translation phase. The `.bin` weights can be found under the respective subdirectory for each benchmark under `model_params/`
......@@ -5,3 +5,4 @@ Developer Documents
:maxdepth: 1
approximation-implementation
cnn-models
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment