{ "best_metric": 1.0, "best_model_checkpoint": "full-sft-tsukuyomi-alpha0.1-google/gemma-2-2b-jpn-it/checkpoint-100", "epoch": 1.1764705882352942, "eval_steps": 100, "global_step": 100, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "custom_loss": 0.84, "epoch": 0, "final_loss": 15.123116493225098, "gpt4_eval_score": 0.16, "loss": 16.71013069152832, "step": 0 }, { "custom_loss": 0.94, "epoch": 0.011764705882352941, "final_loss": 15.45504379272461, "gpt4_eval_score": 0.06000000000000001, "loss": 17.067827224731445, "step": 1 }, { "custom_loss": 0.92, "epoch": 0.023529411764705882, "final_loss": 15.623565673828125, "gpt4_eval_score": 0.08, "loss": 17.257295608520508, "step": 2 }, { "custom_loss": 0.88, "epoch": 0.03529411764705882, "final_loss": 15.859143257141113, "gpt4_eval_score": 0.12000000000000002, "loss": 17.52349281311035, "step": 3 }, { "custom_loss": 0.88, "epoch": 0.047058823529411764, "final_loss": 14.435709953308105, "gpt4_eval_score": 0.12000000000000002, "loss": 15.941900253295898, "step": 4 }, { "custom_loss": 0.96, "epoch": 0.058823529411764705, "final_loss": 13.408587455749512, "gpt4_eval_score": 0.04, "loss": 14.791764259338379, "step": 5 }, { "custom_loss": 0.9, "epoch": 0.07058823529411765, "final_loss": 14.490974426269531, "gpt4_eval_score": 0.1, "loss": 16.001083374023438, "step": 6 }, { "custom_loss": 0.96, "epoch": 0.08235294117647059, "final_loss": 12.650106430053711, "gpt4_eval_score": 0.04, "loss": 13.949007987976074, "step": 7 }, { "custom_loss": 0.8, "epoch": 0.09411764705882353, "final_loss": 13.63055419921875, "gpt4_eval_score": 0.19999999999999998, "loss": 15.056171417236328, "step": 8 }, { "custom_loss": 0.84, "epoch": 0.10588235294117647, "final_loss": 12.176043510437012, "gpt4_eval_score": 0.16, "loss": 13.4356050491333, "step": 9 }, { "custom_loss": 0.72, "epoch": 0.11764705882352941, "final_loss": 11.87337589263916, "gpt4_eval_score": 0.28, "loss": 13.112640380859375, "step": 10 }, { "custom_loss": 0.74, "epoch": 0.12941176470588237, "final_loss": 10.65192985534668, "gpt4_eval_score": 0.26, "loss": 11.753254890441895, "step": 11 }, { "custom_loss": 0.84, "epoch": 0.1411764705882353, "final_loss": 9.939013481140137, "gpt4_eval_score": 0.16, "loss": 10.950016021728516, "step": 12 }, { "custom_loss": 0.8, "epoch": 0.15294117647058825, "final_loss": 8.823448181152344, "gpt4_eval_score": 0.2, "loss": 9.714942932128906, "step": 13 }, { "custom_loss": 0.8, "epoch": 0.16470588235294117, "final_loss": 7.698497295379639, "gpt4_eval_score": 0.2, "loss": 8.464997291564941, "step": 14 }, { "custom_loss": 0.92, "epoch": 0.17647058823529413, "final_loss": 6.19542121887207, "gpt4_eval_score": 0.08, "loss": 6.781579494476318, "step": 15 }, { "custom_loss": 0.9, "epoch": 0.18823529411764706, "final_loss": 6.678496360778809, "gpt4_eval_score": 0.1, "loss": 7.32055139541626, "step": 16 }, { "custom_loss": 0.9, "epoch": 0.2, "final_loss": 4.427515029907227, "gpt4_eval_score": 0.1, "loss": 4.819461345672607, "step": 17 }, { "custom_loss": 0.8200000000000001, "epoch": 0.21176470588235294, "final_loss": 4.527072429656982, "gpt4_eval_score": 0.18, "loss": 4.938969612121582, "step": 18 }, { "custom_loss": 0.76, "epoch": 0.2235294117647059, "final_loss": 3.5684115886688232, "gpt4_eval_score": 0.24000000000000005, "loss": 3.8804574012756348, "step": 19 }, { "custom_loss": 0.76, "epoch": 0.23529411764705882, "final_loss": 2.070030450820923, "gpt4_eval_score": 0.24000000000000005, "loss": 2.2155892848968506, "step": 20 }, { "custom_loss": 0.9, "epoch": 0.24705882352941178, "final_loss": 4.884355068206787, "gpt4_eval_score": 0.1, "loss": 5.327061176300049, "step": 21 }, { "custom_loss": 0.88, "epoch": 0.25882352941176473, "final_loss": 4.8519158363342285, "gpt4_eval_score": 0.12, "loss": 5.293240070343018, "step": 22 }, { "custom_loss": 0.82, "epoch": 0.27058823529411763, "final_loss": 3.471522569656372, "gpt4_eval_score": 0.18000000000000002, "loss": 3.7661361694335938, "step": 23 }, { "custom_loss": 0.84, "epoch": 0.2823529411764706, "final_loss": 3.1397957801818848, "gpt4_eval_score": 0.16, "loss": 3.3953285217285156, "step": 24 }, { "custom_loss": 0.8200000000000001, "epoch": 0.29411764705882354, "final_loss": 2.1852877140045166, "gpt4_eval_score": 0.18, "loss": 2.3369863033294678, "step": 25 }, { "custom_loss": 0.72, "epoch": 0.3058823529411765, "final_loss": 1.739311695098877, "gpt4_eval_score": 0.27999999999999997, "loss": 1.852568507194519, "step": 26 }, { "custom_loss": 0.6, "epoch": 0.3176470588235294, "final_loss": 1.2800674438476562, "gpt4_eval_score": 0.4, "loss": 1.35563063621521, "step": 27 }, { "custom_loss": 0.5, "epoch": 0.32941176470588235, "final_loss": 0.8763731718063354, "gpt4_eval_score": 0.5, "loss": 0.918192446231842, "step": 28 }, { "custom_loss": 0.3400000000000001, "epoch": 0.3411764705882353, "final_loss": 0.4881135821342468, "gpt4_eval_score": 0.6599999999999999, "loss": 0.5045706629753113, "step": 29 }, { "custom_loss": 0.19999999999999996, "epoch": 0.35294117647058826, "final_loss": 0.2739589214324951, "gpt4_eval_score": 0.8, "loss": 0.28217658400535583, "step": 30 }, { "custom_loss": 0.0, "epoch": 0.36470588235294116, "final_loss": 0.09473533928394318, "gpt4_eval_score": 1.0, "loss": 0.10526148974895477, "step": 31 }, { "custom_loss": 0.05999999999999994, "epoch": 0.3764705882352941, "final_loss": 0.08252310007810593, "gpt4_eval_score": 0.9400000000000001, "loss": 0.08502567559480667, "step": 32 }, { "custom_loss": 0.0, "epoch": 0.38823529411764707, "final_loss": 0.020817149430513382, "gpt4_eval_score": 1.0, "loss": 0.02313016727566719, "step": 33 }, { "custom_loss": 0.0, "epoch": 0.4, "final_loss": 0.016543544828891754, "gpt4_eval_score": 1.0, "loss": 0.018381716683506966, "step": 34 }, { "custom_loss": 0.0, "epoch": 0.4117647058823529, "final_loss": 0.005110062193125486, "gpt4_eval_score": 1.0, "loss": 0.005677847191691399, "step": 35 }, { "custom_loss": 0.0, "epoch": 0.4235294117647059, "final_loss": 0.014755932614207268, "gpt4_eval_score": 1.0, "loss": 0.016395481303334236, "step": 36 }, { "custom_loss": 0.18000000000000005, "epoch": 0.43529411764705883, "final_loss": 0.1277998387813568, "gpt4_eval_score": 0.82, "loss": 0.1219998151063919, "step": 37 }, { "custom_loss": 0.0, "epoch": 0.4470588235294118, "final_loss": 0.005542609374970198, "gpt4_eval_score": 1.0, "loss": 0.006158455274999142, "step": 38 }, { "custom_loss": 0.0, "epoch": 0.4588235294117647, "final_loss": 0.011761026456952095, "gpt4_eval_score": 1.0, "loss": 0.013067808002233505, "step": 39 }, { "custom_loss": 0.0, "epoch": 0.47058823529411764, "final_loss": 0.015592307783663273, "gpt4_eval_score": 1.0, "loss": 0.017324786633253098, "step": 40 }, { "custom_loss": 0.0, "epoch": 0.4823529411764706, "final_loss": 0.003539700759574771, "gpt4_eval_score": 1.0, "loss": 0.003933000843971968, "step": 41 }, { "custom_loss": 0.0, "epoch": 0.49411764705882355, "final_loss": 0.41602861881256104, "gpt4_eval_score": 1.0, "loss": 0.46225404739379883, "step": 42 }, { "custom_loss": 0.0, "epoch": 0.5058823529411764, "final_loss": 0.22934108972549438, "gpt4_eval_score": 1.0, "loss": 0.2548234462738037, "step": 43 }, { "custom_loss": 0.040000000000000036, "epoch": 0.5176470588235295, "final_loss": 0.11195889115333557, "gpt4_eval_score": 0.96, "loss": 0.11995432525873184, "step": 44 }, { "custom_loss": 0.0, "epoch": 0.5294117647058824, "final_loss": 0.0058891200460493565, "gpt4_eval_score": 1.0, "loss": 0.006543466821312904, "step": 45 }, { "custom_loss": 0.0, "epoch": 0.5411764705882353, "final_loss": 0.07183459401130676, "gpt4_eval_score": 1.0, "loss": 0.07981621474027634, "step": 46 }, { "custom_loss": 0.0, "epoch": 0.5529411764705883, "final_loss": 0.005363901611417532, "gpt4_eval_score": 1.0, "loss": 0.005959890782833099, "step": 47 }, { "custom_loss": 0.0, "epoch": 0.5647058823529412, "final_loss": 0.003014233196154237, "gpt4_eval_score": 1.0, "loss": 0.003349148202687502, "step": 48 }, { "custom_loss": 0.0, "epoch": 0.5764705882352941, "final_loss": 0.03570278361439705, "gpt4_eval_score": 1.0, "loss": 0.03966975957155228, "step": 49 }, { "custom_loss": 0.0, "epoch": 0.5882352941176471, "final_loss": 0.001202998566441238, "gpt4_eval_score": 1.0, "loss": 0.0013366651255637407, "step": 50 }, { "custom_loss": 0.0, "epoch": 0.6, "final_loss": 0.000932105933316052, "gpt4_eval_score": 1.0, "loss": 0.0010356733109802008, "step": 51 }, { "custom_loss": 0.0, "epoch": 0.611764705882353, "final_loss": 0.0009981885086745024, "gpt4_eval_score": 1.0, "loss": 0.0011090984335169196, "step": 52 }, { "custom_loss": 0.0, "epoch": 0.6235294117647059, "final_loss": 0.0053233616054058075, "gpt4_eval_score": 1.0, "loss": 0.005914846435189247, "step": 53 }, { "custom_loss": 0.0, "epoch": 0.6352941176470588, "final_loss": 0.005819522310048342, "gpt4_eval_score": 1.0, "loss": 0.00646613584831357, "step": 54 }, { "custom_loss": 0.0, "epoch": 0.6470588235294118, "final_loss": 0.0017121024429798126, "gpt4_eval_score": 1.0, "loss": 0.0019023361383005977, "step": 55 }, { "custom_loss": 0.15999999999999992, "epoch": 0.6588235294117647, "final_loss": 0.028725693002343178, "gpt4_eval_score": 0.8400000000000001, "loss": 0.014139658771455288, "step": 56 }, { "custom_loss": 0.0, "epoch": 0.6705882352941176, "final_loss": 0.000470877334009856, "gpt4_eval_score": 1.0, "loss": 0.0005231970571912825, "step": 57 }, { "custom_loss": 0.0, "epoch": 0.6823529411764706, "final_loss": 0.00031207178835757077, "gpt4_eval_score": 1.0, "loss": 0.0003467464412096888, "step": 58 }, { "custom_loss": 0.0, "epoch": 0.6941176470588235, "final_loss": 0.00013724762538913637, "gpt4_eval_score": 1.0, "loss": 0.00015249736316036433, "step": 59 }, { "custom_loss": 0.0, "epoch": 0.7058823529411765, "final_loss": 0.008362076245248318, "gpt4_eval_score": 1.0, "loss": 0.00929119624197483, "step": 60 }, { "custom_loss": 0.0, "epoch": 0.7176470588235294, "final_loss": 0.0010530626168474555, "gpt4_eval_score": 1.0, "loss": 0.0011700695613399148, "step": 61 }, { "custom_loss": 0.0, "epoch": 0.7294117647058823, "final_loss": 0.0001670990459388122, "gpt4_eval_score": 1.0, "loss": 0.00018566561629995704, "step": 62 }, { "custom_loss": 0.05999999999999994, "epoch": 0.7411764705882353, "final_loss": 0.11876700818538666, "gpt4_eval_score": 0.9400000000000001, "loss": 0.1252966821193695, "step": 63 }, { "custom_loss": 0.0, "epoch": 0.7529411764705882, "final_loss": 0.08292262256145477, "gpt4_eval_score": 1.0, "loss": 0.09213624894618988, "step": 64 }, { "custom_loss": 0.0, "epoch": 0.7647058823529411, "final_loss": 0.02646959014236927, "gpt4_eval_score": 1.0, "loss": 0.029410656541585922, "step": 65 }, { "custom_loss": 0.019999999999999907, "epoch": 0.7764705882352941, "final_loss": 0.016994250938296318, "gpt4_eval_score": 0.9800000000000001, "loss": 0.016660278663039207, "step": 66 }, { "custom_loss": 0.019999999999999907, "epoch": 0.788235294117647, "final_loss": 0.052947890013456345, "gpt4_eval_score": 0.9800000000000001, "loss": 0.056608766317367554, "step": 67 }, { "custom_loss": 0.0, "epoch": 0.8, "final_loss": 0.004573707468807697, "gpt4_eval_score": 1.0, "loss": 0.005081897135823965, "step": 68 }, { "custom_loss": 0.0, "epoch": 0.8117647058823529, "final_loss": 0.004014432430267334, "gpt4_eval_score": 1.0, "loss": 0.004460480529814959, "step": 69 }, { "custom_loss": 0.0, "epoch": 0.8235294117647058, "final_loss": 0.003813790623098612, "gpt4_eval_score": 1.0, "loss": 0.004237545188516378, "step": 70 }, { "custom_loss": 0.0, "epoch": 0.8352941176470589, "final_loss": 0.0008132688235491514, "gpt4_eval_score": 1.0, "loss": 0.0009036320261657238, "step": 71 }, { "custom_loss": 0.0, "epoch": 0.8470588235294118, "final_loss": 0.00024077681882772595, "gpt4_eval_score": 1.0, "loss": 0.0002675298019312322, "step": 72 }, { "custom_loss": 0.0, "epoch": 0.8588235294117647, "final_loss": 0.0013859015889465809, "gpt4_eval_score": 1.0, "loss": 0.0015398906543850899, "step": 73 }, { "custom_loss": 0.019999999999999907, "epoch": 0.8705882352941177, "final_loss": 0.02203843742609024, "gpt4_eval_score": 0.9800000000000001, "loss": 0.022264931350946426, "step": 74 }, { "custom_loss": 0.0, "epoch": 0.8823529411764706, "final_loss": 0.0024588347878307104, "gpt4_eval_score": 1.0, "loss": 0.0027320387307554483, "step": 75 }, { "custom_loss": 0.0, "epoch": 0.8941176470588236, "final_loss": 0.00021336309146136045, "gpt4_eval_score": 1.0, "loss": 0.00023707011132501066, "step": 76 }, { "custom_loss": 0.019999999999999907, "epoch": 0.9058823529411765, "final_loss": 0.06269317865371704, "gpt4_eval_score": 0.9800000000000001, "loss": 0.06743686646223068, "step": 77 }, { "custom_loss": 0.0, "epoch": 0.9176470588235294, "final_loss": 0.0004480383940972388, "gpt4_eval_score": 1.0, "loss": 0.0004978204378858209, "step": 78 }, { "custom_loss": 0.040000000000000036, "epoch": 0.9294117647058824, "final_loss": 0.038131676614284515, "gpt4_eval_score": 0.96, "loss": 0.037924084812402725, "step": 79 }, { "custom_loss": 0.0, "epoch": 0.9411764705882353, "final_loss": 0.00026650883955881, "gpt4_eval_score": 1.0, "loss": 0.0002961209393106401, "step": 80 }, { "custom_loss": 0.0, "epoch": 0.9529411764705882, "final_loss": 7.75257867644541e-05, "gpt4_eval_score": 1.0, "loss": 8.61397638800554e-05, "step": 81 }, { "custom_loss": 0.0, "epoch": 0.9647058823529412, "final_loss": 0.00010220255353488028, "gpt4_eval_score": 1.0, "loss": 0.00011355839524185285, "step": 82 }, { "custom_loss": 0.0, "epoch": 0.9764705882352941, "final_loss": 0.0018145291833207011, "gpt4_eval_score": 1.0, "loss": 0.0020161436405032873, "step": 83 }, { "custom_loss": 0.0, "epoch": 0.9882352941176471, "final_loss": 4.244365482009016e-05, "gpt4_eval_score": 1.0, "loss": 4.7159617679426447e-05, "step": 84 }, { "custom_loss": 0.0, "epoch": 1.0, "final_loss": 0.009399116970598698, "gpt4_eval_score": 1.0, "loss": 0.010443463921546936, "step": 85 }, { "custom_loss": 0.0, "epoch": 1.011764705882353, "final_loss": 0.0012943892506882548, "gpt4_eval_score": 1.0, "loss": 0.0014382103690877557, "step": 86 }, { "custom_loss": 0.019999999999999907, "epoch": 1.0235294117647058, "final_loss": 0.03385946527123451, "gpt4_eval_score": 0.9800000000000001, "loss": 0.035399407148361206, "step": 87 }, { "custom_loss": 0.019999999999999907, "epoch": 1.035294117647059, "final_loss": 0.014671835117042065, "gpt4_eval_score": 0.9800000000000001, "loss": 0.01407981663942337, "step": 88 }, { "custom_loss": 0.0, "epoch": 1.0470588235294118, "final_loss": 0.0004618468228727579, "gpt4_eval_score": 1.0, "loss": 0.0005131631623953581, "step": 89 }, { "custom_loss": 0.0, "epoch": 1.0588235294117647, "final_loss": 0.00023186064208857715, "gpt4_eval_score": 1.0, "loss": 0.00025762294535525143, "step": 90 }, { "custom_loss": 0.0, "epoch": 1.0705882352941176, "final_loss": 0.001300316653214395, "gpt4_eval_score": 1.0, "loss": 0.0014447963330894709, "step": 91 }, { "custom_loss": 0.0, "epoch": 1.0823529411764705, "final_loss": 0.001997714163735509, "gpt4_eval_score": 1.0, "loss": 0.0022196825593709946, "step": 92 }, { "custom_loss": 0.0, "epoch": 1.0941176470588236, "final_loss": 0.0008706891676411033, "gpt4_eval_score": 1.0, "loss": 0.0009674324537627399, "step": 93 }, { "custom_loss": 0.0, "epoch": 1.1058823529411765, "final_loss": 0.00027742586098611355, "gpt4_eval_score": 1.0, "loss": 0.00030825097928754985, "step": 94 }, { "custom_loss": 0.0, "epoch": 1.1176470588235294, "final_loss": 0.0004029519041068852, "gpt4_eval_score": 1.0, "loss": 0.00044772433466278017, "step": 95 }, { "custom_loss": 0.0, "epoch": 1.1294117647058823, "final_loss": 0.0004211719788145274, "gpt4_eval_score": 1.0, "loss": 0.00046796887181699276, "step": 96 }, { "custom_loss": 0.0, "epoch": 1.1411764705882352, "final_loss": 0.0004806466167792678, "gpt4_eval_score": 1.0, "loss": 0.0005340517964214087, "step": 97 }, { "custom_loss": 0.0, "epoch": 1.1529411764705881, "final_loss": 0.00013472369755618274, "gpt4_eval_score": 1.0, "loss": 0.00014969300536904484, "step": 98 }, { "custom_loss": 0.0, "epoch": 1.1647058823529413, "final_loss": 5.3839205065742135e-05, "gpt4_eval_score": 1.0, "loss": 5.982134098303504e-05, "step": 99 }, { "epoch": 1.1764705882352942, "grad_norm": 0.01043701171875, "learning_rate": 7.637304728380036e-06, "loss": 2.4434, "step": 100 }, { "custom_loss": 0.0, "epoch": 1.1764705882352942, "final_loss": 0.0015407466562464833, "gpt4_eval_score": 1.0, "loss": 0.001711940742097795, "step": 100 }, { "custom_loss": 0.0, "epoch": 1.1764705882352942, "final_loss": 0.0032093855552375317, "gpt4_eval_score": 1.0, "loss": 0.0035659840796142817, "step": 100 }, { "custom_loss": 0.0, "epoch": 1.1764705882352942, "final_loss": 0.000593104399740696, "gpt4_eval_score": 1.0, "loss": 0.000659004901535809, "step": 100 }, { "epoch": 1.1764705882352942, "eval_gpt4_eval_score": 1.0, "eval_loss": 0.0018327295547351241, "eval_runtime": 33.8528, "eval_samples_per_second": 0.679, "eval_steps_per_second": 0.089, "step": 100 } ], "logging_steps": 100, "max_steps": 255, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 100, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 5, "early_stopping_threshold": 0.0 }, "attributes": { "early_stopping_patience_counter": 0 } }, "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 215319624496128.0, "train_batch_size": 5, "trial_name": null, "trial_params": null }