pytorch-image-models/timm/models
Ross Wightman 8d8677e03b Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 2020-05-04 12:14:43 -07:00
..
layers Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 2020-05-04 12:14:43 -07:00
pruned added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
__init__.py TResNet models 2020-04-12 18:44:12 +03: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 Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08: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 Update version, results csv files, and move remaining dropbox weights to github 2019-07-25 16:54:44 -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 Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. 2019-06-29 15:50:33 -07:00
res2net.py Fix model create fn not passing num_classes through. Fix #135 2020-05-01 18:37:03 -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