| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9826435246995995, | |
| "eval_steps": 3, | |
| "global_step": 46, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.06408544726301736, | |
| "grad_norm": 0.0, | |
| "learning_rate": 0.0, | |
| "loss": 2.8014, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.12817089452603472, | |
| "grad_norm": 0.0, | |
| "learning_rate": 0.0, | |
| "loss": 2.761, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.19225634178905207, | |
| "grad_norm": 0.0, | |
| "learning_rate": 0.0, | |
| "loss": 2.7789, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.25634178905206945, | |
| "grad_norm": 0.2742822766304016, | |
| "learning_rate": 0.002, | |
| "loss": 2.734, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.3204272363150868, | |
| "grad_norm": 0.2908117473125458, | |
| "learning_rate": 0.001414213562373095, | |
| "loss": 2.8294, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.38451268357810414, | |
| "grad_norm": 0.2908117473125458, | |
| "learning_rate": 0.001414213562373095, | |
| "loss": 2.8997, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.4485981308411215, | |
| "grad_norm": 2.0814929008483887, | |
| "learning_rate": 0.0008944271909999159, | |
| "loss": 3.4026, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.5126835781041389, | |
| "grad_norm": 1.6453497409820557, | |
| "learning_rate": 0.0007559289460184544, | |
| "loss": 3.7175, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.5767690253671562, | |
| "grad_norm": 0.6747546792030334, | |
| "learning_rate": 0.0006324555320336759, | |
| "loss": 2.5633, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.6408544726301736, | |
| "grad_norm": 0.4662072956562042, | |
| "learning_rate": 0.0005547001962252292, | |
| "loss": 2.277, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.7049399198931909, | |
| "grad_norm": 0.36622095108032227, | |
| "learning_rate": 0.0005, | |
| "loss": 2.144, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.7690253671562083, | |
| "grad_norm": 0.3787992298603058, | |
| "learning_rate": 0.0004588314677411235, | |
| "loss": 2.1074, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.8331108144192256, | |
| "grad_norm": 0.4321267902851105, | |
| "learning_rate": 0.00042640143271122083, | |
| "loss": 2.0358, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.897196261682243, | |
| "grad_norm": 0.4009685516357422, | |
| "learning_rate": 0.0004, | |
| "loss": 1.9321, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.9612817089452603, | |
| "grad_norm": 0.4228089451789856, | |
| "learning_rate": 0.0003779644730092272, | |
| "loss": 1.9272, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.9826435246995995, | |
| "step": 46, | |
| "total_flos": 4.671973340489974e+17, | |
| "train_loss": 2.5791222867758377, | |
| "train_runtime": 497.9904, | |
| "train_samples_per_second": 12.024, | |
| "train_steps_per_second": 0.092 | |
| } | |
| ], | |
| "logging_steps": 3, | |
| "max_steps": 46, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 3, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 4.671973340489974e+17, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |