From 0cc35a2ed182f9fa566c444f00698f4134096b67 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 10 Dec 2022 00:14:50 -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 0357370f..9caa8cd5 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -26,7 +26,7 @@ import torch.utils.checkpoint as checkpoint from .features import FeatureInfo from .fx_features import register_notrace_function, register_notrace_module -from .helpers import build_model_with_cfg, checkpoint_seq pretrained_cfg_for_features +from .helpers import build_model_with_cfg, checkpoint_seq, pretrained_cfg_for_features from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, ClassifierHead, Mlp from .pretrained import generate_default_cfgs from .registry import register_model