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

124 Commits (87b4d7a29af80d7c1334af36392f744e9382cdf0)

Author SHA1 Message Date
Ross Wightman 010b486590 Add Dino pretrained weights (no head) for vit models. Add support to tests and helpers for models w/ no classifier (num_classes=0 in pretrained cfg) 2022-01-17 12:20:02 -08:00
Ross Wightman a8d103e18b Giant/gigantic vits snuck through in a test a broke GitHub test runner, add filter 2022-01-14 17:23:35 -08:00
Ross Wightman ef72ad4177 Extra vit_huge model likely to cause test issue (non in21k variant), adding to filters 2022-01-14 16:28:27 -08:00
Ross Wightman e967c72875 Update REAMDE.md. Sneak in g/G (giant / gigantic?) ViT defs from scaling paper 2022-01-14 16:28:27 -08:00
Ross Wightman 4df51f3932 Add lcnet_100 and mnasnet_small weights 2022-01-06 22:21:05 -08:00
Ross Wightman 5ccf682a8f Remove deprecated bn-tf train arg and create_model handler. Add evos/evob models back into fx test filter until norm_norm_norm branch merged. 2022-01-06 18:08:39 -08:00
Ross Wightman 25d1526092 Update pytest for GitHub runner to use --forked with xdist, hopefully eliminate memory buildup 2022-01-06 16:04:23 -08:00
Ross Wightman f7d210d759 Remove evonorm models from FX tests 2021-11-24 13:21:24 -08:00
Ross Wightman f83b0b01e3 Would like to pass GitHub tests again disabling both FX feature extract backward and torchscript tests 2021-11-23 22:24:58 -08:00
Ross Wightman 147e1059a8 Remove FX backward test from GitHub actions runs for now. 2021-11-23 14:32:32 -08:00
Ross Wightman 878bee1d5e Add patch8 vit model to FX exclusion filter 2021-11-22 14:00:27 -08:00
Ross Wightman ce76a810c2 New FX test strategy, filter based on param count 2021-11-22 11:48:40 -08:00
Ross Wightman 1e51c2d02e More FX test tweaks 2021-11-22 09:46:43 -08:00
Ross Wightman 90448031ea Filter more large models from FX tests 2021-11-21 21:26:44 -08:00
Ross Wightman 8dc269c303 Filter more models for FX tests 2021-11-21 19:49:33 -08:00
Ross Wightman 2482652027 Add nfnet_f2 to FX test exclusion 2021-11-21 14:08:53 -08:00
Ross Wightman 05092e2fbe Add more models to FX filter 2021-11-20 15:51:48 -08:00
Ross Wightman 3819bef93e Add FX test exclusion since it uses more ram and barfs on GitHub actions. Will take a few iterations to include needed models :( 2021-11-19 17:35:41 -08:00
Ross Wightman 9b3519545d Attempt to reduce memory footprint of FX tests for GitHub actions runs 2021-11-19 14:24:12 -08:00
Ross Wightman bdd3dff0ca beit_large models killing GitHub actions test, filter out 2021-11-19 08:39:48 -08:00
Ross Wightman f2006b2437 Cleanup qkv_bias cat in beit model so it can be traced 2021-11-18 21:25:00 -08:00
Ross Wightman 1076a65df1 Minor post FX merge cleanup 2021-11-18 19:47:07 -08:00
Alexander Soare 0262a0e8e1 fx ready for review 2021-11-13 00:06:33 +00:00
Alexander Soare d2994016e9 Add try/except guards 2021-11-12 21:16:53 +00:00
Alexander Soare b25ff96768 wip - pre-rebase 2021-11-12 20:45:05 +00:00
Alexander Soare a6c24b936b Tests to enforce all models FX traceable 2021-11-12 20:45:05 +00:00
Alexander Soare 6d2acec1bb Fix ordering of tests 2021-10-02 16:10:11 +01:00
Alexander Soare 65c3d78b96 Freeze unfreeze functionality finalized. Tests added 2021-10-02 15:55:08 +01:00
Ross Wightman 24720abe3b Merge branch 'master' into attn_update 2021-09-13 16:51:10 -07:00
Ross Wightman 1c9284c640 Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now 2021-09-13 16:38:23 -07:00
Ross Wightman 7ab2491ab7 Better handling of crossvit for tests / forward_features, fix torchscript regression in my changes 2021-09-13 13:01:05 -07:00
Ross Wightman f1808e0970 Post crossvit merge cleanup, change model names to reflect input size, cleanup img size vs scale handling, fix tests 2021-09-13 11:49:54 -07:00
Ross Wightman a897e0ebcc Merge branch 'feature/crossvit' of https://github.com/chunfuchen/pytorch-image-models into chunfuchen-feature/crossvit 2021-09-10 17:38:37 -07:00
Ross Wightman 8642401e88 Swap botnet 26/50 weights/models after realizing a mistake in arch def, now figuring out why they were so low... 2021-09-05 15:17:19 -07:00
Ross Wightman 5f12de4875 Add initial AttentionPool2d that's being trialed. Fix comment and still trying to improve reliability of sgd test. 2021-09-05 12:41:14 -07:00
Ross Wightman 54e90e82a5 Another attempt at sgd momentum test passing... 2021-09-03 20:50:26 -07:00
Richard Chen 7ab9d4555c add crossvit 2021-09-01 17:13:12 -04:00
Ross Wightman fc894c375c Another attempt at sgd momentum test passing... 2021-08-27 10:39:31 -07:00
Ross Wightman 708d87a813 Fix ViT SAM weight compat as weights at URL changed to not use repr layer. Fix #825. Tweak optim test. 2021-08-27 09:20:13 -07:00
Ross Wightman c207e02782 MOAR optimizer changes. Woo! 2021-08-18 22:20:35 -07:00
Ross Wightman 42c1f0cf6c Fix lars tests 2021-08-18 21:05:34 -07:00
Ross Wightman a426511c95 More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars. 2021-08-18 17:21:56 -07:00
Ross Wightman a6af48be64 add madgradw optimizer 2021-08-17 22:19:27 -07:00
Ross Wightman 55fb5eedf6 Remove experiment from lamb impl 2021-08-17 21:48:26 -07:00
Ross Wightman 959eaff121 Add optimizer tests and update testing to pytorch 1.9 2021-08-17 17:59:15 -07:00
Ross Wightman 01cb46a9a5 Add gc_efficientnetv2_rw_t weights (global context instead of SE attn). Add TF XL weights even though the fine-tuned ones don't validate that well. Change default arg for GlobalContext to use scal (mul) mode. 2021-08-07 16:45:29 -07:00
Ross Wightman ef1e2e12be Attempt to fix xcit test failures on github runner by filter largest models 2021-07-13 16:33:55 -07:00
Alexander Soare 623e8b8eb8 wip xcit 2021-07-11 09:39:38 +01:00
Alexander Soare 7b8a0017f1 wip to review 2021-07-03 12:10:12 +01:00
Ross Wightman b41cffaa93 Fix a few issues loading pretrained vit/bit npz weights w/ num_classes=0 __init__ arg. Missed a few other small classifier handling detail on Mlp, GhostNet, Levit. Should fix #713 2021-06-22 23:16:05 -07:00