Create config.json

config.json

@@ -0,0 +1,33 @@
+{
+  "architectures": ["MiniVisionCNN"],
+  "model_type": "mini-vision",
+  "num_labels": 10,
+  "input_size": [3, 32, 32],
+  "hidden_size": 256,
+  "num_channels": 3,
+  "image_size": 32,
+  "id2label": {
+    "0": "airplane",
+    "1": "automobile",
+    "2": "bird",
+    "3": "cat",
+    "4": "deer",
+    "5": "dog",
+    "6": "frog",
+    "7": "horse",
+    "8": "ship",
+    "9": "truck"
+  },
+  "label2id": {
+    "airplane": 0,
+    "automobile": 1,
+    "bird": 2,
+    "cat": 3,
+    "deer": 4,
+    "dog": 5,
+    "frog": 6,
+    "horse": 7,
+    "ship": 8,
+    "truck": 9
+  }
+}