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