checkpoint-200/trainer_state.json

{
 "best_metric": 0.44470372796058655,
 "best_model_checkpoint": "./output_v2/7b_cluster08_Nous-Hermes-llama-2-7b_partitioned_v3_standardized_08/checkpoint-200",
 "epoch": 0.21633315305570577,
 "global_step": 200,
 "is_hyper_param_search": false,
 "is_local_process_zero": true,
 "is_world_process_zero": true,
 "log_history": [
 {
 "epoch": 0.01,
 "learning_rate": 0.0002,
 "loss": 0.5043,
 "step": 10
 },
 {
 "epoch": 0.02,
 "learning_rate": 0.0002,
 "loss": 0.4612,
 "step": 20
 },
 {
 "epoch": 0.03,
 "learning_rate": 0.0002,
 "loss": 0.4969,
 "step": 30
 },
 {
 "epoch": 0.04,
 "learning_rate": 0.0002,
 "loss": 0.4995,
 "step": 40
 },
 {
 "epoch": 0.05,
 "learning_rate": 0.0002,
 "loss": 0.4247,
 "step": 50
 },
 {
 "epoch": 0.06,
 "learning_rate": 0.0002,
 "loss": 0.4557,
 "step": 60
 },
 {
 "epoch": 0.08,
 "learning_rate": 0.0002,
 "loss": 0.5582,
 "step": 70
 },
 {
 "epoch": 0.09,
 "learning_rate": 0.0002,
 "loss": 0.4484,
 "step": 80
 },
 {
 "epoch": 0.1,
 "learning_rate": 0.0002,
 "loss": 0.438,
 "step": 90
 },
 {
 "epoch": 0.11,
 "learning_rate": 0.0002,
 "loss": 0.4528,
 "step": 100
 },
 {
 "epoch": 0.12,
 "learning_rate": 0.0002,
 "loss": 0.4848,
 "step": 110
 },
 {
 "epoch": 0.13,
 "learning_rate": 0.0002,
 "loss": 0.4192,
 "step": 120
 },
 {
 "epoch": 0.14,
 "learning_rate": 0.0002,
 "loss": 0.4481,
 "step": 130
 },
 {
 "epoch": 0.15,
 "learning_rate": 0.0002,
 "loss": 0.438,
 "step": 140
 },
 {
 "epoch": 0.16,
 "learning_rate": 0.0002,
 "loss": 0.4484,
 "step": 150
 },
 {
 "epoch": 0.17,
 "learning_rate": 0.0002,
 "loss": 0.4156,
 "step": 160
 },
 {
 "epoch": 0.18,
 "learning_rate": 0.0002,
 "loss": 0.3978,
 "step": 170
 },
 {
 "epoch": 0.19,
 "learning_rate": 0.0002,
 "loss": 0.5086,
 "step": 180
 },
 {
 "epoch": 0.21,
 "learning_rate": 0.0002,
 "loss": 0.5037,
 "step": 190
 },
 {
 "epoch": 0.22,
 "learning_rate": 0.0002,
 "loss": 0.4201,
 "step": 200
 },
 {
 "epoch": 0.22,
 "eval_loss": 0.44470372796058655,
 "eval_runtime": 145.3872,
 "eval_samples_per_second": 6.878,
 "eval_steps_per_second": 3.439,
 "step": 200
 },
 {
 "epoch": 0.22,
 "mmlu_eval_accuracy": 0.4603515989210557,
 "mmlu_eval_accuracy_abstract_algebra": 0.18181818181818182,
 "mmlu_eval_accuracy_anatomy": 0.5,
 "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.125,
 "mmlu_eval_accuracy_college_computer_science": 0.36363636363636365,
 "mmlu_eval_accuracy_college_mathematics": 0.18181818181818182,
 "mmlu_eval_accuracy_college_medicine": 0.3181818181818182,
 "mmlu_eval_accuracy_college_physics": 0.36363636363636365,
 "mmlu_eval_accuracy_computer_security": 0.2727272727272727,
 "mmlu_eval_accuracy_conceptual_physics": 0.5,
 "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.2857142857142857,
 "mmlu_eval_accuracy_global_facts": 0.6,
 "mmlu_eval_accuracy_high_school_biology": 0.34375,
 "mmlu_eval_accuracy_high_school_chemistry": 0.4090909090909091,
 "mmlu_eval_accuracy_high_school_computer_science": 0.6666666666666666,
 "mmlu_eval_accuracy_high_school_european_history": 0.6111111111111112,
 "mmlu_eval_accuracy_high_school_geography": 0.7727272727272727,
 "mmlu_eval_accuracy_high_school_government_and_politics": 0.6666666666666666,
 "mmlu_eval_accuracy_high_school_macroeconomics": 0.32558139534883723,
 "mmlu_eval_accuracy_high_school_mathematics": 0.20689655172413793,
 "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.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.5,
 "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.18181818181818182,
 "mmlu_eval_accuracy_management": 0.5454545454545454,
 "mmlu_eval_accuracy_marketing": 0.72,
 "mmlu_eval_accuracy_medical_genetics": 0.7272727272727273,
 "mmlu_eval_accuracy_miscellaneous": 0.6627906976744186,
 "mmlu_eval_accuracy_moral_disputes": 0.42105263157894735,
 "mmlu_eval_accuracy_moral_scenarios": 0.23,
 "mmlu_eval_accuracy_nutrition": 0.5151515151515151,
 "mmlu_eval_accuracy_philosophy": 0.5,
 "mmlu_eval_accuracy_prehistory": 0.42857142857142855,
 "mmlu_eval_accuracy_professional_accounting": 0.3225806451612903,
 "mmlu_eval_accuracy_professional_law": 0.3411764705882353,
 "mmlu_eval_accuracy_professional_medicine": 0.41935483870967744,
 "mmlu_eval_accuracy_professional_psychology": 0.37681159420289856,
 "mmlu_eval_accuracy_public_relations": 0.5833333333333334,
 "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.6842105263157895,
 "mmlu_loss": 1.2632288357763315,
 "step": 200
 }
 ],
 "max_steps": 5000,
 "num_train_epochs": 6,
 "total_flos": 2.878943743210291e+16,
 "trial_name": null,
 "trial_params": null
}