From ec8b69cca6ebc757524cd1c0b0f3b64336f8a074 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Mon, 12 Dec 2022 23:24:51 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index e551cc61..05adc750 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -15,22 +15,21 @@ DaViT model defs and weights adapted from https://github.com/dingmyu/davit, orig
 # FIXME remove unused imports
 
 import itertools
-from typing import Any, Dict, Iterable, Iterator, List, Mapping, Optional, overload, Tuple, TypeVar, Union
-from collections import OrderedDict
 
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from torch import Tensor
-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, 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
+
 from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
+from timm.layers import DropPath, to_2tuple, trunc_normal_, ClassifierHead, Mlp
+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
+
 
 __all__ = ['DaViT']