9aaf579aa5
Update README.md
2024-10-11 15:14:01 -07:00
96cd3f6a7d
Update readme with latest
2024-10-11 15:10:05 -07:00
3236dff91d
Merge pull request #2266 from huggingface/mambaout
...
Mambaout
2024-10-11 12:16:48 -07:00
82ae247879
MambaOut weights on hub, configs finalized
2024-10-11 11:07:40 -07:00
7efb60c299
Add first_conv for mambaout
2024-10-09 14:11:40 -07:00
5dc5ee5b42
Add global_pool to mambaout __init__ and pass to heads
2024-10-09 14:11:40 -07:00
9d1dfe8dbe
Incorrectly named head_hidden_size
2024-10-09 14:11:40 -07:00
91e743f2dd
Mambaout tweaks
2024-10-09 14:11:40 -07:00
4542cf03f9
Add features_only, other bits to mambaout, define different base alternatives
2024-10-09 14:11:40 -07:00
c2da12c7e1
Update rw models, fix heads
2024-10-09 14:11:40 -07:00
f2086f51a0
Add mambaout builder support, pretrained weight remap
2024-10-09 14:11:40 -07:00
c6ef54eefa
Initial mambaout work
2024-10-09 14:11:40 -07:00
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
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
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
72f0edb7e8
missed first_conv for rnv2 18d
2024-10-08 12:38:54 -07:00
3ed603a2ce
Add resnet18/18d pre-act model configs for potential training. Fix #2289
2024-10-08 11:28:07 -07:00
41a79e0fcb
Add overlapped stem convnext zepto weights
2024-10-08 11:26:34 -07:00
4d4bdd64a9
Add --torchcompile-mode args to train, validation, inference, benchmark scripts
2024-10-02 15:17:53 -07:00
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
1a2d8bb5a0
Update pip install to use whl/cpu
2024-10-02 10:26:22 -07:00
036b1f0cc9
no 2.4.1 for cpu
2024-10-02 10:18:34 -07:00
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
fde6719403
relax tolerance on inference test
2024-10-02 09:53:17 -07:00
0e27f302a0
Do full inference test against test vectors for test_* models
2024-10-02 09:39:26 -07:00
44f1a343e7
Merge pull request #2281 from huggingface/convnext_zepto
...
convnext zepto, rmsnorm experiments
2024-09-30 15:37:31 -07:00
545bd4056c
Tag along test_vit3 weights
2024-09-30 12:03:32 -07:00
69b687d4cc
Add zepto weights
2024-09-30 11:43:23 -07:00
c6e5557a5a
Mismatch pretrained_cfg
2024-09-30 11:43:23 -07:00
5d7bd2973e
convnext zepto, rmsnorm experiments
2024-09-30 11:43:23 -07:00
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
305a360f08
Merge branch 'baorepo-patch-1'
2024-09-22 10:27:35 -07:00
c1cb5641c7
Add weight fore mobilenetv4 small 0.5, change 0.25 -> 0.35
2024-09-22 10:27:01 -07:00
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
9067be6a30
Add weights for new tiny test models
2024-09-22 07:59:23 -07:00
93374637c6
Merge pull request #2285 from huggingface/fix_swin_pad
...
Fix reversed H & W padding for swin patch merging
2024-09-21 17:00:49 -07:00
65564f7da5
Fix reversed H & W padding for swin patch merging
2024-09-21 16:51:02 -07:00
a2f539f055
Add a few more test model defs in prep for weight upload
2024-09-21 11:38:38 -07:00
6ab2af610d
Adding some more tiny test models to train
2024-09-06 15:35:57 -07:00
ee5b1e8217
Add missing nhwc bm for 4090
2024-09-05 07:30:24 -07:00
d6b8816eda
MobilenetV4: add two more lightweight models
...
Mobilenetv4 is very fast and ideal for embedded devices. However, for many low-cost, low-power embedded MCU devices, smaller models are required. Hopefully this PR will merge.
2024-09-05 02:34:11 +00:00
ef570e266d
Remove some of the older benchmark files
2024-09-04 15:38:31 -07:00
5c33776aa5
Add a bunch more benchmark sets
2024-09-04 15:35:26 -07:00
f81cbdcca9
Merge pull request #2274 from huggingface/bulk_runner_tweaks
...
Better all res resolution for bulk runner
2024-09-03 12:11:56 -07:00
6c42299b3e
Back to dev version
2024-09-03 10:29:05 -07:00
a50713ce6e
Fix #2272
2024-09-02 13:20:05 -07:00
ebbe530ee4
Add MobileNetV3 RA4 (mnv4 recipe) weights
2024-09-02 13:10:34 -07:00
fa4a1e597f
Better all res resolution for bulk runner
2024-08-26 22:28:01 -07:00
b3f3a005a0
Add RTX4090 inference benchmark numbers for latest models
2024-08-26 17:15:05 -07:00
072783331f
Version 1.0.9
2024-08-23 11:47:42 -07:00
Ross Wightman
Ross Wightman
alias pillar1989