HarukiYqM/Non-Local-Sparse-Attention

My GPU has a small memory

Closed this issue · 2 comments

How to clear video memory during training without affecting training

I am not sure about how to clear the memory and what video memory refer to. Since this code is fully based on EDSR (https://github.com/sanghyun-son/EDSR-PyTorch), you may want to look at that repo about how to use this code.

video memory refers to GPU's memory