mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
2,057 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2057 Commits (05b0aaca51efe6d5f4e7e9a6fd6822361a377e30)
 

Author SHA1 Message Date
a-r-r-o-w 05b0aaca51 improvement: add typehints and docs to timm/models/resnet.py 2023-10-30 13:47:54 -07:00
a-r-r-o-w 564db019f6 fix: typo in CONTRIBUTING.md 2023-10-30 13:47:54 -07:00
a-r-r-o-w c2fe0a2268 improvement: add typehints and docs to timm/models/mobilenetv3.py 2023-10-30 13:47:54 -07:00
Laureηt d023154bb5 Update swin_transformer.py 
make `SwimTransformer`'s `patch_embed` customizable through the constructor
 2023-10-30 13:47:14 -07:00
Ross Wightman 68a121402f Added hub weights for dinov2 register models 2023-10-29 23:03:48 -07:00
Ross Wightman 3f02392488 Add DINOv2 models with register tokens. Convert pos embed to non-overlapping for consistency. 2023-10-29 23:03:48 -07:00
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 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 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