|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 3.0, |
|
"eval_steps": 20, |
|
"global_step": 81, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2e-05, |
|
"loss": 0.9768, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 4e-05, |
|
"loss": 1.0553, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 6e-05, |
|
"loss": 0.9074, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 8e-05, |
|
"loss": 1.0351, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 0.0001, |
|
"loss": 0.9918, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 0.00012, |
|
"loss": 0.9872, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 0.00014, |
|
"loss": 0.9573, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 0.00016, |
|
"loss": 1.0466, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 0.00018, |
|
"loss": 0.8995, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 0.0002, |
|
"loss": 0.9041, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 0.00019996841892833, |
|
"loss": 0.936, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 0.00019987369566060176, |
|
"loss": 0.8254, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 0.0001997158900260614, |
|
"loss": 0.9508, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 0.00019949510169813003, |
|
"loss": 0.929, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 0.0001992114701314478, |
|
"loss": 0.9618, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 0.0001988651744737914, |
|
"loss": 0.9317, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 0.00019845643345292054, |
|
"loss": 0.9399, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 0.0001979855052384247, |
|
"loss": 0.9377, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 0.00019745268727865774, |
|
"loss": 0.9048, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 0.0001968583161128631, |
|
"loss": 0.9311, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"eval_loss": 0.8045752644538879, |
|
"eval_runtime": 2.684, |
|
"eval_samples_per_second": 1.118, |
|
"eval_steps_per_second": 0.745, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 0.0001962027671586086, |
|
"loss": 0.9376, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 0.00019548645447466431, |
|
"loss": 0.8598, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 0.00019470983049947444, |
|
"loss": 0.991, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 0.00019387338576538744, |
|
"loss": 0.8472, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 0.00019297764858882514, |
|
"loss": 0.8818, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 0.00019202318473658705, |
|
"loss": 0.8879, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 0.00019101059706849957, |
|
"loss": 0.8483, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"learning_rate": 0.0001899405251566371, |
|
"loss": 0.9505, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"learning_rate": 0.00018881364488135448, |
|
"loss": 0.9116, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 0.00018763066800438636, |
|
"loss": 0.8575, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 1.15, |
|
"learning_rate": 0.00018639234171928353, |
|
"loss": 0.8093, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 1.19, |
|
"learning_rate": 0.00018509944817946922, |
|
"loss": 0.7966, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"learning_rate": 0.0001837528040042142, |
|
"loss": 0.8263, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 0.00018235325976284275, |
|
"loss": 0.7951, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 1.3, |
|
"learning_rate": 0.00018090169943749476, |
|
"loss": 0.849, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 1.33, |
|
"learning_rate": 0.00017939903986478355, |
|
"loss": 0.863, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"learning_rate": 0.00017784623015670238, |
|
"loss": 0.8144, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 0.0001762442511011448, |
|
"loss": 0.8078, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 1.44, |
|
"learning_rate": 0.00017459411454241822, |
|
"loss": 0.7997, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 1.48, |
|
"learning_rate": 0.00017289686274214118, |
|
"loss": 0.9322, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 1.48, |
|
"eval_loss": 0.7793169617652893, |
|
"eval_runtime": 2.6811, |
|
"eval_samples_per_second": 1.119, |
|
"eval_steps_per_second": 0.746, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 1.52, |
|
"learning_rate": 0.00017115356772092857, |
|
"loss": 0.8279, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 1.56, |
|
"learning_rate": 0.0001693653305812805, |
|
"loss": 0.8759, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 1.59, |
|
"learning_rate": 0.00016753328081210245, |
|
"loss": 0.8748, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 1.63, |
|
"learning_rate": 0.00016565857557529566, |
|
"loss": 0.7638, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 1.67, |
|
"learning_rate": 0.000163742398974869, |
|
"loss": 0.7941, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 1.7, |
|
"learning_rate": 0.00016178596130903344, |
|
"loss": 0.8321, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 1.74, |
|
"learning_rate": 0.0001597904983057519, |
|
"loss": 0.894, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 1.78, |
|
"learning_rate": 0.00015775727034222675, |
|
"loss": 0.9176, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 1.81, |
|
"learning_rate": 0.00015568756164881882, |
|
"loss": 0.8286, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"learning_rate": 0.00015358267949789966, |
|
"loss": 0.9328, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 0.00015144395337815064, |
|
"loss": 0.8644, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 1.93, |
|
"learning_rate": 0.00014927273415482915, |
|
"loss": 0.7769, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 1.96, |
|
"learning_rate": 0.0001470703932165333, |
|
"loss": 0.8, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 0.00014483832160900326, |
|
"loss": 0.7781, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 2.04, |
|
"learning_rate": 0.00014257792915650728, |
|
"loss": 0.7852, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 2.07, |
|
"learning_rate": 0.00014029064357136628, |
|
"loss": 0.7796, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 2.11, |
|
"learning_rate": 0.00013797790955218014, |
|
"loss": 0.8287, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 2.15, |
|
"learning_rate": 0.00013564118787132506, |
|
"loss": 0.6845, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 2.19, |
|
"learning_rate": 0.00013328195445229868, |
|
"loss": 0.7821, |
|
"step": 59 |
|
}, |
|
{ |
|
"epoch": 2.22, |
|
"learning_rate": 0.00013090169943749476, |
|
"loss": 0.708, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 2.22, |
|
"eval_loss": 0.7880761027336121, |
|
"eval_runtime": 2.6843, |
|
"eval_samples_per_second": 1.118, |
|
"eval_steps_per_second": 0.745, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"learning_rate": 0.0001285019262469976, |
|
"loss": 0.8098, |
|
"step": 61 |
|
}, |
|
{ |
|
"epoch": 2.3, |
|
"learning_rate": 0.00012608415062898972, |
|
"loss": 0.82, |
|
"step": 62 |
|
}, |
|
{ |
|
"epoch": 2.33, |
|
"learning_rate": 0.00012364989970237248, |
|
"loss": 0.7187, |
|
"step": 63 |
|
}, |
|
{ |
|
"epoch": 2.37, |
|
"learning_rate": 0.00012120071099220549, |
|
"loss": 0.7802, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 2.41, |
|
"learning_rate": 0.00011873813145857249, |
|
"loss": 0.6834, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"learning_rate": 0.00011626371651948838, |
|
"loss": 0.6808, |
|
"step": 66 |
|
}, |
|
{ |
|
"epoch": 2.48, |
|
"learning_rate": 0.0001137790290684638, |
|
"loss": 0.7881, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 2.52, |
|
"learning_rate": 0.00011128563848734816, |
|
"loss": 0.7281, |
|
"step": 68 |
|
}, |
|
{ |
|
"epoch": 2.56, |
|
"learning_rate": 0.00010878511965507434, |
|
"loss": 0.7231, |
|
"step": 69 |
|
}, |
|
{ |
|
"epoch": 2.59, |
|
"learning_rate": 0.00010627905195293135, |
|
"loss": 0.6938, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 2.63, |
|
"learning_rate": 0.00010376901826699348, |
|
"loss": 0.7633, |
|
"step": 71 |
|
}, |
|
{ |
|
"epoch": 2.67, |
|
"learning_rate": 0.00010125660398833528, |
|
"loss": 0.8253, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 2.7, |
|
"learning_rate": 9.874339601166473e-05, |
|
"loss": 0.8197, |
|
"step": 73 |
|
}, |
|
{ |
|
"epoch": 2.74, |
|
"learning_rate": 9.623098173300654e-05, |
|
"loss": 0.7403, |
|
"step": 74 |
|
}, |
|
{ |
|
"epoch": 2.78, |
|
"learning_rate": 9.372094804706867e-05, |
|
"loss": 0.8175, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 2.81, |
|
"learning_rate": 9.121488034492569e-05, |
|
"loss": 0.7249, |
|
"step": 76 |
|
}, |
|
{ |
|
"epoch": 2.85, |
|
"learning_rate": 8.871436151265184e-05, |
|
"loss": 0.7029, |
|
"step": 77 |
|
}, |
|
{ |
|
"epoch": 2.89, |
|
"learning_rate": 8.62209709315362e-05, |
|
"loss": 0.8081, |
|
"step": 78 |
|
}, |
|
{ |
|
"epoch": 2.93, |
|
"learning_rate": 8.373628348051165e-05, |
|
"loss": 0.7087, |
|
"step": 79 |
|
}, |
|
{ |
|
"epoch": 2.96, |
|
"learning_rate": 8.126186854142752e-05, |
|
"loss": 0.762, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 2.96, |
|
"eval_loss": 0.7806326746940613, |
|
"eval_runtime": 2.6841, |
|
"eval_samples_per_second": 1.118, |
|
"eval_steps_per_second": 0.745, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 7.879928900779456e-05, |
|
"loss": 0.6724, |
|
"step": 81 |
|
} |
|
], |
|
"logging_steps": 1, |
|
"max_steps": 135, |
|
"num_train_epochs": 5, |
|
"save_steps": 500, |
|
"total_flos": 9.83564041740288e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|