OOM for high resolution image

[jingyangcarl]

awesome work!!

I'm testing the pipeline for high-resolution images (2048*2048), which takes 128G GPU. Any suggestions on how to make it work for high-resolution inputs? Thanks!

RuntimeError: CUDA out of memory. Tried to allocate 128.00 GiB (GPU 0; 47.52 GiB total capacity; 5.77 GiB already allocated; 38.53 GiB free; 7.01 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF