search

ChristophReich1996 / Swin-Transformer-V2

PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" [CVPR 2022].
https://arxiv.org/abs/2111.09883
MIT License
173 stars 14 forks source link

How to use for SSL or classifier? #12

Closed sarmientoj24 closed 2 years ago

sarmientoj24 commented 2 years ago

Since it produces feature maps, how do I "flatten" the feature maps for the classifier?

ChristophReich1996 commented 2 years ago

Hi @sarmientoj24, Sorry for my late response. Have a look at the image classification code here. This class just wraps the model for classification.