Ross Wightman 6ca0828166 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

..

__init__.py

Renamed gen_efficientnet.py -> efficientnet.py

2019-11-28 17:04:35 -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

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

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

densenet.py

Fix typo in Densenet default resolutions

2019-07-03 22:11:26 -07:00

dla.py

Add Res2Net and DLA models w/ pretrained weights. Update sotabench.

2019-09-04 17:06:42 -07:00

dpn.py

Register dpn107

2019-06-30 09:57:06 -07: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

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

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

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

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

ResNet updates:

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

gluon_xception.py

A few more additions to Gluon Xception models to match interface of others.

2019-08-11 23:06:23 -07:00

helpers.py

Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup

2019-11-28 17:00:52 -08:00

hrnet.py

Initial HRNet classification model commit

2019-11-28 17:00:52 -08:00

inception_resnet_v2.py

Add model registry and model listing fns, refactor model_factory/create_model fn

2019-06-23 18:22:16 -07: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

Add model registry and model listing fns, refactor model_factory/create_model fn

2019-06-23 18:22:16 -07:00

median_pool.py

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

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

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

nasnet.py

Add NASNet-Large model

2019-07-03 22:10:50 -07:00

pnasnet.py

Add model registry and model listing fns, refactor model_factory/create_model fn

2019-06-23 18:22:16 -07: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

Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.

2019-09-05 15:47:56 -07:00

resnet.py

ResNet updates:

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

senet.py

ResNet updates:

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

test_time_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

xception.py

Add model registry and model listing fns, refactor model_factory/create_model fn

2019-06-23 18:22:16 -07:00