End of training

README.md

@@ -2,7 +2,7 @@
 tags:
 - generated_from_trainer
 datasets:
-- category_cleaning
+- Displate/category_cleaning
 model-index:
 - name: category_cleaning
   results: []
@@ -13,7 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # category_cleaning
 
-This model was trained from scratch on the category_cleaning dataset.
+This model was trained from scratch on the Displate/category_cleaning dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.9225
 
 ## Model description
 

all_results.json

@@ -1,11 +1,11 @@
 {
-    "epoch": 10.0,
-    "eval_loss": 0.9814479351043701,
-    "eval_runtime": 0.6923,
-    "eval_samples_per_second": 108.332,
-    "eval_steps_per_second": 2.889,
-    "train_loss": 1.109443473815918,
-    "train_runtime": 13.1048,
-    "train_samples_per_second": 57.231,
-    "train_steps_per_second": 1.526
+    "epoch": 20.0,
+    "eval_loss": 0.9225080609321594,
+    "eval_runtime": 0.6477,
+    "eval_samples_per_second": 115.8,
+    "eval_steps_per_second": 3.088,
+    "train_loss": 0.8449213981628418,
+    "train_runtime": 25.8102,
+    "train_samples_per_second": 58.117,
+    "train_steps_per_second": 1.55
 }

eval_results.json

@@ -1,7 +1,7 @@
 {
-    "epoch": 10.0,
-    "eval_loss": 0.9814479351043701,
-    "eval_runtime": 0.6923,
-    "eval_samples_per_second": 108.332,
-    "eval_steps_per_second": 2.889
+    "epoch": 20.0,
+    "eval_loss": 0.9225080609321594,
+    "eval_runtime": 0.6477,
+    "eval_samples_per_second": 115.8,
+    "eval_steps_per_second": 3.088
 }

runs/Mar08_11-24-40_9da45c1bcf23/events.out.tfevents.1709897158.9da45c1bcf23.33793.1

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

train_results.json

@@ -1,7 +1,7 @@
 {
-    "epoch": 10.0,
-    "train_loss": 1.109443473815918,
-    "train_runtime": 13.1048,
-    "train_samples_per_second": 57.231,
-    "train_steps_per_second": 1.526
+    "epoch": 20.0,
+    "train_loss": 0.8449213981628418,
+    "train_runtime": 25.8102,
+    "train_samples_per_second": 58.117,
+    "train_steps_per_second": 1.55
 }

trainer_state.json

@@ -1,29 +1,29 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 10.0,
+  "epoch": 20.0,
   "eval_steps": 500,
-  "global_step": 20,
+  "global_step": 40,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 10.0,
-      "step": 20,
-      "total_flos": 43625248474500.0,
-      "train_loss": 1.109443473815918,
-      "train_runtime": 13.1048,
-      "train_samples_per_second": 57.231,
-      "train_steps_per_second": 1.526
+      "epoch": 20.0,
+      "step": 40,
+      "total_flos": 87250496949000.0,
+      "train_loss": 0.8449213981628418,
+      "train_runtime": 25.8102,
+      "train_samples_per_second": 58.117,
+      "train_steps_per_second": 1.55
     }
   ],
   "logging_steps": 500,
-  "max_steps": 20,
+  "max_steps": 40,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 10,
+  "num_train_epochs": 20,
   "save_steps": 500,
-  "total_flos": 43625248474500.0,
+  "total_flos": 87250496949000.0,
   "train_batch_size": 64,
   "trial_name": null,
   "trial_params": null