From 61a792801c44da904dce8c5bbb2b9575713475c3 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 07:08:17 -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 11f6ff11..b11342b8 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -23,7 +23,7 @@ import torch.nn as nn import torch.nn.functional as F from torch import Tensor from .helpers import build_model_with_cfg -from .data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD +from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, ClassifierHead, Mlp from .features import FeatureInfo from collections import OrderedDict