XNOR-Net-Pytorch
这是的PyTorch实现。 我为以下项目实现了二值化神经网络(BNN):
数据集
网络
准确性
浮点精度
MNIST
LeNet-5
99.23%
99.34%
CIFAR-10
网络中网络(NIN)
86.28%
89.67%
影像网
亚历克斯网
前1名:44.87%前5名:69.70%
前1名:57.1%前5名:80.2%
MNIST
我为MNIST数据集实现了LeNet-5结构。 我正在使用提供的数据集阅读器。 要运行培训:
$ cd /MNIST/
$ python main.py
预训练的模型可以在下载。 要评估预训练模型:
$ cp /MNIST/models/
$ python main.py
2021-09-26 12:18:02
2.32MB
Python
1