Update davit.py

3 years ago · 8cc2805e7b
parent 5b26a64d48
commit 8cc2805e7b
1 changed files with 1 additions and 1 deletions
--- 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,