{ "best_metric": 1.4704129695892334, "best_model_checkpoint": "data/outputs/distilbert-base-uncased-pii-200/checkpoint-7", "epoch": 7.0, "eval_steps": 500, "global_step": 7, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "eval_CITY_f1": 0.0, "eval_EMAIL_f1": 0.0, "eval_JOBTYPE_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_USERNAME_f1": 0.0, "eval_loss": 2.5786168575286865, "eval_overall_accuracy": 0.0, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 1.1226, "eval_samples_per_second": 0.891, "eval_steps_per_second": 0.891, "step": 1 }, { "epoch": 2.0, "eval_CITY_f1": 0.0, "eval_EMAIL_f1": 0.0, "eval_JOBTYPE_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_USERNAME_f1": 0.0, "eval_loss": 2.3959434032440186, "eval_overall_accuracy": 0.1774193548387097, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7362, "eval_samples_per_second": 1.358, "eval_steps_per_second": 1.358, "step": 2 }, { "epoch": 3.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 2.069570541381836, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7279, "eval_samples_per_second": 1.374, "eval_steps_per_second": 1.374, "step": 3 }, { "epoch": 4.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 1.8231066465377808, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7362, "eval_samples_per_second": 1.358, "eval_steps_per_second": 1.358, "step": 4 }, { "epoch": 5.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 1.6429799795150757, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7401, "eval_samples_per_second": 1.351, "eval_steps_per_second": 1.351, "step": 5 }, { "epoch": 6.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 1.5266300439834595, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7425, "eval_samples_per_second": 1.347, "eval_steps_per_second": 1.347, "step": 6 }, { "epoch": 7.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 1.4704129695892334, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7107, "eval_samples_per_second": 1.407, "eval_steps_per_second": 1.407, "step": 7 }, { "epoch": 7.0, "step": 7, "total_flos": 139354474896.0, "train_loss": 1.8857054029192244, "train_runtime": 26.5759, "train_samples_per_second": 0.527, "train_steps_per_second": 0.263 }, { "epoch": 7.0, "eval_EMAIL_f1": 0.0, "eval_LASTNAME_f1": 0.0, "eval_PREFIX_f1": 0.0, "eval_loss": 1.4704129695892334, "eval_overall_accuracy": 0.8064516129032258, "eval_overall_f1": 0.0, "eval_overall_precision": 0.0, "eval_overall_recall": 0.0, "eval_runtime": 0.7183, "eval_samples_per_second": 1.392, "eval_steps_per_second": 1.392, "step": 7 } ], "logging_steps": 500, "max_steps": 7, "num_input_tokens_seen": 0, "num_train_epochs": 7, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 139354474896.0, "train_batch_size": 32, "trial_name": null, "trial_params": null }