Out of Memory for 1024x1024 Resolution

GuoLanqing / ShadowFormer

ShadowFormer (AAAI2023), Pytorch implementation

MIT License

141 stars 16 forks source link

Open TolgaAktas opened 1 year ago

TolgaAktas commented 1 year ago

I run into CUDA OOM errors for 1024x1024 even for a batch size of 1, any suggestions how we might be able to fit it to the memory?