Commit Graph

76 Commits (26c400bf1e5e82de3841d993366fe832cd92f9f2)

Author SHA1 Message Date
Ross Wightman ff6a919cf5 Add --fast-norm arg to benchmark.py, train.py, validate.py 2 years ago
Ross Wightman ffaf97f813 MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies.. 2 years ago
Ross Wightman 6e559e9b5f Add MViT (Multi-Scale) V2 2 years ago
Ross Wightman c486aa71f8 Add GCViT 2 years ago
Ross Wightman fba6ecd39b Add EfficientFormer 2 years ago
Ross Wightman ff4a38e2c3 Add PyramidVisionTransformerV2 2 years ago
Ross Wightman 6064d16a2d Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix 3 years ago
Ross Wightman 879df47c0a Support BatchNormAct2d for sync-bn use. Fix 3 years ago
Ross Wightman 9a86b900fa Official SwinV2 models 3 years ago
Ross Wightman d07d015173
Merge pull request from okojoalg/sequencer 3 years ago
okojoalg 578d52e752 Add Sequencer 3 years ago
Ross Wightman 41dc49a337 Vision Transformer refactoring and Rel Pos impl 3 years ago
Ross Wightman b049a5c5c6 Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
Ross Wightman 372ad5fa0d Significant model refactor and additions: 3 years ago
Ross Wightman c6e4b7895a Swin V2 CR impl refactor. 3 years ago
Ross Wightman 58ba49c8ef Add MobileViT models (w/ ByobNet base). Close . 3 years ago
Ross Wightman 5f81d4de23 Move DeiT to own file, vit getting crowded. Working towards fixing , make pooling interface for transformers and mlp closer to convnets. Still working through some details... 3 years ago
Ross Wightman 95cfc9b3e8 Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
Ross Wightman abc9ba2544 Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
Ross Wightman 18934debc5 Add initial ConvNeXt impl (mods of official code) 3 years ago
ICLR Author 44d6d51668 Add ConvMixer 3 years ago
Ross Wightman 1c9284c640 Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now 3 years ago
Ross Wightman f1808e0970 Post crossvit merge cleanup, change model names to reflect input size, cleanup img size vs scale handling, fix tests 3 years ago
Richard Chen 7ab9d4555c add crossvit 3 years ago
Alexander Soare 623e8b8eb8 wip xcit 4 years ago
Alexander Soare 23bb72ce5e nested_transformer wip 4 years ago
Ross Wightman 742c2d5247 Add Gather-Excite and Global Context attn modules. Refactor existing SE-like attn for consistency and refactor byob/byoanet for less redundancy. 4 years ago
Ross Wightman 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. 4 years ago
Ross Wightman f45de37690 Merge branch 'master' into levit_visformer_rednet 4 years ago
Ross Wightman 306c86b668 Merge branch 'convit' of https://github.com/amaarora/pytorch-image-models into amaarora-convit 4 years ago
Aman Arora 40c506ba1e Add ConViT 4 years ago
李鑫杰 00548b8427 Add Twins 4 years ago
Ross Wightman ecc7552c5c Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup. 4 years ago
Ross Wightman 5fcddb96a8 Merge branch 'master' into cait 4 years ago
Ross Wightman 12efffa6b1 Initial MLP-Mixer attempt... 4 years ago
Ross Wightman 1daa15ecc3 Initial Cait commit. Still some cleanup to do. 4 years ago
morizin 1e3b6d4dfc
Update __init__.py 4 years ago
morizin fd022fd6a2
Update __init__.py 4 years ago
Ross Wightman d793deb51a Merge branch 'master' of https://github.com/iamhankai/pytorch-image-models into iamhankai-master 4 years ago
Ross Wightman f606c45c38 Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer 4 years ago
iamhankai de445e7827 Add GhostNet 4 years ago
Ross Wightman 16f7aa9f54 Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models 4 years ago
Ross Wightman ce62f96d4d ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments 4 years ago
Ross Wightman a5310a3451 Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 4 years ago
Ross Wightman 0dfc5a66bb Add PiT model from https://github.com/naver-ai/pit 4 years ago
Ross Wightman 51febd869b Small tweak to tests for tnt model, reorder model imports. 4 years ago
contrastive 4a09bc851e Add TNT model 4 years ago
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs. 4 years ago
AFLALO, Jonathan Isaac 82c1fe958a added hardcorenas model 4 years ago
Ross Wightman 4f49b94311 Initial AGC impl. Still testing. 4 years ago