Sylvain Gugger
|
aa6daf20af
|
Add dependency everywhere
|
4 years ago |
|
Sylvain Gugger
|
614b73bd35
|
Add dependency
|
4 years ago |
|
Sylvain Gugger
|
f269f2d9d5
|
Consistently use hf_model_id
|
4 years ago |
|
Sylvain Gugger
|
482ab548dc
|
Merge branch 'master' into master
|
4 years ago |
|
Julien Chaumond
|
b506f7bc36
|
Add a second example model
cc @sgugger
|
4 years ago |
|
Julien Chaumond
|
dd12f9f2a5
|
Change default weights filename + default branch is main
cc @sgugger
|
4 years ago |
|
Ross Wightman
|
4ea5931964
|
Merge pull request #437 from rwightman/agc
Adaptive Gradient Clipping (AGC) Impl
|
4 years ago |
|
Ross Wightman
|
361fd0fc40
|
Update README.md
|
4 years ago |
|
Ross Wightman
|
9de2ec5e44
|
Update README for AGC and bump version to 0.4.4
|
4 years ago |
|
Ross Wightman
|
01653db104
|
Missed clip-mode arg for repo train script
|
4 years ago |
|
Ross Wightman
|
4f49b94311
|
Initial AGC impl. Still testing.
|
4 years ago |
|
Ross Wightman
|
5f9aff395c
|
Fix stem width in NFNet-F models, add some more comments, add some 'light' NFNet models for testing.
|
4 years ago |
|
Ross Wightman
|
4df513c68f
|
Merge pull request #427 from rwightman/nfnet
Add NFNet-F models and tweak existing NF models.
|
4 years ago |
|
Ross Wightman
|
d86dbe45c2
|
Update README.md and few more comments
|
4 years ago |
|
Ross Wightman
|
0d253e2c5e
|
Fix issue with nfnet tests, bit more cleanup.
|
4 years ago |
|
Ross Wightman
|
cb06c7a910
|
Add NFNet-F models and tweak existing NF models.
|
4 years ago |
|
Ross Wightman
|
607f9149b1
|
Update README.md
|
5 years ago |
|
Sylvain Gugger
|
8c270ed7bf
|
Merge pull request #1 from sgugger/add_hf_hub
PoC for using HF model hub
|
5 years ago |
|
Ross Wightman
|
e4de077021
|
Add first 'Normalizer Free' models. nf_regnet_b1 79.3 @ 288x288 test, and nf_resnet50 80.3 @ 256x256 test (80.68 @ 288x288).
|
5 years ago |
|
Sylvain Gugger
|
5482d2e501
|
Actually load the model
|
5 years ago |
|
Sylvain Gugger
|
7dafa71c0e
|
Use the model hub config
|
5 years ago |
|
Sylvain Gugger
|
4d0b5feed0
|
Make `create_model` work from a checkpoint
|
5 years ago |
|
Sylvain Gugger
|
9857e12c0c
|
Be more opt-in
|
5 years ago |
|
Sylvain Gugger
|
ebe69dd4d3
|
PoC for using HF model hub
|
5 years ago |
|
Ross Wightman
|
d8e69206be
|
Merge pull request #419 from rwightman/byob_vgg_models
More models, GPU-Efficient Nets, RepVGG, classic VGG, and flexible Byob backbone.
|
5 years ago |
|
Ross Wightman
|
ca9b078ac7
|
Update README.md and docs. Version bumped to 0.4.3
|
5 years ago |
|
Ross Wightman
|
6853b07bbd
|
Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments.
|
5 years ago |
|
Ross Wightman
|
0356e773f5
|
Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately.
|
5 years ago |
|
Ross Wightman
|
b9bd960a03
|
Merge pull request #421 from aboutmedicine/master
Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
|
5 years ago |
|
Reuben
|
94ca140b67
|
update collections.abc import
|
5 years ago |
|
Ross Wightman
|
db6128db0f
|
Merge pull request #418 from kecsap/master
Change the Python interpreter to Python 3.x in the scripts
|
5 years ago |
|
Ross Wightman
|
b4e216e377
|
Fix a few small things.
|
5 years ago |
|
Ross Wightman
|
dc85e5a237
|
Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models.
|
5 years ago |
|
Ross Wightman
|
1bcc69e0ad
|
Use in_channels for depthwise groups, allows using `out_channels=N * in_channels` (does not impact existing models). Fix #354.
|
5 years ago |
|
Ross Wightman
|
9811e229f7
|
Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.
|
5 years ago |
|
Csaba Kertesz
|
5114c214fc
|
Change the Python interpreter to Python 3.x in the scripts
|
5 years ago |
|
Ross Wightman
|
aaa715b1e9
|
Update README.md
|
5 years ago |
|
Ross Wightman
|
a7f95818e4
|
Merge pull request #413 from rwightman/eca-weights
Add new weights for ecaresnet26t/50t/269d models.
|
5 years ago |
|
Ross Wightman
|
a39c3ee216
|
Merge branch 'master' into eca-weights
|
5 years ago |
|
Ross Wightman
|
e9d6fe293c
|
Update README for new weights. Version 0.4.2
|
5 years ago |
|
Ross Wightman
|
666de85cf1
|
Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414
|
5 years ago |
|
Ross Wightman
|
3b57490a63
|
Fix some half removed resnet model defs, pooling for ecaresnet269d
|
5 years ago |
|
Ross Wightman
|
2a8c4dc63b
|
Add validation script update for using test_input_size in model default_cfgs
|
5 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.
|
5 years ago |
|
Ross Wightman
|
3a7aa95f7e
|
Update README.md
|
5 years ago |
|
Ross Wightman
|
b9843f954b
|
Merge pull request #282 from tigert1998/patch-1
Add symbolic for SwishJitAutoFn to support onnx
|
5 years ago |
|
Ross Wightman
|
ea36a78cff
|
Merge pull request #401 from hwangdeyu/deyu/add_HardSwishJitAutoFn_operator
add HardSwishJitAutoFn operator export to onnx
|
5 years ago |
|
hwangdeyu
|
7a4be5c035
|
add operator HardSwishJitAutoFn export to onnx
|
5 years ago |
|
Ross Wightman
|
4203efa36d
|
Fix #387 so that checkpoint saver works with max history of 1. Add checkpoint-hist arg to train.py.
|
5 years ago |
|
Ross Wightman
|
99b82ae5ab
|
Merge pull request #389 from rwightman/norm_free_models
Normalizer-Free RegNet and ResNet impl
|
5 years ago |
