diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 8098958c..21f46b5f 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -17,6 +17,7 @@ MetaFormer baselines including IdentityFormer, RandFormer, PoolFormerV2, ConvFormer and CAFormer. Some implementations are modified from timm (https://github.com/rwightman/pytorch-image-models). """ +from collections import OrderedDict from functools import partial import torch import torch.nn as nn