File size: 1,691 Bytes
1f7625e a0fb2e1 bdb1348 a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e a0fb2e1 1f7625e |
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 |
{
"best_metric": 0.7896281800391389,
"best_model_checkpoint": "tiny-bert-sst2-distilled/run-4/checkpoint-192",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 192,
"is_hyper_param_search": true,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 1.90240478515625,
"learning_rate": 0.0008040462940429662,
"loss": 0.4563,
"step": 96
},
{
"epoch": 1.0,
"eval_accuracy": 0.7446183953033269,
"eval_f1": 0.7473378509196515,
"eval_loss": 0.415781170129776,
"eval_precision": 0.7394636015325671,
"eval_recall": 0.7553816046966731,
"eval_runtime": 25.3619,
"eval_samples_per_second": 40.297,
"eval_steps_per_second": 1.262,
"step": 96
},
{
"epoch": 2.0,
"grad_norm": 0.7548394799232483,
"learning_rate": 0.0007147078169270811,
"loss": 0.4243,
"step": 192
},
{
"epoch": 2.0,
"eval_accuracy": 0.7896281800391389,
"eval_f1": 0.7969782813975448,
"eval_loss": 0.39640527963638306,
"eval_precision": 0.7700729927007299,
"eval_recall": 0.8258317025440313,
"eval_runtime": 25.8877,
"eval_samples_per_second": 39.478,
"eval_steps_per_second": 1.236,
"step": 192
}
],
"logging_steps": 500,
"max_steps": 960,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"save_steps": 500,
"total_flos": 471390394560.0,
"train_batch_size": 32,
"trial_name": null,
"trial_params": {
"alpha": 0.6601531096042508,
"learning_rate": 0.0008933847711588513,
"num_train_epochs": 10,
"temperature": 4
}
}
|