Hobe
/
CIM_Training
1
0
Fork 0
Code Issues 0 Pull Requests 0 Projects 0 Releases 0 Wiki Activity
4 Commits
1 Branch
9.8 MiB
 
Branch: master
CIM_Training/BinaryNetpytorch
Hobe baf9d09f67 Upload 8 class CCCB version 3 years ago
..
__pycache__ Upload 8 class CCCB version 3 years ago
models Upload 8 class CCCB version 3 years ago
results Upload 8 class CCCB version 3 years ago
.DS_Store Upload 8 class CCCB version 3 years ago
README.md Upload 8 class CCCB version 3 years ago
data.py Upload 8 class CCCB version 3 years ago
main_binary.py Upload 8 class CCCB version 3 years ago
main_binary_hinge.py Upload 8 class CCCB version 3 years ago
main_mnist.py Upload 8 class CCCB version 3 years ago
preprocess.py Upload 8 class CCCB version 3 years ago
utils.py Upload 8 class CCCB version 3 years ago

README.md

BNN.pytorch

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