hannah-khallaf's picture
Upload best checkpoint for bert-base-multilingual-cased_fold_1
d8c6cc5 verified
Raw
History Blame Contribute Delete
2.15 kB
{
"best_global_step": 1244,
"best_metric": 0.8563748091118463,
"best_model_checkpoint": "./results-PR/bert-base-multilingual-cased_fold_1/checkpoint-1244",
"epoch": 2.0,
"eval_steps": 500,
"global_step": 1244,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.8038585209003215,
"grad_norm": 10.667335510253906,
"learning_rate": 2.0056270096463023e-05,
"loss": 0.5782,
"step": 500
},
{
"epoch": 1.0,
"eval_accuracy": 0.8495575221238938,
"eval_f1": 0.849326689234278,
"eval_loss": 0.4893891513347626,
"eval_precision": 0.8491255592362772,
"eval_recall": 0.8495575221238938,
"eval_runtime": 5.1191,
"eval_samples_per_second": 242.818,
"eval_steps_per_second": 30.474,
"step": 622
},
{
"epoch": 1.607717041800643,
"grad_norm": 6.177433967590332,
"learning_rate": 4.0152733118971066e-05,
"loss": 0.5021,
"step": 1000
},
{
"epoch": 2.0,
"eval_accuracy": 0.8559935639581657,
"eval_f1": 0.8563748091118463,
"eval_loss": 0.49656593799591064,
"eval_precision": 0.8568520299040482,
"eval_recall": 0.8559935639581657,
"eval_runtime": 5.0715,
"eval_samples_per_second": 245.096,
"eval_steps_per_second": 30.76,
"step": 1244
}
],
"logging_steps": 500,
"max_steps": 12440,
"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": 0
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 2615323890278400.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}