Training in progress, epoch 0

.gitignore

@@ -0,0 +1 @@
+checkpoint-*/

config.json

@@ -9,7 +9,6 @@
     23,
     3
   ],
-  "downsample_in_bottleneck": false,
   "downsample_in_first_stage": false,
   "embedding_size": 64,
   "hidden_act": "relu",
@@ -20,36 +19,14 @@
     2048
   ],
   "id2label": {
-    "0": "blue_stain",
-    "1": "crack",
-    "2": "crazing",
-    "3": "dead_knot",
-    "4": "inclusion",
-    "5": "knot_with_crack",
-    "6": "live_knot",
-    "7": "marrow",
-    "8": "patches",
-    "9": "pitted_surface",
-    "10": "resin",
-    "11": "rolled_in_scale",
-    "12": "scratches",
-    "13": "steel_defect"
+    "0": "crazing",
+    "1": "live_knot",
+    "2": "steel_defect"
   },
   "label2id": {
-    "blue_stain": 0,
-    "crack": 1,
-    "crazing": 2,
-    "dead_knot": 3,
-    "inclusion": 4,
-    "knot_with_crack": 5,
-    "live_knot": 6,
-    "marrow": 7,
-    "patches": 8,
-    "pitted_surface": 9,
-    "resin": 10,
-    "rolled_in_scale": 11,
-    "scratches": 12,
-    "steel_defect": 13
+    "crazing": 0,
+    "live_knot": 1,
+    "steel_defect": 2
   },
   "layer_type": "bottleneck",
   "model_type": "resnet",
@@ -69,5 +46,5 @@
     "stage4"
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.38.2"
+  "transformers_version": "4.30.2"
 }

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82d8bd5710597e739dde14c7c95a7af126cc067e5625a6873f5fa453c89ad01a
+size 170661845

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6e2ed147a0e113731dcee064355196221e8730187e96e68d3c4079fcc5735e1f
-size 4920
+oid sha256:8c9033ef5da9d60873df439a2f280fd2d79d4d079325a358a7a0495951b46193
+size 3963