Update config.json

config.json

@@ -1,7 +1,7 @@
 {
   "model_name": "ViT-B/32",
   "fine_tuned": true,
-  "tokenizer": "clip",
+  "tokenizer_class": "CLIPTokenizer",
   "pretrained_weights_path": "./pytorch_model.bin",
   "optimizer": {
     "type": "Adam",
@@ -13,5 +13,18 @@
   "loss_function": "CrossEntropyLoss",
   "num_epochs": 10,
   "batch_size": 16,
-  "device": "cpu"  
+  "device": "cpu",
+  "training_config": {
+    "optimizer": {
+      "type": "Adam",
+      "learning_rate": 1e-5,
+      "betas": [0.9, 0.98],
+      "eps": 1e-6,
+      "weight_decay": 0.2
+    },
+    "loss_function": "CrossEntropyLoss",
+    "num_epochs": 10,
+    "batch_size": 16,
+    "device": "cpu"
+  }
 }