From a050fde5cde892404a5b77973a5916cdd7b602ab Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 1 Jul 2022 15:03:28 -0700 Subject: [PATCH] Add resnet10t (basic block) and resnet14t (bottleneck) with 1,1,1,1 repeats --- timm/models/resnet.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index a7f0c0f6..476ffe91 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -723,6 +723,24 @@ def _create_resnet(variant, pretrained=False, **kwargs): return build_model_with_cfg(ResNet, variant, pretrained, **kwargs) +@register_model +def resnet10t(pretrained=False, **kwargs): + """Constructs a ResNet-10-T model. + """ + model_args = dict( + block=BasicBlock, layers=[1, 1, 1, 1], stem_width=32, stem_type='deep_tiered', avg_down=True, **kwargs) + return _create_resnet('resnet10t', pretrained, **model_args) + + +@register_model +def resnet14t(pretrained=False, **kwargs): + """Constructs a ResNet-14-T model. + """ + model_args = dict( + block=Bottleneck, layers=[1, 1, 1, 1], stem_width=32, stem_type='deep_tiered', avg_down=True, **kwargs) + return _create_resnet('resnet14t', pretrained, **model_args) + + @register_model def resnet18(pretrained=False, **kwargs): """Constructs a ResNet-18 model.