Upload Model configuration

config.json

@@ -2,18 +2,25 @@
   "model_type": "vit-age-gender",
   "architecture": "ViTAgeGenderModel",
   "base_model": "google/vit-base-patch16-224",
-  "num_age_classes": 1,
-  "num_gender_classes": 1,
   "hidden_size": 768,
-  "image_size": 224,
-  "patch_size": 16,
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
+  "image_size": 224,
+  "patch_size": 16,
+  "num_age_outputs": 1,
+  "num_gender_outputs": 1,
   "performance_metrics": {
     "gender_accuracy": 94.3,
     "age_mae": 4.5,
     "training_epochs": 15,
     "dataset": "UTKFace",
     "total_parameters": 86816002
+  },
+  "training_info": {
+    "optimizer": "AdamW",
+    "learning_rate": 2e-05,
+    "batch_size": 32,
+    "training_time_hours": 2.95,
+    "best_epoch": 12
   }
 }