/BinaryNet-Caffe2

Quantized Neural Network on Caffe2

Primary LanguageJupyter Notebook

Note

This repository is not maintian any more.

BNN_Caffe2

Quantized Neural Network on Caffe2

Goal: Porting QNN(BNN) from Theano version to Caffe2 version.

TODO:

  • MNIST with LeNet
  • MNIST with layers from QNN paper, without Batch Normalization, without Binarization.
  • MNIST with layers from QNN paper, with Batch Normalization, without Binarization.
  • Binarization functions (in progress)
  • Layers with Binarization
  • MNIST with layers from QNN paper, with Batch Normalization, with Binarization.
  • Testing and debugging
  • Training / evaluation on other dataset

Acknowledgment

Refs.

QNN Theano version paper https://arxiv.org/abs/1609.07061

Theano version github https://github.com/MatthieuCourbariaux/BinaryNet

License