Possible bug

[JIABI]

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

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

[ranery]

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.