Ross Wightman 2a7d256fd5 Re-enable mem-efficient/jit activations after torchscript tests

2020-02-10 11:59:36 -08:00

..

layers

Re-enable mem-efficient/jit activations after torchscript tests

2020-02-10 11:59:36 -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

…

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

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