itayhubara / BinaryNet.pytorch

Binarized Neural Network (BNN) for pytorch
504 stars 125 forks source link

Change activation to SELU #24

Closed codeprb closed 4 years ago

codeprb commented 4 years ago

Hi ,

I want to implement shifted relu or SELU on the resnet_binary code. But when I change the code to use SELU or even ReLu I get the following error. Could you please give me some hints about what else I might have to change to replace hardtanh to SELU? Any pointers would be really appreciated.

/Users/Desktop/BNN-Imagenet/models/resnet_binary.py(59)forward() -> residual = self.downsample(residual) (Pdb)