BlueSecureBERT / checkpoint-1500 /trainer_state.json
Maxime Turlot
Add BlueSecureBERT weights & tokenizer v1.0
f8766c7
{
"best_global_step": 1500,
"best_metric": 0.9490445859872612,
"best_model_checkpoint": "/workspace/AI/Trend_Primus-FineWeb_Filtering-pipeline/securebert_finetuned/defensive_vs_rest/checkpoint-1500",
"epoch": 2.7272727272727275,
"eval_steps": 300,
"global_step": 1500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.18181818181818182,
"grad_norm": 1.2393633127212524,
"learning_rate": 5.6571428571428576e-06,
"loss": 0.1439,
"step": 100
},
{
"epoch": 0.36363636363636365,
"grad_norm": 1.2253516912460327,
"learning_rate": 1.1371428571428573e-05,
"loss": 0.0855,
"step": 200
},
{
"epoch": 0.5454545454545454,
"grad_norm": 0.6038007140159607,
"learning_rate": 1.708571428571429e-05,
"loss": 0.0369,
"step": 300
},
{
"epoch": 0.5454545454545454,
"eval_f1": 0.8451570393147375,
"eval_f2": 0.9234018045690152,
"eval_loss": 0.022321535274386406,
"eval_precision": 0.7405696689761355,
"eval_recall": 0.9841432225063939,
"eval_runtime": 10.9053,
"eval_samples_per_second": 717.082,
"eval_steps_per_second": 11.279,
"step": 300
},
{
"epoch": 0.7272727272727273,
"grad_norm": 0.47052955627441406,
"learning_rate": 1.924615384615385e-05,
"loss": 0.0203,
"step": 400
},
{
"epoch": 0.9090909090909091,
"grad_norm": 2.0642011165618896,
"learning_rate": 1.770769230769231e-05,
"loss": 0.0176,
"step": 500
},
{
"epoch": 1.0909090909090908,
"grad_norm": 0.28900912404060364,
"learning_rate": 1.6169230769230772e-05,
"loss": 0.0116,
"step": 600
},
{
"epoch": 1.0909090909090908,
"eval_f1": 0.9583852479441815,
"eval_f2": 0.9733751771613687,
"eval_loss": 0.013802087865769863,
"eval_precision": 0.934402332361516,
"eval_recall": 0.9836317135549872,
"eval_runtime": 11.4515,
"eval_samples_per_second": 682.883,
"eval_steps_per_second": 10.741,
"step": 600
},
{
"epoch": 1.2727272727272727,
"grad_norm": 0.4510941803455353,
"learning_rate": 1.4630769230769233e-05,
"loss": 0.0083,
"step": 700
},
{
"epoch": 1.4545454545454546,
"grad_norm": 0.546210765838623,
"learning_rate": 1.3092307692307693e-05,
"loss": 0.0074,
"step": 800
},
{
"epoch": 1.6363636363636362,
"grad_norm": 2.2349956035614014,
"learning_rate": 1.1553846153846156e-05,
"loss": 0.0084,
"step": 900
},
{
"epoch": 1.6363636363636362,
"eval_f1": 0.9500734214390602,
"eval_f2": 0.9752788664455834,
"eval_loss": 0.009357056580483913,
"eval_precision": 0.9108399812294697,
"eval_recall": 0.992838874680307,
"eval_runtime": 11.0303,
"eval_samples_per_second": 708.955,
"eval_steps_per_second": 11.151,
"step": 900
},
{
"epoch": 1.8181818181818183,
"grad_norm": 0.4002629518508911,
"learning_rate": 1.0015384615384615e-05,
"loss": 0.0076,
"step": 1000
},
{
"epoch": 2.0,
"grad_norm": 0.0686814934015274,
"learning_rate": 8.476923076923078e-06,
"loss": 0.0063,
"step": 1100
},
{
"epoch": 2.1818181818181817,
"grad_norm": 0.40794429183006287,
"learning_rate": 6.9384615384615395e-06,
"loss": 0.0032,
"step": 1200
},
{
"epoch": 2.1818181818181817,
"eval_f1": 0.9634601043997018,
"eval_f2": 0.9799757281553398,
"eval_loss": 0.009985245764255524,
"eval_precision": 0.937137330754352,
"eval_recall": 0.991304347826087,
"eval_runtime": 10.5643,
"eval_samples_per_second": 740.231,
"eval_steps_per_second": 11.643,
"step": 1200
},
{
"epoch": 2.3636363636363638,
"grad_norm": 0.26694178581237793,
"learning_rate": 5.400000000000001e-06,
"loss": 0.0031,
"step": 1300
},
{
"epoch": 2.5454545454545454,
"grad_norm": 0.1839500516653061,
"learning_rate": 3.861538461538462e-06,
"loss": 0.0028,
"step": 1400
},
{
"epoch": 2.7272727272727275,
"grad_norm": 0.3117663264274597,
"learning_rate": 2.3230769230769234e-06,
"loss": 0.0021,
"step": 1500
},
{
"epoch": 2.7272727272727275,
"eval_f1": 0.9694694694694694,
"eval_f2": 0.9821519115708346,
"eval_loss": 0.010657339356839657,
"eval_precision": 0.9490445859872612,
"eval_recall": 0.9907928388746803,
"eval_runtime": 10.7067,
"eval_samples_per_second": 730.384,
"eval_steps_per_second": 11.488,
"step": 1500
}
],
"logging_steps": 100,
"max_steps": 1650,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 300,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 2,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 0
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 5.05067981869056e+16,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}