| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 10.0, | |
| "eval_steps": 500, | |
| "global_step": 25070, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.1994415636218588, | |
| "grad_norm": 1.7483305931091309, | |
| "learning_rate": 4.8005584363781416e-05, | |
| "loss": 7.5177, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.3988831272437176, | |
| "grad_norm": 1.5902838706970215, | |
| "learning_rate": 4.601116872756282e-05, | |
| "loss": 6.6658, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.5983246908655764, | |
| "grad_norm": 1.9109225273132324, | |
| "learning_rate": 4.4016753091344236e-05, | |
| "loss": 6.3093, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.7977662544874352, | |
| "grad_norm": 1.8110615015029907, | |
| "learning_rate": 4.202233745512565e-05, | |
| "loss": 6.0429, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.9972078181092939, | |
| "grad_norm": 2.1440045833587646, | |
| "learning_rate": 4.002792181890706e-05, | |
| "loss": 5.8564, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.1966493817311528, | |
| "grad_norm": 2.1070995330810547, | |
| "learning_rate": 3.803350618268847e-05, | |
| "loss": 5.6429, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.3960909453530115, | |
| "grad_norm": 2.0321288108825684, | |
| "learning_rate": 3.6039090546469884e-05, | |
| "loss": 5.5179, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.5955325089748702, | |
| "grad_norm": 2.1192569732666016, | |
| "learning_rate": 3.4044674910251304e-05, | |
| "loss": 5.4088, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.7949740725967291, | |
| "grad_norm": 2.2034354209899902, | |
| "learning_rate": 3.205025927403271e-05, | |
| "loss": 5.3574, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 1.994415636218588, | |
| "grad_norm": 2.3862476348876953, | |
| "learning_rate": 3.0055843637814124e-05, | |
| "loss": 5.2802, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 2.193857199840447, | |
| "grad_norm": 2.287271738052368, | |
| "learning_rate": 2.8061428001595534e-05, | |
| "loss": 5.1049, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 2.3932987634623055, | |
| "grad_norm": 2.282344341278076, | |
| "learning_rate": 2.6067012365376948e-05, | |
| "loss": 5.0663, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 2.5927403270841642, | |
| "grad_norm": 2.3689157962799072, | |
| "learning_rate": 2.4072596729158358e-05, | |
| "loss": 5.0354, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 2.792181890706023, | |
| "grad_norm": 2.4090659618377686, | |
| "learning_rate": 2.207818109293977e-05, | |
| "loss": 4.9782, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 2.9916234543278817, | |
| "grad_norm": 2.3092195987701416, | |
| "learning_rate": 2.008376545672118e-05, | |
| "loss": 4.9434, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 3.191065017949741, | |
| "grad_norm": 2.460700273513794, | |
| "learning_rate": 1.8089349820502595e-05, | |
| "loss": 4.8539, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 3.3905065815715996, | |
| "grad_norm": 2.6232993602752686, | |
| "learning_rate": 1.6098923015556443e-05, | |
| "loss": 4.8056, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 3.5899481451934583, | |
| "grad_norm": 2.5114476680755615, | |
| "learning_rate": 1.4104507379337855e-05, | |
| "loss": 4.7642, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 3.789389708815317, | |
| "grad_norm": 2.446881055831909, | |
| "learning_rate": 1.2110091743119267e-05, | |
| "loss": 4.7694, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 3.988831272437176, | |
| "grad_norm": 2.586047410964966, | |
| "learning_rate": 1.0115676106900679e-05, | |
| "loss": 4.733, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 4.188272836059035, | |
| "grad_norm": 2.6265206336975098, | |
| "learning_rate": 8.125249301954529e-06, | |
| "loss": 4.6612, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 4.387714399680894, | |
| "grad_norm": 2.6361939907073975, | |
| "learning_rate": 6.13083366573594e-06, | |
| "loss": 4.6552, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 4.587155963302752, | |
| "grad_norm": 2.643498182296753, | |
| "learning_rate": 4.136418029517352e-06, | |
| "loss": 4.6594, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 4.786597526924611, | |
| "grad_norm": 2.601949453353882, | |
| "learning_rate": 2.1420023932987634e-06, | |
| "loss": 4.6318, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 4.98603909054647, | |
| "grad_norm": 2.711338996887207, | |
| "learning_rate": 1.515755883526127e-07, | |
| "loss": 4.6389, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 12535, | |
| "total_flos": 1.310011259092992e+16, | |
| "train_loss": 5.274392118328446, | |
| "train_runtime": 3806.2907, | |
| "train_samples_per_second": 52.686, | |
| "train_steps_per_second": 3.293 | |
| }, | |
| { | |
| "epoch": 5.1854806541683285, | |
| "grad_norm": 2.7731709480285645, | |
| "learning_rate": 4.907259672915836e-05, | |
| "loss": 4.7681, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 5.384922217790187, | |
| "grad_norm": 2.5628647804260254, | |
| "learning_rate": 4.8075388911049066e-05, | |
| "loss": 4.7441, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 5.584363781412046, | |
| "grad_norm": 2.7598373889923096, | |
| "learning_rate": 4.707818109293977e-05, | |
| "loss": 4.6634, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 5.783805345033905, | |
| "grad_norm": 2.619565486907959, | |
| "learning_rate": 4.608097327483047e-05, | |
| "loss": 4.6238, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 5.983246908655763, | |
| "grad_norm": 2.6309878826141357, | |
| "learning_rate": 4.508376545672118e-05, | |
| "loss": 4.6015, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 6.182688472277623, | |
| "grad_norm": 2.6872658729553223, | |
| "learning_rate": 4.4086557638611886e-05, | |
| "loss": 4.4218, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 6.382130035899482, | |
| "grad_norm": 2.780477285385132, | |
| "learning_rate": 4.3089349820502596e-05, | |
| "loss": 4.405, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 6.58157159952134, | |
| "grad_norm": 2.700143575668335, | |
| "learning_rate": 4.2092142002393306e-05, | |
| "loss": 4.3785, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 6.781013163143199, | |
| "grad_norm": 2.7577366828918457, | |
| "learning_rate": 4.1094934184284e-05, | |
| "loss": 4.3257, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 6.980454726765058, | |
| "grad_norm": 2.6972854137420654, | |
| "learning_rate": 4.009772636617471e-05, | |
| "loss": 4.3124, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 7.179896290386917, | |
| "grad_norm": 2.8049590587615967, | |
| "learning_rate": 3.9100518548065417e-05, | |
| "loss": 4.1901, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 7.379337854008775, | |
| "grad_norm": 3.0153963565826416, | |
| "learning_rate": 3.810331072995613e-05, | |
| "loss": 4.1466, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 7.578779417630634, | |
| "grad_norm": 3.010798215866089, | |
| "learning_rate": 3.710610291184683e-05, | |
| "loss": 4.1345, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 7.778220981252493, | |
| "grad_norm": 2.857867956161499, | |
| "learning_rate": 3.610889509373754e-05, | |
| "loss": 4.1093, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 7.9776625448743514, | |
| "grad_norm": 2.9866838455200195, | |
| "learning_rate": 3.5111687275628244e-05, | |
| "loss": 4.091, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 8.177104108496211, | |
| "grad_norm": 3.2831859588623047, | |
| "learning_rate": 3.411447945751895e-05, | |
| "loss": 3.9812, | |
| "step": 20500 | |
| }, | |
| { | |
| "epoch": 8.37654567211807, | |
| "grad_norm": 3.1285603046417236, | |
| "learning_rate": 3.311926605504587e-05, | |
| "loss": 3.955, | |
| "step": 21000 | |
| }, | |
| { | |
| "epoch": 8.575987235739928, | |
| "grad_norm": 2.9974467754364014, | |
| "learning_rate": 3.212205823693658e-05, | |
| "loss": 3.9307, | |
| "step": 21500 | |
| }, | |
| { | |
| "epoch": 8.775428799361787, | |
| "grad_norm": 3.0734570026397705, | |
| "learning_rate": 3.1124850418827286e-05, | |
| "loss": 3.9186, | |
| "step": 22000 | |
| }, | |
| { | |
| "epoch": 8.974870362983646, | |
| "grad_norm": 3.024569034576416, | |
| "learning_rate": 3.0129637016354212e-05, | |
| "loss": 3.9455, | |
| "step": 22500 | |
| }, | |
| { | |
| "epoch": 9.174311926605505, | |
| "grad_norm": 3.05592679977417, | |
| "learning_rate": 2.9132429198244916e-05, | |
| "loss": 3.8104, | |
| "step": 23000 | |
| }, | |
| { | |
| "epoch": 9.373753490227363, | |
| "grad_norm": 3.2957406044006348, | |
| "learning_rate": 2.8135221380135622e-05, | |
| "loss": 3.8007, | |
| "step": 23500 | |
| }, | |
| { | |
| "epoch": 9.573195053849222, | |
| "grad_norm": 3.398186683654785, | |
| "learning_rate": 2.7138013562026326e-05, | |
| "loss": 3.8345, | |
| "step": 24000 | |
| }, | |
| { | |
| "epoch": 9.77263661747108, | |
| "grad_norm": 3.1825666427612305, | |
| "learning_rate": 2.6140805743917036e-05, | |
| "loss": 3.7761, | |
| "step": 24500 | |
| }, | |
| { | |
| "epoch": 9.97207818109294, | |
| "grad_norm": 2.9551422595977783, | |
| "learning_rate": 2.514359792580774e-05, | |
| "loss": 3.777, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 25070, | |
| "total_flos": 2.620022518185984e+16, | |
| "train_loss": 2.0909440845904905, | |
| "train_runtime": 3914.0913, | |
| "train_samples_per_second": 102.471, | |
| "train_steps_per_second": 6.405 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 25070, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 10, | |
| "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": 2.620022518185984e+16, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |