Ross Wightman bc264269c9 Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related ...

* add an alternate RMSprop opt that applies eps like TF
* add bn params for passing through alternates and changing defaults to TF style

2019-04-21 15:54:28 -07:00

..

__init__.py

Uniform pretrained model handling.

2019-04-11 21:32:16 -07:00

adaptive_avgmax_pool.py

Fix pooling in mnasnet, more sensible default for AMP opt level

2019-04-17 18:06:37 -07:00

densenet.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

dpn.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

genmobilenet.py

Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related

2019-04-21 15:54:28 -07:00

helpers.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

inception_resnet_v2.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

inception_v4.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

median_pool.py

Initial commit, puting some ol pieces together

2019-02-01 22:07:34 -08:00

model_factory.py

Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related

2019-04-21 15:54:28 -07:00

pnasnet.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

resnet.py

Add checkpoint clean script, add link to pretrained resnext50 weights

2019-04-13 14:15:35 -07:00

senet.py

Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn

2019-04-14 13:43:46 -07:00

test_time_pool.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00

xception.py

Add per model crop pct, interpolation defaults, tie it all together

2019-04-12 22:55:24 -07:00