test error
chichu1021 opened this issue · 4 comments
chichu1021 commented
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
It seems that your input image is too large. You may try to crop your image into patches first.
chichu1021 commented
just 200-300 KB
chichu1021 commented
i try a smaller image,just 57KB,success
lbc12345 commented
"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.