V1 - Tayzzyronth. Finetuning with 53 images. Text, Images and Latex Classes

config.json

@@ -76,19 +76,19 @@
   "freeze_backbone_batch_norms": true,
   "hidden_expansion": 0.5,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3"
+    "0": "word",
+    "1": "text",
+    "2": "latex",
+    "3": "image"
   },
   "initializer_bias_prior_prob": null,
   "initializer_range": 0.01,
   "is_encoder_decoder": true,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3
+    "image": 3,
+    "latex": 2,
+    "text": 1,
+    "word": 0
   },
   "label_noise_ratio": 0.5,
   "layer_norm_eps": 1e-05,