From d8d6109818c9b2c3ed5b34e3dd3bbce11605ff58 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Mon, 12 Dec 2022 23:30:53 -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 43ebc9a3..917c71c8 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -21,9 +21,9 @@ from torch import Tensor from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from timm.layers import DropPath, to_2tuple, trunc_normal_, ClassifierHead, Mlp +from ._builder import build_model_with_cfg from ._features import FeatureInfo from ._features_fx import register_notrace_function -from ._helpers import build_model_with_cfg from ._manipulate import checkpoint_seq from ._pretrained import generate_default_cfgs from ._registry import register_model