Commit Graph

746 Commits (ead80d33c59ca18acaaa594159ca28b6116e9acb)
 

Author SHA1 Message Date
Ross Wightman ead80d33c5 Fix typo, naming consistency 2021-03-16 23:02:51 -07:00
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs.
* improve consistency of model creation helper fns
* add comments to some of the model helpers
* support passing external default_cfgs so they can be sourced from hub
2021-03-16 22:48:26 -07:00
Ross Wightman e5ba5dcbc9 Merge branch 'yoniaflalo-adding_Hardcore_NAS' 2021-03-15 10:21:18 -07:00
Ross Wightman 1f799afcb3 Remove capital letters in model names 2021-03-15 10:20:34 -07:00
Ross Wightman 52b5d0ad0a Merge branch 'adding_Hardcore_NAS' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_Hardcore_NAS 2021-03-15 09:37:25 -07:00
AFLALO, Jonathan Isaac 82c1fe958a added hardcorenas model
refactoring to support feature extraction
2021-03-15 08:06:34 +02:00
Ross Wightman 06aa926fbf Tweak docs model summaries / pages names and layout 2021-03-14 12:45:20 -07:00
Ross Wightman ac9a2c978d Update docs requirements 2021-03-14 12:26:31 -07:00
Ross Wightman dda8fd80d2 Merge branch 'gcucurull-master' 2021-03-14 12:25:08 -07:00
Ross Wightman 74fcd4c50e Update finetune detail, rerun model doc gen. 2021-03-14 12:22:59 -07:00
Ross Wightman e6f5617dcc Fix a few typos and consistency issues in model docs 2021-03-14 12:20:29 -07:00
Ross Wightman 5392c0a4b3 Rerun modelindex generate script 2021-03-14 12:08:56 -07:00
Ross Wightman 36a5648208 Merge branch 'master' of https://github.com/gcucurull/pytorch-image-models into gcucurull-master 2021-03-14 12:01:21 -07:00
Ross Wightman dfae015c5d Move modelindex files into doc tree, add headings to model templates. Tweak a few details. 2021-03-14 12:00:11 -07:00
Marcin Kardas 8936356139 Add results to model-index
Adds evaluation results of models and fixes formatting of yml files.
2021-03-12 19:52:53 +00:00
Guillem Cucurull 4d26eed4d4 fix typo in urls 2021-03-12 12:53:43 +00:00
Guillem Cucurull 3f6501c1ad Add short comment 2021-03-11 17:09:32 +00:00
Guillem Cucurull 9e74125276 Model index files and script to generate it 2021-03-11 16:52:13 +00:00
Ross Wightman 5c7d298234 Update README.md, documentation, add TOC, sponsors, and links to timmdocs and paperswithcode 2021-03-10 10:24:23 -08:00
Ross Wightman f1f332fee5
Update README.md 2021-03-07 16:43:52 -08:00
Ross Wightman 5b28ef4100
Merge pull request #475 from rwightman/pre-release/0.4.5
Prep for PyPi release. Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features. Also, add act args to RexNet #202
2021-03-07 16:18:31 -08:00
Ross Wightman f57db99101 Update README, fix iabn pip version print. 2021-03-07 16:17:06 -08:00
Ross Wightman 7ef7788ee9 Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-03-06 16:56:34 -08:00
Ross Wightman 317ea3e599 Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump. 2021-03-06 12:48:06 -08:00
Ross Wightman 5fcdd4b822
Merge pull request #459 from onatyap/patch-2
corrected typo in feature_extraction.md
2021-03-05 10:27:47 -08:00
Ross Wightman 95feb1da41 Update results csvs, run with PyTorch 1.7.1 cuda 10.2, some differences from prev run with older PT ver. 2021-02-28 17:43:22 -08:00
Onat Yapici 3b95fa4248
corrected typo in feature_extraction.md
severay changed to several
2021-02-28 16:25:00 +03:00
Ross Wightman 48371a33b1
Create FUNDING.yml
$cale up the training...
2021-02-18 16:12:53 -08:00
Ross Wightman da4839530c Fix test model filter to include dm_ variants that break GitHub CI limits 2021-02-18 13:43:04 -08:00
Ross Wightman 8563609b28 Update notes in ScaledStdConv impl 2021-02-18 12:44:08 -08:00
Ross Wightman 678ba4e0a2 Add NFNet-F model weights ported from DeepMind Haiku impl and new set of models w/ compatible config. 2021-02-18 12:28:46 -08:00
Ross Wightman 4ea5931964
Merge pull request #437 from rwightman/agc
Adaptive Gradient Clipping (AGC) Impl
2021-02-16 10:27:58 -08:00
Ross Wightman 361fd0fc40
Update README.md 2021-02-16 10:27:41 -08:00
Ross Wightman 9de2ec5e44 Update README for AGC and bump version to 0.4.4 2021-02-16 09:13:03 -08:00
Ross Wightman 01653db104 Missed clip-mode arg for repo train script 2021-02-15 23:27:16 -08:00
Ross Wightman 4f49b94311 Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00
Ross Wightman 5f9aff395c Fix stem width in NFNet-F models, add some more comments, add some 'light' NFNet models for testing. 2021-02-13 16:58:51 -08:00
Ross Wightman 4df513c68f
Merge pull request #427 from rwightman/nfnet
Add NFNet-F models and tweak existing NF models.
2021-02-12 22:08:03 -08:00
Ross Wightman d86dbe45c2 Update README.md and few more comments 2021-02-12 22:07:18 -08:00
Ross Wightman 0d253e2c5e Fix issue with nfnet tests, bit more cleanup. 2021-02-12 21:05:41 -08:00
Ross Wightman cb06c7a910 Add NFNet-F models and tweak existing NF models. 2021-02-12 18:28:56 -08:00
Ross Wightman 607f9149b1
Update README.md 2021-02-11 22:06:06 -08:00
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). 2021-02-11 13:20:11 -08:00
Ross Wightman d8e69206be
Merge pull request #419 from rwightman/byob_vgg_models
More models, GPU-Efficient Nets, RepVGG, classic VGG, and flexible Byob backbone.
2021-02-10 15:44:09 -08:00
Ross Wightman ca9b078ac7 Update README.md and docs. Version bumped to 0.4.3 2021-02-10 14:46:07 -08:00
Ross Wightman 6853b07bbd Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments. 2021-02-10 14:40:29 -08:00
Ross Wightman 0356e773f5 Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately. 2021-02-10 14:31:18 -08:00
Ross Wightman b9bd960a03
Merge pull request #421 from aboutmedicine/master
Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
2021-02-10 08:11:49 -08:00
Reuben 94ca140b67 update collections.abc import 2021-02-10 23:54:35 +11:00
Ross Wightman db6128db0f
Merge pull request #418 from kecsap/master
Change the Python interpreter to Python 3.x in the scripts
2021-02-09 20:55:55 -08:00