diff --git "a/checkpoint-3200/trainer_state.json" "b/checkpoint-3200/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-3200/trainer_state.json"
@@ -0,0 +1,3072 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.257985257985258,
+  "global_step": 3200,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 0.0002,
+      "loss": 0.6932,
+      "step": 10
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 0.0002,
+      "loss": 0.6073,
+      "step": 20
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 0.0002,
+      "loss": 0.5435,
+      "step": 30
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 0.0002,
+      "loss": 0.5599,
+      "step": 40
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 0.0002,
+      "loss": 0.5723,
+      "step": 50
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 0.0002,
+      "loss": 0.5773,
+      "step": 60
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 0.0002,
+      "loss": 0.574,
+      "step": 70
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 0.0002,
+      "loss": 0.5357,
+      "step": 80
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 0.0002,
+      "loss": 0.5965,
+      "step": 90
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 0.0002,
+      "loss": 0.5547,
+      "step": 100
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 0.0002,
+      "loss": 0.5836,
+      "step": 110
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 0.0002,
+      "loss": 0.5533,
+      "step": 120
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 0.0002,
+      "loss": 0.5412,
+      "step": 130
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 0.0002,
+      "loss": 0.5535,
+      "step": 140
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 0.0002,
+      "loss": 0.5621,
+      "step": 150
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 0.0002,
+      "loss": 0.5295,
+      "step": 160
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 0.0002,
+      "loss": 0.5335,
+      "step": 170
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 0.0002,
+      "loss": 0.5647,
+      "step": 180
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 0.0002,
+      "loss": 0.5521,
+      "step": 190
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 0.0002,
+      "loss": 0.5781,
+      "step": 200
+    },
+    {
+      "epoch": 0.08,
+      "eval_loss": 0.5585418939590454,
+      "eval_runtime": 140.8898,
+      "eval_samples_per_second": 7.098,
+      "eval_steps_per_second": 3.549,
+      "step": 200
+    },
+    {
+      "epoch": 0.08,
+      "mmlu_eval_accuracy": 0.4614629836514187,
+      "mmlu_eval_accuracy_abstract_algebra": 0.18181818181818182,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.41379310344827586,
+      "mmlu_eval_accuracy_college_biology": 0.5,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.2727272727272727,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.5625,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3902439024390244,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.36363636363636365,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.5555555555555556,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.32558139534883723,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.27586206896551724,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.46153846153846156,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7,
+      "mmlu_eval_accuracy_high_school_statistics": 0.30434782608695654,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.2727272727272727,
+      "mmlu_eval_accuracy_management": 0.45454545454545453,
+      "mmlu_eval_accuracy_marketing": 0.72,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.5,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.3411764705882353,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.36231884057971014,
+      "mmlu_eval_accuracy_public_relations": 0.5833333333333334,
+      "mmlu_eval_accuracy_security_studies": 0.48148148148148145,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.5454545454545454,
+      "mmlu_eval_accuracy_virology": 0.3888888888888889,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 1.1617055672267087,
+      "step": 200
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 0.0002,
+      "loss": 0.574,
+      "step": 210
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 0.0002,
+      "loss": 0.5392,
+      "step": 220
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 0.0002,
+      "loss": 0.529,
+      "step": 230
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 0.0002,
+      "loss": 0.5698,
+      "step": 240
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 0.0002,
+      "loss": 0.5554,
+      "step": 250
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 0.0002,
+      "loss": 0.5338,
+      "step": 260
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 0.0002,
+      "loss": 0.5509,
+      "step": 270
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 0.0002,
+      "loss": 0.561,
+      "step": 280
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 0.0002,
+      "loss": 0.5023,
+      "step": 290
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 0.0002,
+      "loss": 0.5104,
+      "step": 300
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 0.0002,
+      "loss": 0.5603,
+      "step": 310
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 0.0002,
+      "loss": 0.5172,
+      "step": 320
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 0.0002,
+      "loss": 0.5496,
+      "step": 330
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 0.0002,
+      "loss": 0.4928,
+      "step": 340
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 0.0002,
+      "loss": 0.5105,
+      "step": 350
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 0.0002,
+      "loss": 0.505,
+      "step": 360
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 0.0002,
+      "loss": 0.5405,
+      "step": 370
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 0.0002,
+      "loss": 0.5046,
+      "step": 380
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 0.0002,
+      "loss": 0.4975,
+      "step": 390
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 0.0002,
+      "loss": 0.485,
+      "step": 400
+    },
+    {
+      "epoch": 0.16,
+      "eval_loss": 0.5374600291252136,
+      "eval_runtime": 140.7817,
+      "eval_samples_per_second": 7.103,
+      "eval_steps_per_second": 3.552,
+      "step": 400
+    },
+    {
+      "epoch": 0.16,
+      "mmlu_eval_accuracy": 0.46126679238699003,
+      "mmlu_eval_accuracy_abstract_algebra": 0.2727272727272727,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.6363636363636364,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.2727272727272727,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3902439024390244,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.4,
+      "mmlu_eval_accuracy_high_school_biology": 0.375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.4090909090909091,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.5555555555555556,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.32558139534883723,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.3103448275862069,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.4230769230769231,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.65,
+      "mmlu_eval_accuracy_high_school_statistics": 0.34782608695652173,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5769230769230769,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.2727272727272727,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.72,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.39473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.48484848484848486,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.45714285714285713,
+      "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
+      "mmlu_eval_accuracy_professional_law": 0.34705882352941175,
+      "mmlu_eval_accuracy_professional_medicine": 0.45161290322580644,
+      "mmlu_eval_accuracy_professional_psychology": 0.34782608695652173,
+      "mmlu_eval_accuracy_public_relations": 0.6666666666666666,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 1.05872035236645,
+      "step": 400
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 0.0002,
+      "loss": 0.4955,
+      "step": 410
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 0.0002,
+      "loss": 0.5569,
+      "step": 420
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 0.0002,
+      "loss": 0.5319,
+      "step": 430
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 0.0002,
+      "loss": 0.522,
+      "step": 440
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 0.0002,
+      "loss": 0.5343,
+      "step": 450
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 0.0002,
+      "loss": 0.5564,
+      "step": 460
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 0.0002,
+      "loss": 0.5689,
+      "step": 470
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 0.0002,
+      "loss": 0.5227,
+      "step": 480
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 0.0002,
+      "loss": 0.5046,
+      "step": 490
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 0.0002,
+      "loss": 0.4907,
+      "step": 500
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 0.0002,
+      "loss": 0.5321,
+      "step": 510
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 0.0002,
+      "loss": 0.4803,
+      "step": 520
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 0.0002,
+      "loss": 0.5385,
+      "step": 530
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 0.0002,
+      "loss": 0.4933,
+      "step": 540
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 0.0002,
+      "loss": 0.5386,
+      "step": 550
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 0.0002,
+      "loss": 0.5205,
+      "step": 560
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 0.0002,
+      "loss": 0.5174,
+      "step": 570
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 0.0002,
+      "loss": 0.5082,
+      "step": 580
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 0.0002,
+      "loss": 0.5137,
+      "step": 590
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 0.0002,
+      "loss": 0.5011,
+      "step": 600
+    },
+    {
+      "epoch": 0.24,
+      "eval_loss": 0.528292179107666,
+      "eval_runtime": 140.9673,
+      "eval_samples_per_second": 7.094,
+      "eval_steps_per_second": 3.547,
+      "step": 600
+    },
+    {
+      "epoch": 0.24,
+      "mmlu_eval_accuracy": 0.46265184845741664,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.6363636363636364,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.2727272727272727,
+      "mmlu_eval_accuracy_conceptual_physics": 0.34615384615384615,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3170731707317073,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5555555555555556,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.3953488372093023,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.2413793103448276,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.46153846153846156,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.6833333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.30434782608695654,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.2727272727272727,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6627906976744186,
+      "mmlu_eval_accuracy_moral_disputes": 0.47368421052631576,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.25806451612903225,
+      "mmlu_eval_accuracy_professional_law": 0.3588235294117647,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.37681159420289856,
+      "mmlu_eval_accuracy_public_relations": 0.6666666666666666,
+      "mmlu_eval_accuracy_security_studies": 0.5555555555555556,
+      "mmlu_eval_accuracy_sociology": 0.5909090909090909,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9400671155388922,
+      "step": 600
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 0.0002,
+      "loss": 0.5095,
+      "step": 610
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 0.0002,
+      "loss": 0.5239,
+      "step": 620
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 0.0002,
+      "loss": 0.5405,
+      "step": 630
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 0.0002,
+      "loss": 0.545,
+      "step": 640
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 0.0002,
+      "loss": 0.5304,
+      "step": 650
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 0.0002,
+      "loss": 0.491,
+      "step": 660
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 0.0002,
+      "loss": 0.4936,
+      "step": 670
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 0.0002,
+      "loss": 0.5233,
+      "step": 680
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 0.0002,
+      "loss": 0.5436,
+      "step": 690
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 0.0002,
+      "loss": 0.5214,
+      "step": 700
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 0.0002,
+      "loss": 0.5166,
+      "step": 710
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 0.0002,
+      "loss": 0.5098,
+      "step": 720
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 0.0002,
+      "loss": 0.528,
+      "step": 730
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 0.0002,
+      "loss": 0.4786,
+      "step": 740
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 0.0002,
+      "loss": 0.4749,
+      "step": 750
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 0.0002,
+      "loss": 0.5029,
+      "step": 760
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 0.0002,
+      "loss": 0.5097,
+      "step": 770
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 0.0002,
+      "loss": 0.4991,
+      "step": 780
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 0.0002,
+      "loss": 0.4829,
+      "step": 790
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 0.0002,
+      "loss": 0.5141,
+      "step": 800
+    },
+    {
+      "epoch": 0.31,
+      "eval_loss": 0.5220857262611389,
+      "eval_runtime": 140.3492,
+      "eval_samples_per_second": 7.125,
+      "eval_steps_per_second": 3.563,
+      "step": 800
+    },
+    {
+      "epoch": 0.31,
+      "mmlu_eval_accuracy": 0.4514231901208302,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.5625,
+      "mmlu_eval_accuracy_business_ethics": 0.6363636363636364,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.41379310344827586,
+      "mmlu_eval_accuracy_college_biology": 0.375,
+      "mmlu_eval_accuracy_college_chemistry": 0.0,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.45454545454545453,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.4146341463414634,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.3125,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.36363636363636365,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7272727272727273,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.37209302325581395,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.1724137931034483,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.4230769230769231,
+      "mmlu_eval_accuracy_high_school_physics": 0.29411764705882354,
+      "mmlu_eval_accuracy_high_school_psychology": 0.6833333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.30434782608695654,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6363636363636364,
+      "mmlu_eval_accuracy_high_school_world_history": 0.4230769230769231,
+      "mmlu_eval_accuracy_human_aging": 0.5652173913043478,
+      "mmlu_eval_accuracy_human_sexuality": 0.3333333333333333,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.6111111111111112,
+      "mmlu_eval_accuracy_machine_learning": 0.2727272727272727,
+      "mmlu_eval_accuracy_management": 0.45454545454545453,
+      "mmlu_eval_accuracy_marketing": 0.72,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.4411764705882353,
+      "mmlu_eval_accuracy_prehistory": 0.42857142857142855,
+      "mmlu_eval_accuracy_professional_accounting": 0.25806451612903225,
+      "mmlu_eval_accuracy_professional_law": 0.35294117647058826,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.42028985507246375,
+      "mmlu_eval_accuracy_public_relations": 0.5833333333333334,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9901202229233386,
+      "step": 800
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 0.0002,
+      "loss": 0.4894,
+      "step": 810
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 0.0002,
+      "loss": 0.5024,
+      "step": 820
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 0.0002,
+      "loss": 0.5535,
+      "step": 830
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 0.0002,
+      "loss": 0.5106,
+      "step": 840
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 0.0002,
+      "loss": 0.4816,
+      "step": 850
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 0.0002,
+      "loss": 0.4978,
+      "step": 860
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 0.0002,
+      "loss": 0.5321,
+      "step": 870
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 0.0002,
+      "loss": 0.5151,
+      "step": 880
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 0.0002,
+      "loss": 0.4796,
+      "step": 890
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 0.0002,
+      "loss": 0.4634,
+      "step": 900
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 0.0002,
+      "loss": 0.5037,
+      "step": 910
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 0.0002,
+      "loss": 0.5253,
+      "step": 920
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 0.0002,
+      "loss": 0.5273,
+      "step": 930
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 0.0002,
+      "loss": 0.509,
+      "step": 940
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 0.0002,
+      "loss": 0.5113,
+      "step": 950
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 0.0002,
+      "loss": 0.5231,
+      "step": 960
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 0.0002,
+      "loss": 0.5452,
+      "step": 970
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 0.0002,
+      "loss": 0.4881,
+      "step": 980
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 0.0002,
+      "loss": 0.4704,
+      "step": 990
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 0.0002,
+      "loss": 0.4986,
+      "step": 1000
+    },
+    {
+      "epoch": 0.39,
+      "eval_loss": 0.5142421722412109,
+      "eval_runtime": 140.711,
+      "eval_samples_per_second": 7.107,
+      "eval_steps_per_second": 3.553,
+      "step": 1000
+    },
+    {
+      "epoch": 0.39,
+      "mmlu_eval_accuracy": 0.4535898038210374,
+      "mmlu_eval_accuracy_abstract_algebra": 0.2727272727272727,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.5625,
+      "mmlu_eval_accuracy_business_ethics": 0.7272727272727273,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.41379310344827586,
+      "mmlu_eval_accuracy_college_biology": 0.375,
+      "mmlu_eval_accuracy_college_chemistry": 0.0,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.36363636363636365,
+      "mmlu_eval_accuracy_computer_security": 0.36363636363636365,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.3125,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.36585365853658536,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.3125,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.4090909090909091,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7272727272727273,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.37209302325581395,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.13793103448275862,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.34615384615384615,
+      "mmlu_eval_accuracy_high_school_physics": 0.29411764705882354,
+      "mmlu_eval_accuracy_high_school_psychology": 0.65,
+      "mmlu_eval_accuracy_high_school_statistics": 0.34782608695652173,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6363636363636364,
+      "mmlu_eval_accuracy_high_school_world_history": 0.46153846153846156,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.6111111111111112,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.42105263157894735,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5151515151515151,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.45714285714285713,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.36470588235294116,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.42028985507246375,
+      "mmlu_eval_accuracy_public_relations": 0.5833333333333334,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9794330676292937,
+      "step": 1000
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 0.0002,
+      "loss": 0.5058,
+      "step": 1010
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 0.0002,
+      "loss": 0.5282,
+      "step": 1020
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 0.0002,
+      "loss": 0.5007,
+      "step": 1030
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 0.0002,
+      "loss": 0.4656,
+      "step": 1040
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 0.0002,
+      "loss": 0.4681,
+      "step": 1050
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 0.0002,
+      "loss": 0.4998,
+      "step": 1060
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 0.0002,
+      "loss": 0.4755,
+      "step": 1070
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 0.0002,
+      "loss": 0.4527,
+      "step": 1080
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 0.0002,
+      "loss": 0.4844,
+      "step": 1090
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 0.0002,
+      "loss": 0.5016,
+      "step": 1100
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 0.0002,
+      "loss": 0.5044,
+      "step": 1110
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 0.0002,
+      "loss": 0.4723,
+      "step": 1120
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 0.0002,
+      "loss": 0.4621,
+      "step": 1130
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 0.0002,
+      "loss": 0.5033,
+      "step": 1140
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 0.0002,
+      "loss": 0.4935,
+      "step": 1150
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 0.0002,
+      "loss": 0.4973,
+      "step": 1160
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 0.0002,
+      "loss": 0.4953,
+      "step": 1170
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 0.0002,
+      "loss": 0.4667,
+      "step": 1180
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 0.0002,
+      "loss": 0.492,
+      "step": 1190
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 0.0002,
+      "loss": 0.4517,
+      "step": 1200
+    },
+    {
+      "epoch": 0.47,
+      "eval_loss": 0.5107662081718445,
+      "eval_runtime": 140.48,
+      "eval_samples_per_second": 7.118,
+      "eval_steps_per_second": 3.559,
+      "step": 1200
+    },
+    {
+      "epoch": 0.47,
+      "mmlu_eval_accuracy": 0.46011784666800093,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.4375,
+      "mmlu_eval_accuracy_business_ethics": 0.6363636363636364,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4827586206896552,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.0,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.36363636363636365,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.3125,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.36585365853658536,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.4375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.4090909090909091,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.37209302325581395,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.27586206896551724,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.4230769230769231,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7,
+      "mmlu_eval_accuracy_high_school_statistics": 0.30434782608695654,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.3333333333333333,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.36363636363636365,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6046511627906976,
+      "mmlu_eval_accuracy_moral_disputes": 0.47368421052631576,
+      "mmlu_eval_accuracy_moral_scenarios": 0.23,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.45714285714285713,
+      "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
+      "mmlu_eval_accuracy_professional_law": 0.36470588235294116,
+      "mmlu_eval_accuracy_professional_medicine": 0.45161290322580644,
+      "mmlu_eval_accuracy_professional_psychology": 0.37681159420289856,
+      "mmlu_eval_accuracy_public_relations": 0.5833333333333334,
+      "mmlu_eval_accuracy_security_studies": 0.4444444444444444,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 1.0901978164133141,
+      "step": 1200
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 0.0002,
+      "loss": 0.472,
+      "step": 1210
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 0.0002,
+      "loss": 0.5237,
+      "step": 1220
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 0.0002,
+      "loss": 0.4689,
+      "step": 1230
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 0.0002,
+      "loss": 0.48,
+      "step": 1240
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 0.0002,
+      "loss": 0.4974,
+      "step": 1250
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 0.0002,
+      "loss": 0.4717,
+      "step": 1260
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 0.0002,
+      "loss": 0.4356,
+      "step": 1270
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 0.0002,
+      "loss": 0.4869,
+      "step": 1280
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 0.0002,
+      "loss": 0.4575,
+      "step": 1290
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 0.0002,
+      "loss": 0.4651,
+      "step": 1300
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 0.0002,
+      "loss": 0.4947,
+      "step": 1310
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 0.0002,
+      "loss": 0.448,
+      "step": 1320
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 0.0002,
+      "loss": 0.4604,
+      "step": 1330
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 0.0002,
+      "loss": 0.5074,
+      "step": 1340
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 0.0002,
+      "loss": 0.4455,
+      "step": 1350
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 0.0002,
+      "loss": 0.4815,
+      "step": 1360
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 0.0002,
+      "loss": 0.5007,
+      "step": 1370
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 0.0002,
+      "loss": 0.4924,
+      "step": 1380
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 0.0002,
+      "loss": 0.5106,
+      "step": 1390
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 0.0002,
+      "loss": 0.4998,
+      "step": 1400
+    },
+    {
+      "epoch": 0.55,
+      "eval_loss": 0.5023671984672546,
+      "eval_runtime": 141.3273,
+      "eval_samples_per_second": 7.076,
+      "eval_steps_per_second": 3.538,
+      "step": 1400
+    },
+    {
+      "epoch": 0.55,
+      "mmlu_eval_accuracy": 0.4616336182493948,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.4375,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.5172413793103449,
+      "mmlu_eval_accuracy_college_biology": 0.375,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.45454545454545453,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.25,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.2926829268292683,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.4,
+      "mmlu_eval_accuracy_high_school_biology": 0.4375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.2727272727272727,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6666666666666666,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.4186046511627907,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.20689655172413793,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5,
+      "mmlu_eval_accuracy_high_school_physics": 0.5294117647058824,
+      "mmlu_eval_accuracy_high_school_psychology": 0.75,
+      "mmlu_eval_accuracy_high_school_statistics": 0.2608695652173913,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.3333333333333333,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.64,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.47368421052631576,
+      "mmlu_eval_accuracy_moral_scenarios": 0.23,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.5,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.34705882352941175,
+      "mmlu_eval_accuracy_professional_medicine": 0.45161290322580644,
+      "mmlu_eval_accuracy_professional_psychology": 0.42028985507246375,
+      "mmlu_eval_accuracy_public_relations": 0.5,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9466302770703019,
+      "step": 1400
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 0.0002,
+      "loss": 0.4895,
+      "step": 1410
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 0.0002,
+      "loss": 0.4573,
+      "step": 1420
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 0.0002,
+      "loss": 0.4813,
+      "step": 1430
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 0.0002,
+      "loss": 0.4755,
+      "step": 1440
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 0.0002,
+      "loss": 0.4699,
+      "step": 1450
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 0.0002,
+      "loss": 0.5089,
+      "step": 1460
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 0.0002,
+      "loss": 0.5035,
+      "step": 1470
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 0.0002,
+      "loss": 0.4757,
+      "step": 1480
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 0.0002,
+      "loss": 0.5067,
+      "step": 1490
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 0.0002,
+      "loss": 0.4848,
+      "step": 1500
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 0.0002,
+      "loss": 0.5219,
+      "step": 1510
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 0.0002,
+      "loss": 0.5037,
+      "step": 1520
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 0.0002,
+      "loss": 0.4931,
+      "step": 1530
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 0.0002,
+      "loss": 0.4946,
+      "step": 1540
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 0.0002,
+      "loss": 0.4873,
+      "step": 1550
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 0.0002,
+      "loss": 0.4809,
+      "step": 1560
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 0.0002,
+      "loss": 0.4705,
+      "step": 1570
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 0.0002,
+      "loss": 0.4921,
+      "step": 1580
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 0.0002,
+      "loss": 0.5159,
+      "step": 1590
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 0.0002,
+      "loss": 0.494,
+      "step": 1600
+    },
+    {
+      "epoch": 0.63,
+      "eval_loss": 0.49854931235313416,
+      "eval_runtime": 140.5627,
+      "eval_samples_per_second": 7.114,
+      "eval_steps_per_second": 3.557,
+      "step": 1600
+    },
+    {
+      "epoch": 0.63,
+      "mmlu_eval_accuracy": 0.45775551084287,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.6428571428571429,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.3793103448275862,
+      "mmlu_eval_accuracy_college_biology": 0.3125,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_medicine": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.36363636363636365,
+      "mmlu_eval_accuracy_conceptual_physics": 0.38461538461538464,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.3125,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3902439024390244,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.36363636363636365,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.3953488372093023,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.2413793103448276,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.4230769230769231,
+      "mmlu_eval_accuracy_high_school_physics": 0.29411764705882354,
+      "mmlu_eval_accuracy_high_school_psychology": 0.6833333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.43478260869565216,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.6923076923076923,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.6111111111111112,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6395348837209303,
+      "mmlu_eval_accuracy_moral_disputes": 0.42105263157894735,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.5,
+      "mmlu_eval_accuracy_prehistory": 0.45714285714285713,
+      "mmlu_eval_accuracy_professional_accounting": 0.25806451612903225,
+      "mmlu_eval_accuracy_professional_law": 0.3411764705882353,
+      "mmlu_eval_accuracy_professional_medicine": 0.3870967741935484,
+      "mmlu_eval_accuracy_professional_psychology": 0.36231884057971014,
+      "mmlu_eval_accuracy_public_relations": 0.5,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5909090909090909,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9470580935517117,
+      "step": 1600
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 0.0002,
+      "loss": 0.4774,
+      "step": 1610
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 0.0002,
+      "loss": 0.4911,
+      "step": 1620
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 0.0002,
+      "loss": 0.4874,
+      "step": 1630
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 0.0002,
+      "loss": 0.4802,
+      "step": 1640
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 0.0002,
+      "loss": 0.5123,
+      "step": 1650
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 0.0002,
+      "loss": 0.4355,
+      "step": 1660
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 0.0002,
+      "loss": 0.4814,
+      "step": 1670
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 0.0002,
+      "loss": 0.4633,
+      "step": 1680
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 0.0002,
+      "loss": 0.4738,
+      "step": 1690
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 0.0002,
+      "loss": 0.4672,
+      "step": 1700
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 0.0002,
+      "loss": 0.4388,
+      "step": 1710
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 0.0002,
+      "loss": 0.4687,
+      "step": 1720
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 0.0002,
+      "loss": 0.4645,
+      "step": 1730
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 0.0002,
+      "loss": 0.4499,
+      "step": 1740
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 0.0002,
+      "loss": 0.4596,
+      "step": 1750
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 0.0002,
+      "loss": 0.5131,
+      "step": 1760
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 0.0002,
+      "loss": 0.4887,
+      "step": 1770
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 0.0002,
+      "loss": 0.4706,
+      "step": 1780
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 0.0002,
+      "loss": 0.454,
+      "step": 1790
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 0.0002,
+      "loss": 0.446,
+      "step": 1800
+    },
+    {
+      "epoch": 0.71,
+      "eval_loss": 0.4937334656715393,
+      "eval_runtime": 140.6554,
+      "eval_samples_per_second": 7.11,
+      "eval_steps_per_second": 3.555,
+      "step": 1800
+    },
+    {
+      "epoch": 0.71,
+      "mmlu_eval_accuracy": 0.4705367508034988,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4827586206896552,
+      "mmlu_eval_accuracy_college_biology": 0.5,
+      "mmlu_eval_accuracy_college_chemistry": 0.375,
+      "mmlu_eval_accuracy_college_computer_science": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_mathematics": 0.45454545454545453,
+      "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.6363636363636364,
+      "mmlu_eval_accuracy_conceptual_physics": 0.46153846153846156,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.3125,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.34146341463414637,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.3,
+      "mmlu_eval_accuracy_high_school_biology": 0.5625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.2727272727272727,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.4186046511627907,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.13793103448275862,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5384615384615384,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7166666666666667,
+      "mmlu_eval_accuracy_high_school_statistics": 0.34782608695652173,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6363636363636364,
+      "mmlu_eval_accuracy_high_school_world_history": 0.4230769230769231,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.36363636363636365,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.72,
+      "mmlu_eval_accuracy_medical_genetics": 0.6363636363636364,
+      "mmlu_eval_accuracy_miscellaneous": 0.6511627906976745,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.5294117647058824,
+      "mmlu_eval_accuracy_prehistory": 0.5428571428571428,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.3588235294117647,
+      "mmlu_eval_accuracy_professional_medicine": 0.3870967741935484,
+      "mmlu_eval_accuracy_professional_psychology": 0.391304347826087,
+      "mmlu_eval_accuracy_public_relations": 0.5,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.6818181818181818,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.5454545454545454,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.8904348751505735,
+      "step": 1800
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 0.0002,
+      "loss": 0.4202,
+      "step": 1810
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 0.0002,
+      "loss": 0.4452,
+      "step": 1820
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 0.0002,
+      "loss": 0.4766,
+      "step": 1830
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 0.0002,
+      "loss": 0.4662,
+      "step": 1840
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 0.0002,
+      "loss": 0.4953,
+      "step": 1850
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 0.0002,
+      "loss": 0.4747,
+      "step": 1860
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 0.0002,
+      "loss": 0.4669,
+      "step": 1870
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 0.0002,
+      "loss": 0.4563,
+      "step": 1880
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 0.0002,
+      "loss": 0.4835,
+      "step": 1890
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 0.0002,
+      "loss": 0.5043,
+      "step": 1900
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 0.0002,
+      "loss": 0.4911,
+      "step": 1910
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 0.0002,
+      "loss": 0.5275,
+      "step": 1920
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 0.0002,
+      "loss": 0.5382,
+      "step": 1930
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 0.0002,
+      "loss": 0.4761,
+      "step": 1940
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 0.0002,
+      "loss": 0.4832,
+      "step": 1950
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 0.0002,
+      "loss": 0.5139,
+      "step": 1960
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 0.0002,
+      "loss": 0.4336,
+      "step": 1970
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 0.0002,
+      "loss": 0.4737,
+      "step": 1980
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 0.0002,
+      "loss": 0.4701,
+      "step": 1990
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 0.0002,
+      "loss": 0.4693,
+      "step": 2000
+    },
+    {
+      "epoch": 0.79,
+      "eval_loss": 0.4895848035812378,
+      "eval_runtime": 140.5189,
+      "eval_samples_per_second": 7.116,
+      "eval_steps_per_second": 3.558,
+      "step": 2000
+    },
+    {
+      "epoch": 0.79,
+      "mmlu_eval_accuracy": 0.4729355673327339,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.375,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4827586206896552,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.25,
+      "mmlu_eval_accuracy_college_computer_science": 0.18181818181818182,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.5454545454545454,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.36585365853658536,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.2727272727272727,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5238095238095238,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.3953488372093023,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.13793103448275862,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5384615384615384,
+      "mmlu_eval_accuracy_high_school_physics": 0.47058823529411764,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7166666666666667,
+      "mmlu_eval_accuracy_high_school_statistics": 0.391304347826087,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6363636363636364,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.8461538461538461,
+      "mmlu_eval_accuracy_jurisprudence": 0.36363636363636365,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.6363636363636364,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.76,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6395348837209303,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.23,
+      "mmlu_eval_accuracy_nutrition": 0.6060606060606061,
+      "mmlu_eval_accuracy_philosophy": 0.4411764705882353,
+      "mmlu_eval_accuracy_prehistory": 0.5428571428571428,
+      "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
+      "mmlu_eval_accuracy_professional_law": 0.3235294117647059,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.37681159420289856,
+      "mmlu_eval_accuracy_public_relations": 0.4166666666666667,
+      "mmlu_eval_accuracy_security_studies": 0.48148148148148145,
+      "mmlu_eval_accuracy_sociology": 0.5909090909090909,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.8803039049102339,
+      "step": 2000
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 0.0002,
+      "loss": 0.4525,
+      "step": 2010
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 0.0002,
+      "loss": 0.4686,
+      "step": 2020
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 0.0002,
+      "loss": 0.5052,
+      "step": 2030
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 0.0002,
+      "loss": 0.5205,
+      "step": 2040
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 0.0002,
+      "loss": 0.4834,
+      "step": 2050
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 0.0002,
+      "loss": 0.5252,
+      "step": 2060
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 0.0002,
+      "loss": 0.4665,
+      "step": 2070
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 0.0002,
+      "loss": 0.4918,
+      "step": 2080
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 0.0002,
+      "loss": 0.4702,
+      "step": 2090
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 0.0002,
+      "loss": 0.4507,
+      "step": 2100
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 0.0002,
+      "loss": 0.5271,
+      "step": 2110
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 0.0002,
+      "loss": 0.4762,
+      "step": 2120
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 0.0002,
+      "loss": 0.4495,
+      "step": 2130
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 0.0002,
+      "loss": 0.4629,
+      "step": 2140
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 0.0002,
+      "loss": 0.4676,
+      "step": 2150
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 0.0002,
+      "loss": 0.4566,
+      "step": 2160
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 0.0002,
+      "loss": 0.4422,
+      "step": 2170
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 0.0002,
+      "loss": 0.4725,
+      "step": 2180
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 0.0002,
+      "loss": 0.4529,
+      "step": 2190
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 0.0002,
+      "loss": 0.4626,
+      "step": 2200
+    },
+    {
+      "epoch": 0.86,
+      "eval_loss": 0.4861636161804199,
+      "eval_runtime": 140.3472,
+      "eval_samples_per_second": 7.125,
+      "eval_steps_per_second": 3.563,
+      "step": 2200
+    },
+    {
+      "epoch": 0.86,
+      "mmlu_eval_accuracy": 0.4631848150398904,
+      "mmlu_eval_accuracy_abstract_algebra": 0.18181818181818182,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.375,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4827586206896552,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.25,
+      "mmlu_eval_accuracy_college_computer_science": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_mathematics": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_medicine": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.45454545454545453,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.34146341463414637,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.8181818181818182,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.4186046511627907,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.2413793103448276,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.46153846153846156,
+      "mmlu_eval_accuracy_high_school_physics": 0.47058823529411764,
+      "mmlu_eval_accuracy_high_school_psychology": 0.6833333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.2608695652173913,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6363636363636364,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.8461538461538461,
+      "mmlu_eval_accuracy_jurisprudence": 0.36363636363636365,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.5454545454545454,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.6363636363636364,
+      "mmlu_eval_accuracy_miscellaneous": 0.6395348837209303,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.23,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
+      "mmlu_eval_accuracy_professional_law": 0.3352941176470588,
+      "mmlu_eval_accuracy_professional_medicine": 0.3870967741935484,
+      "mmlu_eval_accuracy_professional_psychology": 0.391304347826087,
+      "mmlu_eval_accuracy_public_relations": 0.4166666666666667,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5909090909090909,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.5454545454545454,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9769418603174058,
+      "step": 2200
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 0.0002,
+      "loss": 0.4377,
+      "step": 2210
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 0.0002,
+      "loss": 0.4308,
+      "step": 2220
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 0.0002,
+      "loss": 0.4708,
+      "step": 2230
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 0.0002,
+      "loss": 0.4981,
+      "step": 2240
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 0.0002,
+      "loss": 0.4575,
+      "step": 2250
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 0.0002,
+      "loss": 0.4846,
+      "step": 2260
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 0.0002,
+      "loss": 0.4459,
+      "step": 2270
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 0.0002,
+      "loss": 0.4838,
+      "step": 2280
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 0.0002,
+      "loss": 0.4537,
+      "step": 2290
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 0.0002,
+      "loss": 0.4302,
+      "step": 2300
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 0.0002,
+      "loss": 0.4847,
+      "step": 2310
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 0.0002,
+      "loss": 0.4733,
+      "step": 2320
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 0.0002,
+      "loss": 0.4639,
+      "step": 2330
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 0.0002,
+      "loss": 0.4703,
+      "step": 2340
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 0.0002,
+      "loss": 0.4441,
+      "step": 2350
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 0.0002,
+      "loss": 0.4299,
+      "step": 2360
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 0.0002,
+      "loss": 0.4414,
+      "step": 2370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 0.0002,
+      "loss": 0.4285,
+      "step": 2380
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 0.0002,
+      "loss": 0.4821,
+      "step": 2390
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 0.0002,
+      "loss": 0.4382,
+      "step": 2400
+    },
+    {
+      "epoch": 0.94,
+      "eval_loss": 0.4805959463119507,
+      "eval_runtime": 140.4712,
+      "eval_samples_per_second": 7.119,
+      "eval_steps_per_second": 3.559,
+      "step": 2400
+    },
+    {
+      "epoch": 0.94,
+      "mmlu_eval_accuracy": 0.47164059355692883,
+      "mmlu_eval_accuracy_abstract_algebra": 0.2727272727272727,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.4375,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.5,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.5454545454545454,
+      "mmlu_eval_accuracy_conceptual_physics": 0.5,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.4375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3170731707317073,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.34375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.3953488372093023,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.20689655172413793,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.46153846153846156,
+      "mmlu_eval_accuracy_high_school_physics": 0.47058823529411764,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7333333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.4782608695652174,
+      "mmlu_eval_accuracy_high_school_us_history": 0.5909090909090909,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6521739130434783,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.6923076923076923,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.45454545454545453,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.686046511627907,
+      "mmlu_eval_accuracy_moral_disputes": 0.47368421052631576,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.4411764705882353,
+      "mmlu_eval_accuracy_prehistory": 0.5428571428571428,
+      "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
+      "mmlu_eval_accuracy_professional_law": 0.31176470588235294,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.4057971014492754,
+      "mmlu_eval_accuracy_public_relations": 0.4166666666666667,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.6818181818181818,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.5454545454545454,
+      "mmlu_eval_accuracy_virology": 0.3333333333333333,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.8881722550751646,
+      "step": 2400
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 0.0002,
+      "loss": 0.4949,
+      "step": 2410
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 0.0002,
+      "loss": 0.4186,
+      "step": 2420
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 0.0002,
+      "loss": 0.4538,
+      "step": 2430
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 0.0002,
+      "loss": 0.4589,
+      "step": 2440
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 0.0002,
+      "loss": 0.463,
+      "step": 2450
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 0.0002,
+      "loss": 0.4516,
+      "step": 2460
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 0.0002,
+      "loss": 0.4761,
+      "step": 2470
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 0.0002,
+      "loss": 0.4222,
+      "step": 2480
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 0.0002,
+      "loss": 0.4524,
+      "step": 2490
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 0.0002,
+      "loss": 0.4326,
+      "step": 2500
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 0.0002,
+      "loss": 0.4738,
+      "step": 2510
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 0.0002,
+      "loss": 0.4237,
+      "step": 2520
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 0.0002,
+      "loss": 0.4434,
+      "step": 2530
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 0.0002,
+      "loss": 0.4552,
+      "step": 2540
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 0.0002,
+      "loss": 0.434,
+      "step": 2550
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 0.0002,
+      "loss": 0.3825,
+      "step": 2560
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 0.0002,
+      "loss": 0.412,
+      "step": 2570
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 0.0002,
+      "loss": 0.4169,
+      "step": 2580
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 0.0002,
+      "loss": 0.4264,
+      "step": 2590
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 0.0002,
+      "loss": 0.4451,
+      "step": 2600
+    },
+    {
+      "epoch": 1.02,
+      "eval_loss": 0.4820026457309723,
+      "eval_runtime": 140.6209,
+      "eval_samples_per_second": 7.111,
+      "eval_steps_per_second": 3.556,
+      "step": 2600
+    },
+    {
+      "epoch": 1.02,
+      "mmlu_eval_accuracy": 0.46444809086867156,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4827586206896552,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.4090909090909091,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.36363636363636365,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.16666666666666666,
+      "mmlu_eval_accuracy_electrical_engineering": 0.4375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.2926829268292683,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.7272727272727273,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.4186046511627907,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.1724137931034483,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5,
+      "mmlu_eval_accuracy_high_school_physics": 0.47058823529411764,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7,
+      "mmlu_eval_accuracy_high_school_statistics": 0.391304347826087,
+      "mmlu_eval_accuracy_high_school_us_history": 0.7272727272727273,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.5454545454545454,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6627906976744186,
+      "mmlu_eval_accuracy_moral_disputes": 0.39473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.3352941176470588,
+      "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
+      "mmlu_eval_accuracy_professional_psychology": 0.3333333333333333,
+      "mmlu_eval_accuracy_public_relations": 0.4166666666666667,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.6363636363636364,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.3888888888888889,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.8536750212428464,
+      "step": 2600
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 0.0002,
+      "loss": 0.4018,
+      "step": 2610
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 0.0002,
+      "loss": 0.3895,
+      "step": 2620
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 0.0002,
+      "loss": 0.4141,
+      "step": 2630
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 0.0002,
+      "loss": 0.39,
+      "step": 2640
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 0.0002,
+      "loss": 0.4459,
+      "step": 2650
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 0.0002,
+      "loss": 0.405,
+      "step": 2660
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 0.0002,
+      "loss": 0.4372,
+      "step": 2670
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 0.0002,
+      "loss": 0.4247,
+      "step": 2680
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 0.0002,
+      "loss": 0.3909,
+      "step": 2690
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 0.0002,
+      "loss": 0.4143,
+      "step": 2700
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 0.0002,
+      "loss": 0.3966,
+      "step": 2710
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 0.0002,
+      "loss": 0.4216,
+      "step": 2720
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 0.0002,
+      "loss": 0.3964,
+      "step": 2730
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 0.0002,
+      "loss": 0.3929,
+      "step": 2740
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 0.0002,
+      "loss": 0.4062,
+      "step": 2750
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 0.0002,
+      "loss": 0.4071,
+      "step": 2760
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 0.0002,
+      "loss": 0.3655,
+      "step": 2770
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 0.0002,
+      "loss": 0.39,
+      "step": 2780
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 0.0002,
+      "loss": 0.3942,
+      "step": 2790
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 0.0002,
+      "loss": 0.3919,
+      "step": 2800
+    },
+    {
+      "epoch": 1.1,
+      "eval_loss": 0.47912824153900146,
+      "eval_runtime": 140.5539,
+      "eval_samples_per_second": 7.115,
+      "eval_steps_per_second": 3.557,
+      "step": 2800
+    },
+    {
+      "epoch": 1.1,
+      "mmlu_eval_accuracy": 0.4700767413667956,
+      "mmlu_eval_accuracy_abstract_algebra": 0.2727272727272727,
+      "mmlu_eval_accuracy_anatomy": 0.5,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.6363636363636364,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.125,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_medicine": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.5454545454545454,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.4375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3170731707317073,
+      "mmlu_eval_accuracy_formal_logic": 0.2857142857142857,
+      "mmlu_eval_accuracy_global_facts": 0.4,
+      "mmlu_eval_accuracy_high_school_biology": 0.375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.37209302325581395,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.2413793103448276,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5384615384615384,
+      "mmlu_eval_accuracy_high_school_physics": 0.47058823529411764,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7,
+      "mmlu_eval_accuracy_high_school_statistics": 0.34782608695652173,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5769230769230769,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.6923076923076923,
+      "mmlu_eval_accuracy_jurisprudence": 0.45454545454545453,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.68,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6744186046511628,
+      "mmlu_eval_accuracy_moral_disputes": 0.39473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.25,
+      "mmlu_eval_accuracy_nutrition": 0.5757575757575758,
+      "mmlu_eval_accuracy_philosophy": 0.5,
+      "mmlu_eval_accuracy_prehistory": 0.45714285714285713,
+      "mmlu_eval_accuracy_professional_accounting": 0.2903225806451613,
+      "mmlu_eval_accuracy_professional_law": 0.3411764705882353,
+      "mmlu_eval_accuracy_professional_medicine": 0.3548387096774194,
+      "mmlu_eval_accuracy_professional_psychology": 0.36231884057971014,
+      "mmlu_eval_accuracy_public_relations": 0.5,
+      "mmlu_eval_accuracy_security_studies": 0.48148148148148145,
+      "mmlu_eval_accuracy_sociology": 0.6363636363636364,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.5454545454545454,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.985391327085134,
+      "step": 2800
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 0.0002,
+      "loss": 0.4731,
+      "step": 2810
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 0.0002,
+      "loss": 0.4214,
+      "step": 2820
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 0.0002,
+      "loss": 0.3952,
+      "step": 2830
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 0.0002,
+      "loss": 0.3791,
+      "step": 2840
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 0.0002,
+      "loss": 0.4295,
+      "step": 2850
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 0.0002,
+      "loss": 0.4406,
+      "step": 2860
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 0.0002,
+      "loss": 0.3911,
+      "step": 2870
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 0.0002,
+      "loss": 0.3973,
+      "step": 2880
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 0.0002,
+      "loss": 0.387,
+      "step": 2890
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 0.0002,
+      "loss": 0.3885,
+      "step": 2900
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 0.0002,
+      "loss": 0.4061,
+      "step": 2910
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 0.0002,
+      "loss": 0.4004,
+      "step": 2920
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 0.0002,
+      "loss": 0.4214,
+      "step": 2930
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 0.0002,
+      "loss": 0.4001,
+      "step": 2940
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 0.0002,
+      "loss": 0.438,
+      "step": 2950
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 0.0002,
+      "loss": 0.446,
+      "step": 2960
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 0.0002,
+      "loss": 0.4176,
+      "step": 2970
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 0.0002,
+      "loss": 0.3817,
+      "step": 2980
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 0.0002,
+      "loss": 0.3827,
+      "step": 2990
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 0.0002,
+      "loss": 0.4133,
+      "step": 3000
+    },
+    {
+      "epoch": 1.18,
+      "eval_loss": 0.47800615429878235,
+      "eval_runtime": 140.3789,
+      "eval_samples_per_second": 7.124,
+      "eval_steps_per_second": 3.562,
+      "step": 3000
+    },
+    {
+      "epoch": 1.18,
+      "mmlu_eval_accuracy": 0.4646482935507578,
+      "mmlu_eval_accuracy_abstract_algebra": 0.2727272727272727,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.5,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.375,
+      "mmlu_eval_accuracy_college_chemistry": 0.0,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.4090909090909091,
+      "mmlu_eval_accuracy_college_physics": 0.36363636363636365,
+      "mmlu_eval_accuracy_computer_security": 0.5454545454545454,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.3125,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.3902439024390244,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.4375,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.3181818181818182,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.5714285714285714,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.4186046511627907,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.13793103448275862,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.5769230769230769,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.7333333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.391304347826087,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.3333333333333333,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.5454545454545454,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.6363636363636364,
+      "mmlu_eval_accuracy_marketing": 0.64,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6395348837209303,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.24,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.47058823529411764,
+      "mmlu_eval_accuracy_prehistory": 0.4857142857142857,
+      "mmlu_eval_accuracy_professional_accounting": 0.22580645161290322,
+      "mmlu_eval_accuracy_professional_law": 0.3352941176470588,
+      "mmlu_eval_accuracy_professional_medicine": 0.45161290322580644,
+      "mmlu_eval_accuracy_professional_psychology": 0.36231884057971014,
+      "mmlu_eval_accuracy_public_relations": 0.5,
+      "mmlu_eval_accuracy_security_studies": 0.5185185185185185,
+      "mmlu_eval_accuracy_sociology": 0.5454545454545454,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.9537461701556845,
+      "step": 3000
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 0.0002,
+      "loss": 0.4238,
+      "step": 3010
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 0.0002,
+      "loss": 0.4087,
+      "step": 3020
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 0.0002,
+      "loss": 0.3956,
+      "step": 3030
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 0.0002,
+      "loss": 0.4106,
+      "step": 3040
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 0.0002,
+      "loss": 0.4024,
+      "step": 3050
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 0.0002,
+      "loss": 0.4229,
+      "step": 3060
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 0.0002,
+      "loss": 0.4226,
+      "step": 3070
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 0.0002,
+      "loss": 0.3963,
+      "step": 3080
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 0.0002,
+      "loss": 0.4189,
+      "step": 3090
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 0.0002,
+      "loss": 0.42,
+      "step": 3100
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 0.0002,
+      "loss": 0.422,
+      "step": 3110
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 0.0002,
+      "loss": 0.4273,
+      "step": 3120
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 0.0002,
+      "loss": 0.4191,
+      "step": 3130
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 0.0002,
+      "loss": 0.4227,
+      "step": 3140
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 0.0002,
+      "loss": 0.3993,
+      "step": 3150
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 0.0002,
+      "loss": 0.404,
+      "step": 3160
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 0.0002,
+      "loss": 0.3856,
+      "step": 3170
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 0.0002,
+      "loss": 0.4666,
+      "step": 3180
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 0.0002,
+      "loss": 0.3978,
+      "step": 3190
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 0.0002,
+      "loss": 0.4145,
+      "step": 3200
+    },
+    {
+      "epoch": 1.26,
+      "eval_loss": 0.4805873930454254,
+      "eval_runtime": 140.4294,
+      "eval_samples_per_second": 7.121,
+      "eval_steps_per_second": 3.561,
+      "step": 3200
+    },
+    {
+      "epoch": 1.26,
+      "mmlu_eval_accuracy": 0.47096664407805255,
+      "mmlu_eval_accuracy_abstract_algebra": 0.36363636363636365,
+      "mmlu_eval_accuracy_anatomy": 0.5714285714285714,
+      "mmlu_eval_accuracy_astronomy": 0.4375,
+      "mmlu_eval_accuracy_business_ethics": 0.5454545454545454,
+      "mmlu_eval_accuracy_clinical_knowledge": 0.4482758620689655,
+      "mmlu_eval_accuracy_college_biology": 0.4375,
+      "mmlu_eval_accuracy_college_chemistry": 0.25,
+      "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_mathematics": 0.2727272727272727,
+      "mmlu_eval_accuracy_college_medicine": 0.36363636363636365,
+      "mmlu_eval_accuracy_college_physics": 0.45454545454545453,
+      "mmlu_eval_accuracy_computer_security": 0.5454545454545454,
+      "mmlu_eval_accuracy_conceptual_physics": 0.4230769230769231,
+      "mmlu_eval_accuracy_econometrics": 0.25,
+      "mmlu_eval_accuracy_electrical_engineering": 0.375,
+      "mmlu_eval_accuracy_elementary_mathematics": 0.2926829268292683,
+      "mmlu_eval_accuracy_formal_logic": 0.21428571428571427,
+      "mmlu_eval_accuracy_global_facts": 0.5,
+      "mmlu_eval_accuracy_high_school_biology": 0.40625,
+      "mmlu_eval_accuracy_high_school_chemistry": 0.2727272727272727,
+      "mmlu_eval_accuracy_high_school_computer_science": 0.4444444444444444,
+      "mmlu_eval_accuracy_high_school_european_history": 0.5,
+      "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
+      "mmlu_eval_accuracy_high_school_government_and_politics": 0.6190476190476191,
+      "mmlu_eval_accuracy_high_school_macroeconomics": 0.3953488372093023,
+      "mmlu_eval_accuracy_high_school_mathematics": 0.10344827586206896,
+      "mmlu_eval_accuracy_high_school_microeconomics": 0.6153846153846154,
+      "mmlu_eval_accuracy_high_school_physics": 0.35294117647058826,
+      "mmlu_eval_accuracy_high_school_psychology": 0.6833333333333333,
+      "mmlu_eval_accuracy_high_school_statistics": 0.30434782608695654,
+      "mmlu_eval_accuracy_high_school_us_history": 0.6818181818181818,
+      "mmlu_eval_accuracy_high_school_world_history": 0.5384615384615384,
+      "mmlu_eval_accuracy_human_aging": 0.6956521739130435,
+      "mmlu_eval_accuracy_human_sexuality": 0.4166666666666667,
+      "mmlu_eval_accuracy_international_law": 0.7692307692307693,
+      "mmlu_eval_accuracy_jurisprudence": 0.5454545454545454,
+      "mmlu_eval_accuracy_logical_fallacies": 0.5555555555555556,
+      "mmlu_eval_accuracy_machine_learning": 0.36363636363636365,
+      "mmlu_eval_accuracy_management": 0.7272727272727273,
+      "mmlu_eval_accuracy_marketing": 0.64,
+      "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
+      "mmlu_eval_accuracy_miscellaneous": 0.6744186046511628,
+      "mmlu_eval_accuracy_moral_disputes": 0.4473684210526316,
+      "mmlu_eval_accuracy_moral_scenarios": 0.25,
+      "mmlu_eval_accuracy_nutrition": 0.5454545454545454,
+      "mmlu_eval_accuracy_philosophy": 0.4411764705882353,
+      "mmlu_eval_accuracy_prehistory": 0.5142857142857142,
+      "mmlu_eval_accuracy_professional_accounting": 0.25806451612903225,
+      "mmlu_eval_accuracy_professional_law": 0.32941176470588235,
+      "mmlu_eval_accuracy_professional_medicine": 0.3870967741935484,
+      "mmlu_eval_accuracy_professional_psychology": 0.4057971014492754,
+      "mmlu_eval_accuracy_public_relations": 0.4166666666666667,
+      "mmlu_eval_accuracy_security_studies": 0.48148148148148145,
+      "mmlu_eval_accuracy_sociology": 0.6363636363636364,
+      "mmlu_eval_accuracy_us_foreign_policy": 0.6363636363636364,
+      "mmlu_eval_accuracy_virology": 0.4444444444444444,
+      "mmlu_eval_accuracy_world_religions": 0.7368421052631579,
+      "mmlu_loss": 0.854310085155634,
+      "step": 3200
+    }
+  ],
+  "max_steps": 5000,
+  "num_train_epochs": 2,
+  "total_flos": 5.13594124601131e+17,
+  "trial_name": null,
+  "trial_params": null
+}