pytorch-image-models/timm/models
Ross Wightman 50658b9a67 Add RegNet models and weights 2020-05-18 00:08:52 -07:00
..
layers Add RegNet models and weights 2020-05-18 00:08:52 -07:00
pruned added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
__init__.py Add RegNet models and weights 2020-05-18 00:08:52 -07:00
densenet.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
dla.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
dpn.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
efficientnet.py added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
efficientnet_blocks.py Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well. 2020-04-09 01:41:54 -07:00
efficientnet_builder.py Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well. 2020-04-09 01:41:54 -07:00
factory.py Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args. 2020-02-18 14:00:26 -08:00
feature_hooks.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 2019-11-28 17:04:35 -08:00
gluon_resnet.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 2020-02-10 11:55:03 -08:00
gluon_xception.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
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 Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
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 Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
mobilenetv3.py Merge changes in feature extraction interface to MobileNetV3 2020-04-13 02:02:14 +03:00
nasnet.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
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
regnet.py Add RegNet models and weights 2020-05-18 00:08:52 -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 Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
senet.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
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 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference 2020-04-27 17:50:19 -07:00
xception.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00