search

zaccharieramzi / fastmri-reproducible-benchmark

Try several methods for MRI reconstruction on the fastmri dataset. Home to the XPDNet, runner-up of the 2020 fastMRI challenge.
https://fastmri.org/leaderboards
MIT License
151 stars 50 forks source link

Multi gpu #78

Closed zaccharieramzi closed 4 years ago

zaccharieramzi commented 4 years ago

Introducing model sequentialism to have different blocks of the model placed on different GPUs to benefit from more memory.

A little communication overhead is expected so computation should be slower.