Shuffling utility has returns leading axis zer0

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

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

MIT License

103 stars 12 forks source link

Closed kmheckel closed 9 months ago

kmheckel commented 9 months ago

If the batch size perfectly divides the dataset then the shuffling function tool will result in a leading axis of zero, inducing a bug.

kmheckel commented 9 months ago

Fixed in release 0.1.19