Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

..

layers

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

pruned

added efficientnet pruned weights

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

__init__.py

Add RegNet models and weights

2020-05-18 00:08:52 -07:00

densenet.py

Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.

2020-05-22 16:16:45 -07:00

dla.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

dpn.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

efficientnet_blocks.py

Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.

2020-04-09 01:41:54 -07:00

efficientnet_builder.py

Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.

2020-04-09 01:41:54 -07:00

efficientnet.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

factory.py

…

feature_hooks.py

…

gluon_resnet.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

gluon_xception.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

helpers.py

Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop

2020-05-01 17:00:21 -07:00

hrnet.py

Add backward and default_cfg tests and fix a few issues found. Fix #153

2020-05-21 15:30:58 -07:00

inception_resnet_v2.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

inception_v3.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

inception_v4.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

mobilenetv3.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

nasnet.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

pnasnet.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

registry.py

Exclude EfficientNet-L2 models from test

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

regnet.py

Add RegNet models and weights

2020-05-18 00:08:52 -07:00

res2net.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

resnest.py

Add backward and default_cfg tests and fix a few issues found. Fix #153

2020-05-21 15:30:58 -07:00

resnet.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

selecsls.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

senet.py

Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.

2020-05-06 09:54:03 +03:00

sknet.py

Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.

2020-05-04 12:14:43 -07:00

tresnet.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00

xception.py

Merge branch 'master' into densenet_update_and_more

2020-05-22 16:18:10 -07:00