Swin V2 CR impl refactor.

* reformat and change some naming so closer to existing timm vision transformers
* remove typing that wasn't adding clarity (or causing torchscript issues)
* support non-square windows
* auto window size adjust from image size
* post-norm + main-branch no
pull/1150/head
Ross Wightman 3 years ago
parent 67d140446b
commit c6e4b7895a

@ -38,6 +38,7 @@ from .selecsls import *
from .senet import * from .senet import *
from .sknet import * from .sknet import *
from .swin_transformer import * from .swin_transformer import *
from .swin_transformer_v2_cr import *
from .tnt import * from .tnt import *
from .tresnet import * from .tresnet import *
from .twins import * from .twins import *

File diff suppressed because it is too large Load Diff
Loading…
Cancel
Save