End of training

README.md

@@ -2,7 +2,10 @@
 license: apache-2.0
 base_model: albert-base-v2
 tags:
+- question-answering
 - generated_from_trainer
+datasets:
+- squad
 model-index:
 - name: out
   results: []
@@ -13,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # out
 
-This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on the None dataset.
+This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on the squad dataset.
 
 ## Model description
 

all_results.json

@@ -1,11 +1,11 @@
 {
-    "epoch": 5.0,
+    "epoch": 10.0,
     "eval_samples": 3000,
-    "exact_match": 71.46666666666667,
-    "f1": 80.70165088524692,
-    "train_loss": 0.6540570847531583,
-    "train_runtime": 109.2106,
+    "exact_match": 68.83333333333333,
+    "f1": 79.17591999568613,
+    "train_loss": 0.3617458444960574,
+    "train_runtime": 216.5737,
     "train_samples": 3000,
-    "train_samples_per_second": 137.349,
-    "train_steps_per_second": 4.304
+    "train_samples_per_second": 138.521,
+    "train_steps_per_second": 4.34
 }

args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30bf497c8f1d3b0b97d8bda8f659df3c60f82e6b1c22399160ef4e9cb0ff45f3
+oid sha256:d10a59deee26bc53a4b1bfbcd4c175f77919dcdce208eedefe41f5a424d19b28
 size 5652

eval_nbest_predictions.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c80fa9b6a90ff70ce0275553d0d7596167d0287fdde2ff3fa1eb71c10daaf630
-size 13413679
+oid sha256:2dc323de3db5a627f1b03e8ebb643c4237fadecda10144a1f11f1fc0cccd6f57
+size 13413491

eval_results.json

@@ -1,6 +1,6 @@
 {
-    "epoch": 5.0,
+    "epoch": 10.0,
     "eval_samples": 3000,
-    "exact_match": 71.46666666666667,
-    "f1": 80.70165088524692
+    "exact_match": 68.83333333333333,
+    "f1": 79.17591999568613
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 5.0,
-    "train_loss": 0.6540570847531583,
-    "train_runtime": 109.2106,
+    "epoch": 10.0,
+    "train_loss": 0.3617458444960574,
+    "train_runtime": 216.5737,
     "train_samples": 3000,
-    "train_samples_per_second": 137.349,
-    "train_steps_per_second": 4.304
+    "train_samples_per_second": 138.521,
+    "train_steps_per_second": 4.34
 }

trainer_state.json

@@ -1,28 +1,34 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 5.0,
+  "epoch": 10.0,
   "eval_steps": 500,
-  "global_step": 470,
+  "global_step": 940,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 5.0,
-      "step": 470,
-      "total_flos": 248442785280000.0,
-      "train_loss": 0.6540570847531583,
-      "train_runtime": 109.2106,
-      "train_samples_per_second": 137.349,
-      "train_steps_per_second": 4.304
+      "epoch": 5.32,
+      "learning_rate": 2.340425531914894e-05,
+      "loss": 0.6488,
+      "step": 500
+    },
+    {
+      "epoch": 10.0,
+      "step": 940,
+      "total_flos": 496885570560000.0,
+      "train_loss": 0.3617458444960574,
+      "train_runtime": 216.5737,
+      "train_samples_per_second": 138.521,
+      "train_steps_per_second": 4.34
     }
   ],
   "logging_steps": 500,
-  "max_steps": 470,
-  "num_train_epochs": 5,
+  "max_steps": 940,
+  "num_train_epochs": 10,
   "save_steps": 500,
-  "total_flos": 248442785280000.0,
+  "total_flos": 496885570560000.0,
   "trial_name": null,
   "trial_params": null
 }