roberta_large_pl_only / checkpoint-1446 /trainer_state.json
uvegesistvan's picture
Upload folder using huggingface_hub
9695ab8 verified
{
"best_global_step": 1446,
"best_metric": 0.5801101922988892,
"best_model_checkpoint": "/content/drive/MyDrive/modeling/checkpoint-1446",
"epoch": 3.0,
"eval_steps": 100,
"global_step": 1446,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 17.5683536529541,
"learning_rate": 4.501037344398341e-06,
"loss": 0.694,
"step": 482
},
{
"epoch": 1.0,
"eval_accuracy": 0.569055036344756,
"eval_f1": 0.6687948922585794,
"eval_loss": 0.6698787808418274,
"eval_precision": 0.5491480996068152,
"eval_recall": 0.8551020408163266,
"eval_runtime": 20.8749,
"eval_samples_per_second": 46.132,
"eval_steps_per_second": 2.922,
"step": 482
},
{
"epoch": 2.0,
"grad_norm": 17.880165100097656,
"learning_rate": 4.001037344398341e-06,
"loss": 0.6326,
"step": 964
},
{
"epoch": 2.0,
"eval_accuracy": 0.6656282450674974,
"eval_f1": 0.7238421955403087,
"eval_loss": 0.6021382808685303,
"eval_precision": 0.6242603550295858,
"eval_recall": 0.8612244897959184,
"eval_runtime": 20.6115,
"eval_samples_per_second": 46.722,
"eval_steps_per_second": 2.96,
"step": 964
},
{
"epoch": 3.0,
"grad_norm": 16.805747985839844,
"learning_rate": 3.5010373443983405e-06,
"loss": 0.5468,
"step": 1446
},
{
"epoch": 3.0,
"eval_accuracy": 0.6998961578400831,
"eval_f1": 0.7336405529953917,
"eval_loss": 0.5801101922988892,
"eval_precision": 0.66890756302521,
"eval_recall": 0.8122448979591836,
"eval_runtime": 20.6212,
"eval_samples_per_second": 46.699,
"eval_steps_per_second": 2.958,
"step": 1446
}
],
"logging_steps": 100,
"max_steps": 4820,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"save_steps": 100,
"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": 2.153600187514675e+16,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}