fix num_classes not found

timm/models/repvit.py

@@ -176,6 +176,7 @@ class RepViTClassifier(nn.Module):
         super().__init__()
         self.head = NormLinear(dim, num_classes) if num_classes > 0 else nn.Identity()
         self.distillation = distillation
+        self.num_classes=num_classes
         if distillation:
             self.head_dist = NormLinear(dim, num_classes) if num_classes > 0 else nn.Identity()