fix: resnet18

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)