{ "best_metric": 1.3153774738311768, "best_model_checkpoint": "./output/checkpoints/2024-06-11_10-50-46/checkpoint-10", "epoch": 0.5555555555555556, "eval_steps": 1, "global_step": 10, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.05555555555555555, "grad_norm": 1.9254094362258911, "learning_rate": 0.0002, "loss": 3.544, "step": 1 }, { "epoch": 0.05555555555555555, "eval_loss": 3.981394052505493, "eval_runtime": 12.6022, "eval_samples_per_second": 11.347, "eval_steps_per_second": 0.714, "step": 1 }, { "epoch": 0.1111111111111111, "grad_norm": 2.0063693523406982, "learning_rate": 0.0004, "loss": 3.6373, "step": 2 }, { "epoch": 0.1111111111111111, "eval_loss": 3.173820972442627, "eval_runtime": 12.6737, "eval_samples_per_second": 11.283, "eval_steps_per_second": 0.71, "step": 2 }, { "epoch": 0.16666666666666666, "grad_norm": 2.0154714584350586, "learning_rate": 0.000375, "loss": 2.7246, "step": 3 }, { "epoch": 0.16666666666666666, "eval_loss": 1.9138473272323608, "eval_runtime": 12.6464, "eval_samples_per_second": 11.308, "eval_steps_per_second": 0.712, "step": 3 }, { "epoch": 0.2222222222222222, "grad_norm": 1.1926450729370117, "learning_rate": 0.00035, "loss": 1.4998, "step": 4 }, { "epoch": 0.2222222222222222, "eval_loss": 1.4986788034439087, "eval_runtime": 12.7087, "eval_samples_per_second": 11.252, "eval_steps_per_second": 0.708, "step": 4 }, { "epoch": 0.2777777777777778, "grad_norm": 0.8283662796020508, "learning_rate": 0.00032500000000000004, "loss": 1.1373, "step": 5 }, { "epoch": 0.2777777777777778, "eval_loss": 1.3153411149978638, "eval_runtime": 12.7034, "eval_samples_per_second": 11.257, "eval_steps_per_second": 0.708, "step": 5 }, { "epoch": 0.3333333333333333, "grad_norm": 0.6860641241073608, "learning_rate": 0.00030000000000000003, "loss": 0.9828, "step": 6 }, { "epoch": 0.3333333333333333, "eval_loss": 1.247497797012329, "eval_runtime": 12.6669, "eval_samples_per_second": 11.289, "eval_steps_per_second": 0.711, "step": 6 }, { "epoch": 0.3888888888888889, "grad_norm": 0.2969834506511688, "learning_rate": 0.000275, "loss": 0.91, "step": 7 }, { "epoch": 0.3888888888888889, "eval_loss": 1.2433477640151978, "eval_runtime": 12.6841, "eval_samples_per_second": 11.274, "eval_steps_per_second": 0.71, "step": 7 }, { "epoch": 0.4444444444444444, "grad_norm": 0.1941613256931305, "learning_rate": 0.00025, "loss": 0.839, "step": 8 }, { "epoch": 0.4444444444444444, "eval_loss": 1.2696853876113892, "eval_runtime": 12.7426, "eval_samples_per_second": 11.222, "eval_steps_per_second": 0.706, "step": 8 }, { "epoch": 0.5, "grad_norm": 0.19763241708278656, "learning_rate": 0.00022500000000000002, "loss": 0.7744, "step": 9 }, { "epoch": 0.5, "eval_loss": 1.3031294345855713, "eval_runtime": 12.7085, "eval_samples_per_second": 11.252, "eval_steps_per_second": 0.708, "step": 9 }, { "epoch": 0.5555555555555556, "grad_norm": 0.20671840012073517, "learning_rate": 0.0002, "loss": 0.7289, "step": 10 }, { "epoch": 0.5555555555555556, "eval_loss": 1.3153774738311768, "eval_runtime": 12.6745, "eval_samples_per_second": 11.283, "eval_steps_per_second": 0.71, "step": 10 } ], "logging_steps": 1, "max_steps": 18, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 10, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 3708922028359680.0, "train_batch_size": 32, "trial_name": null, "trial_params": null }