pytorch-image-models/timm/models
Ross Wightman 151679c2f1 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
..
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
__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.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
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
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
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 Add backward and default_cfg tests and fix a few issues found. Fix #153 2020-05-21 15:30:58 -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 Update sotabench.py, tweak VovNet cfg 2020-06-11 14:34:25 -07:00
xception.py Merge branch 'master' into densenet_update_and_more 2020-05-22 16:18:10 -07:00