File size: 2,806 Bytes
2803ec7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
{
"best_metric": 0.6896626353263855,
"best_model_checkpoint": "../experiments/attribute_minig_mslacerda/run-2/checkpoint-425",
"epoch": 5.0,
"eval_steps": 500,
"global_step": 425,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_accuracy": 0.15225225225225225,
"eval_f1": 0.02484055052030883,
"eval_loss": 4.083955764770508,
"eval_precision": 0.027106227106227107,
"eval_recall": 0.02292441140024783,
"eval_runtime": 0.593,
"eval_samples_per_second": 490.762,
"eval_steps_per_second": 62.399,
"step": 85
},
{
"epoch": 2.0,
"eval_accuracy": 0.5099099099099099,
"eval_f1": 0.2670894996476392,
"eval_loss": 2.187363862991333,
"eval_precision": 0.309640522875817,
"eval_recall": 0.23482032218091697,
"eval_runtime": 0.3979,
"eval_samples_per_second": 731.333,
"eval_steps_per_second": 92.987,
"step": 170
},
{
"epoch": 3.0,
"eval_accuracy": 0.693993993993994,
"eval_f1": 0.47863247863247865,
"eval_loss": 1.2897497415542603,
"eval_precision": 0.4893203883495146,
"eval_recall": 0.4684014869888476,
"eval_runtime": 0.3852,
"eval_samples_per_second": 755.539,
"eval_steps_per_second": 96.065,
"step": 255
},
{
"epoch": 4.0,
"eval_accuracy": 0.8027027027027027,
"eval_f1": 0.6448598130841122,
"eval_loss": 0.8646674752235413,
"eval_precision": 0.6484962406015038,
"eval_recall": 0.6412639405204461,
"eval_runtime": 0.3916,
"eval_samples_per_second": 743.07,
"eval_steps_per_second": 94.48,
"step": 340
},
{
"epoch": 5.0,
"eval_accuracy": 0.8414414414414414,
"eval_f1": 0.7013867488443759,
"eval_loss": 0.6896626353263855,
"eval_precision": 0.69773145309626,
"eval_recall": 0.7050805452292441,
"eval_runtime": 0.3661,
"eval_samples_per_second": 794.894,
"eval_steps_per_second": 101.069,
"step": 425
}
],
"logging_steps": 500,
"max_steps": 1020,
"num_input_tokens_seen": 0,
"num_train_epochs": 12,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0,
"train_batch_size": 16,
"trial_name": null,
"trial_params": {
"learning_rate": 4.546321141328063e-05,
"num_train_epochs": 12,
"per_device_train_batch_size": 16,
"warmup_steps": 207,
"weight_decay": 0.23618328311920675
}
}
|