You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/models
Ross Wightman 17270c69b9
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.
5 years ago
..
layers 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. 5 years ago
pruned added efficientnet pruned weights 5 years ago
__init__.py Add ResNeSt models 5 years ago
densenet.py Big move, layer modules and fn to timm/models/layers 5 years ago
dla.py Big move, layer modules and fn to timm/models/layers 5 years ago
dpn.py Big move, layer modules and fn to timm/models/layers 5 years ago
efficientnet.py added efficientnet pruned weights 5 years ago
efficientnet_blocks.py Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well. 5 years ago
efficientnet_builder.py Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well. 5 years ago
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. 5 years ago
feature_hooks.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 5 years ago
gluon_resnet.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
gluon_xception.py Big move, layer modules and fn to timm/models/layers 5 years ago
helpers.py Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop 5 years ago
hrnet.py Replace all None by nn.Identity() in HRNet modules 5 years ago
inception_resnet_v2.py Big move, layer modules and fn to timm/models/layers 5 years ago
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. 5 years ago
inception_v4.py Big move, layer modules and fn to timm/models/layers 5 years ago
mobilenetv3.py Merge changes in feature extraction interface to MobileNetV3 5 years ago
nasnet.py Big move, layer modules and fn to timm/models/layers 5 years ago
pnasnet.py Fix model create fn not passing num_classes through. Fix #135 5 years ago
registry.py Exclude EfficientNet-L2 models from test 5 years ago
res2net.py Fix model create fn not passing num_classes through. Fix #135 5 years ago
resnest.py Missed one of the abalation model entrypoints, update README 5 years ago
resnet.py Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop 5 years ago
selecsls.py Big move, layer modules and fn to timm/models/layers 5 years ago
senet.py Big move, layer modules and fn to timm/models/layers 5 years ago
sknet.py Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 5 years ago
tresnet.py Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference 5 years ago
xception.py Big move, layer modules and fn to timm/models/layers 5 years ago