From 7aa3459caf3dd393b28841c8443b3dfdd451f3bb Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sun, 8 Jan 2023 00:14:23 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 1 + 1 file changed, 1 insertion(+) 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