Ross Wightman
|
6f103a442b
|
Add convnext_nano weights, 80.8 @ 224, 81.5 @ 288
|
2 years ago |
|
Ross Wightman
|
4042a94f8f
|
Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks.
|
2 years ago |
|
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`
|
2 years ago |
|
Ross Wightman
|
99af63ca92
|
Merge pull request #1277 from lukasugar/patch-1
Add missing output in Feature extraction docs
|
2 years ago |
|
Ross Wightman
|
45c447fc15
|
Merge pull request #1363 from Jasha10/patch-1
Update type hint for `register_notrace_module`
|
2 years ago |
|
Ceshine Lee
|
0b64117592
|
Take `no_emb_class` into account when calling `resize_pos_embed`
|
2 years ago |
|
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.
|
2 years ago |
|
Ross Wightman
|
d7b55a9429
|
Add gmacs and macts columns to inference benchmark (missed profile in initial run)
|
2 years ago |
|
Ross Wightman
|
1b278136c3
|
Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355
|
2 years ago |
|
Ross Wightman
|
909705e7ff
|
Remove some redundant requires_grad=True from nn.Parameter in third party code
|
2 years ago |
|
Ross Wightman
|
c5e0d1c700
|
Add dilation support to convnext, allows output_stride=8 and 16 use. Fix #1341
|
2 years ago |
|
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...
|
2 years ago |
|
Ross Wightman
|
dc376e3676
|
Ensure all model entrypoint fn default to `pretrained=False` (a few didn't)
|
2 years ago |
|
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)
|
2 years ago |
|
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...
|
2 years ago |
|
Ross Wightman
|
4547920f85
|
Merge pull request #1354 from rwightman/fix_tests
Attempting to fix unit test failures...
|
2 years ago |
|
Ross Wightman
|
29afe79c8b
|
Attempt to fix unit tests by removing subset of tests on mac runner
|
2 years ago |
|
Ross Wightman
|
326ade2999
|
Add updated validation / test set results, benchmarks still running...
|
2 years ago |
|
Ross Wightman
|
92b91af3bb
|
version 0.6.6
|
2 years ago |
|
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.
|
2 years ago |
|
Ross Wightman
|
4283c0c478
|
Merge pull request #1351 from nateraw/use-hf-hub-download
Use hf_hub_download instead of cached_download
|
2 years ago |
|
nateraw
|
51cca82aa1
|
👽 use hf_hub_download instead of cached_download
|
2 years ago |
|
Ross Wightman
|
324a4e58b6
|
disable nvfuser for jit te/legacy modes (for PT 1.12+)
|
2 years ago |
|
Ross Wightman
|
f96da54eb1
|
Merge pull request #1345 from mmaaz60/patch-1
Update README.md
|
2 years ago |
|
Muhammad Maaz
|
4e7ffe5043
|
Update README.md
|
2 years ago |
|
Ross Wightman
|
2898cf6e41
|
version 0.6.5 for pypi release
|
2 years ago |
|
Ross Wightman
|
66393d472f
|
Update README.md
|
2 years ago |
|
Ross Wightman
|
a45b4bce9a
|
x and xx small edgenext models do benefit from larger test input size
|
2 years ago |
|
Ross Wightman
|
a8e34051c1
|
Unbreak gamma remap impacting beit checkpoint load, version bump to 0.6.4
|
2 years ago |
|
Ross Wightman
|
1ccce50d48
|
Merge pull request #1327 from rwightman/edgenext_csp_and_more
EdgeNeXt, additional DarkNets, and more
|
2 years ago |
|
Ross Wightman
|
1c5cb819f9
|
bump version to 0.6.3 before merge
|
2 years ago |
|
Ross Wightman
|
a1cb25066e
|
Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights.
|
2 years ago |
|
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).
|
2 years ago |
|
Ross Wightman
|
ce65a7b29f
|
Update vit_relpos w/ some additional weights, some cleanup to match recent vit updates, more MLP log coord experiments.
|
2 years ago |
|
Ross Wightman
|
58621723bd
|
Add CrossStage3 DarkNet (cs3) weights
|
2 years ago |
|
Ross Wightman
|
9be0c84715
|
Change set -> dict w/ None keys for dataset split synonym search, so always consistent if more than 1 exists. Fix #1224
|
2 years ago |
|
Ross Wightman
|
4670d375c6
|
Reorg benchmark.py import
|
2 years ago |
|
Ross Wightman
|
2456223052
|
Merge pull request #1336 from xwang233/add-local-rank
Make train.py compatible with torchrun
|
2 years ago |
|
Ross Wightman
|
500c190860
|
Add --aot-autograd (functorch efficient mem fusion) support to validate.py
|
2 years ago |
|
Ross Wightman
|
28e0152043
|
Add --no-retry flag to benchmark.py to skip batch_size decay and retry on error. Fix #1226. Update deepspeed profile usage for latest DS releases. Fix # 1333
|
2 years ago |
|
Xiao Wang
|
11060f84c5
|
make train.py compatible with torchrun
|
2 years ago |
|
Ross Wightman
|
db0cee9910
|
Refactor cspnet configuration using dataclasses, update feature extraction for new cs3 variants.
|
2 years ago |
|
Ross Wightman
|
eca09b8642
|
Add MobileVitV2 support. Fix #1332. Move GroupNorm1 to common layers (used in poolformer + mobilevitv2). Keep ol custom ConvNeXt LayerNorm2d impl as LayerNormExp2d for reference.
|
2 years ago |
|
Ross Wightman
|
06307b8b41
|
Remove experimental downsample in block support in ConvNeXt. Experiment further before keeping it in.
|
2 years ago |
|
Ross Wightman
|
bfc0dccb0e
|
Improve image extension handling, add methods to modify / get defaults. Fix #1335 fix #1274.
|
2 years ago |
|
Ross Wightman
|
7d4b3807d5
|
Support DeiT-3 (Revenge of the ViT) checkpoints. Add non-overlapping (w/ class token) pos-embed support to vit.
|
3 years ago |
|
Ross Wightman
|
d0c5bd5722
|
Rename cs2->cs3 for darknets. Fix features_only for cs3 darknets.
|
3 years ago |
|
Ross Wightman
|
d765305821
|
Remove first_conv for resnetaa50 def
|
3 years ago |
|
Ross Wightman
|
dd9b8f57c4
|
Add feature_info to edgenext for features_only support, hopefully fix some fx / test errors
|
3 years ago |
|
Ross Wightman
|
377e9bfa21
|
Add TPU trained darknet53 weights. Add mising pretrain_cfg for some csp/darknet models.
|
3 years ago |
