Learning to Binarized

This is the source code for the paper Learning to Binarize Convolutional Neural Networks with Adaptive Neural Encoder.

Dependencies

• Python 3.5
• Pytorch 1.3
• Cuda

Train

• Download the pretrained teacher model resnet18.pth .
• Modify the run.sh file with your ImageNet dataset directory

clear
python3 trainimagenet.py --data /IMAGENET-DATA | tee -a log/log.txt

• Run the run.sh file in terminal

./run.sh

Reference

• Bi-Real-net

• ReActNet

• MetaQuant