|
3 years ago | |
---|---|---|
.. | ||
__pycache__ | 3 years ago | |
models | 3 years ago | |
results | 3 years ago | |
.DS_Store | 3 years ago | |
README.md | 3 years ago | |
data.py | 3 years ago | |
main_binary.py | 3 years ago | |
main_binary_hinge.py | 3 years ago | |
main_mnist.py | 3 years ago | |
preprocess.py | 3 years ago | |
utils.py | 3 years ago |
Binarized Neural Network (BNN) for pytorch This is the pytorch version for the BNN code, fro VGG and resnet models Link to the paper: https://papers.nips.cc/paper/6573-binarized-neural-networks
The code is based on https://github.com/eladhoffer/convNet.pytorch Please install torch and torchvision by following the instructions at: http://pytorch.org/ To run resnet18 for cifar10 dataset use: python main_binary.py --model resnet_binary --save resnet18_binary --dataset cifar10