{ "best_metric": null, "best_model_checkpoint": null, "epoch": 5.0, "global_step": 49420, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.1, "learning_rate": 4.896398219344395e-05, "loss": 0.2574, "step": 1024 }, { "epoch": 0.1, "eval_loss": 0.7888669967651367, "eval_runtime": 25.6574, "eval_samples_per_second": 81.536, "eval_steps_per_second": 3.43, "step": 1024 }, { "epoch": 0.21, "learning_rate": 4.79279643868879e-05, "loss": 0.2633, "step": 2048 }, { "epoch": 0.21, "eval_loss": 0.780219316482544, "eval_runtime": 25.6023, "eval_samples_per_second": 81.711, "eval_steps_per_second": 3.437, "step": 2048 }, { "epoch": 0.31, "learning_rate": 4.6891946580331854e-05, "loss": 0.2635, "step": 3072 }, { "epoch": 0.31, "eval_loss": 0.7877150774002075, "eval_runtime": 25.6162, "eval_samples_per_second": 81.667, "eval_steps_per_second": 3.435, "step": 3072 }, { "epoch": 0.41, "learning_rate": 4.5855928773775804e-05, "loss": 0.2639, "step": 4096 }, { "epoch": 0.41, "eval_loss": 0.7751328945159912, "eval_runtime": 25.6157, "eval_samples_per_second": 81.669, "eval_steps_per_second": 3.435, "step": 4096 }, { "epoch": 0.52, "learning_rate": 4.4819910967219755e-05, "loss": 0.2625, "step": 5120 }, { "epoch": 0.52, "eval_loss": 0.7836451530456543, "eval_runtime": 25.6812, "eval_samples_per_second": 81.46, "eval_steps_per_second": 3.427, "step": 5120 }, { "epoch": 0.62, "learning_rate": 4.3783893160663705e-05, "loss": 0.2609, "step": 6144 }, { "epoch": 0.62, "eval_loss": 0.7758111357688904, "eval_runtime": 25.6349, "eval_samples_per_second": 81.607, "eval_steps_per_second": 3.433, "step": 6144 }, { "epoch": 0.73, "learning_rate": 4.2747875354107656e-05, "loss": 0.2597, "step": 7168 }, { "epoch": 0.73, "eval_loss": 0.7923067212104797, "eval_runtime": 25.6218, "eval_samples_per_second": 81.649, "eval_steps_per_second": 3.435, "step": 7168 }, { "epoch": 0.83, "learning_rate": 4.17118575475516e-05, "loss": 0.2612, "step": 8192 }, { "epoch": 0.83, "eval_loss": 0.8000349402427673, "eval_runtime": 25.646, "eval_samples_per_second": 81.572, "eval_steps_per_second": 3.431, "step": 8192 }, { "epoch": 0.93, "learning_rate": 4.0676851477134767e-05, "loss": 0.2612, "step": 9216 }, { "epoch": 0.93, "eval_loss": 0.7934667468070984, "eval_runtime": 25.6816, "eval_samples_per_second": 81.459, "eval_steps_per_second": 3.427, "step": 9216 }, { "epoch": 1.04, "learning_rate": 3.964083367057872e-05, "loss": 0.2554, "step": 10240 }, { "epoch": 1.04, "eval_loss": 0.7942602038383484, "eval_runtime": 25.7479, "eval_samples_per_second": 81.249, "eval_steps_per_second": 3.418, "step": 10240 }, { "epoch": 1.14, "learning_rate": 3.860481586402267e-05, "loss": 0.2524, "step": 11264 }, { "epoch": 1.14, "eval_loss": 0.8014633655548096, "eval_runtime": 25.6355, "eval_samples_per_second": 81.606, "eval_steps_per_second": 3.433, "step": 11264 }, { "epoch": 1.24, "learning_rate": 3.7569809793605834e-05, "loss": 0.2504, "step": 12288 }, { "epoch": 1.24, "eval_loss": 0.7962351441383362, "eval_runtime": 25.6019, "eval_samples_per_second": 81.713, "eval_steps_per_second": 3.437, "step": 12288 }, { "epoch": 1.35, "learning_rate": 3.653379198704978e-05, "loss": 0.2524, "step": 13312 }, { "epoch": 1.35, "eval_loss": 0.8086119294166565, "eval_runtime": 25.7264, "eval_samples_per_second": 81.317, "eval_steps_per_second": 3.421, "step": 13312 }, { "epoch": 1.45, "learning_rate": 3.549777418049373e-05, "loss": 0.2521, "step": 14336 }, { "epoch": 1.45, "eval_loss": 0.8062009215354919, "eval_runtime": 25.6245, "eval_samples_per_second": 81.641, "eval_steps_per_second": 3.434, "step": 14336 }, { "epoch": 1.55, "learning_rate": 3.446175637393768e-05, "loss": 0.2503, "step": 15360 }, { "epoch": 1.55, "eval_loss": 0.7997719049453735, "eval_runtime": 25.6588, "eval_samples_per_second": 81.532, "eval_steps_per_second": 3.43, "step": 15360 }, { "epoch": 1.66, "learning_rate": 3.3426750303520846e-05, "loss": 0.2523, "step": 16384 }, { "epoch": 1.66, "eval_loss": 0.8097875118255615, "eval_runtime": 25.6111, "eval_samples_per_second": 81.683, "eval_steps_per_second": 3.436, "step": 16384 }, { "epoch": 1.76, "learning_rate": 3.239073249696479e-05, "loss": 0.251, "step": 17408 }, { "epoch": 1.76, "eval_loss": 0.8213343620300293, "eval_runtime": 25.6868, "eval_samples_per_second": 81.443, "eval_steps_per_second": 3.426, "step": 17408 }, { "epoch": 1.86, "learning_rate": 3.135572642654796e-05, "loss": 0.2509, "step": 18432 }, { "epoch": 1.86, "eval_loss": 0.8137653470039368, "eval_runtime": 25.6223, "eval_samples_per_second": 81.648, "eval_steps_per_second": 3.435, "step": 18432 }, { "epoch": 1.97, "learning_rate": 3.031970861999191e-05, "loss": 0.2533, "step": 19456 }, { "epoch": 1.97, "eval_loss": 0.8182128071784973, "eval_runtime": 25.6272, "eval_samples_per_second": 81.632, "eval_steps_per_second": 3.434, "step": 19456 }, { "epoch": 2.07, "learning_rate": 2.928369081343586e-05, "loss": 0.245, "step": 20480 }, { "epoch": 2.07, "eval_loss": 0.8290057182312012, "eval_runtime": 25.6288, "eval_samples_per_second": 81.627, "eval_steps_per_second": 3.434, "step": 20480 }, { "epoch": 2.18, "learning_rate": 2.824868474301902e-05, "loss": 0.2432, "step": 21504 }, { "epoch": 2.18, "eval_loss": 0.8328016400337219, "eval_runtime": 25.7111, "eval_samples_per_second": 81.366, "eval_steps_per_second": 3.423, "step": 21504 }, { "epoch": 2.28, "learning_rate": 2.7212666936462972e-05, "loss": 0.2435, "step": 22528 }, { "epoch": 2.28, "eval_loss": 0.8186773657798767, "eval_runtime": 25.61, "eval_samples_per_second": 81.687, "eval_steps_per_second": 3.436, "step": 22528 }, { "epoch": 2.38, "learning_rate": 2.617766086604614e-05, "loss": 0.2423, "step": 23552 }, { "epoch": 2.38, "eval_loss": 0.8238145112991333, "eval_runtime": 25.6287, "eval_samples_per_second": 81.627, "eval_steps_per_second": 3.434, "step": 23552 }, { "epoch": 2.49, "learning_rate": 2.51426547956293e-05, "loss": 0.2443, "step": 24576 }, { "epoch": 2.49, "eval_loss": 0.8248941898345947, "eval_runtime": 25.6771, "eval_samples_per_second": 81.473, "eval_steps_per_second": 3.427, "step": 24576 }, { "epoch": 2.59, "learning_rate": 2.410663698907325e-05, "loss": 0.2431, "step": 25600 }, { "epoch": 2.59, "eval_loss": 0.8253324031829834, "eval_runtime": 25.6574, "eval_samples_per_second": 81.536, "eval_steps_per_second": 3.43, "step": 25600 }, { "epoch": 2.69, "learning_rate": 2.30706191825172e-05, "loss": 0.2432, "step": 26624 }, { "epoch": 2.69, "eval_loss": 0.8268725275993347, "eval_runtime": 25.6215, "eval_samples_per_second": 81.65, "eval_steps_per_second": 3.435, "step": 26624 }, { "epoch": 2.8, "learning_rate": 2.203460137596115e-05, "loss": 0.2421, "step": 27648 }, { "epoch": 2.8, "eval_loss": 0.8282283544540405, "eval_runtime": 25.6027, "eval_samples_per_second": 81.71, "eval_steps_per_second": 3.437, "step": 27648 }, { "epoch": 2.9, "learning_rate": 2.09985835694051e-05, "loss": 0.2421, "step": 28672 }, { "epoch": 2.9, "eval_loss": 0.8267728686332703, "eval_runtime": 25.6314, "eval_samples_per_second": 81.619, "eval_steps_per_second": 3.433, "step": 28672 }, { "epoch": 3.0, "learning_rate": 1.9963577498988265e-05, "loss": 0.243, "step": 29696 }, { "epoch": 3.0, "eval_loss": 0.8345196843147278, "eval_runtime": 25.6765, "eval_samples_per_second": 81.475, "eval_steps_per_second": 3.427, "step": 29696 }, { "epoch": 3.11, "learning_rate": 1.8927559692432215e-05, "loss": 0.2367, "step": 30720 }, { "epoch": 3.11, "eval_loss": 0.8424174189567566, "eval_runtime": 25.634, "eval_samples_per_second": 81.61, "eval_steps_per_second": 3.433, "step": 30720 }, { "epoch": 3.21, "learning_rate": 1.789255362201538e-05, "loss": 0.237, "step": 31744 }, { "epoch": 3.21, "eval_loss": 0.8373659253120422, "eval_runtime": 25.5978, "eval_samples_per_second": 81.726, "eval_steps_per_second": 3.438, "step": 31744 }, { "epoch": 3.32, "learning_rate": 1.6857547551598542e-05, "loss": 0.2351, "step": 32768 }, { "epoch": 3.32, "eval_loss": 0.8430664539337158, "eval_runtime": 25.6675, "eval_samples_per_second": 81.504, "eval_steps_per_second": 3.428, "step": 32768 }, { "epoch": 3.42, "learning_rate": 1.5821529745042493e-05, "loss": 0.2374, "step": 33792 }, { "epoch": 3.42, "eval_loss": 0.8425076603889465, "eval_runtime": 25.7071, "eval_samples_per_second": 81.378, "eval_steps_per_second": 3.423, "step": 33792 }, { "epoch": 3.52, "learning_rate": 1.4786523674625658e-05, "loss": 0.2355, "step": 34816 }, { "epoch": 3.52, "eval_loss": 0.8351919054985046, "eval_runtime": 25.6245, "eval_samples_per_second": 81.641, "eval_steps_per_second": 3.434, "step": 34816 }, { "epoch": 3.63, "learning_rate": 1.3750505868069607e-05, "loss": 0.2373, "step": 35840 }, { "epoch": 3.63, "eval_loss": 0.845241904258728, "eval_runtime": 25.6392, "eval_samples_per_second": 81.594, "eval_steps_per_second": 3.432, "step": 35840 }, { "epoch": 3.73, "learning_rate": 1.2714488061513558e-05, "loss": 0.2356, "step": 36864 }, { "epoch": 3.73, "eval_loss": 0.8382672071456909, "eval_runtime": 25.6391, "eval_samples_per_second": 81.594, "eval_steps_per_second": 3.432, "step": 36864 }, { "epoch": 3.83, "learning_rate": 1.1678470254957508e-05, "loss": 0.2343, "step": 37888 }, { "epoch": 3.83, "eval_loss": 0.8443974852561951, "eval_runtime": 25.7018, "eval_samples_per_second": 81.395, "eval_steps_per_second": 3.424, "step": 37888 }, { "epoch": 3.94, "learning_rate": 1.0642452448401457e-05, "loss": 0.2348, "step": 38912 }, { "epoch": 3.94, "eval_loss": 0.8427897095680237, "eval_runtime": 25.6146, "eval_samples_per_second": 81.672, "eval_steps_per_second": 3.436, "step": 38912 }, { "epoch": 4.04, "learning_rate": 9.606434641845407e-06, "loss": 0.2349, "step": 39936 }, { "epoch": 4.04, "eval_loss": 0.8479835987091064, "eval_runtime": 25.6232, "eval_samples_per_second": 81.645, "eval_steps_per_second": 3.434, "step": 39936 }, { "epoch": 4.14, "learning_rate": 8.571428571428573e-06, "loss": 0.2327, "step": 40960 }, { "epoch": 4.14, "eval_loss": 0.8481870889663696, "eval_runtime": 25.6261, "eval_samples_per_second": 81.636, "eval_steps_per_second": 3.434, "step": 40960 }, { "epoch": 4.25, "learning_rate": 7.536422501011736e-06, "loss": 0.2337, "step": 41984 }, { "epoch": 4.25, "eval_loss": 0.850956916809082, "eval_runtime": 25.6867, "eval_samples_per_second": 81.443, "eval_steps_per_second": 3.426, "step": 41984 }, { "epoch": 4.35, "learning_rate": 6.500404694455686e-06, "loss": 0.2288, "step": 43008 }, { "epoch": 4.35, "eval_loss": 0.8499058485031128, "eval_runtime": 25.6219, "eval_samples_per_second": 81.649, "eval_steps_per_second": 3.435, "step": 43008 }, { "epoch": 4.45, "learning_rate": 5.464386887899636e-06, "loss": 0.2299, "step": 44032 }, { "epoch": 4.45, "eval_loss": 0.8522343039512634, "eval_runtime": 25.6037, "eval_samples_per_second": 81.707, "eval_steps_per_second": 3.437, "step": 44032 }, { "epoch": 4.56, "learning_rate": 4.429380817482801e-06, "loss": 0.2277, "step": 45056 }, { "epoch": 4.56, "eval_loss": 0.85256427526474, "eval_runtime": 25.6116, "eval_samples_per_second": 81.682, "eval_steps_per_second": 3.436, "step": 45056 }, { "epoch": 4.66, "learning_rate": 3.3933630109267505e-06, "loss": 0.2301, "step": 46080 }, { "epoch": 4.66, "eval_loss": 0.851773202419281, "eval_runtime": 25.6644, "eval_samples_per_second": 81.514, "eval_steps_per_second": 3.429, "step": 46080 }, { "epoch": 4.77, "learning_rate": 2.3573452043707e-06, "loss": 0.2312, "step": 47104 }, { "epoch": 4.77, "eval_loss": 0.8511470556259155, "eval_runtime": 25.6121, "eval_samples_per_second": 81.68, "eval_steps_per_second": 3.436, "step": 47104 }, { "epoch": 4.87, "learning_rate": 1.32132739781465e-06, "loss": 0.2284, "step": 48128 }, { "epoch": 4.87, "eval_loss": 0.8506836295127869, "eval_runtime": 25.6413, "eval_samples_per_second": 81.587, "eval_steps_per_second": 3.432, "step": 48128 }, { "epoch": 4.97, "learning_rate": 2.8632132739781467e-07, "loss": 0.2294, "step": 49152 }, { "epoch": 4.97, "eval_loss": 0.8519254922866821, "eval_runtime": 25.6136, "eval_samples_per_second": 81.675, "eval_steps_per_second": 3.436, "step": 49152 }, { "epoch": 5.0, "step": 49420, "total_flos": 3.1008084491562394e+17, "train_loss": 0.244437825115332, "train_runtime": 40418.7139, "train_samples_per_second": 25.676, "train_steps_per_second": 1.223 } ], "max_steps": 49420, "num_train_epochs": 5, "total_flos": 3.1008084491562394e+17, "trial_name": null, "trial_params": null }