From 29add820ac0a7f0a11fc3b2648ab697a891f4e00 Mon Sep 17 00:00:00 2001 From: Christoph Reich <34400551+ChristophReich1996@users.noreply.github.com> Date: Sun, 20 Feb 2022 00:46:48 +0100 Subject: [PATCH] Refactor (back to relative imports) --- timm/models/swin_transformer_v2_cr.py | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/timm/models/swin_transformer_v2_cr.py b/timm/models/swin_transformer_v2_cr.py index 6df1ff9b..aeb713a5 100644 --- a/timm/models/swin_transformer_v2_cr.py +++ b/timm/models/swin_transformer_v2_cr.py @@ -22,15 +22,10 @@ import torch.nn.functional as F import torch.utils.checkpoint as checkpoint from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD -# from .helpers import build_model_with_cfg, overlay_external_default_cfg -# from .vision_transformer import checkpoint_filter_fn -# from .registry import register_model -# from .layers import DropPath, Mlp - -from timm.models.helpers import build_model_with_cfg, overlay_external_default_cfg -from timm.models.vision_transformer import checkpoint_filter_fn -from timm.models.registry import register_model -from timm.models.layers import DropPath, Mlp +from .helpers import build_model_with_cfg, overlay_external_default_cfg +from .vision_transformer import checkpoint_filter_fn +from .registry import register_model +from .layers import DropPath, Mlp _logger = logging.getLogger(__name__)