| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.989386248269497, | |
| "eval_steps": 500, | |
| "global_step": 67, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.014766958929395477, | |
| "grad_norm": 2.4889817237854004, | |
| "learning_rate": 0.0, | |
| "loss": 1.6905, | |
| "memory/device_mem_reserved(gib)": 21.22, | |
| "memory/max_mem_active(gib)": 20.26, | |
| "memory/max_mem_allocated(gib)": 20.26, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.029533917858790955, | |
| "grad_norm": 2.847207546234131, | |
| "learning_rate": 3.3333333333333333e-06, | |
| "loss": 1.6684, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.044300876788186436, | |
| "grad_norm": 2.1346399784088135, | |
| "learning_rate": 6.666666666666667e-06, | |
| "loss": 1.6625, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.05906783571758191, | |
| "grad_norm": 1.5967113971710205, | |
| "learning_rate": 1e-05, | |
| "loss": 1.5597, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.07383479464697738, | |
| "grad_norm": 1.1592029333114624, | |
| "learning_rate": 1.3333333333333333e-05, | |
| "loss": 1.61, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.08860175357637287, | |
| "grad_norm": 0.7923451662063599, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 1.6703, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.10336871250576835, | |
| "grad_norm": 0.8879733085632324, | |
| "learning_rate": 2e-05, | |
| "loss": 1.5672, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.11813567143516382, | |
| "grad_norm": 0.8429373502731323, | |
| "learning_rate": 1.9986740898848306e-05, | |
| "loss": 1.5304, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.1329026303645593, | |
| "grad_norm": 0.6310297846794128, | |
| "learning_rate": 1.9946998756145894e-05, | |
| "loss": 1.5365, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.14766958929395477, | |
| "grad_norm": 0.45904460549354553, | |
| "learning_rate": 1.9880878960910772e-05, | |
| "loss": 1.5564, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.16243654822335024, | |
| "grad_norm": 0.4647868573665619, | |
| "learning_rate": 1.978855685095358e-05, | |
| "loss": 1.5871, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.17720350715274574, | |
| "grad_norm": 0.47522127628326416, | |
| "learning_rate": 1.9670277247913205e-05, | |
| "loss": 1.6906, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.19197046608214122, | |
| "grad_norm": 0.4368882477283478, | |
| "learning_rate": 1.9526353808033827e-05, | |
| "loss": 1.5612, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.2067374250115367, | |
| "grad_norm": 0.41860365867614746, | |
| "learning_rate": 1.9357168190404937e-05, | |
| "loss": 1.5157, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.22150438394093216, | |
| "grad_norm": 0.35527029633522034, | |
| "learning_rate": 1.916316904487005e-05, | |
| "loss": 1.6507, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.23627134287032764, | |
| "grad_norm": 0.3510073125362396, | |
| "learning_rate": 1.8944870822287957e-05, | |
| "loss": 1.4494, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.25103830179972314, | |
| "grad_norm": 0.33353254199028015, | |
| "learning_rate": 1.8702852410301556e-05, | |
| "loss": 1.592, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.2658052607291186, | |
| "grad_norm": 0.3858069181442261, | |
| "learning_rate": 1.8437755598231857e-05, | |
| "loss": 1.5903, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.2805722196585141, | |
| "grad_norm": 0.4120069444179535, | |
| "learning_rate": 1.8150283375168112e-05, | |
| "loss": 1.5023, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.29533917858790953, | |
| "grad_norm": 0.37689632177352905, | |
| "learning_rate": 1.7841198065767107e-05, | |
| "loss": 1.4686, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.31010613751730504, | |
| "grad_norm": 0.32370489835739136, | |
| "learning_rate": 1.7511319308705198e-05, | |
| "loss": 1.6619, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.3248730964467005, | |
| "grad_norm": 0.3516235053539276, | |
| "learning_rate": 1.7161521883143936e-05, | |
| "loss": 1.6317, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.339640055376096, | |
| "grad_norm": 0.35210394859313965, | |
| "learning_rate": 1.679273338897293e-05, | |
| "loss": 1.4843, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.3544070143054915, | |
| "grad_norm": 0.31570005416870117, | |
| "learning_rate": 1.6405931786981753e-05, | |
| "loss": 1.4993, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.36917397323488693, | |
| "grad_norm": 0.3055304288864136, | |
| "learning_rate": 1.6002142805483686e-05, | |
| "loss": 1.5752, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.38394093216428243, | |
| "grad_norm": 0.32708096504211426, | |
| "learning_rate": 1.5582437220268648e-05, | |
| "loss": 1.5435, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.3987078910936779, | |
| "grad_norm": 0.3059172034263611, | |
| "learning_rate": 1.5147928015098309e-05, | |
| "loss": 1.5129, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.4134748500230734, | |
| "grad_norm": 0.31772273778915405, | |
| "learning_rate": 1.4699767430273202e-05, | |
| "loss": 1.5566, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.4282418089524688, | |
| "grad_norm": 0.318418949842453, | |
| "learning_rate": 1.423914390709861e-05, | |
| "loss": 1.3997, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.44300876788186433, | |
| "grad_norm": 0.33671969175338745, | |
| "learning_rate": 1.3767278936351853e-05, | |
| "loss": 1.4731, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.45777572681125983, | |
| "grad_norm": 0.32063713669776917, | |
| "learning_rate": 1.3285423819108349e-05, | |
| "loss": 1.5531, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.4725426857406553, | |
| "grad_norm": 0.30657994747161865, | |
| "learning_rate": 1.2794856348516095e-05, | |
| "loss": 1.515, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.4873096446700508, | |
| "grad_norm": 0.2942485213279724, | |
| "learning_rate": 1.2296877421317958e-05, | |
| "loss": 1.5364, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.5020766035994463, | |
| "grad_norm": 0.3146020472049713, | |
| "learning_rate": 1.1792807588107358e-05, | |
| "loss": 1.555, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.5168435625288417, | |
| "grad_norm": 0.3142072260379791, | |
| "learning_rate": 1.1283983551465512e-05, | |
| "loss": 1.4971, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.5316105214582372, | |
| "grad_norm": 0.28682929277420044, | |
| "learning_rate": 1.0771754621266466e-05, | |
| "loss": 1.4582, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.5463774803876327, | |
| "grad_norm": 0.2946212589740753, | |
| "learning_rate": 1.0257479136549889e-05, | |
| "loss": 1.4131, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.5611444393170282, | |
| "grad_norm": 0.30411672592163086, | |
| "learning_rate": 9.742520863450116e-06, | |
| "loss": 1.5548, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.5759113982464237, | |
| "grad_norm": 0.3036080300807953, | |
| "learning_rate": 9.228245378733537e-06, | |
| "loss": 1.6592, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.5906783571758191, | |
| "grad_norm": 0.3091510236263275, | |
| "learning_rate": 8.71601644853449e-06, | |
| "loss": 1.6066, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.6054453161052146, | |
| "grad_norm": 0.31578660011291504, | |
| "learning_rate": 8.207192411892645e-06, | |
| "loss": 1.6179, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.6202122750346101, | |
| "grad_norm": 0.3013366162776947, | |
| "learning_rate": 7.703122578682047e-06, | |
| "loss": 1.4649, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.6349792339640056, | |
| "grad_norm": 0.2899874150753021, | |
| "learning_rate": 7.2051436514839064e-06, | |
| "loss": 1.5725, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.649746192893401, | |
| "grad_norm": 0.28480279445648193, | |
| "learning_rate": 6.714576180891653e-06, | |
| "loss": 1.4933, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.6645131518227965, | |
| "grad_norm": 0.29248034954071045, | |
| "learning_rate": 6.232721063648148e-06, | |
| "loss": 1.5653, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.679280110752192, | |
| "grad_norm": 0.31977611780166626, | |
| "learning_rate": 5.760856092901394e-06, | |
| "loss": 1.661, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.6940470696815875, | |
| "grad_norm": 0.34047219157218933, | |
| "learning_rate": 5.300232569726805e-06, | |
| "loss": 1.5535, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.708814028610983, | |
| "grad_norm": 0.2981061339378357, | |
| "learning_rate": 4.852071984901696e-06, | |
| "loss": 1.5681, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.7235809875403784, | |
| "grad_norm": 0.29522624611854553, | |
| "learning_rate": 4.417562779731355e-06, | |
| "loss": 1.4628, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.7383479464697739, | |
| "grad_norm": 0.29412776231765747, | |
| "learning_rate": 3.997857194516319e-06, | |
| "loss": 1.6033, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.7531149053991694, | |
| "grad_norm": 0.3004733920097351, | |
| "learning_rate": 3.594068213018249e-06, | |
| "loss": 1.5287, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.7678818643285649, | |
| "grad_norm": 0.29276204109191895, | |
| "learning_rate": 3.207266611027069e-06, | |
| "loss": 1.4564, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.7826488232579604, | |
| "grad_norm": 0.30495619773864746, | |
| "learning_rate": 2.8384781168560693e-06, | |
| "loss": 1.4001, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.7974157821873558, | |
| "grad_norm": 0.2985684275627136, | |
| "learning_rate": 2.4886806912948034e-06, | |
| "loss": 1.5332, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.8121827411167513, | |
| "grad_norm": 0.30361080169677734, | |
| "learning_rate": 2.158801934232897e-06, | |
| "loss": 1.4737, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.8269497000461468, | |
| "grad_norm": 0.3165511190891266, | |
| "learning_rate": 1.8497166248318876e-06, | |
| "loss": 1.5006, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.8417166589755423, | |
| "grad_norm": 0.3422015905380249, | |
| "learning_rate": 1.5622444017681438e-06, | |
| "loss": 1.5113, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.8564836179049377, | |
| "grad_norm": 0.3274432420730591, | |
| "learning_rate": 1.2971475896984475e-06, | |
| "loss": 1.5036, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.8712505768343332, | |
| "grad_norm": 0.3002086579799652, | |
| "learning_rate": 1.0551291777120465e-06, | |
| "loss": 1.5217, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.8860175357637287, | |
| "grad_norm": 0.2978605031967163, | |
| "learning_rate": 8.368309551299536e-07, | |
| "loss": 1.5555, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.9007844946931242, | |
| "grad_norm": 0.30598872900009155, | |
| "learning_rate": 6.428318095950648e-07, | |
| "loss": 1.4636, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.9155514536225197, | |
| "grad_norm": 0.3087461292743683, | |
| "learning_rate": 4.73646191966175e-07, | |
| "loss": 1.4255, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.930318412551915, | |
| "grad_norm": 0.30991825461387634, | |
| "learning_rate": 3.2972275208679625e-07, | |
| "loss": 1.5838, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.9450853714813106, | |
| "grad_norm": 0.30382978916168213, | |
| "learning_rate": 2.1144314904642194e-07, | |
| "loss": 1.5332, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.9598523304107061, | |
| "grad_norm": 0.29928240180015564, | |
| "learning_rate": 1.1912103908922945e-07, | |
| "loss": 1.4615, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.9746192893401016, | |
| "grad_norm": 0.298584520816803, | |
| "learning_rate": 5.300124385410943e-08, | |
| "loss": 1.5308, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.989386248269497, | |
| "grad_norm": 0.3029971420764923, | |
| "learning_rate": 1.325910115169471e-08, | |
| "loss": 1.513, | |
| "memory/device_mem_reserved(gib)": 21.23, | |
| "memory/max_mem_active(gib)": 20.61, | |
| "memory/max_mem_allocated(gib)": 20.61, | |
| "step": 67 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 67, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 17, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 2.432859552916439e+18, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |