Ross Wightman
4b30bae67b
Add updated vit_relpos weights, and impl w/ support for official swin-v2 differences for relpos. Add bias control support for MLP layers
3 years ago
Ross Wightman
d4c0588012
Remove persistent buffers from Swin-V2. Change SwinV2Cr cos attn + tau/logit_scale to match official, add ckpt convert, init_value zeros resid LN weight by default
3 years ago
Ross Wightman
27c42f0830
Fix torchscript use for offician Swin-V2, add support for non-square window/shift to WindowAttn/Block
3 years ago
Ross Wightman
2f2b22d8c7
Disable nvfuser fma / opt level overrides per #1244
3 years ago
Ross Wightman
c0211b0bf7
Swin-V2 test fixes, typo
3 years ago
Ross Wightman
9a86b900fa
Official SwinV2 models
3 years ago
Ross Wightman
d07d015173
Merge pull request #1249 from okojoalg/sequencer
...
Add Sequencer
3 years ago
Ross Wightman
d30685c283
Merge pull request #1251 from hankyul2/fix-multistep-scheduler
...
fix: multistep lr decay epoch bugs
3 years ago
han
a16171335b
fix: change milestones to decay-milestones
...
- change argparser option `milestone` to `decay-milestone`
3 years ago
Ross Wightman
39b725e1c9
Fix tests for rank-4 output where feature channels dim is -1 (3) and not 1
3 years ago
Ross Wightman
78a32655fa
Fix poolformer group_matcher to merge proj downsample with previous block, support coarse
3 years ago
Ross Wightman
d79f3d9d1e
Fix torchscript use for sequencer, add group_matcher, forward_head support, minor formatting
3 years ago
Ross Wightman
37b6920df3
Fix group_matcher regex for regnet.py
3 years ago
okojoalg
93a79a3dd9
Fix num_features in Sequencer
3 years ago
okojoalg
2fec08e923
Add Sequencer to non std filters
3 years ago
han
57a988df30
fix: multistep lr decay epoch bugs
...
- add milestones arguments
- change decay_epochs to milestones variable
3 years ago
okojoalg
578d52e752
Add Sequencer
3 years ago
Ross Wightman
6d4665bb52
Merge pull request #1245 from rwightman/vit_relpos_refactor
...
Vision Transformer refactoring and Rel Pos impl
3 years ago
Ross Wightman
f5ca4141f7
Adjust arg order for recent vit model args, add a few comments
3 years ago
Ross Wightman
41dc49a337
Vision Transformer refactoring and Rel Pos impl
3 years ago
Ross Wightman
9c321be330
Merge pull request #1239 from dedeswim/parser-fix
...
Add support for different TFDS `BuilderConfig`s
3 years ago
Edoardo Debenedetti
c76d772670
Add support for different TFDS `BuilderConfig`s
3 years ago
Ross Wightman
b7cb8d0337
Add Swin-V2 Small-NS weights (83.5 @ 224). Add layer scale like 'init_values' via post-norm LN weight scaling
3 years ago
Ross Wightman
001688dabf
Merge pull request #1233 from jjsjann123/nhwc_cond_conv2d
...
fixing channels_last on cond_conv2d; update nvfuser debug env variable
3 years ago
jjsjann123
f88c606fcf
fixing channels_last on cond_conv2d; update nvfuser debug env variable
3 years ago
Ross Wightman
7d235c5a5f
Merge pull request #1230 from donglixp/patch-1
...
migrate azure blob for beit checkpoints
3 years ago
Li Dong
09e9f3defb
migrate azure blob for beit checkpoints
...
## Motivation
We are going to use a new blob account to store the checkpoints.
## Modification
Modify the azure blob storage URLs for BEiT checkpoints.
3 years ago
Ross Wightman
52ac881402
Missed first_conv in latest seresnext 'D' default_cfgs
3 years ago
Ross Wightman
7629d8264d
Add two new SE-ResNeXt101-D 32x8d weights, one anti-aliased and one not. Reshuffle default_cfgs vs model entrypoints for resnet.py so they are better aligned.
3 years ago
Ross Wightman
fbf597049c
Update README and change timmdocs link in documentation
3 years ago
Ross Wightman
01a0e25a67
Merge pull request #1208 from seefun/master
3 years ago
SeeFun
8f0bc0591e
fix convnext args
3 years ago
SeeFun
b0d2fcf647
Merge branch 'rwightman:master' into master
3 years ago
Ross Wightman
754e11402a
Merge remote-tracking branch 'origin/master' into bits_and_tpu
3 years ago
Ross Wightman
eac2df3d2c
Update PyTorch 1.10 benchmark numbers for latest code
3 years ago
Ross Wightman
c5a8e929fb
Add initial swinv2 tiny / small weights
3 years ago
Ross Wightman
83d7a11eec
Update .gitignore, remove out of date notebooks
3 years ago
Ross Wightman
02b806e00a
Add PyTorch 1.11 train benchmark numbers
3 years ago
Ross Wightman
c9f208f7f8
Update eval results and add latest PyTorch 1.11 inference benchmarks
3 years ago
Ross Wightman
1ba0ec4c18
Merge remote-tracking branch 'origin/master' into bits_and_tpu
3 years ago
Ross Wightman
f670d98cb8
Make a few more layers symbolically traceable (remove from FX leaf modules)
...
* remove dtype kwarg from .to() calls in EvoNorm as it messed up script + trace combo
* BatchNormAct2d always uses custom forward (cut & paste from original) instead of super().forward. Fixes #1176
* BlurPool groups==channels, no need to use input.dim[1]
3 years ago
Ross Wightman
a9ecb880e5
Merge pull request #1190 from seefun/ConvNeXt-pretrain
...
Add ConvNeXt tiny and small pretrain in22k
3 years ago
SeeFun
5f4de2334b
Merge pull request #1 from seefun/ConvNeXt-pretrain
...
Add ConvNeXt tiny and small pretrain in22k
3 years ago
SeeFun
ec4e9aa5a0
Add ConvNeXt tiny and small pretrain in22k
...
Add ConvNeXt tiny and small pretrain in22k from ConvNeXt repo:
06f7b05f92
3 years ago
Ross Wightman
73ffade1f8
Update README.md
3 years ago
Ross Wightman
575924ed60
Update test crop for new RegNet-V weights to match Y
3 years ago
Ross Wightman
8f6d638887
Update README.md
3 years ago
Ross Wightman
1618527098
Add layer scale and parallel blocks to vision_transformer
3 years ago
Ross Wightman
c42be74621
Add attrib / comments about Swin-S3 (AutoFormerV2) weights
3 years ago
Ross Wightman
474ac906a2
Add 'head norm first' convnext_tiny_hnf weights
3 years ago