c3be79a8b7
Rename _pretrained.py -> pretrained.py, not feasible to change the other files to same scheme without breaking uses
1 year ago
23b357f1df
Add ported Tensorflow MaxVit weights. Add a few more CLIP ViT fine-tunes. Tweak some model tag names. Improve model tag name sorting. Update HF hub push config layout.
1 year ago
ebb99a1f8d
Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.
2 years ago
ff6a919cf5
Add --fast-norm arg to benchmark.py, train.py, validate.py
2 years ago
ffaf97f813
MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies..
2 years ago
6e559e9b5f
Add MViT (Multi-Scale) V2
2 years ago
c486aa71f8
Add GCViT
2 years ago
fba6ecd39b
Add EfficientFormer
2 years ago
ff4a38e2c3
Add PyramidVisionTransformerV2
2 years ago
6064d16a2d
Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320
...
* edgenext refactored for torchscript compat, stage base organization
* slight refactor of ConvNeXt to match some EdgeNeXt additions
* remove use of funky LayerNorm layer in ConvNeXt and just use nn.LayerNorm and LayerNorm2d (permute)
2 years ago
879df47c0a
Support BatchNormAct2d for sync-bn use. Fix #1254
2 years ago
9a86b900fa
Official SwinV2 models
2 years ago
d07d015173
Merge pull request #1249 from okojoalg/sequencer
...
Add Sequencer
2 years ago
578d52e752
Add Sequencer
2 years ago
41dc49a337
Vision Transformer refactoring and Rel Pos impl
2 years ago
b049a5c5c6
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2 years ago
372ad5fa0d
Significant model refactor and additions:
...
* All models updated with revised foward_features / forward_head interface
* Vision transformer and MLP based models consistently output sequence from forward_features (pooling or token selection considered part of 'head')
* WIP param grouping interface to allow consistent grouping of parameters for layer-wise decay across all model types
* Add gradient checkpointing support to a significant % of models, especially popular architectures
* Formatting and interface consistency improvements across models
* layer-wise LR decay impl part of optimizer factory w/ scale support in scheduler
* Poolformer and Volo architectures added
2 years ago
c6e4b7895a
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
2 years ago
58ba49c8ef
Add MobileViT models (w/ ByobNet base). Close #1038 .
2 years ago
5f81d4de23
Move DeiT to own file, vit getting crowded. Working towards fixing #1029 , make pooling interface for transformers and mlp closer to convnets. Still working through some details...
2 years ago
95cfc9b3e8
Merge remote-tracking branch 'origin/master' into norm_norm_norm
2 years ago
abc9ba2544
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2 years ago
18934debc5
Add initial ConvNeXt impl (mods of official code)
2 years ago
44d6d51668
Add ConvMixer
3 years ago
1c9284c640
Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now
3 years ago
f1808e0970
Post crossvit merge cleanup, change model names to reflect input size, cleanup img size vs scale handling, fix tests
3 years ago
7ab9d4555c
add crossvit
3 years ago
623e8b8eb8
wip xcit
3 years ago
23bb72ce5e
nested_transformer wip
3 years ago
742c2d5247
Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy.
3 years ago
bfc72f75d3
Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.
3 years ago
f45de37690
Merge branch 'master' into levit_visformer_rednet
3 years ago
306c86b668
Merge branch 'convit' of https://github.com/amaarora/pytorch-image-models into amaarora-convit
3 years ago
40c506ba1e
Add ConViT
3 years ago
00548b8427
Add Twins
3 years ago
ecc7552c5c
Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.
3 years ago
5fcddb96a8
Merge branch 'master' into cait
3 years ago
12efffa6b1
Initial MLP-Mixer attempt...
3 years ago
1daa15ecc3
Initial Cait commit. Still some cleanup to do.
3 years ago
1e3b6d4dfc
Update __init__.py
3 years ago
fd022fd6a2
Update __init__.py
3 years ago
d793deb51a
Merge branch 'master' of https://github.com/iamhankai/pytorch-image-models into iamhankai-master
3 years ago
f606c45c38
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
3 years ago
de445e7827
Add GhostNet
3 years ago
16f7aa9f54
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
3 years ago
ce62f96d4d
ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments
3 years ago
a5310a3451
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
3 years ago
0dfc5a66bb
Add PiT model from https://github.com/naver-ai/pit
3 years ago
51febd869b
Small tweak to tests for tnt model, reorder model imports.
3 years ago
4a09bc851e
Add TNT model
3 years ago
Ross Wightman
Ross Wightman
okojoalg
ICLR Author
Richard Chen
Alexander Soare
Aman Arora
李鑫杰
morizin
iamhankai
contrastive