Adding grad check in test cases

paganpasta / eqxvision

A Python package of computer vision models for the Equinox ecosystem.

https://eqxvision.readthedocs.io

MIT License

100 stars 12 forks source link

Adding grad check in test cases #13

Closed paganpasta closed 2 years ago

paganpasta commented 2 years ago

Currently, gradient computation is not supported with BatchNorm models. With the next release of Equinox, it would be nice to have test cases to check that networks are passing gradient computation.

paganpasta commented 2 years ago

Added in https://github.com/paganpasta/eqxvision/releases/tag/v0.1.4