lbc12345/SeD

test error

chichu1021 opened this issue · 4 comments

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)

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

just 200-300 KB

i try a smaller image,just 57KB,success

"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.