Upload T5 spelling corrector checkpoint from local

added_tokens.json

@@ -0,0 +1,102 @@
+{
+  "<extra_id_0>": 32099,
+  "<extra_id_10>": 32089,
+  "<extra_id_11>": 32088,
+  "<extra_id_12>": 32087,
+  "<extra_id_13>": 32086,
+  "<extra_id_14>": 32085,
+  "<extra_id_15>": 32084,
+  "<extra_id_16>": 32083,
+  "<extra_id_17>": 32082,
+  "<extra_id_18>": 32081,
+  "<extra_id_19>": 32080,
+  "<extra_id_1>": 32098,
+  "<extra_id_20>": 32079,
+  "<extra_id_21>": 32078,
+  "<extra_id_22>": 32077,
+  "<extra_id_23>": 32076,
+  "<extra_id_24>": 32075,
+  "<extra_id_25>": 32074,
+  "<extra_id_26>": 32073,
+  "<extra_id_27>": 32072,
+  "<extra_id_28>": 32071,
+  "<extra_id_29>": 32070,
+  "<extra_id_2>": 32097,
+  "<extra_id_30>": 32069,
+  "<extra_id_31>": 32068,
+  "<extra_id_32>": 32067,
+  "<extra_id_33>": 32066,
+  "<extra_id_34>": 32065,
+  "<extra_id_35>": 32064,
+  "<extra_id_36>": 32063,
+  "<extra_id_37>": 32062,
+  "<extra_id_38>": 32061,
+  "<extra_id_39>": 32060,
+  "<extra_id_3>": 32096,
+  "<extra_id_40>": 32059,
+  "<extra_id_41>": 32058,
+  "<extra_id_42>": 32057,
+  "<extra_id_43>": 32056,
+  "<extra_id_44>": 32055,
+  "<extra_id_45>": 32054,
+  "<extra_id_46>": 32053,
+  "<extra_id_47>": 32052,
+  "<extra_id_48>": 32051,
+  "<extra_id_49>": 32050,
+  "<extra_id_4>": 32095,
+  "<extra_id_50>": 32049,
+  "<extra_id_51>": 32048,
+  "<extra_id_52>": 32047,
+  "<extra_id_53>": 32046,
+  "<extra_id_54>": 32045,
+  "<extra_id_55>": 32044,
+  "<extra_id_56>": 32043,
+  "<extra_id_57>": 32042,
+  "<extra_id_58>": 32041,
+  "<extra_id_59>": 32040,
+  "<extra_id_5>": 32094,
+  "<extra_id_60>": 32039,
+  "<extra_id_61>": 32038,
+  "<extra_id_62>": 32037,
+  "<extra_id_63>": 32036,
+  "<extra_id_64>": 32035,
+  "<extra_id_65>": 32034,
+  "<extra_id_66>": 32033,
+  "<extra_id_67>": 32032,
+  "<extra_id_68>": 32031,
+  "<extra_id_69>": 32030,
+  "<extra_id_6>": 32093,
+  "<extra_id_70>": 32029,
+  "<extra_id_71>": 32028,
+  "<extra_id_72>": 32027,
+  "<extra_id_73>": 32026,
+  "<extra_id_74>": 32025,
+  "<extra_id_75>": 32024,
+  "<extra_id_76>": 32023,
+  "<extra_id_77>": 32022,
+  "<extra_id_78>": 32021,
+  "<extra_id_79>": 32020,
+  "<extra_id_7>": 32092,
+  "<extra_id_80>": 32019,
+  "<extra_id_81>": 32018,
+  "<extra_id_82>": 32017,
+  "<extra_id_83>": 32016,
+  "<extra_id_84>": 32015,
+  "<extra_id_85>": 32014,
+  "<extra_id_86>": 32013,
+  "<extra_id_87>": 32012,
+  "<extra_id_88>": 32011,
+  "<extra_id_89>": 32010,
+  "<extra_id_8>": 32091,
+  "<extra_id_90>": 32009,
+  "<extra_id_91>": 32008,
+  "<extra_id_92>": 32007,
+  "<extra_id_93>": 32006,
+  "<extra_id_94>": 32005,
+  "<extra_id_95>": 32004,
+  "<extra_id_96>": 32003,
+  "<extra_id_97>": 32002,
+  "<extra_id_98>": 32001,
+  "<extra_id_99>": 32000,
+  "<extra_id_9>": 32090
+}

config.json

@@ -0,0 +1,60 @@
+{
+  "architectures": [
+    "T5ForConditionalGeneration"
+  ],
+  "classifier_dropout": 0.0,
+  "d_ff": 3072,
+  "d_kv": 64,
+  "d_model": 768,
+  "decoder_start_token_id": 0,
+  "dense_act_fn": "relu",
+  "dropout_rate": 0.1,
+  "dtype": "float32",
+  "eos_token_id": 1,
+  "feed_forward_proj": "relu",
+  "initializer_factor": 1.0,
+  "is_encoder_decoder": true,
+  "is_gated_act": false,
+  "layer_norm_epsilon": 1e-06,
+  "model_type": "t5",
+  "n_positions": 512,
+  "num_decoder_layers": 12,
+  "num_heads": 12,
+  "num_layers": 12,
+  "output_past": true,
+  "pad_token_id": 0,
+  "relative_attention_max_distance": 128,
+  "relative_attention_num_buckets": 32,
+  "task_specific_params": {
+    "summarization": {
+      "early_stopping": true,
+      "length_penalty": 2.0,
+      "max_length": 200,
+      "min_length": 30,
+      "no_repeat_ngram_size": 3,
+      "num_beams": 4,
+      "prefix": "summarize: "
+    },
+    "translation_en_to_de": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to German: "
+    },
+    "translation_en_to_fr": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to French: "
+    },
+    "translation_en_to_ro": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to Romanian: "
+    }
+  },
+  "transformers_version": "4.56.2",
+  "use_cache": true,
+  "vocab_size": 32128
+}

generation_config.json

@@ -0,0 +1,8 @@
+{
+  "decoder_start_token_id": 0,
+  "eos_token_id": [
+    1
+  ],
+  "pad_token_id": 0,
+  "transformers_version": "4.56.2"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c8aa3ef4c7676976682613f4ba8e3ea5d6eab4422dc760c6b1816196100fb56
+size 891644712

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74810134dfa66422735b506f52eb0ac6755641d6512c0c9c5e2cecc52d6dd755
+size 1783448843

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27b327dabfb282f32c20a8cc969f7757e3fd968319bbdd4a51288cc959104b94
+size 14645

scaler.pt

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

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8128bd31b2985a3c95d0dc3e72c07aac0ce98ee14f3794ea2b5cfcb6a19689f5
+size 1465

special_tokens_map.json

@@ -0,0 +1,125 @@
+{
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

spiece.model

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

tokenizer_config.json

@@ -0,0 +1,941 @@
+{
+  "add_prefix_space": true,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "extra_ids": 100,
+  "extra_special_tokens": {},
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}

trainer_state.json

@@ -0,0 +1,3729 @@
+{
+  "best_global_step": 5000,
+  "best_metric": 0.992226148409894,
+  "best_model_checkpoint": "./models/t5_spelling_corrector_finetuned_v3/checkpoint-5000",
+  "epoch": 50.0,
+  "eval_steps": 1000,
+  "global_step": 5200,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.09615384615384616,
+      "grad_norm": 3.347588539123535,
+      "learning_rate": 5.4e-07,
+      "loss": 0.6699,
+      "step": 10
+    },
+    {
+      "epoch": 0.19230769230769232,
+      "grad_norm": 2.8302624225616455,
+      "learning_rate": 1.14e-06,
+      "loss": 0.5993,
+      "step": 20
+    },
+    {
+      "epoch": 0.28846153846153844,
+      "grad_norm": 3.3392577171325684,
+      "learning_rate": 1.74e-06,
+      "loss": 0.564,
+      "step": 30
+    },
+    {
+      "epoch": 0.38461538461538464,
+      "grad_norm": 3.260563611984253,
+      "learning_rate": 2.34e-06,
+      "loss": 0.5126,
+      "step": 40
+    },
+    {
+      "epoch": 0.4807692307692308,
+      "grad_norm": 2.92582631111145,
+      "learning_rate": 2.9400000000000002e-06,
+      "loss": 0.4923,
+      "step": 50
+    },
+    {
+      "epoch": 0.5769230769230769,
+      "grad_norm": 2.35099720954895,
+      "learning_rate": 3.54e-06,
+      "loss": 0.4266,
+      "step": 60
+    },
+    {
+      "epoch": 0.6730769230769231,
+      "grad_norm": 2.393505334854126,
+      "learning_rate": 4.14e-06,
+      "loss": 0.3655,
+      "step": 70
+    },
+    {
+      "epoch": 0.7692307692307693,
+      "grad_norm": 1.5848568677902222,
+      "learning_rate": 4.74e-06,
+      "loss": 0.3313,
+      "step": 80
+    },
+    {
+      "epoch": 0.8653846153846154,
+      "grad_norm": 1.625791072845459,
+      "learning_rate": 5.34e-06,
+      "loss": 0.2682,
+      "step": 90
+    },
+    {
+      "epoch": 0.9615384615384616,
+      "grad_norm": 1.9399585723876953,
+      "learning_rate": 5.940000000000001e-06,
+      "loss": 0.2452,
+      "step": 100
+    },
+    {
+      "epoch": 1.0576923076923077,
+      "grad_norm": 1.909173607826233,
+      "learning_rate": 6.54e-06,
+      "loss": 0.2577,
+      "step": 110
+    },
+    {
+      "epoch": 1.1538461538461537,
+      "grad_norm": 1.0260469913482666,
+      "learning_rate": 7.14e-06,
+      "loss": 0.1986,
+      "step": 120
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 1.2334129810333252,
+      "learning_rate": 7.74e-06,
+      "loss": 0.1966,
+      "step": 130
+    },
+    {
+      "epoch": 1.3461538461538463,
+      "grad_norm": 1.1841124296188354,
+      "learning_rate": 8.340000000000001e-06,
+      "loss": 0.1764,
+      "step": 140
+    },
+    {
+      "epoch": 1.4423076923076923,
+      "grad_norm": 1.069820523262024,
+      "learning_rate": 8.939999999999999e-06,
+      "loss": 0.172,
+      "step": 150
+    },
+    {
+      "epoch": 1.5384615384615383,
+      "grad_norm": 0.9867880344390869,
+      "learning_rate": 9.54e-06,
+      "loss": 0.1585,
+      "step": 160
+    },
+    {
+      "epoch": 1.6346153846153846,
+      "grad_norm": 1.1286975145339966,
+      "learning_rate": 1.0140000000000001e-05,
+      "loss": 0.1385,
+      "step": 170
+    },
+    {
+      "epoch": 1.7307692307692308,
+      "grad_norm": 1.0937390327453613,
+      "learning_rate": 1.074e-05,
+      "loss": 0.1412,
+      "step": 180
+    },
+    {
+      "epoch": 1.8269230769230769,
+      "grad_norm": 1.139171838760376,
+      "learning_rate": 1.134e-05,
+      "loss": 0.1217,
+      "step": 190
+    },
+    {
+      "epoch": 1.9230769230769231,
+      "grad_norm": 0.9474167227745056,
+      "learning_rate": 1.1940000000000001e-05,
+      "loss": 0.13,
+      "step": 200
+    },
+    {
+      "epoch": 2.019230769230769,
+      "grad_norm": 0.8131914138793945,
+      "learning_rate": 1.254e-05,
+      "loss": 0.1128,
+      "step": 210
+    },
+    {
+      "epoch": 2.1153846153846154,
+      "grad_norm": 1.041396975517273,
+      "learning_rate": 1.314e-05,
+      "loss": 0.1158,
+      "step": 220
+    },
+    {
+      "epoch": 2.2115384615384617,
+      "grad_norm": 0.9056050777435303,
+      "learning_rate": 1.374e-05,
+      "loss": 0.1131,
+      "step": 230
+    },
+    {
+      "epoch": 2.3076923076923075,
+      "grad_norm": 0.8580456972122192,
+      "learning_rate": 1.434e-05,
+      "loss": 0.1107,
+      "step": 240
+    },
+    {
+      "epoch": 2.4038461538461537,
+      "grad_norm": 1.473669409751892,
+      "learning_rate": 1.4940000000000001e-05,
+      "loss": 0.1127,
+      "step": 250
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 0.9101526141166687,
+      "learning_rate": 1.554e-05,
+      "loss": 0.1022,
+      "step": 260
+    },
+    {
+      "epoch": 2.5961538461538463,
+      "grad_norm": 0.8412496447563171,
+      "learning_rate": 1.614e-05,
+      "loss": 0.1023,
+      "step": 270
+    },
+    {
+      "epoch": 2.6923076923076925,
+      "grad_norm": 1.4617328643798828,
+      "learning_rate": 1.6740000000000002e-05,
+      "loss": 0.0978,
+      "step": 280
+    },
+    {
+      "epoch": 2.7884615384615383,
+      "grad_norm": 1.095248818397522,
+      "learning_rate": 1.734e-05,
+      "loss": 0.0946,
+      "step": 290
+    },
+    {
+      "epoch": 2.8846153846153846,
+      "grad_norm": 0.5798925757408142,
+      "learning_rate": 1.794e-05,
+      "loss": 0.0916,
+      "step": 300
+    },
+    {
+      "epoch": 2.980769230769231,
+      "grad_norm": 0.708819568157196,
+      "learning_rate": 1.854e-05,
+      "loss": 0.0866,
+      "step": 310
+    },
+    {
+      "epoch": 3.076923076923077,
+      "grad_norm": 0.47592493891716003,
+      "learning_rate": 1.914e-05,
+      "loss": 0.0742,
+      "step": 320
+    },
+    {
+      "epoch": 3.173076923076923,
+      "grad_norm": 1.1844018697738647,
+      "learning_rate": 1.974e-05,
+      "loss": 0.0974,
+      "step": 330
+    },
+    {
+      "epoch": 3.269230769230769,
+      "grad_norm": 0.6659021973609924,
+      "learning_rate": 2.0340000000000002e-05,
+      "loss": 0.071,
+      "step": 340
+    },
+    {
+      "epoch": 3.3653846153846154,
+      "grad_norm": 0.8126431107521057,
+      "learning_rate": 2.094e-05,
+      "loss": 0.0718,
+      "step": 350
+    },
+    {
+      "epoch": 3.4615384615384617,
+      "grad_norm": 0.6788111925125122,
+      "learning_rate": 2.154e-05,
+      "loss": 0.0801,
+      "step": 360
+    },
+    {
+      "epoch": 3.5576923076923075,
+      "grad_norm": 0.8465282917022705,
+      "learning_rate": 2.214e-05,
+      "loss": 0.0696,
+      "step": 370
+    },
+    {
+      "epoch": 3.6538461538461537,
+      "grad_norm": 0.6793688535690308,
+      "learning_rate": 2.274e-05,
+      "loss": 0.0806,
+      "step": 380
+    },
+    {
+      "epoch": 3.75,
+      "grad_norm": 0.7204192876815796,
+      "learning_rate": 2.334e-05,
+      "loss": 0.0604,
+      "step": 390
+    },
+    {
+      "epoch": 3.8461538461538463,
+      "grad_norm": 0.5197739601135254,
+      "learning_rate": 2.394e-05,
+      "loss": 0.0696,
+      "step": 400
+    },
+    {
+      "epoch": 3.9423076923076925,
+      "grad_norm": 0.5837198495864868,
+      "learning_rate": 2.454e-05,
+      "loss": 0.0675,
+      "step": 410
+    },
+    {
+      "epoch": 4.038461538461538,
+      "grad_norm": 0.6246194243431091,
+      "learning_rate": 2.514e-05,
+      "loss": 0.0585,
+      "step": 420
+    },
+    {
+      "epoch": 4.134615384615385,
+      "grad_norm": 0.8931935429573059,
+      "learning_rate": 2.574e-05,
+      "loss": 0.0626,
+      "step": 430
+    },
+    {
+      "epoch": 4.230769230769231,
+      "grad_norm": 0.8182059526443481,
+      "learning_rate": 2.6340000000000002e-05,
+      "loss": 0.0574,
+      "step": 440
+    },
+    {
+      "epoch": 4.326923076923077,
+      "grad_norm": 0.6059229373931885,
+      "learning_rate": 2.6940000000000003e-05,
+      "loss": 0.0571,
+      "step": 450
+    },
+    {
+      "epoch": 4.423076923076923,
+      "grad_norm": 0.7983167171478271,
+      "learning_rate": 2.754e-05,
+      "loss": 0.051,
+      "step": 460
+    },
+    {
+      "epoch": 4.519230769230769,
+      "grad_norm": 1.0164097547531128,
+      "learning_rate": 2.8139999999999998e-05,
+      "loss": 0.0487,
+      "step": 470
+    },
+    {
+      "epoch": 4.615384615384615,
+      "grad_norm": 0.7057549357414246,
+      "learning_rate": 2.874e-05,
+      "loss": 0.0587,
+      "step": 480
+    },
+    {
+      "epoch": 4.711538461538462,
+      "grad_norm": 1.007047176361084,
+      "learning_rate": 2.934e-05,
+      "loss": 0.0545,
+      "step": 490
+    },
+    {
+      "epoch": 4.8076923076923075,
+      "grad_norm": 0.9160193204879761,
+      "learning_rate": 2.994e-05,
+      "loss": 0.05,
+      "step": 500
+    },
+    {
+      "epoch": 4.903846153846154,
+      "grad_norm": 0.9796598553657532,
+      "learning_rate": 2.9942553191489364e-05,
+      "loss": 0.0556,
+      "step": 510
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 1.1414951086044312,
+      "learning_rate": 2.987872340425532e-05,
+      "loss": 0.05,
+      "step": 520
+    },
+    {
+      "epoch": 5.096153846153846,
+      "grad_norm": 0.4623892605304718,
+      "learning_rate": 2.9814893617021278e-05,
+      "loss": 0.0502,
+      "step": 530
+    },
+    {
+      "epoch": 5.1923076923076925,
+      "grad_norm": 0.6435359120368958,
+      "learning_rate": 2.9751063829787234e-05,
+      "loss": 0.045,
+      "step": 540
+    },
+    {
+      "epoch": 5.288461538461538,
+      "grad_norm": 1.070151925086975,
+      "learning_rate": 2.9687234042553193e-05,
+      "loss": 0.0499,
+      "step": 550
+    },
+    {
+      "epoch": 5.384615384615385,
+      "grad_norm": 0.5936897993087769,
+      "learning_rate": 2.962340425531915e-05,
+      "loss": 0.0519,
+      "step": 560
+    },
+    {
+      "epoch": 5.480769230769231,
+      "grad_norm": 0.47224923968315125,
+      "learning_rate": 2.9559574468085107e-05,
+      "loss": 0.0485,
+      "step": 570
+    },
+    {
+      "epoch": 5.576923076923077,
+      "grad_norm": 0.8727656602859497,
+      "learning_rate": 2.9495744680851063e-05,
+      "loss": 0.043,
+      "step": 580
+    },
+    {
+      "epoch": 5.673076923076923,
+      "grad_norm": 1.1207246780395508,
+      "learning_rate": 2.9431914893617022e-05,
+      "loss": 0.0408,
+      "step": 590
+    },
+    {
+      "epoch": 5.769230769230769,
+      "grad_norm": 0.4979076683521271,
+      "learning_rate": 2.9368085106382978e-05,
+      "loss": 0.0419,
+      "step": 600
+    },
+    {
+      "epoch": 5.865384615384615,
+      "grad_norm": 0.6043898463249207,
+      "learning_rate": 2.9304255319148937e-05,
+      "loss": 0.0445,
+      "step": 610
+    },
+    {
+      "epoch": 5.961538461538462,
+      "grad_norm": 0.8977653384208679,
+      "learning_rate": 2.9240425531914892e-05,
+      "loss": 0.0456,
+      "step": 620
+    },
+    {
+      "epoch": 6.0576923076923075,
+      "grad_norm": 0.44534873962402344,
+      "learning_rate": 2.917659574468085e-05,
+      "loss": 0.0368,
+      "step": 630
+    },
+    {
+      "epoch": 6.153846153846154,
+      "grad_norm": 1.2936347723007202,
+      "learning_rate": 2.911276595744681e-05,
+      "loss": 0.0373,
+      "step": 640
+    },
+    {
+      "epoch": 6.25,
+      "grad_norm": 0.6867542862892151,
+      "learning_rate": 2.9048936170212766e-05,
+      "loss": 0.0355,
+      "step": 650
+    },
+    {
+      "epoch": 6.346153846153846,
+      "grad_norm": 0.8258110880851746,
+      "learning_rate": 2.8985106382978725e-05,
+      "loss": 0.0389,
+      "step": 660
+    },
+    {
+      "epoch": 6.4423076923076925,
+      "grad_norm": 0.42925509810447693,
+      "learning_rate": 2.892127659574468e-05,
+      "loss": 0.0353,
+      "step": 670
+    },
+    {
+      "epoch": 6.538461538461538,
+      "grad_norm": 0.9368314146995544,
+      "learning_rate": 2.885744680851064e-05,
+      "loss": 0.0378,
+      "step": 680
+    },
+    {
+      "epoch": 6.634615384615385,
+      "grad_norm": 0.5876572132110596,
+      "learning_rate": 2.8793617021276595e-05,
+      "loss": 0.0329,
+      "step": 690
+    },
+    {
+      "epoch": 6.730769230769231,
+      "grad_norm": 0.6402647495269775,
+      "learning_rate": 2.8729787234042554e-05,
+      "loss": 0.027,
+      "step": 700
+    },
+    {
+      "epoch": 6.826923076923077,
+      "grad_norm": 0.5723885297775269,
+      "learning_rate": 2.866595744680851e-05,
+      "loss": 0.0327,
+      "step": 710
+    },
+    {
+      "epoch": 6.923076923076923,
+      "grad_norm": 0.4314801096916199,
+      "learning_rate": 2.860212765957447e-05,
+      "loss": 0.0331,
+      "step": 720
+    },
+    {
+      "epoch": 7.019230769230769,
+      "grad_norm": 0.5018507838249207,
+      "learning_rate": 2.8538297872340424e-05,
+      "loss": 0.0351,
+      "step": 730
+    },
+    {
+      "epoch": 7.115384615384615,
+      "grad_norm": 0.49081480503082275,
+      "learning_rate": 2.8474468085106383e-05,
+      "loss": 0.0308,
+      "step": 740
+    },
+    {
+      "epoch": 7.211538461538462,
+      "grad_norm": 0.7369358539581299,
+      "learning_rate": 2.841063829787234e-05,
+      "loss": 0.0276,
+      "step": 750
+    },
+    {
+      "epoch": 7.3076923076923075,
+      "grad_norm": 0.49376147985458374,
+      "learning_rate": 2.8346808510638298e-05,
+      "loss": 0.0323,
+      "step": 760
+    },
+    {
+      "epoch": 7.403846153846154,
+      "grad_norm": 0.7800844311714172,
+      "learning_rate": 2.8282978723404253e-05,
+      "loss": 0.0287,
+      "step": 770
+    },
+    {
+      "epoch": 7.5,
+      "grad_norm": 0.9613171815872192,
+      "learning_rate": 2.8219148936170212e-05,
+      "loss": 0.0279,
+      "step": 780
+    },
+    {
+      "epoch": 7.596153846153846,
+      "grad_norm": 1.0874011516571045,
+      "learning_rate": 2.8155319148936168e-05,
+      "loss": 0.0371,
+      "step": 790
+    },
+    {
+      "epoch": 7.6923076923076925,
+      "grad_norm": 0.3330044746398926,
+      "learning_rate": 2.8091489361702127e-05,
+      "loss": 0.0283,
+      "step": 800
+    },
+    {
+      "epoch": 7.788461538461538,
+      "grad_norm": 1.4728511571884155,
+      "learning_rate": 2.8027659574468086e-05,
+      "loss": 0.0325,
+      "step": 810
+    },
+    {
+      "epoch": 7.884615384615385,
+      "grad_norm": 1.2733508348464966,
+      "learning_rate": 2.796382978723404e-05,
+      "loss": 0.0257,
+      "step": 820
+    },
+    {
+      "epoch": 7.980769230769231,
+      "grad_norm": 0.6438696980476379,
+      "learning_rate": 2.79e-05,
+      "loss": 0.0312,
+      "step": 830
+    },
+    {
+      "epoch": 8.076923076923077,
+      "grad_norm": 0.6629941463470459,
+      "learning_rate": 2.783617021276596e-05,
+      "loss": 0.0298,
+      "step": 840
+    },
+    {
+      "epoch": 8.173076923076923,
+      "grad_norm": 0.3993019461631775,
+      "learning_rate": 2.777234042553192e-05,
+      "loss": 0.0325,
+      "step": 850
+    },
+    {
+      "epoch": 8.26923076923077,
+      "grad_norm": 0.6544252634048462,
+      "learning_rate": 2.7708510638297874e-05,
+      "loss": 0.0271,
+      "step": 860
+    },
+    {
+      "epoch": 8.365384615384615,
+      "grad_norm": 0.581778883934021,
+      "learning_rate": 2.7644680851063833e-05,
+      "loss": 0.0287,
+      "step": 870
+    },
+    {
+      "epoch": 8.461538461538462,
+      "grad_norm": 0.7206617593765259,
+      "learning_rate": 2.758085106382979e-05,
+      "loss": 0.0243,
+      "step": 880
+    },
+    {
+      "epoch": 8.557692307692308,
+      "grad_norm": 0.7053912878036499,
+      "learning_rate": 2.7517021276595748e-05,
+      "loss": 0.0257,
+      "step": 890
+    },
+    {
+      "epoch": 8.653846153846153,
+      "grad_norm": 0.9742723107337952,
+      "learning_rate": 2.7453191489361703e-05,
+      "loss": 0.0256,
+      "step": 900
+    },
+    {
+      "epoch": 8.75,
+      "grad_norm": 1.0441752672195435,
+      "learning_rate": 2.7389361702127662e-05,
+      "loss": 0.0273,
+      "step": 910
+    },
+    {
+      "epoch": 8.846153846153847,
+      "grad_norm": 0.571890115737915,
+      "learning_rate": 2.7325531914893618e-05,
+      "loss": 0.0218,
+      "step": 920
+    },
+    {
+      "epoch": 8.942307692307692,
+      "grad_norm": 0.6239679455757141,
+      "learning_rate": 2.7261702127659577e-05,
+      "loss": 0.0266,
+      "step": 930
+    },
+    {
+      "epoch": 9.038461538461538,
+      "grad_norm": 0.6254406571388245,
+      "learning_rate": 2.7197872340425533e-05,
+      "loss": 0.0207,
+      "step": 940
+    },
+    {
+      "epoch": 9.134615384615385,
+      "grad_norm": 0.4847211241722107,
+      "learning_rate": 2.713404255319149e-05,
+      "loss": 0.0285,
+      "step": 950
+    },
+    {
+      "epoch": 9.23076923076923,
+      "grad_norm": 0.7274488806724548,
+      "learning_rate": 2.7070212765957447e-05,
+      "loss": 0.0233,
+      "step": 960
+    },
+    {
+      "epoch": 9.326923076923077,
+      "grad_norm": 0.5963655710220337,
+      "learning_rate": 2.7006382978723406e-05,
+      "loss": 0.023,
+      "step": 970
+    },
+    {
+      "epoch": 9.423076923076923,
+      "grad_norm": 0.4576031267642975,
+      "learning_rate": 2.6942553191489362e-05,
+      "loss": 0.0241,
+      "step": 980
+    },
+    {
+      "epoch": 9.51923076923077,
+      "grad_norm": 0.38139963150024414,
+      "learning_rate": 2.687872340425532e-05,
+      "loss": 0.0251,
+      "step": 990
+    },
+    {
+      "epoch": 9.615384615384615,
+      "grad_norm": 0.4351661801338196,
+      "learning_rate": 2.681489361702128e-05,
+      "loss": 0.0288,
+      "step": 1000
+    },
+    {
+      "epoch": 9.615384615384615,
+      "eval_correct_predictions": 1324,
+      "eval_exact_match_accuracy": 0.9356890459363958,
+      "eval_loss": 0.018085243180394173,
+      "eval_runtime": 67.218,
+      "eval_samples_per_second": 21.051,
+      "eval_steps_per_second": 0.342,
+      "eval_total_predictions": 1415,
+      "step": 1000
+    },
+    {
+      "epoch": 9.711538461538462,
+      "grad_norm": 1.3073899745941162,
+      "learning_rate": 2.6751063829787235e-05,
+      "loss": 0.0228,
+      "step": 1010
+    },
+    {
+      "epoch": 9.807692307692308,
+      "grad_norm": 0.844657838344574,
+      "learning_rate": 2.6687234042553194e-05,
+      "loss": 0.0224,
+      "step": 1020
+    },
+    {
+      "epoch": 9.903846153846153,
+      "grad_norm": 0.7185432314872742,
+      "learning_rate": 2.662340425531915e-05,
+      "loss": 0.0196,
+      "step": 1030
+    },
+    {
+      "epoch": 10.0,
+      "grad_norm": 1.6467353105545044,
+      "learning_rate": 2.655957446808511e-05,
+      "loss": 0.0223,
+      "step": 1040
+    },
+    {
+      "epoch": 10.096153846153847,
+      "grad_norm": 0.581244945526123,
+      "learning_rate": 2.6495744680851065e-05,
+      "loss": 0.0212,
+      "step": 1050
+    },
+    {
+      "epoch": 10.192307692307692,
+      "grad_norm": 0.8405095338821411,
+      "learning_rate": 2.6431914893617024e-05,
+      "loss": 0.0247,
+      "step": 1060
+    },
+    {
+      "epoch": 10.288461538461538,
+      "grad_norm": 0.7707756757736206,
+      "learning_rate": 2.636808510638298e-05,
+      "loss": 0.0183,
+      "step": 1070
+    },
+    {
+      "epoch": 10.384615384615385,
+      "grad_norm": 0.5692296624183655,
+      "learning_rate": 2.6304255319148938e-05,
+      "loss": 0.0221,
+      "step": 1080
+    },
+    {
+      "epoch": 10.48076923076923,
+      "grad_norm": 0.2527153789997101,
+      "learning_rate": 2.6240425531914894e-05,
+      "loss": 0.0228,
+      "step": 1090
+    },
+    {
+      "epoch": 10.576923076923077,
+      "grad_norm": 0.6079349517822266,
+      "learning_rate": 2.6176595744680853e-05,
+      "loss": 0.0223,
+      "step": 1100
+    },
+    {
+      "epoch": 10.673076923076923,
+      "grad_norm": 0.43173831701278687,
+      "learning_rate": 2.611276595744681e-05,
+      "loss": 0.017,
+      "step": 1110
+    },
+    {
+      "epoch": 10.76923076923077,
+      "grad_norm": 0.5763304829597473,
+      "learning_rate": 2.6048936170212767e-05,
+      "loss": 0.0183,
+      "step": 1120
+    },
+    {
+      "epoch": 10.865384615384615,
+      "grad_norm": 0.5463092923164368,
+      "learning_rate": 2.5985106382978723e-05,
+      "loss": 0.0197,
+      "step": 1130
+    },
+    {
+      "epoch": 10.961538461538462,
+      "grad_norm": 1.0987874269485474,
+      "learning_rate": 2.5921276595744682e-05,
+      "loss": 0.0201,
+      "step": 1140
+    },
+    {
+      "epoch": 11.057692307692308,
+      "grad_norm": 0.25165650248527527,
+      "learning_rate": 2.5857446808510638e-05,
+      "loss": 0.0154,
+      "step": 1150
+    },
+    {
+      "epoch": 11.153846153846153,
+      "grad_norm": 0.8446175456047058,
+      "learning_rate": 2.5793617021276597e-05,
+      "loss": 0.0189,
+      "step": 1160
+    },
+    {
+      "epoch": 11.25,
+      "grad_norm": 0.597083568572998,
+      "learning_rate": 2.5729787234042556e-05,
+      "loss": 0.0169,
+      "step": 1170
+    },
+    {
+      "epoch": 11.346153846153847,
+      "grad_norm": 0.5754260420799255,
+      "learning_rate": 2.566595744680851e-05,
+      "loss": 0.0179,
+      "step": 1180
+    },
+    {
+      "epoch": 11.442307692307692,
+      "grad_norm": 0.3986382782459259,
+      "learning_rate": 2.560212765957447e-05,
+      "loss": 0.0187,
+      "step": 1190
+    },
+    {
+      "epoch": 11.538461538461538,
+      "grad_norm": 0.649178147315979,
+      "learning_rate": 2.5538297872340426e-05,
+      "loss": 0.019,
+      "step": 1200
+    },
+    {
+      "epoch": 11.634615384615385,
+      "grad_norm": 0.5855590105056763,
+      "learning_rate": 2.5474468085106385e-05,
+      "loss": 0.0179,
+      "step": 1210
+    },
+    {
+      "epoch": 11.73076923076923,
+      "grad_norm": 0.7554678916931152,
+      "learning_rate": 2.541063829787234e-05,
+      "loss": 0.018,
+      "step": 1220
+    },
+    {
+      "epoch": 11.826923076923077,
+      "grad_norm": 0.417162150144577,
+      "learning_rate": 2.53468085106383e-05,
+      "loss": 0.0244,
+      "step": 1230
+    },
+    {
+      "epoch": 11.923076923076923,
+      "grad_norm": 1.6350237131118774,
+      "learning_rate": 2.5282978723404255e-05,
+      "loss": 0.0207,
+      "step": 1240
+    },
+    {
+      "epoch": 12.01923076923077,
+      "grad_norm": 0.41877123713493347,
+      "learning_rate": 2.5219148936170214e-05,
+      "loss": 0.019,
+      "step": 1250
+    },
+    {
+      "epoch": 12.115384615384615,
+      "grad_norm": 0.48958903551101685,
+      "learning_rate": 2.515531914893617e-05,
+      "loss": 0.0172,
+      "step": 1260
+    },
+    {
+      "epoch": 12.211538461538462,
+      "grad_norm": 0.8494559526443481,
+      "learning_rate": 2.509148936170213e-05,
+      "loss": 0.0177,
+      "step": 1270
+    },
+    {
+      "epoch": 12.307692307692308,
+      "grad_norm": 0.31908470392227173,
+      "learning_rate": 2.5027659574468084e-05,
+      "loss": 0.0145,
+      "step": 1280
+    },
+    {
+      "epoch": 12.403846153846153,
+      "grad_norm": 1.0067028999328613,
+      "learning_rate": 2.4963829787234043e-05,
+      "loss": 0.0245,
+      "step": 1290
+    },
+    {
+      "epoch": 12.5,
+      "grad_norm": 0.999679684638977,
+      "learning_rate": 2.49e-05,
+      "loss": 0.0151,
+      "step": 1300
+    },
+    {
+      "epoch": 12.596153846153847,
+      "grad_norm": 1.5106195211410522,
+      "learning_rate": 2.4836170212765958e-05,
+      "loss": 0.014,
+      "step": 1310
+    },
+    {
+      "epoch": 12.692307692307692,
+      "grad_norm": 0.810861349105835,
+      "learning_rate": 2.4772340425531913e-05,
+      "loss": 0.0179,
+      "step": 1320
+    },
+    {
+      "epoch": 12.788461538461538,
+      "grad_norm": 0.45111358165740967,
+      "learning_rate": 2.4708510638297872e-05,
+      "loss": 0.0128,
+      "step": 1330
+    },
+    {
+      "epoch": 12.884615384615385,
+      "grad_norm": 0.6044699549674988,
+      "learning_rate": 2.4644680851063828e-05,
+      "loss": 0.0195,
+      "step": 1340
+    },
+    {
+      "epoch": 12.98076923076923,
+      "grad_norm": 0.5198887586593628,
+      "learning_rate": 2.4580851063829787e-05,
+      "loss": 0.0154,
+      "step": 1350
+    },
+    {
+      "epoch": 13.076923076923077,
+      "grad_norm": 0.7923489809036255,
+      "learning_rate": 2.4517021276595746e-05,
+      "loss": 0.0155,
+      "step": 1360
+    },
+    {
+      "epoch": 13.173076923076923,
+      "grad_norm": 0.6800804734230042,
+      "learning_rate": 2.44531914893617e-05,
+      "loss": 0.0144,
+      "step": 1370
+    },
+    {
+      "epoch": 13.26923076923077,
+      "grad_norm": 0.8769320845603943,
+      "learning_rate": 2.438936170212766e-05,
+      "loss": 0.0142,
+      "step": 1380
+    },
+    {
+      "epoch": 13.365384615384615,
+      "grad_norm": 0.14133420586585999,
+      "learning_rate": 2.4325531914893616e-05,
+      "loss": 0.0137,
+      "step": 1390
+    },
+    {
+      "epoch": 13.461538461538462,
+      "grad_norm": 0.7058908343315125,
+      "learning_rate": 2.4261702127659575e-05,
+      "loss": 0.0206,
+      "step": 1400
+    },
+    {
+      "epoch": 13.557692307692308,
+      "grad_norm": 0.4690243899822235,
+      "learning_rate": 2.419787234042553e-05,
+      "loss": 0.0196,
+      "step": 1410
+    },
+    {
+      "epoch": 13.653846153846153,
+      "grad_norm": 0.4434458017349243,
+      "learning_rate": 2.413404255319149e-05,
+      "loss": 0.0156,
+      "step": 1420
+    },
+    {
+      "epoch": 13.75,
+      "grad_norm": 1.199776530265808,
+      "learning_rate": 2.4070212765957445e-05,
+      "loss": 0.0129,
+      "step": 1430
+    },
+    {
+      "epoch": 13.846153846153847,
+      "grad_norm": 0.999998927116394,
+      "learning_rate": 2.4006382978723404e-05,
+      "loss": 0.0153,
+      "step": 1440
+    },
+    {
+      "epoch": 13.942307692307692,
+      "grad_norm": 0.4517202079296112,
+      "learning_rate": 2.394255319148936e-05,
+      "loss": 0.0135,
+      "step": 1450
+    },
+    {
+      "epoch": 14.038461538461538,
+      "grad_norm": 3.2634222507476807,
+      "learning_rate": 2.387872340425532e-05,
+      "loss": 0.0134,
+      "step": 1460
+    },
+    {
+      "epoch": 14.134615384615385,
+      "grad_norm": 0.2920875549316406,
+      "learning_rate": 2.3814893617021275e-05,
+      "loss": 0.0137,
+      "step": 1470
+    },
+    {
+      "epoch": 14.23076923076923,
+      "grad_norm": 0.38768115639686584,
+      "learning_rate": 2.3751063829787234e-05,
+      "loss": 0.0122,
+      "step": 1480
+    },
+    {
+      "epoch": 14.326923076923077,
+      "grad_norm": 0.26864001154899597,
+      "learning_rate": 2.368723404255319e-05,
+      "loss": 0.0117,
+      "step": 1490
+    },
+    {
+      "epoch": 14.423076923076923,
+      "grad_norm": 0.6234428286552429,
+      "learning_rate": 2.3623404255319148e-05,
+      "loss": 0.0182,
+      "step": 1500
+    },
+    {
+      "epoch": 14.51923076923077,
+      "grad_norm": 0.464770644903183,
+      "learning_rate": 2.3559574468085107e-05,
+      "loss": 0.017,
+      "step": 1510
+    },
+    {
+      "epoch": 14.615384615384615,
+      "grad_norm": 0.4075222313404083,
+      "learning_rate": 2.3495744680851066e-05,
+      "loss": 0.0131,
+      "step": 1520
+    },
+    {
+      "epoch": 14.711538461538462,
+      "grad_norm": 0.5356881022453308,
+      "learning_rate": 2.3431914893617022e-05,
+      "loss": 0.0123,
+      "step": 1530
+    },
+    {
+      "epoch": 14.807692307692308,
+      "grad_norm": 0.4046107828617096,
+      "learning_rate": 2.336808510638298e-05,
+      "loss": 0.013,
+      "step": 1540
+    },
+    {
+      "epoch": 14.903846153846153,
+      "grad_norm": 0.5966467261314392,
+      "learning_rate": 2.330425531914894e-05,
+      "loss": 0.0153,
+      "step": 1550
+    },
+    {
+      "epoch": 15.0,
+      "grad_norm": 2.5298049449920654,
+      "learning_rate": 2.3240425531914895e-05,
+      "loss": 0.0213,
+      "step": 1560
+    },
+    {
+      "epoch": 15.096153846153847,
+      "grad_norm": 0.5523245930671692,
+      "learning_rate": 2.3176595744680854e-05,
+      "loss": 0.0185,
+      "step": 1570
+    },
+    {
+      "epoch": 15.192307692307692,
+      "grad_norm": 1.1738141775131226,
+      "learning_rate": 2.311276595744681e-05,
+      "loss": 0.0152,
+      "step": 1580
+    },
+    {
+      "epoch": 15.288461538461538,
+      "grad_norm": 0.4074403941631317,
+      "learning_rate": 2.304893617021277e-05,
+      "loss": 0.0153,
+      "step": 1590
+    },
+    {
+      "epoch": 15.384615384615385,
+      "grad_norm": 0.5409718751907349,
+      "learning_rate": 2.2985106382978725e-05,
+      "loss": 0.0115,
+      "step": 1600
+    },
+    {
+      "epoch": 15.48076923076923,
+      "grad_norm": 0.39902809262275696,
+      "learning_rate": 2.2921276595744684e-05,
+      "loss": 0.0126,
+      "step": 1610
+    },
+    {
+      "epoch": 15.576923076923077,
+      "grad_norm": 0.2555994391441345,
+      "learning_rate": 2.285744680851064e-05,
+      "loss": 0.015,
+      "step": 1620
+    },
+    {
+      "epoch": 15.673076923076923,
+      "grad_norm": 0.5419341325759888,
+      "learning_rate": 2.2793617021276598e-05,
+      "loss": 0.0146,
+      "step": 1630
+    },
+    {
+      "epoch": 15.76923076923077,
+      "grad_norm": 0.31709060072898865,
+      "learning_rate": 2.2729787234042554e-05,
+      "loss": 0.0103,
+      "step": 1640
+    },
+    {
+      "epoch": 15.865384615384615,
+      "grad_norm": 0.514011800289154,
+      "learning_rate": 2.2665957446808513e-05,
+      "loss": 0.0186,
+      "step": 1650
+    },
+    {
+      "epoch": 15.961538461538462,
+      "grad_norm": 1.1153632402420044,
+      "learning_rate": 2.260212765957447e-05,
+      "loss": 0.0107,
+      "step": 1660
+    },
+    {
+      "epoch": 16.057692307692307,
+      "grad_norm": 0.3839642405509949,
+      "learning_rate": 2.2538297872340427e-05,
+      "loss": 0.0137,
+      "step": 1670
+    },
+    {
+      "epoch": 16.153846153846153,
+      "grad_norm": 0.42906543612480164,
+      "learning_rate": 2.2474468085106383e-05,
+      "loss": 0.0115,
+      "step": 1680
+    },
+    {
+      "epoch": 16.25,
+      "grad_norm": 0.35228168964385986,
+      "learning_rate": 2.2410638297872342e-05,
+      "loss": 0.0119,
+      "step": 1690
+    },
+    {
+      "epoch": 16.346153846153847,
+      "grad_norm": 0.9752399325370789,
+      "learning_rate": 2.2346808510638298e-05,
+      "loss": 0.0122,
+      "step": 1700
+    },
+    {
+      "epoch": 16.442307692307693,
+      "grad_norm": 0.5118827223777771,
+      "learning_rate": 2.2282978723404257e-05,
+      "loss": 0.0105,
+      "step": 1710
+    },
+    {
+      "epoch": 16.53846153846154,
+      "grad_norm": 0.6164939999580383,
+      "learning_rate": 2.2219148936170216e-05,
+      "loss": 0.0125,
+      "step": 1720
+    },
+    {
+      "epoch": 16.634615384615383,
+      "grad_norm": 0.5313933491706848,
+      "learning_rate": 2.215531914893617e-05,
+      "loss": 0.011,
+      "step": 1730
+    },
+    {
+      "epoch": 16.73076923076923,
+      "grad_norm": 0.5187636017799377,
+      "learning_rate": 2.209148936170213e-05,
+      "loss": 0.0144,
+      "step": 1740
+    },
+    {
+      "epoch": 16.826923076923077,
+      "grad_norm": 1.1867964267730713,
+      "learning_rate": 2.2027659574468086e-05,
+      "loss": 0.0093,
+      "step": 1750
+    },
+    {
+      "epoch": 16.923076923076923,
+      "grad_norm": 0.7687259316444397,
+      "learning_rate": 2.1963829787234045e-05,
+      "loss": 0.0115,
+      "step": 1760
+    },
+    {
+      "epoch": 17.01923076923077,
+      "grad_norm": 0.28656214475631714,
+      "learning_rate": 2.19e-05,
+      "loss": 0.0098,
+      "step": 1770
+    },
+    {
+      "epoch": 17.115384615384617,
+      "grad_norm": 0.9335207939147949,
+      "learning_rate": 2.183617021276596e-05,
+      "loss": 0.0122,
+      "step": 1780
+    },
+    {
+      "epoch": 17.21153846153846,
+      "grad_norm": 0.21729937195777893,
+      "learning_rate": 2.1772340425531915e-05,
+      "loss": 0.0083,
+      "step": 1790
+    },
+    {
+      "epoch": 17.307692307692307,
+      "grad_norm": 0.4466613233089447,
+      "learning_rate": 2.1708510638297874e-05,
+      "loss": 0.0122,
+      "step": 1800
+    },
+    {
+      "epoch": 17.403846153846153,
+      "grad_norm": 1.2216111421585083,
+      "learning_rate": 2.164468085106383e-05,
+      "loss": 0.0169,
+      "step": 1810
+    },
+    {
+      "epoch": 17.5,
+      "grad_norm": 0.4963133633136749,
+      "learning_rate": 2.158085106382979e-05,
+      "loss": 0.0105,
+      "step": 1820
+    },
+    {
+      "epoch": 17.596153846153847,
+      "grad_norm": 0.701065719127655,
+      "learning_rate": 2.1517021276595744e-05,
+      "loss": 0.0168,
+      "step": 1830
+    },
+    {
+      "epoch": 17.692307692307693,
+      "grad_norm": 0.8817301392555237,
+      "learning_rate": 2.1453191489361703e-05,
+      "loss": 0.008,
+      "step": 1840
+    },
+    {
+      "epoch": 17.78846153846154,
+      "grad_norm": 0.5563378930091858,
+      "learning_rate": 2.138936170212766e-05,
+      "loss": 0.011,
+      "step": 1850
+    },
+    {
+      "epoch": 17.884615384615383,
+      "grad_norm": 0.16618973016738892,
+      "learning_rate": 2.1325531914893618e-05,
+      "loss": 0.0103,
+      "step": 1860
+    },
+    {
+      "epoch": 17.98076923076923,
+      "grad_norm": 0.2868179678916931,
+      "learning_rate": 2.1261702127659573e-05,
+      "loss": 0.012,
+      "step": 1870
+    },
+    {
+      "epoch": 18.076923076923077,
+      "grad_norm": 0.6677718758583069,
+      "learning_rate": 2.1197872340425532e-05,
+      "loss": 0.0082,
+      "step": 1880
+    },
+    {
+      "epoch": 18.173076923076923,
+      "grad_norm": 0.3902364671230316,
+      "learning_rate": 2.1134042553191488e-05,
+      "loss": 0.0116,
+      "step": 1890
+    },
+    {
+      "epoch": 18.26923076923077,
+      "grad_norm": 0.563302218914032,
+      "learning_rate": 2.1070212765957447e-05,
+      "loss": 0.0093,
+      "step": 1900
+    },
+    {
+      "epoch": 18.365384615384617,
+      "grad_norm": 0.8351249098777771,
+      "learning_rate": 2.1006382978723406e-05,
+      "loss": 0.0072,
+      "step": 1910
+    },
+    {
+      "epoch": 18.46153846153846,
+      "grad_norm": 0.35901957750320435,
+      "learning_rate": 2.094255319148936e-05,
+      "loss": 0.0145,
+      "step": 1920
+    },
+    {
+      "epoch": 18.557692307692307,
+      "grad_norm": 1.6402753591537476,
+      "learning_rate": 2.087872340425532e-05,
+      "loss": 0.0148,
+      "step": 1930
+    },
+    {
+      "epoch": 18.653846153846153,
+      "grad_norm": 0.6766220331192017,
+      "learning_rate": 2.0814893617021276e-05,
+      "loss": 0.0093,
+      "step": 1940
+    },
+    {
+      "epoch": 18.75,
+      "grad_norm": 0.3588562607765198,
+      "learning_rate": 2.0751063829787235e-05,
+      "loss": 0.0111,
+      "step": 1950
+    },
+    {
+      "epoch": 18.846153846153847,
+      "grad_norm": 0.3064340651035309,
+      "learning_rate": 2.068723404255319e-05,
+      "loss": 0.0084,
+      "step": 1960
+    },
+    {
+      "epoch": 18.942307692307693,
+      "grad_norm": 0.9118555188179016,
+      "learning_rate": 2.062340425531915e-05,
+      "loss": 0.0142,
+      "step": 1970
+    },
+    {
+      "epoch": 19.03846153846154,
+      "grad_norm": 0.47235992550849915,
+      "learning_rate": 2.0559574468085105e-05,
+      "loss": 0.0146,
+      "step": 1980
+    },
+    {
+      "epoch": 19.134615384615383,
+      "grad_norm": 0.7751461863517761,
+      "learning_rate": 2.0495744680851064e-05,
+      "loss": 0.0077,
+      "step": 1990
+    },
+    {
+      "epoch": 19.23076923076923,
+      "grad_norm": 0.4031819701194763,
+      "learning_rate": 2.043191489361702e-05,
+      "loss": 0.0107,
+      "step": 2000
+    },
+    {
+      "epoch": 19.23076923076923,
+      "eval_correct_predictions": 1382,
+      "eval_exact_match_accuracy": 0.976678445229682,
+      "eval_loss": 0.007804804015904665,
+      "eval_runtime": 58.2622,
+      "eval_samples_per_second": 24.287,
+      "eval_steps_per_second": 0.395,
+      "eval_total_predictions": 1415,
+      "step": 2000
+    },
+    {
+      "epoch": 19.326923076923077,
+      "grad_norm": 0.42076176404953003,
+      "learning_rate": 2.036808510638298e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 19.423076923076923,
+      "grad_norm": 0.8498334884643555,
+      "learning_rate": 2.0304255319148935e-05,
+      "loss": 0.01,
+      "step": 2020
+    },
+    {
+      "epoch": 19.51923076923077,
+      "grad_norm": 0.5926603674888611,
+      "learning_rate": 2.0240425531914894e-05,
+      "loss": 0.009,
+      "step": 2030
+    },
+    {
+      "epoch": 19.615384615384617,
+      "grad_norm": 0.6045398712158203,
+      "learning_rate": 2.017659574468085e-05,
+      "loss": 0.0094,
+      "step": 2040
+    },
+    {
+      "epoch": 19.71153846153846,
+      "grad_norm": 0.4028799831867218,
+      "learning_rate": 2.0112765957446808e-05,
+      "loss": 0.012,
+      "step": 2050
+    },
+    {
+      "epoch": 19.807692307692307,
+      "grad_norm": 0.4177253544330597,
+      "learning_rate": 2.0048936170212764e-05,
+      "loss": 0.012,
+      "step": 2060
+    },
+    {
+      "epoch": 19.903846153846153,
+      "grad_norm": 0.30318310856819153,
+      "learning_rate": 1.9985106382978723e-05,
+      "loss": 0.0119,
+      "step": 2070
+    },
+    {
+      "epoch": 20.0,
+      "grad_norm": 1.5503188371658325,
+      "learning_rate": 1.9921276595744682e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 20.096153846153847,
+      "grad_norm": 0.6722589731216431,
+      "learning_rate": 1.9857446808510637e-05,
+      "loss": 0.0069,
+      "step": 2090
+    },
+    {
+      "epoch": 20.192307692307693,
+      "grad_norm": 0.2268471121788025,
+      "learning_rate": 1.9793617021276596e-05,
+      "loss": 0.0081,
+      "step": 2100
+    },
+    {
+      "epoch": 20.28846153846154,
+      "grad_norm": 0.4761890769004822,
+      "learning_rate": 1.9729787234042552e-05,
+      "loss": 0.0109,
+      "step": 2110
+    },
+    {
+      "epoch": 20.384615384615383,
+      "grad_norm": 0.37841594219207764,
+      "learning_rate": 1.966595744680851e-05,
+      "loss": 0.0107,
+      "step": 2120
+    },
+    {
+      "epoch": 20.48076923076923,
+      "grad_norm": 0.4525921046733856,
+      "learning_rate": 1.9602127659574467e-05,
+      "loss": 0.0107,
+      "step": 2130
+    },
+    {
+      "epoch": 20.576923076923077,
+      "grad_norm": 0.1672481745481491,
+      "learning_rate": 1.9538297872340426e-05,
+      "loss": 0.0093,
+      "step": 2140
+    },
+    {
+      "epoch": 20.673076923076923,
+      "grad_norm": 0.8809177279472351,
+      "learning_rate": 1.947446808510638e-05,
+      "loss": 0.0118,
+      "step": 2150
+    },
+    {
+      "epoch": 20.76923076923077,
+      "grad_norm": 1.2535909414291382,
+      "learning_rate": 1.941063829787234e-05,
+      "loss": 0.0095,
+      "step": 2160
+    },
+    {
+      "epoch": 20.865384615384617,
+      "grad_norm": 0.2467496544122696,
+      "learning_rate": 1.9346808510638296e-05,
+      "loss": 0.0083,
+      "step": 2170
+    },
+    {
+      "epoch": 20.96153846153846,
+      "grad_norm": 0.3241216838359833,
+      "learning_rate": 1.9282978723404255e-05,
+      "loss": 0.0102,
+      "step": 2180
+    },
+    {
+      "epoch": 21.057692307692307,
+      "grad_norm": 1.0612328052520752,
+      "learning_rate": 1.9219148936170214e-05,
+      "loss": 0.0079,
+      "step": 2190
+    },
+    {
+      "epoch": 21.153846153846153,
+      "grad_norm": 0.6267879605293274,
+      "learning_rate": 1.9155319148936173e-05,
+      "loss": 0.0111,
+      "step": 2200
+    },
+    {
+      "epoch": 21.25,
+      "grad_norm": 0.34094300866127014,
+      "learning_rate": 1.909148936170213e-05,
+      "loss": 0.0082,
+      "step": 2210
+    },
+    {
+      "epoch": 21.346153846153847,
+      "grad_norm": 0.3252468705177307,
+      "learning_rate": 1.9027659574468087e-05,
+      "loss": 0.0076,
+      "step": 2220
+    },
+    {
+      "epoch": 21.442307692307693,
+      "grad_norm": 0.18613576889038086,
+      "learning_rate": 1.8963829787234043e-05,
+      "loss": 0.0111,
+      "step": 2230
+    },
+    {
+      "epoch": 21.53846153846154,
+      "grad_norm": 0.5395278334617615,
+      "learning_rate": 1.8900000000000002e-05,
+      "loss": 0.008,
+      "step": 2240
+    },
+    {
+      "epoch": 21.634615384615383,
+      "grad_norm": 1.6617084741592407,
+      "learning_rate": 1.8836170212765958e-05,
+      "loss": 0.0092,
+      "step": 2250
+    },
+    {
+      "epoch": 21.73076923076923,
+      "grad_norm": 0.4416300654411316,
+      "learning_rate": 1.8772340425531917e-05,
+      "loss": 0.0075,
+      "step": 2260
+    },
+    {
+      "epoch": 21.826923076923077,
+      "grad_norm": 0.25209808349609375,
+      "learning_rate": 1.8708510638297876e-05,
+      "loss": 0.0084,
+      "step": 2270
+    },
+    {
+      "epoch": 21.923076923076923,
+      "grad_norm": 0.47792643308639526,
+      "learning_rate": 1.864468085106383e-05,
+      "loss": 0.0124,
+      "step": 2280
+    },
+    {
+      "epoch": 22.01923076923077,
+      "grad_norm": 0.49080875515937805,
+      "learning_rate": 1.858085106382979e-05,
+      "loss": 0.0096,
+      "step": 2290
+    },
+    {
+      "epoch": 22.115384615384617,
+      "grad_norm": 0.6654906868934631,
+      "learning_rate": 1.8517021276595746e-05,
+      "loss": 0.0066,
+      "step": 2300
+    },
+    {
+      "epoch": 22.21153846153846,
+      "grad_norm": 0.9110317230224609,
+      "learning_rate": 1.8453191489361705e-05,
+      "loss": 0.011,
+      "step": 2310
+    },
+    {
+      "epoch": 22.307692307692307,
+      "grad_norm": 0.9715136885643005,
+      "learning_rate": 1.838936170212766e-05,
+      "loss": 0.0094,
+      "step": 2320
+    },
+    {
+      "epoch": 22.403846153846153,
+      "grad_norm": 1.4085966348648071,
+      "learning_rate": 1.832553191489362e-05,
+      "loss": 0.0086,
+      "step": 2330
+    },
+    {
+      "epoch": 22.5,
+      "grad_norm": 0.3735029995441437,
+      "learning_rate": 1.8261702127659575e-05,
+      "loss": 0.009,
+      "step": 2340
+    },
+    {
+      "epoch": 22.596153846153847,
+      "grad_norm": 0.6319369077682495,
+      "learning_rate": 1.8197872340425534e-05,
+      "loss": 0.0075,
+      "step": 2350
+    },
+    {
+      "epoch": 22.692307692307693,
+      "grad_norm": 1.088571548461914,
+      "learning_rate": 1.813404255319149e-05,
+      "loss": 0.0075,
+      "step": 2360
+    },
+    {
+      "epoch": 22.78846153846154,
+      "grad_norm": 0.1355327069759369,
+      "learning_rate": 1.807021276595745e-05,
+      "loss": 0.0084,
+      "step": 2370
+    },
+    {
+      "epoch": 22.884615384615383,
+      "grad_norm": 0.1889447569847107,
+      "learning_rate": 1.8006382978723404e-05,
+      "loss": 0.0092,
+      "step": 2380
+    },
+    {
+      "epoch": 22.98076923076923,
+      "grad_norm": 0.1720777302980423,
+      "learning_rate": 1.7942553191489363e-05,
+      "loss": 0.0092,
+      "step": 2390
+    },
+    {
+      "epoch": 23.076923076923077,
+      "grad_norm": 0.7916960716247559,
+      "learning_rate": 1.787872340425532e-05,
+      "loss": 0.0068,
+      "step": 2400
+    },
+    {
+      "epoch": 23.173076923076923,
+      "grad_norm": 0.3657436668872833,
+      "learning_rate": 1.7814893617021278e-05,
+      "loss": 0.0082,
+      "step": 2410
+    },
+    {
+      "epoch": 23.26923076923077,
+      "grad_norm": 1.1953802108764648,
+      "learning_rate": 1.7751063829787233e-05,
+      "loss": 0.0108,
+      "step": 2420
+    },
+    {
+      "epoch": 23.365384615384617,
+      "grad_norm": 1.18259859085083,
+      "learning_rate": 1.7687234042553192e-05,
+      "loss": 0.007,
+      "step": 2430
+    },
+    {
+      "epoch": 23.46153846153846,
+      "grad_norm": 0.5571394562721252,
+      "learning_rate": 1.762340425531915e-05,
+      "loss": 0.0091,
+      "step": 2440
+    },
+    {
+      "epoch": 23.557692307692307,
+      "grad_norm": 0.6768787503242493,
+      "learning_rate": 1.7559574468085107e-05,
+      "loss": 0.0098,
+      "step": 2450
+    },
+    {
+      "epoch": 23.653846153846153,
+      "grad_norm": 1.5829925537109375,
+      "learning_rate": 1.7495744680851066e-05,
+      "loss": 0.0104,
+      "step": 2460
+    },
+    {
+      "epoch": 23.75,
+      "grad_norm": 0.7399948239326477,
+      "learning_rate": 1.743191489361702e-05,
+      "loss": 0.0093,
+      "step": 2470
+    },
+    {
+      "epoch": 23.846153846153847,
+      "grad_norm": 0.14280381798744202,
+      "learning_rate": 1.736808510638298e-05,
+      "loss": 0.0079,
+      "step": 2480
+    },
+    {
+      "epoch": 23.942307692307693,
+      "grad_norm": 0.14676937460899353,
+      "learning_rate": 1.7304255319148936e-05,
+      "loss": 0.0067,
+      "step": 2490
+    },
+    {
+      "epoch": 24.03846153846154,
+      "grad_norm": 0.29147419333457947,
+      "learning_rate": 1.7240425531914895e-05,
+      "loss": 0.007,
+      "step": 2500
+    },
+    {
+      "epoch": 24.134615384615383,
+      "grad_norm": 0.773118793964386,
+      "learning_rate": 1.717659574468085e-05,
+      "loss": 0.006,
+      "step": 2510
+    },
+    {
+      "epoch": 24.23076923076923,
+      "grad_norm": 1.2377712726593018,
+      "learning_rate": 1.711276595744681e-05,
+      "loss": 0.005,
+      "step": 2520
+    },
+    {
+      "epoch": 24.326923076923077,
+      "grad_norm": 0.9301372170448303,
+      "learning_rate": 1.7048936170212765e-05,
+      "loss": 0.0082,
+      "step": 2530
+    },
+    {
+      "epoch": 24.423076923076923,
+      "grad_norm": 0.3903908133506775,
+      "learning_rate": 1.6985106382978724e-05,
+      "loss": 0.007,
+      "step": 2540
+    },
+    {
+      "epoch": 24.51923076923077,
+      "grad_norm": 0.8196276426315308,
+      "learning_rate": 1.692127659574468e-05,
+      "loss": 0.0064,
+      "step": 2550
+    },
+    {
+      "epoch": 24.615384615384617,
+      "grad_norm": 0.3960936963558197,
+      "learning_rate": 1.685744680851064e-05,
+      "loss": 0.0066,
+      "step": 2560
+    },
+    {
+      "epoch": 24.71153846153846,
+      "grad_norm": 0.43898555636405945,
+      "learning_rate": 1.6793617021276595e-05,
+      "loss": 0.0101,
+      "step": 2570
+    },
+    {
+      "epoch": 24.807692307692307,
+      "grad_norm": 0.3561786115169525,
+      "learning_rate": 1.6729787234042554e-05,
+      "loss": 0.0122,
+      "step": 2580
+    },
+    {
+      "epoch": 24.903846153846153,
+      "grad_norm": 0.5492029786109924,
+      "learning_rate": 1.666595744680851e-05,
+      "loss": 0.0093,
+      "step": 2590
+    },
+    {
+      "epoch": 25.0,
+      "grad_norm": 1.1004884243011475,
+      "learning_rate": 1.6602127659574468e-05,
+      "loss": 0.0083,
+      "step": 2600
+    },
+    {
+      "epoch": 25.096153846153847,
+      "grad_norm": 0.4236828088760376,
+      "learning_rate": 1.6538297872340424e-05,
+      "loss": 0.0092,
+      "step": 2610
+    },
+    {
+      "epoch": 25.192307692307693,
+      "grad_norm": 0.642759382724762,
+      "learning_rate": 1.6474468085106383e-05,
+      "loss": 0.0066,
+      "step": 2620
+    },
+    {
+      "epoch": 25.28846153846154,
+      "grad_norm": 1.0845800638198853,
+      "learning_rate": 1.6410638297872342e-05,
+      "loss": 0.0067,
+      "step": 2630
+    },
+    {
+      "epoch": 25.384615384615383,
+      "grad_norm": 0.4961228370666504,
+      "learning_rate": 1.6346808510638297e-05,
+      "loss": 0.0051,
+      "step": 2640
+    },
+    {
+      "epoch": 25.48076923076923,
+      "grad_norm": 0.7525473237037659,
+      "learning_rate": 1.6282978723404256e-05,
+      "loss": 0.008,
+      "step": 2650
+    },
+    {
+      "epoch": 25.576923076923077,
+      "grad_norm": 0.5340065956115723,
+      "learning_rate": 1.6219148936170212e-05,
+      "loss": 0.0086,
+      "step": 2660
+    },
+    {
+      "epoch": 25.673076923076923,
+      "grad_norm": 0.9437041282653809,
+      "learning_rate": 1.615531914893617e-05,
+      "loss": 0.0073,
+      "step": 2670
+    },
+    {
+      "epoch": 25.76923076923077,
+      "grad_norm": 0.1908496618270874,
+      "learning_rate": 1.6091489361702127e-05,
+      "loss": 0.0054,
+      "step": 2680
+    },
+    {
+      "epoch": 25.865384615384617,
+      "grad_norm": 0.7633543014526367,
+      "learning_rate": 1.6027659574468086e-05,
+      "loss": 0.0062,
+      "step": 2690
+    },
+    {
+      "epoch": 25.96153846153846,
+      "grad_norm": 0.36000245809555054,
+      "learning_rate": 1.596382978723404e-05,
+      "loss": 0.0087,
+      "step": 2700
+    },
+    {
+      "epoch": 26.057692307692307,
+      "grad_norm": 0.24572286009788513,
+      "learning_rate": 1.59e-05,
+      "loss": 0.0137,
+      "step": 2710
+    },
+    {
+      "epoch": 26.153846153846153,
+      "grad_norm": 0.7748898863792419,
+      "learning_rate": 1.5836170212765956e-05,
+      "loss": 0.007,
+      "step": 2720
+    },
+    {
+      "epoch": 26.25,
+      "grad_norm": 0.18942640721797943,
+      "learning_rate": 1.5772340425531915e-05,
+      "loss": 0.005,
+      "step": 2730
+    },
+    {
+      "epoch": 26.346153846153847,
+      "grad_norm": 0.5449305772781372,
+      "learning_rate": 1.570851063829787e-05,
+      "loss": 0.0093,
+      "step": 2740
+    },
+    {
+      "epoch": 26.442307692307693,
+      "grad_norm": 0.7563040256500244,
+      "learning_rate": 1.564468085106383e-05,
+      "loss": 0.0055,
+      "step": 2750
+    },
+    {
+      "epoch": 26.53846153846154,
+      "grad_norm": 0.024510016664862633,
+      "learning_rate": 1.5580851063829785e-05,
+      "loss": 0.0091,
+      "step": 2760
+    },
+    {
+      "epoch": 26.634615384615383,
+      "grad_norm": 0.9824059009552002,
+      "learning_rate": 1.5517021276595744e-05,
+      "loss": 0.007,
+      "step": 2770
+    },
+    {
+      "epoch": 26.73076923076923,
+      "grad_norm": 1.0016579627990723,
+      "learning_rate": 1.54531914893617e-05,
+      "loss": 0.0082,
+      "step": 2780
+    },
+    {
+      "epoch": 26.826923076923077,
+      "grad_norm": 0.3715296685695648,
+      "learning_rate": 1.538936170212766e-05,
+      "loss": 0.0051,
+      "step": 2790
+    },
+    {
+      "epoch": 26.923076923076923,
+      "grad_norm": 0.609065592288971,
+      "learning_rate": 1.5325531914893618e-05,
+      "loss": 0.0081,
+      "step": 2800
+    },
+    {
+      "epoch": 27.01923076923077,
+      "grad_norm": 0.8376663327217102,
+      "learning_rate": 1.5261702127659573e-05,
+      "loss": 0.0082,
+      "step": 2810
+    },
+    {
+      "epoch": 27.115384615384617,
+      "grad_norm": 0.08116922527551651,
+      "learning_rate": 1.5197872340425534e-05,
+      "loss": 0.0063,
+      "step": 2820
+    },
+    {
+      "epoch": 27.21153846153846,
+      "grad_norm": 0.4511634409427643,
+      "learning_rate": 1.513404255319149e-05,
+      "loss": 0.0061,
+      "step": 2830
+    },
+    {
+      "epoch": 27.307692307692307,
+      "grad_norm": 0.8804696202278137,
+      "learning_rate": 1.5070212765957449e-05,
+      "loss": 0.0063,
+      "step": 2840
+    },
+    {
+      "epoch": 27.403846153846153,
+      "grad_norm": 0.2572877109050751,
+      "learning_rate": 1.5006382978723404e-05,
+      "loss": 0.0058,
+      "step": 2850
+    },
+    {
+      "epoch": 27.5,
+      "grad_norm": 1.1330682039260864,
+      "learning_rate": 1.4942553191489361e-05,
+      "loss": 0.0077,
+      "step": 2860
+    },
+    {
+      "epoch": 27.596153846153847,
+      "grad_norm": 0.5823140740394592,
+      "learning_rate": 1.4878723404255319e-05,
+      "loss": 0.0068,
+      "step": 2870
+    },
+    {
+      "epoch": 27.692307692307693,
+      "grad_norm": 0.20316189527511597,
+      "learning_rate": 1.4814893617021276e-05,
+      "loss": 0.007,
+      "step": 2880
+    },
+    {
+      "epoch": 27.78846153846154,
+      "grad_norm": 0.86988365650177,
+      "learning_rate": 1.4751063829787235e-05,
+      "loss": 0.0053,
+      "step": 2890
+    },
+    {
+      "epoch": 27.884615384615383,
+      "grad_norm": 0.4562746584415436,
+      "learning_rate": 1.4687234042553192e-05,
+      "loss": 0.0058,
+      "step": 2900
+    },
+    {
+      "epoch": 27.98076923076923,
+      "grad_norm": 0.7957671284675598,
+      "learning_rate": 1.462340425531915e-05,
+      "loss": 0.0061,
+      "step": 2910
+    },
+    {
+      "epoch": 28.076923076923077,
+      "grad_norm": 0.2981686592102051,
+      "learning_rate": 1.4559574468085107e-05,
+      "loss": 0.0091,
+      "step": 2920
+    },
+    {
+      "epoch": 28.173076923076923,
+      "grad_norm": 0.34361180663108826,
+      "learning_rate": 1.4495744680851064e-05,
+      "loss": 0.0045,
+      "step": 2930
+    },
+    {
+      "epoch": 28.26923076923077,
+      "grad_norm": 0.3850739598274231,
+      "learning_rate": 1.4431914893617022e-05,
+      "loss": 0.0093,
+      "step": 2940
+    },
+    {
+      "epoch": 28.365384615384617,
+      "grad_norm": 0.2868598699569702,
+      "learning_rate": 1.4368085106382979e-05,
+      "loss": 0.0048,
+      "step": 2950
+    },
+    {
+      "epoch": 28.46153846153846,
+      "grad_norm": 0.20695632696151733,
+      "learning_rate": 1.4304255319148936e-05,
+      "loss": 0.0054,
+      "step": 2960
+    },
+    {
+      "epoch": 28.557692307692307,
+      "grad_norm": 0.13250812888145447,
+      "learning_rate": 1.4240425531914893e-05,
+      "loss": 0.0038,
+      "step": 2970
+    },
+    {
+      "epoch": 28.653846153846153,
+      "grad_norm": 0.9032599329948425,
+      "learning_rate": 1.417659574468085e-05,
+      "loss": 0.007,
+      "step": 2980
+    },
+    {
+      "epoch": 28.75,
+      "grad_norm": 0.2160140872001648,
+      "learning_rate": 1.4112765957446808e-05,
+      "loss": 0.0051,
+      "step": 2990
+    },
+    {
+      "epoch": 28.846153846153847,
+      "grad_norm": 0.8250911831855774,
+      "learning_rate": 1.4048936170212765e-05,
+      "loss": 0.0079,
+      "step": 3000
+    },
+    {
+      "epoch": 28.846153846153847,
+      "eval_correct_predictions": 1391,
+      "eval_exact_match_accuracy": 0.9830388692579505,
+      "eval_loss": 0.006084157153964043,
+      "eval_runtime": 58.5417,
+      "eval_samples_per_second": 24.171,
+      "eval_steps_per_second": 0.393,
+      "eval_total_predictions": 1415,
+      "step": 3000
+    },
+    {
+      "epoch": 28.942307692307693,
+      "grad_norm": 2.7227625846862793,
+      "learning_rate": 1.3985106382978723e-05,
+      "loss": 0.0085,
+      "step": 3010
+    },
+    {
+      "epoch": 29.03846153846154,
+      "grad_norm": 0.41631412506103516,
+      "learning_rate": 1.3921276595744682e-05,
+      "loss": 0.0065,
+      "step": 3020
+    },
+    {
+      "epoch": 29.134615384615383,
+      "grad_norm": 0.566443681716919,
+      "learning_rate": 1.3857446808510639e-05,
+      "loss": 0.0044,
+      "step": 3030
+    },
+    {
+      "epoch": 29.23076923076923,
+      "grad_norm": 0.34673500061035156,
+      "learning_rate": 1.3793617021276596e-05,
+      "loss": 0.0081,
+      "step": 3040
+    },
+    {
+      "epoch": 29.326923076923077,
+      "grad_norm": 0.10608164221048355,
+      "learning_rate": 1.3729787234042554e-05,
+      "loss": 0.0059,
+      "step": 3050
+    },
+    {
+      "epoch": 29.423076923076923,
+      "grad_norm": 0.258186012506485,
+      "learning_rate": 1.3665957446808511e-05,
+      "loss": 0.0042,
+      "step": 3060
+    },
+    {
+      "epoch": 29.51923076923077,
+      "grad_norm": 0.8078503608703613,
+      "learning_rate": 1.360212765957447e-05,
+      "loss": 0.0056,
+      "step": 3070
+    },
+    {
+      "epoch": 29.615384615384617,
+      "grad_norm": 0.3078606128692627,
+      "learning_rate": 1.3538297872340427e-05,
+      "loss": 0.0065,
+      "step": 3080
+    },
+    {
+      "epoch": 29.71153846153846,
+      "grad_norm": 0.5479371547698975,
+      "learning_rate": 1.3474468085106384e-05,
+      "loss": 0.005,
+      "step": 3090
+    },
+    {
+      "epoch": 29.807692307692307,
+      "grad_norm": 0.43646132946014404,
+      "learning_rate": 1.3410638297872342e-05,
+      "loss": 0.0075,
+      "step": 3100
+    },
+    {
+      "epoch": 29.903846153846153,
+      "grad_norm": 0.15388578176498413,
+      "learning_rate": 1.3346808510638299e-05,
+      "loss": 0.0066,
+      "step": 3110
+    },
+    {
+      "epoch": 30.0,
+      "grad_norm": 1.9249227046966553,
+      "learning_rate": 1.3282978723404256e-05,
+      "loss": 0.0073,
+      "step": 3120
+    },
+    {
+      "epoch": 30.096153846153847,
+      "grad_norm": 0.027103092521429062,
+      "learning_rate": 1.3219148936170214e-05,
+      "loss": 0.0057,
+      "step": 3130
+    },
+    {
+      "epoch": 30.192307692307693,
+      "grad_norm": 0.8037792444229126,
+      "learning_rate": 1.3155319148936171e-05,
+      "loss": 0.0092,
+      "step": 3140
+    },
+    {
+      "epoch": 30.28846153846154,
+      "grad_norm": 1.0261093378067017,
+      "learning_rate": 1.3091489361702128e-05,
+      "loss": 0.0069,
+      "step": 3150
+    },
+    {
+      "epoch": 30.384615384615383,
+      "grad_norm": 0.049915142357349396,
+      "learning_rate": 1.3027659574468086e-05,
+      "loss": 0.0074,
+      "step": 3160
+    },
+    {
+      "epoch": 30.48076923076923,
+      "grad_norm": 0.51163250207901,
+      "learning_rate": 1.2963829787234043e-05,
+      "loss": 0.0044,
+      "step": 3170
+    },
+    {
+      "epoch": 30.576923076923077,
+      "grad_norm": 1.2742948532104492,
+      "learning_rate": 1.29e-05,
+      "loss": 0.0072,
+      "step": 3180
+    },
+    {
+      "epoch": 30.673076923076923,
+      "grad_norm": 2.13923978805542,
+      "learning_rate": 1.2836170212765957e-05,
+      "loss": 0.005,
+      "step": 3190
+    },
+    {
+      "epoch": 30.76923076923077,
+      "grad_norm": 0.12393970787525177,
+      "learning_rate": 1.2772340425531915e-05,
+      "loss": 0.004,
+      "step": 3200
+    },
+    {
+      "epoch": 30.865384615384617,
+      "grad_norm": 0.8308172821998596,
+      "learning_rate": 1.2708510638297872e-05,
+      "loss": 0.0057,
+      "step": 3210
+    },
+    {
+      "epoch": 30.96153846153846,
+      "grad_norm": 0.19924144446849823,
+      "learning_rate": 1.264468085106383e-05,
+      "loss": 0.0046,
+      "step": 3220
+    },
+    {
+      "epoch": 31.057692307692307,
+      "grad_norm": 0.7076709270477295,
+      "learning_rate": 1.2580851063829787e-05,
+      "loss": 0.0047,
+      "step": 3230
+    },
+    {
+      "epoch": 31.153846153846153,
+      "grad_norm": 0.5298773050308228,
+      "learning_rate": 1.2517021276595744e-05,
+      "loss": 0.0076,
+      "step": 3240
+    },
+    {
+      "epoch": 31.25,
+      "grad_norm": 0.11929306387901306,
+      "learning_rate": 1.2453191489361703e-05,
+      "loss": 0.0043,
+      "step": 3250
+    },
+    {
+      "epoch": 31.346153846153847,
+      "grad_norm": 0.3417571485042572,
+      "learning_rate": 1.238936170212766e-05,
+      "loss": 0.0073,
+      "step": 3260
+    },
+    {
+      "epoch": 31.442307692307693,
+      "grad_norm": 0.5745849609375,
+      "learning_rate": 1.2325531914893618e-05,
+      "loss": 0.0048,
+      "step": 3270
+    },
+    {
+      "epoch": 31.53846153846154,
+      "grad_norm": 0.7137449979782104,
+      "learning_rate": 1.2261702127659575e-05,
+      "loss": 0.0041,
+      "step": 3280
+    },
+    {
+      "epoch": 31.634615384615383,
+      "grad_norm": 1.3129104375839233,
+      "learning_rate": 1.2197872340425532e-05,
+      "loss": 0.0033,
+      "step": 3290
+    },
+    {
+      "epoch": 31.73076923076923,
+      "grad_norm": 1.0879548788070679,
+      "learning_rate": 1.213404255319149e-05,
+      "loss": 0.0052,
+      "step": 3300
+    },
+    {
+      "epoch": 31.826923076923077,
+      "grad_norm": 0.42951247096061707,
+      "learning_rate": 1.2070212765957447e-05,
+      "loss": 0.0072,
+      "step": 3310
+    },
+    {
+      "epoch": 31.923076923076923,
+      "grad_norm": 0.2130812406539917,
+      "learning_rate": 1.2006382978723404e-05,
+      "loss": 0.0062,
+      "step": 3320
+    },
+    {
+      "epoch": 32.01923076923077,
+      "grad_norm": 0.13096341490745544,
+      "learning_rate": 1.1942553191489361e-05,
+      "loss": 0.0047,
+      "step": 3330
+    },
+    {
+      "epoch": 32.11538461538461,
+      "grad_norm": 0.42370834946632385,
+      "learning_rate": 1.1878723404255319e-05,
+      "loss": 0.0053,
+      "step": 3340
+    },
+    {
+      "epoch": 32.21153846153846,
+      "grad_norm": 0.6214959621429443,
+      "learning_rate": 1.1814893617021276e-05,
+      "loss": 0.0057,
+      "step": 3350
+    },
+    {
+      "epoch": 32.30769230769231,
+      "grad_norm": 0.24280411005020142,
+      "learning_rate": 1.1751063829787235e-05,
+      "loss": 0.0055,
+      "step": 3360
+    },
+    {
+      "epoch": 32.40384615384615,
+      "grad_norm": 0.22130908071994781,
+      "learning_rate": 1.1687234042553192e-05,
+      "loss": 0.0074,
+      "step": 3370
+    },
+    {
+      "epoch": 32.5,
+      "grad_norm": 0.5882713794708252,
+      "learning_rate": 1.162340425531915e-05,
+      "loss": 0.0065,
+      "step": 3380
+    },
+    {
+      "epoch": 32.59615384615385,
+      "grad_norm": 0.09809905290603638,
+      "learning_rate": 1.1559574468085107e-05,
+      "loss": 0.0061,
+      "step": 3390
+    },
+    {
+      "epoch": 32.69230769230769,
+      "grad_norm": 0.6485664248466492,
+      "learning_rate": 1.1495744680851064e-05,
+      "loss": 0.0058,
+      "step": 3400
+    },
+    {
+      "epoch": 32.78846153846154,
+      "grad_norm": 0.07379120588302612,
+      "learning_rate": 1.1431914893617021e-05,
+      "loss": 0.005,
+      "step": 3410
+    },
+    {
+      "epoch": 32.88461538461539,
+      "grad_norm": 0.2927326261997223,
+      "learning_rate": 1.1368085106382979e-05,
+      "loss": 0.0041,
+      "step": 3420
+    },
+    {
+      "epoch": 32.98076923076923,
+      "grad_norm": 0.3531413674354553,
+      "learning_rate": 1.1304255319148938e-05,
+      "loss": 0.0035,
+      "step": 3430
+    },
+    {
+      "epoch": 33.07692307692308,
+      "grad_norm": 0.10925979912281036,
+      "learning_rate": 1.1240425531914895e-05,
+      "loss": 0.0057,
+      "step": 3440
+    },
+    {
+      "epoch": 33.17307692307692,
+      "grad_norm": 0.2170642763376236,
+      "learning_rate": 1.1176595744680852e-05,
+      "loss": 0.0048,
+      "step": 3450
+    },
+    {
+      "epoch": 33.26923076923077,
+      "grad_norm": 0.774167001247406,
+      "learning_rate": 1.111276595744681e-05,
+      "loss": 0.0041,
+      "step": 3460
+    },
+    {
+      "epoch": 33.36538461538461,
+      "grad_norm": 0.22706744074821472,
+      "learning_rate": 1.1048936170212767e-05,
+      "loss": 0.0042,
+      "step": 3470
+    },
+    {
+      "epoch": 33.46153846153846,
+      "grad_norm": 0.2846420407295227,
+      "learning_rate": 1.0985106382978724e-05,
+      "loss": 0.0056,
+      "step": 3480
+    },
+    {
+      "epoch": 33.55769230769231,
+      "grad_norm": 1.1403943300247192,
+      "learning_rate": 1.0921276595744682e-05,
+      "loss": 0.0093,
+      "step": 3490
+    },
+    {
+      "epoch": 33.65384615384615,
+      "grad_norm": 1.2854979038238525,
+      "learning_rate": 1.0857446808510639e-05,
+      "loss": 0.0056,
+      "step": 3500
+    },
+    {
+      "epoch": 33.75,
+      "grad_norm": 0.6309840679168701,
+      "learning_rate": 1.0793617021276596e-05,
+      "loss": 0.0051,
+      "step": 3510
+    },
+    {
+      "epoch": 33.84615384615385,
+      "grad_norm": 0.35604169964790344,
+      "learning_rate": 1.0729787234042553e-05,
+      "loss": 0.0048,
+      "step": 3520
+    },
+    {
+      "epoch": 33.94230769230769,
+      "grad_norm": 0.038923054933547974,
+      "learning_rate": 1.066595744680851e-05,
+      "loss": 0.0039,
+      "step": 3530
+    },
+    {
+      "epoch": 34.03846153846154,
+      "grad_norm": 0.10073179751634598,
+      "learning_rate": 1.0602127659574468e-05,
+      "loss": 0.0036,
+      "step": 3540
+    },
+    {
+      "epoch": 34.13461538461539,
+      "grad_norm": 0.10360430926084518,
+      "learning_rate": 1.0538297872340425e-05,
+      "loss": 0.0041,
+      "step": 3550
+    },
+    {
+      "epoch": 34.23076923076923,
+      "grad_norm": 0.1899426281452179,
+      "learning_rate": 1.0474468085106383e-05,
+      "loss": 0.0049,
+      "step": 3560
+    },
+    {
+      "epoch": 34.32692307692308,
+      "grad_norm": 0.7377802133560181,
+      "learning_rate": 1.041063829787234e-05,
+      "loss": 0.0077,
+      "step": 3570
+    },
+    {
+      "epoch": 34.42307692307692,
+      "grad_norm": 0.056161731481552124,
+      "learning_rate": 1.0346808510638297e-05,
+      "loss": 0.0049,
+      "step": 3580
+    },
+    {
+      "epoch": 34.51923076923077,
+      "grad_norm": 0.7138869762420654,
+      "learning_rate": 1.0282978723404255e-05,
+      "loss": 0.0037,
+      "step": 3590
+    },
+    {
+      "epoch": 34.61538461538461,
+      "grad_norm": 0.47865352034568787,
+      "learning_rate": 1.0219148936170212e-05,
+      "loss": 0.0049,
+      "step": 3600
+    },
+    {
+      "epoch": 34.71153846153846,
+      "grad_norm": 0.032178666442632675,
+      "learning_rate": 1.015531914893617e-05,
+      "loss": 0.0045,
+      "step": 3610
+    },
+    {
+      "epoch": 34.80769230769231,
+      "grad_norm": 0.7013983130455017,
+      "learning_rate": 1.0091489361702128e-05,
+      "loss": 0.0043,
+      "step": 3620
+    },
+    {
+      "epoch": 34.90384615384615,
+      "grad_norm": 0.3409075438976288,
+      "learning_rate": 1.0027659574468085e-05,
+      "loss": 0.0046,
+      "step": 3630
+    },
+    {
+      "epoch": 35.0,
+      "grad_norm": 0.06704340875148773,
+      "learning_rate": 9.963829787234043e-06,
+      "loss": 0.0069,
+      "step": 3640
+    },
+    {
+      "epoch": 35.09615384615385,
+      "grad_norm": 1.4539157152175903,
+      "learning_rate": 9.9e-06,
+      "loss": 0.0047,
+      "step": 3650
+    },
+    {
+      "epoch": 35.19230769230769,
+      "grad_norm": 0.1617375612258911,
+      "learning_rate": 9.836170212765957e-06,
+      "loss": 0.0036,
+      "step": 3660
+    },
+    {
+      "epoch": 35.28846153846154,
+      "grad_norm": 0.190397247672081,
+      "learning_rate": 9.772340425531915e-06,
+      "loss": 0.0049,
+      "step": 3670
+    },
+    {
+      "epoch": 35.38461538461539,
+      "grad_norm": 0.31659790873527527,
+      "learning_rate": 9.708510638297872e-06,
+      "loss": 0.0064,
+      "step": 3680
+    },
+    {
+      "epoch": 35.48076923076923,
+      "grad_norm": 0.3006196618080139,
+      "learning_rate": 9.64468085106383e-06,
+      "loss": 0.0053,
+      "step": 3690
+    },
+    {
+      "epoch": 35.57692307692308,
+      "grad_norm": 0.07199200987815857,
+      "learning_rate": 9.580851063829788e-06,
+      "loss": 0.0045,
+      "step": 3700
+    },
+    {
+      "epoch": 35.67307692307692,
+      "grad_norm": 0.5229556560516357,
+      "learning_rate": 9.517021276595746e-06,
+      "loss": 0.0038,
+      "step": 3710
+    },
+    {
+      "epoch": 35.76923076923077,
+      "grad_norm": 0.6748898029327393,
+      "learning_rate": 9.453191489361703e-06,
+      "loss": 0.0086,
+      "step": 3720
+    },
+    {
+      "epoch": 35.86538461538461,
+      "grad_norm": 0.24701105058193207,
+      "learning_rate": 9.38936170212766e-06,
+      "loss": 0.0047,
+      "step": 3730
+    },
+    {
+      "epoch": 35.96153846153846,
+      "grad_norm": 0.4330107271671295,
+      "learning_rate": 9.325531914893617e-06,
+      "loss": 0.0039,
+      "step": 3740
+    },
+    {
+      "epoch": 36.05769230769231,
+      "grad_norm": 0.24795526266098022,
+      "learning_rate": 9.261702127659575e-06,
+      "loss": 0.0037,
+      "step": 3750
+    },
+    {
+      "epoch": 36.15384615384615,
+      "grad_norm": 0.5435231328010559,
+      "learning_rate": 9.197872340425532e-06,
+      "loss": 0.0047,
+      "step": 3760
+    },
+    {
+      "epoch": 36.25,
+      "grad_norm": 0.5274829864501953,
+      "learning_rate": 9.13404255319149e-06,
+      "loss": 0.0048,
+      "step": 3770
+    },
+    {
+      "epoch": 36.34615384615385,
+      "grad_norm": 0.40137821435928345,
+      "learning_rate": 9.070212765957447e-06,
+      "loss": 0.0054,
+      "step": 3780
+    },
+    {
+      "epoch": 36.44230769230769,
+      "grad_norm": 0.2212786227464676,
+      "learning_rate": 9.006382978723404e-06,
+      "loss": 0.0052,
+      "step": 3790
+    },
+    {
+      "epoch": 36.53846153846154,
+      "grad_norm": 0.43687814474105835,
+      "learning_rate": 8.942553191489363e-06,
+      "loss": 0.0035,
+      "step": 3800
+    },
+    {
+      "epoch": 36.63461538461539,
+      "grad_norm": 0.21894030272960663,
+      "learning_rate": 8.87872340425532e-06,
+      "loss": 0.0048,
+      "step": 3810
+    },
+    {
+      "epoch": 36.73076923076923,
+      "grad_norm": 1.4914783239364624,
+      "learning_rate": 8.814893617021278e-06,
+      "loss": 0.0037,
+      "step": 3820
+    },
+    {
+      "epoch": 36.82692307692308,
+      "grad_norm": 0.32890626788139343,
+      "learning_rate": 8.751063829787235e-06,
+      "loss": 0.0043,
+      "step": 3830
+    },
+    {
+      "epoch": 36.92307692307692,
+      "grad_norm": 0.3183184266090393,
+      "learning_rate": 8.687234042553192e-06,
+      "loss": 0.0026,
+      "step": 3840
+    },
+    {
+      "epoch": 37.01923076923077,
+      "grad_norm": 0.4841859042644501,
+      "learning_rate": 8.62340425531915e-06,
+      "loss": 0.003,
+      "step": 3850
+    },
+    {
+      "epoch": 37.11538461538461,
+      "grad_norm": 0.4076668620109558,
+      "learning_rate": 8.559574468085107e-06,
+      "loss": 0.0032,
+      "step": 3860
+    },
+    {
+      "epoch": 37.21153846153846,
+      "grad_norm": 0.05997830256819725,
+      "learning_rate": 8.495744680851064e-06,
+      "loss": 0.0043,
+      "step": 3870
+    },
+    {
+      "epoch": 37.30769230769231,
+      "grad_norm": 0.20216123759746552,
+      "learning_rate": 8.431914893617021e-06,
+      "loss": 0.003,
+      "step": 3880
+    },
+    {
+      "epoch": 37.40384615384615,
+      "grad_norm": 0.19002175331115723,
+      "learning_rate": 8.368085106382979e-06,
+      "loss": 0.0061,
+      "step": 3890
+    },
+    {
+      "epoch": 37.5,
+      "grad_norm": 0.051696181297302246,
+      "learning_rate": 8.304255319148936e-06,
+      "loss": 0.0026,
+      "step": 3900
+    },
+    {
+      "epoch": 37.59615384615385,
+      "grad_norm": 0.1388806402683258,
+      "learning_rate": 8.240425531914893e-06,
+      "loss": 0.0043,
+      "step": 3910
+    },
+    {
+      "epoch": 37.69230769230769,
+      "grad_norm": 0.24038566648960114,
+      "learning_rate": 8.17659574468085e-06,
+      "loss": 0.0059,
+      "step": 3920
+    },
+    {
+      "epoch": 37.78846153846154,
+      "grad_norm": 0.04175787419080734,
+      "learning_rate": 8.112765957446808e-06,
+      "loss": 0.0037,
+      "step": 3930
+    },
+    {
+      "epoch": 37.88461538461539,
+      "grad_norm": 0.15831300616264343,
+      "learning_rate": 8.048936170212765e-06,
+      "loss": 0.0048,
+      "step": 3940
+    },
+    {
+      "epoch": 37.98076923076923,
+      "grad_norm": 0.40547874569892883,
+      "learning_rate": 7.985106382978723e-06,
+      "loss": 0.0054,
+      "step": 3950
+    },
+    {
+      "epoch": 38.07692307692308,
+      "grad_norm": 0.4601777493953705,
+      "learning_rate": 7.92127659574468e-06,
+      "loss": 0.0046,
+      "step": 3960
+    },
+    {
+      "epoch": 38.17307692307692,
+      "grad_norm": 1.0779550075531006,
+      "learning_rate": 7.857446808510637e-06,
+      "loss": 0.0048,
+      "step": 3970
+    },
+    {
+      "epoch": 38.26923076923077,
+      "grad_norm": 1.490932822227478,
+      "learning_rate": 7.793617021276596e-06,
+      "loss": 0.0068,
+      "step": 3980
+    },
+    {
+      "epoch": 38.36538461538461,
+      "grad_norm": 0.7527866363525391,
+      "learning_rate": 7.729787234042553e-06,
+      "loss": 0.0044,
+      "step": 3990
+    },
+    {
+      "epoch": 38.46153846153846,
+      "grad_norm": 0.030520111322402954,
+      "learning_rate": 7.66595744680851e-06,
+      "loss": 0.0041,
+      "step": 4000
+    },
+    {
+      "epoch": 38.46153846153846,
+      "eval_correct_predictions": 1400,
+      "eval_exact_match_accuracy": 0.9893992932862191,
+      "eval_loss": 0.004963812418282032,
+      "eval_runtime": 75.6553,
+      "eval_samples_per_second": 18.703,
+      "eval_steps_per_second": 0.304,
+      "eval_total_predictions": 1415,
+      "step": 4000
+    },
+    {
+      "epoch": 38.55769230769231,
+      "grad_norm": 0.2399033159017563,
+      "learning_rate": 7.602127659574469e-06,
+      "loss": 0.0049,
+      "step": 4010
+    },
+    {
+      "epoch": 38.65384615384615,
+      "grad_norm": 0.300660103559494,
+      "learning_rate": 7.538297872340426e-06,
+      "loss": 0.0039,
+      "step": 4020
+    },
+    {
+      "epoch": 38.75,
+      "grad_norm": 0.08755325525999069,
+      "learning_rate": 7.4744680851063834e-06,
+      "loss": 0.0069,
+      "step": 4030
+    },
+    {
+      "epoch": 38.84615384615385,
+      "grad_norm": 0.5832145810127258,
+      "learning_rate": 7.410638297872341e-06,
+      "loss": 0.0051,
+      "step": 4040
+    },
+    {
+      "epoch": 38.94230769230769,
+      "grad_norm": 0.31291982531547546,
+      "learning_rate": 7.346808510638298e-06,
+      "loss": 0.003,
+      "step": 4050
+    },
+    {
+      "epoch": 39.03846153846154,
+      "grad_norm": 0.39460045099258423,
+      "learning_rate": 7.282978723404255e-06,
+      "loss": 0.0029,
+      "step": 4060
+    },
+    {
+      "epoch": 39.13461538461539,
+      "grad_norm": 0.4764561951160431,
+      "learning_rate": 7.219148936170213e-06,
+      "loss": 0.0031,
+      "step": 4070
+    },
+    {
+      "epoch": 39.23076923076923,
+      "grad_norm": 0.05070360749959946,
+      "learning_rate": 7.15531914893617e-06,
+      "loss": 0.0057,
+      "step": 4080
+    },
+    {
+      "epoch": 39.32692307692308,
+      "grad_norm": 0.23319944739341736,
+      "learning_rate": 7.091489361702127e-06,
+      "loss": 0.0046,
+      "step": 4090
+    },
+    {
+      "epoch": 39.42307692307692,
+      "grad_norm": 0.10411890596151352,
+      "learning_rate": 7.0276595744680846e-06,
+      "loss": 0.0038,
+      "step": 4100
+    },
+    {
+      "epoch": 39.51923076923077,
+      "grad_norm": 0.3427359163761139,
+      "learning_rate": 6.963829787234043e-06,
+      "loss": 0.0058,
+      "step": 4110
+    },
+    {
+      "epoch": 39.61538461538461,
+      "grad_norm": 0.7232850790023804,
+      "learning_rate": 6.900000000000001e-06,
+      "loss": 0.0035,
+      "step": 4120
+    },
+    {
+      "epoch": 39.71153846153846,
+      "grad_norm": 0.012613312341272831,
+      "learning_rate": 6.836170212765958e-06,
+      "loss": 0.0022,
+      "step": 4130
+    },
+    {
+      "epoch": 39.80769230769231,
+      "grad_norm": 0.922091543674469,
+      "learning_rate": 6.7723404255319155e-06,
+      "loss": 0.0048,
+      "step": 4140
+    },
+    {
+      "epoch": 39.90384615384615,
+      "grad_norm": 0.4784044325351715,
+      "learning_rate": 6.708510638297873e-06,
+      "loss": 0.0044,
+      "step": 4150
+    },
+    {
+      "epoch": 40.0,
+      "grad_norm": 4.750979900360107,
+      "learning_rate": 6.64468085106383e-06,
+      "loss": 0.0045,
+      "step": 4160
+    },
+    {
+      "epoch": 40.09615384615385,
+      "grad_norm": 0.5721331834793091,
+      "learning_rate": 6.580851063829787e-06,
+      "loss": 0.0061,
+      "step": 4170
+    },
+    {
+      "epoch": 40.19230769230769,
+      "grad_norm": 0.9367163777351379,
+      "learning_rate": 6.517021276595745e-06,
+      "loss": 0.0052,
+      "step": 4180
+    },
+    {
+      "epoch": 40.28846153846154,
+      "grad_norm": 0.008925567381083965,
+      "learning_rate": 6.453191489361702e-06,
+      "loss": 0.0043,
+      "step": 4190
+    },
+    {
+      "epoch": 40.38461538461539,
+      "grad_norm": 0.5253669023513794,
+      "learning_rate": 6.38936170212766e-06,
+      "loss": 0.0049,
+      "step": 4200
+    },
+    {
+      "epoch": 40.48076923076923,
+      "grad_norm": 0.048985060304403305,
+      "learning_rate": 6.325531914893617e-06,
+      "loss": 0.0064,
+      "step": 4210
+    },
+    {
+      "epoch": 40.57692307692308,
+      "grad_norm": 0.1342606544494629,
+      "learning_rate": 6.261702127659575e-06,
+      "loss": 0.0034,
+      "step": 4220
+    },
+    {
+      "epoch": 40.67307692307692,
+      "grad_norm": 0.06901761889457703,
+      "learning_rate": 6.197872340425532e-06,
+      "loss": 0.0035,
+      "step": 4230
+    },
+    {
+      "epoch": 40.76923076923077,
+      "grad_norm": 0.13464286923408508,
+      "learning_rate": 6.134042553191489e-06,
+      "loss": 0.005,
+      "step": 4240
+    },
+    {
+      "epoch": 40.86538461538461,
+      "grad_norm": 0.43832606077194214,
+      "learning_rate": 6.070212765957447e-06,
+      "loss": 0.0031,
+      "step": 4250
+    },
+    {
+      "epoch": 40.96153846153846,
+      "grad_norm": 0.44453272223472595,
+      "learning_rate": 6.006382978723404e-06,
+      "loss": 0.0048,
+      "step": 4260
+    },
+    {
+      "epoch": 41.05769230769231,
+      "grad_norm": 0.47965118288993835,
+      "learning_rate": 5.942553191489361e-06,
+      "loss": 0.0042,
+      "step": 4270
+    },
+    {
+      "epoch": 41.15384615384615,
+      "grad_norm": 0.10220217704772949,
+      "learning_rate": 5.878723404255319e-06,
+      "loss": 0.0049,
+      "step": 4280
+    },
+    {
+      "epoch": 41.25,
+      "grad_norm": 0.4104725420475006,
+      "learning_rate": 5.8148936170212775e-06,
+      "loss": 0.0036,
+      "step": 4290
+    },
+    {
+      "epoch": 41.34615384615385,
+      "grad_norm": 0.1370197832584381,
+      "learning_rate": 5.751063829787235e-06,
+      "loss": 0.0022,
+      "step": 4300
+    },
+    {
+      "epoch": 41.44230769230769,
+      "grad_norm": 0.6965671181678772,
+      "learning_rate": 5.687234042553192e-06,
+      "loss": 0.0047,
+      "step": 4310
+    },
+    {
+      "epoch": 41.53846153846154,
+      "grad_norm": 0.17734394967556,
+      "learning_rate": 5.623404255319149e-06,
+      "loss": 0.003,
+      "step": 4320
+    },
+    {
+      "epoch": 41.63461538461539,
+      "grad_norm": 0.01661546155810356,
+      "learning_rate": 5.559574468085107e-06,
+      "loss": 0.0038,
+      "step": 4330
+    },
+    {
+      "epoch": 41.73076923076923,
+      "grad_norm": 0.17337529361248016,
+      "learning_rate": 5.495744680851064e-06,
+      "loss": 0.0045,
+      "step": 4340
+    },
+    {
+      "epoch": 41.82692307692308,
+      "grad_norm": 0.40350341796875,
+      "learning_rate": 5.431914893617021e-06,
+      "loss": 0.0061,
+      "step": 4350
+    },
+    {
+      "epoch": 41.92307692307692,
+      "grad_norm": 0.17201651632785797,
+      "learning_rate": 5.368085106382979e-06,
+      "loss": 0.004,
+      "step": 4360
+    },
+    {
+      "epoch": 42.01923076923077,
+      "grad_norm": 0.5255150198936462,
+      "learning_rate": 5.304255319148936e-06,
+      "loss": 0.0033,
+      "step": 4370
+    },
+    {
+      "epoch": 42.11538461538461,
+      "grad_norm": 0.758439302444458,
+      "learning_rate": 5.240425531914894e-06,
+      "loss": 0.005,
+      "step": 4380
+    },
+    {
+      "epoch": 42.21153846153846,
+      "grad_norm": 0.7857176661491394,
+      "learning_rate": 5.176595744680851e-06,
+      "loss": 0.004,
+      "step": 4390
+    },
+    {
+      "epoch": 42.30769230769231,
+      "grad_norm": 0.38413724303245544,
+      "learning_rate": 5.112765957446809e-06,
+      "loss": 0.0022,
+      "step": 4400
+    },
+    {
+      "epoch": 42.40384615384615,
+      "grad_norm": 0.457057923078537,
+      "learning_rate": 5.048936170212766e-06,
+      "loss": 0.0041,
+      "step": 4410
+    },
+    {
+      "epoch": 42.5,
+      "grad_norm": 0.028009217232465744,
+      "learning_rate": 4.985106382978723e-06,
+      "loss": 0.0025,
+      "step": 4420
+    },
+    {
+      "epoch": 42.59615384615385,
+      "grad_norm": 0.7792872190475464,
+      "learning_rate": 4.9212765957446806e-06,
+      "loss": 0.0052,
+      "step": 4430
+    },
+    {
+      "epoch": 42.69230769230769,
+      "grad_norm": 0.46134594082832336,
+      "learning_rate": 4.857446808510638e-06,
+      "loss": 0.0017,
+      "step": 4440
+    },
+    {
+      "epoch": 42.78846153846154,
+      "grad_norm": 0.43057435750961304,
+      "learning_rate": 4.793617021276596e-06,
+      "loss": 0.0038,
+      "step": 4450
+    },
+    {
+      "epoch": 42.88461538461539,
+      "grad_norm": 0.2747887372970581,
+      "learning_rate": 4.729787234042553e-06,
+      "loss": 0.0052,
+      "step": 4460
+    },
+    {
+      "epoch": 42.98076923076923,
+      "grad_norm": 0.053347088396549225,
+      "learning_rate": 4.6659574468085115e-06,
+      "loss": 0.0046,
+      "step": 4470
+    },
+    {
+      "epoch": 43.07692307692308,
+      "grad_norm": 0.5245047807693481,
+      "learning_rate": 4.602127659574469e-06,
+      "loss": 0.004,
+      "step": 4480
+    },
+    {
+      "epoch": 43.17307692307692,
+      "grad_norm": 0.14071467518806458,
+      "learning_rate": 4.538297872340426e-06,
+      "loss": 0.0036,
+      "step": 4490
+    },
+    {
+      "epoch": 43.26923076923077,
+      "grad_norm": 0.5309870839118958,
+      "learning_rate": 4.474468085106383e-06,
+      "loss": 0.0056,
+      "step": 4500
+    },
+    {
+      "epoch": 43.36538461538461,
+      "grad_norm": 0.6158878803253174,
+      "learning_rate": 4.410638297872341e-06,
+      "loss": 0.0032,
+      "step": 4510
+    },
+    {
+      "epoch": 43.46153846153846,
+      "grad_norm": 0.3933374881744385,
+      "learning_rate": 4.346808510638298e-06,
+      "loss": 0.0027,
+      "step": 4520
+    },
+    {
+      "epoch": 43.55769230769231,
+      "grad_norm": 0.32318615913391113,
+      "learning_rate": 4.282978723404255e-06,
+      "loss": 0.0031,
+      "step": 4530
+    },
+    {
+      "epoch": 43.65384615384615,
+      "grad_norm": 0.4188336730003357,
+      "learning_rate": 4.219148936170213e-06,
+      "loss": 0.0038,
+      "step": 4540
+    },
+    {
+      "epoch": 43.75,
+      "grad_norm": 0.13132989406585693,
+      "learning_rate": 4.15531914893617e-06,
+      "loss": 0.0046,
+      "step": 4550
+    },
+    {
+      "epoch": 43.84615384615385,
+      "grad_norm": 0.5103133916854858,
+      "learning_rate": 4.091489361702128e-06,
+      "loss": 0.004,
+      "step": 4560
+    },
+    {
+      "epoch": 43.94230769230769,
+      "grad_norm": 0.11087504774332047,
+      "learning_rate": 4.027659574468085e-06,
+      "loss": 0.003,
+      "step": 4570
+    },
+    {
+      "epoch": 44.03846153846154,
+      "grad_norm": 0.9140055179595947,
+      "learning_rate": 3.963829787234043e-06,
+      "loss": 0.0038,
+      "step": 4580
+    },
+    {
+      "epoch": 44.13461538461539,
+      "grad_norm": 0.41617265343666077,
+      "learning_rate": 3.9e-06,
+      "loss": 0.0036,
+      "step": 4590
+    },
+    {
+      "epoch": 44.23076923076923,
+      "grad_norm": 0.14915060997009277,
+      "learning_rate": 3.836170212765957e-06,
+      "loss": 0.0026,
+      "step": 4600
+    },
+    {
+      "epoch": 44.32692307692308,
+      "grad_norm": 0.045890841633081436,
+      "learning_rate": 3.772340425531915e-06,
+      "loss": 0.0063,
+      "step": 4610
+    },
+    {
+      "epoch": 44.42307692307692,
+      "grad_norm": 0.35580122470855713,
+      "learning_rate": 3.7085106382978723e-06,
+      "loss": 0.0049,
+      "step": 4620
+    },
+    {
+      "epoch": 44.51923076923077,
+      "grad_norm": 0.26410701870918274,
+      "learning_rate": 3.64468085106383e-06,
+      "loss": 0.0029,
+      "step": 4630
+    },
+    {
+      "epoch": 44.61538461538461,
+      "grad_norm": 0.5982622504234314,
+      "learning_rate": 3.5808510638297873e-06,
+      "loss": 0.0023,
+      "step": 4640
+    },
+    {
+      "epoch": 44.71153846153846,
+      "grad_norm": 0.44741129875183105,
+      "learning_rate": 3.5170212765957446e-06,
+      "loss": 0.0039,
+      "step": 4650
+    },
+    {
+      "epoch": 44.80769230769231,
+      "grad_norm": 0.20336119830608368,
+      "learning_rate": 3.4531914893617023e-06,
+      "loss": 0.0025,
+      "step": 4660
+    },
+    {
+      "epoch": 44.90384615384615,
+      "grad_norm": 0.0518646277487278,
+      "learning_rate": 3.3893617021276596e-06,
+      "loss": 0.0031,
+      "step": 4670
+    },
+    {
+      "epoch": 45.0,
+      "grad_norm": 0.021453868597745895,
+      "learning_rate": 3.3255319148936173e-06,
+      "loss": 0.0038,
+      "step": 4680
+    },
+    {
+      "epoch": 45.09615384615385,
+      "grad_norm": 0.059526897966861725,
+      "learning_rate": 3.2617021276595746e-06,
+      "loss": 0.0044,
+      "step": 4690
+    },
+    {
+      "epoch": 45.19230769230769,
+      "grad_norm": 0.03833373263478279,
+      "learning_rate": 3.197872340425532e-06,
+      "loss": 0.0039,
+      "step": 4700
+    },
+    {
+      "epoch": 45.28846153846154,
+      "grad_norm": 0.054637033492326736,
+      "learning_rate": 3.1340425531914892e-06,
+      "loss": 0.0069,
+      "step": 4710
+    },
+    {
+      "epoch": 45.38461538461539,
+      "grad_norm": 0.32997268438339233,
+      "learning_rate": 3.070212765957447e-06,
+      "loss": 0.0033,
+      "step": 4720
+    },
+    {
+      "epoch": 45.48076923076923,
+      "grad_norm": 0.01344509981572628,
+      "learning_rate": 3.0063829787234043e-06,
+      "loss": 0.0047,
+      "step": 4730
+    },
+    {
+      "epoch": 45.57692307692308,
+      "grad_norm": 0.1493314951658249,
+      "learning_rate": 2.942553191489362e-06,
+      "loss": 0.0042,
+      "step": 4740
+    },
+    {
+      "epoch": 45.67307692307692,
+      "grad_norm": 0.5637978911399841,
+      "learning_rate": 2.8787234042553193e-06,
+      "loss": 0.0038,
+      "step": 4750
+    },
+    {
+      "epoch": 45.76923076923077,
+      "grad_norm": 0.8575944304466248,
+      "learning_rate": 2.8148936170212766e-06,
+      "loss": 0.0046,
+      "step": 4760
+    },
+    {
+      "epoch": 45.86538461538461,
+      "grad_norm": 0.810599148273468,
+      "learning_rate": 2.7510638297872343e-06,
+      "loss": 0.0035,
+      "step": 4770
+    },
+    {
+      "epoch": 45.96153846153846,
+      "grad_norm": 0.031988173723220825,
+      "learning_rate": 2.6872340425531916e-06,
+      "loss": 0.0022,
+      "step": 4780
+    },
+    {
+      "epoch": 46.05769230769231,
+      "grad_norm": 0.06916001439094543,
+      "learning_rate": 2.623404255319149e-06,
+      "loss": 0.004,
+      "step": 4790
+    },
+    {
+      "epoch": 46.15384615384615,
+      "grad_norm": 0.04660704359412193,
+      "learning_rate": 2.5595744680851062e-06,
+      "loss": 0.0027,
+      "step": 4800
+    },
+    {
+      "epoch": 46.25,
+      "grad_norm": 1.0712189674377441,
+      "learning_rate": 2.495744680851064e-06,
+      "loss": 0.0032,
+      "step": 4810
+    },
+    {
+      "epoch": 46.34615384615385,
+      "grad_norm": 0.17620952427387238,
+      "learning_rate": 2.4319148936170212e-06,
+      "loss": 0.002,
+      "step": 4820
+    },
+    {
+      "epoch": 46.44230769230769,
+      "grad_norm": 0.29380717873573303,
+      "learning_rate": 2.368085106382979e-06,
+      "loss": 0.0029,
+      "step": 4830
+    },
+    {
+      "epoch": 46.53846153846154,
+      "grad_norm": 0.015064992010593414,
+      "learning_rate": 2.3042553191489363e-06,
+      "loss": 0.0034,
+      "step": 4840
+    },
+    {
+      "epoch": 46.63461538461539,
+      "grad_norm": 0.9566193222999573,
+      "learning_rate": 2.2404255319148936e-06,
+      "loss": 0.0039,
+      "step": 4850
+    },
+    {
+      "epoch": 46.73076923076923,
+      "grad_norm": 0.4771270751953125,
+      "learning_rate": 2.1765957446808513e-06,
+      "loss": 0.0041,
+      "step": 4860
+    },
+    {
+      "epoch": 46.82692307692308,
+      "grad_norm": 0.10530088096857071,
+      "learning_rate": 2.1127659574468086e-06,
+      "loss": 0.0056,
+      "step": 4870
+    },
+    {
+      "epoch": 46.92307692307692,
+      "grad_norm": 0.38113558292388916,
+      "learning_rate": 2.048936170212766e-06,
+      "loss": 0.0044,
+      "step": 4880
+    },
+    {
+      "epoch": 47.01923076923077,
+      "grad_norm": 0.34304213523864746,
+      "learning_rate": 1.985106382978723e-06,
+      "loss": 0.0036,
+      "step": 4890
+    },
+    {
+      "epoch": 47.11538461538461,
+      "grad_norm": 0.3129338026046753,
+      "learning_rate": 1.921276595744681e-06,
+      "loss": 0.0045,
+      "step": 4900
+    },
+    {
+      "epoch": 47.21153846153846,
+      "grad_norm": 0.04093209281563759,
+      "learning_rate": 1.8574468085106384e-06,
+      "loss": 0.0044,
+      "step": 4910
+    },
+    {
+      "epoch": 47.30769230769231,
+      "grad_norm": 0.19558611512184143,
+      "learning_rate": 1.7936170212765957e-06,
+      "loss": 0.0038,
+      "step": 4920
+    },
+    {
+      "epoch": 47.40384615384615,
+      "grad_norm": 0.07514757663011551,
+      "learning_rate": 1.7297872340425532e-06,
+      "loss": 0.0024,
+      "step": 4930
+    },
+    {
+      "epoch": 47.5,
+      "grad_norm": 0.6852735877037048,
+      "learning_rate": 1.6659574468085108e-06,
+      "loss": 0.0031,
+      "step": 4940
+    },
+    {
+      "epoch": 47.59615384615385,
+      "grad_norm": 0.6539517641067505,
+      "learning_rate": 1.602127659574468e-06,
+      "loss": 0.0039,
+      "step": 4950
+    },
+    {
+      "epoch": 47.69230769230769,
+      "grad_norm": 0.06155241280794144,
+      "learning_rate": 1.5382978723404256e-06,
+      "loss": 0.0031,
+      "step": 4960
+    },
+    {
+      "epoch": 47.78846153846154,
+      "grad_norm": 0.3951682150363922,
+      "learning_rate": 1.4744680851063829e-06,
+      "loss": 0.0032,
+      "step": 4970
+    },
+    {
+      "epoch": 47.88461538461539,
+      "grad_norm": 0.0964864119887352,
+      "learning_rate": 1.4106382978723406e-06,
+      "loss": 0.002,
+      "step": 4980
+    },
+    {
+      "epoch": 47.98076923076923,
+      "grad_norm": 0.17981211841106415,
+      "learning_rate": 1.346808510638298e-06,
+      "loss": 0.003,
+      "step": 4990
+    },
+    {
+      "epoch": 48.07692307692308,
+      "grad_norm": 0.7259301543235779,
+      "learning_rate": 1.2829787234042554e-06,
+      "loss": 0.0028,
+      "step": 5000
+    },
+    {
+      "epoch": 48.07692307692308,
+      "eval_correct_predictions": 1404,
+      "eval_exact_match_accuracy": 0.992226148409894,
+      "eval_loss": 0.004099734127521515,
+      "eval_runtime": 71.3472,
+      "eval_samples_per_second": 19.833,
+      "eval_steps_per_second": 0.322,
+      "eval_total_predictions": 1415,
+      "step": 5000
+    },
+    {
+      "epoch": 48.17307692307692,
+      "grad_norm": 0.2118825614452362,
+      "learning_rate": 1.2191489361702127e-06,
+      "loss": 0.0032,
+      "step": 5010
+    },
+    {
+      "epoch": 48.26923076923077,
+      "grad_norm": 1.0008749961853027,
+      "learning_rate": 1.1553191489361702e-06,
+      "loss": 0.0028,
+      "step": 5020
+    },
+    {
+      "epoch": 48.36538461538461,
+      "grad_norm": 0.6345369815826416,
+      "learning_rate": 1.0914893617021277e-06,
+      "loss": 0.0032,
+      "step": 5030
+    },
+    {
+      "epoch": 48.46153846153846,
+      "grad_norm": 0.7323829531669617,
+      "learning_rate": 1.027659574468085e-06,
+      "loss": 0.006,
+      "step": 5040
+    },
+    {
+      "epoch": 48.55769230769231,
+      "grad_norm": 0.22454875707626343,
+      "learning_rate": 9.638297872340426e-07,
+      "loss": 0.0043,
+      "step": 5050
+    },
+    {
+      "epoch": 48.65384615384615,
+      "grad_norm": 0.4376821219921112,
+      "learning_rate": 9e-07,
+      "loss": 0.0028,
+      "step": 5060
+    },
+    {
+      "epoch": 48.75,
+      "grad_norm": 0.21368823945522308,
+      "learning_rate": 8.361702127659575e-07,
+      "loss": 0.0027,
+      "step": 5070
+    },
+    {
+      "epoch": 48.84615384615385,
+      "grad_norm": 0.6260131001472473,
+      "learning_rate": 7.723404255319149e-07,
+      "loss": 0.0032,
+      "step": 5080
+    },
+    {
+      "epoch": 48.94230769230769,
+      "grad_norm": 0.27165454626083374,
+      "learning_rate": 7.085106382978724e-07,
+      "loss": 0.0057,
+      "step": 5090
+    },
+    {
+      "epoch": 49.03846153846154,
+      "grad_norm": 0.3616870641708374,
+      "learning_rate": 6.446808510638298e-07,
+      "loss": 0.0076,
+      "step": 5100
+    },
+    {
+      "epoch": 49.13461538461539,
+      "grad_norm": 0.1850527971982956,
+      "learning_rate": 5.808510638297873e-07,
+      "loss": 0.0029,
+      "step": 5110
+    },
+    {
+      "epoch": 49.23076923076923,
+      "grad_norm": 0.09594523161649704,
+      "learning_rate": 5.170212765957447e-07,
+      "loss": 0.0053,
+      "step": 5120
+    },
+    {
+      "epoch": 49.32692307692308,
+      "grad_norm": 0.2822889983654022,
+      "learning_rate": 4.531914893617021e-07,
+      "loss": 0.0034,
+      "step": 5130
+    },
+    {
+      "epoch": 49.42307692307692,
+      "grad_norm": 0.37851470708847046,
+      "learning_rate": 3.893617021276596e-07,
+      "loss": 0.005,
+      "step": 5140
+    },
+    {
+      "epoch": 49.51923076923077,
+      "grad_norm": 0.257566899061203,
+      "learning_rate": 3.2553191489361704e-07,
+      "loss": 0.0028,
+      "step": 5150
+    },
+    {
+      "epoch": 49.61538461538461,
+      "grad_norm": 0.022371169179677963,
+      "learning_rate": 2.6170212765957445e-07,
+      "loss": 0.0028,
+      "step": 5160
+    },
+    {
+      "epoch": 49.71153846153846,
+      "grad_norm": 0.9283726215362549,
+      "learning_rate": 1.978723404255319e-07,
+      "loss": 0.0038,
+      "step": 5170
+    },
+    {
+      "epoch": 49.80769230769231,
+      "grad_norm": 0.5324678421020508,
+      "learning_rate": 1.3404255319148937e-07,
+      "loss": 0.0022,
+      "step": 5180
+    },
+    {
+      "epoch": 49.90384615384615,
+      "grad_norm": 0.11308519542217255,
+      "learning_rate": 7.021276595744682e-08,
+      "loss": 0.0023,
+      "step": 5190
+    },
+    {
+      "epoch": 50.0,
+      "grad_norm": 0.00943013746291399,
+      "learning_rate": 6.3829787234042555e-09,
+      "loss": 0.0022,
+      "step": 5200
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 5200,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 50,
+  "save_steps": 1000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2.51195129856e+16,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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

upload.py

@@ -0,0 +1,25 @@
+import os
+from huggingface_hub import HfApi, upload_folder
+
+token = os.environ.get("HF_TOKEN")
+if not token:
+    raise SystemExit("No HF token provided. Set HF_TOKEN env var.")
+
+api = HfApi(token=token)
+user = api.whoami()['name']
+repo_name = "t5-spelling-corrector-finetuned-v3"
+repo_id = f"{user}/{repo_name}"
+
+print(f"Creating repo: {repo_id}")
+api.create_repo(repo_id=repo_id, repo_type="model", exist_ok=True, private=False)
+
+folder_path = "."  # current directory
+print(f"Uploading folder {folder_path} to {repo_id} ...")
+upload_folder(
+    folder_path=folder_path,
+    repo_id=repo_id,
+    repo_type="model",
+    token=token,
+    commit_message="Upload T5 spelling corrector checkpoint from local"
+)
+print(f"Upload complete. Model available at: https://huggingface.co/{repo_id}")