sajaddarabi / BiRealNet

Bi-Real-Net Model in Pytorch https://arxiv.org/abs/1808.00278 + pre-trained fp model weights

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BiRealNet

Implementation of Bi-Real-Net Model in Pytorch https://arxiv.org/abs/1808.00278

Below you can find pretrained weights on imagenet for full-precision bireal18 and bireal34 network.

Both networks are trained by setting initial lr=1.0e-1, weight_decay=1e-4 and momentum=0.9. The learning rate is decayed by 0.1 using a step-lr schedule every 30 epochs for a total of 120 epochs.

bireal18 bireal34

Related Work

BNN+: Improved Binary Network Training: https://arxiv.org/abs/1812.11800

About

Bi-Real-Net Model in Pytorch https://arxiv.org/abs/1808.00278 + pre-trained fp model weights


Languages

Language:Python 100.0%