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

Switch to tf 2.4 #127

Closed zaccharieramzi closed 3 years ago

zaccharieramzi commented 3 years ago

This should allow the use of mixed precision even for multi scale models

zaccharieramzi commented 3 years ago

This indeed allowed the use of mixed precision for multi-scale models. I also managed to corrected the image size adjustment (And added a unit test for it).

This PR also contains various fixes, and extensions to different data settings of pipelines that were already in place.

Finally it also enables a distributed setting in a SLURM cluster.