roberta-base-mnli_AppE / trainer_state.json
liuyanchen1015's picture
Training in progress, step 2000
812be88
{
"best_metric": 0.8634595701125896,
"best_model_checkpoint": "./outputs/finetuning/mnli_AppE/checkpoint-26000",
"epoch": 5.0,
"global_step": 60000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.17,
"learning_rate": 1.9333333333333333e-05,
"loss": 0.3562,
"step": 2000
},
{
"epoch": 0.17,
"eval_acc": 0.8524053224155578,
"eval_loss": 0.41664522886276245,
"eval_runtime": 16.7668,
"eval_samples_per_second": 582.7,
"eval_steps_per_second": 18.25,
"step": 2000
},
{
"epoch": 0.33,
"learning_rate": 1.866666666666667e-05,
"loss": 0.3414,
"step": 4000
},
{
"epoch": 0.33,
"eval_acc": 0.8475946775844422,
"eval_loss": 0.43986669182777405,
"eval_runtime": 16.7213,
"eval_samples_per_second": 584.285,
"eval_steps_per_second": 18.3,
"step": 4000
},
{
"epoch": 0.5,
"learning_rate": 1.8e-05,
"loss": 0.3308,
"step": 6000
},
{
"epoch": 0.5,
"eval_acc": 0.8560900716479017,
"eval_loss": 0.395113080739975,
"eval_runtime": 16.7306,
"eval_samples_per_second": 583.96,
"eval_steps_per_second": 18.29,
"step": 6000
},
{
"epoch": 0.67,
"learning_rate": 1.7333333333333336e-05,
"loss": 0.3306,
"step": 8000
},
{
"epoch": 0.67,
"eval_acc": 0.8488229273285568,
"eval_loss": 0.41524460911750793,
"eval_runtime": 16.7003,
"eval_samples_per_second": 585.021,
"eval_steps_per_second": 18.323,
"step": 8000
},
{
"epoch": 0.83,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.3295,
"step": 10000
},
{
"epoch": 0.83,
"eval_acc": 0.8575230296827021,
"eval_loss": 0.39166104793548584,
"eval_runtime": 16.6831,
"eval_samples_per_second": 585.622,
"eval_steps_per_second": 18.342,
"step": 10000
},
{
"epoch": 1.0,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.3288,
"step": 12000
},
{
"epoch": 1.0,
"eval_acc": 0.8581371545547595,
"eval_loss": 0.40951114892959595,
"eval_runtime": 16.6934,
"eval_samples_per_second": 585.263,
"eval_steps_per_second": 18.331,
"step": 12000
},
{
"epoch": 1.17,
"learning_rate": 1.5333333333333334e-05,
"loss": 0.229,
"step": 14000
},
{
"epoch": 1.17,
"eval_acc": 0.8611054247697032,
"eval_loss": 0.4420296251773834,
"eval_runtime": 16.7189,
"eval_samples_per_second": 584.367,
"eval_steps_per_second": 18.303,
"step": 14000
},
{
"epoch": 1.33,
"learning_rate": 1.4666666666666666e-05,
"loss": 0.2307,
"step": 16000
},
{
"epoch": 1.33,
"eval_acc": 0.8560900716479017,
"eval_loss": 0.4876081645488739,
"eval_runtime": 16.6966,
"eval_samples_per_second": 585.148,
"eval_steps_per_second": 18.327,
"step": 16000
},
{
"epoch": 1.5,
"learning_rate": 1.4e-05,
"loss": 0.2334,
"step": 18000
},
{
"epoch": 1.5,
"eval_acc": 0.857727737973388,
"eval_loss": 0.449561208486557,
"eval_runtime": 16.7074,
"eval_samples_per_second": 584.771,
"eval_steps_per_second": 18.315,
"step": 18000
},
{
"epoch": 1.67,
"learning_rate": 1.3333333333333333e-05,
"loss": 0.2392,
"step": 20000
},
{
"epoch": 1.67,
"eval_acc": 0.8607983623336745,
"eval_loss": 0.4188033640384674,
"eval_runtime": 16.694,
"eval_samples_per_second": 585.239,
"eval_steps_per_second": 18.33,
"step": 20000
},
{
"epoch": 1.83,
"learning_rate": 1.2666666666666667e-05,
"loss": 0.233,
"step": 22000
},
{
"epoch": 1.83,
"eval_acc": 0.8578300921187308,
"eval_loss": 0.44934678077697754,
"eval_runtime": 16.6959,
"eval_samples_per_second": 585.174,
"eval_steps_per_second": 18.328,
"step": 22000
},
{
"epoch": 2.0,
"learning_rate": 1.2e-05,
"loss": 0.2343,
"step": 24000
},
{
"epoch": 2.0,
"eval_acc": 0.8602865916069601,
"eval_loss": 0.42783743143081665,
"eval_runtime": 16.7147,
"eval_samples_per_second": 584.517,
"eval_steps_per_second": 18.307,
"step": 24000
},
{
"epoch": 2.17,
"learning_rate": 1.1333333333333334e-05,
"loss": 0.163,
"step": 26000
},
{
"epoch": 2.17,
"eval_acc": 0.8634595701125896,
"eval_loss": 0.5700035095214844,
"eval_runtime": 16.7469,
"eval_samples_per_second": 583.393,
"eval_steps_per_second": 18.272,
"step": 26000
},
{
"epoch": 2.33,
"learning_rate": 1.0666666666666667e-05,
"loss": 0.1657,
"step": 28000
},
{
"epoch": 2.33,
"eval_acc": 0.8560900716479017,
"eval_loss": 0.5322555303573608,
"eval_runtime": 16.7051,
"eval_samples_per_second": 584.851,
"eval_steps_per_second": 18.318,
"step": 28000
},
{
"epoch": 2.5,
"learning_rate": 1e-05,
"loss": 0.1652,
"step": 30000
},
{
"epoch": 2.5,
"eval_acc": 0.8588536335721597,
"eval_loss": 0.5046815872192383,
"eval_runtime": 16.7454,
"eval_samples_per_second": 583.445,
"eval_steps_per_second": 18.274,
"step": 30000
},
{
"epoch": 2.67,
"learning_rate": 9.333333333333334e-06,
"loss": 0.1654,
"step": 32000
},
{
"epoch": 2.67,
"eval_acc": 0.8589559877175026,
"eval_loss": 0.5081538558006287,
"eval_runtime": 16.6963,
"eval_samples_per_second": 585.158,
"eval_steps_per_second": 18.327,
"step": 32000
},
{
"epoch": 2.83,
"learning_rate": 8.666666666666668e-06,
"loss": 0.1659,
"step": 34000
},
{
"epoch": 2.83,
"eval_acc": 0.8630501535312181,
"eval_loss": 0.5075551271438599,
"eval_runtime": 16.7114,
"eval_samples_per_second": 584.632,
"eval_steps_per_second": 18.311,
"step": 34000
},
{
"epoch": 3.0,
"learning_rate": 8.000000000000001e-06,
"loss": 0.1654,
"step": 36000
},
{
"epoch": 3.0,
"eval_acc": 0.8612077789150461,
"eval_loss": 0.5035998821258545,
"eval_runtime": 16.7468,
"eval_samples_per_second": 583.394,
"eval_steps_per_second": 18.272,
"step": 36000
},
{
"epoch": 3.17,
"learning_rate": 7.333333333333333e-06,
"loss": 0.1195,
"step": 38000
},
{
"epoch": 3.17,
"eval_acc": 0.8599795291709315,
"eval_loss": 0.6220871806144714,
"eval_runtime": 16.6963,
"eval_samples_per_second": 585.161,
"eval_steps_per_second": 18.327,
"step": 38000
},
{
"epoch": 3.33,
"learning_rate": 6.666666666666667e-06,
"loss": 0.1194,
"step": 40000
},
{
"epoch": 3.33,
"eval_acc": 0.8575230296827021,
"eval_loss": 0.6466721892356873,
"eval_runtime": 16.6726,
"eval_samples_per_second": 585.991,
"eval_steps_per_second": 18.353,
"step": 40000
},
{
"epoch": 3.5,
"learning_rate": 6e-06,
"loss": 0.1205,
"step": 42000
},
{
"epoch": 3.5,
"eval_acc": 0.8578300921187308,
"eval_loss": 0.6742368340492249,
"eval_runtime": 16.6999,
"eval_samples_per_second": 585.034,
"eval_steps_per_second": 18.323,
"step": 42000
},
{
"epoch": 3.67,
"learning_rate": 5.333333333333334e-06,
"loss": 0.1209,
"step": 44000
},
{
"epoch": 3.67,
"eval_acc": 0.8580348004094166,
"eval_loss": 0.6536660194396973,
"eval_runtime": 16.7907,
"eval_samples_per_second": 581.87,
"eval_steps_per_second": 18.224,
"step": 44000
},
{
"epoch": 3.83,
"learning_rate": 4.666666666666667e-06,
"loss": 0.1234,
"step": 46000
},
{
"epoch": 3.83,
"eval_acc": 0.8580348004094166,
"eval_loss": 0.6385037899017334,
"eval_runtime": 16.6773,
"eval_samples_per_second": 585.826,
"eval_steps_per_second": 18.348,
"step": 46000
},
{
"epoch": 4.0,
"learning_rate": 4.000000000000001e-06,
"loss": 0.122,
"step": 48000
},
{
"epoch": 4.0,
"eval_acc": 0.8628454452405322,
"eval_loss": 0.6103753447532654,
"eval_runtime": 16.681,
"eval_samples_per_second": 585.697,
"eval_steps_per_second": 18.344,
"step": 48000
},
{
"epoch": 4.17,
"learning_rate": 3.3333333333333333e-06,
"loss": 0.0937,
"step": 50000
},
{
"epoch": 4.17,
"eval_acc": 0.8602865916069601,
"eval_loss": 0.7260550260543823,
"eval_runtime": 16.6866,
"eval_samples_per_second": 585.501,
"eval_steps_per_second": 18.338,
"step": 50000
},
{
"epoch": 4.33,
"learning_rate": 2.666666666666667e-06,
"loss": 0.0941,
"step": 52000
},
{
"epoch": 4.33,
"eval_acc": 0.8606960081883316,
"eval_loss": 0.7633941769599915,
"eval_runtime": 16.696,
"eval_samples_per_second": 585.17,
"eval_steps_per_second": 18.328,
"step": 52000
},
{
"epoch": 4.5,
"learning_rate": 2.0000000000000003e-06,
"loss": 0.0926,
"step": 54000
},
{
"epoch": 4.5,
"eval_acc": 0.8617195496417605,
"eval_loss": 0.7736791968345642,
"eval_runtime": 16.6764,
"eval_samples_per_second": 585.859,
"eval_steps_per_second": 18.349,
"step": 54000
},
{
"epoch": 4.67,
"learning_rate": 1.3333333333333334e-06,
"loss": 0.093,
"step": 56000
},
{
"epoch": 4.67,
"eval_acc": 0.8632548618219038,
"eval_loss": 0.757666289806366,
"eval_runtime": 16.6803,
"eval_samples_per_second": 585.722,
"eval_steps_per_second": 18.345,
"step": 56000
},
{
"epoch": 4.83,
"learning_rate": 6.666666666666667e-07,
"loss": 0.0918,
"step": 58000
},
{
"epoch": 4.83,
"eval_acc": 0.8630501535312181,
"eval_loss": 0.7534608840942383,
"eval_runtime": 16.6933,
"eval_samples_per_second": 585.263,
"eval_steps_per_second": 18.331,
"step": 58000
},
{
"epoch": 5.0,
"learning_rate": 0.0,
"loss": 0.0935,
"step": 60000
},
{
"epoch": 5.0,
"eval_acc": 0.8622313203684749,
"eval_loss": 0.743323802947998,
"eval_runtime": 16.7307,
"eval_samples_per_second": 583.956,
"eval_steps_per_second": 18.29,
"step": 60000
},
{
"epoch": 5.0,
"step": 60000,
"total_flos": 1.2628556043159168e+17,
"train_loss": 0.18972002029418944,
"train_runtime": 11504.4253,
"train_samples_per_second": 166.881,
"train_steps_per_second": 5.215
}
],
"max_steps": 60000,
"num_train_epochs": 5,
"total_flos": 1.2628556043159168e+17,
"trial_name": null,
"trial_params": null
}