Malicious_Prompt_Classifier / trainer_state.json
Al-Chan's picture
initial commit
c9d6bb2 verified
{
"best_metric": 0.0444386750459671,
"best_model_checkpoint": "572project/checkpoint-939",
"epoch": 5.0,
"eval_steps": 500,
"global_step": 4695,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.53,
"grad_norm": 0.0010740529978647828,
"learning_rate": 1.7870074547390843e-05,
"loss": 0.029,
"step": 500
},
{
"epoch": 1.0,
"eval_f1": 0.983853606027987,
"eval_loss": 0.0444386750459671,
"eval_runtime": 5.0208,
"eval_samples_per_second": 187.023,
"eval_steps_per_second": 23.502,
"step": 939
},
{
"epoch": 1.06,
"grad_norm": 0.029427560046315193,
"learning_rate": 1.5740149094781685e-05,
"loss": 0.0461,
"step": 1000
},
{
"epoch": 1.6,
"grad_norm": 78.75341796875,
"learning_rate": 1.3610223642172523e-05,
"loss": 0.0114,
"step": 1500
},
{
"epoch": 2.0,
"eval_f1": 0.9924324324324325,
"eval_loss": 0.054377324879169464,
"eval_runtime": 5.0274,
"eval_samples_per_second": 186.775,
"eval_steps_per_second": 23.471,
"step": 1878
},
{
"epoch": 2.13,
"grad_norm": 0.0012783489655703306,
"learning_rate": 1.1480298189563365e-05,
"loss": 0.0062,
"step": 2000
},
{
"epoch": 2.66,
"grad_norm": 0.00034116956521756947,
"learning_rate": 9.350372736954207e-06,
"loss": 0.0013,
"step": 2500
},
{
"epoch": 3.0,
"eval_f1": 0.9934924078091107,
"eval_loss": 0.047857534140348434,
"eval_runtime": 5.0032,
"eval_samples_per_second": 187.679,
"eval_steps_per_second": 23.585,
"step": 2817
},
{
"epoch": 3.19,
"grad_norm": 0.00018943840404972434,
"learning_rate": 7.220447284345049e-06,
"loss": 0.0015,
"step": 3000
},
{
"epoch": 3.73,
"grad_norm": 0.00020874786423519254,
"learning_rate": 5.090521831735889e-06,
"loss": 0.002,
"step": 3500
},
{
"epoch": 4.0,
"eval_f1": 0.992399565689468,
"eval_loss": 0.054781731218099594,
"eval_runtime": 5.0055,
"eval_samples_per_second": 187.594,
"eval_steps_per_second": 23.574,
"step": 3756
},
{
"epoch": 4.26,
"grad_norm": 0.00014990718045737594,
"learning_rate": 2.9605963791267307e-06,
"loss": 0.002,
"step": 4000
},
{
"epoch": 4.79,
"grad_norm": 0.00011578563135117292,
"learning_rate": 8.306709265175719e-07,
"loss": 0.0,
"step": 4500
},
{
"epoch": 5.0,
"eval_f1": 0.9913419913419914,
"eval_loss": 0.06532972306013107,
"eval_runtime": 5.0316,
"eval_samples_per_second": 186.621,
"eval_steps_per_second": 23.452,
"step": 4695
}
],
"logging_steps": 500,
"max_steps": 4695,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"total_flos": 1871333823963648.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}