dc376e3676
Ensure all model entrypoint fn default to `pretrained=False` (a few didn't)
2 years ago
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
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
4547920f85
Merge pull request #1354 from rwightman/fix_tests
...
Attempting to fix unit test failures...
2 years ago
29afe79c8b
Attempt to fix unit tests by removing subset of tests on mac runner
2 years ago
326ade2999
Add updated validation / test set results, benchmarks still running...
2 years ago
92b91af3bb
version 0.6.6
2 years ago
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
4283c0c478
Merge pull request #1351 from nateraw/use-hf-hub-download
...
Use hf_hub_download instead of cached_download
2 years ago
51cca82aa1
👽 use hf_hub_download instead of cached_download
2 years ago
324a4e58b6
disable nvfuser for jit te/legacy modes (for PT 1.12+)
2 years ago
f96da54eb1
Merge pull request #1345 from mmaaz60/patch-1
...
Update README.md
2 years ago
4e7ffe5043
Update README.md
2 years ago
2898cf6e41
version 0.6.5 for pypi release
2 years ago
66393d472f
Update README.md
2 years ago
a45b4bce9a
x and xx small edgenext models do benefit from larger test input size
2 years ago
a8e34051c1
Unbreak gamma remap impacting beit checkpoint load, version bump to 0.6.4
2 years ago
1ccce50d48
Merge pull request #1327 from rwightman/edgenext_csp_and_more
...
EdgeNeXt, additional DarkNets, and more
2 years ago
1c5cb819f9
bump version to 0.6.3 before merge
2 years ago
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
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
ce65a7b29f
Update vit_relpos w/ some additional weights, some cleanup to match recent vit updates, more MLP log coord experiments.
2 years ago
58621723bd
Add CrossStage3 DarkNet (cs3) weights
2 years ago
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
4670d375c6
Reorg benchmark.py import
2 years ago
2456223052
Merge pull request #1336 from xwang233/add-local-rank
...
Make train.py compatible with torchrun
2 years ago
500c190860
Add --aot-autograd (functorch efficient mem fusion) support to validate.py
2 years ago
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
11060f84c5
make train.py compatible with torchrun
2 years ago
db0cee9910
Refactor cspnet configuration using dataclasses, update feature extraction for new cs3 variants.
2 years ago
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
06307b8b41
Remove experimental downsample in block support in ConvNeXt. Experiment further before keeping it in.
2 years ago
bfc0dccb0e
Improve image extension handling, add methods to modify / get defaults. Fix #1335 fix #1274 .
2 years ago
7d4b3807d5
Support DeiT-3 (Revenge of the ViT) checkpoints. Add non-overlapping (w/ class token) pos-embed support to vit.
2 years ago
d0c5bd5722
Rename cs2->cs3 for darknets. Fix features_only for cs3 darknets.
2 years ago
d765305821
Remove first_conv for resnetaa50 def
2 years ago
dd9b8f57c4
Add feature_info to edgenext for features_only support, hopefully fix some fx / test errors
2 years ago
377e9bfa21
Add TPU trained darknet53 weights. Add mising pretrain_cfg for some csp/darknet models.
2 years ago
c170ba3173
Add weights for resnet10t, resnet14t, and resnetaa50 models. Fix #1314
2 years ago
188c194b0f
Left some experiment stem code in convnext by mistake
2 years ago
70d6d2c484
support test_crop_size in data config resolve
2 years ago
6064d16a2d
Add initial EdgeNeXt import. Significant cleanup / reorg (like ConvNeXt). Fix #1320
...
* edgenext refactored for torchscript compat, stage base organization
* slight refactor of ConvNeXt to match some EdgeNeXt additions
* remove use of funky LayerNorm layer in ConvNeXt and just use nn.LayerNorm and LayerNorm2d (permute)
2 years ago
7a9c6811c9
Add eps arg to LayerNorm2d, add 'tf' (tensorflow) variant of trunc_normal_ that applies scale/shift after sampling (instead of needing to move a/b)
2 years ago
82c311d082
Add more experimental darknet and 'cs2' darknet variants (different cross stage setup, closer to newer YOLO backbones) for train trials.
2 years ago
a050fde5cd
Add resnet10t (basic block) and resnet14t (bottleneck) with 1,1,1,1 repeats
2 years ago
34f382f8f6
move dataconfig before script, scripting killing metadata now (PyTorch 1.12? just nvfuser?)
2 years ago
beef62e7ab
Merge pull request #1317 from rwightman/fixes-syncbn_pretrain_cfg_resolve
...
Fix SyncBatchNorm for BatchNormAc2d, improve resolve_pretrained_cfg behaviour, other mix fixes.
2 years ago
e6d7df40ec
no longer a point using kwargs for pretrain_cfg resolve, just pass explicit arg
2 years ago
a29fba307d
disable dist_bn when sync_bn active
2 years ago
07d0c4ae96
Improve repr for DropPath module
2 years ago
Ross Wightman