Fix torch 1.10 errors + tests on multiple pytorch versions

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

Closed pierreablin closed 2 years ago

pierreablin commented 2 years ago

This fixes torch 1.10 errors and adds unit testing on different pytorch versions.