Update cvt.py

Author: fffffgggg54

timm/models/cvt.py

@@ -297,7 +297,7 @@ def forward(self, x: torch.Tensor, cls_token: Optional[torch.Tensor]) -> Tuple[t
             cls_token, x = torch.split(x, [1, H*W], 1)
 
         x = x.transpose(1, 2).reshape(B, C, H, W)
-        self.probe(x)
+        x = self.probe(x)
 
         return x, cls_token