Mehrdad 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

..

layers

Update README, fix iabn pip version print.

2021-03-07 16:17:06 -08:00

pruned

…

__init__.py

added hardcorenas model

2021-03-15 08:06:34 +02:00

byobnet.py

Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments.

2021-02-10 14:40:29 -08:00

cspnet.py

Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk()

2021-03-06 16:56:34 -08:00

densenet.py

…

dla.py

…

dpn.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

efficientnet_blocks.py

Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414

2021-02-07 22:10:18 -08:00

efficientnet_builder.py

…

efficientnet.py

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

factory.py

ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.

2020-12-28 16:59:15 -08:00

features.py

…

gluon_resnet.py

fix test_model_default_cfgs

2020-08-11 23:23:57 +09:00

gluon_xception.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

hardcorenas.py

Remove capital letters in model names

2021-03-15 10:20:34 -07:00

helpers.py

Initial AGC impl. Still testing.

2021-02-15 23:22:44 -08:00

hrnet.py

Bug in last mod for features_only default_cfg

2020-12-03 12:33:01 -08:00

inception_resnet_v2.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

inception_v3.py

Fix a few small things.

2021-02-09 17:33:43 -08:00

inception_v4.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

mobilenetv3.py

Bug in last mod for features_only default_cfg

2020-12-03 12:33:01 -08:00

nasnet.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

nfnet.py

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

pnasnet.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

registry.py

…

regnet.py

Add DropPath (stochastic depth) to ReXNet and VoVNet. RegNet DropPath impl tweak and dedupe se args.

2020-09-24 18:20:36 -07:00

res2net.py

…

resnest.py

fix test_model_default_cfgs

2020-08-11 23:23:57 +09:00

resnet.py

Fix some half removed resnet model defs, pooling for ecaresnet269d

2021-02-07 22:09:25 -08:00

resnetv2.py

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

rexnet.py

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

selecsls.py

fix test_model_default_cfgs

2020-08-11 23:23:57 +09:00

senet.py

AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992

2020-09-01 16:05:32 -07:00

sknet.py

fix test_model_default_cfgs

2020-08-11 23:23:57 +09:00

tresnet.py

Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'

2020-09-02 09:11:48 -07:00

vgg.py

Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models.

2021-02-09 16:22:52 -08:00

vision_transformer.py

Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.

2021-01-27 22:06:57 -08:00

vovnet.py

Add DropPath (stochastic depth) to ReXNet and VoVNet. RegNet DropPath impl tweak and dedupe se args.

2020-09-24 18:20:36 -07:00

xception_aligned.py

Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.

2021-02-09 16:22:52 -08:00

xception.py

…