itayhubara / BinaryNet.pytorch

Binarized Neural Network (BNN) for pytorch
505 stars 126 forks source link

Replace deprecated part and fix hinge train script. #16

Closed Enderdead closed 4 years ago

Enderdead commented 5 years ago

I fix some mistake due to deprecated usage. Here some error fixed :

And I fix some mistake about the Hinge loss label given and Net output dim.