Training in progress, epoch 0

config.json

@@ -9,57 +9,15 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "brick",
-    "1": "carpet",
-    "10": "mirror",
-    "11": "other",
-    "12": "painted",
-    "13": "paper",
-    "14": "plastic",
-    "15": "polishedstone",
-    "16": "skin",
-    "17": "sky",
-    "18": "stone",
-    "19": "tile",
-    "2": "ceramic",
-    "20": "wallpaper",
-    "21": "water",
-    "22": "wood",
-    "3": "fabric",
-    "4": "foliage",
-    "5": "food",
-    "6": "glass",
-    "7": "hair",
-    "8": "leather",
-    "9": "metal"
+    "0": "metal",
+    "1": "wood"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "brick": "0",
-    "carpet": "1",
-    "ceramic": "2",
-    "fabric": "3",
-    "foliage": "4",
-    "food": "5",
-    "glass": "6",
-    "hair": "7",
-    "leather": "8",
-    "metal": "9",
-    "mirror": "10",
-    "other": "11",
-    "painted": "12",
-    "paper": "13",
-    "plastic": "14",
-    "polishedstone": "15",
-    "skin": "16",
-    "sky": "17",
-    "stone": "18",
-    "tile": "19",
-    "wallpaper": "20",
-    "water": "21",
-    "wood": "22"
+    "metal": "0",
+    "wood": "1"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cafee6604ac4a2e12c637047cbb9c9f6f82aed0779407ba4e947c44cc6e72080
-size 343333738
+oid sha256:f48198a6d9147dbfcec27c99be040beac65faf7b5c56916262038dfa23e2587f
+size 343269162

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:93adce7e2154704e155026f0f4390b534bde1fff0bd21cd7464d97a73f9240bb
+oid sha256:1e11533c84789a80eed05ee7f33ceebc3f0a06e97434cfa72b4bda98f4659699
 size 4536