mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
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
This commit is contained in:
parent
67d140446b
commit
c6e4b7895a
@ -38,6 +38,7 @@ from .selecsls import *
|
||||
from .senet import *
|
||||
from .sknet import *
|
||||
from .swin_transformer import *
|
||||
from .swin_transformer_v2_cr import *
|
||||
from .tnt import *
|
||||
from .tresnet import *
|
||||
from .twins import *
|
||||
|
File diff suppressed because it is too large
Load Diff
Loading…
x
Reference in New Issue
Block a user