Ross Wightman cefc9b7761 Move SelectKernelConv to conv2d_layers and more ...

* always apply attention in SelectKernelConv, leave MixedConv for no attention alternative
* make MixedConv torchscript compatible
* refactor first/previous dilation name to make more sense in ResNet* networks

2020-02-01 22:11:00 -08:00

..

__init__.py

Move Selective Kernel blocks/convs to their own sknet.py file

2020-02-01 22:11:00 -08:00

activations.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

adaptive_avgmax_pool.py

Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py

2019-06-20 17:29:25 -07:00

conv2d_layers.py

Move SelectKernelConv to conv2d_layers and more

2020-02-01 22:11:00 -08:00

densenet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

dla.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

dpn.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

efficientnet_blocks.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

efficientnet_builder.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

efficientnet.py

Add newly added TF ported EfficientNet-B8 weights (RandAugment)

2020-02-01 18:01:14 -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

Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d

2019-12-28 11:43:50 -08:00

gluon_xception.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -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

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

inception_resnet_v2.py

InceptionResNetV2 torchscript compatible

2020-01-01 17:13:37 -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

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

median_pool.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

mobilenetv3.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

nasnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

pnasnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -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

Move SelectKernelConv to conv2d_layers and more

2020-02-01 22:11:00 -08:00

resnet.py

Move SelectKernelConv to conv2d_layers and more

2020-02-01 22:11:00 -08:00

selecsls.py

Add selecsls60 weights

2019-12-31 16:49:04 -08:00

senet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

sknet.py

Move SelectKernelConv to conv2d_layers and more

2020-02-01 22:11:00 -08:00

split_batchnorm.py

Update AugMix, JSD, etc comments and references

2020-01-11 12:02:05 -08:00

test_time_pool.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

xception.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00