Ross Wightman d6ac5bbc48 EfficientNet and related cleanup ...

* remove folded_bn support and corresponding untrainable tflite ported weights
* combine bn args into dict
* add inplace support to activations and use where possible for reduced mem on large models

2019-07-22 09:29:58 -07:00

..

__init__.py

Add NASNet-Large model

2019-07-03 22:10:50 -07: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_same.py

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

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

densenet.py

Fix typo in Densenet default resolutions

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

dpn.py

Register dpn107

2019-06-30 09:57:06 -07:00

factory.py

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

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

gen_efficientnet.py

EfficientNet and related cleanup

2019-07-22 09:29:58 -07:00

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

helpers.py

A number of tweaks to arguments, epoch handling, config

2019-06-28 13:49:20 -07: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

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

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

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

resnet.py

A bit more ResNet cleanup.

2019-07-19 16:44:35 -07:00

senet.py

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

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