be4e0d8f76
Update attrib comment to include v2
2023-08-19 23:39:09 -07:00
126a58e563
Combine ghostnetv2 with ghostnet, reduec redundancy, add weights to hf hub.
2023-08-19 23:33:43 -07:00
3f320a9e57
Merge branch 'Add-GhostNetV2' of github.com:yehuitang/pytorch-image-models into yehuitang-Add-GhostNetV2
2023-08-19 22:07:54 -07:00
7c2728c6fe
Merge pull request #1919 from ChengpengChen/main
...
Add RepGhost models and weights
2023-08-19 16:26:45 -07:00
69e0ca2e36
Weights on hf hub, bicubic yields slightly better eval
2023-08-19 16:25:45 -07:00
b8011565bd
Merge pull request #1894 from seefun/master
...
add two different EfficientViT models
2023-08-19 09:24:14 -07:00
7d7589e8da
Fixing efficient_vit torchscript, fx, default_cfg issues
2023-08-18 23:23:11 -07:00
58ea1c02c4
Add fixed_input_size flag to msra efficient_vit
2023-08-18 16:48:17 -07:00
c28324a150
Update efficient_vit (msra), hf hub weights
2023-08-18 16:45:37 -07:00
e700a32626
Cleanup of efficient_vit (mit), tweak eps for better AMP behaviour, formatting/cleanup, weights on hf hub
2023-08-18 16:06:07 -07:00
00f670fa69
fix bug in ci for efficientvits
2023-08-17 14:40:17 +08:00
e7f97cb5ce
Fix typos RepGhost models
2023-08-16 14:27:45 +08:00
d1d0193615
Add RepGhost models and weights
2023-08-16 11:54:53 +08:00
7938f28542
Fix typo in efficientformer_v2
2023-08-16 03:29:01 +09:00
b407794e3a
Add GhostNetV2
2023-08-13 18:20:27 +08:00
fc865282e5
Add ghostnetv2.py
2023-08-13 18:16:26 +08:00
da75cdd212
Merge pull request #1900 from huggingface/swin_maxvit_resize
...
Add support for resizing swin transformer, maxvit, coatnet at creation time
2023-08-11 15:05:28 -07:00
78a04a0e7d
Merge pull request #1911 from dsuess/1910-fixes-batchnormact-fx
...
Register norm_act layers as leaf modules
2023-08-11 14:34:16 -07:00
2048f6f20f
set self.num_features to neck_chans if neck_chans > 0
2023-08-11 13:45:06 +09:00
3a44e6c602
Fix #1912 CoaT model not loading w/ return_interm_layers
2023-08-10 11:15:58 -07:00
986de90360
Register orm_act layers as leaf modules
2023-08-10 15:37:26 +10:00
c692715388
Some RepVit tweaks
...
* add head dropout to RepVit as all models have that arg
* default train to non-distilled head output via distilled_training flag (set_distilled_training) so fine-tune works by default w/o distillation script
* camel case naming tweaks to match other models
2023-08-09 12:41:12 -07:00
c153cd4a3e
Add more advanced interpolation method from BEiT and support non-square window & image size adaptation for
...
* beit/beit-v2
* maxxvit/coatnet
* swin transformer
And non-square windows for swin-v2
2023-08-08 16:41:16 -07:00
bb2b6b5f09
fix num_classes not found
2023-08-07 15:16:03 +08:00
1dab536cb1
Fix torch.fx for swin padding change
2023-08-05 13:09:55 -07:00
7c0f492dbb
Fix type annotation for torchscript
2023-08-04 23:03:52 -07:00
7790ea709b
Add support for resizing swin transformer img_size and window_size on init and load from pretrained weights. Add support for non-square window_size to both swin v1/v2
2023-08-04 22:10:46 -07:00
81089b10a2
Remove unecessary LongTensor in EfficientFormer. Possibly maybe fix #1878
2023-08-03 16:38:53 -07:00
4224529ebe
Version 0.9.5 prep for release. README update
2023-08-03 15:16:46 -07:00
d138a9bf88
Add gluon hrnet small weights, fix #1895
2023-08-03 12:15:04 -07:00
76d166981d
Fix missing norm call in Mlp forward (not used by default, but can be enabled for normformer MLP scale). Fix #1851 fix #1852
2023-08-03 11:36:30 -07:00
8e4480e4b6
Patch and pos embed resample done in float32 always (cast to float and back). Fix #1811
2023-08-03 11:32:17 -07:00
150356c493
Fix unfortunate selecsls case bug caused by aggressive IDE rename
2023-08-03 10:37:06 -07:00
6e8c53d0d3
Comment out beit url, no longer valid as now require long query string, leave for reference, must use HF hub now.
2023-08-03 10:00:46 -07:00
a56e2bbf19
fix efficientvit_msra pretrained load
2023-08-03 18:44:38 +08:00
e94c60b546
efficientvit_msra refactor
2023-08-03 17:45:50 +08:00
047bab6ab2
efficientvit_mit stage refactor
2023-08-03 14:59:35 +08:00
e8fb866ccf
fix efficientvit_msra pool
2023-08-02 14:40:01 +08:00
43443f64eb
fix efficientvits
2023-08-02 14:12:37 +08:00
82d1e99e1a
add efficientvit(msra)
2023-08-01 18:51:08 +08:00
b91a77fab7
add EfficientVit (MIT)
2023-08-01 12:42:21 +08:00
40a518c194
use float in resample_abs_pos_embed_nhwc
...
since F.interpolate doesn't always support BFloat16
2023-07-28 16:01:42 -07:00
8cb0ddac45
Update README, version 0.9.4dev0
2023-07-27 17:07:31 -07:00
a9d0615f42
Fix ijepa vit issue with 448 model, minor formatting fixes
2023-07-26 20:46:27 -07:00
942726db31
import lion in __init__.py
2023-07-27 09:26:57 +08:00
5874d1bfc7
Merge pull request #1876 from jameslahm/main
...
Add RepViT models
2023-07-26 14:38:41 -07:00
b10310cc27
Add proper pool size for new resnexts
2023-07-26 14:36:03 -07:00
b71d60cdb7
Two small fixes, num_classes in base class, add model tag
2023-07-26 13:18:49 -07:00
3561f8e885
Add seresnextaa201d_32x8d 12k and 1k weights
2023-07-26 13:17:05 -07:00
3318e7614d
Add RepViT models
2023-07-21 14:56:53 +08:00
Ross Wightman
Ross Wightman
方曦
Chengpeng Chen
Minseo Kang
yehuitang
Yonghye Kwon
Daniel Suess
alec.tu
Sepehr Sameni
jameslahm