How to make nn.Linear binary weights

allenai / XNOR-Net

ImageNet classification using binary Convolutional Neural Networks

https://xnor.ai/

Other

856 stars 239 forks source link

How to make nn.Linear binary weights #24

Open xiao1228 opened 7 years ago

xiao1228 commented 7 years ago

If I'm adding nn.linear into the network. Can it get binarized? I have passed it through BinActiveZ . Is this all I need to do? Where should I change in the updateBinaryGradWeight and binarizeConvParms Because in binarizeConvParms it only binarized the ConvNode right?