End of training

README.md

@@ -33,11 +33,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/swin-base-patch4-window7-224-in22k](https://huggingface.co/microsoft/swin-base-patch4-window7-224-in22k) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9920
+- Loss: 1.0399
 - Accuracy: 0.3333
-- Auc Overall: 0.75
-- Auc Class 0: 0.5
-- Auc Class 1: 1.0
+- Auc Overall: 0.6667
+- Auc Class 0: 0.375
+- Auc Class 1: 0.875
 - Auc Class 2: 0.75
 
 ## Model description

all_results.json

@@ -1,22 +1,22 @@
 {
     "epoch": 3.0,
-    "eval_accuracy": 0.5,
-    "eval_auc_class_0": 0.25,
-    "eval_auc_class_1": 0.5,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 0.5833333333333334,
-    "eval_loss": 1.0559865236282349,
-    "eval_runtime": 0.6886,
-    "eval_samples_per_second": 8.713,
-    "eval_steps_per_second": 1.452,
+    "eval_accuracy": 0.3333333333333333,
+    "eval_auc_class_0": 0.375,
+    "eval_auc_class_1": 0.875,
+    "eval_auc_class_2": 0.75,
+    "eval_auc_overall": 0.6666666666666666,
+    "eval_loss": 1.0399374961853027,
+    "eval_runtime": 0.6861,
+    "eval_samples_per_second": 8.745,
+    "eval_steps_per_second": 1.458,
     "total_flos": 3525591661885440.0,
     "train_accuracy": 0.6666666666666666,
     "train_auc_class_0": 0.8200000000000001,
     "train_auc_class_1": 0.92,
     "train_auc_class_2": 0.7200000000000001,
     "train_auc_overall": 0.8200000000000002,
-    "train_loss": 0.9375088214874268,
-    "train_runtime": 1.5198,
-    "train_samples_per_second": 9.87,
-    "train_steps_per_second": 0.658
+    "train_loss": 0.25828051567077637,
+    "train_runtime": 73.8621,
+    "train_samples_per_second": 0.609,
+    "train_steps_per_second": 0.041
 }

eval_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 3.0,
-    "eval_accuracy": 0.5,
-    "eval_auc_class_0": 0.25,
-    "eval_auc_class_1": 0.5,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 0.5833333333333334,
-    "eval_loss": 1.0559865236282349,
-    "eval_runtime": 0.6886,
-    "eval_samples_per_second": 8.713,
-    "eval_steps_per_second": 1.452
+    "eval_accuracy": 0.3333333333333333,
+    "eval_auc_class_0": 0.375,
+    "eval_auc_class_1": 0.875,
+    "eval_auc_class_2": 0.75,
+    "eval_auc_overall": 0.6666666666666666,
+    "eval_loss": 1.0399374961853027,
+    "eval_runtime": 0.6861,
+    "eval_samples_per_second": 8.745,
+    "eval_steps_per_second": 1.458
 }

runs/Sep04_11-19-25_data-science-wbi/events.out.tfevents.1725448843.data-science-wbi

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:64ce92500ffe72bc5aa1d8e81055b04c82d01bfa2bd75047313bdc77d7c595fa
-size 152
+oid sha256:3f92c3b9149a82ac2d08fa7da5ba4020210ecc8e1560e05ac60283e6431d15dd
+size 636

train_results.json

@@ -1,12 +1,8 @@
 {
     "epoch": 3.0,
-    "train_accuracy": 0.6666666666666666,
-    "train_auc_class_0": 0.8200000000000001,
-    "train_auc_class_1": 0.92,
-    "train_auc_class_2": 0.7200000000000001,
-    "train_auc_overall": 0.8200000000000002,
-    "train_loss": 0.9375088214874268,
-    "train_runtime": 1.5198,
-    "train_samples_per_second": 9.87,
-    "train_steps_per_second": 0.658
+    "total_flos": 3525591661885440.0,
+    "train_loss": 0.25828051567077637,
+    "train_runtime": 73.8621,
+    "train_samples_per_second": 0.609,
+    "train_steps_per_second": 0.041
 }

trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.5,
+  "best_metric": 0.3333333333333333,
   "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-2",
   "epoch": 3.0,
   "eval_steps": 500,
@@ -11,102 +11,102 @@
     {
       "epoch": 1.0,
       "step": 1,
-      "train_accuracy": 0.5333333333333333,
-      "train_auc_class_0": 0.34,
-      "train_auc_class_1": 0.6599999999999999,
-      "train_auc_class_2": 0.5599999999999999,
-      "train_auc_overall": 0.52,
-      "train_loss": 1.1574488878250122,
-      "train_runtime": 1.6568,
-      "train_samples_per_second": 9.054,
-      "train_steps_per_second": 0.604
+      "train_accuracy": 0.2,
+      "train_auc_class_0": 0.52,
+      "train_auc_class_1": 0.32,
+      "train_auc_class_2": 0.44000000000000006,
+      "train_auc_overall": 0.42666666666666675,
+      "train_loss": 1.2085740566253662,
+      "train_runtime": 1.8632,
+      "train_samples_per_second": 8.051,
+      "train_steps_per_second": 0.537
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.3333333333333333,
-      "eval_auc_class_0": 0.25,
-      "eval_auc_class_1": 0.375,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 0.5416666666666666,
-      "eval_loss": 1.2032381296157837,
-      "eval_runtime": 0.6994,
-      "eval_samples_per_second": 8.579,
-      "eval_steps_per_second": 1.43,
+      "eval_accuracy": 0.16666666666666666,
+      "eval_auc_class_0": 0.375,
+      "eval_auc_class_1": 0.25,
+      "eval_auc_class_2": 0.625,
+      "eval_auc_overall": 0.4166666666666667,
+      "eval_loss": 1.1931848526000977,
+      "eval_runtime": 0.71,
+      "eval_samples_per_second": 8.451,
+      "eval_steps_per_second": 1.409,
       "step": 1
     },
     {
       "epoch": 2.0,
       "step": 2,
       "train_accuracy": 0.6,
-      "train_auc_class_0": 0.5,
-      "train_auc_class_1": 0.92,
-      "train_auc_class_2": 0.8,
-      "train_auc_overall": 0.7399999999999999,
-      "train_loss": 0.9586163759231567,
-      "train_runtime": 1.6488,
-      "train_samples_per_second": 9.097,
-      "train_steps_per_second": 0.606
+      "train_auc_class_0": 0.8800000000000001,
+      "train_auc_class_1": 0.78,
+      "train_auc_class_2": 0.96,
+      "train_auc_overall": 0.8733333333333334,
+      "train_loss": 0.8179279565811157,
+      "train_runtime": 1.6625,
+      "train_samples_per_second": 9.022,
+      "train_steps_per_second": 0.601
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.5,
-      "eval_auc_class_0": 0.25,
-      "eval_auc_class_1": 0.5,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 0.5833333333333334,
-      "eval_loss": 1.0559865236282349,
-      "eval_runtime": 0.6879,
-      "eval_samples_per_second": 8.723,
-      "eval_steps_per_second": 1.454,
+      "eval_accuracy": 0.3333333333333333,
+      "eval_auc_class_0": 0.375,
+      "eval_auc_class_1": 0.875,
+      "eval_auc_class_2": 0.75,
+      "eval_auc_overall": 0.6666666666666666,
+      "eval_loss": 1.0399374961853027,
+      "eval_runtime": 0.7076,
+      "eval_samples_per_second": 8.48,
+      "eval_steps_per_second": 1.413,
       "step": 2
     },
     {
       "epoch": 3.0,
       "step": 3,
       "train_accuracy": 0.6666666666666666,
-      "train_auc_class_0": 0.7399999999999999,
-      "train_auc_class_1": 0.8800000000000001,
-      "train_auc_class_2": 0.8600000000000001,
-      "train_auc_overall": 0.8266666666666668,
-      "train_loss": 0.8743086457252502,
-      "train_runtime": 1.6839,
-      "train_samples_per_second": 8.908,
-      "train_steps_per_second": 0.594
+      "train_auc_class_0": 0.88,
+      "train_auc_class_1": 0.9400000000000001,
+      "train_auc_class_2": 0.96,
+      "train_auc_overall": 0.9266666666666667,
+      "train_loss": 0.7590439319610596,
+      "train_runtime": 1.5813,
+      "train_samples_per_second": 9.486,
+      "train_steps_per_second": 0.632
     },
     {
       "epoch": 3.0,
-      "eval_accuracy": 0.5,
-      "eval_auc_class_0": 0.25,
-      "eval_auc_class_1": 0.625,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 0.625,
-      "eval_loss": 0.9944595694541931,
-      "eval_runtime": 0.6872,
-      "eval_samples_per_second": 8.73,
-      "eval_steps_per_second": 1.455,
+      "eval_accuracy": 0.3333333333333333,
+      "eval_auc_class_0": 0.5,
+      "eval_auc_class_1": 1.0,
+      "eval_auc_class_2": 0.75,
+      "eval_auc_overall": 0.75,
+      "eval_loss": 0.9920186996459961,
+      "eval_runtime": 0.6966,
+      "eval_samples_per_second": 8.613,
+      "eval_steps_per_second": 1.435,
       "step": 3
     },
     {
       "epoch": 3.0,
       "step": 3,
       "total_flos": 3525591661885440.0,
-      "train_loss": 0.2689722975095113,
-      "train_runtime": 72.4115,
-      "train_samples_per_second": 0.621,
+      "train_loss": 0.25828051567077637,
+      "train_runtime": 73.8621,
+      "train_samples_per_second": 0.609,
       "train_steps_per_second": 0.041
     },
     {
       "epoch": 3.0,
       "step": 3,
-      "train_accuracy": 0.6,
-      "train_auc_class_0": 0.7,
-      "train_auc_class_1": 0.96,
-      "train_auc_class_2": 0.8200000000000001,
-      "train_auc_overall": 0.8266666666666667,
-      "train_loss": 0.9045717120170593,
-      "train_runtime": 1.6579,
-      "train_samples_per_second": 9.047,
-      "train_steps_per_second": 0.603
+      "train_accuracy": 0.6666666666666666,
+      "train_auc_class_0": 0.8200000000000001,
+      "train_auc_class_1": 0.92,
+      "train_auc_class_2": 0.7200000000000001,
+      "train_auc_overall": 0.8200000000000002,
+      "train_loss": 0.9375088214874268,
+      "train_runtime": 1.5198,
+      "train_samples_per_second": 9.87,
+      "train_steps_per_second": 0.658
     }
   ],
   "logging_steps": 10,