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.9299
+- Loss: 0.9763
 - Accuracy: 0.6667
-- Auc Overall: 0.7917
-- Auc Class 0: 0.75
-- Auc Class 1: 0.625
+- Auc Overall: 0.7083
+- Auc Class 0: 0.625
+- Auc Class 1: 0.5
 - Auc Class 2: 1.0
 
 ## Model description

all_results.json

@@ -1,22 +1,22 @@
 {
     "epoch": 3.0,
-    "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,
+    "eval_accuracy": 0.6666666666666666,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.5,
+    "eval_auc_class_2": 1.0,
+    "eval_auc_overall": 0.7083333333333334,
+    "eval_loss": 0.9763297438621521,
+    "eval_runtime": 0.692,
+    "eval_samples_per_second": 8.67,
+    "eval_steps_per_second": 1.445,
     "total_flos": 3525591661885440.0,
     "train_accuracy": 0.7333333333333333,
     "train_auc_class_0": 0.8400000000000001,
     "train_auc_class_1": 0.9199999999999999,
     "train_auc_class_2": 1.0,
     "train_auc_overall": 0.9199999999999999,
-    "train_loss": 0.7121346592903137,
-    "train_runtime": 1.6755,
-    "train_samples_per_second": 8.952,
-    "train_steps_per_second": 0.597
+    "train_loss": 0.24405852953592935,
+    "train_runtime": 72.8636,
+    "train_samples_per_second": 0.618,
+    "train_steps_per_second": 0.041
 }

eval_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 3.0,
-    "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
+    "eval_accuracy": 0.6666666666666666,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.5,
+    "eval_auc_class_2": 1.0,
+    "eval_auc_overall": 0.7083333333333334,
+    "eval_loss": 0.9763297438621521,
+    "eval_runtime": 0.692,
+    "eval_samples_per_second": 8.67,
+    "eval_steps_per_second": 1.445
 }

runs/Sep04_11-58-40_data-science-wbi/events.out.tfevents.1725451197.data-science-wbi

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

train_results.json

@@ -1,12 +1,8 @@
 {
     "epoch": 3.0,
-    "train_accuracy": 0.7333333333333333,
-    "train_auc_class_0": 0.8400000000000001,
-    "train_auc_class_1": 0.9199999999999999,
-    "train_auc_class_2": 1.0,
-    "train_auc_overall": 0.9199999999999999,
-    "train_loss": 0.7121346592903137,
-    "train_runtime": 1.6755,
-    "train_samples_per_second": 8.952,
-    "train_steps_per_second": 0.597
+    "total_flos": 3525591661885440.0,
+    "train_loss": 0.24405852953592935,
+    "train_runtime": 72.8636,
+    "train_samples_per_second": 0.618,
+    "train_steps_per_second": 0.041
 }

trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.3333333333333333,
+  "best_metric": 0.6666666666666666,
   "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.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
+      "train_accuracy": 0.3333333333333333,
+      "train_auc_class_0": 0.6000000000000001,
+      "train_auc_class_1": 0.42,
+      "train_auc_class_2": 0.41999999999999993,
+      "train_auc_overall": 0.48,
+      "train_loss": 1.1545398235321045,
+      "train_runtime": 1.8894,
+      "train_samples_per_second": 7.939,
+      "train_steps_per_second": 0.529
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.16666666666666666,
-      "eval_auc_class_0": 0.375,
+      "eval_accuracy": 0.5,
+      "eval_auc_class_0": 0.625,
       "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,
+      "eval_auc_class_2": 0.75,
+      "eval_auc_overall": 0.5416666666666666,
+      "eval_loss": 1.1536046266555786,
+      "eval_runtime": 0.6925,
+      "eval_samples_per_second": 8.665,
+      "eval_steps_per_second": 1.444,
       "step": 1
     },
     {
       "epoch": 2.0,
       "step": 2,
       "train_accuracy": 0.6,
-      "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
+      "train_auc_class_0": 0.8400000000000001,
+      "train_auc_class_1": 0.7200000000000001,
+      "train_auc_class_2": 0.8600000000000001,
+      "train_auc_overall": 0.8066666666666666,
+      "train_loss": 0.87682044506073,
+      "train_runtime": 1.6553,
+      "train_samples_per_second": 9.062,
+      "train_steps_per_second": 0.604
     },
     {
       "epoch": 2.0,
-      "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,
+      "eval_accuracy": 0.6666666666666666,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 0.5,
+      "eval_auc_class_2": 1.0,
+      "eval_auc_overall": 0.7083333333333334,
+      "eval_loss": 0.9763297438621521,
+      "eval_runtime": 0.6997,
+      "eval_samples_per_second": 8.575,
+      "eval_steps_per_second": 1.429,
       "step": 2
     },
     {
       "epoch": 3.0,
       "step": 3,
-      "train_accuracy": 0.6666666666666666,
+      "train_accuracy": 0.8666666666666667,
       "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
+      "train_auc_class_1": 0.96,
+      "train_auc_class_2": 1.0,
+      "train_auc_overall": 0.9466666666666667,
+      "train_loss": 0.6894626617431641,
+      "train_runtime": 1.7265,
+      "train_samples_per_second": 8.688,
+      "train_steps_per_second": 0.579
     },
     {
       "epoch": 3.0,
-      "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,
+      "eval_accuracy": 0.6666666666666666,
+      "eval_auc_class_0": 0.75,
+      "eval_auc_class_1": 0.625,
+      "eval_auc_class_2": 1.0,
+      "eval_auc_overall": 0.7916666666666666,
+      "eval_loss": 0.9298850893974304,
+      "eval_runtime": 0.7147,
+      "eval_samples_per_second": 8.395,
+      "eval_steps_per_second": 1.399,
       "step": 3
     },
     {
       "epoch": 3.0,
       "step": 3,
       "total_flos": 3525591661885440.0,
-      "train_loss": 0.25828051567077637,
-      "train_runtime": 73.8621,
-      "train_samples_per_second": 0.609,
+      "train_loss": 0.24405852953592935,
+      "train_runtime": 72.8636,
+      "train_samples_per_second": 0.618,
       "train_steps_per_second": 0.041
     },
     {
       "epoch": 3.0,
       "step": 3,
-      "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_accuracy": 0.7333333333333333,
+      "train_auc_class_0": 0.8400000000000001,
+      "train_auc_class_1": 0.9199999999999999,
+      "train_auc_class_2": 1.0,
+      "train_auc_overall": 0.9199999999999999,
+      "train_loss": 0.7121346592903137,
+      "train_runtime": 1.6755,
+      "train_samples_per_second": 8.952,
+      "train_steps_per_second": 0.597
     }
   ],
   "logging_steps": 10,