End of training

README.md

@@ -15,6 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
 # clip-finetuned
 
 This model is a fine-tuned version of [openai/clip-vit-base-patch32](https://huggingface.co/openai/clip-vit-base-patch32) on the imagefolder dataset.
+It achieves the following results on the evaluation set:
+- Loss: 5.6469
 
 ## Model description
 

all_results.json

@@ -1,15 +1,15 @@
 {
-    "epoch": 20.0,
-    "eval_loss": 4.885480880737305,
-    "eval_runtime": 2.1722,
-    "eval_samples_per_second": 19.335,
-    "eval_steps_per_second": 0.921,
-    "test_loss": 4.23183536529541,
-    "test_runtime": 0.9267,
-    "test_samples_per_second": 90.649,
-    "test_steps_per_second": 3.237,
-    "train_loss": 0.9488969976251775,
-    "train_runtime": 209.8213,
-    "train_samples_per_second": 32.313,
-    "train_steps_per_second": 1.049
+    "epoch": 30.0,
+    "eval_loss": 5.64693546295166,
+    "eval_runtime": 1.9784,
+    "eval_samples_per_second": 21.229,
+    "eval_steps_per_second": 1.011,
+    "test_loss": 5.721177577972412,
+    "test_runtime": 0.9455,
+    "test_samples_per_second": 88.843,
+    "test_steps_per_second": 3.173,
+    "train_loss": 0.7521804347182766,
+    "train_runtime": 313.3926,
+    "train_samples_per_second": 32.451,
+    "train_steps_per_second": 1.053
 }

eval_results.json

@@ -1,7 +1,7 @@
 {
-    "epoch": 20.0,
-    "eval_loss": 4.885480880737305,
-    "eval_runtime": 2.1722,
-    "eval_samples_per_second": 19.335,
-    "eval_steps_per_second": 0.921
+    "epoch": 30.0,
+    "eval_loss": 5.64693546295166,
+    "eval_runtime": 1.9784,
+    "eval_samples_per_second": 21.229,
+    "eval_steps_per_second": 1.011
 }

test_results.json

@@ -1,6 +1,6 @@
 {
-    "test_loss": 4.23183536529541,
-    "test_runtime": 0.9267,
-    "test_samples_per_second": 90.649,
-    "test_steps_per_second": 3.237
+    "test_loss": 5.721177577972412,
+    "test_runtime": 0.9455,
+    "test_samples_per_second": 88.843,
+    "test_steps_per_second": 3.173
 }

train_results.json

@@ -1,7 +1,7 @@
 {
-    "epoch": 20.0,
-    "train_loss": 0.9488969976251775,
-    "train_runtime": 209.8213,
-    "train_samples_per_second": 32.313,
-    "train_steps_per_second": 1.049
+    "epoch": 30.0,
+    "train_loss": 0.7521804347182766,
+    "train_runtime": 313.3926,
+    "train_samples_per_second": 32.451,
+    "train_steps_per_second": 1.053
 }

trainer_state.json

@@ -1,29 +1,29 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 20.0,
+  "epoch": 30.0,
   "eval_steps": 500,
-  "global_step": 220,
+  "global_step": 330,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 20.0,
-      "step": 220,
-      "total_flos": 394372246209480.0,
-      "train_loss": 0.9488969976251775,
-      "train_runtime": 209.8213,
-      "train_samples_per_second": 32.313,
-      "train_steps_per_second": 1.049
+      "epoch": 30.0,
+      "step": 330,
+      "total_flos": 591558369314220.0,
+      "train_loss": 0.7521804347182766,
+      "train_runtime": 313.3926,
+      "train_samples_per_second": 32.451,
+      "train_steps_per_second": 1.053
     }
   ],
   "logging_steps": 500,
-  "max_steps": 220,
+  "max_steps": 330,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 20,
+  "num_train_epochs": 30,
   "save_steps": 500,
-  "total_flos": 394372246209480.0,
+  "total_flos": 591558369314220.0,
   "train_batch_size": 32,
   "trial_name": null,
   "trial_params": null