Ross Wightman 0004f37d25 Some TResNet cleanup. ...

* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences

2020-04-27 13:21:05 -07:00

..

layers

Some TResNet cleanup.

2020-04-27 13:21:05 -07: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_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

efficientnet.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

…

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

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

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

Fix minor typos in create_attn.py and resnet.py

2020-04-07 21:15:57 +09: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 comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights

2020-02-18 13:58:30 -08:00

tresnet.py

Some TResNet cleanup.

2020-04-27 13:21:05 -07:00

xception.py

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

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