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,532 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2532 Commits (e31e5d2d6430d8cd14664d88f680aa8d62ad4726)
 

Author SHA1 Message Date
mrT23 e31e5d2d64 imports 2024-11-12 07:53:39 -08:00
Tal 68d5a64e45 extend existing unittests 2024-11-12 07:53:39 -08:00
Ross Wightman 9f5c279bad Update log to describe scheduling behaviour diff w/ warmup_prefix 2024-11-08 11:01:11 -08:00
Ross Wightman 363b043c13 Extend train epoch schedule by warmup_epochs if warmup_prefix enable, allows schedule to reach end w/ prefix enabledy 2024-11-08 11:01:11 -08:00
Augustin Godinot 7f0c1b1f30 Add trust_remote_code argument to ReaderHfds 2024-11-08 08:16:36 -08:00
Wojtek Jasiński eb94efb218 fix pos embed dynamic resampling for eva 2024-11-06 16:03:27 -08:00
Wojtek Jasiński 3c7822c621 fix pos embed dynamic resampling for deit 2024-11-06 16:03:27 -08:00
Wojtek Jasiński 3ae3f44288 Fix positional embedding resampling for non-square inputs in ViT 2024-11-06 16:03:27 -08:00
Josua Rieder 51ac8d2efb fix typo in train.py: bathes > batches 2024-11-05 08:53:55 -08:00
Josua Rieder 7e5477acf5 Replace deprecated positional argument with --data-dir 2024-11-05 08:53:36 -08:00
Ross Wightman d4dde48dd5 Missed first_conv from resnet18d 2024-10-31 19:29:53 -07:00
Ross Wightman e6263bf64d Add resnet and resnet-v2 18/34 weights trained with mnv4 small based recipe 2024-10-31 16:39:35 -07:00
Ross Wightman f5b58e31a2 Allow non train mode for wds reader to operate w/o sample count, exhaust iterator 2024-10-31 16:39:35 -07:00
Ross Wightman f689c850b9 One more small c&p issue 2024-10-23 21:51:09 -07:00
Ross Wightman baa7242dd3 Fix c&p error, slight reformat 2024-10-23 21:51:09 -07:00
Ross Wightman 1b5cae681c Update some clip pretrained weights to point to new hub locations, add a few missing weights 2024-10-23 21:51:09 -07:00
Ross Wightman 310ffa32c5
Update version.py 
dev version 1.0.12.dev0
 2024-10-19 09:56:17 -07:00
Ross Wightman c93567280f
Update README.md 2024-10-19 08:23:54 -07:00
Ross Wightman 5081b53e48
Merge pull request #2308 from huggingface/device_amp_cleanup 
Cleanup some amp related behaviour to better support different (non-cuda) devices
 2024-10-19 08:19:27 -07:00
Ross Wightman c3992d5c4c Remove extra space 2024-10-18 14:54:16 -07:00
Ross Wightman 015fbe457a Merge branch 'MengqingCao-npu_support' into device_amp_cleanup 2024-10-18 14:50:44 -07:00
Ross Wightman 81b59faf77 Merge branch 'npu_support' of github.com:MengqingCao/pytorch-image-models into MengqingCao-npu_support 2024-10-18 14:50:00 -07:00
Ross Wightman 1766a01f96 Cleanup some amp related behaviour to better support different (non-cuda) devices 2024-10-18 13:54:16 -07:00
MengqingCao 37c731ca37 fix device check 2024-10-17 12:38:02 +00:00
Ross Wightman a852318b63
Merge pull request #2305 from NightMachinery/patch-2 
mambaout.py: fixed bug
 2024-10-16 14:39:43 -07:00
Feraidoon Mehri ca20e102fe
mambaout.py: fixed bug 2024-10-17 01:03:28 +03:30
Ross Wightman 8cb2548962 Version 1.0.11 2024-10-16 14:14:44 -07:00
Ross Wightman 65e8e9ca12
Merge pull request #2304 from huggingface/intern300m 
Add intern300m vit w/ converted timm weights. Fix #2300
 2024-10-16 14:00:22 -07:00
Ross Wightman 89dffc5ff0 Another small fix for original mambaout models, no classifier nn.Linear when num_classe=0 on init 2024-10-16 12:36:36 -07:00
Ross Wightman fad4538801 Elevate import deprecation warnings from DeprecationWarning to FutureWarning so messages are now seen 2024-10-16 11:30:01 -07:00
Ross Wightman a1f379e712 Add intern300m vit w/ converted timm weights. Fix #2300 2024-10-16 10:29:06 -07:00
MengqingCao 234f975787 add npu support 2024-10-16 07:13:45 +00:00
Ross Wightman 60f517c883 Fix wrong name in _all_ for models._registry 2024-10-15 07:39:46 -07:00
Ross Wightman b4a9a166c3 Version 1.0.10 2024-10-14 21:40:30 -07:00
Ross Wightman c3052fa19e
Merge pull request #2298 from huggingface/preact_resnet18 
Add resnet18/18d pre-act model configs for potential training.
 2024-10-14 19:39:04 -07:00
Ross Wightman abdf33145c Add 34/34d pre-act resnet variants 2024-10-14 13:23:50 -07:00
Ross Wightman c82ce86f8f Add 384x384 mambaout_base_plus model weights 2024-10-14 12:28:57 -07:00
Ross Wightman 2703d155c8
Update README.md 2024-10-11 16:59:06 -07:00
Ross Wightman 9aaf579aa5
Update README.md 2024-10-11 15:14:01 -07:00
Ross Wightman 96cd3f6a7d Update readme with latest 2024-10-11 15:10:05 -07:00
Ross Wightman 3236dff91d
Merge pull request #2266 from huggingface/mambaout 
Mambaout
 2024-10-11 12:16:48 -07:00
Ross Wightman 82ae247879 MambaOut weights on hub, configs finalized 2024-10-11 11:07:40 -07:00
Ross Wightman 7efb60c299 Add first_conv for mambaout 2024-10-09 14:11:40 -07:00
Ross Wightman 5dc5ee5b42 Add global_pool to mambaout __init__ and pass to heads 2024-10-09 14:11:40 -07:00
Ross Wightman 9d1dfe8dbe Incorrectly named head_hidden_size 2024-10-09 14:11:40 -07:00
Ross Wightman 91e743f2dd Mambaout tweaks 2024-10-09 14:11:40 -07:00
Ross Wightman 4542cf03f9 Add features_only, other bits to mambaout, define different base alternatives 2024-10-09 14:11:40 -07:00
Ross Wightman c2da12c7e1 Update rw models, fix heads 2024-10-09 14:11:40 -07:00
Ross Wightman f2086f51a0 Add mambaout builder support, pretrained weight remap 2024-10-09 14:11:40 -07:00
Ross Wightman c6ef54eefa Initial mambaout work 2024-10-09 14:11:40 -07:00