{ "best_metric": 0.7650397419929504, "best_model_checkpoint": "ckpt/origin/pedes_location_v2/checkpoint-91", "epoch": 4.859813084112149, "eval_steps": 7, "global_step": 130, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04, "learning_rate": 5e-06, "loss": 1.0785, "step": 1 }, { "epoch": 0.07, "learning_rate": 1e-05, "loss": 1.0074, "step": 2 }, { "epoch": 0.11, "learning_rate": 9.998494093481022e-06, "loss": 0.9961, "step": 3 }, { "epoch": 0.15, "learning_rate": 9.993977281025862e-06, "loss": 1.0135, "step": 4 }, { "epoch": 0.19, "learning_rate": 9.986452283393452e-06, "loss": 0.9407, "step": 5 }, { "epoch": 0.22, "learning_rate": 9.975923633360985e-06, "loss": 0.9739, "step": 6 }, { "epoch": 0.26, "learning_rate": 9.962397672993552e-06, "loss": 0.9871, "step": 7 }, { "epoch": 0.26, "eval_loss": 0.8237647414207458, "eval_runtime": 37.148, "eval_samples_per_second": 1.911, "eval_steps_per_second": 1.911, "step": 7 }, { "epoch": 0.3, "learning_rate": 9.945882549823906e-06, "loss": 1.0363, "step": 8 }, { "epoch": 0.34, "learning_rate": 9.926388211944707e-06, "loss": 0.954, "step": 9 }, { "epoch": 0.37, "learning_rate": 9.903926402016153e-06, "loss": 0.9186, "step": 10 }, { "epoch": 0.41, "learning_rate": 9.878510650192644e-06, "loss": 0.8576, "step": 11 }, { "epoch": 0.45, "learning_rate": 9.850156265972722e-06, "loss": 0.9378, "step": 12 }, { "epoch": 0.49, "learning_rate": 9.8188803289772e-06, "loss": 0.8016, "step": 13 }, { "epoch": 0.52, "learning_rate": 9.784701678661045e-06, "loss": 0.8307, "step": 14 }, { "epoch": 0.52, "eval_loss": 0.8120225071907043, "eval_runtime": 36.7585, "eval_samples_per_second": 1.932, "eval_steps_per_second": 1.932, "step": 14 }, { "epoch": 0.56, "learning_rate": 9.747640902965185e-06, "loss": 0.9499, "step": 15 }, { "epoch": 0.6, "learning_rate": 9.707720325915105e-06, "loss": 0.92, "step": 16 }, { "epoch": 0.64, "learning_rate": 9.664963994173695e-06, "loss": 0.8689, "step": 17 }, { "epoch": 0.67, "learning_rate": 9.619397662556434e-06, "loss": 0.9116, "step": 18 }, { "epoch": 0.71, "learning_rate": 9.571048778517655e-06, "loss": 0.8537, "step": 19 }, { "epoch": 0.75, "learning_rate": 9.519946465617217e-06, "loss": 0.8569, "step": 20 }, { "epoch": 0.79, "learning_rate": 9.466121505977577e-06, "loss": 0.9572, "step": 21 }, { "epoch": 0.79, "eval_loss": 0.8014978170394897, "eval_runtime": 36.792, "eval_samples_per_second": 1.93, "eval_steps_per_second": 1.93, "step": 21 }, { "epoch": 0.82, "learning_rate": 9.409606321741776e-06, "loss": 0.7823, "step": 22 }, { "epoch": 0.86, "learning_rate": 9.350434955543557e-06, "loss": 0.8996, "step": 23 }, { "epoch": 0.9, "learning_rate": 9.288643050001362e-06, "loss": 0.8973, "step": 24 }, { "epoch": 0.93, "learning_rate": 9.224267826248536e-06, "loss": 0.8862, "step": 25 }, { "epoch": 0.97, "learning_rate": 9.157348061512728e-06, "loss": 0.9096, "step": 26 }, { "epoch": 1.01, "learning_rate": 9.08792406575792e-06, "loss": 0.8754, "step": 27 }, { "epoch": 1.05, "learning_rate": 9.016037657403225e-06, "loss": 0.8163, "step": 28 }, { "epoch": 1.05, "eval_loss": 0.7943388819694519, "eval_runtime": 36.78, "eval_samples_per_second": 1.93, "eval_steps_per_second": 1.93, "step": 28 }, { "epoch": 1.08, "learning_rate": 8.941732138133032e-06, "loss": 0.8693, "step": 29 }, { "epoch": 1.12, "learning_rate": 8.865052266813686e-06, "loss": 0.825, "step": 30 }, { "epoch": 1.16, "learning_rate": 8.786044232532423e-06, "loss": 0.8679, "step": 31 }, { "epoch": 1.2, "learning_rate": 8.704755626774796e-06, "loss": 0.8532, "step": 32 }, { "epoch": 1.23, "learning_rate": 8.621235414757337e-06, "loss": 0.843, "step": 33 }, { "epoch": 1.27, "learning_rate": 8.535533905932739e-06, "loss": 0.7849, "step": 34 }, { "epoch": 1.31, "learning_rate": 8.447702723685335e-06, "loss": 0.8735, "step": 35 }, { "epoch": 1.31, "eval_loss": 0.7872768640518188, "eval_runtime": 36.778, "eval_samples_per_second": 1.93, "eval_steps_per_second": 1.93, "step": 35 }, { "epoch": 1.35, "learning_rate": 8.357794774235094e-06, "loss": 0.8562, "step": 36 }, { "epoch": 1.38, "learning_rate": 8.265864214768883e-06, "loss": 0.7895, "step": 37 }, { "epoch": 1.42, "learning_rate": 8.171966420818227e-06, "loss": 0.9451, "step": 38 }, { "epoch": 1.46, "learning_rate": 8.076157952903134e-06, "loss": 0.9048, "step": 39 }, { "epoch": 1.5, "learning_rate": 7.978496522462167e-06, "loss": 0.9119, "step": 40 }, { "epoch": 1.53, "learning_rate": 7.879040957089229e-06, "loss": 0.8287, "step": 41 }, { "epoch": 1.57, "learning_rate": 7.777851165098012e-06, "loss": 0.8644, "step": 42 }, { "epoch": 1.57, "eval_loss": 0.7796598672866821, "eval_runtime": 36.8039, "eval_samples_per_second": 1.929, "eval_steps_per_second": 1.929, "step": 42 }, { "epoch": 1.61, "learning_rate": 7.674988099435487e-06, "loss": 0.9097, "step": 43 }, { "epoch": 1.64, "learning_rate": 7.570513720966108e-06, "loss": 0.7788, "step": 44 }, { "epoch": 1.68, "learning_rate": 7.464490961148921e-06, "loss": 0.8286, "step": 45 }, { "epoch": 1.72, "learning_rate": 7.3569836841299905e-06, "loss": 0.811, "step": 46 }, { "epoch": 1.76, "learning_rate": 7.248056648273034e-06, "loss": 0.793, "step": 47 }, { "epoch": 1.79, "learning_rate": 7.137775467151411e-06, "loss": 0.8564, "step": 48 }, { "epoch": 1.83, "learning_rate": 7.026206570024949e-06, "loss": 0.8223, "step": 49 }, { "epoch": 1.83, "eval_loss": 0.7741303443908691, "eval_runtime": 37.1509, "eval_samples_per_second": 1.911, "eval_steps_per_second": 1.911, "step": 49 }, { "epoch": 1.87, "learning_rate": 6.913417161825449e-06, "loss": 0.9124, "step": 50 }, { "epoch": 1.91, "learning_rate": 6.799475182674942e-06, "loss": 0.8556, "step": 51 }, { "epoch": 1.94, "learning_rate": 6.684449266961101e-06, "loss": 0.8013, "step": 52 }, { "epoch": 1.98, "learning_rate": 6.568408701994459e-06, "loss": 0.8264, "step": 53 }, { "epoch": 2.02, "learning_rate": 6.451423386272312e-06, "loss": 0.8864, "step": 54 }, { "epoch": 2.06, "learning_rate": 6.333563787374493e-06, "loss": 0.8105, "step": 55 }, { "epoch": 2.09, "learning_rate": 6.21490089951632e-06, "loss": 0.8621, "step": 56 }, { "epoch": 2.09, "eval_loss": 0.7700739502906799, "eval_runtime": 37.1567, "eval_samples_per_second": 1.911, "eval_steps_per_second": 1.911, "step": 56 }, { "epoch": 2.13, "learning_rate": 6.095506200784349e-06, "loss": 0.8165, "step": 57 }, { "epoch": 2.17, "learning_rate": 5.975451610080643e-06, "loss": 0.8335, "step": 58 }, { "epoch": 2.21, "learning_rate": 5.8548094438015065e-06, "loss": 0.8128, "step": 59 }, { "epoch": 2.24, "learning_rate": 5.733652372276809e-06, "loss": 0.9178, "step": 60 }, { "epoch": 2.28, "learning_rate": 5.612053375996082e-06, "loss": 0.8001, "step": 61 }, { "epoch": 2.32, "learning_rate": 5.490085701647805e-06, "loss": 0.8148, "step": 62 }, { "epoch": 2.36, "learning_rate": 5.367822817998338e-06, "loss": 0.89, "step": 63 }, { "epoch": 2.36, "eval_loss": 0.7689735293388367, "eval_runtime": 36.9469, "eval_samples_per_second": 1.922, "eval_steps_per_second": 1.922, "step": 63 }, { "epoch": 2.39, "learning_rate": 5.245338371637091e-06, "loss": 0.73, "step": 64 }, { "epoch": 2.43, "learning_rate": 5.122706142614562e-06, "loss": 0.8476, "step": 65 }, { "epoch": 2.47, "learning_rate": 5e-06, "loss": 0.8808, "step": 66 }, { "epoch": 2.5, "learning_rate": 4.87729385738544e-06, "loss": 0.88, "step": 67 }, { "epoch": 2.54, "learning_rate": 4.75466162836291e-06, "loss": 0.7914, "step": 68 }, { "epoch": 2.58, "learning_rate": 4.6321771820016635e-06, "loss": 0.8507, "step": 69 }, { "epoch": 2.62, "learning_rate": 4.509914298352197e-06, "loss": 0.8229, "step": 70 }, { "epoch": 2.62, "eval_loss": 0.7681828141212463, "eval_runtime": 37.0363, "eval_samples_per_second": 1.917, "eval_steps_per_second": 1.917, "step": 70 }, { "epoch": 2.65, "learning_rate": 4.38794662400392e-06, "loss": 0.8205, "step": 71 }, { "epoch": 2.69, "learning_rate": 4.266347627723192e-06, "loss": 0.785, "step": 72 }, { "epoch": 2.73, "learning_rate": 4.145190556198494e-06, "loss": 0.723, "step": 73 }, { "epoch": 2.77, "learning_rate": 4.02454838991936e-06, "loss": 0.7553, "step": 74 }, { "epoch": 2.8, "learning_rate": 3.904493799215652e-06, "loss": 0.8349, "step": 75 }, { "epoch": 2.84, "learning_rate": 3.7850991004836813e-06, "loss": 0.9063, "step": 76 }, { "epoch": 2.88, "learning_rate": 3.6664362126255087e-06, "loss": 0.7675, "step": 77 }, { "epoch": 2.88, "eval_loss": 0.7669386267662048, "eval_runtime": 37.0181, "eval_samples_per_second": 1.918, "eval_steps_per_second": 1.918, "step": 77 }, { "epoch": 2.92, "learning_rate": 3.5485766137276894e-06, "loss": 0.8301, "step": 78 }, { "epoch": 2.95, "learning_rate": 3.4315912980055433e-06, "loss": 0.7527, "step": 79 }, { "epoch": 2.99, "learning_rate": 3.3155507330389004e-06, "loss": 0.8311, "step": 80 }, { "epoch": 3.03, "learning_rate": 3.2005248173250593e-06, "loss": 0.7622, "step": 81 }, { "epoch": 3.07, "learning_rate": 3.0865828381745515e-06, "loss": 0.802, "step": 82 }, { "epoch": 3.1, "learning_rate": 2.9737934299750514e-06, "loss": 0.7803, "step": 83 }, { "epoch": 3.14, "learning_rate": 2.862224532848591e-06, "loss": 0.8235, "step": 84 }, { "epoch": 3.14, "eval_loss": 0.76643306016922, "eval_runtime": 37.3567, "eval_samples_per_second": 1.901, "eval_steps_per_second": 1.901, "step": 84 }, { "epoch": 3.18, "learning_rate": 2.7519433517269665e-06, "loss": 0.8095, "step": 85 }, { "epoch": 3.21, "learning_rate": 2.6430163158700116e-06, "loss": 0.7871, "step": 86 }, { "epoch": 3.25, "learning_rate": 2.5355090388510806e-06, "loss": 0.9052, "step": 87 }, { "epoch": 3.29, "learning_rate": 2.429486279033892e-06, "loss": 0.7754, "step": 88 }, { "epoch": 3.33, "learning_rate": 2.325011900564515e-06, "loss": 0.8105, "step": 89 }, { "epoch": 3.36, "learning_rate": 2.2221488349019903e-06, "loss": 0.7491, "step": 90 }, { "epoch": 3.4, "learning_rate": 2.1209590429107734e-06, "loss": 0.8888, "step": 91 }, { "epoch": 3.4, "eval_loss": 0.7650397419929504, "eval_runtime": 37.1172, "eval_samples_per_second": 1.913, "eval_steps_per_second": 1.913, "step": 91 }, { "epoch": 3.44, "learning_rate": 2.0215034775378336e-06, "loss": 0.8014, "step": 92 }, { "epoch": 3.48, "learning_rate": 1.9238420470968665e-06, "loss": 0.8284, "step": 93 }, { "epoch": 3.51, "learning_rate": 1.8280335791817733e-06, "loss": 0.8674, "step": 94 }, { "epoch": 3.55, "learning_rate": 1.7341357852311175e-06, "loss": 0.8278, "step": 95 }, { "epoch": 3.59, "learning_rate": 1.642205225764908e-06, "loss": 0.798, "step": 96 }, { "epoch": 3.63, "learning_rate": 1.5522972763146653e-06, "loss": 0.8205, "step": 97 }, { "epoch": 3.66, "learning_rate": 1.4644660940672628e-06, "loss": 0.7885, "step": 98 }, { "epoch": 3.66, "eval_loss": 0.7652753591537476, "eval_runtime": 37.5305, "eval_samples_per_second": 1.892, "eval_steps_per_second": 1.892, "step": 98 }, { "epoch": 3.7, "learning_rate": 1.3787645852426663e-06, "loss": 0.8437, "step": 99 }, { "epoch": 3.74, "learning_rate": 1.2952443732252058e-06, "loss": 0.7953, "step": 100 }, { "epoch": 3.78, "learning_rate": 1.2139557674675773e-06, "loss": 0.8059, "step": 101 }, { "epoch": 3.81, "learning_rate": 1.134947733186315e-06, "loss": 0.8249, "step": 102 }, { "epoch": 3.85, "learning_rate": 1.058267861866969e-06, "loss": 0.8576, "step": 103 }, { "epoch": 3.89, "learning_rate": 9.83962342596776e-07, "loss": 0.7749, "step": 104 }, { "epoch": 3.93, "learning_rate": 9.120759342420821e-07, "loss": 0.8581, "step": 105 }, { "epoch": 3.93, "eval_loss": 0.7648642659187317, "eval_runtime": 37.1908, "eval_samples_per_second": 1.909, "eval_steps_per_second": 1.909, "step": 105 }, { "epoch": 3.96, "learning_rate": 8.426519384872733e-07, "loss": 0.7704, "step": 106 }, { "epoch": 4.0, "learning_rate": 7.757321737514645e-07, "loss": 0.8748, "step": 107 }, { "epoch": 4.04, "learning_rate": 7.113569499986401e-07, "loss": 0.804, "step": 108 }, { "epoch": 4.07, "learning_rate": 6.495650444564433e-07, "loss": 0.797, "step": 109 }, { "epoch": 4.11, "learning_rate": 5.903936782582253e-07, "loss": 0.7378, "step": 110 }, { "epoch": 4.15, "learning_rate": 5.338784940224239e-07, "loss": 0.8688, "step": 111 }, { "epoch": 4.19, "learning_rate": 4.800535343827834e-07, "loss": 0.749, "step": 112 }, { "epoch": 4.19, "eval_loss": 0.7643035054206848, "eval_runtime": 37.2542, "eval_samples_per_second": 1.906, "eval_steps_per_second": 1.906, "step": 112 }, { "epoch": 4.22, "learning_rate": 4.289512214823466e-07, "loss": 0.7901, "step": 113 }, { "epoch": 4.26, "learning_rate": 3.8060233744356634e-07, "loss": 0.8472, "step": 114 }, { "epoch": 4.3, "learning_rate": 3.350360058263058e-07, "loss": 0.8008, "step": 115 }, { "epoch": 4.34, "learning_rate": 2.9227967408489653e-07, "loss": 0.7978, "step": 116 }, { "epoch": 4.37, "learning_rate": 2.523590970348166e-07, "loss": 0.863, "step": 117 }, { "epoch": 4.41, "learning_rate": 2.152983213389559e-07, "loss": 0.8246, "step": 118 }, { "epoch": 4.45, "learning_rate": 1.8111967102280082e-07, "loss": 0.8035, "step": 119 }, { "epoch": 4.45, "eval_loss": 0.7640846967697144, "eval_runtime": 36.8656, "eval_samples_per_second": 1.926, "eval_steps_per_second": 1.926, "step": 119 }, { "epoch": 4.49, "learning_rate": 1.4984373402728014e-07, "loss": 0.8703, "step": 120 }, { "epoch": 4.52, "learning_rate": 1.2148934980735772e-07, "loss": 0.778, "step": 121 }, { "epoch": 4.56, "learning_rate": 9.607359798384785e-08, "loss": 0.7844, "step": 122 }, { "epoch": 4.6, "learning_rate": 7.36117880552939e-08, "loss": 0.7964, "step": 123 }, { "epoch": 4.64, "learning_rate": 5.411745017609493e-08, "loss": 0.7973, "step": 124 }, { "epoch": 4.67, "learning_rate": 3.7602327006450166e-08, "loss": 0.8472, "step": 125 }, { "epoch": 4.71, "learning_rate": 2.4076366639015914e-08, "loss": 0.8169, "step": 126 }, { "epoch": 4.71, "eval_loss": 0.7648836374282837, "eval_runtime": 37.2706, "eval_samples_per_second": 1.905, "eval_steps_per_second": 1.905, "step": 126 }, { "epoch": 4.75, "learning_rate": 1.3547716606548967e-08, "loss": 0.7949, "step": 127 }, { "epoch": 4.79, "learning_rate": 6.022718974137976e-09, "loss": 0.8412, "step": 128 }, { "epoch": 4.82, "learning_rate": 1.5059065189787502e-09, "loss": 0.8343, "step": 129 }, { "epoch": 4.86, "learning_rate": 0.0, "loss": 0.8341, "step": 130 }, { "epoch": 4.86, "step": 130, "total_flos": 3.500008779892654e+17, "train_loss": 0.8453521888989669, "train_runtime": 7729.7132, "train_samples_per_second": 0.829, "train_steps_per_second": 0.017 } ], "logging_steps": 1.0, "max_steps": 130, "num_train_epochs": 5, "save_steps": 13, "total_flos": 3.500008779892654e+17, "trial_name": null, "trial_params": null }