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,