|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.9914893617021276, |
|
"eval_steps": 5, |
|
"global_step": 59, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 2e-05, |
|
"loss": 1.9849, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"eval_loss": 2.0810964107513428, |
|
"eval_runtime": 6.2278, |
|
"eval_samples_per_second": 44.96, |
|
"eval_steps_per_second": 5.62, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 4e-05, |
|
"loss": 1.8931, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 6e-05, |
|
"loss": 1.763, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 8e-05, |
|
"loss": 1.6688, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 0.0001, |
|
"loss": 1.3107, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"eval_loss": 1.1991726160049438, |
|
"eval_runtime": 6.3173, |
|
"eval_samples_per_second": 44.322, |
|
"eval_steps_per_second": 5.54, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 0.00012, |
|
"loss": 1.1899, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 0.00014, |
|
"loss": 1.041, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 0.00016, |
|
"loss": 0.9779, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 0.00018, |
|
"loss": 0.7671, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 0.0002, |
|
"loss": 0.6399, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"eval_loss": 0.6359391212463379, |
|
"eval_runtime": 6.2721, |
|
"eval_samples_per_second": 44.642, |
|
"eval_steps_per_second": 5.58, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 0.0001999167799344583, |
|
"loss": 0.6777, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 0.00019966725824941932, |
|
"loss": 0.492, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 0.00019925185024910277, |
|
"loss": 0.3781, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 0.0001986712473391289, |
|
"loss": 0.4503, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 0.00019792641587574212, |
|
"loss": 0.2779, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"eval_loss": 0.28619876503944397, |
|
"eval_runtime": 6.3023, |
|
"eval_samples_per_second": 44.428, |
|
"eval_steps_per_second": 5.554, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 0.00019701859555740648, |
|
"loss": 0.2658, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 0.00019594929736144976, |
|
"loss": 0.2282, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 0.000194720301029191, |
|
"loss": 0.2149, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 0.0001933336521037367, |
|
"loss": 0.1792, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 0.00019179165852537596, |
|
"loss": 0.1807, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"eval_loss": 0.1634436696767807, |
|
"eval_runtime": 6.2724, |
|
"eval_samples_per_second": 44.64, |
|
"eval_steps_per_second": 5.58, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 0.0001900968867902419, |
|
"loss": 0.1425, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 0.00018825215767863214, |
|
"loss": 0.1377, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 0.00018626054156009806, |
|
"loss": 0.1173, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 0.00018412535328311814, |
|
"loss": 0.1338, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 0.00018185014665785936, |
|
"loss": 0.1256, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"eval_loss": 0.1177145466208458, |
|
"eval_runtime": 6.2558, |
|
"eval_samples_per_second": 44.758, |
|
"eval_steps_per_second": 5.595, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 0.00017943870854121124, |
|
"loss": 0.1234, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 0.0001768950525339362, |
|
"loss": 0.1164, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 0.000174223412300427, |
|
"loss": 0.1077, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 0.00017142823452219038, |
|
"loss": 0.0877, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 0.00016851417149678444, |
|
"loss": 0.097, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_loss": 0.08908110111951828, |
|
"eval_runtime": 6.274, |
|
"eval_samples_per_second": 44.628, |
|
"eval_steps_per_second": 5.579, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"learning_rate": 0.00016548607339452853, |
|
"loss": 0.0741, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"learning_rate": 0.00016234898018587337, |
|
"loss": 0.0811, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 0.00015910811325286768, |
|
"loss": 0.0743, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 1.14, |
|
"learning_rate": 0.00015576886669868296, |
|
"loss": 0.0799, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"learning_rate": 0.00015233679836966122, |
|
"loss": 0.1063, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"eval_loss": 0.07344743609428406, |
|
"eval_runtime": 6.3055, |
|
"eval_samples_per_second": 44.406, |
|
"eval_steps_per_second": 5.551, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 1.21, |
|
"learning_rate": 0.00014881762060482814, |
|
"loss": 0.0835, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 1.24, |
|
"learning_rate": 0.00014521719072826858, |
|
"loss": 0.0661, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 1.28, |
|
"learning_rate": 0.00014154150130018866, |
|
"loss": 0.0637, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 1.31, |
|
"learning_rate": 0.00013779667014289065, |
|
"loss": 0.0642, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"learning_rate": 0.00013398893015826167, |
|
"loss": 0.0769, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"eval_loss": 0.07225509732961655, |
|
"eval_runtime": 6.2831, |
|
"eval_samples_per_second": 44.564, |
|
"eval_steps_per_second": 5.571, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 1.38, |
|
"learning_rate": 0.00013012461895372344, |
|
"loss": 0.0746, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 0.00012621016829391022, |
|
"loss": 0.0592, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"learning_rate": 0.00012225209339563145, |
|
"loss": 0.078, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 1.48, |
|
"learning_rate": 0.00011825698208393619, |
|
"loss": 0.088, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 0.00011423148382732853, |
|
"loss": 0.0694, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"eval_loss": 0.06333146244287491, |
|
"eval_runtime": 6.2803, |
|
"eval_samples_per_second": 44.584, |
|
"eval_steps_per_second": 5.573, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 1.55, |
|
"learning_rate": 0.00011018229867038356, |
|
"loss": 0.0591, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 1.58, |
|
"learning_rate": 0.00010611616608218429, |
|
"loss": 0.0646, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 1.62, |
|
"learning_rate": 0.00010203985373914056, |
|
"loss": 0.068, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"learning_rate": 9.79601462608595e-05, |
|
"loss": 0.055, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 1.69, |
|
"learning_rate": 9.388383391781575e-05, |
|
"loss": 0.0687, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 1.69, |
|
"eval_loss": 0.06235363706946373, |
|
"eval_runtime": 6.2849, |
|
"eval_samples_per_second": 44.551, |
|
"eval_steps_per_second": 5.569, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 1.72, |
|
"learning_rate": 8.981770132961649e-05, |
|
"loss": 0.0643, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 1.75, |
|
"learning_rate": 8.57685161726715e-05, |
|
"loss": 0.0589, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 1.79, |
|
"learning_rate": 8.174301791606385e-05, |
|
"loss": 0.0729, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 1.82, |
|
"learning_rate": 7.774790660436858e-05, |
|
"loss": 0.0553, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 1.86, |
|
"learning_rate": 7.378983170608982e-05, |
|
"loss": 0.0575, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 1.86, |
|
"eval_loss": 0.062243152409791946, |
|
"eval_runtime": 6.2854, |
|
"eval_samples_per_second": 44.548, |
|
"eval_steps_per_second": 5.568, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 6.98753810462766e-05, |
|
"loss": 0.0675, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 1.92, |
|
"learning_rate": 6.601106984173835e-05, |
|
"loss": 0.068, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 1.96, |
|
"learning_rate": 6.220332985710936e-05, |
|
"loss": 0.0601, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 1.99, |
|
"learning_rate": 5.845849869981137e-05, |
|
"loss": 0.0764, |
|
"step": 59 |
|
} |
|
], |
|
"logging_steps": 1, |
|
"max_steps": 87, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"total_flos": 3.83737422209876e+17, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|