End of training

README.md

@@ -3,6 +3,8 @@ library_name: transformers
 license: apache-2.0
 base_model: google/efficientnet-b0
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 metrics:
 - accuracy
@@ -16,10 +18,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # resnet-50-cocoa
 
-This model is a fine-tuned version of [google/efficientnet-b0](https://huggingface.co/google/efficientnet-b0) on an unknown dataset.
+This model is a fine-tuned version of [google/efficientnet-b0](https://huggingface.co/google/efficientnet-b0) on the SemilleroCV/Cocoa-dataset dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3624
-- Accuracy: 0.8809
+- Loss: 0.2657
+- Accuracy: 0.9097
 
 ## Model description
 

all_results.json

@@ -1,13 +1,13 @@
 {
-    "epoch": 5.0,
-    "eval_accuracy": 0.8628158844765343,
-    "eval_loss": 0.897123396396637,
-    "eval_runtime": 16.5935,
-    "eval_samples_per_second": 16.693,
-    "eval_steps_per_second": 2.109,
-    "total_flos": 1.6633116935737344e+17,
-    "train_loss": 1.1215976812401596,
-    "train_runtime": 1562.8364,
-    "train_samples_per_second": 5.01,
-    "train_steps_per_second": 0.627
+    "epoch": 100.0,
+    "eval_accuracy": 0.9097472924187726,
+    "eval_loss": 0.26566287875175476,
+    "eval_runtime": 0.9771,
+    "eval_samples_per_second": 283.48,
+    "eval_steps_per_second": 35.819,
+    "total_flos": 5.678990728814592e+17,
+    "train_loss": 0.30055975163317455,
+    "train_runtime": 951.6633,
+    "train_samples_per_second": 164.554,
+    "train_steps_per_second": 20.596
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 5.0,
-    "eval_accuracy": 0.8628158844765343,
-    "eval_loss": 0.897123396396637,
-    "eval_runtime": 16.5935,
-    "eval_samples_per_second": 16.693,
-    "eval_steps_per_second": 2.109
+    "epoch": 100.0,
+    "eval_accuracy": 0.9097472924187726,
+    "eval_loss": 0.26566287875175476,
+    "eval_runtime": 0.9771,
+    "eval_samples_per_second": 283.48,
+    "eval_steps_per_second": 35.819
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 5.0,
-    "total_flos": 1.6633116935737344e+17,
-    "train_loss": 1.1215976812401596,
-    "train_runtime": 1562.8364,
-    "train_samples_per_second": 5.01,
-    "train_steps_per_second": 0.627
+    "epoch": 100.0,
+    "total_flos": 5.678990728814592e+17,
+    "train_loss": 0.30055975163317455,
+    "train_runtime": 951.6633,
+    "train_samples_per_second": 164.554,
+    "train_steps_per_second": 20.596
 }