End of training

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.3333333333333333
+      value: 0.5
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -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: 1.0246
-- Accuracy: 0.3333
-- Auc Overall: 0.625
+- Loss: 1.2537
+- Accuracy: 0.5
+- Auc Overall: 0.3333
 - Auc Class 0: 0.625
-- Auc Class 1: 1.0
-- Auc Class 2: 0.25
+- Auc Class 1: 0.375
+- Auc Class 2: 0.0
 
 ## Model description
 

all_results.json

@@ -1,22 +1,22 @@
 {
     "epoch": 3.0,
-    "eval_accuracy": 0.8333333333333334,
-    "eval_auc_class_0": 1.0,
-    "eval_auc_class_1": 0.5,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 0.8333333333333334,
-    "eval_loss": 0.8253422379493713,
-    "eval_runtime": 0.6818,
-    "eval_samples_per_second": 8.8,
-    "eval_steps_per_second": 1.467,
+    "eval_accuracy": 0.5,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.375,
+    "eval_auc_class_2": 0.0,
+    "eval_auc_overall": 0.3333333333333333,
+    "eval_loss": 1.2536860704421997,
+    "eval_runtime": 0.7286,
+    "eval_samples_per_second": 8.234,
+    "eval_steps_per_second": 1.372,
     "total_flos": 3525591661885440.0,
     "train_accuracy": 0.13333333333333333,
     "train_auc_class_0": 0.39999999999999997,
     "train_auc_class_1": 0.6000000000000001,
     "train_auc_class_2": 0.2,
     "train_auc_overall": 0.39999999999999997,
-    "train_loss": 1.3039982318878174,
-    "train_runtime": 1.7052,
-    "train_samples_per_second": 8.797,
-    "train_steps_per_second": 0.586
+    "train_loss": 0.2987220684687297,
+    "train_runtime": 73.0126,
+    "train_samples_per_second": 0.616,
+    "train_steps_per_second": 0.041
 }

eval_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 3.0,
-    "eval_accuracy": 0.8333333333333334,
-    "eval_auc_class_0": 1.0,
-    "eval_auc_class_1": 0.5,
-    "eval_auc_class_2": 1.0,
-    "eval_auc_overall": 0.8333333333333334,
-    "eval_loss": 0.8253422379493713,
-    "eval_runtime": 0.6818,
-    "eval_samples_per_second": 8.8,
-    "eval_steps_per_second": 1.467
+    "eval_accuracy": 0.5,
+    "eval_auc_class_0": 0.625,
+    "eval_auc_class_1": 0.375,
+    "eval_auc_class_2": 0.0,
+    "eval_auc_overall": 0.3333333333333333,
+    "eval_loss": 1.2536860704421997,
+    "eval_runtime": 0.7286,
+    "eval_samples_per_second": 8.234,
+    "eval_steps_per_second": 1.372
 }

runs/Sep04_10-53-36_data-science-wbi/events.out.tfevents.1725447292.data-science-wbi

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:92225dd26d505b60761ab96eb49bba68695bab7ee0b0dcce26942589966119df
-size 213
+oid sha256:35798a006af59ba3567cfa7d2f9ddfe6a8a1952d41c0f99fbf32a79af12040f3
+size 636

train_results.json

@@ -1,12 +1,8 @@
 {
     "epoch": 3.0,
-    "train_accuracy": 0.13333333333333333,
-    "train_auc_class_0": 0.39999999999999997,
-    "train_auc_class_1": 0.6000000000000001,
-    "train_auc_class_2": 0.2,
-    "train_auc_overall": 0.39999999999999997,
-    "train_loss": 1.3039982318878174,
-    "train_runtime": 1.7052,
-    "train_samples_per_second": 8.797,
-    "train_steps_per_second": 0.586
+    "total_flos": 3525591661885440.0,
+    "train_loss": 0.2987220684687297,
+    "train_runtime": 73.0126,
+    "train_samples_per_second": 0.616,
+    "train_steps_per_second": 0.041
 }

trainer_state.json

@@ -1,6 +1,6 @@
 {
-  "best_metric": 0.8333333333333334,
-  "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-2",
+  "best_metric": 0.5,
+  "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-1",
   "epoch": 3.0,
   "eval_steps": 500,
   "global_step": 3,
@@ -11,89 +11,102 @@
     {
       "epoch": 1.0,
       "step": 1,
-      "train_accuracy": 0.6,
-      "train_auc_class_0": 0.64,
-      "train_auc_class_1": 0.46,
-      "train_auc_class_2": 0.8,
-      "train_auc_overall": 0.6333333333333334,
-      "train_loss": 1.0219470262527466,
-      "train_runtime": 1.8345,
-      "train_samples_per_second": 8.177,
-      "train_steps_per_second": 0.545
+      "train_accuracy": 0.3333333333333333,
+      "train_auc_class_0": 0.52,
+      "train_auc_class_1": 0.5800000000000001,
+      "train_auc_class_2": 0.22,
+      "train_auc_overall": 0.44,
+      "train_loss": 1.3131554126739502,
+      "train_runtime": 1.6412,
+      "train_samples_per_second": 9.14,
+      "train_steps_per_second": 0.609
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.6666666666666666,
-      "eval_auc_class_0": 1.0,
-      "eval_auc_class_1": 0.5,
-      "eval_auc_class_2": 0.875,
-      "eval_auc_overall": 0.7916666666666666,
-      "eval_loss": 0.9240671992301941,
-      "eval_runtime": 0.7631,
-      "eval_samples_per_second": 7.863,
-      "eval_steps_per_second": 1.31,
+      "eval_accuracy": 0.5,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 0.375,
+      "eval_auc_class_2": 0.0,
+      "eval_auc_overall": 0.3333333333333333,
+      "eval_loss": 1.2536860704421997,
+      "eval_runtime": 0.6979,
+      "eval_samples_per_second": 8.597,
+      "eval_steps_per_second": 1.433,
       "step": 1
     },
     {
       "epoch": 2.0,
       "step": 2,
-      "train_accuracy": 0.8,
-      "train_auc_class_0": 1.0,
-      "train_auc_class_1": 0.96,
-      "train_auc_class_2": 1.0,
-      "train_auc_overall": 0.9866666666666667,
-      "train_loss": 0.6553783416748047,
-      "train_runtime": 1.6643,
-      "train_samples_per_second": 9.013,
-      "train_steps_per_second": 0.601
+      "train_accuracy": 0.6666666666666666,
+      "train_auc_class_0": 0.7799999999999999,
+      "train_auc_class_1": 0.8799999999999999,
+      "train_auc_class_2": 0.64,
+      "train_auc_overall": 0.7666666666666666,
+      "train_loss": 0.9476243853569031,
+      "train_runtime": 1.6529,
+      "train_samples_per_second": 9.075,
+      "train_steps_per_second": 0.605
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.8333333333333334,
-      "eval_auc_class_0": 1.0,
-      "eval_auc_class_1": 0.5,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 0.8333333333333334,
-      "eval_loss": 0.8253422379493713,
-      "eval_runtime": 0.6909,
-      "eval_samples_per_second": 8.685,
-      "eval_steps_per_second": 1.447,
+      "eval_accuracy": 0.3333333333333333,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 0.75,
+      "eval_auc_class_2": 0.125,
+      "eval_auc_overall": 0.5,
+      "eval_loss": 1.089605689048767,
+      "eval_runtime": 0.7552,
+      "eval_samples_per_second": 7.945,
+      "eval_steps_per_second": 1.324,
       "step": 2
     },
     {
       "epoch": 3.0,
       "step": 3,
-      "train_accuracy": 0.8666666666666667,
-      "train_auc_class_0": 0.96,
-      "train_auc_class_1": 1.0,
-      "train_auc_class_2": 0.98,
-      "train_auc_overall": 0.98,
-      "train_loss": 0.7010973691940308,
-      "train_runtime": 1.7627,
-      "train_samples_per_second": 8.51,
-      "train_steps_per_second": 0.567
+      "train_accuracy": 0.8,
+      "train_auc_class_0": 1.0,
+      "train_auc_class_1": 0.98,
+      "train_auc_class_2": 0.9199999999999999,
+      "train_auc_overall": 0.9666666666666667,
+      "train_loss": 0.7161972522735596,
+      "train_runtime": 1.7658,
+      "train_samples_per_second": 8.495,
+      "train_steps_per_second": 0.566
     },
     {
       "epoch": 3.0,
-      "eval_accuracy": 0.8333333333333334,
-      "eval_auc_class_0": 1.0,
-      "eval_auc_class_1": 0.625,
-      "eval_auc_class_2": 1.0,
-      "eval_auc_overall": 0.875,
-      "eval_loss": 0.7836728096008301,
-      "eval_runtime": 0.7323,
-      "eval_samples_per_second": 8.193,
-      "eval_steps_per_second": 1.366,
+      "eval_accuracy": 0.3333333333333333,
+      "eval_auc_class_0": 0.625,
+      "eval_auc_class_1": 1.0,
+      "eval_auc_class_2": 0.25,
+      "eval_auc_overall": 0.625,
+      "eval_loss": 1.0246309041976929,
+      "eval_runtime": 0.7191,
+      "eval_samples_per_second": 8.344,
+      "eval_steps_per_second": 1.391,
       "step": 3
     },
     {
       "epoch": 3.0,
       "step": 3,
       "total_flos": 3525591661885440.0,
-      "train_loss": 0.2555307348569234,
-      "train_runtime": 72.9305,
-      "train_samples_per_second": 0.617,
+      "train_loss": 0.2987220684687297,
+      "train_runtime": 73.0126,
+      "train_samples_per_second": 0.616,
       "train_steps_per_second": 0.041
+    },
+    {
+      "epoch": 3.0,
+      "step": 3,
+      "train_accuracy": 0.13333333333333333,
+      "train_auc_class_0": 0.39999999999999997,
+      "train_auc_class_1": 0.6000000000000001,
+      "train_auc_class_2": 0.2,
+      "train_auc_overall": 0.39999999999999997,
+      "train_loss": 1.3039982318878174,
+      "train_runtime": 1.7052,
+      "train_samples_per_second": 8.797,
+      "train_steps_per_second": 0.586
     }
   ],
   "logging_steps": 10,