mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Refactor (back to relative imports)
This commit is contained in:
parent
74a04e0016
commit
29add820ac
@ -22,15 +22,10 @@ import torch.nn.functional as F
|
|||||||
import torch.utils.checkpoint as checkpoint
|
import torch.utils.checkpoint as checkpoint
|
||||||
|
|
||||||
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
|
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
|
||||||
# from .helpers import build_model_with_cfg, overlay_external_default_cfg
|
from .helpers import build_model_with_cfg, overlay_external_default_cfg
|
||||||
# from .vision_transformer import checkpoint_filter_fn
|
from .vision_transformer import checkpoint_filter_fn
|
||||||
# from .registry import register_model
|
from .registry import register_model
|
||||||
# from .layers import DropPath, Mlp
|
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
|
|
||||||
|
|
||||||
_logger = logging.getLogger(__name__)
|
_logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user