2051 Commits

Author SHA1 Message Date
Laureηt
fe92fd93e5 fix adaptive_avgmax_pool.py
remove extra whitespace in `SelectAdaptivePool2d`'s `__repr__`
2023-10-29 23:03:36 -07:00
Patrick Labatut
97450d618a Update DINOv2 license to Apache 2.0 2023-10-27 09:12:51 -07:00
mjamroz
7a6369156f avoid getting undefined 2023-10-22 21:36:23 -07:00
Tush9905
89ba0da910 Fixed Typos
Fixed the typos in helpers.py and CONTRIBUTING.md
2023-10-21 21:46:31 -07:00
Ross Wightman
9afe0bb78e Update README, prep for 0.9.8 release v0.9.8 2023-10-20 13:57:23 -07:00
pUmpKin-Co
8556462a18 fix doc typo in resnetv2 2023-10-20 11:56:50 -07:00
Ross Wightman
a517b82ccc Merge branch 'jameslahm-main' 2023-10-20 11:50:18 -07:00
Ross Wightman
462fb3ec9f Push new repvit weights to hub, tweak tag names 2023-10-20 11:49:29 -07:00
Ross Wightman
5309424d5e Merge branch 'main' of https://github.com/jameslahm/pytorch-image-models into jameslahm-main 2023-10-20 11:08:12 -07:00
Ross Wightman
d3ebdcfd93 Disable strict load when siglip vit pooling removed 2023-10-19 12:03:40 -07:00
Remy
963d80b299
fix: remove useless token (#1995) 2023-10-19 14:58:52 +02:00
Ross Wightman
e1e7cf5275
Merge pull request #1973 from huggingface/vit_siglip_and_reg
Working on support for SigLIP (w/ attn pool) ViT backbone and registers
2023-10-17 17:09:25 -07:00
Ross Wightman
e728f3efdb Cleanup ijepa models, they're just gap (global-avg-pool) models w/o heads. fc-norm conversion was wrong, gigantic should have been giant 2023-10-17 15:44:46 -07:00
Ross Wightman
49a459e8f1 Merge remote-tracking branch 'upstream/main' into vit_siglip_and_reg 2023-10-17 09:36:48 -07:00
Ross Wightman
a58f9162d7 Missed __init__.py update for attention pooling layer add 2023-10-17 09:28:21 -07:00
Ross Wightman
59b622233b Change ijepa names, add pretrain cfg for reg experimentts 2023-10-17 07:16:17 -07:00
Ross Wightman
71365165a2 Add SigLIP weights 2023-10-16 23:26:08 -07:00
Ross Wightman
68b2824e49
Update optim test to remove Variable/.data and fix _state_dict optim test for PyTorch 2.1 (#1988)
* Update optim test to remove Variable/.data and fix _state_dict optim test

* Attempt to run python 3.11 w/ 2.1

* Try factoring out testmarker to common var

* More fiddling

* Abandon attempt to reduce redunancy

* Another try
2023-10-12 11:53:10 -07:00
Ross Wightman
42daa3b497 Add full set of SigLIP models 2023-10-10 22:15:45 -07:00
lucapericlp
7ce65a83a2 Removing unused self.drop 2023-10-05 11:20:57 -07:00
Yassine
884ef88818 fix all SDPA dropouts 2023-10-05 08:58:41 -07:00
Yassine
b500cae4c5 fastvit: don't dropout in eval 2023-10-05 08:58:41 -07:00
Ross Wightman
b9dde58076 Fixup attention pooling in siglip vit support 2023-10-02 11:44:12 -07:00
jameslahm
f061b539d7 Update RepViT models 2023-10-01 14:00:53 +08:00
Ross Wightman
99cfd6702f Use global pool arg to select attention pooling in head 2023-09-30 16:16:21 -07:00
Ross Wightman
82cc53237e Working on support for siglip (w/ attn pool) vit backbone, and adding registers (reg tokens) 2023-09-30 16:03:01 -07:00
Ross Wightman
054c763fca Bump to dev 0.9.8 version 2023-09-27 10:27:47 -07:00
Ross Wightman
6bae514656 Add pretrained patch embed resizing to swin 2023-09-27 10:27:28 -07:00
Yassine
5c504b4ded flip these two 2023-09-27 10:24:12 -07:00
Yassine
8ba2038e6b fast_vit: propagate act_layer argument 2023-09-27 10:24:12 -07:00
Nguyen Nhat Hoang
95ba90157f Update tiny_vit.py to fix bug 2023-09-23 10:05:52 -07:00
Youssef Boulaoaune
a9c29a05de add input_size & batch_size to onnx_export 2023-09-21 14:46:25 -07:00
belfner
245ad4f41a Added missing RuntimeError to builder functions of models that do not currently support feature extraction 2023-09-19 08:19:14 -07:00
Thorsten Hempel
d4c21b95f4 Update repghost.py 2023-09-15 11:41:56 -07:00
Thorsten Hempel
7eb7d13845 Fix in_features for linear layer in reset_classifier. 2023-09-13 09:29:38 -07:00
Ross Wightman
730b907b4d
Update README.md v0.9.7 2023-09-02 11:16:21 -07:00
Ross Wightman
34ae2861f4 Version 0.9.7 2023-09-01 15:06:55 -07:00
Ross Wightman
0d124ffd4f Update README. Fine-grained layer-wise lr decay working for tiny_vit and both efficientvits. Minor fixes. 2023-09-01 15:05:29 -07:00
Ross Wightman
2f0fbb59b3 TinyViT weights on HF hub 2023-09-01 11:05:56 -07:00
Ross Wightman
507cb08acf TinyVitBlock needs adding as leaf for FX now, tweak a few dim names 2023-09-01 11:05:56 -07:00
Ross Wightman
9caf32b93f Move levit style pos bias resize with other rel pos bias utils 2023-09-01 11:05:56 -07:00
Ross Wightman
63417b438f TinyViT adjustments
* keep most of net in BCHW layout, performance appears same, can remove static resolution attribs and features easier to use
* add F.sdpa, decent gains in pt 2.1
* tweak crop pct based on eval
2023-09-01 11:05:56 -07:00
方曦
39aa44b192 Fixing tinyvit trace issue 2023-09-01 11:05:56 -07:00
方曦
aea3b9c854 Fixing tinyvit input_size issue 2023-09-01 11:05:56 -07:00
方曦
fabc4e5bcd Fixing tinyvit torchscript issue 2023-09-01 11:05:56 -07:00
方曦
bae949f830 fix attention_bias_cache in tinyvit 2023-09-01 11:05:56 -07:00
方曦
0198a2019e tinyvit skip ci test 2023-09-01 11:05:56 -07:00
方曦
170a5b6e27 add tinyvit 2023-09-01 11:05:56 -07:00
Ross Wightman
983310d6a2 Fix #1935 , torch.amp.autocast -> torch.autocast (namespace issue for 1.10 pt compat) 2023-08-30 15:03:28 -07:00
Ross Wightman
9260cf517d
Update README.md 2023-08-29 12:16:11 -07:00