Ross Wightman d72ac0db25 Fix #173, lr cycle default 0 vs 1. Fix #177, mirror resnest weights for future stability.

2020-06-29 21:31:48 -07:00

..

layers

Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory

2020-06-11 14:49:23 -07:00

pruned

added efficientnet pruned weights

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

__init__.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

densenet.py

Remove pointless densenet configs, add an iabn version of 264 as it makes more sense to try someday...

2020-06-03 17:13:52 -07:00

dla.py

Merge branch 'master' into densenet_update_and_more

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

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

efficientnet_blocks.py

Fix an an untested change, remove a debug print

2020-06-01 17:26:42 -07:00

efficientnet_builder.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

efficientnet.py

Add missing feature_info() on MobileNetV3, make hook feature output order/type consistent with bottleneck (list, decreasing fmap size)

2020-06-03 00:00:37 -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

feature_hooks.py

Add missing feature_info() on MobileNetV3, make hook feature output order/type consistent with bottleneck (list, decreasing fmap size)

2020-06-03 00:00:37 -07:00

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

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

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

Add missing feature_info() on MobileNetV3, make hook feature output order/type consistent with bottleneck (list, decreasing fmap size)

2020-06-03 00:00:37 -07:00

nasnet.py

Merge branch 'master' into densenet_update_and_more

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

pnasnet.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

registry.py

Exclude EfficientNet-L2 models from test

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

regnet.py

Update regnet cfg keys to match model names so registry works properly

2020-06-09 14:37:45 -07:00

res2net.py

Fix two regressions

2020-06-05 11:04:51 -07:00

resnest.py

Fix #173, lr cycle default 0 vs 1. Fix #177, mirror resnest weights for future stability.

2020-06-29 21:31:48 -07:00

resnet.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

selecsls.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -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

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

vovnet.py

Add ESE-VoVNet-19-DW weights

2020-06-15 14:31:55 -07:00

xception.py

Merge branch 'master' into densenet_update_and_more

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