the memory is nearly twice that of swin transformer

[cjt222]

Hi，Thanks for great jobs, I try to infer a image ， I tried to use the segmentation model by cswin transformer small to test some pictures, and found that the memory is nearly twice that of swin transformer small. Is this normal?

[vidit98]

Hi, you can try to set use_chk param of CSwin to True for reducing memory footprint. I suppose you are already using checkpointing in Swin model.