mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
2,510 Commits 141 Branches 65 Tags
Commit Graph

2510 Commits

This Branch
This Branch
All Branches
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
MengqingCao
37c731ca37 fix device check 2024-10-17 12:38:02 +00:00
Ross Wightman
a852318b63
Merge pull request #2305 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 v1.0.11 2024-10-16 14:14:44 -07:00
Ross Wightman
65e8e9ca12
Merge pull request #2304 from huggingface/intern300m 
Add intern300m vit w/ converted timm weights. Fix #2300
 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 #2300 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 v1.0.10 2024-10-14 21:40:30 -07:00
Ross Wightman
c3052fa19e
Merge pull request #2298 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 #2266 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 #2299 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 #2289 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 #2292 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 #2281 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 #2277 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
Ross Wightman
a22ce0a329 Merge branch 'patch-1' of https://github.com/baorepo/pytorch-image-models into baorepo-patch-1 2024-09-22 10:14:35 -07:00