adversarial_qa_dbert_based_on / trainer_state.json
xummer's picture
End of training
b6f1657 verified
{
"best_metric": 0.3466796875,
"best_model_checkpoint": "/data/leeye/checkpoints_fft_t5_large/adversarial_qa_dbert_based_on/checkpoint-315",
"epoch": 5.0,
"eval_steps": 500,
"global_step": 315,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.31746031746031744,
"grad_norm": 18.570829391479492,
"learning_rate": 0.00018730158730158731,
"loss": 1.2416,
"step": 20
},
{
"epoch": 0.6349206349206349,
"grad_norm": 15.798161506652832,
"learning_rate": 0.00017460317460317462,
"loss": 1.0892,
"step": 40
},
{
"epoch": 0.9523809523809523,
"grad_norm": 7.127645015716553,
"learning_rate": 0.00016190476190476192,
"loss": 1.0162,
"step": 60
},
{
"epoch": 1.0,
"eval_bleu": 0.2748783583138883,
"eval_exact_match": 0.275390625,
"eval_loss": 0.7607467174530029,
"eval_runtime": 22.1309,
"eval_samples_per_second": 45.186,
"eval_steps_per_second": 0.723,
"step": 63
},
{
"epoch": 1.2698412698412698,
"grad_norm": 6.7987775802612305,
"learning_rate": 0.00014920634920634923,
"loss": 0.4064,
"step": 80
},
{
"epoch": 1.5873015873015874,
"grad_norm": 10.752936363220215,
"learning_rate": 0.0001365079365079365,
"loss": 0.3513,
"step": 100
},
{
"epoch": 1.9047619047619047,
"grad_norm": 7.6475348472595215,
"learning_rate": 0.0001238095238095238,
"loss": 0.3929,
"step": 120
},
{
"epoch": 2.0,
"eval_bleu": 0.24117803737423354,
"eval_exact_match": 0.2958984375,
"eval_loss": 0.794335126876831,
"eval_runtime": 24.0083,
"eval_samples_per_second": 41.652,
"eval_steps_per_second": 0.666,
"step": 126
},
{
"epoch": 2.2222222222222223,
"grad_norm": 2.862144947052002,
"learning_rate": 0.00011111111111111112,
"loss": 0.2161,
"step": 140
},
{
"epoch": 2.5396825396825395,
"grad_norm": 5.910730838775635,
"learning_rate": 9.841269841269841e-05,
"loss": 0.138,
"step": 160
},
{
"epoch": 2.857142857142857,
"grad_norm": 1.7475248575210571,
"learning_rate": 8.571428571428571e-05,
"loss": 0.1542,
"step": 180
},
{
"epoch": 3.0,
"eval_bleu": 0.27200128946842234,
"eval_exact_match": 0.3017578125,
"eval_loss": 1.0052680969238281,
"eval_runtime": 21.8202,
"eval_samples_per_second": 45.829,
"eval_steps_per_second": 0.733,
"step": 189
},
{
"epoch": 3.1746031746031744,
"grad_norm": 2.242680549621582,
"learning_rate": 7.301587301587302e-05,
"loss": 0.095,
"step": 200
},
{
"epoch": 3.492063492063492,
"grad_norm": 1.32904851436615,
"learning_rate": 6.0317460317460316e-05,
"loss": 0.0614,
"step": 220
},
{
"epoch": 3.8095238095238093,
"grad_norm": 37.23604202270508,
"learning_rate": 4.761904761904762e-05,
"loss": 0.0544,
"step": 240
},
{
"epoch": 4.0,
"eval_bleu": 0.31847923264557687,
"eval_exact_match": 0.345703125,
"eval_loss": 1.1005158424377441,
"eval_runtime": 18.54,
"eval_samples_per_second": 53.937,
"eval_steps_per_second": 0.863,
"step": 252
},
{
"epoch": 4.1269841269841265,
"grad_norm": 0.15852811932563782,
"learning_rate": 3.492063492063492e-05,
"loss": 0.0319,
"step": 260
},
{
"epoch": 4.444444444444445,
"grad_norm": 0.4201483130455017,
"learning_rate": 2.2222222222222223e-05,
"loss": 0.0249,
"step": 280
},
{
"epoch": 4.761904761904762,
"grad_norm": 0.46498769521713257,
"learning_rate": 9.523809523809523e-06,
"loss": 0.0239,
"step": 300
},
{
"epoch": 5.0,
"eval_bleu": 0.30828732530539343,
"eval_exact_match": 0.3466796875,
"eval_loss": 1.138079285621643,
"eval_runtime": 20.547,
"eval_samples_per_second": 48.669,
"eval_steps_per_second": 0.779,
"step": 315
},
{
"epoch": 5.0,
"step": 315,
"total_flos": 2586783250382848.0,
"train_loss": 0.3393354445222824,
"train_runtime": 835.7154,
"train_samples_per_second": 2.991,
"train_steps_per_second": 0.377
}
],
"logging_steps": 20,
"max_steps": 315,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 2586783250382848.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}