{ "best_metric": 2.537315607070923, "best_model_checkpoint": "output/led-zeppelin/checkpoint-66", "epoch": 2.0, "global_step": 66, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.15, "learning_rate": 0.00012957411177772776, "loss": 3.2905, "step": 5 }, { "epoch": 0.3, "learning_rate": 0.0001083919039965842, "loss": 2.9278, "step": 10 }, { "epoch": 0.45, "learning_rate": 7.836279790554737e-05, "loss": 2.9731, "step": 15 }, { "epoch": 0.61, "learning_rate": 4.616313771642488e-05, "loss": 2.8657, "step": 20 }, { "epoch": 0.76, "learning_rate": 1.8951844985992186e-05, "loss": 2.9708, "step": 25 }, { "epoch": 0.91, "learning_rate": 2.778782010045488e-06, "loss": 2.9287, "step": 30 }, { "epoch": 1.0, "eval_loss": 2.8552775382995605, "eval_runtime": 2.0716, "eval_samples_per_second": 22.688, "eval_steps_per_second": 2.896, "step": 33 }, { "epoch": 1.06, "learning_rate": 1.2396913677783124e-06, "loss": 2.9767, "step": 35 }, { "epoch": 1.21, "learning_rate": 1.4676757700644785e-05, "loss": 2.7654, "step": 40 }, { "epoch": 1.36, "learning_rate": 4.0102530108070535e-05, "loss": 2.8035, "step": 45 }, { "epoch": 1.52, "learning_rate": 7.186411942550872e-05, "loss": 2.7337, "step": 50 }, { "epoch": 1.67, "learning_rate": 0.00010290000000000001, "loss": 2.8151, "step": 55 }, { "epoch": 1.82, "learning_rate": 0.000126309992352219, "loss": 2.6463, "step": 60 }, { "epoch": 1.97, "learning_rate": 0.0001368893738885136, "loss": 2.5568, "step": 65 }, { "epoch": 2.0, "eval_loss": 2.537315607070923, "eval_runtime": 0.5866, "eval_samples_per_second": 75.011, "eval_steps_per_second": 10.229, "step": 66 } ], "max_steps": 66, "num_train_epochs": 2, "total_flos": 67413344256000.0, "trial_name": null, "trial_params": null }