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 nopull/1150/head
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…
Reference in New Issue