Ross Wightman
b9bd960a03
Merge pull request #421 from aboutmedicine/master
...
Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
4 years ago
Reuben
94ca140b67
update collections.abc import
4 years ago
Ross Wightman
db6128db0f
Merge pull request #418 from kecsap/master
...
Change the Python interpreter to Python 3.x in the scripts
4 years ago
Csaba Kertesz
5114c214fc
Change the Python interpreter to Python 3.x in the scripts
4 years ago
Ross Wightman
aaa715b1e9
Update README.md
4 years ago
Ross Wightman
a7f95818e4
Merge pull request #413 from rwightman/eca-weights
...
Add new weights for ecaresnet26t/50t/269d models.
4 years ago
Ross Wightman
a39c3ee216
Merge branch 'master' into eca-weights
4 years ago
Ross Wightman
e9d6fe293c
Update README for new weights. Version 0.4.2
4 years ago
Ross Wightman
666de85cf1
Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414
4 years ago
Ross Wightman
3b57490a63
Fix some half removed resnet model defs, pooling for ecaresnet269d
4 years ago
Ross Wightman
2a8c4dc63b
Add validation script update for using test_input_size in model default_cfgs
4 years ago
Ross Wightman
68a4144882
Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg.
4 years ago
Ross Wightman
3a7aa95f7e
Update README.md
4 years ago
Ross Wightman
b9843f954b
Merge pull request #282 from tigert1998/patch-1
...
Add symbolic for SwishJitAutoFn to support onnx
4 years ago
Ross Wightman
ea36a78cff
Merge pull request #401 from hwangdeyu/deyu/add_HardSwishJitAutoFn_operator
...
add HardSwishJitAutoFn operator export to onnx
4 years ago
hwangdeyu
7a4be5c035
add operator HardSwishJitAutoFn export to onnx
4 years ago
Ross Wightman
4203efa36d
Fix #387 so that checkpoint saver works with max history of 1. Add checkpoint-hist arg to train.py.
4 years ago
Ross Wightman
99b82ae5ab
Merge pull request #389 from rwightman/norm_free_models
...
Normalizer-Free RegNet and ResNet impl
4 years ago
Ross Wightman
f0e65e37b7
Fix NF-ResNet101 model defs
4 years ago
Ross Wightman
2c988c3b6e
Update README.md for NF-nets, bump version to 0.4.1 for merge
4 years ago
Ross Wightman
2de54d174a
Fix pool size defs for NFNet models, add a comment.
4 years ago
Ross Wightman
90980de4a9
Fix up a few details in NFResNet models, managed stable training. Add support for gamma gain to be applied in activation or ScaleStdConv. Some tweaks to ScaledStdConv.
4 years ago
Ross Wightman
5a8e1e643e
Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.
4 years ago
Ross Wightman
9a38416fbd
Merge pull request #323 from rwightman/imagenet21k_datasets_more
...
BiT (Big Transfer) ResNetV2 models, Official ViT Hybrid R50 weights, VIT IN21K weights updated w/ repr layer, ImageNet21k and dataset / parser refactor
4 years ago
Ross Wightman
745bc5f723
Mention 21k ViT weights in README.md
4 years ago
Ross Wightman
6306aa79e9
Merge branch 'master' into imagenet21k_datasets_more
4 years ago
Ross Wightman
38d8f67570
Fix potential issue with change to num_classes arg in train/validate.py defaulting to None (rely on model def / default_cfg)
4 years ago
Ross Wightman
587780e56b
Update README.md and bump version to 0.4.0
4 years ago
Ross Wightman
bb50ac4708
Add DeiT distilled weights and distilled model def. Remove some redudant ViT model args.
4 years ago
Ross Wightman
c16e965037
Add some ViT comments and fix a few minor issues.
4 years ago
Ross Wightman
22748f1a2d
Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py
4 years ago
Ross Wightman
5d4c3d0af3
Add enhanced ParserImageInTar that can read images from tars within tars, folders with multiple tars, etc. Additional comment cleanup.
4 years ago
Ross Wightman
55f7dfa9ea
Refactor vision_transformer entrpy fns, add pos embedding resize support for fine tuning, add some deit models for testing
4 years ago
Ross Wightman
9d5d4b8df6
Fix silly train.py typo during dataset work
4 years ago
Ross Wightman
d55bcc0fee
Finishing adding stochastic depth support to BiT ResNetV2 models
4 years ago
Ross Wightman
0a1668f63e
Update tests
4 years ago
Ross Wightman
58ccf43150
Add BiT references and knowledge distill links to readme/docs
4 years ago
Ross Wightman
855d6cc217
More dataset work including factories and a tensorflow datasets (TFDS) wrapper
...
* Add parser/dataset factory methods for more flexible dataset & parser creation
* Add dataset parser that wraps TFDS image classification datasets
* Tweak num_classes handling bug for 21k models
* Add initial deit models so they can be benchmarked in next csv results runs
4 years ago
Ross Wightman
f8463b8fa9
Version 0.3.4. Tweak setup.cfg and update setup.py metadata
4 years ago
Ross Wightman
20516abc18
Fix some broken tests for ResNetV2 BiT models
4 years ago
Ross Wightman
fd9061dbf7
Remove debug print from train.py
4 years ago
Ross Wightman
59ec7e6a53
Merge branch 'master' into imagenet21k_datasets_more
4 years ago
Ross Wightman
fc3d9183e8
Merge pull request #335 from kecsap/new_option
...
Add --input-size option to scripts to specify full input dimensions f…
4 years ago
Ross Wightman
e7a9ddf982
Merge pull request #334 from kecsap/links
...
Follow symbolic links during dataset scanning
4 years ago
Ross Wightman
19816fe226
Add citation info
4 years ago
Csaba Kertesz
e42b140ade
Add --input-size option to scripts to specify full input dimensions from command-line
4 years ago
Csaba Kertesz
7cae7e7035
Follow links during dataset scanning
4 years ago
Ross Wightman
1d01c2b68c
Update README.md
4 years ago
Ross Wightman
c96e9f99a0
Update version to 0.3.3
4 years ago
Ross Wightman
a7d0a8b5b2
Update results csv files with latest models, incl 101D, 152D, 200D, SE152D ResNets and yet to be merged BiT and ViT-R50 models.
4 years ago