End of training

README.md

@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 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 imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.6590
+- Loss: 1.7052
 - Accuracy: 0.2
 
 ## Model description

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.2,
+    "eval_loss": 1.7051823139190674,
+    "eval_runtime": 0.2027,
+    "eval_samples_per_second": 24.669,
+    "eval_steps_per_second": 4.934,
+    "total_flos": 120858889052160.0,
+    "train_loss": 0.39613640308380127,
+    "train_runtime": 15.9621,
+    "train_samples_per_second": 3.759,
+    "train_steps_per_second": 0.188
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.2,
+    "eval_loss": 1.7051823139190674,
+    "eval_runtime": 0.2027,
+    "eval_samples_per_second": 24.669,
+    "eval_steps_per_second": 4.934
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "total_flos": 120858889052160.0,
+    "train_loss": 0.39613640308380127,
+    "train_runtime": 15.9621,
+    "train_samples_per_second": 3.759,
+    "train_steps_per_second": 0.188
+}

trainer_state.json

@@ -0,0 +1,55 @@
+{
+  "best_metric": 0.2,
+  "best_model_checkpoint": "mobilenet_v2_1.0_224-finetuned-prueba/checkpoint-1",
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 3,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.2,
+      "eval_loss": 1.7051823139190674,
+      "eval_runtime": 0.2078,
+      "eval_samples_per_second": 24.063,
+      "eval_steps_per_second": 4.813,
+      "step": 1
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.2,
+      "eval_loss": 1.6696685552597046,
+      "eval_runtime": 0.2794,
+      "eval_samples_per_second": 17.898,
+      "eval_steps_per_second": 3.58,
+      "step": 2
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.2,
+      "eval_loss": 1.6589889526367188,
+      "eval_runtime": 0.2093,
+      "eval_samples_per_second": 23.894,
+      "eval_steps_per_second": 4.779,
+      "step": 3
+    },
+    {
+      "epoch": 3.0,
+      "step": 3,
+      "total_flos": 120858889052160.0,
+      "train_loss": 0.39613640308380127,
+      "train_runtime": 15.9621,
+      "train_samples_per_second": 3.759,
+      "train_steps_per_second": 0.188
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 3,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 120858889052160.0,
+  "trial_name": null,
+  "trial_params": null
+}