search

GATECH-EIC / ShiftAddNet

[NeurIPS 2020] ShiftAddNet: A Hardware-Inspired Deep Network
MIT License
68 stars 18 forks source link

Possible bug #7

Open JIABI opened 3 years ago

JIABI commented 3 years ago

https://github.com/RICE-EIC/ShiftAddNet/blob/654a6327444ba471ab6a80410d8835b5dd24c3a3/prune.py#L416

It should be as if isinstance(m, adder_module):

ranery commented 1 year ago

Thanks for checking! As we did not use random pruning for adder layers, I did not carefully check that part.

Feel free to adapt for your usage.