Hobe
/
CIM_Training
1
0
Fork 0
Code Issues 0 Pull Requests 0 Projects 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
3 Commits
1 Branch
9.8 MiB
 
Tree: baf9d09f67
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'baf9d09f67'
${ noResults }
CIM_Training/BinaryNetpytorch
Hobe baf9d09f67 Upload 8 class CCCB version 4 years ago
..
__pycache__ Upload 8 class CCCB version 4 years ago
models Upload 8 class CCCB version 4 years ago
results Upload 8 class CCCB version 4 years ago
.DS_Store Upload 8 class CCCB version 4 years ago
README.md Upload 8 class CCCB version 4 years ago
data.py Upload 8 class CCCB version 4 years ago
main_binary.py Upload 8 class CCCB version 4 years ago
main_binary_hinge.py Upload 8 class CCCB version 4 years ago
main_mnist.py Upload 8 class CCCB version 4 years ago
preprocess.py Upload 8 class CCCB version 4 years ago
utils.py Upload 8 class CCCB version 4 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