Skip to content

Commit e5e1575

Browse files
committed
Fix coat first conv ident
1 parent 76739a7 commit e5e1575

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/coat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def _cfg_coat(url='', **kwargs):
3636
'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None,
3737
'crop_pct': .9, 'interpolation': 'bicubic',
3838
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
39-
'first_conv': 'patch_embed.proj', 'classifier': 'head',
39+
'first_conv': 'patch_embed1.proj', 'classifier': 'head',
4040
**kwargs
4141
}
4242

@@ -654,7 +654,7 @@ def coat_lite_tiny(pretrained=False, **kwargs):
654654
patch_size=4, embed_dims=[64, 128, 256, 320], serial_depths=[2, 2, 2, 2], parallel_depth=0,
655655
num_heads=8, mlp_ratios=[8, 8, 4, 4], **kwargs)
656656
# FIXME use builder
657-
model.default_cfg = default_cfgs['coat_lite_mini']
657+
model.default_cfg = default_cfgs['coat_lite_tiny']
658658
if pretrained:
659659
load_pretrained(model, num_classes=model.num_classes, in_chans=kwargs.get('in_chans', 3))
660660
return model

0 commit comments

Comments
 (0)