wyf0912/SinSR

torch.OutOfMemoryError: CUDA out of memory. even for BS =1

Opened this issue · 0 comments

Even with a batch size of 1 for train & val , I still get Out of Memory error
Screenshot from 2024-10-23 10-18-47

My config yaml file is like this : SinSR.txt

Training log is: training.log

How to fix this please ?

Screenshot from 2024-10-23 10-28-10