search

guopengf / FL-MRCM

Multi-institutional Collaborations for Improving Deep Learning-based Magnetic Resonance Image Reconstruction Using Federated Learning
MIT License
45 stars 10 forks source link

GPU Memory Leakage #2

Closed bbbbbbzhou closed 3 years ago

bbbbbbzhou commented 3 years ago

Thanks for sharing

guopengf commented 3 years ago

You may add 'with torch.no_grad():' to line 172 in test.py. Memory leakage might result from it. I will fix and update the code. Thanks for pointing out this problem.