NKI-AI / direct

Deep learning framework for MRI reconstruction
https://docs.aiforoncology.nl/direct
Apache License 2.0
228 stars 40 forks source link

High memory consumption during validation #171

Closed jonasteuwen closed 2 years ago

jonasteuwen commented 2 years ago

Describe the bug When there is a lot of volumes to validate, there is a very high memory use

To Reproduce E.g. run FastMRI

Expected behavior A slight increase of memory is acceptable, because you keep the whole volume in memory. But not double. Likely we also need to delete the training dataloaders during validation.