End of training

README.md

@@ -2,6 +2,8 @@
 library_name: transformers
 base_model: google/mobilenet_v2_1.0_224
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 metrics:
 - accuracy
@@ -18,28 +20,28 @@ should probably proofread and complete it, then remove this comment. -->
 
 # mobilenetv2-typecoffee-2
 
-This model is a fine-tuned version of [google/mobilenet_v2_1.0_224](https://huggingface.co/google/mobilenet_v2_1.0_224) on an unknown dataset.
+This model is a fine-tuned version of [google/mobilenet_v2_1.0_224](https://huggingface.co/google/mobilenet_v2_1.0_224) on the Master-Rapha7/TypeCoffee_128x128 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.6716
-- Accuracy: 0.6708
-- Precision: 0.8169
-- Recall: 0.6696
-- F1: 0.6597
-- Precision Durariadorio 128x128: 1.0
-- Recall Durariadorio 128x128: 0.2569
-- F1 Durariadorio 128x128: 0.4088
-- Precision Mole 128x128: 0.9839
-- Recall Mole 128x128: 0.4236
-- F1 Mole 128x128: 0.5922
-- Precision Quebrado 128x128: 0.8633
-- Recall Quebrado 128x128: 0.8333
-- F1 Quebrado 128x128: 0.8481
-- Precision Riadorio 128x128: 0.4050
-- Recall Riadorio 128x128: 0.8553
-- F1 Riadorio 128x128: 0.5497
-- Precision Riofechado 128x128: 0.8323
-- Recall Riofechado 128x128: 0.9789
-- F1 Riofechado 128x128: 0.8997
+- Loss: 0.3384
+- Accuracy: 0.8939
+- Precision: 0.8983
+- Recall: 0.8949
+- F1: 0.8958
+- Precision Durariadorio 128x128: 0.9545
+- Recall Durariadorio 128x128: 0.875
+- F1 Durariadorio 128x128: 0.9130
+- Precision Mole 128x128: 0.9858
+- Recall Mole 128x128: 0.9653
+- F1 Mole 128x128: 0.9754
+- Precision Quebrado 128x128: 0.8063
+- Recall Quebrado 128x128: 0.8958
+- F1 Quebrado 128x128: 0.8487
+- Precision Riadorio 128x128: 0.8158
+- Recall Riadorio 128x128: 0.8158
+- F1 Riadorio 128x128: 0.8158
+- Precision Riofechado 128x128: 0.9291
+- Recall Riofechado 128x128: 0.9225
+- F1 Riofechado 128x128: 0.9258
 
 ## Model description
 

all_results.json

@@ -1,13 +1,31 @@
 {
-    "epoch": 50.0,
-    "eval_accuracy": 0.8705234159779615,
-    "eval_loss": 0.3891112208366394,
-    "eval_runtime": 1.9855,
-    "eval_samples_per_second": 365.644,
-    "eval_steps_per_second": 23.168,
-    "total_flos": 7.644201407152128e+17,
-    "train_loss": 0.40744263351618587,
-    "train_runtime": 1591.5969,
-    "train_samples_per_second": 182.553,
-    "train_steps_per_second": 11.435
+    "epoch": 100.0,
+    "eval_accuracy": 0.8939393939393939,
+    "eval_f1": 0.8957501623932316,
+    "eval_f1_DuraRiadoRio_128x128": 0.9130434782608695,
+    "eval_f1_Mole_128x128": 0.9754385964912281,
+    "eval_f1_Quebrado_128x128": 0.8486842105263158,
+    "eval_f1_RiadoRio_128x128": 0.8157894736842105,
+    "eval_f1_RioFechado_128x128": 0.9257950530035336,
+    "eval_loss": 0.3383881151676178,
+    "eval_precision": 0.8982957090501884,
+    "eval_precision_DuraRiadoRio_128x128": 0.9545454545454546,
+    "eval_precision_Mole_128x128": 0.9858156028368794,
+    "eval_precision_Quebrado_128x128": 0.80625,
+    "eval_precision_RiadoRio_128x128": 0.8157894736842105,
+    "eval_precision_RioFechado_128x128": 0.9290780141843972,
+    "eval_recall": 0.8948871592125854,
+    "eval_recall_DuraRiadoRio_128x128": 0.875,
+    "eval_recall_Mole_128x128": 0.9652777777777778,
+    "eval_recall_Quebrado_128x128": 0.8958333333333334,
+    "eval_recall_RiadoRio_128x128": 0.8157894736842105,
+    "eval_recall_RioFechado_128x128": 0.9225352112676056,
+    "eval_runtime": 2.1767,
+    "eval_samples_per_second": 333.527,
+    "eval_steps_per_second": 21.133,
+    "total_flos": 1.5288402814304256e+18,
+    "train_loss": 0.30421131687504904,
+    "train_runtime": 3220.4178,
+    "train_samples_per_second": 180.442,
+    "train_steps_per_second": 11.303
 }

eval_results.json

@@ -1,8 +1,26 @@
 {
-    "epoch": 50.0,
-    "eval_accuracy": 0.8705234159779615,
-    "eval_loss": 0.3891112208366394,
-    "eval_runtime": 1.9855,
-    "eval_samples_per_second": 365.644,
-    "eval_steps_per_second": 23.168
+    "epoch": 100.0,
+    "eval_accuracy": 0.8939393939393939,
+    "eval_f1": 0.8957501623932316,
+    "eval_f1_DuraRiadoRio_128x128": 0.9130434782608695,
+    "eval_f1_Mole_128x128": 0.9754385964912281,
+    "eval_f1_Quebrado_128x128": 0.8486842105263158,
+    "eval_f1_RiadoRio_128x128": 0.8157894736842105,
+    "eval_f1_RioFechado_128x128": 0.9257950530035336,
+    "eval_loss": 0.3383881151676178,
+    "eval_precision": 0.8982957090501884,
+    "eval_precision_DuraRiadoRio_128x128": 0.9545454545454546,
+    "eval_precision_Mole_128x128": 0.9858156028368794,
+    "eval_precision_Quebrado_128x128": 0.80625,
+    "eval_precision_RiadoRio_128x128": 0.8157894736842105,
+    "eval_precision_RioFechado_128x128": 0.9290780141843972,
+    "eval_recall": 0.8948871592125854,
+    "eval_recall_DuraRiadoRio_128x128": 0.875,
+    "eval_recall_Mole_128x128": 0.9652777777777778,
+    "eval_recall_Quebrado_128x128": 0.8958333333333334,
+    "eval_recall_RiadoRio_128x128": 0.8157894736842105,
+    "eval_recall_RioFechado_128x128": 0.9225352112676056,
+    "eval_runtime": 2.1767,
+    "eval_samples_per_second": 333.527,
+    "eval_steps_per_second": 21.133
 }

runs/Mar06_19-41-56_12dd6e624592/events.out.tfevents.1741293350.12dd6e624592.35890.1

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

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 50.0,
-    "total_flos": 7.644201407152128e+17,
-    "train_loss": 0.40744263351618587,
-    "train_runtime": 1591.5969,
-    "train_samples_per_second": 182.553,
-    "train_steps_per_second": 11.435
+    "epoch": 100.0,
+    "total_flos": 1.5288402814304256e+18,
+    "train_loss": 0.30421131687504904,
+    "train_runtime": 3220.4178,
+    "train_samples_per_second": 180.442,
+    "train_steps_per_second": 11.303
 }