{ "best_metric": null, "best_model_checkpoint": null, "epoch": 3.764705882352941, "eval_steps": 25, "global_step": 8, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.47058823529411764, "grad_norm": 1.1976841688156128, "learning_rate": 0.000175, "loss": 9.5754, "step": 1 }, { "epoch": 0.9411764705882353, "grad_norm": 1.1408287286758423, "learning_rate": 0.00015000000000000001, "loss": 9.3978, "step": 2 }, { "epoch": 1.4117647058823528, "grad_norm": 1.1529213190078735, "learning_rate": 0.000125, "loss": 9.2123, "step": 3 }, { "epoch": 1.8823529411764706, "grad_norm": 1.337565302848816, "learning_rate": 0.0001, "loss": 8.9941, "step": 4 }, { "epoch": 2.3529411764705883, "grad_norm": 1.5665826797485352, "learning_rate": 7.500000000000001e-05, "loss": 8.8097, "step": 5 }, { "epoch": 2.8235294117647056, "grad_norm": 1.7597465515136719, "learning_rate": 5e-05, "loss": 8.7077, "step": 6 }, { "epoch": 3.2941176470588234, "grad_norm": 2.1072466373443604, "learning_rate": 2.5e-05, "loss": 8.4972, "step": 7 }, { "epoch": 3.764705882352941, "grad_norm": 2.2672441005706787, "learning_rate": 0.0, "loss": 8.4222, "step": 8 }, { "epoch": 3.764705882352941, "step": 8, "total_flos": 3464493556500.0, "train_loss": 8.952053427696228, "train_runtime": 296.0676, "train_samples_per_second": 0.446, "train_steps_per_second": 0.027 } ], "logging_steps": 1.0, "max_steps": 8, "num_input_tokens_seen": 0, "num_train_epochs": 4, "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": 3464493556500.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }