7077f16c6a
Change 21k model naming from _21k to _in21k for consistency with existing 21k models.
2021-05-15 12:42:26 -07:00
94d4b53352
Add temporary default_cfgs to visformer models so they pass tests
2021-05-15 08:41:31 -07:00
3bffc701f1
Merge branch 'master' into levit_visformer_rednet
2021-05-14 23:02:12 -07:00
ecc7552c5c
Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.
2021-05-14 17:16:34 -07:00
165fb354b2
Add initial RedNet model / Involution layer impl for testing
2021-05-14 17:16:34 -07:00
328249f11a
Update README, tweak fine-tune effv2 model names.
2021-05-14 16:37:43 -07:00
c4f482a08b
EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.
2021-05-14 15:50:00 -07:00
4fbc32d3d0
Fix crop_pct for cait models.
2021-05-11 10:50:15 -07:00
715519a5ef
Rethink name of patch embed grid info
2021-05-06 14:08:20 -07:00
b2c305c2aa
Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...
2021-05-06 14:03:23 -07:00
3ba6b55cb2
More adjustments to ByoaNet models for further experiments.
2021-05-06 13:59:44 -07:00
5fcddb96a8
Merge branch 'master' into cait
2021-05-05 17:29:38 -07:00
3db12b4b6a
Finish CaiT cleanup
2021-05-05 17:28:19 -07:00
2d8b09fe8b
Add official pretrained weights to MLP-Mixer, complete model cfgs.
2021-05-05 15:59:40 -07:00
12efffa6b1
Initial MLP-Mixer attempt...
2021-05-05 00:59:45 -07:00
0721559511
Improved (hopefully) init for SA/SA-like layers used in ByoaNets
2021-05-04 21:40:39 -07:00
d5473c17f7
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
0d87650fea
Remove filter hack from BlurPool w/ non-persistent buffer. Use BlurPool2d instead of AntiAliasing.. for TResNet. Breaks PyTorch < 1.6.
2021-05-04 16:56:28 -07:00
ddc743fdf8
Update ResNet-RS models to EMA weights
2021-05-04 16:16:55 -07:00
08d60f4a9a
resnetrs50 pool sizing wrong
2021-05-04 12:41:09 -07:00
1daa15ecc3
Initial Cait commit. Still some cleanup to do.
2021-05-04 11:19:27 -07:00
67d0665b46
Post ResNet-RS merge cleanup. Add weight urls, adjust train/test/crop pct.
2021-05-04 11:04:23 -07:00
560eae38f5
[WIP] Add ResNet-RS models ( #554 )
...
* Add ResNet-RS models
* Only include resnet-rs changes
* remove whitespace diff
* EOF newline
* Update time
* increase time
* Add first conv
* Try running only resnetv2_101x1_bitm on Linux runner
* Add to exclude filter
* Run test_model_forward_features for all
* Add to exclude ftrs
* back to defaults
* only run test_forward_features
* run all tests
* Run all tests
* Add bigger resnetrs to model filters to fix Github CLI
* Remove resnetv2_101x1_bitm from exclude feat features
* Remove hardcoded values
* Make sure reduction ratio in resnetrs is 0.25
* There is no bias in replaced maxpool so remove it
2021-05-04 10:59:44 -07:00
9cc7dda6e5
Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.
2021-04-29 21:08:37 -07:00
e15c3886ba
Defaul lambda r=7. Define '26t' stage 4/5 256x256 variants for all of bot/halo/lambda nets for experiment. Add resnet50t for exp. Fix a few comments.
2021-04-29 10:58:49 -07:00
e5e15754c9
Fix coat first conv ident
2021-04-28 18:09:23 -07:00
76739a7589
CoaT merge. Bit of formatting, fix torchscript (for non features), remove einops/einsum dep, add pretrained weight hub (url) support.
2021-04-28 16:31:35 -07:00
026430c083
Merge branch 'master' of https://github.com/morizin/pytorch-image-models-1 into morizin-master
2021-04-28 11:24:27 -07:00
a0492e3b48
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
2021-04-28 10:33:02 -07:00
8c1f03e56c
comment
2021-04-28 09:04:10 +03:00
19e1b67a84
old spaces
2021-04-28 09:01:29 +03:00
a443865876
update naming and scores
2021-04-28 08:57:06 +03:00
cf0e371594
84_0
2021-04-27 22:33:55 +03:00
0968bdeca3
vit, tresnet and mobilenetV3 ImageNet-21K-P weights
2021-04-27 20:54:06 +03:00
1e3b6d4dfc
Update __init__.py
2021-04-24 18:22:59 +05:30
fd022fd6a2
Update __init__.py
2021-04-24 18:22:36 +05:30
c2d5087eae
Add files via upload
2021-04-24 17:47:57 +05:30
79640fcc1f
Enable uniform augmentation magnitude sampling and set AugMix default
2021-04-19 14:21:12 -07:00
c1cf9712fc
Add updated EfficientNet-V2S weights, 83.8 @ 384x384 test. Add PyTorch trained EfficientNet-B4 weights, 83.4 @ 384x384 test. Tweak non TF EfficientNet B1-B4 train/test res scaling.
2021-04-19 10:42:56 -07:00
e8a64fb881
Test input size for efficientnet_v2s was wrong in last results run
2021-04-17 16:17:41 -07:00
2df77ee5cb
Fix torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers.
2021-04-15 10:20:26 -07:00
d793deb51a
Merge branch 'master' of https://github.com/iamhankai/pytorch-image-models into iamhankai-master
2021-04-15 09:30:25 -07:00
e685618f45
Merge pull request #550 from amaarora/wandb
...
Wandb Support
2021-04-15 09:26:35 -07:00
f606c45c38
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
2021-04-13 12:17:21 -07:00
de445e7827
Add GhostNet
2021-04-13 23:19:51 +08:00
5a196dddf6
Update README.md with latest, bump version to 0.4.8
2021-04-12 13:15:00 -07:00
b3d7580df1
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
2021-04-12 12:11:35 -07:00
16f7aa9f54
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
2021-04-12 11:54:22 -07:00
4e4b863b15
Missed norm.py
2021-04-12 09:57:56 -07:00
7c97e66f7c
Remove commented code, add more consistent seed fn
2021-04-12 09:51:36 -07:00
364dd6a58e
Merge branch 'master' into byoanet-self_attn
2021-04-12 09:38:59 -07:00
ce62f96d4d
ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments
2021-04-12 09:38:02 -07:00
cd3dc4979f
Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like
2021-04-12 08:25:31 -07:00
21812d33aa
Add prelim efficientnet_v2s weights from 224x224 train, eval 83.3 @ 288. Add eca_nfnet_l1 weights, train at 256, eval 84 @ 320.
2021-04-11 23:26:13 -07:00
5772c55c57
Make wandb optional
2021-04-10 01:34:20 -04:00
f54897cc0b
make wandb not required but rather optional as huggingface_hub
2021-04-10 01:27:23 -04:00
3f028ebc0f
import wandb in summary.py
2021-04-08 03:48:51 -04:00
624c9b6949
log to wandb only if using using wandb
2021-04-08 03:40:22 -04:00
addfc7c1ac
adabelief
2021-04-04 23:48:15 -04:00
fb896c0b26
Update some comments re preliminary EfficientNet-V2 assumptions
2021-04-03 12:00:25 -07:00
2b49ab7a36
Fix ResNetV2 pretrained classifier issue. Fixes #540
2021-04-03 11:18:12 -07:00
de9dff933a
EfficientNet-V2S preliminary model def (for experimentation)
2021-04-02 09:36:51 -07:00
37c71a5609
Some further create_optimizer_v2 tweaks, remove some redudnant code, add back safe model str. Benchmark step times per batch.
2021-04-01 22:34:55 -07:00
2bb65bd875
Wrong default_cfg pool_size for L1
2021-04-01 20:00:41 -07:00
bf2ca6bdf4
Merge jax and original weight init
2021-04-01 18:11:51 -07:00
acbd698c83
Update README.md with updates. Small tweak to head_dist handling.
2021-04-01 17:49:05 -07:00
9071568f0e
Add weights for SE NFNet-L0 model, rename nfnet_l0b -> nfnet_l0. 82.75 top-1 @ 288. Add nfnet_l1 model def for training.
2021-04-01 17:22:27 -07:00
c468c47a9c
Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size.
2021-04-01 16:41:04 -07:00
288682796f
Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7
2021-04-01 16:40:12 -07:00
ea9c9550b2
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
2021-04-01 14:17:38 -07:00
a5310a3451
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
2021-04-01 12:15:34 -07:00
7953e5d11a
Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433
2021-03-31 23:11:28 -07:00
a760a4c3f4
Some ViT cleanup, merge distilled model with main, fixup torchscript support for distilled models
2021-03-31 18:21:02 -07:00
0dfc5a66bb
Add PiT model from https://github.com/naver-ai/pit
2021-03-31 18:20:14 -07:00
51febd869b
Small tweak to tests for tnt model, reorder model imports.
2021-03-29 11:33:08 -07:00
b27a4e0d88
Merge branch 'master' of https://github.com/contrastive/pytorch-image-models into contrastive-master
2021-03-29 10:37:05 -07:00
6b18061773
Add GIST to docstring for quick access
2021-03-29 15:33:31 +11:00
de86314655
Update TNT
2021-03-29 08:23:34 +08:00
92b1db9a79
update docstrings and add check on and
2021-03-29 10:04:51 +11:00
b85be24054
update to work with fnmatch
2021-03-29 09:36:31 +11:00
cfc15283a4
Update TNT url
2021-03-28 23:19:15 +08:00
4a09bc851e
Add TNT model
2021-03-28 19:53:42 +08:00
20626e8387
Add to extract stats for SPP
2021-03-27 05:40:04 +11:00
cf5fec5047
Cleanup experimental vit weight init a bit
2021-03-20 09:44:24 -07:00
f42f1df26c
Improve evenness of per-worker split for validation set with TFDS
2021-03-18 23:16:14 -07:00
cbcb76d72c
Should have included Conv2d layers in original weight init. Lets see what the impact is...
2021-03-18 23:15:48 -07:00
4de57ccf01
Add weight init scheme that's closer to JAX impl
2021-03-18 15:35:22 -07:00
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
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
1f799afcb3
Remove capital letters in model names
2021-03-15 10:20:34 -07:00
82c1fe958a
added hardcorenas model
...
refactoring to support feature extraction
2021-03-15 08:06:34 +02: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
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
Ross Wightman
Aman Arora
talrid
morizin
Norman Mu
Ross Wightman
iamhankai
Aman Arora
juntang
contrastive
Mehrdad
AFLALO, Jonathan Isaac