mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
2,711 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2711 Commits (c8c4f256b8c279575ed87012f38620a4cd25df57)
 

Author SHA1 Message Date
Ross Wightman 81b59faf77 Merge branch 'npu_support' of github.com:MengqingCao/pytorch-image-models into MengqingCao-npu_support 2024-10-18 14:50:00 -07:00
Ross Wightman 1766a01f96 Cleanup some amp related behaviour to better support different (non-cuda) devices 2024-10-18 13:54:16 -07:00
MengqingCao 37c731ca37 fix device check 2024-10-17 12:38:02 +00:00
Ross Wightman a852318b63
Merge pull request from NightMachinery/patch-2 
mambaout.py: fixed bug
 2024-10-16 14:39:43 -07:00
Feraidoon Mehri ca20e102fe
mambaout.py: fixed bug 2024-10-17 01:03:28 +03:30
Ross Wightman 8cb2548962 Version 1.0.11 2024-10-16 14:14:44 -07:00
Ross Wightman 65e8e9ca12
Merge pull request from huggingface/intern300m 
Add intern300m vit w/ converted timm weights. Fix
2024-10-16 14:00:22 -07:00
Ross Wightman 89dffc5ff0 Another small fix for original mambaout models, no classifier nn.Linear when num_classe=0 on init 2024-10-16 12:36:36 -07:00
Ross Wightman fad4538801 Elevate import deprecation warnings from DeprecationWarning to FutureWarning so messages are now seen 2024-10-16 11:30:01 -07:00
Ross Wightman a1f379e712 Add intern300m vit w/ converted timm weights. Fix 2024-10-16 10:29:06 -07:00
MengqingCao 234f975787 add npu support 2024-10-16 07:13:45 +00:00
Ross Wightman 60f517c883 Fix wrong name in _all_ for models._registry 2024-10-15 07:39:46 -07:00
Ross Wightman b4a9a166c3 Version 1.0.10 2024-10-14 21:40:30 -07:00
Ross Wightman c3052fa19e
Merge pull request from huggingface/preact_resnet18 
Add resnet18/18d pre-act model configs for potential training.
 2024-10-14 19:39:04 -07:00
Ross Wightman abdf33145c Add 34/34d pre-act resnet variants 2024-10-14 13:23:50 -07:00
Ross Wightman c82ce86f8f Add 384x384 mambaout_base_plus model weights 2024-10-14 12:28:57 -07:00
Ross Wightman 2703d155c8
Update README.md 2024-10-11 16:59:06 -07:00
Ross Wightman 9aaf579aa5
Update README.md 2024-10-11 15:14:01 -07:00
Ross Wightman 96cd3f6a7d Update readme with latest 2024-10-11 15:10:05 -07:00
Ross Wightman 3236dff91d
Merge pull request from huggingface/mambaout 
Mambaout
 2024-10-11 12:16:48 -07:00
Ross Wightman 82ae247879 MambaOut weights on hub, configs finalized 2024-10-11 11:07:40 -07:00
Ross Wightman 7efb60c299 Add first_conv for mambaout 2024-10-09 14:11:40 -07:00
Ross Wightman 5dc5ee5b42 Add global_pool to mambaout __init__ and pass to heads 2024-10-09 14:11:40 -07:00
Ross Wightman 9d1dfe8dbe Incorrectly named head_hidden_size 2024-10-09 14:11:40 -07:00
Ross Wightman 91e743f2dd Mambaout tweaks 2024-10-09 14:11:40 -07:00
Ross Wightman 4542cf03f9 Add features_only, other bits to mambaout, define different base alternatives 2024-10-09 14:11:40 -07:00
Ross Wightman c2da12c7e1 Update rw models, fix heads 2024-10-09 14:11:40 -07:00
Ross Wightman f2086f51a0 Add mambaout builder support, pretrained weight remap 2024-10-09 14:11:40 -07:00
Ross Wightman c6ef54eefa Initial mambaout work 2024-10-09 14:11:40 -07:00
Ross Wightman 6ee638a095
Merge pull request from huggingface/siglip_update 
Add i18n variant of so400m model w/ weights. Add two in1k fine-tunes
 2024-10-09 10:25:32 -07:00
Ross Wightman d9321b0e10 Add weights for fine-tuned siglip so400m. Add webli_i18n pretrained tags for the multi-lingual model variants (incl older base) 2024-10-09 09:04:44 -07:00
Ross Wightman 01b62264af Add i18n variant of so400m model w/ weights. Add two in1k fine-tunes of original so400m 384x384 but at 378x378 (better matches patch14) 2024-10-08 23:40:24 -07:00
Ross Wightman 72f0edb7e8 missed first_conv for rnv2 18d 2024-10-08 12:38:54 -07:00
Ross Wightman 3ed603a2ce Add resnet18/18d pre-act model configs for potential training. Fix 2024-10-08 11:28:07 -07:00
Ross Wightman 41a79e0fcb Add overlapped stem convnext zepto weights 2024-10-08 11:26:34 -07:00
Ross Wightman 4d4bdd64a9 Add --torchcompile-mode args to train, validation, inference, benchmark scripts 2024-10-02 15:17:53 -07:00
Ross Wightman 14d55a7cf3
Merge pull request from huggingface/inference_tests 
Do full inference test against test vectors for test_* models
 2024-10-02 11:48:21 -07:00
Ross Wightman 1a2d8bb5a0 Update pip install to use whl/cpu 2024-10-02 10:26:22 -07:00
Ross Wightman 036b1f0cc9 no 2.4.1 for cpu 2024-10-02 10:18:34 -07:00
Ross Wightman 95907e69c2 Further reduce atol for model comparison, move python 3.11 + torch 2.2 -> python 3.12 + torch 2.4.1 2024-10-02 10:05:27 -07:00
Ross Wightman fde6719403 relax tolerance on inference test 2024-10-02 09:53:17 -07:00
Ross Wightman 0e27f302a0 Do full inference test against test vectors for test_* models 2024-10-02 09:39:26 -07:00
Ross Wightman 44f1a343e7
Merge pull request from huggingface/convnext_zepto 
convnext zepto, rmsnorm experiments
 2024-09-30 15:37:31 -07:00
Ross Wightman 545bd4056c Tag along test_vit3 weights 2024-09-30 12:03:32 -07:00
Ross Wightman 69b687d4cc Add zepto weights 2024-09-30 11:43:23 -07:00
Ross Wightman c6e5557a5a Mismatch pretrained_cfg 2024-09-30 11:43:23 -07:00
Ross Wightman 5d7bd2973e convnext zepto, rmsnorm experiments 2024-09-30 11:43:23 -07:00
Ross Wightman e3242a5258
Merge pull request from huggingface/more_tiny_test_models 
Adding some more tiny test models to train...
 2024-09-22 10:28:29 -07:00
Ross Wightman 305a360f08 Merge branch 'baorepo-patch-1' 2024-09-22 10:27:35 -07:00
Ross Wightman c1cb5641c7 Add weight fore mobilenetv4 small 0.5, change 0.25 -> 0.35 2024-09-22 10:27:01 -07:00