End of training

README.md

@@ -33,12 +33,12 @@ 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.6634
+- Loss: 0.9276
 - Accuracy: 0.6667
-- Auc Overall: 1.0
-- Auc Class 0: 1.0
-- Auc Class 1: 1.0
-- Auc Class 2: 1.0
+- Auc Overall: 0.75
+- Auc Class 0: 0.625
+- Auc Class 1: 0.75
+- Auc Class 2: 0.875
 
 ## Model description
 

all_results.json

@@ -1,17 +1,17 @@
 {
     "epoch": 7.0,
-    "eval_accuracy": 0.8333333333333334,
-    "eval_auc_class_0": 1.0,
-    "eval_auc_class_1": 1.0,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 1.0,
-    "eval_loss": 0.8139967918395996,
-    "eval_runtime": 0.735,
-    "eval_samples_per_second": 8.164,
-    "eval_steps_per_second": 1.361,
+    "eval_accuracy": 0.6666666666666666,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.75,
+    "eval_auc_class_2": 0.875,
+    "eval_auc_overall": 0.75,
+    "eval_loss": 0.9276485443115234,
+    "eval_runtime": 0.7838,
+    "eval_samples_per_second": 7.655,
+    "eval_steps_per_second": 1.276,
     "total_flos": 8226380544399360.0,
-    "train_loss": 0.17954584530421666,
-    "train_runtime": 78.6151,
-    "train_samples_per_second": 1.336,
-    "train_steps_per_second": 0.089
+    "train_loss": 0.14502547468457902,
+    "train_runtime": 156.7549,
+    "train_samples_per_second": 0.67,
+    "train_steps_per_second": 0.045
 }

eval_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 7.0,
-    "eval_accuracy": 0.8333333333333334,
-    "eval_auc_class_0": 1.0,
-    "eval_auc_class_1": 1.0,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 1.0,
-    "eval_loss": 0.8139967918395996,
-    "eval_runtime": 0.735,
-    "eval_samples_per_second": 8.164,
-    "eval_steps_per_second": 1.361
+    "eval_accuracy": 0.6666666666666666,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.75,
+    "eval_auc_class_2": 0.875,
+    "eval_auc_overall": 0.75,
+    "eval_loss": 0.9276485443115234,
+    "eval_runtime": 0.7838,
+    "eval_samples_per_second": 7.655,
+    "eval_steps_per_second": 1.276
 }

runs/Sep03_08-41-07_data-science-wbi/events.out.tfevents.1725353077.data-science-wbi

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:04ce1f0a35609bff3610bd0c3b549c2e596446e849a65a06d2d9cbce39ea16ab
+size 40

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 7.0,
     "total_flos": 8226380544399360.0,
-    "train_loss": 0.17954584530421666,
-    "train_runtime": 78.6151,
-    "train_samples_per_second": 1.336,
-    "train_steps_per_second": 0.089
+    "train_loss": 0.14502547468457902,
+    "train_runtime": 156.7549,
+    "train_samples_per_second": 0.67,
+    "train_steps_per_second": 0.045
 }

trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.8333333333333334,
+  "best_metric": 0.6666666666666666,
   "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-2",
   "epoch": 7.0,
   "eval_steps": 500,
@@ -11,79 +11,79 @@
     {
       "epoch": 1.0,
       "step": 1,
-      "train_accuracy": 0.26666666666666666,
-      "train_auc_class_0": 0.32000000000000006,
-      "train_auc_class_1": 0.6200000000000001,
-      "train_auc_class_2": 0.39999999999999997,
-      "train_auc_overall": 0.4466666666666667,
-      "train_loss": 1.1479226350784302,
-      "train_runtime": 1.6122,
-      "train_samples_per_second": 9.304,
-      "train_steps_per_second": 0.62
+      "train_accuracy": 0.6,
+      "train_auc_class_0": 0.36,
+      "train_auc_class_1": 0.8,
+      "train_auc_class_2": 0.8400000000000001,
+      "train_auc_overall": 0.6666666666666666,
+      "train_loss": 0.9687549471855164,
+      "train_runtime": 1.665,
+      "train_samples_per_second": 9.009,
+      "train_steps_per_second": 0.601
     },
     {
       "epoch": 1.0,
       "eval_accuracy": 0.5,
-      "eval_auc_class_0": 0.875,
-      "eval_auc_class_1": 0.5,
-      "eval_auc_class_2": 0.625,
-      "eval_auc_overall": 0.6666666666666666,
-      "eval_loss": 0.9469685554504395,
-      "eval_runtime": 0.6769,
-      "eval_samples_per_second": 8.863,
-      "eval_steps_per_second": 1.477,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 0.75,
+      "eval_auc_class_2": 0.75,
+      "eval_auc_overall": 0.7083333333333334,
+      "eval_loss": 1.028936505317688,
+      "eval_runtime": 0.6838,
+      "eval_samples_per_second": 8.774,
+      "eval_steps_per_second": 1.462,
       "step": 1
     },
     {
       "epoch": 2.0,
       "step": 2,
-      "train_accuracy": 0.6,
-      "train_auc_class_0": 0.82,
-      "train_auc_class_1": 0.96,
-      "train_auc_class_2": 0.9,
-      "train_auc_overall": 0.8933333333333332,
-      "train_loss": 0.814429521560669,
-      "train_runtime": 1.6517,
-      "train_samples_per_second": 9.081,
-      "train_steps_per_second": 0.605
+      "train_accuracy": 0.8,
+      "train_auc_class_0": 0.98,
+      "train_auc_class_1": 1.0,
+      "train_auc_class_2": 1.0,
+      "train_auc_overall": 0.9933333333333333,
+      "train_loss": 0.6606505513191223,
+      "train_runtime": 1.7751,
+      "train_samples_per_second": 8.45,
+      "train_steps_per_second": 0.563
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.8333333333333334,
-      "eval_auc_class_0": 1.0,
-      "eval_auc_class_1": 1.0,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 1.0,
-      "eval_loss": 0.8139967918395996,
-      "eval_runtime": 0.7014,
-      "eval_samples_per_second": 8.555,
-      "eval_steps_per_second": 1.426,
+      "eval_accuracy": 0.6666666666666666,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 0.75,
+      "eval_auc_class_2": 0.875,
+      "eval_auc_overall": 0.75,
+      "eval_loss": 0.9276485443115234,
+      "eval_runtime": 0.7125,
+      "eval_samples_per_second": 8.421,
+      "eval_steps_per_second": 1.404,
       "step": 2
     },
     {
       "epoch": 3.0,
       "step": 3,
-      "train_accuracy": 0.8666666666666667,
+      "train_accuracy": 1.0,
       "train_auc_class_0": 1.0,
       "train_auc_class_1": 1.0,
-      "train_auc_class_2": 0.9600000000000001,
-      "train_auc_overall": 0.9866666666666667,
-      "train_loss": 0.6608889698982239,
-      "train_runtime": 1.7063,
-      "train_samples_per_second": 8.791,
-      "train_steps_per_second": 0.586
+      "train_auc_class_2": 1.0,
+      "train_auc_overall": 1.0,
+      "train_loss": 0.5248541831970215,
+      "train_runtime": 1.627,
+      "train_samples_per_second": 9.219,
+      "train_steps_per_second": 0.615
     },
     {
       "epoch": 3.0,
-      "eval_accuracy": 0.8333333333333334,
-      "eval_auc_class_0": 1.0,
+      "eval_accuracy": 0.6666666666666666,
+      "eval_auc_class_0": 0.625,
       "eval_auc_class_1": 1.0,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 1.0,
-      "eval_loss": 0.7299937605857849,
-      "eval_runtime": 0.7538,
-      "eval_samples_per_second": 7.959,
-      "eval_steps_per_second": 1.327,
+      "eval_auc_class_2": 0.875,
+      "eval_auc_overall": 0.8333333333333334,
+      "eval_loss": 0.8520984053611755,
+      "eval_runtime": 0.7245,
+      "eval_samples_per_second": 8.281,
+      "eval_steps_per_second": 1.38,
       "step": 3
     },
     {
@@ -94,74 +94,74 @@
       "train_auc_class_1": 1.0,
       "train_auc_class_2": 1.0,
       "train_auc_overall": 1.0,
-      "train_loss": 0.42420145869255066,
-      "train_runtime": 1.6223,
-      "train_samples_per_second": 9.246,
-      "train_steps_per_second": 0.616
+      "train_loss": 0.36024707555770874,
+      "train_runtime": 1.7829,
+      "train_samples_per_second": 8.413,
+      "train_steps_per_second": 0.561
     },
     {
       "epoch": 4.0,
-      "eval_accuracy": 0.8333333333333334,
-      "eval_auc_class_0": 1.0,
+      "eval_accuracy": 0.6666666666666666,
+      "eval_auc_class_0": 0.875,
       "eval_auc_class_1": 1.0,
       "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 1.0,
-      "eval_loss": 0.6729159355163574,
-      "eval_runtime": 0.772,
-      "eval_samples_per_second": 7.772,
-      "eval_steps_per_second": 1.295,
+      "eval_auc_overall": 0.9583333333333334,
+      "eval_loss": 0.7887496948242188,
+      "eval_runtime": 0.8054,
+      "eval_samples_per_second": 7.45,
+      "eval_steps_per_second": 1.242,
       "step": 4
     },
     {
       "epoch": 5.0,
       "step": 5,
-      "train_accuracy": 0.9333333333333333,
+      "train_accuracy": 1.0,
       "train_auc_class_0": 1.0,
       "train_auc_class_1": 1.0,
       "train_auc_class_2": 1.0,
       "train_auc_overall": 1.0,
-      "train_loss": 0.42396074533462524,
-      "train_runtime": 1.6518,
-      "train_samples_per_second": 9.081,
-      "train_steps_per_second": 0.605
+      "train_loss": 0.31448933482170105,
+      "train_runtime": 1.9887,
+      "train_samples_per_second": 7.543,
+      "train_steps_per_second": 0.503
     },
     {
       "epoch": 5.0,
-      "eval_accuracy": 0.8333333333333334,
+      "eval_accuracy": 0.6666666666666666,
       "eval_auc_class_0": 1.0,
       "eval_auc_class_1": 1.0,
       "eval_auc_class_2": 1.0,
       "eval_auc_overall": 1.0,
-      "eval_loss": 0.6391382813453674,
-      "eval_runtime": 0.6917,
-      "eval_samples_per_second": 8.675,
-      "eval_steps_per_second": 1.446,
+      "eval_loss": 0.7302556037902832,
+      "eval_runtime": 0.719,
+      "eval_samples_per_second": 8.345,
+      "eval_steps_per_second": 1.391,
       "step": 5
     },
     {
       "epoch": 6.0,
       "step": 6,
-      "train_accuracy": 1.0,
+      "train_accuracy": 0.9333333333333333,
       "train_auc_class_0": 1.0,
       "train_auc_class_1": 1.0,
       "train_auc_class_2": 1.0,
       "train_auc_overall": 1.0,
-      "train_loss": 0.29362058639526367,
-      "train_runtime": 1.6474,
-      "train_samples_per_second": 9.105,
-      "train_steps_per_second": 0.607
+      "train_loss": 0.31753572821617126,
+      "train_runtime": 1.8048,
+      "train_samples_per_second": 8.311,
+      "train_steps_per_second": 0.554
     },
     {
       "epoch": 6.0,
-      "eval_accuracy": 0.8333333333333334,
+      "eval_accuracy": 0.6666666666666666,
       "eval_auc_class_0": 1.0,
       "eval_auc_class_1": 1.0,
       "eval_auc_class_2": 1.0,
       "eval_auc_overall": 1.0,
-      "eval_loss": 0.6181725859642029,
-      "eval_runtime": 0.6858,
-      "eval_samples_per_second": 8.748,
-      "eval_steps_per_second": 1.458,
+      "eval_loss": 0.688157320022583,
+      "eval_runtime": 0.719,
+      "eval_samples_per_second": 8.345,
+      "eval_steps_per_second": 1.391,
       "step": 6
     },
     {
@@ -172,32 +172,32 @@
       "train_auc_class_1": 1.0,
       "train_auc_class_2": 1.0,
       "train_auc_overall": 1.0,
-      "train_loss": 0.35040372610092163,
-      "train_runtime": 1.7724,
-      "train_samples_per_second": 8.463,
-      "train_steps_per_second": 0.564
+      "train_loss": 0.27592089772224426,
+      "train_runtime": 2.0142,
+      "train_samples_per_second": 7.447,
+      "train_steps_per_second": 0.496
     },
     {
       "epoch": 7.0,
-      "eval_accuracy": 0.8333333333333334,
+      "eval_accuracy": 0.6666666666666666,
       "eval_auc_class_0": 1.0,
       "eval_auc_class_1": 1.0,
       "eval_auc_class_2": 1.0,
       "eval_auc_overall": 1.0,
-      "eval_loss": 0.6054350733757019,
-      "eval_runtime": 0.7011,
-      "eval_samples_per_second": 8.558,
-      "eval_steps_per_second": 1.426,
+      "eval_loss": 0.6634145379066467,
+      "eval_runtime": 0.7768,
+      "eval_samples_per_second": 7.724,
+      "eval_steps_per_second": 1.287,
       "step": 7
     },
     {
       "epoch": 7.0,
       "step": 7,
       "total_flos": 8226380544399360.0,
-      "train_loss": 0.17954584530421666,
-      "train_runtime": 78.6151,
-      "train_samples_per_second": 1.336,
-      "train_steps_per_second": 0.089
+      "train_loss": 0.14502547468457902,
+      "train_runtime": 156.7549,
+      "train_samples_per_second": 0.67,
+      "train_steps_per_second": 0.045
     }
   ],
   "logging_steps": 10,