From ff1445952ac676ee8b99b58c3dbc922d9a1652d5 Mon Sep 17 00:00:00 2001 From: Ajay Uppili Arasanipalai Date: Sun, 31 Jan 2021 12:23:53 +0530 Subject: [PATCH] fix: model name --- timm/models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 184aaf65..288e9018 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -1298,7 +1298,7 @@ def senet154(pretrained=False, **kwargs): @register_model def triplet_resnet18d(pretrained=False, **kwargs): model_args = dict(block=BasicBlock, layers=[2, 2, 2, 2], block_args=dict(attn_layer='triplet'), **kwargs) - return _create_resnet('triplet_resnet18', pretrained, **model_args) + return _create_resnet('triplet_resnet18d', pretrained, **model_args) @register_model @@ -1306,7 +1306,7 @@ def triplet_resnet50d(pretrained=False, **kwargs): model_args = dict( block=Bottleneck, layers=[3, 4, 6, 3], stem_width=32, stem_type='deep', avg_down=True, block_args=dict(attn_layer='triplet'), **kwargs) - return _create_resnet('triplet_resnet50', pretrained, **model_args) + return _create_resnet('triplet_resnet50d', pretrained, **model_args) @register_model @@ -1314,4 +1314,4 @@ def triplet_resnet101d(pretrained=False, **kwargs): model_args = dict( block=Bottleneck, layers=[3, 4, 23, 3], stem_width=32, stem_type='deep', avg_down=True, block_args=dict(attn_layer='triplet'), **kwargs) - return _create_resnet('triplet_resnet101', pretrained, **model_args) \ No newline at end of file + return _create_resnet('triplet_resnet101d', pretrained, **model_args) \ No newline at end of file