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": 100.0,
+    "epoch": 3.0,
     "eval_samples": 3000,
-    "exact_match": 35.0,
-    "f1": 47.40760446523187,
-    "train_loss": 0.3965238536672389,
-    "train_runtime": 2138.0517,
+    "exact_match": 64.2,
+    "f1": 74.31179260848076,
+    "train_loss": 2.188524670071072,
+    "train_runtime": 53.6084,
     "train_samples": 3000,
-    "train_samples_per_second": 140.315,
-    "train_steps_per_second": 4.397
+    "train_samples_per_second": 167.884,
+    "train_steps_per_second": 0.672
 }

args.bin

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

eval_nbest_predictions.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:193e91bbd5318fc8026a67f82b00acffe3890058b1f9f5f790485e349b7414d9
-size 13407354
+oid sha256:b82a97ed87fdc47a92d9f9cc868907de41476cd54d137c4eec77a44457560143
+size 13662949

eval_results.json

@@ -1,6 +1,6 @@
 {
-    "epoch": 100.0,
+    "epoch": 3.0,
     "eval_samples": 3000,
-    "exact_match": 35.0,
-    "f1": 47.40760446523187
+    "exact_match": 64.2,
+    "f1": 74.31179260848076
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 100.0,
-    "train_loss": 0.3965238536672389,
-    "train_runtime": 2138.0517,
+    "epoch": 3.0,
+    "train_loss": 2.188524670071072,
+    "train_runtime": 53.6084,
     "train_samples": 3000,
-    "train_samples_per_second": 140.315,
-    "train_steps_per_second": 4.397
+    "train_samples_per_second": 167.884,
+    "train_steps_per_second": 0.672
 }

trainer_state.json

@@ -1,136 +1,28 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 100.0,
+  "epoch": 3.0,
   "eval_steps": 500,
-  "global_step": 9400,
+  "global_step": 36,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 5.32,
-      "learning_rate": 4.734042553191489e-05,
-      "loss": 0.6879,
-      "step": 500
-    },
-    {
-      "epoch": 10.64,
-      "learning_rate": 4.468085106382979e-05,
-      "loss": 0.1213,
-      "step": 1000
-    },
-    {
-      "epoch": 15.96,
-      "learning_rate": 4.2021276595744684e-05,
-      "loss": 0.0743,
-      "step": 1500
-    },
-    {
-      "epoch": 21.28,
-      "learning_rate": 3.936170212765958e-05,
-      "loss": 0.0487,
-      "step": 2000
-    },
-    {
-      "epoch": 26.6,
-      "learning_rate": 3.670212765957447e-05,
-      "loss": 0.0615,
-      "step": 2500
-    },
-    {
-      "epoch": 31.91,
-      "learning_rate": 3.4042553191489365e-05,
-      "loss": 0.0804,
-      "step": 3000
-    },
-    {
-      "epoch": 37.23,
-      "learning_rate": 3.1382978723404254e-05,
-      "loss": 0.0936,
-      "step": 3500
-    },
-    {
-      "epoch": 42.55,
-      "learning_rate": 2.8723404255319154e-05,
-      "loss": 0.1289,
-      "step": 4000
-    },
-    {
-      "epoch": 47.87,
-      "learning_rate": 2.6063829787234046e-05,
-      "loss": 0.1842,
-      "step": 4500
-    },
-    {
-      "epoch": 53.19,
-      "learning_rate": 2.340425531914894e-05,
-      "loss": 0.2967,
-      "step": 5000
-    },
-    {
-      "epoch": 58.51,
-      "learning_rate": 2.074468085106383e-05,
-      "loss": 0.394,
-      "step": 5500
-    },
-    {
-      "epoch": 63.83,
-      "learning_rate": 1.8085106382978724e-05,
-      "loss": 0.6998,
-      "step": 6000
-    },
-    {
-      "epoch": 69.15,
-      "learning_rate": 1.5425531914893617e-05,
-      "loss": 0.7916,
-      "step": 6500
-    },
-    {
-      "epoch": 74.47,
-      "learning_rate": 1.2765957446808511e-05,
-      "loss": 0.7954,
-      "step": 7000
-    },
-    {
-      "epoch": 79.79,
-      "learning_rate": 1.0106382978723404e-05,
-      "loss": 0.8401,
-      "step": 7500
-    },
-    {
-      "epoch": 85.11,
-      "learning_rate": 7.446808510638298e-06,
-      "loss": 0.8077,
-      "step": 8000
-    },
-    {
-      "epoch": 90.43,
-      "learning_rate": 4.787234042553191e-06,
-      "loss": 0.6265,
-      "step": 8500
-    },
-    {
-      "epoch": 95.74,
-      "learning_rate": 2.1276595744680853e-06,
-      "loss": 0.511,
-      "step": 9000
-    },
-    {
-      "epoch": 100.0,
-      "step": 9400,
-      "total_flos": 4968855705600000.0,
-      "train_loss": 0.3965238536672389,
-      "train_runtime": 2138.0517,
-      "train_samples_per_second": 140.315,
-      "train_steps_per_second": 4.397
+      "epoch": 3.0,
+      "step": 36,
+      "total_flos": 149065671168000.0,
+      "train_loss": 2.188524670071072,
+      "train_runtime": 53.6084,
+      "train_samples_per_second": 167.884,
+      "train_steps_per_second": 0.672
     }
   ],
   "logging_steps": 500,
-  "max_steps": 9400,
-  "num_train_epochs": 100,
+  "max_steps": 36,
+  "num_train_epochs": 3,
   "save_steps": 500,
-  "total_flos": 4968855705600000.0,
+  "total_flos": 149065671168000.0,
   "trial_name": null,
   "trial_params": null
 }