pytorch-image-models/timm/models
Ross Wightman 53c47479c4 Batch validation batch size adjustment, tweak L2 crop pct 2020-02-15 20:37:04 -08:00
..
layers Add CBAM for experimentation 2020-02-10 16:23:09 -08:00
__init__.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08: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 Batch validation batch size adjustment, tweak L2 crop pct 2020-02-15 20:37:04 -08:00
efficientnet_blocks.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 2020-02-10 11:55:03 -08:00
efficientnet_builder.py A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility. 2020-02-09 14:46:28 -08:00
factory.py Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2 2019-11-28 17:43:00 -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 Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts 2020-01-03 14:57:46 -08: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 Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 2020-02-10 11:55:03 -08:00
nasnet.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
pnasnet.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00
registry.py
res2net.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 2020-02-10 11:55:03 -08:00
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
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 Update SK network configs, add weights for skresnet8 and skresnext50 2020-02-14 15:37:00 -08:00
xception.py Big move, layer modules and fn to timm/models/layers 2020-02-09 13:13:08 -08:00