Brain MRI 🧠 fine tune II run - 10 epochs.

README.md

@@ -3,6 +3,7 @@ library_name: transformers
 license: apache-2.0
 base_model: google/vit-base-patch16-224
 tags:
+- image-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -16,10 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-brain-mri
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the BrainMRI dataset.
 It achieves the following results on the evaluation set:
-- Accuracy: 0.8223
-- Loss: 0.5857
+- Loss: 1.1297
+- Accuracy: 0.5685
 
 ## Model description
 

all_results.json

@@ -2,12 +2,12 @@
     "epoch": 14.0,
     "eval_accuracy": 0.5685279187817259,
     "eval_loss": 1.129746437072754,
-    "eval_runtime": 3.2309,
-    "eval_samples_per_second": 121.946,
-    "eval_steps_per_second": 15.475,
+    "eval_runtime": 3.146,
+    "eval_samples_per_second": 125.24,
+    "eval_steps_per_second": 15.893,
     "total_flos": 2.4909471645141565e+18,
-    "train_loss": 0.15913245034596277,
-    "train_runtime": 101.1425,
-    "train_samples_per_second": 317.809,
-    "train_steps_per_second": 9.966
+    "train_loss": 0.0,
+    "train_runtime": 0.147,
+    "train_samples_per_second": 218667.975,
+    "train_steps_per_second": 6857.184
 }

config.json

@@ -29,6 +29,7 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.44.2"

eval_results.json

@@ -2,7 +2,7 @@
     "epoch": 14.0,
     "eval_accuracy": 0.5685279187817259,
     "eval_loss": 1.129746437072754,
-    "eval_runtime": 3.2309,
-    "eval_samples_per_second": 121.946,
-    "eval_steps_per_second": 15.475
+    "eval_runtime": 3.146,
+    "eval_samples_per_second": 125.24,
+    "eval_steps_per_second": 15.893
 }

runs/Aug25_15-54-39_DESKTOP-4SGMSGR/events.out.tfevents.1724590487.DESKTOP-4SGMSGR.27212.1

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

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 14.0,
     "total_flos": 2.4909471645141565e+18,
-    "train_loss": 0.15913245034596277,
-    "train_runtime": 101.1425,
-    "train_samples_per_second": 317.809,
-    "train_steps_per_second": 9.966
+    "train_loss": 0.0,
+    "train_runtime": 0.147,
+    "train_samples_per_second": 218667.975,
+    "train_steps_per_second": 6857.184
 }

trainer_state.json

@@ -208,10 +208,10 @@
       "epoch": 14.0,
       "step": 1008,
       "total_flos": 2.4909471645141565e+18,
-      "train_loss": 0.15913245034596277,
-      "train_runtime": 101.1425,
-      "train_samples_per_second": 317.809,
-      "train_steps_per_second": 9.966
+      "train_loss": 0.0,
+      "train_runtime": 0.147,
+      "train_samples_per_second": 218667.975,
+      "train_steps_per_second": 6857.184
     }
   ],
   "logging_steps": 100,