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

1620 Commits (927f031293a30afb940fff0bee34b85d9c059b0e)
 

Author SHA1 Message Date
Ross Wightman 56596e4e84 jit trace comparisons snuck into torchscript part of validate.py, fixed 2022-07-31 21:13:56 -07:00
Ross Wightman 8ad4bdfa06 Allow ntuple to be used with string values 2022-07-28 16:18:18 -07:00
Christoph Reich faae93e62d
Fix typo in PositionalEncodingFourier 2022-07-28 19:08:08 -04:00
Ross Wightman 7430a85d07 Update README, bump version to 0.6.8 2022-07-28 15:07:11 -07:00
Ross Wightman ec6a28830f Add DeiT-III 'medium' model defs and weights 2022-07-28 15:03:20 -07:00
Ross Wightman 7cd4204a28 Add TPU TRC acknowledge 2022-07-27 14:07:37 -07:00
Ross Wightman 7d44d65bf5 Update README and changelogs 2022-07-27 14:04:29 -07:00
Ross Wightman d875a1d3f6 version 0.6.7 2022-07-27 12:41:06 -07:00
Ross Wightman c865028c34 Update benchmark with latest model adds 2022-07-27 12:40:22 -07:00
Ross Wightman 30bd1746c5 Improve csv table result processing for better sort when updating 2022-07-27 12:26:52 -07:00
Ross Wightman e987e29036 Add convnext_nano and few cs3 models to existing results tables 2022-07-27 12:26:34 -07:00
Ross Wightman 6f103a442b Add convnext_nano weights, 80.8 @ 224, 81.5 @ 288 2022-07-26 16:40:27 -07:00
Ross Wightman 4042a94f8f Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks. 2022-07-26 16:40:27 -07:00
Ross Wightman c8f69e04a9
Merge pull request #1365 from veritable-tech/fix-resize-pos-embed 
Take `no_emb_class` into account when calling  `resize_pos_embed`
 2022-07-24 21:03:01 -07:00
Ross Wightman 99af63ca92
Merge pull request #1277 from lukasugar/patch-1 
Add missing output in Feature extraction docs
 2022-07-24 15:35:48 -07:00
Ross Wightman 45c447fc15
Merge pull request #1363 from Jasha10/patch-1 
Update type hint for `register_notrace_module`
 2022-07-24 15:34:14 -07:00
Ceshine Lee 0b64117592 Take `no_emb_class` into account when calling `resize_pos_embed` 2022-07-24 19:11:45 +08:00
Jasha10 56c3a84db3
Update type hint for `register_notrace_module` 
register_notrace_module is used to decorate types (i.e. subclasses of nn.Module).
It is not called on module instances.
 2022-07-22 16:59:55 -05:00
Ross Wightman d7b55a9429 Add gmacs and macts columns to inference benchmark (missed profile in initial run) 2022-07-22 12:12:37 -07:00
Ross Wightman 1b278136c3 Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355 2022-07-21 17:36:15 -07:00
Ross Wightman 909705e7ff Remove some redundant requires_grad=True from nn.Parameter in third party code 2022-07-20 12:37:41 -07:00
Ross Wightman c5e0d1c700 Add dilation support to convnext, allows output_stride=8 and 16 use. Fix #1341 2022-07-19 17:52:10 -07:00
Ross Wightman 5e7d47ca10 Add pytorch 1.12 benchmark csv files w/ 0.6.6 code. Remove pytorch 1.10 results. Deciding whether to update 1.11 results or remove... 2022-07-19 15:46:03 -07:00
Ross Wightman dc376e3676 Ensure all model entrypoint fn default to `pretrained=False` (a few didn't) 2022-07-19 13:58:41 -07:00
Ross Wightman 23b102064a Add cs3sedarknet_x weights w/ 82.65 @ 288 top1. Add 2 cs3 edgenet models (w/ 3x3-1x1 block), remove aa from cspnet blocks (not needed) 2022-07-19 13:56:44 -07:00
Ross Wightman 0dbd9352ce Add bulk_runner script and updates to benchmark.py and validate.py for better error handling in bulk runs (used for benchmark and validation result runs). Improved batch size decay stepping on retry... 2022-07-18 18:04:54 -07:00
Ross Wightman 4547920f85
Merge pull request #1354 from rwightman/fix_tests 
Attempting to fix unit test failures...
 2022-07-17 21:37:16 -07:00
Ross Wightman 29afe79c8b Attempt to fix unit tests by removing subset of tests on mac runner 2022-07-17 14:55:47 -07:00
Ross Wightman 326ade2999 Add updated validation / test set results, benchmarks still running... 2022-07-16 16:15:28 -07:00
Ross Wightman 92b91af3bb version 0.6.6 2022-07-15 17:00:18 -07:00
Ross Wightman 05313940e2 Add cs3darknet_x, cs3sedarknet_l, and darknetaa53 weights from TPU sessions. Move SE btwn conv1 & conv2 in DarkBlock. Improve SE/attn handling in Csp/DarkNet. Fix leaky_relu bug on older csp models. 2022-07-15 16:55:16 -07:00
Ross Wightman 4283c0c478
Merge pull request #1351 from nateraw/use-hf-hub-download 
Use hf_hub_download instead of cached_download
 2022-07-14 14:51:02 -07:00
nateraw 51cca82aa1 👽 use hf_hub_download instead of cached_download 2022-07-14 16:41:45 -04:00
Ross Wightman 324a4e58b6 disable nvfuser for jit te/legacy modes (for PT 1.12+) 2022-07-13 10:34:34 -07:00
Ross Wightman f96da54eb1
Merge pull request #1345 from mmaaz60/patch-1 
Update README.md
 2022-07-11 20:34:46 -07:00
Muhammad Maaz 4e7ffe5043
Update README.md 2022-07-12 05:08:34 +04:00
Ross Wightman 2898cf6e41 version 0.6.5 for pypi release 2022-07-10 16:43:23 -07:00
Ross Wightman 66393d472f Update README.md 2022-07-08 12:21:23 -07:00
Ross Wightman a45b4bce9a x and xx small edgenext models do benefit from larger test input size 2022-07-08 10:53:27 -07:00
Ross Wightman a8e34051c1 Unbreak gamma remap impacting beit checkpoint load, version bump to 0.6.4 2022-07-07 23:07:43 -07:00
Ross Wightman 1ccce50d48
Merge pull request #1327 from rwightman/edgenext_csp_and_more 
EdgeNeXt, additional DarkNets, and more
 2022-07-07 22:06:47 -07:00
Ross Wightman 1c5cb819f9 bump version to 0.6.3 before merge 2022-07-07 22:05:56 -07:00
Ross Wightman a1cb25066e Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights. 2022-07-07 22:02:57 -07:00
Ross Wightman 7c7ecd2492 Add --use-train-size flag to force use of train input_size (over test input size) for validation. Default test-time pooling to use train input size (fixes issues). 2022-07-07 22:01:24 -07:00
Ross Wightman ce65a7b29f Update vit_relpos w/ some additional weights, some cleanup to match recent vit updates, more MLP log coord experiments. 2022-07-07 21:33:25 -07:00
Ross Wightman 58621723bd Add CrossStage3 DarkNet (cs3) weights 2022-07-07 17:43:38 -07:00
Ross Wightman 9be0c84715 Change set -> dict w/ None keys for dataset split synonym search, so always consistent if more than 1 exists. Fix #1224 2022-07-07 15:33:53 -07:00
Ross Wightman 4670d375c6 Reorg benchmark.py import 2022-07-07 15:21:29 -07:00
Ross Wightman 2456223052
Merge pull request #1336 from xwang233/add-local-rank 
Make train.py compatible with torchrun
 2022-07-07 15:20:30 -07:00
Ross Wightman 500c190860 Add --aot-autograd (functorch efficient mem fusion) support to validate.py 2022-07-07 15:15:25 -07:00