Ross Wightman 2ac663f340 Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing.

2020-07-21 11:15:30 -07:00

..

layers

Back out some activation hacks trialing upcoming pytorch changes

2020-07-17 18:41:37 -07:00

pruned

added efficientnet pruned weights

2020-05-03 09:29:45 +03:00

__init__.py

Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs.

2020-07-13 15:01:06 -07:00

csp.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

densenet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

dla.py

More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.

2020-07-19 14:02:02 -07:00

dpn.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

efficientnet_blocks.py

Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work

2020-07-21 01:21:38 -07:00

efficientnet_builder.py

Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work

2020-07-21 01:21:38 -07:00

efficientnet.py

Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work

2020-07-21 01:21:38 -07:00

factory.py

Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.

2020-06-02 21:06:10 -07:00

features.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

gluon_resnet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

gluon_xception.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

helpers.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

hrnet.py

More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.

2020-07-19 14:02:02 -07:00

inception_resnet_v2.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

inception_v3.py

More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.

2020-07-19 14:02:02 -07:00

inception_v4.py

More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.

2020-07-19 14:02:02 -07:00

mobilenetv3.py

Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work

2020-07-21 01:21:38 -07:00

nasnet.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

pnasnet.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

registry.py

Exclude EfficientNet-L2 models from test

2020-05-12 13:07:03 -07:00

regnet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

res2net.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

resnest.py

Merge branch 'master' into features

2020-07-20 16:11:38 -07:00

resnet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

selecsls.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

senet.py

Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing.

2020-07-21 11:15:30 -07:00

sknet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

tresnet.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00

vovnet.py

Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces

2020-07-17 17:54:26 -07:00

xception_aligned.py

Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing.

2020-07-21 11:15:30 -07:00

xception.py

Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript

2020-07-20 16:10:31 -07:00