Ross Wightman 59ec7e6a53 Merge branch 'master' into imagenet21k_datasets_more

2021-01-04 12:11:05 -08:00

..

layers

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

pruned

…

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

cspnet.py

…

densenet.py

…

dla.py

…

dpn.py

Remove some old __future__ imports

2021-01-03 11:24:16 -08:00

efficientnet_blocks.py

…

efficientnet_builder.py

…

efficientnet.py

Bug in last mod for features_only default_cfg

2020-12-03 12:33:01 -08: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

…

gluon_xception.py

…

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

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

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

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 320x320 model default cfgs for 101D and 152D ResNets. Add SEResNet-152D weights and 320x320 cfg.

2021-01-03 12:10:25 -08:00

resnetv2.py

Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models

2020-12-28 16:59:15 -08: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

…

sknet.py

…

tresnet.py

…

vision_transformer.py

Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models

2020-12-28 16:59:15 -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

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

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

xception.py

…