| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.9760765550239237, | |
| "eval_steps": 500, | |
| "global_step": 312, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.09569377990430622, | |
| "grad_norm": 4.315772756794048, | |
| "learning_rate": 3.125e-06, | |
| "loss": 0.4212, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.19138755980861244, | |
| "grad_norm": 3.3227666892135432, | |
| "learning_rate": 6.25e-06, | |
| "loss": 0.4736, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.28708133971291866, | |
| "grad_norm": 7.502806440154027, | |
| "learning_rate": 9.375000000000001e-06, | |
| "loss": 0.4117, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.3827751196172249, | |
| "grad_norm": 4.812933618699202, | |
| "learning_rate": 9.979871469976197e-06, | |
| "loss": 0.3306, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.4784688995215311, | |
| "grad_norm": 5.113380763093044, | |
| "learning_rate": 9.898376992116179e-06, | |
| "loss": 0.3924, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.5741626794258373, | |
| "grad_norm": 5.2850983406475915, | |
| "learning_rate": 9.755282581475769e-06, | |
| "loss": 0.4365, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.6698564593301436, | |
| "grad_norm": 5.364137037341764, | |
| "learning_rate": 9.552387733294081e-06, | |
| "loss": 0.3638, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.7655502392344498, | |
| "grad_norm": 4.0260153866397355, | |
| "learning_rate": 9.292243968009332e-06, | |
| "loss": 0.5465, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.861244019138756, | |
| "grad_norm": 3.8837148354409137, | |
| "learning_rate": 8.978122744408905e-06, | |
| "loss": 0.4152, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.9569377990430622, | |
| "grad_norm": 3.207603400930971, | |
| "learning_rate": 8.613974319136959e-06, | |
| "loss": 0.406, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.0478468899521531, | |
| "grad_norm": 3.081310895258368, | |
| "learning_rate": 8.204378069925121e-06, | |
| "loss": 0.273, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 1.1435406698564594, | |
| "grad_norm": 6.305626396570556, | |
| "learning_rate": 7.754484907260513e-06, | |
| "loss": 0.2318, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.2392344497607655, | |
| "grad_norm": 5.149450095126118, | |
| "learning_rate": 7.269952498697734e-06, | |
| "loss": 0.2031, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.3349282296650717, | |
| "grad_norm": 3.597480792122046, | |
| "learning_rate": 6.7568741204067145e-06, | |
| "loss": 0.1868, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.430622009569378, | |
| "grad_norm": 3.7719825769304873, | |
| "learning_rate": 6.2217020306894705e-06, | |
| "loss": 0.2089, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.526315789473684, | |
| "grad_norm": 6.079710603280231, | |
| "learning_rate": 5.671166329088278e-06, | |
| "loss": 0.2477, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.6220095693779903, | |
| "grad_norm": 2.5888279440746964, | |
| "learning_rate": 5.112190321479026e-06, | |
| "loss": 0.1991, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.7177033492822966, | |
| "grad_norm": 3.9779920431632423, | |
| "learning_rate": 4.551803455482833e-06, | |
| "loss": 0.1879, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.813397129186603, | |
| "grad_norm": 2.5621003966734115, | |
| "learning_rate": 3.997052921083637e-06, | |
| "loss": 0.1741, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.9090909090909092, | |
| "grad_norm": 4.103803645019158, | |
| "learning_rate": 3.4549150281252635e-06, | |
| "loss": 0.2304, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 10.76835936227807, | |
| "learning_rate": 2.932207475167398e-06, | |
| "loss": 0.1943, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 2.0956937799043063, | |
| "grad_norm": 3.1049956168395, | |
| "learning_rate": 2.43550361297047e-06, | |
| "loss": 0.0971, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 2.1913875598086126, | |
| "grad_norm": 1.8626051426196015, | |
| "learning_rate": 1.971049780795901e-06, | |
| "loss": 0.0747, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 2.287081339712919, | |
| "grad_norm": 2.2249948115867717, | |
| "learning_rate": 1.544686755065677e-06, | |
| "loss": 0.0828, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 2.382775119617225, | |
| "grad_norm": 3.645465893241635, | |
| "learning_rate": 1.1617762982099446e-06, | |
| "loss": 0.0859, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 2.478468899521531, | |
| "grad_norm": 4.11800452730375, | |
| "learning_rate": 8.271337313934869e-07, | |
| "loss": 0.0715, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 2.574162679425837, | |
| "grad_norm": 2.0043986949135855, | |
| "learning_rate": 5.449673790581611e-07, | |
| "loss": 0.095, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 2.6698564593301435, | |
| "grad_norm": 3.2194517540717276, | |
| "learning_rate": 3.18825646801314e-07, | |
| "loss": 0.1088, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 2.7655502392344498, | |
| "grad_norm": 5.7653256964411055, | |
| "learning_rate": 1.5155239811656562e-07, | |
| "loss": 0.0886, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 2.861244019138756, | |
| "grad_norm": 4.725414597201169, | |
| "learning_rate": 4.52511911603265e-08, | |
| "loss": 0.0909, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.9569377990430623, | |
| "grad_norm": 2.4916384390306168, | |
| "learning_rate": 1.2588252874673469e-09, | |
| "loss": 0.0912, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.9760765550239237, | |
| "step": 312, | |
| "total_flos": 1915811659776.0, | |
| "train_loss": 0.23802761278616694, | |
| "train_runtime": 1918.6836, | |
| "train_samples_per_second": 0.98, | |
| "train_steps_per_second": 0.163 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 312, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 50000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1915811659776.0, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |