Ross Wightman 22748f1a2d Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py

2021-01-22 16:54:33 -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

added efficientnet pruned weights

2020-05-03 09:29:45 +03:00

__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

Add CSPResNet50 weights, 79.6 top-1 at 256x256

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

densenet.py

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

dla.py

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

dpn.py

Remove some old __future__ imports

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

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

Merge branch 'logger' into features. Change 'logger' to '_logger'.

2020-07-27 18:00:46 -07:00

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

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

gluon_resnet.py

fix test_model_default_cfgs

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

gluon_xception.py

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

helpers.py

More dataset work including factories and a tensorflow datasets (TFDS) wrapper

2021-01-15 17:26:20 -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

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

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

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

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

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00

registry.py

Fix some attributions, add copyrights to some file docstrings

2020-07-27 13:44:56 -07:00

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

Fix some checkpoint / model str regressions

2020-07-29 19:43:01 -07:00

resnest.py

fix test_model_default_cfgs

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

resnet.py

Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py

2021-01-22 16:54:33 -08:00

resnetv2.py

Finishing adding stochastic depth support to BiT ResNetV2 models

2021-01-16 16:32:03 -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

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

vision_transformer.py

Refactor vision_transformer entrpy fns, add pos embedding resize support for fine tuning, add some deit models for testing

2021-01-18 16:11:02 -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

More uniform treatment of classifiers across all models, reduce code duplication.

2020-08-03 22:18:24 -07:00