Update config.json

config.json

@@ -1,17 +1,30 @@
 {
-        "num_classes": 1,
-        "image_size": 640,
-        "architectures": [
-    "YolosForObjectDetection"
-  ],
+    "architectures": [
+        "YolosForObjectDetection"
+    ],
+    "attention_probs_dropout_prob": 0.1,
+    "hidden_dropout_prob": 0.1,
+    "hidden_size": 768,
+    "image_size": 640,
+    "initializer_range": 0.02,
+    "layer_norm_eps": 1e-12,
     "model_type": "yolos",
+    "num_attention_heads": 12,
+    "num_channels": 3,
+    "num_hidden_layers": 12,
+    "patch_size": 16,
+    "qkv_bias": true,
+    "encoder_stride": 16,
+    "num_classes": 2,
     "id2label": {
-        "1": "roof",
-        "0": "N/A"
-        },
+        "0": "N/A",
+        "1": "roof"
+    },
     "label2id": {
-        "roof": 1,
-        "N/A": 0
-    }
-
+        "N/A": 0,
+        "roof": 1
+    },
+    "use_mid_position_embeddings": true,
+    "use_relative_position_bias": true,
+    "intermediate_size": 3072
 }