timm (pytorch-image-models)

May 14, 2024

• Support loading PaliGemma jax weights into SigLIP ViT models with average pooling.
• Add Hiera models from Meta (https://github.com/facebookresearch/hiera).
• Add normalize= flag for transorms, return non-normalized torch.Tensor with original dytpe (for chug)
• Version 1.0.3 release

May 11, 2024

• Searching for Better ViT Baselines (For the GPU Poor) weights and vit variants released. Exploring model shapes between Tiny and Base.

• AttentionExtract helper added to extract attention maps from timm models. See example in https://github.com/huggingface/pytorch-image-models/discussions/1232#discussioncomment-9320949
• forward_intermediates() API refined and added to more models including some ConvNets that have other extraction methods.
• 1017 of 1047 model architectures support features_only=True feature extraction. Remaining 34 architectures can be supported but based on priority requests.
• Remove torch.jit.script annotated functions including old JIT activations. Conflict with dynamo and dynamo does a much better job when used.

April 11, 2024

• Prepping for a long overdue 1.0 release, things have been stable for a while now.
• Significant feature that's been missing for a while, features_only=True support for ViT models with flat hidden states or non-std module layouts (so far covering 'vit_*', 'twins_*', 'deit*', 'beit*', 'mvitv2*', 'eva*', 'samvit_*', 'flexivit*')
• Above feature support achieved through a new forward_intermediates() API that can be used with a feature wrapping module or direclty.

model = timm.create_model('vit_base_patch16_224')
final_feat, intermediates = model.forward_intermediates(input) 
output = model.forward_head(final_feat)  # pooling + classifier head

print(final_feat.shape)
torch.Size([2, 197, 768])

for f in intermediates:
    print(f.shape)
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])
torch.Size([2, 768, 14, 14])

print(output.shape)
torch.Size([2, 1000])

model = timm.create_model('eva02_base_patch16_clip_224', pretrained=True, img_size=512, features_only=True, out_indices=(-3, -2,))
output = model(torch.randn(2, 3, 512, 512))

for o in output:    
    print(o.shape)   
torch.Size([2, 768, 32, 32])
torch.Size([2, 768, 32, 32])

• TinyCLIP vision tower weights added, thx Thien Tran

Feb 19, 2024

• Next-ViT models added. Adapted from https://github.com/bytedance/Next-ViT
• HGNet and PP-HGNetV2 models added. Adapted from https://github.com/PaddlePaddle/PaddleClas by SeeFun
• Removed setup.py, moved to pyproject.toml based build supported by PDM
• Add updated model EMA impl using _for_each for less overhead
• Support device args in train script for non GPU devices
• Other misc fixes and small additions
• Min supported Python version increased to 3.8
• Release 0.9.16

Jan 8, 2024

Datasets & transform refactoring

• HuggingFace streaming (iterable) dataset support (--dataset hfids:org/dataset)
• Webdataset wrapper tweaks for improved split info fetching, can auto fetch splits from supported HF hub webdataset
• Tested HF datasets and webdataset wrapper streaming from HF hub with recent timm ImageNet uploads to https://huggingface.co/timm
• Make input & target column/field keys consistent across datasets and pass via args
• Full monochrome support when using e:g: --input-size 1 224 224 or --in-chans 1, sets PIL image conversion appropriately in dataset
• Improved several alternate crop & resize transforms (ResizeKeepRatio, RandomCropOrPad, etc) for use in PixParse document AI project
• Add SimCLR style color jitter prob along with grayscale and gaussian blur options to augmentations and args
• Allow train without validation set (--val-split '') in train script
• Add --bce-sum (sum over class dim) and --bce-pos-weight (positive weighting) args for training as they're common BCE loss tweaks I was often hard coding

Nov 23, 2023

• Added EfficientViT-Large models, thanks SeeFun
• Fix Python 3.7 compat, will be dropping support for it soon
• Other misc fixes
• Release 0.9.12

Nov 20, 2023

• Added significant flexibility for Hugging Face Hub based timm models via model_args config entry. model_args will be passed as kwargs through to models on creation.
  • See example at https://huggingface.co/gaunernst/vit_base_patch16_1024_128.audiomae_as2m_ft_as20k/blob/main/config.json
  • Usage: https://github.com/huggingface/pytorch-image-models/discussions/2035
• Updated imagenet eval and test set csv files with latest models
• vision_transformer.py typing and doc cleanup by Laureηt
• 0.9.11 release

Nov 4

• Patch fix for 0.9.9 to fix FrozenBatchnorm2d import path for old torchvision (~2 years )

Nov 3, 2023

• DFN (Data Filtering Networks) and MetaCLIP ViT weights added
• DINOv2 'register' ViT model weights added
• Add quickgelu ViT variants for OpenAI, DFN, MetaCLIP weights that use it (less efficient)
• Improved typing added to ResNet, MobileNet-v3 thanks to Aryan
• ImageNet-12k fine-tuned (from LAION-2B CLIP) convnext_xxlarge
• 0.9.9 release

Nov 3, 2023

• DFN (Data Filtering Networks) and MetaCLIP ViT weights added
• DINOv2 'register' ViT model weights added
• Add quickgelu ViT variants for OpenAI, DFN, MetaCLIP weights that use it (less efficient)
• Improved typing added to ResNet, MobileNet-v3 thanks to Aryan
• ImageNet-12k fine-tuned (from LAION-2B CLIP) convnext_xxlarge
• 0.9.9 release

Oct 20, 2023

• SigLIP image tower weights supported in vision_transformer.py.
  • Great potential for fine-tune and downstream feature use.
• Experimental 'register' support in vit models as per Vision Transformers Need Registers
• Updated RepViT with new weight release. Thanks wangao
• Add patch resizing support (on pretrained weight load) to Swin models
• 0.9.8 release

Small bug fix & extra model from v0.9.6

Sep 1, 2023

• TinyViT added by SeeFun
• Fix EfficientViT (MIT) to use torch.autocast so it works back to PT 1.10
• 0.9.7 release

Aug 28, 2023

• Add dynamic img size support to models in vision_transformer.py, vision_transformer_hybrid.py, deit.py, and eva.py w/o breaking backward compat.
  • Add dynamic_img_size=True to args at model creation time to allow changing the grid size (interpolate abs and/or ROPE pos embed each forward pass).
  • Add dynamic_img_pad=True to allow image sizes that aren't divisible by patch size (pad bottom right to patch size each forward pass).
  • Enabling either dynamic mode will break FX tracing unless PatchEmbed module added as leaf.
  • Existing method of resizing position embedding by passing different img_size (interpolate pretrained embed weights once) on creation still works.
  • Existing method of changing patch_size (resize pretrained patch_embed weights once) on creation still works.
  • Example validation cmd python validate.py /imagenet --model vit_base_patch16_224 --amp --amp-dtype bfloat16 --img-size 255 --crop-pct 1.0 --model-kwargs dynamic_img_size=True dyamic_img_pad=True

Aug 25, 2023

• Many new models since last release
  • FastViT - https://arxiv.org/abs/2303.14189
  • MobileOne - https://arxiv.org/abs/2206.04040
  • InceptionNeXt - https://arxiv.org/abs/2303.16900
  • RepGhostNet - https://arxiv.org/abs/2211.06088 (thanks https://github.com/ChengpengChen)
  • GhostNetV2 - https://arxiv.org/abs/2211.12905 (thanks https://github.com/yehuitang)
  • EfficientViT (MSRA) - https://arxiv.org/abs/2305.07027 (thanks https://github.com/seefun)
  • EfficientViT (MIT) - https://arxiv.org/abs/2205.14756 (thanks https://github.com/seefun)
• Add --reparam arg to benchmark.py, onnx_export.py, and validate.py to trigger layer reparameterization / fusion for models with any one of reparameterize(), switch_to_deploy() or fuse()
  • Including FastViT, MobileOne, RepGhostNet, EfficientViT (MSRA), RepViT, RepVGG, and LeViT
• Preparing 0.9.6 'back to school' release

Aug 11, 2023

• Swin, MaxViT, CoAtNet, and BEiT models support resizing of image/window size on creation with adaptation of pretrained weights
• Example validation cmd to test w/ non-square resize python validate.py /imagenet --model swin_base_patch4_window7_224.ms_in22k_ft_in1k --amp --amp-dtype bfloat16 --input-size 3 256 320 --model-kwargs window_size=8,10 img_size=256,320

Minor updates and bug fixes. New ResNeXT w/ highest ImageNet eval I'm aware of in the ResNe(X)t family (seresnextaa201d_32x8d.sw_in12k_ft_in1k_384)

Aug 3, 2023

• Add GluonCV weights for HRNet w18_small and w18_small_v2. Converted by SeeFun
• Fix selecsls* model naming regression
• Patch and position embedding for ViT/EVA works for bfloat16/float16 weights on load (or activations for on-the-fly resize)
• v0.9.5 release prep

July 27, 2023

• Added timm trained seresnextaa201d_32x8d.sw_in12k_ft_in1k_384 weights (and .sw_in12k pretrain) with 87.3% top-1 on ImageNet-1k, best ImageNet ResNet family model I'm aware of.
• RepViT model and weights (https://arxiv.org/abs/2307.09283) added by wangao
• I-JEPA ViT feature weights (no classifier) added by SeeFun
• SAM-ViT (segment anything) feature weights (no classifier) added by SeeFun
• Add support for alternative feat extraction methods and -ve indices to EfficientNet
• Add NAdamW optimizer
• Misc fixes

• Fix _hub deprecation pass through import

The first non pre-release since Oct 2022 with a long list of changes from 0.6.x releases...

May 12, 2023

• Fix Python 3.7 import error re Final[] typing annotation

May 11, 2023

• timm 0.9 released, transition from 0.8.xdev releases

May 10, 2023

• Hugging Face Hub downloading is now default, 1132 models on https://huggingface.co/timm, 1163 weights in timm
• DINOv2 vit feature backbone weights added thanks to Leng Yue
• FB MAE vit feature backbone weights added
• OpenCLIP DataComp-XL L/14 feat backbone weights added
• MetaFormer (poolformer-v2, caformer, convformer, updated poolformer (v1)) w/ weights added by Fredo Guan
• Experimental get_intermediate_layers function on vit/deit models for grabbing hidden states (inspired by DINO impl). This is WIP and may change significantly... feedback welcome.
• Model creation throws error if pretrained=True and no weights exist (instead of continuing with random initialization)
• Fix regression with inception / nasnet TF sourced weights with 1001 classes in original classifiers
• bitsandbytes (https://github.com/TimDettmers/bitsandbytes) optimizers added to factory, use bnb prefix, ie bnbadam8bit
• Misc cleanup and fixes
• Final testing before switching to a 0.9 and bringing timm out of pre-release state

April 27, 2023

• 97% of timm models uploaded to HF Hub and almost all updated to support multi-weight pretrained configs
• Minor cleanup and refactoring of another batch of models as multi-weight added. More fused_attn (F.sdpa) and features_only support, and torchscript fixes.

April 21, 2023

• Gradient accumulation support added to train script and tested (--grad-accum-steps), thanks Taeksang Kim
• More weights on HF Hub (cspnet, cait, volo, xcit, tresnet, hardcorenas, densenet, dpn, vovnet, xception_aligned)
• Added --head-init-scale and --head-init-bias to train.py to scale classiifer head and set fixed bias for fine-tune
• Remove all InplaceABN (inplace_abn) use, replaced use in tresnet with standard BatchNorm (modified weights accordingly).

April 12, 2023

• Add ONNX export script, validate script, helpers that I've had kicking around for along time. Tweak 'same' padding for better export w/ recent ONNX + pytorch.
• Refactor dropout args for vit and vit-like models, separate drop_rate into drop_rate (classifier dropout), proj_drop_rate (block mlp / out projections), pos_drop_rate (position embedding drop), attn_drop_rate (attention dropout). Also add patch dropout (FLIP) to vit and eva models.
• fused F.scaled_dot_product_attention support to more vit models, add env var (TIMM_FUSED_ATTN) to control, and config interface to enable/disable
• Add EVA-CLIP backbones w/ image tower weights, all the way up to 4B param 'enormous' model, and 336x336 OpenAI ViT mode that was missed.

April 5, 2023

• ALL ResNet models pushed to Hugging Face Hub with multi-weight support
  • All past timm trained weights added with recipe based tags to differentiate
  • All ResNet strikes back A1/A2/A3 (seed 0) and R50 example B/C1/C2/D weights available
  • Add torchvision v2 recipe weights to existing torchvision originals
  • See comparison table in https://huggingface.co/timm/seresnextaa101d_32x8d.sw_in12k_ft_in1k_288#model-comparison
• New ImageNet-12k + ImageNet-1k fine-tunes available for a few anti-aliased ResNet models
  • resnetaa50d.sw_in12k_ft_in1k - 81.7 @ 224, 82.6 @ 288
  • resnetaa101d.sw_in12k_ft_in1k - 83.5 @ 224, 84.1 @ 288
  • seresnextaa101d_32x8d.sw_in12k_ft_in1k - 86.0 @ 224, 86.5 @ 288
  • seresnextaa101d_32x8d.sw_in12k_ft_in1k_288 - 86.5 @ 288, 86.7 @ 320

March 31, 2023

• Add first ConvNext-XXLarge CLIP -> IN-1k fine-tune and IN-12k intermediate fine-tunes for convnext-base/large CLIP models.

• Add EVA-02 MIM pretrained and fine-tuned weights, push to HF hub and update model cards for all EVA models. First model over 90% top-1 (99% top-5)! Check out the original code & weights at https://github.com/baaivision/EVA for more details on their work blending MIM, CLIP w/ many model, dataset, and train recipe tweaks.

• Multi-weight and HF hub for DeiT and MLP-Mixer based models

March 22, 2023

• More weights pushed to HF hub along with multi-weight support, including: regnet.py, rexnet.py, byobnet.py, resnetv2.py, swin_transformer.py, swin_transformer_v2.py, swin_transformer_v2_cr.py
• Swin Transformer models support feature extraction (NCHW feat maps for swinv2_cr_*, and NHWC for all others) and spatial embedding outputs.
• FocalNet (from https://github.com/microsoft/FocalNet) models and weights added with significant refactoring, feature extraction, no fixed resolution / sizing constraint
• RegNet weights increased with HF hub push, SWAG, SEER, and torchvision v2 weights. SEER is pretty poor wrt to performance for model size, but possibly useful.
• More ImageNet-12k pretrained and 1k fine-tuned timm weights:
  • rexnetr_200.sw_in12k_ft_in1k - 82.6 @ 224, 83.2 @ 288
  • rexnetr_300.sw_in12k_ft_in1k - 84.0 @ 224, 84.5 @ 288
  • regnety_120.sw_in12k_ft_in1k - 85.0 @ 224, 85.4 @ 288
  • regnety_160.lion_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288
  • regnety_160.sw_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288 (compare to SWAG PT + 1k FT this is same BUT much lower res, blows SEER FT away)
• Model name deprecation + remapping functionality added (a milestone for bringing 0.8.x out of pre-release). Mappings being added...
• Minor bug fixes and improvements.

Feb 26, 2023

• Add ConvNeXt-XXLarge CLIP pretrained image tower weights for fine-tune & features (fine-tuning TBD) -- see model card
• Update convnext_xxlarge default LayerNorm eps to 1e-5 (for CLIP weights, improved stability)
• 0.8.15dev0

Feb 20, 2023

• Add 320x320 convnext_large_mlp.clip_laion2b_ft_320 and convnext_lage_mlp.clip_laion2b_ft_soup_320 CLIP image tower weights for features & fine-tune
• 0.8.13dev0 pypi release for latest changes w/ move to huggingface org

Feb 16, 2023

• safetensor checkpoint support added
• Add ideas from 'Scaling Vision Transformers to 22 B. Params' (https://arxiv.org/abs/2302.05442) -- qk norm, RmsNorm, parallel block
• Add F.scaled_dot_product_attention support (PyTorch 2.0 only) to vit_*, vit_relpos*, coatnet / maxxvit (to start)
• Lion optimizer (w/ multi-tensor option) added (https://arxiv.org/abs/2302.06675)
• gradient checkpointing works with features_only=True

Feb 7, 2023

• New inference benchmark numbers added in results folder.
• Add convnext LAION CLIP trained weights and initial set of in1k fine-tunes
  • convnext_base.clip_laion2b_augreg_ft_in1k - 86.2% @ 256x256
  • convnext_base.clip_laiona_augreg_ft_in1k_384 - 86.5% @ 384x384
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k - 87.3% @ 256x256
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384 - 87.9% @ 384x384
• Add DaViT models. Supports features_only=True. Adapted from https://github.com/dingmyu/davit by Fredo.
• Use a common NormMlpClassifierHead across MaxViT, ConvNeXt, DaViT
• Add EfficientFormer-V2 model, update EfficientFormer, and refactor LeViT (closely related architectures). Weights on HF hub.
  • New EfficientFormer-V2 arch, significant refactor from original at (https://github.com/snap-research/EfficientFormer). Supports features_only=True.
  • Minor updates to EfficientFormer.
  • Refactor LeViT models to stages, add features_only=True support to new conv variants, weight remap required.
• Move ImageNet meta-data (synsets, indices) from /results to timm/data/_info.
• Add ImageNetInfo / DatasetInfo classes to provide labelling for various ImageNet classifier layouts in timm
  • Update inference.py to use, try: python inference.py /folder/to/images --model convnext_small.in12k --label-type detail --topk 5
• Ready for 0.8.10 pypi pre-release (final testing).

Jan 20, 2023

• Add two convnext 12k -> 1k fine-tunes at 384x384

  • convnext_tiny.in12k_ft_in1k_384 - 85.1 @ 384
  • convnext_small.in12k_ft_in1k_384 - 86.2 @ 384

• Push all MaxxViT weights to HF hub, and add new ImageNet-12k -> 1k fine-tunes for rw base MaxViT and CoAtNet 1/2 models

Jan 11, 2023

• Update ConvNeXt ImageNet-12k pretrain series w/ two new fine-tuned weights (and pre FT .in12k tags)
  • convnext_nano.in12k_ft_in1k - 82.3 @ 224, 82.9 @ 288 (previously released)
  • convnext_tiny.in12k_ft_in1k - 84.2 @ 224, 84.5 @ 288
  • convnext_small.in12k_ft_in1k - 85.2 @ 224, 85.3 @ 288

Jan 6, 2023

• Finally got around to adding --model-kwargs and --opt-kwargs to scripts to pass through rare args directly to model classes from cmd line
  • train.py /imagenet --model resnet50 --amp --model-kwargs output_stride=16 act_layer=silu
  • train.py /imagenet --model vit_base_patch16_clip_224 --img-size 240 --amp --model-kwargs img_size=240 patch_size=12
• Cleanup some popular models to better support arg passthrough / merge with model configs, more to go.

Jan 5, 2023

• ConvNeXt-V2 models and weights added to existing convnext.py
  • Paper: ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders
  • Reference impl: https://github.com/facebookresearch/ConvNeXt-V2 (NOTE: weights currently CC-BY-NC)

Dec 23, 2022 🎄☃

• Add FlexiViT models and weights from https://github.com/google-research/big_vision (check out paper at https://arxiv.org/abs/2212.08013)
  • NOTE currently resizing is static on model creation, on-the-fly dynamic / train patch size sampling is a WIP
• Many more models updated to multi-weight and downloadable via HF hub now (convnext, efficientnet, mobilenet, vision_transformer*, beit)
• More model pretrained tag and adjustments, some model names changed (working on deprecation translations, consider main branch DEV branch right now, use 0.6.x for stable use)
• More ImageNet-12k (subset of 22k) pretrain models popping up:
  • efficientnet_b5.in12k_ft_in1k - 85.9 @ 448x448
  • vit_medium_patch16_gap_384.in12k_ft_in1k - 85.5 @ 384x384
  • vit_medium_patch16_gap_256.in12k_ft_in1k - 84.5 @ 256x256
  • convnext_nano.in12k_ft_in1k - 82.9 @ 288x288

Dec 8, 2022

• Add 'EVA l' to vision_transformer.py, MAE style ViT-L/14 MIM pretrain w/ EVA-CLIP targets, FT on ImageNet-1k (w/ ImageNet-22k intermediate for some)
  • original source: https://github.com/baaivision/EVA

Dec 6, 2022

• Add 'EVA g', BEiT style ViT-g/14 model weights w/ both MIM pretrain and CLIP pretrain to beit.py.
  • original source: https://github.com/baaivision/EVA
  • paper: https://arxiv.org/abs/2211.07636

Dec 5, 2022

• Pre-release (0.8.0dev0) of multi-weight support (model_arch.pretrained_tag). Install with pip install --pre timm
  • vision_transformer, maxvit, convnext are the first three model impl w/ support
  • model names are changing with this (previous _21k, etc. fn will merge), still sorting out deprecation handling
  • bugs are likely, but I need feedback so please try it out
  • if stability is needed, please use 0.6.x pypi releases or clone from 0.6.x branch
• Support for PyTorch 2.0 compile is added in train/validate/inference/benchmark, use --torchcompile argument
• Inference script allows more control over output, select k for top-class index + prob json, csv or parquet output
• Add a full set of fine-tuned CLIP image tower weights from both LAION-2B and original OpenAI CLIP models

• Port of MaxViT Tensorflow Weights from official impl at https://github.com/google-research/maxvit
  • There was larger than expected drops for the upscaled 384/512 in21k fine-tune weights, possible detail missing, but the 21k FT did seem sensitive to small preprocessing

Oct 15, 2022

• Train and validation script enhancements
• Non-GPU (ie CPU) device support
• SLURM compatibility for train script
• HF datasets support (via ReaderHfds)
• TFDS/WDS dataloading improvements (sample padding/wrap for distributed use fixed wrt sample count estimate)
• in_chans !=3 support for scripts / loader
• Adan optimizer
• Can enable per-step LR scheduling via args
• Dataset 'parsers' renamed to 'readers', more descriptive of purpose
• AMP args changed, APEX via --amp-impl apex, bfloat16 supportedf via --amp-dtype bfloat16
• main branch switched to 0.7.x version, 0.6x forked for stable release of weight only adds
• master -> main branch rename

First non pre-release in a loooong while, changelog from 0.6.x below...

May 11, 2023

• timm 0.9 released, transition from 0.8.xdev releases

May 10, 2023

• Hugging Face Hub downloading is now default, 1132 models on https://huggingface.co/timm, 1163 weights in timm
• DINOv2 vit feature backbone weights added thanks to Leng Yue
• FB MAE vit feature backbone weights added
• OpenCLIP DataComp-XL L/14 feat backbone weights added
• MetaFormer (poolformer-v2, caformer, convformer, updated poolformer (v1)) w/ weights added by Fredo Guan
• Experimental get_intermediate_layers function on vit/deit models for grabbing hidden states (inspired by DINO impl). This is WIP and may change significantly... feedback welcome.
• Model creation throws error if pretrained=True and no weights exist (instead of continuing with random initialization)
• Fix regression with inception / nasnet TF sourced weights with 1001 classes in original classifiers
• bitsandbytes (https://github.com/TimDettmers/bitsandbytes) optimizers added to factory, use bnb prefix, ie bnbadam8bit
• Misc cleanup and fixes
• Final testing before switching to a 0.9 and bringing timm out of pre-release state

April 27, 2023

• 97% of timm models uploaded to HF Hub and almost all updated to support multi-weight pretrained configs
• Minor cleanup and refactoring of another batch of models as multi-weight added. More fused_attn (F.sdpa) and features_only support, and torchscript fixes.

April 21, 2023

• Gradient accumulation support added to train script and tested (--grad-accum-steps), thanks Taeksang Kim
• More weights on HF Hub (cspnet, cait, volo, xcit, tresnet, hardcorenas, densenet, dpn, vovnet, xception_aligned)
• Added --head-init-scale and --head-init-bias to train.py to scale classiifer head and set fixed bias for fine-tune
• Remove all InplaceABN (inplace_abn) use, replaced use in tresnet with standard BatchNorm (modified weights accordingly).

April 12, 2023

• Add ONNX export script, validate script, helpers that I've had kicking around for along time. Tweak 'same' padding for better export w/ recent ONNX + pytorch.
• Refactor dropout args for vit and vit-like models, separate drop_rate into drop_rate (classifier dropout), proj_drop_rate (block mlp / out projections), pos_drop_rate (position embedding drop), attn_drop_rate (attention dropout). Also add patch dropout (FLIP) to vit and eva models.
• fused F.scaled_dot_product_attention support to more vit models, add env var (TIMM_FUSED_ATTN) to control, and config interface to enable/disable
• Add EVA-CLIP backbones w/ image tower weights, all the way up to 4B param 'enormous' model, and 336x336 OpenAI ViT mode that was missed.

April 5, 2023

• ALL ResNet models pushed to Hugging Face Hub with multi-weight support
  • All past timm trained weights added with recipe based tags to differentiate
  • All ResNet strikes back A1/A2/A3 (seed 0) and R50 example B/C1/C2/D weights available
  • Add torchvision v2 recipe weights to existing torchvision originals
  • See comparison table in https://huggingface.co/timm/seresnextaa101d_32x8d.sw_in12k_ft_in1k_288#model-comparison
• New ImageNet-12k + ImageNet-1k fine-tunes available for a few anti-aliased ResNet models
  • resnetaa50d.sw_in12k_ft_in1k - 81.7 @ 224, 82.6 @ 288
  • resnetaa101d.sw_in12k_ft_in1k - 83.5 @ 224, 84.1 @ 288
  • seresnextaa101d_32x8d.sw_in12k_ft_in1k - 86.0 @ 224, 86.5 @ 288
  • seresnextaa101d_32x8d.sw_in12k_ft_in1k_288 - 86.5 @ 288, 86.7 @ 320

March 31, 2023

• Add first ConvNext-XXLarge CLIP -> IN-1k fine-tune and IN-12k intermediate fine-tunes for convnext-base/large CLIP models.

• Add EVA-02 MIM pretrained and fine-tuned weights, push to HF hub and update model cards for all EVA models. First model over 90% top-1 (99% top-5)! Check out the original code & weights at https://github.com/baaivision/EVA for more details on their work blending MIM, CLIP w/ many model, dataset, and train recipe tweaks.

• Multi-weight and HF hub for DeiT and MLP-Mixer based models

March 22, 2023

• More weights pushed to HF hub along with multi-weight support, including: regnet.py, rexnet.py, byobnet.py, resnetv2.py, swin_transformer.py, swin_transformer_v2.py, swin_transformer_v2_cr.py
• Swin Transformer models support feature extraction (NCHW feat maps for swinv2_cr_*, and NHWC for all others) and spatial embedding outputs.
• FocalNet (from https://github.com/microsoft/FocalNet) models and weights added with significant refactoring, feature extraction, no fixed resolution / sizing constraint
• RegNet weights increased with HF hub push, SWAG, SEER, and torchvision v2 weights. SEER is pretty poor wrt to performance for model size, but possibly useful.
• More ImageNet-12k pretrained and 1k fine-tuned timm weights:
  • rexnetr_200.sw_in12k_ft_in1k - 82.6 @ 224, 83.2 @ 288
  • rexnetr_300.sw_in12k_ft_in1k - 84.0 @ 224, 84.5 @ 288
  • regnety_120.sw_in12k_ft_in1k - 85.0 @ 224, 85.4 @ 288
  • regnety_160.lion_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288
  • regnety_160.sw_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288 (compare to SWAG PT + 1k FT this is same BUT much lower res, blows SEER FT away)
• Model name deprecation + remapping functionality added (a milestone for bringing 0.8.x out of pre-release). Mappings being added...
• Minor bug fixes and improvements.

Feb 26, 2023

• Add ConvNeXt-XXLarge CLIP pretrained image tower weights for fine-tune & features (fine-tuning TBD) -- see model card
• Update convnext_xxlarge default LayerNorm eps to 1e-5 (for CLIP weights, improved stability)
• 0.8.15dev0

Feb 20, 2023

• Add 320x320 convnext_large_mlp.clip_laion2b_ft_320 and convnext_lage_mlp.clip_laion2b_ft_soup_320 CLIP image tower weights for features & fine-tune
• 0.8.13dev0 pypi release for latest changes w/ move to huggingface org

Feb 16, 2023

• safetensor checkpoint support added
• Add ideas from 'Scaling Vision Transformers to 22 B. Params' (https://arxiv.org/abs/2302.05442) -- qk norm, RmsNorm, parallel block
• Add F.scaled_dot_product_attention support (PyTorch 2.0 only) to vit_*, vit_relpos*, coatnet / maxxvit (to start)
• Lion optimizer (w/ multi-tensor option) added (https://arxiv.org/abs/2302.06675)
• gradient checkpointing works with features_only=True

Feb 7, 2023

• New inference benchmark numbers added in results folder.
• Add convnext LAION CLIP trained weights and initial set of in1k fine-tunes
  • convnext_base.clip_laion2b_augreg_ft_in1k - 86.2% @ 256x256
  • convnext_base.clip_laiona_augreg_ft_in1k_384 - 86.5% @ 384x384
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k - 87.3% @ 256x256
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384 - 87.9% @ 384x384
• Add DaViT models. Supports features_only=True. Adapted from https://github.com/dingmyu/davit by Fredo.
• Use a common NormMlpClassifierHead across MaxViT, ConvNeXt, DaViT
• Add EfficientFormer-V2 model, update EfficientFormer, and refactor LeViT (closely related architectures). Weights on HF hub.
  • New EfficientFormer-V2 arch, significant refactor from original at (https://github.com/snap-research/EfficientFormer). Supports features_only=True.
  • Minor updates to EfficientFormer.
  • Refactor LeViT models to stages, add features_only=True support to new conv variants, weight remap required.
• Move ImageNet meta-data (synsets, indices) from /results to timm/data/_info.
• Add ImageNetInfo / DatasetInfo classes to provide labelling for various ImageNet classifier layouts in timm
  • Update inference.py to use, try: python inference.py /folder/to/images --model convnext_small.in12k --label-type detail --topk 5
• Ready for 0.8.10 pypi pre-release (final testing).

Jan 20, 2023

• Add two convnext 12k -> 1k fine-tunes at 384x384

  • convnext_tiny.in12k_ft_in1k_384 - 85.1 @ 384
  • convnext_small.in12k_ft_in1k_384 - 86.2 @ 384

• Push all MaxxViT weights to HF hub, and add new ImageNet-12k -> 1k fine-tunes for rw base MaxViT and CoAtNet 1/2 models

Jan 11, 2023

• Update ConvNeXt ImageNet-12k pretrain series w/ two new fine-tuned weights (and pre FT .in12k tags)
  • convnext_nano.in12k_ft_in1k - 82.3 @ 224, 82.9 @ 288 (previously released)
  • convnext_tiny.in12k_ft_in1k - 84.2 @ 224, 84.5 @ 288
  • convnext_small.in12k_ft_in1k - 85.2 @ 224, 85.3 @ 288

Jan 6, 2023

• Finally got around to adding --model-kwargs and --opt-kwargs to scripts to pass through rare args directly to model classes from cmd line
  • train.py /imagenet --model resnet50 --amp --model-kwargs output_stride=16 act_layer=silu
  • train.py /imagenet --model vit_base_patch16_clip_224 --img-size 240 --amp --model-kwargs img_size=240 patch_size=12
• Cleanup some popular models to better support arg passthrough / merge with model configs, more to go.

Jan 5, 2023

• ConvNeXt-V2 models and weights added to existing convnext.py
  • Paper: ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders
  • Reference impl: https://github.com/facebookresearch/ConvNeXt-V2 (NOTE: weights currently CC-BY-NC)

Dec 23, 2022 🎄☃

• Add FlexiViT models and weights from https://github.com/google-research/big_vision (check out paper at https://arxiv.org/abs/2212.08013)
  • NOTE currently resizing is static on model creation, on-the-fly dynamic / train patch size sampling is a WIP
• Many more models updated to multi-weight and downloadable via HF hub now (convnext, efficientnet, mobilenet, vision_transformer*, beit)
• More model pretrained tag and adjustments, some model names changed (working on deprecation translations, consider main branch DEV branch right now, use 0.6.x for stable use)
• More ImageNet-12k (subset of 22k) pretrain models popping up:
  • efficientnet_b5.in12k_ft_in1k - 85.9 @ 448x448
  • vit_medium_patch16_gap_384.in12k_ft_in1k - 85.5 @ 384x384
  • vit_medium_patch16_gap_256.in12k_ft_in1k - 84.5 @ 256x256
  • convnext_nano.in12k_ft_in1k - 82.9 @ 288x288

Dec 8, 2022

• Add 'EVA l' to vision_transformer.py, MAE style ViT-L/14 MIM pretrain w/ EVA-CLIP targets, FT on ImageNet-1k (w/ ImageNet-22k intermediate for some)
  • original source: https://github.com/baaivision/EVA

Dec 6, 2022

• Add 'EVA g', BEiT style ViT-g/14 model weights w/ both MIM pretrain and CLIP pretrain to beit.py.
  • original source: https://github.com/baaivision/EVA
  • paper: https://arxiv.org/abs/2211.07636

Dec 5, 2022

• Pre-release (0.8.0dev0) of multi-weight support (model_arch.pretrained_tag). Install with pip install --pre timm
  • vision_transformer, maxvit, convnext are the first three model impl w/ support
  • model names are changing with this (previous _21k, etc. fn will merge), still sorting out deprecation handling
  • bugs are likely, but I need feedback so please try it out
  • if stability is needed, please use 0.6.x pypi releases or clone from 0.6.x branch
• Support for PyTorch 2.0 compile is added in train/validate/inference/benchmark, use --torchcompile argument
• Inference script allows more control over output, select k for top-class index + prob json, csv or parquet output
• Add a full set of fine-tuned CLIP image tower weights from both LAION-2B and original OpenAI CLIP models

• Port of MaxViT Tensorflow Weights from official impl at https://github.com/google-research/maxvit
  • There was larger than expected drops for the upscaled 384/512 in21k fine-tune weights, possible detail missing, but the 21k FT did seem sensitive to small preprocessing

Oct 15, 2022

• Train and validation script enhancements
• Non-GPU (ie CPU) device support
• SLURM compatibility for train script
• HF datasets support (via ReaderHfds)
• TFDS/WDS dataloading improvements (sample padding/wrap for distributed use fixed wrt sample count estimate)
• in_chans !=3 support for scripts / loader
• Adan optimizer
• Can enable per-step LR scheduling via args
• Dataset 'parsers' renamed to 'readers', more descriptive of purpose
• AMP args changed, APEX via --amp-impl apex, bfloat16 supportedf via --amp-dtype bfloat16
• main branch switched to 0.7.x version, 0.6x forked for stable release of weight only adds
• master -> main branch rename

Release from 0.6.x stable branch with fix for Python 3.11. NOTE original 0.6.13 release tag was against wrong branch.

March 22, 2023

• More weights pushed to HF hub along with multi-weight support, including: regnet.py, rexnet.py, byobnet.py, resnetv2.py, swin_transformer.py, swin_transformer_v2.py, swin_transformer_v2_cr.py
• Swin Transformer models support feature extraction (NCHW feat maps for swinv2_cr_*, and NHWC for all others) and spatial embedding outputs.
• FocalNet (from https://github.com/microsoft/FocalNet) models and weights added with significant refactoring, feature extraction, no fixed resolution / sizing constraint
• RegNet weights increased with HF hub push, SWAG, SEER, and torchvision v2 weights. SEER is pretty poor wrt to performance for model size, but possibly useful.
• More ImageNet-12k pretrained and 1k fine-tuned timm weights:
  • rexnetr_200.sw_in12k_ft_in1k - 82.6 @ 224, 83.2 @ 288
  • rexnetr_300.sw_in12k_ft_in1k - 84.0 @ 224, 84.5 @ 288
  • regnety_120.sw_in12k_ft_in1k - 85.0 @ 224, 85.4 @ 288
  • regnety_160.lion_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288
  • regnety_160.sw_in12k_ft_in1k - 85.6 @ 224, 86.0 @ 288 (compare to SWAG PT + 1k FT this is same BUT much lower res, blows SEER FT away)
• Model name deprecation + remapping functionality added (a milestone for bringing 0.8.x out of pre-release). Mappings being added...
• Minor bug fixes and improvements.

Feb 26, 2023

• Add ConvNeXt-XXLarge CLIP pretrained image tower weights for fine-tune & features (fine-tuning TBD) -- see model card
• Update convnext_xxlarge default LayerNorm eps to 1e-5 (for CLIP weights, improved stability)
• 0.8.15dev0

Feb 20, 2023

• Add 320x320 convnext_large_mlp.clip_laion2b_ft_320 and convnext_lage_mlp.clip_laion2b_ft_soup_320 CLIP image tower weights for features & fine-tune
• 0.8.13dev0 pypi release for latest changes w/ move to huggingface org

Feb 16, 2023

• safetensor checkpoint support added
• Add ideas from 'Scaling Vision Transformers to 22 B. Params' (https://arxiv.org/abs/2302.05442) -- qk norm, RmsNorm, parallel block
• Add F.scaled_dot_product_attention support (PyTorch 2.0 only) to vit_*, vit_relpos*, coatnet / maxxvit (to start)
• Lion optimizer (w/ multi-tensor option) added (https://arxiv.org/abs/2302.06675)

Feb 7, 2023

• New inference benchmark numbers added in results folder.
• Add convnext LAION CLIP trained weights and initial set of in1k fine-tunes
  • convnext_base.clip_laion2b_augreg_ft_in1k - 86.2% @ 256x256
  • convnext_base.clip_laiona_augreg_ft_in1k_384 - 86.5% @ 384x384
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k - 87.3% @ 256x256
  • convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384 - 87.9% @ 384x384
• Add DaViT models. Supports features_only=True. Adapted from https://github.com/dingmyu/davit by Fredo.
• Use a common NormMlpClassifierHead across MaxViT, ConvNeXt, DaViT
• Add EfficientFormer-V2 model, update EfficientFormer, and refactor LeViT (closely related architectures). Weights on HF hub.
  • New EfficientFormer-V2 arch, significant refactor from original at (https://github.com/snap-research/EfficientFormer). Supports features_only=True.
  • Minor updates to EfficientFormer.
  • Refactor LeViT models to stages, add features_only=True support to new conv variants, weight remap required.
• Move ImageNet meta-data (synsets, indices) from /results to timm/data/_info.
• Add ImageNetInfo / DatasetInfo classes to provide labelling for various ImageNet classifier layouts in timm
  • Update inference.py to use, try: python inference.py /folder/to/images --model convnext_small.in12k --label-type detail --topk 5
• Ready for 0.8.10 pypi pre-release (final testing).

Jan 20, 2023

• Add two convnext 12k -> 1k fine-tunes at 384x384

  • convnext_tiny.in12k_ft_in1k_384 - 85.1 @ 384
  • convnext_small.in12k_ft_in1k_384 - 86.2 @ 384

• Push all MaxxViT weights to HF hub, and add new ImageNet-12k -> 1k fine-tunes for rw base MaxViT and CoAtNet 1/2 models

Jan 11, 2023

• Update ConvNeXt ImageNet-12k pretrain series w/ two new fine-tuned weights (and pre FT .in12k tags)
  • convnext_nano.in12k_ft_in1k - 82.3 @ 224, 82.9 @ 288 (previously released)
  • convnext_tiny.in12k_ft_in1k - 84.2 @ 224, 84.5 @ 288
  • convnext_small.in12k_ft_in1k - 85.2 @ 224, 85.3 @ 288

Jan 6, 2023

• Finally got around to adding --model-kwargs and --opt-kwargs to scripts to pass through rare args directly to model classes from cmd line
  • train.py /imagenet --model resnet50 --amp --model-kwargs output_stride=16 act_layer=silu
  • train.py /imagenet --model vit_base_patch16_clip_224 --img-size 240 --amp --model-kwargs img_size=240 patch_size=12
• Cleanup some popular models to better support arg passthrough / merge with model configs, more to go.

Jan 5, 2023

• ConvNeXt-V2 models and weights added to existing convnext.py
  • Paper: ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders
  • Reference impl: https://github.com/facebookresearch/ConvNeXt-V2 (NOTE: weights currently CC-BY-NC)

Part way through the conversion of models to multi-weight support (model_arch.pretrain_tag), module reorg for future building, and lots of new weights and model additions as we go...

This is considered a development release. Please stick to 0.6.x if you need stability. Some of the model names, tags will shift a bit, some old names have already been deprecated and remapping support not added yet. For code 0.6.x branch is considered 'stable' https://github.com/rwightman/pytorch-image-models/tree/0.6.x

Dec 23, 2022 🎄☃

• Add FlexiViT models and weights from https://github.com/google-research/big_vision (check out paper at https://arxiv.org/abs/2212.08013)
  • NOTE currently resizing is static on model creation, on-the-fly dynamic / train patch size sampling is a WIP
• Many more models updated to multi-weight and downloadable via HF hub now (convnext, efficientnet, mobilenet, vision_transformer*, beit)
• More model pretrained tag and adjustments, some model names changed (working on deprecation translations, consider main branch DEV branch right now, use 0.6.x for stable use)
• More ImageNet-12k (subset of 22k) pretrain models popping up:
  • efficientnet_b5.in12k_ft_in1k - 85.9 @ 448x448
  • vit_medium_patch16_gap_384.in12k_ft_in1k - 85.5 @ 384x384
  • vit_medium_patch16_gap_256.in12k_ft_in1k - 84.5 @ 256x256
  • convnext_nano.in12k_ft_in1k - 82.9 @ 288x288

Dec 8, 2022

• Add 'EVA l' to vision_transformer.py, MAE style ViT-L/14 MIM pretrain w/ EVA-CLIP targets, FT on ImageNet-1k (w/ ImageNet-22k intermediate for some)
  • original source: https://github.com/baaivision/EVA

Dec 6, 2022

• Add 'EVA g', BEiT style ViT-g/14 model weights w/ both MIM pretrain and CLIP pretrain to beit.py.
  • original source: https://github.com/baaivision/EVA
  • paper: https://arxiv.org/abs/2211.07636

Dec 5, 2022

• Pre-release (0.8.0dev0) of multi-weight support (model_arch.pretrained_tag). Install with pip install --pre timm
  • vision_transformer, maxvit, convnext are the first three model impl w/ support
  • model names are changing with this (previous _21k, etc. fn will merge), still sorting out deprecation handling
  • bugs are likely, but I need feedback so please try it out
  • if stability is needed, please use 0.6.x pypi releases or clone from 0.6.x branch
• Support for PyTorch 2.0 compile is added in train/validate/inference/benchmark, use --torchcompile argument
• Inference script allows more control over output, select k for top-class index + prob json, csv or parquet output
• Add a full set of fine-tuned CLIP image tower weights from both LAION-2B and original OpenAI CLIP models

• Port of MaxViT Tensorflow Weights from official impl at https://github.com/google-research/maxvit
  • There was larger than expected drops for the upscaled 384/512 in21k fine-tune weights, possible detail missing, but the 21k FT did seem sensitive to small preprocessing

Oct 15, 2022

• Train and validation script enhancements
• Non-GPU (ie CPU) device support
• SLURM compatibility for train script
• HF datasets support (via ReaderHfds)
• TFDS/WDS dataloading improvements (sample padding/wrap for distributed use fixed wrt sample count estimate)
• in_chans !=3 support for scripts / loader
• Adan optimizer
• Can enable per-step LR scheduling via args
• Dataset 'parsers' renamed to 'readers', more descriptive of purpose
• AMP args changed, APEX via --amp-impl apex, bfloat16 supportedf via --amp-dtype bfloat16
• main branch switched to 0.7.x version, 0.6x forked for stable release of weight only adds
• master -> main branch rename

Minor bug fixes to HF push_to_hub, plus some more MaxVit weights

Oct 10, 2022

• More weights in maxxvit series, incl first ConvNeXt block based coatnext and maxxvit experiments:
  • coatnext_nano_rw_224 - 82.0 @ 224 (G) -- (uses ConvNeXt conv block, no BatchNorm)
  • maxxvit_rmlp_nano_rw_256 - 83.0 @ 256, 83.7 @ 320 (G) (uses ConvNeXt conv block, no BN)
  • maxvit_rmlp_small_rw_224 - 84.5 @ 224, 85.1 @ 320 (G)
  • maxxvit_rmlp_small_rw_256 - 84.6 @ 256, 84.9 @ 288 (G) -- could be trained better, hparams need tuning (uses ConvNeXt block, no BN)
  • coatnet_rmlp_2_rw_224 - 84.6 @ 224, 85 @ 320 (T)