| { |
| "best_metric": 0.3849843144416809, |
| "best_model_checkpoint": "autotrain-rj5xv-b9wsb/checkpoint-1584", |
| "epoch": 3.0, |
| "eval_steps": 500, |
| "global_step": 1584, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.04734848484848485, |
| "grad_norm": 4.511193752288818, |
| "learning_rate": 7.861635220125786e-06, |
| "loss": 2.8758, |
| "step": 25 |
| }, |
| { |
| "epoch": 0.0946969696969697, |
| "grad_norm": 3.8025200366973877, |
| "learning_rate": 1.572327044025157e-05, |
| "loss": 2.3043, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.14204545454545456, |
| "grad_norm": 2.95987868309021, |
| "learning_rate": 2.358490566037736e-05, |
| "loss": 1.5836, |
| "step": 75 |
| }, |
| { |
| "epoch": 0.1893939393939394, |
| "grad_norm": 3.758180618286133, |
| "learning_rate": 3.144654088050314e-05, |
| "loss": 1.4148, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.23674242424242425, |
| "grad_norm": 1.9288899898529053, |
| "learning_rate": 3.9308176100628936e-05, |
| "loss": 1.0447, |
| "step": 125 |
| }, |
| { |
| "epoch": 0.2840909090909091, |
| "grad_norm": 2.4479756355285645, |
| "learning_rate": 4.716981132075472e-05, |
| "loss": 0.7194, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.3314393939393939, |
| "grad_norm": 3.4983065128326416, |
| "learning_rate": 4.943859649122807e-05, |
| "loss": 0.8141, |
| "step": 175 |
| }, |
| { |
| "epoch": 0.3787878787878788, |
| "grad_norm": 2.1139252185821533, |
| "learning_rate": 4.856140350877193e-05, |
| "loss": 0.8187, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.42613636363636365, |
| "grad_norm": 2.1289165019989014, |
| "learning_rate": 4.7684210526315794e-05, |
| "loss": 0.8632, |
| "step": 225 |
| }, |
| { |
| "epoch": 0.4734848484848485, |
| "grad_norm": 3.188091516494751, |
| "learning_rate": 4.680701754385965e-05, |
| "loss": 0.7706, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.5208333333333334, |
| "grad_norm": 3.666327714920044, |
| "learning_rate": 4.592982456140351e-05, |
| "loss": 0.6436, |
| "step": 275 |
| }, |
| { |
| "epoch": 0.5681818181818182, |
| "grad_norm": 3.153630256652832, |
| "learning_rate": 4.5052631578947366e-05, |
| "loss": 0.5896, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.615530303030303, |
| "grad_norm": 1.9767557382583618, |
| "learning_rate": 4.417543859649123e-05, |
| "loss": 0.6594, |
| "step": 325 |
| }, |
| { |
| "epoch": 0.6628787878787878, |
| "grad_norm": 1.0973279476165771, |
| "learning_rate": 4.329824561403509e-05, |
| "loss": 0.7358, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.7102272727272727, |
| "grad_norm": 3.707737922668457, |
| "learning_rate": 4.242105263157895e-05, |
| "loss": 0.5802, |
| "step": 375 |
| }, |
| { |
| "epoch": 0.7575757575757576, |
| "grad_norm": 1.6052640676498413, |
| "learning_rate": 4.1543859649122806e-05, |
| "loss": 0.5345, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.8049242424242424, |
| "grad_norm": 2.5281245708465576, |
| "learning_rate": 4.066666666666667e-05, |
| "loss": 0.5426, |
| "step": 425 |
| }, |
| { |
| "epoch": 0.8522727272727273, |
| "grad_norm": 1.1469014883041382, |
| "learning_rate": 3.978947368421053e-05, |
| "loss": 0.4517, |
| "step": 450 |
| }, |
| { |
| "epoch": 0.8996212121212122, |
| "grad_norm": 2.1793298721313477, |
| "learning_rate": 3.891228070175439e-05, |
| "loss": 0.5795, |
| "step": 475 |
| }, |
| { |
| "epoch": 0.946969696969697, |
| "grad_norm": 1.2433311939239502, |
| "learning_rate": 3.8035087719298247e-05, |
| "loss": 0.4883, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.9943181818181818, |
| "grad_norm": 1.057558536529541, |
| "learning_rate": 3.715789473684211e-05, |
| "loss": 0.6399, |
| "step": 525 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_gen_len": 18.9849, |
| "eval_loss": 0.45811647176742554, |
| "eval_rouge1": 45.943, |
| "eval_rouge2": 41.8728, |
| "eval_rougeL": 45.3719, |
| "eval_rougeLsum": 45.5844, |
| "eval_runtime": 638.6799, |
| "eval_samples_per_second": 0.415, |
| "eval_steps_per_second": 0.105, |
| "step": 528 |
| }, |
| { |
| "epoch": 1.0416666666666667, |
| "grad_norm": 0.9785796999931335, |
| "learning_rate": 3.628070175438596e-05, |
| "loss": 0.4255, |
| "step": 550 |
| }, |
| { |
| "epoch": 1.0890151515151516, |
| "grad_norm": 2.2381393909454346, |
| "learning_rate": 3.5403508771929825e-05, |
| "loss": 0.4439, |
| "step": 575 |
| }, |
| { |
| "epoch": 1.1363636363636362, |
| "grad_norm": 2.753933906555176, |
| "learning_rate": 3.452631578947369e-05, |
| "loss": 0.5136, |
| "step": 600 |
| }, |
| { |
| "epoch": 1.183712121212121, |
| "grad_norm": 1.5527116060256958, |
| "learning_rate": 3.364912280701755e-05, |
| "loss": 0.3932, |
| "step": 625 |
| }, |
| { |
| "epoch": 1.231060606060606, |
| "grad_norm": 0.6465177536010742, |
| "learning_rate": 3.2771929824561403e-05, |
| "loss": 0.5194, |
| "step": 650 |
| }, |
| { |
| "epoch": 1.2784090909090908, |
| "grad_norm": 1.6484010219573975, |
| "learning_rate": 3.1894736842105265e-05, |
| "loss": 0.4773, |
| "step": 675 |
| }, |
| { |
| "epoch": 1.3257575757575757, |
| "grad_norm": 1.9873112440109253, |
| "learning_rate": 3.101754385964912e-05, |
| "loss": 0.4183, |
| "step": 700 |
| }, |
| { |
| "epoch": 1.3731060606060606, |
| "grad_norm": 2.486020803451538, |
| "learning_rate": 3.0140350877192985e-05, |
| "loss": 0.6344, |
| "step": 725 |
| }, |
| { |
| "epoch": 1.4204545454545454, |
| "grad_norm": 1.8892446756362915, |
| "learning_rate": 2.9263157894736844e-05, |
| "loss": 0.4328, |
| "step": 750 |
| }, |
| { |
| "epoch": 1.4678030303030303, |
| "grad_norm": 0.961203932762146, |
| "learning_rate": 2.8385964912280705e-05, |
| "loss": 0.5562, |
| "step": 775 |
| }, |
| { |
| "epoch": 1.5151515151515151, |
| "grad_norm": 0.44476789236068726, |
| "learning_rate": 2.750877192982456e-05, |
| "loss": 0.3306, |
| "step": 800 |
| }, |
| { |
| "epoch": 1.5625, |
| "grad_norm": 0.9097229838371277, |
| "learning_rate": 2.6631578947368426e-05, |
| "loss": 0.3695, |
| "step": 825 |
| }, |
| { |
| "epoch": 1.6098484848484849, |
| "grad_norm": 1.3498631715774536, |
| "learning_rate": 2.575438596491228e-05, |
| "loss": 0.3705, |
| "step": 850 |
| }, |
| { |
| "epoch": 1.6571969696969697, |
| "grad_norm": 0.6882970929145813, |
| "learning_rate": 2.4877192982456142e-05, |
| "loss": 0.5327, |
| "step": 875 |
| }, |
| { |
| "epoch": 1.7045454545454546, |
| "grad_norm": 0.7746265530586243, |
| "learning_rate": 2.4e-05, |
| "loss": 0.3751, |
| "step": 900 |
| }, |
| { |
| "epoch": 1.7518939393939394, |
| "grad_norm": 1.2026286125183105, |
| "learning_rate": 2.312280701754386e-05, |
| "loss": 0.2608, |
| "step": 925 |
| }, |
| { |
| "epoch": 1.7992424242424243, |
| "grad_norm": 5.0856099128723145, |
| "learning_rate": 2.224561403508772e-05, |
| "loss": 0.4432, |
| "step": 950 |
| }, |
| { |
| "epoch": 1.8465909090909092, |
| "grad_norm": 2.294633150100708, |
| "learning_rate": 2.136842105263158e-05, |
| "loss": 0.5151, |
| "step": 975 |
| }, |
| { |
| "epoch": 1.893939393939394, |
| "grad_norm": 1.039154052734375, |
| "learning_rate": 2.0491228070175437e-05, |
| "loss": 0.3376, |
| "step": 1000 |
| }, |
| { |
| "epoch": 1.941287878787879, |
| "grad_norm": 1.9636164903640747, |
| "learning_rate": 1.96140350877193e-05, |
| "loss": 0.4545, |
| "step": 1025 |
| }, |
| { |
| "epoch": 1.9886363636363638, |
| "grad_norm": 1.2051464319229126, |
| "learning_rate": 1.8736842105263158e-05, |
| "loss": 0.5154, |
| "step": 1050 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_gen_len": 18.9887, |
| "eval_loss": 0.3989172577857971, |
| "eval_rouge1": 46.097, |
| "eval_rouge2": 42.2086, |
| "eval_rougeL": 45.6885, |
| "eval_rougeLsum": 45.7885, |
| "eval_runtime": 648.8496, |
| "eval_samples_per_second": 0.408, |
| "eval_steps_per_second": 0.103, |
| "step": 1056 |
| }, |
| { |
| "epoch": 2.0359848484848486, |
| "grad_norm": 0.8088118433952332, |
| "learning_rate": 1.785964912280702e-05, |
| "loss": 0.4052, |
| "step": 1075 |
| }, |
| { |
| "epoch": 2.0833333333333335, |
| "grad_norm": 0.8511497974395752, |
| "learning_rate": 1.6982456140350878e-05, |
| "loss": 0.2922, |
| "step": 1100 |
| }, |
| { |
| "epoch": 2.1306818181818183, |
| "grad_norm": 1.41793954372406, |
| "learning_rate": 1.6105263157894736e-05, |
| "loss": 0.1893, |
| "step": 1125 |
| }, |
| { |
| "epoch": 2.178030303030303, |
| "grad_norm": 0.5501357316970825, |
| "learning_rate": 1.5228070175438596e-05, |
| "loss": 0.3815, |
| "step": 1150 |
| }, |
| { |
| "epoch": 2.225378787878788, |
| "grad_norm": 0.7588028311729431, |
| "learning_rate": 1.4350877192982456e-05, |
| "loss": 0.3866, |
| "step": 1175 |
| }, |
| { |
| "epoch": 2.2727272727272725, |
| "grad_norm": 1.9546937942504883, |
| "learning_rate": 1.3473684210526316e-05, |
| "loss": 0.3575, |
| "step": 1200 |
| }, |
| { |
| "epoch": 2.320075757575758, |
| "grad_norm": 0.6162445545196533, |
| "learning_rate": 1.2596491228070176e-05, |
| "loss": 0.5128, |
| "step": 1225 |
| }, |
| { |
| "epoch": 2.367424242424242, |
| "grad_norm": 2.0828588008880615, |
| "learning_rate": 1.1719298245614036e-05, |
| "loss": 0.4164, |
| "step": 1250 |
| }, |
| { |
| "epoch": 2.4147727272727275, |
| "grad_norm": 1.3566968441009521, |
| "learning_rate": 1.0842105263157895e-05, |
| "loss": 0.3453, |
| "step": 1275 |
| }, |
| { |
| "epoch": 2.462121212121212, |
| "grad_norm": 0.7142326235771179, |
| "learning_rate": 9.964912280701755e-06, |
| "loss": 0.4117, |
| "step": 1300 |
| }, |
| { |
| "epoch": 2.5094696969696972, |
| "grad_norm": 2.7209577560424805, |
| "learning_rate": 9.087719298245615e-06, |
| "loss": 0.3381, |
| "step": 1325 |
| }, |
| { |
| "epoch": 2.5568181818181817, |
| "grad_norm": 2.0031161308288574, |
| "learning_rate": 8.210526315789475e-06, |
| "loss": 0.2646, |
| "step": 1350 |
| }, |
| { |
| "epoch": 2.6041666666666665, |
| "grad_norm": 0.36383265256881714, |
| "learning_rate": 7.333333333333334e-06, |
| "loss": 0.3025, |
| "step": 1375 |
| }, |
| { |
| "epoch": 2.6515151515151514, |
| "grad_norm": 2.464855194091797, |
| "learning_rate": 6.456140350877193e-06, |
| "loss": 0.3774, |
| "step": 1400 |
| }, |
| { |
| "epoch": 2.6988636363636362, |
| "grad_norm": 2.341742515563965, |
| "learning_rate": 5.578947368421053e-06, |
| "loss": 0.3929, |
| "step": 1425 |
| }, |
| { |
| "epoch": 2.746212121212121, |
| "grad_norm": 0.6026067137718201, |
| "learning_rate": 4.7017543859649125e-06, |
| "loss": 0.4215, |
| "step": 1450 |
| }, |
| { |
| "epoch": 2.793560606060606, |
| "grad_norm": 1.8426591157913208, |
| "learning_rate": 3.8245614035087725e-06, |
| "loss": 0.2499, |
| "step": 1475 |
| }, |
| { |
| "epoch": 2.840909090909091, |
| "grad_norm": 1.0914125442504883, |
| "learning_rate": 2.9473684210526317e-06, |
| "loss": 0.3535, |
| "step": 1500 |
| }, |
| { |
| "epoch": 2.8882575757575757, |
| "grad_norm": 0.6658899784088135, |
| "learning_rate": 2.0701754385964913e-06, |
| "loss": 0.3272, |
| "step": 1525 |
| }, |
| { |
| "epoch": 2.9356060606060606, |
| "grad_norm": 0.7760673761367798, |
| "learning_rate": 1.192982456140351e-06, |
| "loss": 0.4175, |
| "step": 1550 |
| }, |
| { |
| "epoch": 2.9829545454545454, |
| "grad_norm": 1.258583426475525, |
| "learning_rate": 3.1578947368421055e-07, |
| "loss": 0.4278, |
| "step": 1575 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_gen_len": 18.9849, |
| "eval_loss": 0.3849843144416809, |
| "eval_rouge1": 46.2037, |
| "eval_rouge2": 42.3541, |
| "eval_rougeL": 45.8784, |
| "eval_rougeLsum": 45.9787, |
| "eval_runtime": 656.7129, |
| "eval_samples_per_second": 0.404, |
| "eval_steps_per_second": 0.102, |
| "step": 1584 |
| } |
| ], |
| "logging_steps": 25, |
| "max_steps": 1584, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 3, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "EarlyStoppingCallback": { |
| "args": { |
| "early_stopping_patience": 5, |
| "early_stopping_threshold": 0.01 |
| }, |
| "attributes": { |
| "early_stopping_patience_counter": 0 |
| } |
| }, |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": true |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 390532125671424.0, |
| "train_batch_size": 2, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
