diff --git a/timm/models/davit.py b/timm/models/davit.py index af70c4d4..f7607f2c 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -560,14 +560,14 @@ class DaViT(nn.Module): def _cfg(url='', **kwargs): # not sure how this should be set up - return { - 'url': url, - 'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7), - 'crop_pct': 0.875, 'interpolation': 'bilinear', - 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, - 'first_conv': 'patch_embeds.0.proj', 'classifier': 'head.fc', - **kwargs - } + return { + 'url': url, + 'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7), + 'crop_pct': 0.875, 'interpolation': 'bilinear', + 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, + 'first_conv': 'patch_embeds.0.proj', 'classifier': 'head.fc', + **kwargs + }