microsoft / CSWin-Transformer

CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022
MIT License
539 stars 78 forks source link

why gpu memory of cswin-tiny is twice than swin-tiny? #12

Open UcanSee opened 3 years ago

UcanSee commented 3 years ago

I have transformed cswin-tiny to object detection task, just adding some padding operation. When I used Faster-RCNN model, I found that gpu memory of cswin-tiny is twice than that of swin-tiny, is there something wrong I made?