24f6d4f7f8
Fix #2127 move to ema device
2024-04-10 21:29:09 -07:00
e25bbfceec
Fix #2097 a small typo in train.py
2024-04-10 09:40:14 -07:00
195d12fdff
Add missing device argument to inference.py dataloader
2024-04-10 09:34:34 -07:00
c28ee2e904
Merge pull request #2145 from huggingface/fix_imagenet22k_ms_mapping
...
Add teddy-bear class back to first 1000 classes of imagenet22k_ms_synsets (line 851, index 850)
2024-04-09 14:56:31 -07:00
f5ea076a46
Merge pull request #2143 from huggingface/fix_asymm_set_grad_enable
...
Fix #2132 , remove use of _C.set_grad_enable. Line endings were messed up too
2024-04-09 10:14:13 -07:00
9531eb793c
Merge pull request #2142 from huggingface/fix_hrnet_head
...
Fix #2139 , disable strict weight loading when head changes from classification
2024-04-09 10:04:09 -07:00
286d941923
Add teddy-bear class back to first 1000 classes of imagenet22k_ms_synsets (index 851)
2024-04-09 09:33:08 -07:00
5c5ae8d401
Fix #2132 , remove use of _C.set_grad_enable. Line endings were messed up too
2024-04-09 09:00:23 -07:00
17b892f703
Fix #2139 , disable strict weight loading when head changes from classification
2024-04-09 08:41:37 -07:00
59b3d86c1d
Merge pull request #2124 from fzyzcjy/patch-1
...
Fix super tiny type error
2024-04-02 14:31:38 -07:00
b44e4e45a2
more
2024-04-02 10:25:30 +08:00
8880a5cd5c
Update scheduler.py
2024-03-23 11:27:33 +08:00
67b0b3d7c7
Merge pull request #2121 from huggingface/cleanup_vit_convert
...
Improve vit conversions. OpenAI convert pass through main convert
2024-03-21 13:13:53 -07:00
c559c3911f
Improve vit conversions. OpenAI convert pass through main convert for patch & pos resize. Fix #2120
2024-03-21 10:00:43 -07:00
492947d129
Merge pull request #2113 from gau-nernst/tinyclip
...
Add TinyCLIP
2024-03-20 16:22:08 -07:00
256cf19148
Rename tinyclip models to fit existing 'clip' variants, use consistently mapped OpenCLIP compatible checkpoint on hf hub
2024-03-20 15:21:46 -07:00
111fad1a9f
Add first CPU inference benchmark, using torch.compile
2024-03-20 08:43:29 -07:00
4c743769ff
Merge pull request #2117 from mwalmsley/patch-1
...
Update hf_hub.mdx with new import, rename labels
2024-03-19 12:54:07 -07:00
496ba54485
Update hf_hub.mdx with new import, rename labels
...
Change suggested import from timm.models.hub.push_to_hf_hub to timm.models.push_to_hf_hub
timm.models.hub.push_to_hf_hub import fails because of refactor. timm now uses _hub.py
https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/_hub.py#L270
which is imported into timm.models.__init__
6ccb7d6a7c/timm/models/__init__.py (L87)
2024-03-19 13:39:21 -04:00
1a1d07d479
add other tinyclip
2024-03-19 07:27:09 +08:00
dfffffac55
add tinyclip 8m
2024-03-19 07:02:17 +08:00
6ccb7d6a7c
Merge pull request #2111 from jamesljlster/enhance_vit_get_intermediate_layers
...
Vision Transformer (ViT) get_intermediate_layers: enhanced to support dynamic image size and saved computational costs from unused blocks
2024-03-18 13:41:18 -07:00
db06b56d34
Saved computational costs of get_intermediate_layers() from unused blocks
2024-03-17 21:34:06 +08:00
4731e4efc4
Modified ViT get_intermediate_layers() to support dynamic image size
2024-03-16 23:07:21 +08:00
70ccf00c95
Merge pull request #2108 from huggingface/onnx_dynamo
...
Add support for dynamo based onnx export
2024-03-13 13:24:03 -07:00
ba641e07ae
Add support for dynamo based onnx export
2024-03-13 12:05:26 -07:00
2ec2f1aa73
Merge pull request #2105 from SmilingWolf/main
...
SwinV2: add configurable act_layer argument
2024-03-05 22:17:10 -08:00
59cb0be595
SwinV2: add configurable act_layer argument
...
Defaults to "gelu", but makes it possible to pass "gelu_tanh".
Makes it easier to port weights from JAX/Flax, where the tanh
approximation is the default.
2024-03-05 22:04:17 +01:00
6e6f3686a7
Update README.md
2024-02-19 11:32:03 -08:00
49992b0dc7
Update version.py
...
Update to 0.9.16 for release
2024-02-19 11:08:17 -08:00
79f2ce99aa
Update changes.md
2024-02-19 11:07:28 -08:00
9926be4706
Update README.md
2024-02-19 11:06:32 -08:00
3df4ffe914
Delete requirements-docs.txt
2024-02-16 09:15:38 -08:00
155b32a1c2
Merge pull request #2096 from huggingface/pyproject_pdm
...
Remove setup.py, replace with pyproject.toml and pdm helpers
2024-02-16 09:04:46 -08:00
35d6eef0df
Version bump, add test markers back to toml
2024-02-16 09:04:00 -08:00
01616aa314
Remove setup.py, replace with pyproject.toml and pdm helpers
2024-02-15 17:39:22 -08:00
8a713b09e5
Merge branch 'seefun-master'
2024-02-12 15:01:45 -08:00
31e0dc0a5d
Tweak hgnet before merge
2024-02-12 15:00:32 -08:00
3e03491e49
Merge branch 'master' of https://github.com/seefun/pytorch-image-models into seefun-master
2024-02-12 14:59:54 -08:00
958938845a
Update version.py
2024-02-10 23:10:50 -08:00
1b50b15145
Merge pull request #2092 from huggingface/mesa_ema
...
ModelEMAV3 + MESA experiments
2024-02-10 23:10:27 -08:00
47c9bc4dc6
Fix device idx split
2024-02-10 21:41:14 -08:00
59239d9df5
Cleanup imports for vit relpos
2024-02-10 21:40:57 -08:00
ac1b08deb6
fix_init on vit & relpos vit
2024-02-10 20:15:37 -08:00
935950cc11
Fix F.sdpa attn drop prob
2024-02-10 20:14:47 -08:00
0737cf231d
Add Next-ViT
2024-02-10 17:05:16 -08:00
d6c2cc91af
Make NormMlpClassifier head reset args consistent with ClassifierHead
2024-02-10 16:25:33 -08:00
87fec3dc14
Update experimental vit model configs
2024-02-10 16:05:58 -08:00
7d3c2dc993
Add group_matcher for DaViT
2024-02-10 14:58:45 -08:00
7bc7798d0e
Type annotation correctness for create_act
2024-02-10 14:57:58 -08:00
Ross Wightman
Ross Wightman
fzyzcjy
fzyzcjy
Mike Walmsley
Thien Tran
Cheng-Ling Lai
SmilingWolf