lbc12345 / SeD

Semantic-Aware Discriminator for Image Super-Resolution
94 stars 5 forks source link

test error #5

Open chichu1021 opened 3 months ago

chichu1021 commented 3 months ago

This error occurred when I was testing, only loading a picture, why

RuntimeError: CUDA out of memory. Tried to allocate 87.95 GiB (GPU 0; 44.48 GiB total capacity; 22.69 GiB already allocated; 19.28 GiB free; 23.92 GiB reserved in total by PyTorch)

lbc12345 commented 3 months ago

It seems that your input image is too large. You may try to crop your image into patches first.

chichu1021 commented 3 months ago

just 200-300 KB

chichu1021 commented 3 months ago

i try a smaller image,just 57KB,success

lbc12345 commented 3 months ago

"Large image" refers to the resolution of image, not the storage size of the image. If you want to try a 2K image, you may need to crop it into patches first.