Skip to content

Pytorch implementations of the BNN, XNOR-Net and BiReal-Net

Notifications You must be signed in to change notification settings

ShuaiZ1037/bnn-xnor-bireal

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BNN-XNOR-BiReal

This is a project contains Pytorch implementations of the BNN*, XNOR-Net and BiReal-Net. (NOTE: the first and last layer of BNN* is floating point.)


Results

Method Acc
BNN 71.70%
XNOR 80.77%
BiReal 83.17%

Dependencies

  • Python 3.5
  • Pytorch 1.3
  • Cuda

Train

  • Train a binary model with bireal:
python train.py --binary_method bireal
  • Train a binary model with xnor:
python train.py --binary_method xnor
  • Train a binary model with bnn:
python train.py --binary_method bnn

Todo

  • Pretrain model on cifar10

Reference

About

Pytorch implementations of the BNN, XNOR-Net and BiReal-Net

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages