From 225d45afac61657fc7ceb9c59f506edf25e46b9e Mon Sep 17 00:00:00 2001 From: iyaja Date: Sun, 31 Jan 2021 13:56:13 +0530 Subject: [PATCH] fix: resnet18 --- timm/models/resnet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 288e9018..c4938664 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -1297,7 +1297,9 @@ 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) + model_args = dict( + block=BasicBlock, layers=[2, 2, 2, 2], stem_width=32, stem_type='deep', avg_down=True, + block_args=dict(attn_layer='triplet'), **kwargs) return _create_resnet('triplet_resnet18d', pretrained, **model_args)