PyTorch Image Models Files

Feb 21, 2025

• SigLIP 2 ViT image encoders added (https://huggingface.co/collections/timm/siglip-2-67b8e72ba08b09dd97aecaf9)
• Variable resolution / aspect NaFlex versions are a WIP
• Add 'SO150M2' ViT weights trained with SBB recipes, great results, better for ImageNet than previous attempt w/ less training.
• vit_so150m2_patch16_reg1_gap_448.sbb_e200_in12k_ft_in1k - 88.1% top-1
• vit_so150m2_patch16_reg1_gap_384.sbb_e200_in12k_ft_in1k - 87.9% top-1
• vit_so150m2_patch16_reg1_gap_256.sbb_e200_in12k_ft_in1k - 87.3% top-1
• vit_so150m2_patch16_reg4_gap_256.sbb_e200_in12k
• Updated InternViT-300M '2.5' weights
• Release 1.0.15

Feb 1, 2025

• FYI PyTorch 2.6 & Python 3.13 are tested and working w/ current main and released version of timm

Jan 27, 2025

• Add Kron Optimizer (PSGD w/ Kronecker-factored preconditioner)
• Code from https://github.com/evanatyourservice/kron_torch
• See also https://sites.google.com/site/lixilinx/home/psgd

What's Changed

• Fix metavar for --input-size by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2417
• Add arguments to the respective argument groups by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2416
• Add missing training flag to convert_sync_batchnorm by @collinmccarthy in https://github.com/huggingface/pytorch-image-models/pull/2423
• Fix num_classes update in reset_classifier and RDNet forward head call by @brianhou0208 in https://github.com/huggingface/pytorch-image-models/pull/2421
• timm: add all to init by @adamjstewart in https://github.com/huggingface/pytorch-image-models/pull/2399
• Fiddling with Kron (PSGD) optimizer by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2427
• Try to force numpy<2.0 for torch 1.13 tests, update newest tested torch to 2.5.1 by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2429
• Kron flatten improvements + stochastic weight decay by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2431
• PSGD: unify RNG by @ClashLuke in https://github.com/huggingface/pytorch-image-models/pull/2433
• Add vit so150m2 weights by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2439
• adapt_input_conv: add type hints by @adamjstewart in https://github.com/huggingface/pytorch-image-models/pull/2441
• SigLIP 2 by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2440
• timm.models: explicitly export attributes by @adamjstewart in https://github.com/huggingface/pytorch-image-models/pull/2442

New Contributors

• @collinmccarthy made their first contribution in https://github.com/huggingface/pytorch-image-models/pull/2423
• @ClashLuke made their first contribution in https://github.com/huggingface/pytorch-image-models/pull/2433

Full Changelog: https://github.com/huggingface/pytorch-image-models/compare/v1.0.14...v1.0.15