307a935b79
Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA.
3 years ago
8bf63b6c6c
Able to use other attn layer in EfficientNet now. Create test ECA + GC B0 configs. Make ECA more configurable.
3 years ago
23c18a33e4
Add efficientnetv2_rw_m weights trained in PyTorch. 84.8 top-1 @ 416 test. 53M params.
3 years ago
c2ba229d99
Prep for effcientnetv2_rw_m model weights that started training before official release..
3 years ago
7077f16c6a
Change 21k model naming from _21k to _in21k for consistency with existing 21k models.
3 years ago
328249f11a
Update README, tweak fine-tune effv2 model names.
3 years ago
c4f482a08b
EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.
3 years ago
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.
3 years ago
e8a64fb881
Test input size for efficientnet_v2s was wrong in last results run
3 years ago
21812d33aa
Add prelim efficientnet_v2s weights from 224x224 train, eval 83.3 @ 288. Add eca_nfnet_l1 weights, train at 256, eval 84 @ 320.
3 years ago
fb896c0b26
Update some comments re preliminary EfficientNet-V2 assumptions
3 years ago
de9dff933a
EfficientNet-V2S preliminary model def (for experimentation)
3 years ago
5e2e4e7fb6
Merge pull request #502 from khatami-mehrdad/efficientnet_dg
...
adding efficientnet_el, efficientnet_es_pruned and efficientnet_el_pruned pre-trained models
3 years ago
97f49d163c
adding efficientnet_el, efficientnet_es_prunedefficientnet_es_pruned and efficientnet_el_pruned pretrained models to timm/models
3 years ago
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
3 years ago
cd72e66eff
Bug in last mod for features_only default_cfg
4 years ago
867a0e5a04
Add default_cfg back to models wrapped in feature extraction module as per discussion in #294 .
4 years ago
9c406532bd
Add EfficientNet-EdgeTPU-M (efficientnet_em) model trained natively in PyTorch. More sotabench fiddling.
4 years ago
512b2dd645
Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
4 years ago
b1f1a54de9
More uniform treatment of classifiers across all models, reduce code duplication.
4 years ago
ec4976fdba
Add EfficientNet-Lite0 weights trained with this code by @hal-314, 75.484 top-1
4 years ago
6c17d57a2c
Fix some attributions, add copyrights to some file docstrings
4 years ago
08016e839d
Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights
4 years ago
c146b54abc
Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work
4 years ago
6eec3fb4a4
Move FeatureHooks into features.py, switch EfficientNet, MobileNetV3 to use build model helper
4 years ago
3b9004bef9
Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
4 years ago
d23a2697d0
Working on feature extraction, interfaces refined, a number of models working, some in progress.
4 years ago
7be299504f
Add missing feature_info() on MobileNetV3, make hook feature output order/type consistent with bottleneck (list, decreasing fmap size)
4 years ago
eb7653614f
Monster commit, activation refactor, VoVNet, norm_act improvements, more
...
* refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto
* implement hard-mish, better grad for hard-swish
* add initial VovNet V1/V2 impl, fix #151
* VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN)
* Wrap IABN for any models that use it
* make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests
4 years ago
a7ebe09029
Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
...
Make small code refactoring
4 years ago
9c15d57505
added efficientnet pruned weights
4 years ago
1a8f5900ab
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
4 years ago
e34074b4da
Add final weights for MobileNet-V2 experiments
4 years ago
a6a5565de7
Fiddling...
4 years ago
bc998cad91
Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite
4 years ago
3406e582cf
Add EfficientNet-Lite results, update README
4 years ago
bd05258f7b
EfficientNet-Lite model added w/ converted checkpoints, validation in progress...
4 years ago
43225d110c
Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.
4 years ago
53c47479c4
Batch validation batch size adjustment, tweak L2 crop pct
4 years ago
08553e16b3
Merge branch 'master' into attention
4 years ago
ba15ca47e8
Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU
4 years ago
f902bcd54c
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
...
* select_conv2d -> create_conv2d
* added create_attn to create attention module from string/bool/module
* factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same
* add some more test eca resnet variants
* minor tweaks, naming, comments, consistency
4 years ago
a99ec4e7d1
A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
4 years ago
13746a33fc
Big move, layer modules and fn to timm/models/layers
4 years ago
5c4991a088
Add PyTorch trained EfficientNet-ES weights from Andrew Lavin
4 years ago
91534522f9
Add newly added TF ported EfficientNet-B8 weights (RandAugment)
4 years ago
ec0dd4053a
Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko
4 years ago
acc3ed2b8c
Add EfficientNet-B3 weights, trained from scratch with RA.
4 years ago
73b78459dc
Add update RandAugment MixNet-XL weights
5 years ago
ff421e5e09
New PyTorch trained EfficientNet-B2 weights with my RandAugment impl
5 years ago
Ross Wightman
Ross Wightman
Mehrdad
Vyacheslav Shults
AFLALO, Jonathan Isaac