| { |
| "best_global_step": null, |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 1.0, |
| "eval_steps": 1000, |
| "global_step": 6148, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.00016265452179570593, |
| "grad_norm": 6.144397258758545, |
| "learning_rate": 0.0, |
| "loss": 3.8954, |
| "step": 1 |
| }, |
| { |
| "epoch": 0.16265452179570591, |
| "grad_norm": 4.344635009765625, |
| "learning_rate": 1.624390243902439e-05, |
| "loss": 3.0884, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.16265452179570591, |
| "eval_cosine_accuracy": 0.9446439743041992, |
| "eval_loss": 1.014482855796814, |
| "eval_runtime": 39.0277, |
| "eval_samples_per_second": 242.545, |
| "eval_steps_per_second": 0.948, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.32530904359141183, |
| "grad_norm": 4.419331073760986, |
| "learning_rate": 2.9721669980119285e-05, |
| "loss": 2.3127, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.32530904359141183, |
| "eval_cosine_accuracy": 0.9529896378517151, |
| "eval_loss": 0.9663239121437073, |
| "eval_runtime": 32.6394, |
| "eval_samples_per_second": 290.018, |
| "eval_steps_per_second": 1.134, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.48796356538711777, |
| "grad_norm": 4.359193325042725, |
| "learning_rate": 2.7914332188686065e-05, |
| "loss": 2.192, |
| "step": 3000 |
| }, |
| { |
| "epoch": 0.48796356538711777, |
| "eval_cosine_accuracy": 0.9564757943153381, |
| "eval_loss": 0.921445369720459, |
| "eval_runtime": 31.8747, |
| "eval_samples_per_second": 296.975, |
| "eval_steps_per_second": 1.161, |
| "step": 3000 |
| }, |
| { |
| "epoch": 0.6506180871828237, |
| "grad_norm": 4.2792558670043945, |
| "learning_rate": 2.6106994397252848e-05, |
| "loss": 2.466, |
| "step": 4000 |
| }, |
| { |
| "epoch": 0.6506180871828237, |
| "eval_cosine_accuracy": 0.9553137421607971, |
| "eval_loss": 0.8910627365112305, |
| "eval_runtime": 31.5443, |
| "eval_samples_per_second": 300.086, |
| "eval_steps_per_second": 1.173, |
| "step": 4000 |
| }, |
| { |
| "epoch": 0.8132726089785296, |
| "grad_norm": 3.616072177886963, |
| "learning_rate": 2.430146394361106e-05, |
| "loss": 3.0285, |
| "step": 5000 |
| }, |
| { |
| "epoch": 0.8132726089785296, |
| "eval_cosine_accuracy": 0.947284996509552, |
| "eval_loss": 0.8859251141548157, |
| "eval_runtime": 31.4273, |
| "eval_samples_per_second": 301.203, |
| "eval_steps_per_second": 1.177, |
| "step": 5000 |
| }, |
| { |
| "epoch": 0.9759271307742355, |
| "grad_norm": 3.35726261138916, |
| "learning_rate": 2.249412615217784e-05, |
| "loss": 3.3706, |
| "step": 6000 |
| }, |
| { |
| "epoch": 0.9759271307742355, |
| "eval_cosine_accuracy": 0.9297485947608948, |
| "eval_loss": 0.9370076060295105, |
| "eval_runtime": 31.4172, |
| "eval_samples_per_second": 301.3, |
| "eval_steps_per_second": 1.178, |
| "step": 6000 |
| } |
| ], |
| "logging_steps": 1000, |
| "max_steps": 18444, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 3, |
| "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": 0.0, |
| "train_batch_size": 256, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
