From d17bcb10a56dc95ed7fbd5cb6bdd49a2ed4bd3b6 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 7 Jan 2023 21:14:21 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 92d7a0c5..a24e3377 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -21,7 +21,7 @@ from functools import partial import torch import torch.nn as nn from timm.layers import trunc_normal_, DropPath -from timm.models.registry import register_model +from ._registry import register_model from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from timm.layers.helpers import to_2tuple