10k_test3_nli_finetuned_canine_c / trainer_state.json
Gwatk's picture
End of training
a541ac1
{
"best_metric": 0.35333333333333333,
"best_model_checkpoint": "nli_results/10k_test3/models/canine_c/checkpoint-625",
"epoch": 2.9952,
"global_step": 936,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4.99465811965812e-05,
"loss": 1.3267,
"step": 1
},
{
"epoch": 0.32,
"learning_rate": 4.465811965811966e-05,
"loss": 1.1043,
"step": 100
},
{
"epoch": 0.64,
"learning_rate": 3.931623931623932e-05,
"loss": 1.102,
"step": 200
},
{
"epoch": 0.96,
"learning_rate": 3.397435897435898e-05,
"loss": 1.1017,
"step": 300
},
{
"epoch": 1.0,
"eval_accuracy": 0.32466666666666666,
"eval_f1_weighted": 0.15914712296594533,
"eval_loss": 1.1029257774353027,
"eval_precision_weighted": 0.10540844444444444,
"eval_recall_weighted": 0.32466666666666666,
"eval_runtime": 5.4193,
"eval_samples_per_second": 276.79,
"eval_steps_per_second": 69.197,
"step": 312
},
{
"epoch": 1.28,
"learning_rate": 2.863247863247863e-05,
"loss": 1.0994,
"step": 400
},
{
"epoch": 1.6,
"learning_rate": 2.3290598290598293e-05,
"loss": 1.1003,
"step": 500
},
{
"epoch": 1.92,
"learning_rate": 1.794871794871795e-05,
"loss": 1.0999,
"step": 600
},
{
"epoch": 2.0,
"eval_accuracy": 0.35333333333333333,
"eval_f1_weighted": 0.1844991789819376,
"eval_loss": 1.0978268384933472,
"eval_precision_weighted": 0.12484444444444445,
"eval_recall_weighted": 0.35333333333333333,
"eval_runtime": 6.1353,
"eval_samples_per_second": 244.485,
"eval_steps_per_second": 61.121,
"step": 625
},
{
"epoch": 2.24,
"learning_rate": 1.2606837606837608e-05,
"loss": 1.0999,
"step": 700
},
{
"epoch": 2.56,
"learning_rate": 7.264957264957266e-06,
"loss": 1.0995,
"step": 800
},
{
"epoch": 2.88,
"learning_rate": 1.9230769230769234e-06,
"loss": 1.099,
"step": 900
},
{
"epoch": 3.0,
"eval_accuracy": 0.32466666666666666,
"eval_f1_weighted": 0.15914712296594533,
"eval_loss": 1.0989837646484375,
"eval_precision_weighted": 0.10540844444444444,
"eval_recall_weighted": 0.32466666666666666,
"eval_runtime": 5.3918,
"eval_samples_per_second": 278.202,
"eval_steps_per_second": 69.55,
"step": 936
},
{
"epoch": 3.0,
"step": 936,
"total_flos": 5792280106446720.0,
"train_loss": 1.100847668117947,
"train_runtime": 456.1632,
"train_samples_per_second": 65.766,
"train_steps_per_second": 2.052
}
],
"max_steps": 936,
"num_train_epochs": 3,
"total_flos": 5792280106446720.0,
"trial_name": null,
"trial_params": null
}