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
}