From 2de09f6bb7abd2a66d85bd3b596d7ab7f0b50dbf Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 07:52:00 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index cee1b404..8d6e66e6 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -28,7 +28,7 @@ from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, Cl from .features import FeatureInfo from collections import OrderedDict import torch.utils.checkpoint as checkpoint -from .pretrained import generate_default_cfgs +from .pretrained import generate_default_cfgs, pretrained_cfg_for_features from .registry import register_model