{ "best_metric": 1.7459986209869385, "best_model_checkpoint": "output/big-baby-tape/checkpoint-520", "epoch": 4.0, "global_step": 520, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04, "learning_rate": 5.001712368734975e-07, "loss": 1.923, "step": 5 }, { "epoch": 0.08, "learning_rate": 1.9933913245728244e-06, "loss": 2.0321, "step": 10 }, { "epoch": 0.12, "learning_rate": 4.457885751780542e-06, "loss": 2.0426, "step": 15 }, { "epoch": 0.15, "learning_rate": 7.857716640189785e-06, "loss": 2.0776, "step": 20 }, { "epoch": 0.19, "learning_rate": 1.2143306799695167e-05, "loss": 1.9711, "step": 25 }, { "epoch": 0.23, "learning_rate": 1.7252162675462473e-05, "loss": 2.0637, "step": 30 }, { "epoch": 0.27, "learning_rate": 2.3109785644681427e-05, "loss": 1.8713, "step": 35 }, { "epoch": 0.31, "learning_rate": 2.9630758374242703e-05, "loss": 2.088, "step": 40 }, { "epoch": 0.35, "learning_rate": 3.671999039779749e-05, "loss": 1.9995, "step": 45 }, { "epoch": 0.38, "learning_rate": 4.427410474888204e-05, "loss": 2.0004, "step": 50 }, { "epoch": 0.42, "learning_rate": 5.218294542987354e-05, "loss": 1.9791, "step": 55 }, { "epoch": 0.46, "learning_rate": 6.03311837344848e-05, "loss": 1.882, "step": 60 }, { "epoch": 0.5, "learning_rate": 6.859999999999999e-05, "loss": 1.8736, "step": 65 }, { "epoch": 0.54, "learning_rate": 7.686881626551516e-05, "loss": 1.9173, "step": 70 }, { "epoch": 0.58, "learning_rate": 8.501705457012643e-05, "loss": 2.0122, "step": 75 }, { "epoch": 0.62, "learning_rate": 9.292589525111794e-05, "loss": 1.871, "step": 80 }, { "epoch": 0.65, "learning_rate": 0.00010048000960220248, "loss": 2.0619, "step": 85 }, { "epoch": 0.69, "learning_rate": 0.00010756924162575728, "loss": 1.9238, "step": 90 }, { "epoch": 0.73, "learning_rate": 0.00011409021435531856, "loss": 2.0238, "step": 95 }, { "epoch": 0.77, "learning_rate": 0.0001199478373245375, "loss": 1.9745, "step": 100 }, { "epoch": 0.81, "learning_rate": 0.00012505669320030482, "loss": 1.9809, "step": 105 }, { "epoch": 0.85, "learning_rate": 0.0001293422833598102, "loss": 1.8881, "step": 110 }, { "epoch": 0.88, "learning_rate": 0.00013274211424821946, "loss": 1.8788, "step": 115 }, { "epoch": 0.92, "learning_rate": 0.00013520660867542716, "loss": 1.984, "step": 120 }, { "epoch": 0.96, "learning_rate": 0.00013669982876312649, "loss": 1.9443, "step": 125 }, { "epoch": 1.0, "learning_rate": 0.0001372, "loss": 1.9006, "step": 130 }, { "epoch": 1.0, "eval_loss": 1.9121332168579102, "eval_runtime": 9.6453, "eval_samples_per_second": 20.217, "eval_steps_per_second": 2.592, "step": 130 }, { "epoch": 1.04, "learning_rate": 0.0001366998287631265, "loss": 1.8831, "step": 135 }, { "epoch": 1.08, "learning_rate": 0.00013520660867542716, "loss": 1.8547, "step": 140 }, { "epoch": 1.12, "learning_rate": 0.00013274211424821946, "loss": 1.87, "step": 145 }, { "epoch": 1.15, "learning_rate": 0.00012934228335981023, "loss": 1.8612, "step": 150 }, { "epoch": 1.19, "learning_rate": 0.00012505669320030482, "loss": 1.7875, "step": 155 }, { "epoch": 1.23, "learning_rate": 0.00011994783732453754, "loss": 1.8327, "step": 160 }, { "epoch": 1.27, "learning_rate": 0.00011409021435531858, "loss": 1.9541, "step": 165 }, { "epoch": 1.31, "learning_rate": 0.00010756924162575734, "loss": 1.8205, "step": 170 }, { "epoch": 1.35, "learning_rate": 0.00010048000960220251, "loss": 1.8993, "step": 175 }, { "epoch": 1.38, "learning_rate": 9.292589525111797e-05, "loss": 1.8431, "step": 180 }, { "epoch": 1.42, "learning_rate": 8.501705457012652e-05, "loss": 1.8127, "step": 185 }, { "epoch": 1.46, "learning_rate": 7.686881626551514e-05, "loss": 1.8428, "step": 190 }, { "epoch": 1.5, "learning_rate": 6.860000000000001e-05, "loss": 1.7509, "step": 195 }, { "epoch": 1.54, "learning_rate": 6.03311837344849e-05, "loss": 1.7859, "step": 200 }, { "epoch": 1.58, "learning_rate": 5.218294542987351e-05, "loss": 1.8358, "step": 205 }, { "epoch": 1.62, "learning_rate": 4.427410474888207e-05, "loss": 1.7393, "step": 210 }, { "epoch": 1.65, "learning_rate": 3.6719990397797524e-05, "loss": 1.8051, "step": 215 }, { "epoch": 1.69, "learning_rate": 2.9630758374242683e-05, "loss": 1.8083, "step": 220 }, { "epoch": 1.73, "learning_rate": 2.310978564468145e-05, "loss": 1.7062, "step": 225 }, { "epoch": 1.77, "learning_rate": 1.7252162675462497e-05, "loss": 1.7829, "step": 230 }, { "epoch": 1.81, "learning_rate": 1.214330679969522e-05, "loss": 1.7874, "step": 235 }, { "epoch": 1.85, "learning_rate": 7.8577166401898e-06, "loss": 1.8435, "step": 240 }, { "epoch": 1.88, "learning_rate": 4.457885751780558e-06, "loss": 1.8288, "step": 245 }, { "epoch": 1.92, "learning_rate": 1.9933913245728472e-06, "loss": 1.7984, "step": 250 }, { "epoch": 1.96, "learning_rate": 5.001712368734975e-07, "loss": 1.8331, "step": 255 }, { "epoch": 2.0, "learning_rate": 0.0, "loss": 1.8175, "step": 260 }, { "epoch": 2.0, "eval_loss": 1.8163535594940186, "eval_runtime": 9.657, "eval_samples_per_second": 20.193, "eval_steps_per_second": 2.589, "step": 260 }, { "epoch": 2.04, "learning_rate": 5.001712368734899e-07, "loss": 1.7297, "step": 265 }, { "epoch": 2.08, "learning_rate": 1.9933913245728396e-06, "loss": 1.5719, "step": 270 }, { "epoch": 2.12, "learning_rate": 4.457885751780535e-06, "loss": 1.7261, "step": 275 }, { "epoch": 2.15, "learning_rate": 7.857716640189778e-06, "loss": 1.6366, "step": 280 }, { "epoch": 2.19, "learning_rate": 1.2143306799695189e-05, "loss": 1.8032, "step": 285 }, { "epoch": 2.23, "learning_rate": 1.725216267546246e-05, "loss": 1.8041, "step": 290 }, { "epoch": 2.27, "learning_rate": 2.310978564468141e-05, "loss": 1.752, "step": 295 }, { "epoch": 2.31, "learning_rate": 2.9630758374242642e-05, "loss": 1.831, "step": 300 }, { "epoch": 2.35, "learning_rate": 3.671999039779748e-05, "loss": 1.6489, "step": 305 }, { "epoch": 2.38, "learning_rate": 4.427410474888202e-05, "loss": 1.7224, "step": 310 }, { "epoch": 2.42, "learning_rate": 5.218294542987346e-05, "loss": 1.6862, "step": 315 }, { "epoch": 2.46, "learning_rate": 6.033118373448485e-05, "loss": 1.7129, "step": 320 }, { "epoch": 2.5, "learning_rate": 6.859999999999997e-05, "loss": 1.9049, "step": 325 }, { "epoch": 2.54, "learning_rate": 7.68688162655151e-05, "loss": 1.8259, "step": 330 }, { "epoch": 2.58, "learning_rate": 8.501705457012648e-05, "loss": 1.7537, "step": 335 }, { "epoch": 2.62, "learning_rate": 9.292589525111793e-05, "loss": 1.7193, "step": 340 }, { "epoch": 2.65, "learning_rate": 0.00010048000960220248, "loss": 1.7921, "step": 345 }, { "epoch": 2.69, "learning_rate": 0.00010756924162575731, "loss": 1.7727, "step": 350 }, { "epoch": 2.73, "learning_rate": 0.00011409021435531856, "loss": 1.8267, "step": 355 }, { "epoch": 2.77, "learning_rate": 0.0001199478373245375, "loss": 1.6066, "step": 360 }, { "epoch": 2.81, "learning_rate": 0.0001250566932003048, "loss": 1.757, "step": 365 }, { "epoch": 2.85, "learning_rate": 0.00012934228335981018, "loss": 1.7909, "step": 370 }, { "epoch": 2.88, "learning_rate": 0.00013274211424821943, "loss": 1.7481, "step": 375 }, { "epoch": 2.92, "learning_rate": 0.00013520660867542716, "loss": 1.7786, "step": 380 }, { "epoch": 2.96, "learning_rate": 0.00013669982876312649, "loss": 1.689, "step": 385 }, { "epoch": 3.0, "learning_rate": 0.0001372, "loss": 1.7754, "step": 390 }, { "epoch": 3.0, "eval_loss": 1.8065567016601562, "eval_runtime": 9.6575, "eval_samples_per_second": 20.192, "eval_steps_per_second": 2.589, "step": 390 }, { "epoch": 3.04, "learning_rate": 0.0001366998287631265, "loss": 1.6829, "step": 395 }, { "epoch": 3.08, "learning_rate": 0.0001352066086754272, "loss": 1.7295, "step": 400 }, { "epoch": 3.12, "learning_rate": 0.0001327421142482195, "loss": 1.6853, "step": 405 }, { "epoch": 3.15, "learning_rate": 0.00012934228335981015, "loss": 1.7343, "step": 410 }, { "epoch": 3.19, "learning_rate": 0.00012505669320030482, "loss": 1.7685, "step": 415 }, { "epoch": 3.23, "learning_rate": 0.00011994783732453755, "loss": 1.6945, "step": 420 }, { "epoch": 3.27, "learning_rate": 0.00011409021435531858, "loss": 1.7482, "step": 425 }, { "epoch": 3.31, "learning_rate": 0.00010756924162575738, "loss": 1.5529, "step": 430 }, { "epoch": 3.35, "learning_rate": 0.00010048000960220263, "loss": 1.7141, "step": 435 }, { "epoch": 3.38, "learning_rate": 9.292589525111788e-05, "loss": 1.6804, "step": 440 }, { "epoch": 3.42, "learning_rate": 8.501705457012643e-05, "loss": 1.7525, "step": 445 }, { "epoch": 3.46, "learning_rate": 7.686881626551516e-05, "loss": 1.6325, "step": 450 }, { "epoch": 3.5, "learning_rate": 6.860000000000003e-05, "loss": 1.6941, "step": 455 }, { "epoch": 3.54, "learning_rate": 6.033118373448492e-05, "loss": 1.5753, "step": 460 }, { "epoch": 3.58, "learning_rate": 5.218294542987365e-05, "loss": 1.6565, "step": 465 }, { "epoch": 3.62, "learning_rate": 4.42741047488822e-05, "loss": 1.6553, "step": 470 }, { "epoch": 3.65, "learning_rate": 3.671999039779743e-05, "loss": 1.5705, "step": 475 }, { "epoch": 3.69, "learning_rate": 2.9630758374242696e-05, "loss": 1.7043, "step": 480 }, { "epoch": 3.73, "learning_rate": 2.3109785644681465e-05, "loss": 1.5999, "step": 485 }, { "epoch": 3.77, "learning_rate": 1.7252162675462504e-05, "loss": 1.675, "step": 490 }, { "epoch": 3.81, "learning_rate": 1.2143306799695228e-05, "loss": 1.6187, "step": 495 }, { "epoch": 3.85, "learning_rate": 7.857716640189861e-06, "loss": 1.5483, "step": 500 }, { "epoch": 3.88, "learning_rate": 4.4578857517805195e-06, "loss": 1.6583, "step": 505 }, { "epoch": 3.92, "learning_rate": 1.9933913245728244e-06, "loss": 1.6295, "step": 510 }, { "epoch": 3.96, "learning_rate": 5.001712368734975e-07, "loss": 1.5502, "step": 515 }, { "epoch": 4.0, "learning_rate": 0.0, "loss": 1.7657, "step": 520 }, { "epoch": 4.0, "eval_loss": 1.7459986209869385, "eval_runtime": 9.6571, "eval_samples_per_second": 20.192, "eval_steps_per_second": 2.589, "step": 520 } ], "max_steps": 520, "num_train_epochs": 4, "total_flos": 542964842496000.0, "trial_name": null, "trial_params": null }