Deep-MRI-Reconstruction icon indicating copy to clipboard operation
Deep-MRI-Reconstruction copied to clipboard

Unnecessary GPU memory usage in PyTorch Implementation

Open tinyRattar opened this issue 6 years ago • 1 comments

I noticed that in main_crnn.py, line 167 : validate_err += err as err is a Varaible on GPU, such usage will take extra GPU memory (with PyTorch 1.0.1), maybe you should use err.item() instead like line 147. line 188 has the same problem as well

tinyRattar avatar Mar 22 '19 12:03 tinyRattar

Thanks!

Alxemade avatar Dec 23 '19 12:12 Alxemade