69e0ca2e36
Weights on hf hub, bicubic yields slightly better eval
2023-08-19 16:25:45 -07: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
24fc720d34
Merge pull request #1918 from Mingosnake/fix-efficientformerv2-typo
...
Fix typo in efficientformer_v2
2023-08-15 20:04:35 -07:00
7938f28542
Fix typo in efficientformer_v2
2023-08-16 03:29:01 +09: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
379998ed5d
Merge pull request #1914 from developer0hye/developer0hye-patch-1
...
Set self.num_features to neck_chans if neck_chans > 0 for vision_transformer_sam
2023-08-11 11:23:57 -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
f6771909ff
Merge pull request #1903 from twmht/fix_num_classes
...
fix num_classes not found in repvit
2023-08-07 16:35:36 -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
3b8ef3f32f
Merge pull request #1890 from Separius/patch-1
...
use float in resample_abs_pos_embed_nhwc
2023-07-28 21:47:26 -07: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
e590ec51b7
Merge pull request #1889 from twmht/import_lion
...
import lion in __init__.py
2023-07-26 20:33:32 -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
394e814555
Merge pull request #1866 from lRomul/replace_deprecated_np_types
...
Replace deprecated NumPy aliases of builtin types
2023-07-07 16:07:50 -07:00
158bf129c4
Replace deprecated NumPy aliases of builtin types
2023-07-03 22:24:25 +03:00
c241081251
Merge pull request #1850 from huggingface/effnet_improve_features_only
...
Support other features only modes for EfficientNet. Fix #1848 fix #1849
2023-06-23 22:56:08 -07:00
f9a24fa19f
Merge pull request #1846 from seefun/master
...
add I-JEPA pretrained weight for ViT
2023-06-15 11:12:53 -07:00
47517dbefd
Clean more feature extract issues
...
* EfficientNet/MobileNetV3/HRNetFeatures cls and FX mode support -ve index
* MobileNetV3 allows feature_cfg mode to bypass MobileNetV3Features
2023-06-14 14:46:22 -07:00
a09c88ed0f
Support other features only modes for EfficientNet
2023-06-14 12:57:39 -07:00
c3f24a5ae5
‘add ViT weight from I-JEPA pretrain’
2023-06-14 22:30:31 +08:00
2d597b126d
Missed extra nadam algo step for capturable path
2023-06-13 20:51:31 -07:00
4790c0fa16
Missed nadamw.py
2023-06-13 20:45:58 -07:00
dab0360e00
Add NadamW based on mlcommons algorithm, added multi-tensor step
2023-06-13 20:45:17 -07:00
fb4f220c2e
Merge pull request #1841 from mishig25/update-doc-build-actions
...
[doc build] Use secrets
2023-06-09 07:04:06 -07:00
3ebbe172ec
[doc build] Use secrets
2023-06-09 10:47:32 +02:00
2d0dbd17e3
Merge pull request #1837 from lorenzbaraldi/fix_help_string
...
Changed help_string of args worker
2023-06-02 09:22:32 -07:00
700aebcdc4
Fix Pytorch 2.0 breakage for Lookahead optimizer adapter
2023-06-02 08:39:07 -07:00
Ross Wightman
Chengpeng Chen
Ross Wightman
Minseo Kang
Yonghye Kwon
Daniel Suess
alec.tu
Sepehr Sameni
jameslahm
Ruslan Baikulov
SeeFun
Mishig