torch.OutOfMemoryError: CUDA out of memory. even for BS =1
Opened this issue · 0 comments
ManuBN786 commented
Even with a batch size of 1 for train & val , I still get Out of Memory error
My config yaml file is like this : SinSR.txt
Training log is: training.log
How to fix this please ?