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

Dataparallel #26

Closed michaelsdr closed 2 years ago

michaelsdr commented 2 years ago

Add of method find_parameters() to enable dataparralelism