Training in progress, epoch 0

all_results.json

@@ -1,8 +1,11 @@
 {
     "epoch": 0.0,
-    "eval_f1": 0.08855052368356824,
-    "eval_loss": 2.330942153930664,
-    "eval_runtime": 11.8647,
-    "eval_samples_per_second": 21.577,
-    "eval_steps_per_second": 5.394
+    "eval_accuracy": 0.15708812260536398,
+    "eval_f1": 0.10951914110924829,
+    "eval_loss": 2.2501182556152344,
+    "eval_precision": 0.11397786789695766,
+    "eval_recall": 0.14094382256290108,
+    "eval_runtime": 37.0603,
+    "eval_samples_per_second": 7.043,
+    "eval_steps_per_second": 1.781
 }

config.json

@@ -9,25 +9,15 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 192,
   "id2label": {
-    "0": "CONTROL SHOT",
-    "1": "COVER",
-    "2": "EDGE + SPINE",
-    "3": "FLYSHEET",
-    "4": "OTHER",
-    "5": "PAGE + FOLIO",
-    "6": "SCROLL"
+    "0": "control",
+    "1": "flysheet"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 768,
   "label2id": {
-    "CONTROL SHOT": 0,
-    "COVER": 1,
-    "EDGE + SPINE": 2,
-    "FLYSHEET": 3,
-    "OTHER": 4,
-    "PAGE + FOLIO": 5,
-    "SCROLL": 6
+    "control": 0,
+    "flysheet": 1
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

eval_results.json

@@ -1,8 +1,11 @@
 {
     "epoch": 0.0,
-    "eval_f1": 0.08855052368356824,
-    "eval_loss": 2.330942153930664,
-    "eval_runtime": 11.8647,
-    "eval_samples_per_second": 21.577,
-    "eval_steps_per_second": 5.394
+    "eval_accuracy": 0.15708812260536398,
+    "eval_f1": 0.10951914110924829,
+    "eval_loss": 2.2501182556152344,
+    "eval_precision": 0.11397786789695766,
+    "eval_recall": 0.14094382256290108,
+    "eval_runtime": 37.0603,
+    "eval_samples_per_second": 7.043,
+    "eval_steps_per_second": 1.781
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cf31d3e9c8fd7f6e1c88cd9b68c1f39603724771b652258377b5329c53a73ec3
-size 22167725
+oid sha256:ab9a363b4c2b647e249e54a9a770676b0f0a8b368c3856ae01a9a3b8bdbdc437
+size 22163885