Don't register mem as buffer

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

https://snntorch.readthedocs.io/en/latest/

MIT License

1.17k stars 202 forks source link

Closed gekkom closed 2 months ago

gekkom commented 2 months ago

Fixes #314

stevenabreu7 commented 2 months ago

This fixes the membrane potential error when loading a state_dict for me, thanks!

jeshraghian commented 2 months ago

hero.