New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

ManuBN786 opened this issue Oct 23, 2024 · 2 comments

ManuBN786 commented Oct 23, 2024 •

edited

Loading

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 ?

y2k0999 commented Nov 25, 2024

what's your image shape?

zouiner commented Jan 31, 2025 •

edited

Loading

what's your image shape?

I got the same problem with image size 64,64
@y2k0999

@ManuBN786 Can you fix it out?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment