Ross Wightman cd72e66eff Bug in last mod for features_only default_cfg

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

..

layers

A few more changes for 0.3.2 maint release. Linear layer change for mobilenetv3 and inception_v3, support no bias for linear wrapper.

2020-11-30 16:19:52 -08:00

pruned

…

__init__.py

Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.

2020-10-13 13:33:44 -07:00

cspnet.py

Add CSPResNet50 weights, 79.6 top-1 at 256x256

2020-08-12 11:20:11 -07:00

densenet.py

…

dla.py

…

dpn.py

…

efficientnet_blocks.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

efficientnet_builder.py

…

efficientnet.py

Bug in last mod for features_only default_cfg

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

factory.py

Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool

2020-09-02 16:13:47 -07:00

features.py

…

gluon_resnet.py

…

gluon_xception.py

…

helpers.py

Add default_cfg back to models wrapped in feature extraction module as per discussion in #294.

2020-12-03 10:24:35 -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

…

inception_v3.py

A few more changes for 0.3.2 maint release. Linear layer change for mobilenetv3 and inception_v3, support no bias for linear wrapper.

2020-11-30 16:19:52 -08:00

inception_v4.py

…

mobilenetv3.py

Bug in last mod for features_only default_cfg

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

nasnet.py

…

pnasnet.py

…

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

…

resnet.py

Add ResNet weights. 80.5 (top-1) ResNet-50-D, 77.1 ResNet-34-D, 72.7 ResNet-18-D.

2020-09-18 12:05:37 -07:00

rexnet.py

Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.

2020-10-13 13:33:44 -07:00

selecsls.py

…

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

…

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

vision_transformer.py

Add newly added vision transformer large/base 224x224 weights ported from JAX official repo

2020-10-29 17:31:01 -07: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

Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.

2020-10-13 13:33:44 -07:00

xception.py

…