|
{ |
|
"best_metric": 0.9298920007321985, |
|
"best_model_checkpoint": "./fp32/models/qnli-roberta-base/checkpoint-9000", |
|
"epoch": 6.108735491753207, |
|
"global_step": 20000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 5.0890585241730285e-06, |
|
"loss": 0.5591, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"eval_accuracy": 0.8694856306058942, |
|
"eval_loss": 0.31558236479759216, |
|
"eval_runtime": 21.239, |
|
"eval_samples_per_second": 257.216, |
|
"eval_steps_per_second": 8.051, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 9.988627132412675e-06, |
|
"loss": 0.3512, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"eval_accuracy": 0.9007871133077063, |
|
"eval_loss": 0.2616254389286041, |
|
"eval_runtime": 15.8012, |
|
"eval_samples_per_second": 345.733, |
|
"eval_steps_per_second": 10.822, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 9.663688058489034e-06, |
|
"loss": 0.3075, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"eval_accuracy": 0.9126853377265239, |
|
"eval_loss": 0.23381730914115906, |
|
"eval_runtime": 15.8636, |
|
"eval_samples_per_second": 344.372, |
|
"eval_steps_per_second": 10.779, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"learning_rate": 9.338748984565394e-06, |
|
"loss": 0.2678, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"eval_accuracy": 0.920190371590701, |
|
"eval_loss": 0.22113902866840363, |
|
"eval_runtime": 18.4732, |
|
"eval_samples_per_second": 295.725, |
|
"eval_steps_per_second": 9.257, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"learning_rate": 9.013809910641755e-06, |
|
"loss": 0.2453, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"eval_accuracy": 0.9103056928427604, |
|
"eval_loss": 0.221624493598938, |
|
"eval_runtime": 15.1776, |
|
"eval_samples_per_second": 359.939, |
|
"eval_steps_per_second": 11.267, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"learning_rate": 8.688870836718116e-06, |
|
"loss": 0.2364, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"eval_accuracy": 0.923302214900238, |
|
"eval_loss": 0.20321357250213623, |
|
"eval_runtime": 14.6843, |
|
"eval_samples_per_second": 372.03, |
|
"eval_steps_per_second": 11.645, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 8.363931762794477e-06, |
|
"loss": 0.2109, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"eval_accuracy": 0.9240344133260113, |
|
"eval_loss": 0.20744949579238892, |
|
"eval_runtime": 15.2273, |
|
"eval_samples_per_second": 358.765, |
|
"eval_steps_per_second": 11.23, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"learning_rate": 8.038992688870837e-06, |
|
"loss": 0.1861, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"eval_accuracy": 0.9253157605711148, |
|
"eval_loss": 0.22822654247283936, |
|
"eval_runtime": 14.5847, |
|
"eval_samples_per_second": 374.572, |
|
"eval_steps_per_second": 11.725, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"learning_rate": 7.714053614947198e-06, |
|
"loss": 0.1857, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"eval_accuracy": 0.9298920007321985, |
|
"eval_loss": 0.20272189378738403, |
|
"eval_runtime": 14.8122, |
|
"eval_samples_per_second": 368.818, |
|
"eval_steps_per_second": 11.545, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"learning_rate": 7.389114541023559e-06, |
|
"loss": 0.1741, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"eval_accuracy": 0.9174446274940509, |
|
"eval_loss": 0.2495354562997818, |
|
"eval_runtime": 14.8558, |
|
"eval_samples_per_second": 367.736, |
|
"eval_steps_per_second": 11.511, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"learning_rate": 7.0641754670999195e-06, |
|
"loss": 0.1442, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"eval_accuracy": 0.9242174629324547, |
|
"eval_loss": 0.2571660578250885, |
|
"eval_runtime": 18.4719, |
|
"eval_samples_per_second": 295.746, |
|
"eval_steps_per_second": 9.257, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 6.73923639317628e-06, |
|
"loss": 0.1453, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"eval_accuracy": 0.9256818597840014, |
|
"eval_loss": 0.23131424188613892, |
|
"eval_runtime": 14.7976, |
|
"eval_samples_per_second": 369.181, |
|
"eval_steps_per_second": 11.556, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"learning_rate": 6.41429731925264e-06, |
|
"loss": 0.1458, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"eval_accuracy": 0.923851363719568, |
|
"eval_loss": 0.25046634674072266, |
|
"eval_runtime": 14.7742, |
|
"eval_samples_per_second": 369.767, |
|
"eval_steps_per_second": 11.574, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 4.28, |
|
"learning_rate": 6.089358245329001e-06, |
|
"loss": 0.1195, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 4.28, |
|
"eval_accuracy": 0.923851363719568, |
|
"eval_loss": 0.28182199597358704, |
|
"eval_runtime": 14.6725, |
|
"eval_samples_per_second": 372.328, |
|
"eval_steps_per_second": 11.654, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"learning_rate": 5.7644191714053624e-06, |
|
"loss": 0.1209, |
|
"step": 15000 |
|
}, |
|
{ |
|
"epoch": 4.58, |
|
"eval_accuracy": 0.9223869668680212, |
|
"eval_loss": 0.293856680393219, |
|
"eval_runtime": 15.1476, |
|
"eval_samples_per_second": 360.65, |
|
"eval_steps_per_second": 11.289, |
|
"step": 15000 |
|
}, |
|
{ |
|
"epoch": 4.89, |
|
"learning_rate": 5.439480097481723e-06, |
|
"loss": 0.12, |
|
"step": 16000 |
|
}, |
|
{ |
|
"epoch": 4.89, |
|
"eval_accuracy": 0.9245835621453414, |
|
"eval_loss": 0.28331321477890015, |
|
"eval_runtime": 17.008, |
|
"eval_samples_per_second": 321.202, |
|
"eval_steps_per_second": 10.054, |
|
"step": 16000 |
|
}, |
|
{ |
|
"epoch": 5.19, |
|
"learning_rate": 5.114541023558083e-06, |
|
"loss": 0.1086, |
|
"step": 17000 |
|
}, |
|
{ |
|
"epoch": 5.19, |
|
"eval_accuracy": 0.9249496613582281, |
|
"eval_loss": 0.3361791968345642, |
|
"eval_runtime": 16.1915, |
|
"eval_samples_per_second": 337.399, |
|
"eval_steps_per_second": 10.561, |
|
"step": 17000 |
|
}, |
|
{ |
|
"epoch": 5.5, |
|
"learning_rate": 4.789601949634444e-06, |
|
"loss": 0.0991, |
|
"step": 18000 |
|
}, |
|
{ |
|
"epoch": 5.5, |
|
"eval_accuracy": 0.923302214900238, |
|
"eval_loss": 0.3357701301574707, |
|
"eval_runtime": 15.4017, |
|
"eval_samples_per_second": 354.702, |
|
"eval_steps_per_second": 11.103, |
|
"step": 18000 |
|
}, |
|
{ |
|
"epoch": 5.8, |
|
"learning_rate": 4.4646628757108045e-06, |
|
"loss": 0.1033, |
|
"step": 19000 |
|
}, |
|
{ |
|
"epoch": 5.8, |
|
"eval_accuracy": 0.9264140582097748, |
|
"eval_loss": 0.36776551604270935, |
|
"eval_runtime": 18.05, |
|
"eval_samples_per_second": 302.659, |
|
"eval_steps_per_second": 9.474, |
|
"step": 19000 |
|
}, |
|
{ |
|
"epoch": 6.11, |
|
"learning_rate": 4.139723801787165e-06, |
|
"loss": 0.0952, |
|
"step": 20000 |
|
}, |
|
{ |
|
"epoch": 6.11, |
|
"eval_accuracy": 0.9267801574226615, |
|
"eval_loss": 0.36395037174224854, |
|
"eval_runtime": 20.1267, |
|
"eval_samples_per_second": 271.431, |
|
"eval_steps_per_second": 8.496, |
|
"step": 20000 |
|
} |
|
], |
|
"max_steps": 32740, |
|
"num_train_epochs": 10, |
|
"total_flos": 4.2087902193024e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|