From 8cc2805e7beb4a126e148e6b32dd0d52849818a2 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 22:01:19 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index 7de3c2bb..eda928e4 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -67,7 +67,7 @@ class ConvPosEnc(nn.Module): # reason: dim in control sequence -# FIXME reimplement in a way that allows tracing +# FIXME reimplement to allow tracing @register_notrace_module class PatchEmbed(nn.Module): """ Size-agnostic implementation of 2D image to patch embedding,