roberta-base-ca_fold_1 / trainer_state.json
hannah-khallaf's picture
Upload best checkpoint for roberta-base-ca_fold_1
1ad8cc5 verified
Raw
History Blame Contribute Delete
4.33 kB
{
"best_global_step": 216,
"best_metric": 0.8041111111111111,
"best_model_checkpoint": "./results/PlanTL-GOB-ES/roberta-base-ca_fold_1/checkpoint-216",
"epoch": 9.0,
"eval_steps": 500,
"global_step": 324,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_accuracy": 0.8055555555555556,
"eval_f1": 0.7188034188034188,
"eval_loss": 1.0945801734924316,
"eval_precision": 0.6489197530864197,
"eval_recall": 0.8055555555555556,
"eval_runtime": 0.215,
"eval_samples_per_second": 334.959,
"eval_steps_per_second": 41.87,
"step": 36
},
{
"epoch": 2.0,
"eval_accuracy": 0.8055555555555556,
"eval_f1": 0.7188034188034188,
"eval_loss": 1.0182355642318726,
"eval_precision": 0.6489197530864197,
"eval_recall": 0.8055555555555556,
"eval_runtime": 0.2151,
"eval_samples_per_second": 334.721,
"eval_steps_per_second": 41.84,
"step": 72
},
{
"epoch": 3.0,
"eval_accuracy": 0.8333333333333334,
"eval_f1": 0.7751322751322751,
"eval_loss": 0.9727938771247864,
"eval_precision": 0.727328431372549,
"eval_recall": 0.8333333333333334,
"eval_runtime": 0.2102,
"eval_samples_per_second": 342.517,
"eval_steps_per_second": 42.815,
"step": 108
},
{
"epoch": 4.0,
"eval_accuracy": 0.8333333333333334,
"eval_f1": 0.7937361324458098,
"eval_loss": 1.00287926197052,
"eval_precision": 0.7703072390572391,
"eval_recall": 0.8333333333333334,
"eval_runtime": 0.2118,
"eval_samples_per_second": 339.898,
"eval_steps_per_second": 42.487,
"step": 144
},
{
"epoch": 5.0,
"eval_accuracy": 0.8194444444444444,
"eval_f1": 0.7840406405980177,
"eval_loss": 1.0656781196594238,
"eval_precision": 0.76171875,
"eval_recall": 0.8194444444444444,
"eval_runtime": 0.213,
"eval_samples_per_second": 337.978,
"eval_steps_per_second": 42.247,
"step": 180
},
{
"epoch": 6.0,
"eval_accuracy": 0.8472222222222222,
"eval_f1": 0.8041111111111111,
"eval_loss": 1.0739294290542603,
"eval_precision": 0.806851160862355,
"eval_recall": 0.8472222222222222,
"eval_runtime": 0.2141,
"eval_samples_per_second": 336.243,
"eval_steps_per_second": 42.03,
"step": 216
},
{
"epoch": 7.0,
"eval_accuracy": 0.8472222222222222,
"eval_f1": 0.8041111111111111,
"eval_loss": 1.0710959434509277,
"eval_precision": 0.806851160862355,
"eval_recall": 0.8472222222222222,
"eval_runtime": 0.2124,
"eval_samples_per_second": 339.012,
"eval_steps_per_second": 42.376,
"step": 252
},
{
"epoch": 8.0,
"eval_accuracy": 0.8472222222222222,
"eval_f1": 0.8041111111111111,
"eval_loss": 1.088027000427246,
"eval_precision": 0.806851160862355,
"eval_recall": 0.8472222222222222,
"eval_runtime": 0.2114,
"eval_samples_per_second": 340.652,
"eval_steps_per_second": 42.582,
"step": 288
},
{
"epoch": 9.0,
"eval_accuracy": 0.8472222222222222,
"eval_f1": 0.8041111111111111,
"eval_loss": 1.0886331796646118,
"eval_precision": 0.806851160862355,
"eval_recall": 0.8472222222222222,
"eval_runtime": 0.2146,
"eval_samples_per_second": 335.545,
"eval_steps_per_second": 41.943,
"step": 324
}
],
"logging_steps": 500,
"max_steps": 720,
"num_input_tokens_seen": 0,
"num_train_epochs": 20,
"save_steps": 500,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 3,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 3
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 54236147045652.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}