diff --git a/timm/models/_hub.py b/timm/models/_hub.py index 61594753..e247382f 100644 --- a/timm/models/_hub.py +++ b/timm/models/_hub.py @@ -16,7 +16,6 @@ except ImportError: from torch.hub import _get_torch_home as get_dir from timm import __version__ -from timm.layers import ClassifierHead, NormMlpClassifierHead from timm.models._pretrained import filter_pretrained_cfg try: