方曦
|
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 |
Sepehr Sameni
|
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 |
Ross Wightman
|
8cb0ddac45
|
Update README, version 0.9.4dev0
|
2023-07-27 17:07:31 -07:00 |
|
Ross Wightman
|
a9d0615f42
|
Fix ijepa vit issue with 448 model, minor formatting fixes
|
2023-07-26 20:46:27 -07:00 |
alec.tu
|
942726db31
|
import lion in __init__.py
|
2023-07-27 09:26:57 +08:00 |
Ross Wightman
|
5874d1bfc7
|
Merge pull request #1876 from jameslahm/main
Add RepViT models
|
2023-07-26 14:38:41 -07:00 |
|
Ross Wightman
|
b10310cc27
|
Add proper pool size for new resnexts
|
2023-07-26 14:36:03 -07:00 |
|
Ross Wightman
|
b71d60cdb7
|
Two small fixes, num_classes in base class, add model tag
|
2023-07-26 13:18:49 -07:00 |
|
Ross Wightman
|
3561f8e885
|
Add seresnextaa201d_32x8d 12k and 1k weights
|
2023-07-26 13:17:05 -07:00 |
jameslahm
|
3318e7614d
|
Add RepViT models
|
2023-07-21 14:56:53 +08:00 |
Ruslan Baikulov
|
158bf129c4
|
Replace deprecated NumPy aliases of builtin types
|
2023-07-03 22:24:25 +03:00 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
a09c88ed0f
|
Support other features only modes for EfficientNet
|
2023-06-14 12:57:39 -07:00 |
SeeFun
|
c3f24a5ae5
|
‘add ViT weight from I-JEPA pretrain’
|
2023-06-14 22:30:31 +08:00 |
|
Ross Wightman
|
2d597b126d
|
Missed extra nadam algo step for capturable path
|
2023-06-13 20:51:31 -07:00 |
|
Ross Wightman
|
4790c0fa16
|
Missed nadamw.py
|
2023-06-13 20:45:58 -07:00 |
|
Ross Wightman
|
dab0360e00
|
Add NadamW based on mlcommons algorithm, added multi-tensor step
|
2023-06-13 20:45:17 -07:00 |
|
Ross Wightman
|
700aebcdc4
|
Fix Pytorch 2.0 breakage for Lookahead optimizer adapter
|
2023-06-02 08:39:07 -07:00 |
Lengyue
|
c308dbc6f2
|
update dinov2 layerscale init values
|
2023-05-24 12:20:17 -04:00 |
|
Ross Wightman
|
7cea88e2c4
|
Pop eps for lion optimizer
|
2023-05-21 15:20:03 -07:00 |
|
Ross Wightman
|
e9373b1b92
|
Cleanup before samvit merge. Resize abs posembed on the fly, undo some line-wraps, remove redundant unbind, fix HF hub weight load
|
2023-05-18 16:43:48 -07:00 |
|
方曦
|
c1c6eeb909
|
fix loading pretrained weight for samvit
|
2023-05-18 08:49:29 +08:00 |
|
方曦
|
15de561f2c
|
fix unit test for samvit
|
2023-05-17 12:51:12 +08:00 |
|
方曦
|
ea1f52df3e
|
add ViT for Segment-Anything Model
|
2023-05-17 11:39:29 +08:00 |
|
Ross Wightman
|
960202cfcc
|
Dev version 0.9.3 for main
|
2023-05-16 11:28:00 -07:00 |
|
Ross Wightman
|
c5d3ee47f3
|
Add B/16 datacompxl CLIP weights
|
2023-05-16 11:27:20 -07:00 |
|
Ross Wightman
|
3d05c0e86f
|
Version 0.9.2
|
2023-05-14 08:03:04 -07:00 |
Philip Keller
|
fc77e9ecc5
|
Update hub.py
fixed import of _hub modules
|
2023-05-12 21:48:46 +02:00 |
|
Ross Wightman
|
cc77096350
|
Version 0.9.1
|
2023-05-12 09:47:47 -07:00 |
|
Ross Wightman
|
f744bda994
|
use torch.jit.Final instead of Final for beit, eva
|
2023-05-12 09:12:14 -07:00 |
|
Ross Wightman
|
2e99bcaedd
|
Update README, prep for version 0.9.0 release
|
2023-05-11 15:22:50 -07:00 |
|
Ross Wightman
|
3eaf729f3f
|
F.sdpa for visformer fails w/o contiguous on qkv, make experimental
|
2023-05-11 11:37:37 -07:00 |
|
Ross Wightman
|
cf1884bfeb
|
Add 21k maxvit tf weights
|
2023-05-10 18:23:32 -07:00 |
|
Ross Wightman
|
6c2edf4d74
|
Missed hub_id entries for byoanet models
|
2023-05-10 15:58:55 -07:00 |
|
Ross Wightman
|
cf101b0097
|
Version 0.8.23dev0 and README update
|
2023-05-10 14:41:22 -07:00 |
|
Ross Wightman
|
850ab4931f
|
Missed a few pretrained tags...
|
2023-05-10 12:16:30 -07:00 |
|
Ross Wightman
|
ff2464e2a0
|
Throw when pretrained weights not available and pretrained=True (principle of least surprise).
|
2023-05-10 10:44:34 -07:00 |
|
Ross Wightman
|
8ce9a2c00a
|
Merge pull request #1222 from Leoooo333/master
Fix mixup/one_hot device problem
|
2023-05-10 08:59:15 -07:00 |
|
Ross Wightman
|
fd592ec86c
|
Fix an issue with FastCollateMixup still using device
|
2023-05-10 08:55:38 -07:00 |
|
Ross Wightman
|
e0ec0f7252
|
Merge pull request #1643 from nateraw/docstrings-update
Update Docstring for create_model
|
2023-05-09 21:33:20 -07:00 |
|
Ross Wightman
|
627b6315ba
|
Add typing to dinov2 entrypt fns, use hf hub for mae & dinov2 weights
|
2023-05-09 20:42:11 -07:00 |
|
Ross Wightman
|
b9d43c7dca
|
Version 0.8.22dev0
|
2023-05-09 20:38:10 -07:00 |
|
Ross Wightman
|
960a882510
|
Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights
|
2023-05-09 18:00:41 -07:00 |