issues
search
kmheckel
/
spyx
Spyx: Spiking Neural Networks in JAX
https://spyx.readthedocs.io/en/latest/
MIT License
98
stars
11
forks
source link
Create optimize utility/quick train and eval loop
#26
Open
kmheckel
opened
6 months ago
kmheckel
commented
6 months ago
Make spyx.optimize:
[ ] surrogate gradient training loop
[ ] take SNN, params, Dataset in form (#batches, batchsize, timesteps, ...), optimizer
[ ] also take number of epochs, maybe eval frequency.
Make spyx.optimize: