File size: 1,727 Bytes
e12486c |
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 |
{
"best_metric": 0.7827788649706457,
"best_model_checkpoint": "tiny-bert-sst2-distilled/run-9/checkpoint-162",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 162,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 0.9513042569160461,
"learning_rate": 0.0005954099159667608,
"loss": 0.499,
"step": 81
},
{
"epoch": 1.0,
"eval_accuracy": 0.6829745596868885,
"eval_f1": 0.6067961165048544,
"eval_loss": 0.4608267843723297,
"eval_precision": 0.7987220447284346,
"eval_recall": 0.4892367906066536,
"eval_runtime": 28.3141,
"eval_samples_per_second": 36.095,
"eval_steps_per_second": 1.13,
"step": 81
},
{
"epoch": 2.0,
"grad_norm": 1.7321057319641113,
"learning_rate": 0.0004961749299723006,
"loss": 0.4578,
"step": 162
},
{
"epoch": 2.0,
"eval_accuracy": 0.7827788649706457,
"eval_f1": 0.798913043478261,
"eval_loss": 0.42703160643577576,
"eval_precision": 0.7436762225969646,
"eval_recall": 0.863013698630137,
"eval_runtime": 28.5611,
"eval_samples_per_second": 35.783,
"eval_steps_per_second": 1.12,
"step": 162
}
],
"logging_steps": 500,
"max_steps": 567,
"num_input_tokens_seen": 0,
"num_train_epochs": 7,
"save_steps": 500,
"total_flos": 471390394560.0,
"train_batch_size": 38,
"trial_name": null,
"trial_params": {
"alpha": 0.7591558093291033,
"learning_rate": 0.0006946449019612209,
"num_train_epochs": 7,
"per_device_train_batch_size": 38,
"temperature": 16
}
}
|