Ross Wightman 33f8a1bf36 Updated README, add wide_resnet50_2 and seresnext50_32x4d weights

2020-09-03 10:45:17 -07:00

..

layers

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

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

pruned

…

__init__.py

…

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

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

2020-08-03 22:18:24 -07: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

Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1

2020-08-11 14:18:51 -07: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

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

Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.

2020-09-02 15:12:59 -07:00

hrnet.py

hrnet features_only pretrained weight loading issue. Fix #232.

2020-09-01 17:37:55 -07: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

fix test_model_default_cfgs

2020-08-11 23:23:57 +09: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

Fix MobileNetV3 crash with global_pool='', output consistent with other models but not equivalent due to efficient head.

2020-08-18 14:11:30 -07:00

nasnet.py

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

2020-08-03 22:18:24 -07: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 new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1

2020-08-11 14:18:51 -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

Updated README, add wide_resnet50_2 and seresnext50_32x4d weights

2020-09-03 10:45:17 -07:00

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

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

vovnet.py

Fix some attributions, add copyrights to some file docstrings

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

xception_aligned.py

fix test_model_default_cfgs

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

xception.py

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

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