Add num_classes suggestion by CodeRabbit. Good rabbit.

Signed-off-by: Cory Ye <cye@nvidia.com>

Author: cspades

recipes/vit/config/defaults.yaml

@@ -3,7 +3,7 @@ model:
     img_size: 224
     patch_size: 16
     in_chans: 3
-    num_classes: 100000
+    num_classes: ${dataset.num_classes}
     global_pool: "token"
     embed_dim: 768
     depth: 12
@@ -68,6 +68,7 @@ training:
     resume_from_metric: null
 
 dataset:
+  num_classes: 100000
   num_workers: 0
   train:
     root: null

recipes/vit/config/vit_base_patch16_224.yaml

@@ -7,7 +7,7 @@ model:
     img_size: 224
     patch_size: 16
     in_chans: 3
-    num_classes: 100000
+    num_classes: ${dataset.num_classes}
     global_pool: "map"
     embed_dim: 768
     depth: 12
@@ -66,6 +66,7 @@ training:
     resume_from_metric: "-"   # + = Highest Metric (Score), - = Lowest Metric (Loss)
 
 dataset:
+  num_classes: 100000
   num_workers: 4
   train:
     root: "./data/super-tiny-imagenet-5/train"