Create optimize utility/quick train and eval loop

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

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

MIT License

98 stars 11 forks source link

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.