初始化项目,由ModelHub XC社区提供模型
Model: hongzhouyu/FineMedLM-o1 Source: Original Platform
This commit is contained in:
409
trainer_state.json
Normal file
409
trainer_state.json
Normal file
@@ -0,0 +1,409 @@
|
||||
{
|
||||
"best_global_step": null,
|
||||
"best_metric": null,
|
||||
"best_model_checkpoint": null,
|
||||
"epoch": 0.9992709599027947,
|
||||
"eval_steps": 500,
|
||||
"global_step": 257,
|
||||
"is_hyper_param_search": false,
|
||||
"is_local_process_zero": true,
|
||||
"is_world_process_zero": true,
|
||||
"log_history": [
|
||||
{
|
||||
"epoch": 0.038882138517618466,
|
||||
"grad_norm": 160.0573020755792,
|
||||
"learning_rate": 3.846153846153846e-08,
|
||||
"logits/chosen": -0.7651611566543579,
|
||||
"logits/rejected": -0.9917968511581421,
|
||||
"logps/chosen": -1039.6500244140625,
|
||||
"logps/rejected": -269.35626220703125,
|
||||
"loss": 0.7032,
|
||||
"rewards/accuracies": 0.17109374701976776,
|
||||
"rewards/chosen": -0.00718765240162611,
|
||||
"rewards/margins": -0.006218528840690851,
|
||||
"rewards/rejected": -0.0009794235229492188,
|
||||
"step": 10
|
||||
},
|
||||
{
|
||||
"epoch": 0.07776427703523693,
|
||||
"grad_norm": 160.5753020431693,
|
||||
"learning_rate": 7.692307692307692e-08,
|
||||
"logits/chosen": -0.7669433355331421,
|
||||
"logits/rejected": -0.991040050983429,
|
||||
"logps/chosen": -1027.425048828125,
|
||||
"logps/rejected": -273.2875061035156,
|
||||
"loss": 0.6959,
|
||||
"rewards/accuracies": 0.23593750596046448,
|
||||
"rewards/chosen": 0.008073806762695312,
|
||||
"rewards/margins": 0.009277725592255592,
|
||||
"rewards/rejected": -0.0012153625721111894,
|
||||
"step": 20
|
||||
},
|
||||
{
|
||||
"epoch": 0.1166464155528554,
|
||||
"grad_norm": 144.69837671419592,
|
||||
"learning_rate": 9.992603458948281e-08,
|
||||
"logits/chosen": -0.766284167766571,
|
||||
"logits/rejected": -0.9904540777206421,
|
||||
"logps/chosen": -1034.0250244140625,
|
||||
"logps/rejected": -269.0625,
|
||||
"loss": 0.6534,
|
||||
"rewards/accuracies": 0.4437499940395355,
|
||||
"rewards/chosen": 0.06487135589122772,
|
||||
"rewards/margins": 0.10399703681468964,
|
||||
"rewards/rejected": -0.03914298862218857,
|
||||
"step": 30
|
||||
},
|
||||
{
|
||||
"epoch": 0.15552855407047386,
|
||||
"grad_norm": 114.38616995997658,
|
||||
"learning_rate": 9.909643486313532e-08,
|
||||
"logits/chosen": -0.7626953125,
|
||||
"logits/rejected": -0.987060546875,
|
||||
"logps/chosen": -1035.2249755859375,
|
||||
"logps/rejected": -268.70001220703125,
|
||||
"loss": 0.5462,
|
||||
"rewards/accuracies": 0.81640625,
|
||||
"rewards/chosen": 0.22022247314453125,
|
||||
"rewards/margins": 0.36003416776657104,
|
||||
"rewards/rejected": -0.1398262083530426,
|
||||
"step": 40
|
||||
},
|
||||
{
|
||||
"epoch": 0.19441069258809235,
|
||||
"grad_norm": 95.18572051179285,
|
||||
"learning_rate": 9.73601505145955e-08,
|
||||
"logits/chosen": -0.7583252191543579,
|
||||
"logits/rejected": -0.9866698980331421,
|
||||
"logps/chosen": -1038.300048828125,
|
||||
"logps/rejected": -277.33123779296875,
|
||||
"loss": 0.4385,
|
||||
"rewards/accuracies": 0.983593761920929,
|
||||
"rewards/chosen": 0.379730224609375,
|
||||
"rewards/margins": 0.663256824016571,
|
||||
"rewards/rejected": -0.283303827047348,
|
||||
"step": 50
|
||||
},
|
||||
{
|
||||
"epoch": 0.2332928311057108,
|
||||
"grad_norm": 58.648166089179284,
|
||||
"learning_rate": 9.474924624108548e-08,
|
||||
"logits/chosen": -0.7514404058456421,
|
||||
"logits/rejected": -0.9781738519668579,
|
||||
"logps/chosen": -1039.574951171875,
|
||||
"logps/rejected": -274.51873779296875,
|
||||
"loss": 0.3168,
|
||||
"rewards/accuracies": 0.999218761920929,
|
||||
"rewards/chosen": 0.5596984624862671,
|
||||
"rewards/margins": 1.0992431640625,
|
||||
"rewards/rejected": -0.539135754108429,
|
||||
"step": 60
|
||||
},
|
||||
{
|
||||
"epoch": 0.27217496962332927,
|
||||
"grad_norm": 38.61380385007199,
|
||||
"learning_rate": 9.131193871579974e-08,
|
||||
"logits/chosen": -0.740234375,
|
||||
"logits/rejected": -0.972973644733429,
|
||||
"logps/chosen": -1028.800048828125,
|
||||
"logps/rejected": -280.6625061035156,
|
||||
"loss": 0.1797,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 0.835314929485321,
|
||||
"rewards/margins": 1.780029296875,
|
||||
"rewards/rejected": -0.9450439214706421,
|
||||
"step": 70
|
||||
},
|
||||
{
|
||||
"epoch": 0.3110571081409477,
|
||||
"grad_norm": 24.436947703379207,
|
||||
"learning_rate": 8.71117061502135e-08,
|
||||
"logits/chosen": -0.7481689453125,
|
||||
"logits/rejected": -0.9714599847793579,
|
||||
"logps/chosen": -1023.0499877929688,
|
||||
"logps/rejected": -284.95623779296875,
|
||||
"loss": 0.1221,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 0.9628051519393921,
|
||||
"rewards/margins": 2.2330079078674316,
|
||||
"rewards/rejected": -1.270166039466858,
|
||||
"step": 80
|
||||
},
|
||||
{
|
||||
"epoch": 0.34993924665856624,
|
||||
"grad_norm": 31.163290960443195,
|
||||
"learning_rate": 8.222611601511082e-08,
|
||||
"logits/chosen": -0.7407470941543579,
|
||||
"logits/rejected": -0.968579113483429,
|
||||
"logps/chosen": -1046.9000244140625,
|
||||
"logps/rejected": -283.3500061035156,
|
||||
"loss": 0.0878,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.033911108970642,
|
||||
"rewards/margins": 2.625439405441284,
|
||||
"rewards/rejected": -1.591650366783142,
|
||||
"step": 90
|
||||
},
|
||||
{
|
||||
"epoch": 0.3888213851761847,
|
||||
"grad_norm": 10.132641150484275,
|
||||
"learning_rate": 7.674539256930362e-08,
|
||||
"logits/chosen": -0.73974609375,
|
||||
"logits/rejected": -0.960034191608429,
|
||||
"logps/chosen": -1007.0999755859375,
|
||||
"logps/rejected": -294.1812438964844,
|
||||
"loss": 0.0515,
|
||||
"rewards/accuracies": 0.999218761920929,
|
||||
"rewards/chosen": 1.075415015220642,
|
||||
"rewards/margins": 3.2860350608825684,
|
||||
"rewards/rejected": -2.210693359375,
|
||||
"step": 100
|
||||
},
|
||||
{
|
||||
"epoch": 0.42770352369380316,
|
||||
"grad_norm": 3.381766003891569,
|
||||
"learning_rate": 7.077075065009432e-08,
|
||||
"logits/chosen": -0.735546886920929,
|
||||
"logits/rejected": -0.957446277141571,
|
||||
"logps/chosen": -1014.7249755859375,
|
||||
"logps/rejected": -305.45623779296875,
|
||||
"loss": 0.0199,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.205664038658142,
|
||||
"rewards/margins": 4.345410346984863,
|
||||
"rewards/rejected": -3.1407227516174316,
|
||||
"step": 110
|
||||
},
|
||||
{
|
||||
"epoch": 0.4665856622114216,
|
||||
"grad_norm": 2.9204173229145636,
|
||||
"learning_rate": 6.441252649607854e-08,
|
||||
"logits/chosen": -0.729931652545929,
|
||||
"logits/rejected": -0.951171875,
|
||||
"logps/chosen": -1013.6749877929688,
|
||||
"logps/rejected": -311.09375,
|
||||
"loss": 0.0101,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.242578148841858,
|
||||
"rewards/margins": 5.182031154632568,
|
||||
"rewards/rejected": -3.9403319358825684,
|
||||
"step": 120
|
||||
},
|
||||
{
|
||||
"epoch": 0.5054678007290401,
|
||||
"grad_norm": 1.4356612196732965,
|
||||
"learning_rate": 5.7788140121173145e-08,
|
||||
"logits/chosen": -0.7265380620956421,
|
||||
"logits/rejected": -0.9493408203125,
|
||||
"logps/chosen": -1042.8499755859375,
|
||||
"logps/rejected": -315.17498779296875,
|
||||
"loss": 0.0053,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.3319091796875,
|
||||
"rewards/margins": 5.879687309265137,
|
||||
"rewards/rejected": -4.5478515625,
|
||||
"step": 130
|
||||
},
|
||||
{
|
||||
"epoch": 0.5443499392466585,
|
||||
"grad_norm": 1.344341217145263,
|
||||
"learning_rate": 5.1019926869570276e-08,
|
||||
"logits/chosen": -0.7250732183456421,
|
||||
"logits/rejected": -0.9361572265625,
|
||||
"logps/chosen": -1033.800048828125,
|
||||
"logps/rejected": -322.75,
|
||||
"loss": 0.0038,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.329248070716858,
|
||||
"rewards/margins": 6.33203125,
|
||||
"rewards/rejected": -5.001562595367432,
|
||||
"step": 140
|
||||
},
|
||||
{
|
||||
"epoch": 0.583232077764277,
|
||||
"grad_norm": 0.5751713296126671,
|
||||
"learning_rate": 4.4232878197208657e-08,
|
||||
"logits/chosen": -0.720263659954071,
|
||||
"logits/rejected": -0.9346923828125,
|
||||
"logps/chosen": -1047.300048828125,
|
||||
"logps/rejected": -326.04998779296875,
|
||||
"loss": 0.0028,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.3618042469024658,
|
||||
"rewards/margins": 6.713671684265137,
|
||||
"rewards/rejected": -5.349413871765137,
|
||||
"step": 150
|
||||
},
|
||||
{
|
||||
"epoch": 0.6221142162818954,
|
||||
"grad_norm": 0.8087854397697334,
|
||||
"learning_rate": 3.75523334017065e-08,
|
||||
"logits/chosen": -0.7223144769668579,
|
||||
"logits/rejected": -0.9326171875,
|
||||
"logps/chosen": -1028.300048828125,
|
||||
"logps/rejected": -328.9125061035156,
|
||||
"loss": 0.0021,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.3705565929412842,
|
||||
"rewards/margins": 6.984765529632568,
|
||||
"rewards/rejected": -5.613085746765137,
|
||||
"step": 160
|
||||
},
|
||||
{
|
||||
"epoch": 0.660996354799514,
|
||||
"grad_norm": 1.4449125849513227,
|
||||
"learning_rate": 3.1101664928554675e-08,
|
||||
"logits/chosen": -0.729809582233429,
|
||||
"logits/rejected": -0.9361816644668579,
|
||||
"logps/chosen": -1021.7249755859375,
|
||||
"logps/rejected": -330.04998779296875,
|
||||
"loss": 0.0019,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.3477783203125,
|
||||
"rewards/margins": 7.177734375,
|
||||
"rewards/rejected": -5.828906059265137,
|
||||
"step": 170
|
||||
},
|
||||
{
|
||||
"epoch": 0.6998784933171325,
|
||||
"grad_norm": 1.1450098797738104,
|
||||
"learning_rate": 2.500000000000001e-08,
|
||||
"logits/chosen": -0.722607433795929,
|
||||
"logits/rejected": -0.932812511920929,
|
||||
"logps/chosen": -1011.75,
|
||||
"logps/rejected": -328.375,
|
||||
"loss": 0.0016,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.342016577720642,
|
||||
"rewards/margins": 7.329297065734863,
|
||||
"rewards/rejected": -5.984179496765137,
|
||||
"step": 180
|
||||
},
|
||||
{
|
||||
"epoch": 0.7387606318347509,
|
||||
"grad_norm": 0.43432731079566683,
|
||||
"learning_rate": 1.9360020642261153e-08,
|
||||
"logits/chosen": -0.7257324457168579,
|
||||
"logits/rejected": -0.9291747808456421,
|
||||
"logps/chosen": -1048.449951171875,
|
||||
"logps/rejected": -332.2875061035156,
|
||||
"loss": 0.0021,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.404748558998108,
|
||||
"rewards/margins": 7.463281154632568,
|
||||
"rewards/rejected": -6.056640625,
|
||||
"step": 190
|
||||
},
|
||||
{
|
||||
"epoch": 0.7776427703523694,
|
||||
"grad_norm": 0.42312272714084886,
|
||||
"learning_rate": 1.428588273890482e-08,
|
||||
"logits/chosen": -0.725415050983429,
|
||||
"logits/rejected": -0.9350341558456421,
|
||||
"logps/chosen": -1040.5999755859375,
|
||||
"logps/rejected": -332.82501220703125,
|
||||
"loss": 0.0014,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.384423851966858,
|
||||
"rewards/margins": 7.4599609375,
|
||||
"rewards/rejected": -6.073828220367432,
|
||||
"step": 200
|
||||
},
|
||||
{
|
||||
"epoch": 0.8165249088699879,
|
||||
"grad_norm": 2.545449065189939,
|
||||
"learning_rate": 9.871292540103377e-09,
|
||||
"logits/chosen": -0.722412109375,
|
||||
"logits/rejected": -0.929858386516571,
|
||||
"logps/chosen": -1036.6500244140625,
|
||||
"logps/rejected": -329.54998779296875,
|
||||
"loss": 0.0014,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.4049561023712158,
|
||||
"rewards/margins": 7.583984375,
|
||||
"rewards/rejected": -6.179296970367432,
|
||||
"step": 210
|
||||
},
|
||||
{
|
||||
"epoch": 0.8554070473876063,
|
||||
"grad_norm": 0.43163218586837404,
|
||||
"learning_rate": 6.1977761496908695e-09,
|
||||
"logits/chosen": -0.7271728515625,
|
||||
"logits/rejected": -0.9333740472793579,
|
||||
"logps/chosen": -1029.324951171875,
|
||||
"logps/rejected": -330.95001220703125,
|
||||
"loss": 0.0014,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.387841820716858,
|
||||
"rewards/margins": 7.583984375,
|
||||
"rewards/rejected": -6.195703029632568,
|
||||
"step": 220
|
||||
},
|
||||
{
|
||||
"epoch": 0.8942891859052248,
|
||||
"grad_norm": 1.1334297545543626,
|
||||
"learning_rate": 3.3331739481316624e-09,
|
||||
"logits/chosen": -0.7221924066543579,
|
||||
"logits/rejected": -0.9290771484375,
|
||||
"logps/chosen": -1033.625,
|
||||
"logps/rejected": -335.51251220703125,
|
||||
"loss": 0.0015,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.386499047279358,
|
||||
"rewards/margins": 7.572070121765137,
|
||||
"rewards/rejected": -6.1845703125,
|
||||
"step": 230
|
||||
},
|
||||
{
|
||||
"epoch": 0.9331713244228432,
|
||||
"grad_norm": 0.2245653138567375,
|
||||
"learning_rate": 1.3303877555292442e-09,
|
||||
"logits/chosen": -0.725512683391571,
|
||||
"logits/rejected": -0.934619128704071,
|
||||
"logps/chosen": -1013.4249877929688,
|
||||
"logps/rejected": -333.98748779296875,
|
||||
"loss": 0.0012,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.3474853038787842,
|
||||
"rewards/margins": 7.607226371765137,
|
||||
"rewards/rejected": -6.259961128234863,
|
||||
"step": 240
|
||||
},
|
||||
{
|
||||
"epoch": 0.9720534629404617,
|
||||
"grad_norm": 0.34410802938606294,
|
||||
"learning_rate": 2.2640387134577055e-10,
|
||||
"logits/chosen": -0.724291980266571,
|
||||
"logits/rejected": -0.932788074016571,
|
||||
"logps/chosen": -1036.6500244140625,
|
||||
"logps/rejected": -330.67498779296875,
|
||||
"loss": 0.0014,
|
||||
"rewards/accuracies": 1.0,
|
||||
"rewards/chosen": 1.348962426185608,
|
||||
"rewards/margins": 7.596093654632568,
|
||||
"rewards/rejected": -6.248437404632568,
|
||||
"step": 250
|
||||
}
|
||||
],
|
||||
"logging_steps": 10,
|
||||
"max_steps": 257,
|
||||
"num_input_tokens_seen": 0,
|
||||
"num_train_epochs": 1,
|
||||
"save_steps": 500,
|
||||
"stateful_callbacks": {
|
||||
"TrainerControl": {
|
||||
"args": {
|
||||
"should_epoch_stop": false,
|
||||
"should_evaluate": false,
|
||||
"should_log": false,
|
||||
"should_save": true,
|
||||
"should_training_stop": true
|
||||
},
|
||||
"attributes": {}
|
||||
}
|
||||
},
|
||||
"total_flos": 0.0,
|
||||
"train_batch_size": 1,
|
||||
"trial_name": null,
|
||||
"trial_params": null
|
||||
}
|
||||
Reference in New Issue
Block a user