| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.0, | |
| "eval_steps": 500, | |
| "global_step": 148, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.033783783783783786, | |
| "grad_norm": 4.190485489816713, | |
| "learning_rate": 3.125e-05, | |
| "loss": 1.4275, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.06756756756756757, | |
| "grad_norm": 2.6414426334400147, | |
| "learning_rate": 4.9977343997179584e-05, | |
| "loss": 1.1039, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.10135135135135136, | |
| "grad_norm": 1.686024999224537, | |
| "learning_rate": 4.97229876633906e-05, | |
| "loss": 0.9965, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.13513513513513514, | |
| "grad_norm": 1.7115424530922185, | |
| "learning_rate": 4.91891643656567e-05, | |
| "loss": 0.9521, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.16891891891891891, | |
| "grad_norm": 1.4990237498171883, | |
| "learning_rate": 4.838258724167838e-05, | |
| "loss": 0.9455, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.20270270270270271, | |
| "grad_norm": 1.6486883561613281, | |
| "learning_rate": 4.731339946677661e-05, | |
| "loss": 0.8812, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.23648648648648649, | |
| "grad_norm": 1.5741498095587194, | |
| "learning_rate": 4.599504669757798e-05, | |
| "loss": 0.871, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.2702702702702703, | |
| "grad_norm": 1.3455492691957438, | |
| "learning_rate": 4.444410798508125e-05, | |
| "loss": 0.8531, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.30405405405405406, | |
| "grad_norm": 1.259851496424771, | |
| "learning_rate": 4.268008728347168e-05, | |
| "loss": 0.823, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.33783783783783783, | |
| "grad_norm": 1.3658683302758547, | |
| "learning_rate": 4.072516817658065e-05, | |
| "loss": 0.8246, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.3716216216216216, | |
| "grad_norm": 1.291538016192788, | |
| "learning_rate": 3.860393490644781e-05, | |
| "loss": 0.8263, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.40540540540540543, | |
| "grad_norm": 1.1645381626538085, | |
| "learning_rate": 3.634306321221328e-05, | |
| "loss": 0.8048, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.4391891891891892, | |
| "grad_norm": 1.2798574384936259, | |
| "learning_rate": 3.397098486722039e-05, | |
| "loss": 0.8026, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.47297297297297297, | |
| "grad_norm": 1.2283284037526505, | |
| "learning_rate": 3.1517530132969326e-05, | |
| "loss": 0.7856, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.5067567567567568, | |
| "grad_norm": 1.3455606050143976, | |
| "learning_rate": 2.9013552626270894e-05, | |
| "loss": 0.8108, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.5405405405405406, | |
| "grad_norm": 1.1852287249106976, | |
| "learning_rate": 2.6490541317113427e-05, | |
| "loss": 0.7486, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.5743243243243243, | |
| "grad_norm": 1.1999962718351136, | |
| "learning_rate": 2.3980224536594803e-05, | |
| "loss": 0.7655, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.6081081081081081, | |
| "grad_norm": 1.2218896130828723, | |
| "learning_rate": 2.1514170974749814e-05, | |
| "loss": 0.7524, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.6418918918918919, | |
| "grad_norm": 1.163223199140077, | |
| "learning_rate": 1.9123392685956238e-05, | |
| "loss": 0.739, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.6756756756756757, | |
| "grad_norm": 1.1184385135414245, | |
| "learning_rate": 1.6837955094357533e-05, | |
| "loss": 0.7278, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.7094594594594594, | |
| "grad_norm": 1.1118477248764975, | |
| "learning_rate": 1.468659890370983e-05, | |
| "loss": 0.7373, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.7432432432432432, | |
| "grad_norm": 1.1604515193370903, | |
| "learning_rate": 1.2696378666356468e-05, | |
| "loss": 0.7237, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.777027027027027, | |
| "grad_norm": 1.1199470362056234, | |
| "learning_rate": 1.0892322556534839e-05, | |
| "loss": 0.7066, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.8108108108108109, | |
| "grad_norm": 1.078017791089081, | |
| "learning_rate": 9.297117626563687e-06, | |
| "loss": 0.7106, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.8445945945945946, | |
| "grad_norm": 1.1306273346210831, | |
| "learning_rate": 7.930824503996856e-06, | |
| "loss": 0.7274, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.8783783783783784, | |
| "grad_norm": 1.1083189976901686, | |
| "learning_rate": 6.810625117592363e-06, | |
| "loss": 0.7067, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.9121621621621622, | |
| "grad_norm": 1.0417923261832651, | |
| "learning_rate": 5.950606624589065e-06, | |
| "loss": 0.7093, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.9459459459459459, | |
| "grad_norm": 1.0363022521447358, | |
| "learning_rate": 5.361584256530833e-06, | |
| "loss": 0.6924, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.9797297297297297, | |
| "grad_norm": 1.0728825316546002, | |
| "learning_rate": 5.050965311454739e-06, | |
| "loss": 0.6906, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 148, | |
| "total_flos": 147125908537344.0, | |
| "train_loss": 0.8190691213350039, | |
| "train_runtime": 2476.8024, | |
| "train_samples_per_second": 3.819, | |
| "train_steps_per_second": 0.06 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 148, | |
| "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": 147125908537344.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |