Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases ...

* select_conv2d -> create_conv2d
* added create_attn to create attention module from string/bool/module
* factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same
* add some more test eca resnet variants
* minor tweaks, naming, comments, consistency

2020-02-10 11:55:03 -08:00

..

layers

Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases

2020-02-10 11:55:03 -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_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

efficientnet.py

Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases

2020-02-10 11:55:03 -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

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

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

Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases

2020-02-10 11:55:03 -08:00

xception.py

Big move, layer modules and fn to timm/models/layers

2020-02-09 13:13:08 -08:00