search

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.
https://spikingjelly.readthedocs.io
Other
1.35k stars 239 forks source link

Optimizing voltage initialization of multi-step neurons #179

Open fangwei123456 opened 2 years ago

fangwei123456 commented 2 years ago

https://github.com/fangwei123456/spikingjelly/blob/29b1bb3c09db3f87786b34c03f4fa781c2fe3b66/spikingjelly/clock_driven/neuron.py#L277

image

fangwei123456 commented 2 years ago

The tensor filled with v0 should not be re-newed after each reset.