Update cvt.py

Author: fffffgggg54

timm/models/cvt.py

@@ -372,11 +372,13 @@ def __init__(
             )
             blocks.append(block)
         self.blocks = nn.ModuleList(blocks)
+        self.probe = nn.Identity()
 
         if self.cls_token is not None:
             trunc_normal_(self.cls_token, std=.02)
 
     def forward(self, x: torch.Tensor) -> torch.Tensor:
+        self.probe(x)
         x = self.conv_embed(x)
         x = self.embed_drop(x)