PyTorch Image Models Files

Jan 19, 2025

• Fix loading of LeViT safetensor weights, remove conversion code which should have been deactivated
• Add 'SO150M' ViT weights trained with SBB recipes, decent results, but not optimal shape for ImageNet-12k/1k pretrain/ft
• vit_so150m_patch16_reg4_gap_256.sbb_e250_in12k_ft_in1k - 86.7% top-1
• vit_so150m_patch16_reg4_gap_384.sbb_e250_in12k_ft_in1k - 87.4% top-1
• vit_so150m_patch16_reg4_gap_256.sbb_e250_in12k
• Misc typing, typo, etc. cleanup
• 1.0.14 release to get above LeViT fix out

What's Changed

• Fix nn.Module type hints by @adamjstewart in https://github.com/huggingface/pytorch-image-models/pull/2400
• Add missing paper title by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2405
• fix 'timm recipe scripts' link by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2404
• fix typo in EfficientNet docs by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2403
• disable abbreviating csv inference output with ellipses by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2402
• fix incorrect LaTeX formulas by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2406
• VGG ConvMlp: fix layer defaults/types by @adamjstewart in https://github.com/huggingface/pytorch-image-models/pull/2409
• Implement --no-console-results in inference.py by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2408
• LeViT safetensors load is broken by conversion code that wasn't deactivated by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2412
• A few more weights by @rwightman in https://github.com/huggingface/pytorch-image-models/pull/2413
• Fix typos by @JosuaRieder in https://github.com/huggingface/pytorch-image-models/pull/2415

New Contributors

• @adamjstewart made their first contribution in https://github.com/huggingface/pytorch-image-models/pull/2400

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