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

Fix DC for batch_size>1 #133

Closed chaithyagr closed 3 years ago

chaithyagr commented 3 years ago

Currently, the NCPDNet works only for batch_size = 1.

This is a minute bug.

chaithyagr commented 3 years ago

I have also updated this branch with a method to save optimizer weights at model checkpoint... This is particularly useful in case the run crashes as then we will have last model and its associated weights.