Commit Graph

2695 Commits (9ce824c39a01e56e1d4214fa439edd7011b58b07)
 

Author SHA1 Message Date
Ross Wightman 9ce824c39a Add vit so150m2 weights 2025-02-14 15:55:51 -08:00
Ross Wightman a49b020eff Merge branch 'ClashLuke-patch-1' 2025-01-31 12:53:29 -08:00
Ross Wightman 490d222dd8 Fix issue taking device from V before V exists 2025-01-31 12:52:47 -08:00
Ross Wightman 875c19d0c9 Merge branch 'patch-1' of github.com:ClashLuke/pytorch-image-models into ClashLuke-patch-1 2025-01-31 12:43:28 -08:00
Ross Wightman 8b3c07a841
Update README.md 2025-01-31 10:37:32 -08:00
Lucas Nestler e025328f96
simplify RNG 2025-01-31 17:26:14 +01:00
Lucas Nestler 6367267298
unify RNG 2025-01-31 17:23:53 +01:00
Ross Wightman 872978ccfe Fix comment, add 'stochastic weight decay' idea because why not 2025-01-30 18:22:36 -08:00
Ross Wightman 510bbd5389 Change start/end args 2025-01-30 18:22:36 -08:00
Ross Wightman 31831f5948 Change flattening behaviour in Kron 2025-01-30 18:22:36 -08:00
Ross Wightman cdbafd9057 Try to force numpy<2.0 for torch 1.13 tests, update newest tested torch to 2.5.1 2025-01-28 20:56:30 -08:00
Ross Wightman b1752eefb5 Fix missing model key in bulk validate results on error 2025-01-28 13:20:40 -08:00
Ross Wightman b3a83b81d6 Prep Kron for merge, add detail to attributions note, README. 2025-01-27 21:02:26 -08:00
Ross Wightman 67ef6f0a92 Move opt_einsum import back out of class __init__ 2025-01-27 21:02:26 -08:00
Ross Wightman 9ab5464e4d More additions to Kron 2025-01-27 21:02:26 -08:00
Ross Wightman 5f10450235 Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now. 2025-01-27 21:02:26 -08:00
Ross Wightman cd21e80d03 Fiddling with Kron (PSGD) 2025-01-27 21:02:26 -08:00
Adam J. Stewart d81da93c16 Use import alias 2025-01-22 10:27:17 -08:00
Adam J. Stewart 4de1abf837 timm: add __all__ to __init__ 2025-01-22 10:27:17 -08:00
Ryan bda46f8e6f Add num_classes assertion after reset_classifier 2025-01-21 11:52:05 -08:00
Ryan 17eabaad17 Fix RDNet forward call 2025-01-21 11:52:05 -08:00
Ryan 80a4877376 Fix self.reset_classifier num_classes update 2025-01-21 11:52:05 -08:00
Collin McCarthy 84631cb5c6 Add missing training flag to convert_sync_batchnorm 2025-01-21 11:51:55 -08:00
Josua Rieder cb4cea561a add arguments to the respective argument groups 2025-01-20 10:54:35 -08:00
Josua Rieder 634b68ae50 Fix metavar for --input-size 2025-01-20 10:53:46 -08:00
Ross Wightman 5d535d7a2d Version 1.0.14, update README & changelog 2025-01-19 13:53:09 -08:00
Ross Wightman c6b74eb5bd Remove numpy ver constraint, not relevant for latest PyTorch version 2025-01-19 13:41:16 -08:00
Ross Wightman aa333079da Tweak so150m2 def 2025-01-19 13:40:53 -08:00
Josua Rieder 8d81fdf3d9 Fix typos 2025-01-19 13:39:40 -08:00
Ross Wightman 3677f67902 Add the 256x256 in1k ft of the so150m, add an alternate so150m def 2025-01-18 15:51:57 -08:00
Ross Wightman 2a84d68d02 Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe 2025-01-18 15:51:57 -08:00
Ross Wightman 9265d54a3a LeViT safetensors load is broken by conversion code that wasn't deactivated 2025-01-16 11:37:00 -08:00
Ross Wightman 21e75a9d25
Update version.py
Back to dev version
2025-01-16 11:23:17 -08:00
Ross Wightman c96e9e7ce0
Merge pull request #2408 from JosuaRieder/no_console_results
Implement --no-console-results in inference.py
2025-01-15 08:11:31 -08:00
Ross Wightman 63b2de7a6c
Merge pull request #2409 from adamjstewart/models/vgg
VGG ConvMlp: fix layer defaults/types
2025-01-15 07:37:26 -08:00
Adam J. Stewart 6d21eb0d37
VGG ConvMlp: fix layer defaults/types 2025-01-15 12:11:56 +01:00
Josua Rieder 47790637f2 Implement --no-console-results in inference.py 2025-01-15 11:37:11 +01:00
Ross Wightman ef7dec8e43
Merge pull request #2406 from JosuaRieder/fix_latex
fix incorrect LaTeX formulas
2025-01-14 11:03:15 -08:00
Ross Wightman 1572769059
Merge pull request #2402 from JosuaRieder/fix_inference_csv_export
disable abbreviating csv inference output with ellipses
2025-01-14 11:01:48 -08:00
Ross Wightman fc0609bcb6 Add --model-dtype (pure bfloat16/float16) support to inference.py 2025-01-14 11:00:16 -08:00
Ross Wightman 53c3c89a86
Merge pull request #2403 from JosuaRieder/efficientnet_typo
fix typo in EfficientNet docs
2025-01-14 10:57:46 -08:00
Ross Wightman 5e0df16fc0
Merge pull request #2404 from JosuaRieder/fix_training_scripts_doc_link
fix 'timm recipe scripts' link
2025-01-14 10:57:25 -08:00
Ross Wightman 7fa37c1f26
Merge pull request #2405 from JosuaRieder/add_missing_paper_title
Add missing paper title
2025-01-14 10:57:02 -08:00
Josua Rieder adf9efeac7 fix incorrect LaTeX formulas 2025-01-14 19:17:52 +01:00
Josua Rieder 1dab96c637 fix small copy paste mistake 2025-01-14 18:58:54 +01:00
Josua Rieder cac1899ba1 add missing paper title DLA: Deep Layer Aggregation 2025-01-14 18:56:23 +01:00
Josua Rieder 676c734796 fix 'timm recipe scripts' link 2025-01-14 18:49:45 +01:00
Josua Rieder 17a1abfc0d fix typo in EfficientNet docs 2025-01-14 18:36:51 +01:00
Josua Rieder 8ce197e33a disable abbreviating csv inference output with ellipses 2025-01-14 17:39:48 +01:00
Ross Wightman ff77dfa825
Merge pull request #2400 from adamjstewart/types/nn-module
Fix nn.Module type hints
2025-01-14 08:23:57 -08:00