Refactored U-Net model code

facebookresearch / fastMRI

A large-scale dataset of both raw MRI measurements and clinical MRI images.

https://fastmri.org

MIT License

1.34k stars 373 forks source link

Closed anuroopsriram closed 4 years ago

anuroopsriram commented 4 years ago

Refactored U-Net model code to use pytorch-lightning.

tullie commented 4 years ago

One more thing I just noticed. Can you update the requirements.txt file to include pytorch-lighting?