14ac4abf74
Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example.
2021-03-18 11:38:06 -07:00
740f32c96a
Add ECA-NFNet-L0 weights and update model name. Update README and bump version to 0.4.6
2021-03-17 13:55:32 -07:00
5e2e4e7fb6
Merge pull request #502 from khatami-mehrdad/efficientnet_dg
...
adding efficientnet_el, efficientnet_es_pruned and efficientnet_el_pruned pre-trained models
2021-03-17 13:35:22 -07:00
3eac7dc5a3
Merge pull request #501 from rwightman/hf_hub_revisit
...
Support for huggingface hub via create_model and default_cfgs.
2021-03-17 13:26:50 -07:00
45c048ba13
A few minor fixes and bit more cleanup on the huggingface hub integration.
2021-03-17 13:18:52 -07:00
97f49d163c
adding efficientnet_el, efficientnet_es_prunedefficientnet_es_pruned and efficientnet_el_pruned pretrained models to timm/models
2021-03-17 10:51:51 -07:00
ead80d33c5
Fix typo, naming consistency
2021-03-16 23:02:51 -07:00
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
e5ba5dcbc9
Merge branch 'yoniaflalo-adding_Hardcore_NAS'
2021-03-15 10:21:18 -07:00
1f799afcb3
Remove capital letters in model names
2021-03-15 10:20:34 -07:00
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
82c1fe958a
added hardcorenas model
...
refactoring to support feature extraction
2021-03-15 08:06:34 +02:00
06aa926fbf
Tweak docs model summaries / pages names and layout
2021-03-14 12:45:20 -07:00
ac9a2c978d
Update docs requirements
2021-03-14 12:26:31 -07:00
dda8fd80d2
Merge branch 'gcucurull-master'
2021-03-14 12:25:08 -07:00
74fcd4c50e
Update finetune detail, rerun model doc gen.
2021-03-14 12:22:59 -07:00
e6f5617dcc
Fix a few typos and consistency issues in model docs
2021-03-14 12:20:29 -07:00
5392c0a4b3
Rerun modelindex generate script
2021-03-14 12:08:56 -07:00
36a5648208
Merge branch 'master' of https://github.com/gcucurull/pytorch-image-models into gcucurull-master
2021-03-14 12:01:21 -07:00
dfae015c5d
Move modelindex files into doc tree, add headings to model templates. Tweak a few details.
2021-03-14 12:00:11 -07:00
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
4d26eed4d4
fix typo in urls
2021-03-12 12:53:43 +00:00
3f6501c1ad
Add short comment
2021-03-11 17:09:32 +00:00
9e74125276
Model index files and script to generate it
2021-03-11 16:52:13 +00:00
5c7d298234
Update README.md, documentation, add TOC, sponsors, and links to timmdocs and paperswithcode
2021-03-10 10:24:23 -08:00
f1f332fee5
Update README.md
2021-03-07 16:43:52 -08:00
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
f57db99101
Update README, fix iabn pip version print.
2021-03-07 16:17:06 -08:00
7ef7788ee9
Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk()
2021-03-06 16:56:34 -08:00
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
4445eaa470
Add img_size to benchmark output
2021-03-05 16:48:31 -08:00
5fcdd4b822
Merge pull request #459 from onatyap/patch-2
...
corrected typo in feature_extraction.md
2021-03-05 10:27:47 -08:00
17cdee7354
Fix C&P patch_size error, and order of op patch_size arg resolution bug. Remove a test vit model.
2021-03-01 16:53:32 -08:00
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
0706d05d52
Benchmark models listed in txt file. Add more hybrid vit variants for testing
2021-02-28 16:00:33 -08:00
3b95fa4248
corrected typo in feature_extraction.md
...
severay changed to several
2021-02-28 16:25:00 +03:00
2db2d87ff7
Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.
2021-02-23 17:31:42 -08:00
de97be9146
Spell out diff between my small and deit small vit models.
2021-02-23 16:22:55 -08:00
f0ffdf89b3
Add numerous experimental ViT Hybrid models w/ ResNetV2 base. Update the ViT naming for hybrids. Fix #426 for pretrained vit resizing.
2021-02-23 15:54:55 -08:00
0e16d4e9fb
Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.
2021-02-23 15:38:12 -08:00
4bc103f504
Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk()
2021-02-23 13:15:52 -08:00
48371a33b1
Create FUNDING.yml
...
$cale up the training...
2021-02-18 16:12:53 -08:00
da4839530c
Fix test model filter to include dm_ variants that break GitHub CI limits
2021-02-18 13:43:04 -08:00
8563609b28
Update notes in ScaledStdConv impl
2021-02-18 12:44:08 -08:00
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
4ea5931964
Merge pull request #437 from rwightman/agc
...
Adaptive Gradient Clipping (AGC) Impl
2021-02-16 10:27:58 -08:00
361fd0fc40
Update README.md
2021-02-16 10:27:41 -08:00
9de2ec5e44
Update README for AGC and bump version to 0.4.4
2021-02-16 09:13:03 -08:00
01653db104
Missed clip-mode arg for repo train script
2021-02-15 23:27:16 -08:00
4f49b94311
Initial AGC impl. Still testing.
2021-02-15 23:22:44 -08:00
Ross Wightman
Ross Wightman
Mehrdad
AFLALO, Jonathan Isaac
Marcin Kardas
Guillem Cucurull
Onat Yapici