search

michaelsdr / momentumnet

Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
https://michaelsdr.github.io/momentumnet/
MIT License
207 stars 19 forks source link

new test with batch norms #29

Open michaelsdr opened 1 year ago

michaelsdr commented 1 year ago

Tests the gradients with batch norms

pierreablin commented 1 year ago

@michaelsdr do you know how to fix the unit tests?