Ross Wightman 6853b07bbd Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments.		2021-02-10 14:40:29 -08:00
..
layers	Use in_channels for depthwise groups, allows using `out_channels=N * in_channels` (does not impact existing models). Fix #354.	2021-02-09 16:22:52 -08:00
pruned	…
__init__.py	Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models.	2021-02-09 16:22:52 -08: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	Add CSPResNet50 weights, 79.6 top-1 at 256x256	2020-08-12 11:20:11 -07: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.py	Bug in last mod for features_only default_cfg	2020-12-03 12:33:01 -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	…
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	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
helpers.py	Default to native PyTorch AMP instead of APEX amp. Too many APEX issues cropping up lately.	2021-02-10 14:31:18 -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	Add ByobNet w/ GPU-EfficientNets and RepVGG. Also add classic vgg models.	2021-02-09 16:22:52 -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	…
resnet.py	Fix some half removed resnet model defs, pooling for ecaresnet269d	2021-02-07 22:09:25 -08:00
resnetv2.py	Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.	2021-01-27 22:06:57 -08:00
rexnet.py	Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.	2021-01-27 22:06:57 -08: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
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.py	…
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