lllqaq's picture
upload R2EGym 32B mid-post step155 final model (no ckpt)
d0fceb0 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 500,
"global_step": 540,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03710575139146568,
"grad_norm": 75.80557250976562,
"learning_rate": 1.6666666666666667e-06,
"loss": 3.1368,
"step": 10
},
{
"epoch": 0.07421150278293136,
"grad_norm": 7.823807716369629,
"learning_rate": 3.5185185185185187e-06,
"loss": 1.9929,
"step": 20
},
{
"epoch": 0.11131725417439703,
"grad_norm": 1.7641185522079468,
"learning_rate": 5.370370370370371e-06,
"loss": 0.901,
"step": 30
},
{
"epoch": 0.14842300556586271,
"grad_norm": 2.0621747970581055,
"learning_rate": 7.222222222222223e-06,
"loss": 0.5934,
"step": 40
},
{
"epoch": 0.18552875695732837,
"grad_norm": 0.9267826676368713,
"learning_rate": 9.074074074074075e-06,
"loss": 0.396,
"step": 50
},
{
"epoch": 0.22263450834879406,
"grad_norm": 0.8804956078529358,
"learning_rate": 9.997388623947927e-06,
"loss": 0.3619,
"step": 60
},
{
"epoch": 0.2597402597402597,
"grad_norm": 1.133553385734558,
"learning_rate": 9.976513978965829e-06,
"loss": 0.2826,
"step": 70
},
{
"epoch": 0.29684601113172543,
"grad_norm": 0.861992597579956,
"learning_rate": 9.934851884702415e-06,
"loss": 0.2703,
"step": 80
},
{
"epoch": 0.3339517625231911,
"grad_norm": 0.6837931275367737,
"learning_rate": 9.872576368333152e-06,
"loss": 0.2481,
"step": 90
},
{
"epoch": 0.37105751391465674,
"grad_norm": 0.9555836319923401,
"learning_rate": 9.789947561577445e-06,
"loss": 0.2355,
"step": 100
},
{
"epoch": 0.40816326530612246,
"grad_norm": 1.1039456129074097,
"learning_rate": 9.687310614099676e-06,
"loss": 0.2368,
"step": 110
},
{
"epoch": 0.4452690166975881,
"grad_norm": 0.8142388463020325,
"learning_rate": 9.565094251780872e-06,
"loss": 0.2282,
"step": 120
},
{
"epoch": 0.48237476808905383,
"grad_norm": 0.6265286803245544,
"learning_rate": 9.423808985883289e-06,
"loss": 0.2268,
"step": 130
},
{
"epoch": 0.5194805194805194,
"grad_norm": 0.7751391530036926,
"learning_rate": 9.264044980588415e-06,
"loss": 0.2231,
"step": 140
},
{
"epoch": 0.5565862708719852,
"grad_norm": 0.6618830561637878,
"learning_rate": 9.086469587815904e-06,
"loss": 0.2274,
"step": 150
},
{
"epoch": 0.5936920222634509,
"grad_norm": 1.1778677701950073,
"learning_rate": 8.891824559620801e-06,
"loss": 0.2214,
"step": 160
},
{
"epoch": 0.6307977736549165,
"grad_norm": 0.6525479555130005,
"learning_rate": 8.680922949813177e-06,
"loss": 0.2152,
"step": 170
},
{
"epoch": 0.6679035250463822,
"grad_norm": 1.056062936782837,
"learning_rate": 8.45464571774244e-06,
"loss": 0.2148,
"step": 180
},
{
"epoch": 0.7050092764378478,
"grad_norm": 0.506918728351593,
"learning_rate": 8.213938048432697e-06,
"loss": 0.2064,
"step": 190
},
{
"epoch": 0.7421150278293135,
"grad_norm": 0.6317462921142578,
"learning_rate": 7.95980540444038e-06,
"loss": 0.2136,
"step": 200
},
{
"epoch": 0.7792207792207793,
"grad_norm": 0.888347864151001,
"learning_rate": 7.69330932592594e-06,
"loss": 0.2214,
"step": 210
},
{
"epoch": 0.8163265306122449,
"grad_norm": 0.8190760612487793,
"learning_rate": 7.415562996483193e-06,
"loss": 0.2112,
"step": 220
},
{
"epoch": 0.8534322820037106,
"grad_norm": 0.6531184315681458,
"learning_rate": 7.127726593248337e-06,
"loss": 0.204,
"step": 230
},
{
"epoch": 0.8905380333951762,
"grad_norm": 0.6250878572463989,
"learning_rate": 6.8310024407117405e-06,
"loss": 0.2078,
"step": 240
},
{
"epoch": 0.9276437847866419,
"grad_norm": 0.590032160282135,
"learning_rate": 6.526629988475567e-06,
"loss": 0.2078,
"step": 250
},
{
"epoch": 0.9647495361781077,
"grad_norm": 0.8457366824150085,
"learning_rate": 6.215880633935709e-06,
"loss": 0.1985,
"step": 260
},
{
"epoch": 1.0,
"grad_norm": 0.721627950668335,
"learning_rate": 5.900052411514257e-06,
"loss": 0.195,
"step": 270
},
{
"epoch": 1.0371057513914657,
"grad_norm": 1.2528144121170044,
"learning_rate": 5.5804645706261515e-06,
"loss": 0.1468,
"step": 280
},
{
"epoch": 1.0742115027829313,
"grad_norm": 0.8960326910018921,
"learning_rate": 5.258452065028473e-06,
"loss": 0.1426,
"step": 290
},
{
"epoch": 1.111317254174397,
"grad_norm": 0.7007734775543213,
"learning_rate": 4.9353599765709585e-06,
"loss": 0.1366,
"step": 300
},
{
"epoch": 1.1484230055658626,
"grad_norm": 0.480663537979126,
"learning_rate": 4.6125378966403465e-06,
"loss": 0.1395,
"step": 310
},
{
"epoch": 1.1855287569573283,
"grad_norm": 0.47765257954597473,
"learning_rate": 4.291334288767899e-06,
"loss": 0.1372,
"step": 320
},
{
"epoch": 1.222634508348794,
"grad_norm": 1.1163313388824463,
"learning_rate": 3.9730908559480904e-06,
"loss": 0.1467,
"step": 330
},
{
"epoch": 1.2597402597402598,
"grad_norm": 0.9910896420478821,
"learning_rate": 3.6591369361968127e-06,
"loss": 0.1395,
"step": 340
},
{
"epoch": 1.2968460111317255,
"grad_norm": 0.852510929107666,
"learning_rate": 3.350783949759462e-06,
"loss": 0.1415,
"step": 350
},
{
"epoch": 1.3339517625231911,
"grad_norm": 0.5408625602722168,
"learning_rate": 3.049319921163526e-06,
"loss": 0.134,
"step": 360
},
{
"epoch": 1.3710575139146568,
"grad_norm": 0.5254811644554138,
"learning_rate": 2.7560040989976894e-06,
"loss": 0.1398,
"step": 370
},
{
"epoch": 1.4081632653061225,
"grad_norm": 0.8783926367759705,
"learning_rate": 2.4720616958912054e-06,
"loss": 0.1434,
"step": 380
},
{
"epoch": 1.4452690166975881,
"grad_norm": 1.284106969833374,
"learning_rate": 2.1986787706652377e-06,
"loss": 0.1377,
"step": 390
},
{
"epoch": 1.4823747680890538,
"grad_norm": 0.5269446969032288,
"learning_rate": 1.936997274033986e-06,
"loss": 0.1354,
"step": 400
},
{
"epoch": 1.5194805194805194,
"grad_norm": 0.5373296141624451,
"learning_rate": 1.6881102785502618e-06,
"loss": 0.1313,
"step": 410
},
{
"epoch": 1.556586270871985,
"grad_norm": 0.5133374333381653,
"learning_rate": 1.453057412720536e-06,
"loss": 0.1326,
"step": 420
},
{
"epoch": 1.593692022263451,
"grad_norm": 1.2796943187713623,
"learning_rate": 1.2328205183616964e-06,
"loss": 0.1403,
"step": 430
},
{
"epoch": 1.6307977736549164,
"grad_norm": 0.8333746790885925,
"learning_rate": 1.0283195493391823e-06,
"loss": 0.135,
"step": 440
},
{
"epoch": 1.6679035250463823,
"grad_norm": 0.5386627912521362,
"learning_rate": 8.404087288179425e-07,
"loss": 0.1317,
"step": 450
},
{
"epoch": 1.7050092764378477,
"grad_norm": 0.5091426968574524,
"learning_rate": 6.698729810778065e-07,
"loss": 0.1287,
"step": 460
},
{
"epoch": 1.7421150278293136,
"grad_norm": 0.4761359393596649,
"learning_rate": 5.174246527979532e-07,
"loss": 0.1312,
"step": 470
},
{
"epoch": 1.7792207792207793,
"grad_norm": 0.7398775219917297,
"learning_rate": 3.837005375060482e-07,
"loss": 0.128,
"step": 480
},
{
"epoch": 1.816326530612245,
"grad_norm": 0.614173948764801,
"learning_rate": 2.6925921562124867e-07,
"loss": 0.1305,
"step": 490
},
{
"epoch": 1.8534322820037106,
"grad_norm": 0.9018604159355164,
"learning_rate": 1.745787212020178e-07,
"loss": 0.136,
"step": 500
},
{
"epoch": 1.8905380333951762,
"grad_norm": 0.5306493639945984,
"learning_rate": 1.0005454514499413e-07,
"loss": 0.1221,
"step": 510
},
{
"epoch": 1.927643784786642,
"grad_norm": 0.5025076866149902,
"learning_rate": 4.599798317577342e-08,
"loss": 0.1307,
"step": 520
},
{
"epoch": 1.9647495361781075,
"grad_norm": 1.3446109294891357,
"learning_rate": 1.2634835532233658e-08,
"loss": 0.1358,
"step": 530
},
{
"epoch": 2.0,
"grad_norm": 1.5992753505706787,
"learning_rate": 1.0446377197104174e-10,
"loss": 0.1257,
"step": 540
},
{
"epoch": 2.0,
"step": 540,
"total_flos": 2.1175585841610752e+17,
"train_loss": 0.2914459398499242,
"train_runtime": 58643.1461,
"train_samples_per_second": 0.11,
"train_steps_per_second": 0.009
}
],
"logging_steps": 10,
"max_steps": 540,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 100000,
"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.1175585841610752e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}