Binarized Neural Network : BNN二值神经网络代码实例
我自己在研究BNN,苦于找不到代码(没有一个人写一个吗???)自己写了一个,也不知道是不是标准的BNN,反正就是把所有参数都二值化了,用的MNIST,效果一般。如果只二值权值的话就需要分别对每一层进行二值,这个代码量挺大的而且没有意义。from torchvision import datasets, transformsimport torchimport torch.nn as nnimport torch.nn.functional as Fimport numpy as npimpor
原创
2020-12-30 16:41:23 ·
1164 阅读 ·
2 评论