Ross Wightman 1904ed8fec Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups

2020-05-13 15:17:08 -07:00

..

layers

Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups

2020-05-13 15:17:08 -07:00

pruned

added efficientnet pruned weights

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

__init__.py

Add ResNeSt models

2020-05-11 23:27:09 -07:00

densenet.py

…

dla.py

…

dpn.py

…

efficientnet_blocks.py

…

efficientnet_builder.py

…

efficientnet.py

added efficientnet pruned weights

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

factory.py

…

feature_hooks.py

…

gluon_resnet.py

…

gluon_xception.py

…

helpers.py

Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop

2020-05-01 17:00:21 -07:00

hrnet.py

Replace all None by nn.Identity() in HRNet modules

2020-05-05 21:09:35 +03:00

inception_resnet_v2.py

…

inception_v3.py

Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.

2020-05-12 21:59:34 -07:00

inception_v4.py

…

mobilenetv3.py

…

nasnet.py

…

pnasnet.py

Fix model create fn not passing num_classes through. Fix #135

2020-05-01 18:37:03 -07:00

registry.py

Exclude EfficientNet-L2 models from test

2020-05-12 13:07:03 -07:00

res2net.py

Fix model create fn not passing num_classes through. Fix #135

2020-05-01 18:37:03 -07:00

resnest.py

Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups

2020-05-13 15:17:08 -07:00

resnet.py

Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop

2020-05-01 17:00:21 -07:00

selecsls.py

…

senet.py

…

sknet.py

Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.

2020-05-04 12:14:43 -07:00

tresnet.py

…

xception.py

…