{ "best_metric": null, "best_model_checkpoint": null, "epoch": 4.901960784313726, "eval_steps": 500, "global_step": 1500, "is_hyper_param_search": true, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "eval_f1": 0.0, "eval_f1_sklearn": 0.9551506074222, "eval_loss": 0.19100283086299896, "eval_precision": 0.0, "eval_precision_sklearn": 0.9551506074222, "eval_recall": 0.0, "eval_recall_sklearn": 0.9551506074222, "eval_runtime": 4.3455, "eval_samples_per_second": 93.89, "eval_steps_per_second": 23.472, "step": 306 }, { "epoch": 1.6339869281045751, "grad_norm": 1.0916732549667358, "learning_rate": 6.73202614379085e-07, "loss": 0.3505, "step": 500 }, { "epoch": 2.0, "eval_f1": 0.0, "eval_f1_sklearn": 0.9551506074222, "eval_loss": 0.17071670293807983, "eval_precision": 0.0, "eval_precision_sklearn": 0.9551506074222, "eval_recall": 0.0, "eval_recall_sklearn": 0.9551506074222, "eval_runtime": 4.8134, "eval_samples_per_second": 84.764, "eval_steps_per_second": 21.191, "step": 612 }, { "epoch": 3.0, "eval_f1": 0.0, "eval_f1_sklearn": 0.9551506074222, "eval_loss": 0.15558268129825592, "eval_precision": 0.0, "eval_precision_sklearn": 0.9551506074222, "eval_recall": 0.0, "eval_recall_sklearn": 0.9551506074222, "eval_runtime": 4.3951, "eval_samples_per_second": 92.83, "eval_steps_per_second": 23.208, "step": 918 }, { "epoch": 3.2679738562091503, "grad_norm": 1.2043535709381104, "learning_rate": 3.464052287581699e-07, "loss": 0.1655, "step": 1000 }, { "epoch": 4.0, "eval_f1": 0.0, "eval_f1_sklearn": 0.9551506074222, "eval_loss": 0.1459077149629593, "eval_precision": 0.0, "eval_precision_sklearn": 0.9551506074222, "eval_recall": 0.0, "eval_recall_sklearn": 0.9551506074222, "eval_runtime": 5.1168, "eval_samples_per_second": 79.737, "eval_steps_per_second": 19.934, "step": 1224 }, { "epoch": 4.901960784313726, "grad_norm": 3.849971294403076, "learning_rate": 1.9607843137254902e-08, "loss": 0.1449, "step": 1500 } ], "logging_steps": 500, "max_steps": 1530, "num_input_tokens_seen": 0, "num_train_epochs": 5, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 454778868404304.0, "train_batch_size": 4, "trial_name": null, "trial_params": { "_wandb": {}, "assignments": {}, "learning_rate": 1e-06, "metric": "eval/loss", "per_device_train_batch_size": 4 } }