|
{ |
|
"best_metric": 0.7928669410150893, |
|
"best_model_checkpoint": "/scratch/mrahma45/pixel/finetuned_models/mbert/mbert-base-finetuned-masakhaner-yor/checkpoint-4000", |
|
"epoch": 73.52941176470588, |
|
"global_step": 5000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 2.94, |
|
"eval_accuracy_score": 0.9617709065354884, |
|
"eval_f1": 0.7467362924281985, |
|
"eval_loss": 0.16836212575435638, |
|
"eval_precision": 0.7044334975369458, |
|
"eval_recall": 0.7944444444444444, |
|
"eval_runtime": 4.2439, |
|
"eval_samples_per_second": 71.868, |
|
"eval_steps_per_second": 9.19, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 5.88, |
|
"eval_accuracy_score": 0.9564300773014758, |
|
"eval_f1": 0.7172774869109947, |
|
"eval_loss": 0.20965266227722168, |
|
"eval_precision": 0.6782178217821783, |
|
"eval_recall": 0.7611111111111111, |
|
"eval_runtime": 4.2334, |
|
"eval_samples_per_second": 72.046, |
|
"eval_steps_per_second": 9.212, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 7.35, |
|
"learning_rate": 4.865771812080537e-05, |
|
"loss": 0.1207, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 8.82, |
|
"eval_accuracy_score": 0.9651440618411806, |
|
"eval_f1": 0.7470198675496688, |
|
"eval_loss": 0.19184653460979462, |
|
"eval_precision": 0.7139240506329114, |
|
"eval_recall": 0.7833333333333333, |
|
"eval_runtime": 4.2345, |
|
"eval_samples_per_second": 72.028, |
|
"eval_steps_per_second": 9.21, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 11.76, |
|
"eval_accuracy_score": 0.9675333801827126, |
|
"eval_f1": 0.7758389261744967, |
|
"eval_loss": 0.20845761895179749, |
|
"eval_precision": 0.7506493506493507, |
|
"eval_recall": 0.8027777777777778, |
|
"eval_runtime": 4.2327, |
|
"eval_samples_per_second": 72.058, |
|
"eval_steps_per_second": 9.214, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 14.71, |
|
"learning_rate": 4.697986577181208e-05, |
|
"loss": 0.0064, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 14.71, |
|
"eval_accuracy_score": 0.9676739283204497, |
|
"eval_f1": 0.768, |
|
"eval_loss": 0.20797914266586304, |
|
"eval_precision": 0.7384615384615385, |
|
"eval_recall": 0.8, |
|
"eval_runtime": 4.225, |
|
"eval_samples_per_second": 72.189, |
|
"eval_steps_per_second": 9.231, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 17.65, |
|
"eval_accuracy_score": 0.9612087139845397, |
|
"eval_f1": 0.7708609271523178, |
|
"eval_loss": 0.2564171552658081, |
|
"eval_precision": 0.7367088607594937, |
|
"eval_recall": 0.8083333333333333, |
|
"eval_runtime": 4.2295, |
|
"eval_samples_per_second": 72.112, |
|
"eval_steps_per_second": 9.221, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 20.59, |
|
"eval_accuracy_score": 0.967111735769501, |
|
"eval_f1": 0.7891156462585034, |
|
"eval_loss": 0.2191239595413208, |
|
"eval_precision": 0.7733333333333333, |
|
"eval_recall": 0.8055555555555556, |
|
"eval_runtime": 4.2308, |
|
"eval_samples_per_second": 72.09, |
|
"eval_steps_per_second": 9.218, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 22.06, |
|
"learning_rate": 4.530201342281879e-05, |
|
"loss": 0.0037, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 23.53, |
|
"eval_accuracy_score": 0.9680955727336613, |
|
"eval_f1": 0.7924016282225237, |
|
"eval_loss": 0.213937446475029, |
|
"eval_precision": 0.7745358090185677, |
|
"eval_recall": 0.8111111111111111, |
|
"eval_runtime": 4.2289, |
|
"eval_samples_per_second": 72.122, |
|
"eval_steps_per_second": 9.222, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 26.47, |
|
"eval_accuracy_score": 0.9631763879128602, |
|
"eval_f1": 0.7652645861601086, |
|
"eval_loss": 0.2679533362388611, |
|
"eval_precision": 0.7480106100795756, |
|
"eval_recall": 0.7833333333333333, |
|
"eval_runtime": 4.2299, |
|
"eval_samples_per_second": 72.106, |
|
"eval_steps_per_second": 9.22, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 29.41, |
|
"learning_rate": 4.36241610738255e-05, |
|
"loss": 0.0033, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 29.41, |
|
"eval_accuracy_score": 0.9633169360505973, |
|
"eval_f1": 0.7709190672153635, |
|
"eval_loss": 0.2611113488674164, |
|
"eval_precision": 0.7615176151761518, |
|
"eval_recall": 0.7805555555555556, |
|
"eval_runtime": 4.2112, |
|
"eval_samples_per_second": 72.426, |
|
"eval_steps_per_second": 9.261, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 32.35, |
|
"eval_accuracy_score": 0.9657062543921293, |
|
"eval_f1": 0.7735099337748345, |
|
"eval_loss": 0.2416420429944992, |
|
"eval_precision": 0.739240506329114, |
|
"eval_recall": 0.8111111111111111, |
|
"eval_runtime": 4.2178, |
|
"eval_samples_per_second": 72.312, |
|
"eval_steps_per_second": 9.246, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 35.29, |
|
"eval_accuracy_score": 0.9575544624033732, |
|
"eval_f1": 0.7470198675496688, |
|
"eval_loss": 0.26062682271003723, |
|
"eval_precision": 0.7139240506329114, |
|
"eval_recall": 0.7833333333333333, |
|
"eval_runtime": 4.2283, |
|
"eval_samples_per_second": 72.134, |
|
"eval_steps_per_second": 9.224, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 36.76, |
|
"learning_rate": 4.194630872483222e-05, |
|
"loss": 0.0033, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 38.24, |
|
"eval_accuracy_score": 0.9621925509486999, |
|
"eval_f1": 0.7560321715817695, |
|
"eval_loss": 0.2517440915107727, |
|
"eval_precision": 0.7305699481865285, |
|
"eval_recall": 0.7833333333333333, |
|
"eval_runtime": 4.221, |
|
"eval_samples_per_second": 72.257, |
|
"eval_steps_per_second": 9.239, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 41.18, |
|
"eval_accuracy_score": 0.9619114546732256, |
|
"eval_f1": 0.7779273216689098, |
|
"eval_loss": 0.25837406516075134, |
|
"eval_precision": 0.7545691906005222, |
|
"eval_recall": 0.8027777777777778, |
|
"eval_runtime": 4.2231, |
|
"eval_samples_per_second": 72.221, |
|
"eval_steps_per_second": 9.235, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 44.12, |
|
"learning_rate": 4.026845637583892e-05, |
|
"loss": 0.0021, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 44.12, |
|
"eval_accuracy_score": 0.9619114546732256, |
|
"eval_f1": 0.7852257181942545, |
|
"eval_loss": 0.2560689449310303, |
|
"eval_precision": 0.7735849056603774, |
|
"eval_recall": 0.7972222222222223, |
|
"eval_runtime": 4.2232, |
|
"eval_samples_per_second": 72.22, |
|
"eval_steps_per_second": 9.235, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 47.06, |
|
"eval_accuracy_score": 0.9631763879128602, |
|
"eval_f1": 0.757123473541384, |
|
"eval_loss": 0.24956972897052765, |
|
"eval_precision": 0.7400530503978779, |
|
"eval_recall": 0.775, |
|
"eval_runtime": 4.2237, |
|
"eval_samples_per_second": 72.212, |
|
"eval_steps_per_second": 9.234, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 50.0, |
|
"eval_accuracy_score": 0.9655657062543921, |
|
"eval_f1": 0.7722222222222223, |
|
"eval_loss": 0.28158125281333923, |
|
"eval_precision": 0.7722222222222223, |
|
"eval_recall": 0.7722222222222223, |
|
"eval_runtime": 4.2211, |
|
"eval_samples_per_second": 72.256, |
|
"eval_steps_per_second": 9.239, |
|
"step": 3400 |
|
}, |
|
{ |
|
"epoch": 51.47, |
|
"learning_rate": 3.859060402684564e-05, |
|
"loss": 0.002, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 52.94, |
|
"eval_accuracy_score": 0.9645818692902319, |
|
"eval_f1": 0.772117962466488, |
|
"eval_loss": 0.24671129882335663, |
|
"eval_precision": 0.7461139896373057, |
|
"eval_recall": 0.8, |
|
"eval_runtime": 4.2219, |
|
"eval_samples_per_second": 72.243, |
|
"eval_steps_per_second": 9.238, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 55.88, |
|
"eval_accuracy_score": 0.9659873506676037, |
|
"eval_f1": 0.7923497267759563, |
|
"eval_loss": 0.2602536678314209, |
|
"eval_precision": 0.7795698924731183, |
|
"eval_recall": 0.8055555555555556, |
|
"eval_runtime": 4.2241, |
|
"eval_samples_per_second": 72.205, |
|
"eval_steps_per_second": 9.233, |
|
"step": 3800 |
|
}, |
|
{ |
|
"epoch": 58.82, |
|
"learning_rate": 3.6912751677852356e-05, |
|
"loss": 0.0013, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 58.82, |
|
"eval_accuracy_score": 0.9665495432185524, |
|
"eval_f1": 0.7928669410150893, |
|
"eval_loss": 0.291377454996109, |
|
"eval_precision": 0.7831978319783198, |
|
"eval_recall": 0.8027777777777778, |
|
"eval_runtime": 4.2136, |
|
"eval_samples_per_second": 72.384, |
|
"eval_steps_per_second": 9.256, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 61.76, |
|
"eval_accuracy_score": 0.9658468025298664, |
|
"eval_f1": 0.7842605156037992, |
|
"eval_loss": 0.280934602022171, |
|
"eval_precision": 0.76657824933687, |
|
"eval_recall": 0.8027777777777778, |
|
"eval_runtime": 4.2253, |
|
"eval_samples_per_second": 72.183, |
|
"eval_steps_per_second": 9.23, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 64.71, |
|
"eval_accuracy_score": 0.9621925509486999, |
|
"eval_f1": 0.7779273216689098, |
|
"eval_loss": 0.3064756691455841, |
|
"eval_precision": 0.7545691906005222, |
|
"eval_recall": 0.8027777777777778, |
|
"eval_runtime": 4.2222, |
|
"eval_samples_per_second": 72.237, |
|
"eval_steps_per_second": 9.237, |
|
"step": 4400 |
|
}, |
|
{ |
|
"epoch": 66.18, |
|
"learning_rate": 3.523489932885906e-05, |
|
"loss": 0.0021, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 67.65, |
|
"eval_accuracy_score": 0.9673928320449754, |
|
"eval_f1": 0.7667121418826739, |
|
"eval_loss": 0.2502681016921997, |
|
"eval_precision": 0.7533512064343163, |
|
"eval_recall": 0.7805555555555556, |
|
"eval_runtime": 4.2225, |
|
"eval_samples_per_second": 72.232, |
|
"eval_steps_per_second": 9.236, |
|
"step": 4600 |
|
}, |
|
{ |
|
"epoch": 70.59, |
|
"eval_accuracy_score": 0.9641602248770204, |
|
"eval_f1": 0.7788331071913162, |
|
"eval_loss": 0.2737264335155487, |
|
"eval_precision": 0.7612732095490716, |
|
"eval_recall": 0.7972222222222223, |
|
"eval_runtime": 4.2229, |
|
"eval_samples_per_second": 72.225, |
|
"eval_steps_per_second": 9.235, |
|
"step": 4800 |
|
}, |
|
{ |
|
"epoch": 73.53, |
|
"learning_rate": 3.3557046979865775e-05, |
|
"loss": 0.0013, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 73.53, |
|
"eval_accuracy_score": 0.963879128601546, |
|
"eval_f1": 0.7750677506775068, |
|
"eval_loss": 0.2737545967102051, |
|
"eval_precision": 0.7566137566137566, |
|
"eval_recall": 0.7944444444444444, |
|
"eval_runtime": 4.2199, |
|
"eval_samples_per_second": 72.277, |
|
"eval_steps_per_second": 9.242, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 73.53, |
|
"step": 5000, |
|
"total_flos": 2.085737378893056e+16, |
|
"train_loss": 0.014607224118709565, |
|
"train_runtime": 5667.0091, |
|
"train_samples_per_second": 84.701, |
|
"train_steps_per_second": 2.647 |
|
} |
|
], |
|
"max_steps": 15000, |
|
"num_train_epochs": 221, |
|
"total_flos": 2.085737378893056e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|