| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 144, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.020833333333333332, | |
| "grad_norm": 5.9375, | |
| "learning_rate": 0.0, | |
| "loss": 1.457, | |
| "memory/device_reserved (GiB)": 68.17, | |
| "memory/max_active (GiB)": 62.24, | |
| "memory/max_allocated (GiB)": 62.24, | |
| "step": 1, | |
| "tokens_per_second_per_gpu": 4754.85, | |
| "total_tokens": 193735 | |
| }, | |
| { | |
| "epoch": 0.041666666666666664, | |
| "grad_norm": 6.40625, | |
| "learning_rate": 1.4285714285714286e-06, | |
| "loss": 1.5117, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 2, | |
| "tokens_per_second_per_gpu": 8987.92, | |
| "total_tokens": 377645 | |
| }, | |
| { | |
| "epoch": 0.0625, | |
| "grad_norm": 6.34375, | |
| "learning_rate": 2.8571428571428573e-06, | |
| "loss": 1.5352, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 3, | |
| "tokens_per_second_per_gpu": 10584.88, | |
| "total_tokens": 572231 | |
| }, | |
| { | |
| "epoch": 0.08333333333333333, | |
| "grad_norm": 5.75, | |
| "learning_rate": 4.2857142857142855e-06, | |
| "loss": 1.4023, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 4, | |
| "tokens_per_second_per_gpu": 10641.72, | |
| "total_tokens": 771980 | |
| }, | |
| { | |
| "epoch": 0.10416666666666667, | |
| "grad_norm": 5.8125, | |
| "learning_rate": 5.7142857142857145e-06, | |
| "loss": 1.5234, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 5, | |
| "tokens_per_second_per_gpu": 10136.19, | |
| "total_tokens": 960629 | |
| }, | |
| { | |
| "epoch": 0.125, | |
| "grad_norm": 5.5, | |
| "learning_rate": 7.1428571428571436e-06, | |
| "loss": 1.4961, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 6, | |
| "tokens_per_second_per_gpu": 10066.87, | |
| "total_tokens": 1145765 | |
| }, | |
| { | |
| "epoch": 0.14583333333333334, | |
| "grad_norm": 4.59375, | |
| "learning_rate": 8.571428571428571e-06, | |
| "loss": 1.3828, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 7, | |
| "tokens_per_second_per_gpu": 10793.17, | |
| "total_tokens": 1339007 | |
| }, | |
| { | |
| "epoch": 0.16666666666666666, | |
| "grad_norm": 3.53125, | |
| "learning_rate": 1e-05, | |
| "loss": 1.3359, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 8, | |
| "tokens_per_second_per_gpu": 10042.61, | |
| "total_tokens": 1533294 | |
| }, | |
| { | |
| "epoch": 0.1875, | |
| "grad_norm": 2.625, | |
| "learning_rate": 1.1428571428571429e-05, | |
| "loss": 1.3203, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 9, | |
| "tokens_per_second_per_gpu": 9880.69, | |
| "total_tokens": 1715421 | |
| }, | |
| { | |
| "epoch": 0.20833333333333334, | |
| "grad_norm": 1.859375, | |
| "learning_rate": 1.2857142857142859e-05, | |
| "loss": 1.2695, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 10, | |
| "tokens_per_second_per_gpu": 10012.45, | |
| "total_tokens": 1899665 | |
| }, | |
| { | |
| "epoch": 0.22916666666666666, | |
| "grad_norm": 1.359375, | |
| "learning_rate": 1.4285714285714287e-05, | |
| "loss": 1.2539, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 11, | |
| "tokens_per_second_per_gpu": 10498.2, | |
| "total_tokens": 2090232 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.359375, | |
| "learning_rate": 1.5714285714285715e-05, | |
| "loss": 1.2383, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 12, | |
| "tokens_per_second_per_gpu": 10504.24, | |
| "total_tokens": 2277678 | |
| }, | |
| { | |
| "epoch": 0.2708333333333333, | |
| "grad_norm": 1.3828125, | |
| "learning_rate": 1.7142857142857142e-05, | |
| "loss": 1.2031, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 13, | |
| "tokens_per_second_per_gpu": 11150.71, | |
| "total_tokens": 2475912 | |
| }, | |
| { | |
| "epoch": 0.2916666666666667, | |
| "grad_norm": 1.3125, | |
| "learning_rate": 1.8571428571428575e-05, | |
| "loss": 1.1172, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 14, | |
| "tokens_per_second_per_gpu": 10549.83, | |
| "total_tokens": 2677381 | |
| }, | |
| { | |
| "epoch": 0.3125, | |
| "grad_norm": 1.3671875, | |
| "learning_rate": 2e-05, | |
| "loss": 1.1602, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 15, | |
| "tokens_per_second_per_gpu": 10901.98, | |
| "total_tokens": 2876565 | |
| }, | |
| { | |
| "epoch": 0.3333333333333333, | |
| "grad_norm": 1.203125, | |
| "learning_rate": 1.9997080140801932e-05, | |
| "loss": 1.1445, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 16, | |
| "tokens_per_second_per_gpu": 10121.73, | |
| "total_tokens": 3065392 | |
| }, | |
| { | |
| "epoch": 0.3541666666666667, | |
| "grad_norm": 1.109375, | |
| "learning_rate": 1.998832226832327e-05, | |
| "loss": 1.1016, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 17, | |
| "tokens_per_second_per_gpu": 10594.37, | |
| "total_tokens": 3260102 | |
| }, | |
| { | |
| "epoch": 0.375, | |
| "grad_norm": 0.93359375, | |
| "learning_rate": 1.9973731496914914e-05, | |
| "loss": 1.168, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 18, | |
| "tokens_per_second_per_gpu": 9736.14, | |
| "total_tokens": 3442701 | |
| }, | |
| { | |
| "epoch": 0.3958333333333333, | |
| "grad_norm": 0.79296875, | |
| "learning_rate": 1.995331634717649e-05, | |
| "loss": 1.082, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 19, | |
| "tokens_per_second_per_gpu": 10128.02, | |
| "total_tokens": 3638349 | |
| }, | |
| { | |
| "epoch": 0.4166666666666667, | |
| "grad_norm": 0.75, | |
| "learning_rate": 1.992708874098054e-05, | |
| "loss": 1.1055, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 20, | |
| "tokens_per_second_per_gpu": 10243.56, | |
| "total_tokens": 3826742 | |
| }, | |
| { | |
| "epoch": 0.4375, | |
| "grad_norm": 0.69140625, | |
| "learning_rate": 1.9895063994510512e-05, | |
| "loss": 1.1016, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 21, | |
| "tokens_per_second_per_gpu": 9967.12, | |
| "total_tokens": 4014944 | |
| }, | |
| { | |
| "epoch": 0.4583333333333333, | |
| "grad_norm": 0.6796875, | |
| "learning_rate": 1.985726080931651e-05, | |
| "loss": 1.1211, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 22, | |
| "tokens_per_second_per_gpu": 10518.66, | |
| "total_tokens": 4208308 | |
| }, | |
| { | |
| "epoch": 0.4791666666666667, | |
| "grad_norm": 0.67578125, | |
| "learning_rate": 1.9813701261394136e-05, | |
| "loss": 1.1523, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 23, | |
| "tokens_per_second_per_gpu": 10033.96, | |
| "total_tokens": 4394123 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "grad_norm": 0.64453125, | |
| "learning_rate": 1.9764410788292724e-05, | |
| "loss": 1.0898, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 24, | |
| "tokens_per_second_per_gpu": 10669.49, | |
| "total_tokens": 4588895 | |
| }, | |
| { | |
| "epoch": 0.5208333333333334, | |
| "grad_norm": 0.62109375, | |
| "learning_rate": 1.9709418174260523e-05, | |
| "loss": 1.1172, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 25, | |
| "tokens_per_second_per_gpu": 10276.83, | |
| "total_tokens": 4781600 | |
| }, | |
| { | |
| "epoch": 0.5416666666666666, | |
| "grad_norm": 0.6328125, | |
| "learning_rate": 1.9648755533435517e-05, | |
| "loss": 1.1094, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 26, | |
| "tokens_per_second_per_gpu": 9934.02, | |
| "total_tokens": 4962927 | |
| }, | |
| { | |
| "epoch": 0.5625, | |
| "grad_norm": 0.546875, | |
| "learning_rate": 1.9582458291091664e-05, | |
| "loss": 1.0898, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 27, | |
| "tokens_per_second_per_gpu": 10213.04, | |
| "total_tokens": 5154156 | |
| }, | |
| { | |
| "epoch": 0.5833333333333334, | |
| "grad_norm": 0.5390625, | |
| "learning_rate": 1.9510565162951538e-05, | |
| "loss": 1.0625, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 28, | |
| "tokens_per_second_per_gpu": 10650.84, | |
| "total_tokens": 5346313 | |
| }, | |
| { | |
| "epoch": 0.6041666666666666, | |
| "grad_norm": 0.52734375, | |
| "learning_rate": 1.9433118132577432e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 29, | |
| "tokens_per_second_per_gpu": 10206.96, | |
| "total_tokens": 5537564 | |
| }, | |
| { | |
| "epoch": 0.625, | |
| "grad_norm": 0.52734375, | |
| "learning_rate": 1.9350162426854152e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 30, | |
| "tokens_per_second_per_gpu": 9831.26, | |
| "total_tokens": 5720388 | |
| }, | |
| { | |
| "epoch": 0.6458333333333334, | |
| "grad_norm": 0.54296875, | |
| "learning_rate": 1.9261746489577767e-05, | |
| "loss": 1.0781, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 31, | |
| "tokens_per_second_per_gpu": 10701.1, | |
| "total_tokens": 5906962 | |
| }, | |
| { | |
| "epoch": 0.6666666666666666, | |
| "grad_norm": 0.52734375, | |
| "learning_rate": 1.9167921953165827e-05, | |
| "loss": 1.0938, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 32, | |
| "tokens_per_second_per_gpu": 10732.53, | |
| "total_tokens": 6104218 | |
| }, | |
| { | |
| "epoch": 0.6875, | |
| "grad_norm": 0.52734375, | |
| "learning_rate": 1.9068743608505454e-05, | |
| "loss": 1.1133, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 33, | |
| "tokens_per_second_per_gpu": 10156.96, | |
| "total_tokens": 6291571 | |
| }, | |
| { | |
| "epoch": 0.7083333333333334, | |
| "grad_norm": 0.5, | |
| "learning_rate": 1.896426937295704e-05, | |
| "loss": 1.0781, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 34, | |
| "tokens_per_second_per_gpu": 10651.58, | |
| "total_tokens": 6476350 | |
| }, | |
| { | |
| "epoch": 0.7291666666666666, | |
| "grad_norm": 0.5390625, | |
| "learning_rate": 1.8854560256532098e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 35, | |
| "tokens_per_second_per_gpu": 10027.99, | |
| "total_tokens": 6660433 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 0.49609375, | |
| "learning_rate": 1.873968032626518e-05, | |
| "loss": 1.0625, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 36, | |
| "tokens_per_second_per_gpu": 10538.25, | |
| "total_tokens": 6853661 | |
| }, | |
| { | |
| "epoch": 0.7708333333333334, | |
| "grad_norm": 0.470703125, | |
| "learning_rate": 1.8619696668800494e-05, | |
| "loss": 1.1055, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 37, | |
| "tokens_per_second_per_gpu": 10499.61, | |
| "total_tokens": 7041887 | |
| }, | |
| { | |
| "epoch": 0.7916666666666666, | |
| "grad_norm": 0.4609375, | |
| "learning_rate": 1.8494679351215212e-05, | |
| "loss": 1.0117, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 38, | |
| "tokens_per_second_per_gpu": 10892.76, | |
| "total_tokens": 7242129 | |
| }, | |
| { | |
| "epoch": 0.8125, | |
| "grad_norm": 0.4765625, | |
| "learning_rate": 1.8364701380102267e-05, | |
| "loss": 1.0391, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 39, | |
| "tokens_per_second_per_gpu": 10826.51, | |
| "total_tokens": 7442101 | |
| }, | |
| { | |
| "epoch": 0.8333333333333334, | |
| "grad_norm": 0.494140625, | |
| "learning_rate": 1.8229838658936566e-05, | |
| "loss": 0.998, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 40, | |
| "tokens_per_second_per_gpu": 9529.96, | |
| "total_tokens": 7637450 | |
| }, | |
| { | |
| "epoch": 0.8541666666666666, | |
| "grad_norm": 0.458984375, | |
| "learning_rate": 1.8090169943749477e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 41, | |
| "tokens_per_second_per_gpu": 10530.91, | |
| "total_tokens": 7830567 | |
| }, | |
| { | |
| "epoch": 0.875, | |
| "grad_norm": 0.470703125, | |
| "learning_rate": 1.7945776797137544e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 42, | |
| "tokens_per_second_per_gpu": 10621.21, | |
| "total_tokens": 8025656 | |
| }, | |
| { | |
| "epoch": 0.8958333333333334, | |
| "grad_norm": 0.466796875, | |
| "learning_rate": 1.7796743540632226e-05, | |
| "loss": 1.0254, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 43, | |
| "tokens_per_second_per_gpu": 10096.46, | |
| "total_tokens": 8212765 | |
| }, | |
| { | |
| "epoch": 0.9166666666666666, | |
| "grad_norm": 0.447265625, | |
| "learning_rate": 1.7643157205458483e-05, | |
| "loss": 1.0332, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 44, | |
| "tokens_per_second_per_gpu": 10109.75, | |
| "total_tokens": 8408896 | |
| }, | |
| { | |
| "epoch": 0.9375, | |
| "grad_norm": 0.439453125, | |
| "learning_rate": 1.7485107481711014e-05, | |
| "loss": 1.002, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 45, | |
| "tokens_per_second_per_gpu": 10350.81, | |
| "total_tokens": 8608276 | |
| }, | |
| { | |
| "epoch": 0.9583333333333334, | |
| "grad_norm": 0.53125, | |
| "learning_rate": 1.7322686665977738e-05, | |
| "loss": 1.1094, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 46, | |
| "tokens_per_second_per_gpu": 10131.42, | |
| "total_tokens": 8799852 | |
| }, | |
| { | |
| "epoch": 0.9791666666666666, | |
| "grad_norm": 0.435546875, | |
| "learning_rate": 1.715598960744121e-05, | |
| "loss": 0.9707, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 47, | |
| "tokens_per_second_per_gpu": 9848.57, | |
| "total_tokens": 8991281 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "grad_norm": 0.466796875, | |
| "learning_rate": 1.6985113652489374e-05, | |
| "loss": 1.0391, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 48, | |
| "tokens_per_second_per_gpu": 10189.26, | |
| "total_tokens": 9182092 | |
| }, | |
| { | |
| "epoch": 1.0208333333333333, | |
| "grad_norm": 0.44921875, | |
| "learning_rate": 1.6810158587867973e-05, | |
| "loss": 1.0117, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 49, | |
| "tokens_per_second_per_gpu": 10109.46, | |
| "total_tokens": 9375827 | |
| }, | |
| { | |
| "epoch": 1.0416666666666667, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.6631226582407954e-05, | |
| "loss": 1.0312, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 50, | |
| "tokens_per_second_per_gpu": 9167.36, | |
| "total_tokens": 9559737 | |
| }, | |
| { | |
| "epoch": 1.0625, | |
| "grad_norm": 0.458984375, | |
| "learning_rate": 1.6448422127361707e-05, | |
| "loss": 1.0703, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 51, | |
| "tokens_per_second_per_gpu": 10531.34, | |
| "total_tokens": 9754323 | |
| }, | |
| { | |
| "epoch": 1.0833333333333333, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 1.626185197538314e-05, | |
| "loss": 0.9688, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 52, | |
| "tokens_per_second_per_gpu": 10570.62, | |
| "total_tokens": 9954072 | |
| }, | |
| { | |
| "epoch": 1.1041666666666667, | |
| "grad_norm": 0.4609375, | |
| "learning_rate": 1.6071625078187113e-05, | |
| "loss": 1.0859, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 53, | |
| "tokens_per_second_per_gpu": 10132.93, | |
| "total_tokens": 10142721 | |
| }, | |
| { | |
| "epoch": 1.125, | |
| "grad_norm": 0.4609375, | |
| "learning_rate": 1.5877852522924733e-05, | |
| "loss": 1.0703, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 54, | |
| "tokens_per_second_per_gpu": 10263.22, | |
| "total_tokens": 10327857 | |
| }, | |
| { | |
| "epoch": 1.1458333333333333, | |
| "grad_norm": 0.44921875, | |
| "learning_rate": 1.568064746731156e-05, | |
| "loss": 1.0176, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 55, | |
| "tokens_per_second_per_gpu": 10786.39, | |
| "total_tokens": 10521099 | |
| }, | |
| { | |
| "epoch": 1.1666666666666667, | |
| "grad_norm": 0.443359375, | |
| "learning_rate": 1.5480125073546705e-05, | |
| "loss": 1.0352, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 56, | |
| "tokens_per_second_per_gpu": 10023.87, | |
| "total_tokens": 10715386 | |
| }, | |
| { | |
| "epoch": 1.1875, | |
| "grad_norm": 0.45703125, | |
| "learning_rate": 1.527640244106133e-05, | |
| "loss": 1.043, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 57, | |
| "tokens_per_second_per_gpu": 9848.52, | |
| "total_tokens": 10897513 | |
| }, | |
| { | |
| "epoch": 1.2083333333333333, | |
| "grad_norm": 0.482421875, | |
| "learning_rate": 1.5069598538135905e-05, | |
| "loss": 1.0352, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 58, | |
| "tokens_per_second_per_gpu": 10004.24, | |
| "total_tokens": 11081757 | |
| }, | |
| { | |
| "epoch": 1.2291666666666667, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.485983413242606e-05, | |
| "loss": 1.0195, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 59, | |
| "tokens_per_second_per_gpu": 10483.32, | |
| "total_tokens": 11272324 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 0.4609375, | |
| "learning_rate": 1.4647231720437687e-05, | |
| "loss": 1.0293, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 60, | |
| "tokens_per_second_per_gpu": 10478.12, | |
| "total_tokens": 11459770 | |
| }, | |
| { | |
| "epoch": 1.2708333333333333, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.4431915455992416e-05, | |
| "loss": 1.0039, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 61, | |
| "tokens_per_second_per_gpu": 11157.45, | |
| "total_tokens": 11658004 | |
| }, | |
| { | |
| "epoch": 1.2916666666666667, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 1.4214011077725293e-05, | |
| "loss": 0.9414, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 62, | |
| "tokens_per_second_per_gpu": 10464.97, | |
| "total_tokens": 11859473 | |
| }, | |
| { | |
| "epoch": 1.3125, | |
| "grad_norm": 0.45703125, | |
| "learning_rate": 1.3993645835656955e-05, | |
| "loss": 0.9844, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 63, | |
| "tokens_per_second_per_gpu": 10802.11, | |
| "total_tokens": 12058657 | |
| }, | |
| { | |
| "epoch": 1.3333333333333333, | |
| "grad_norm": 0.435546875, | |
| "learning_rate": 1.3770948416883205e-05, | |
| "loss": 0.9746, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 64, | |
| "tokens_per_second_per_gpu": 10169.34, | |
| "total_tokens": 12247484 | |
| }, | |
| { | |
| "epoch": 1.3541666666666667, | |
| "grad_norm": 0.470703125, | |
| "learning_rate": 1.3546048870425356e-05, | |
| "loss": 0.9434, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 65, | |
| "tokens_per_second_per_gpu": 10566.69, | |
| "total_tokens": 12442194 | |
| }, | |
| { | |
| "epoch": 1.375, | |
| "grad_norm": 0.4453125, | |
| "learning_rate": 1.3319078531285286e-05, | |
| "loss": 1.002, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 66, | |
| "tokens_per_second_per_gpu": 9699.95, | |
| "total_tokens": 12624793 | |
| }, | |
| { | |
| "epoch": 1.3958333333333333, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 1.3090169943749475e-05, | |
| "loss": 0.9336, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 67, | |
| "tokens_per_second_per_gpu": 10187.26, | |
| "total_tokens": 12820441 | |
| }, | |
| { | |
| "epoch": 1.4166666666666667, | |
| "grad_norm": 0.458984375, | |
| "learning_rate": 1.2859456783986892e-05, | |
| "loss": 0.9629, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 68, | |
| "tokens_per_second_per_gpu": 10254.48, | |
| "total_tokens": 13008834 | |
| }, | |
| { | |
| "epoch": 1.4375, | |
| "grad_norm": 0.447265625, | |
| "learning_rate": 1.262707378198587e-05, | |
| "loss": 0.9648, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 69, | |
| "tokens_per_second_per_gpu": 9940.27, | |
| "total_tokens": 13197036 | |
| }, | |
| { | |
| "epoch": 1.4583333333333333, | |
| "grad_norm": 0.4609375, | |
| "learning_rate": 1.2393156642875579e-05, | |
| "loss": 0.9902, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 70, | |
| "tokens_per_second_per_gpu": 10499.3, | |
| "total_tokens": 13390400 | |
| }, | |
| { | |
| "epoch": 1.4791666666666667, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.2157841967678064e-05, | |
| "loss": 1.0215, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 71, | |
| "tokens_per_second_per_gpu": 10092.49, | |
| "total_tokens": 13576215 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "grad_norm": 0.427734375, | |
| "learning_rate": 1.1921267173537085e-05, | |
| "loss": 0.9707, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 72, | |
| "tokens_per_second_per_gpu": 10734.78, | |
| "total_tokens": 13770987 | |
| }, | |
| { | |
| "epoch": 1.5208333333333335, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.1683570413470384e-05, | |
| "loss": 1.0039, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 73, | |
| "tokens_per_second_per_gpu": 10374.13, | |
| "total_tokens": 13963692 | |
| }, | |
| { | |
| "epoch": 1.5416666666666665, | |
| "grad_norm": 0.451171875, | |
| "learning_rate": 1.1444890495692214e-05, | |
| "loss": 0.9863, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 74, | |
| "tokens_per_second_per_gpu": 9925.41, | |
| "total_tokens": 14145019 | |
| }, | |
| { | |
| "epoch": 1.5625, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 1.1205366802553231e-05, | |
| "loss": 0.9844, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 75, | |
| "tokens_per_second_per_gpu": 10193.32, | |
| "total_tokens": 14336248 | |
| }, | |
| { | |
| "epoch": 1.5833333333333335, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 1.0965139209145153e-05, | |
| "loss": 0.957, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 76, | |
| "tokens_per_second_per_gpu": 10648.45, | |
| "total_tokens": 14528405 | |
| }, | |
| { | |
| "epoch": 1.6041666666666665, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 1.0724348001617626e-05, | |
| "loss": 0.9902, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 77, | |
| "tokens_per_second_per_gpu": 10212.71, | |
| "total_tokens": 14719656 | |
| }, | |
| { | |
| "epoch": 1.625, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 1.0483133795255072e-05, | |
| "loss": 0.9941, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 78, | |
| "tokens_per_second_per_gpu": 9784.17, | |
| "total_tokens": 14902480 | |
| }, | |
| { | |
| "epoch": 1.6458333333333335, | |
| "grad_norm": 0.453125, | |
| "learning_rate": 1.0241637452361323e-05, | |
| "loss": 0.9883, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 79, | |
| "tokens_per_second_per_gpu": 10663.55, | |
| "total_tokens": 15089054 | |
| }, | |
| { | |
| "epoch": 1.6666666666666665, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 1e-05, | |
| "loss": 1.0078, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 80, | |
| "tokens_per_second_per_gpu": 10776.49, | |
| "total_tokens": 15286310 | |
| }, | |
| { | |
| "epoch": 1.6875, | |
| "grad_norm": 0.4453125, | |
| "learning_rate": 9.75836254763868e-06, | |
| "loss": 1.0234, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 81, | |
| "tokens_per_second_per_gpu": 10157.52, | |
| "total_tokens": 15473663 | |
| }, | |
| { | |
| "epoch": 1.7083333333333335, | |
| "grad_norm": 0.470703125, | |
| "learning_rate": 9.516866204744932e-06, | |
| "loss": 1.0, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 82, | |
| "tokens_per_second_per_gpu": 10701.67, | |
| "total_tokens": 15658442 | |
| }, | |
| { | |
| "epoch": 1.7291666666666665, | |
| "grad_norm": 0.466796875, | |
| "learning_rate": 9.275651998382377e-06, | |
| "loss": 1.0039, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 83, | |
| "tokens_per_second_per_gpu": 9974.39, | |
| "total_tokens": 15842525 | |
| }, | |
| { | |
| "epoch": 1.75, | |
| "grad_norm": 0.455078125, | |
| "learning_rate": 9.034860790854848e-06, | |
| "loss": 0.9863, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 84, | |
| "tokens_per_second_per_gpu": 10560.42, | |
| "total_tokens": 16035753 | |
| }, | |
| { | |
| "epoch": 1.7708333333333335, | |
| "grad_norm": 0.455078125, | |
| "learning_rate": 8.79463319744677e-06, | |
| "loss": 1.0312, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 85, | |
| "tokens_per_second_per_gpu": 10498.04, | |
| "total_tokens": 16223979 | |
| }, | |
| { | |
| "epoch": 1.7916666666666665, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 8.55510950430779e-06, | |
| "loss": 0.9492, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 86, | |
| "tokens_per_second_per_gpu": 10922.51, | |
| "total_tokens": 16424221 | |
| }, | |
| { | |
| "epoch": 1.8125, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 8.316429586529616e-06, | |
| "loss": 0.9746, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 87, | |
| "tokens_per_second_per_gpu": 10848.98, | |
| "total_tokens": 16624193 | |
| }, | |
| { | |
| "epoch": 1.8333333333333335, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 8.078732826462917e-06, | |
| "loss": 0.9355, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 88, | |
| "tokens_per_second_per_gpu": 9565.65, | |
| "total_tokens": 16819542 | |
| }, | |
| { | |
| "epoch": 1.8541666666666665, | |
| "grad_norm": 0.43359375, | |
| "learning_rate": 7.84215803232194e-06, | |
| "loss": 1.0234, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 89, | |
| "tokens_per_second_per_gpu": 10503.54, | |
| "total_tokens": 17012659 | |
| }, | |
| { | |
| "epoch": 1.875, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 7.606843357124426e-06, | |
| "loss": 1.0215, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 90, | |
| "tokens_per_second_per_gpu": 10559.42, | |
| "total_tokens": 17207748 | |
| }, | |
| { | |
| "epoch": 1.8958333333333335, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 7.372926218014131e-06, | |
| "loss": 0.9668, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 91, | |
| "tokens_per_second_per_gpu": 10102.48, | |
| "total_tokens": 17394857 | |
| }, | |
| { | |
| "epoch": 1.9166666666666665, | |
| "grad_norm": 0.4453125, | |
| "learning_rate": 7.140543216013109e-06, | |
| "loss": 0.9785, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 92, | |
| "tokens_per_second_per_gpu": 10165.22, | |
| "total_tokens": 17590988 | |
| }, | |
| { | |
| "epoch": 1.9375, | |
| "grad_norm": 0.423828125, | |
| "learning_rate": 6.909830056250527e-06, | |
| "loss": 0.9492, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 93, | |
| "tokens_per_second_per_gpu": 10355.66, | |
| "total_tokens": 17790368 | |
| }, | |
| { | |
| "epoch": 1.9583333333333335, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 6.680921468714718e-06, | |
| "loss": 1.0508, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 94, | |
| "tokens_per_second_per_gpu": 10134.55, | |
| "total_tokens": 17981944 | |
| }, | |
| { | |
| "epoch": 1.9791666666666665, | |
| "grad_norm": 0.40625, | |
| "learning_rate": 6.453951129574644e-06, | |
| "loss": 0.9199, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 95, | |
| "tokens_per_second_per_gpu": 9778.3, | |
| "total_tokens": 18173373 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 0.43359375, | |
| "learning_rate": 6.229051583116796e-06, | |
| "loss": 0.9863, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 96, | |
| "tokens_per_second_per_gpu": 10231.38, | |
| "total_tokens": 18364184 | |
| }, | |
| { | |
| "epoch": 2.0208333333333335, | |
| "grad_norm": 0.546875, | |
| "learning_rate": 6.006354164343047e-06, | |
| "loss": 0.959, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 97, | |
| "tokens_per_second_per_gpu": 10122.23, | |
| "total_tokens": 18557919 | |
| }, | |
| { | |
| "epoch": 2.0416666666666665, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 5.785988922274711e-06, | |
| "loss": 0.9824, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 98, | |
| "tokens_per_second_per_gpu": 9172.58, | |
| "total_tokens": 18741829 | |
| }, | |
| { | |
| "epoch": 2.0625, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 5.5680845440075885e-06, | |
| "loss": 1.0273, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 99, | |
| "tokens_per_second_per_gpu": 10557.98, | |
| "total_tokens": 18936415 | |
| }, | |
| { | |
| "epoch": 2.0833333333333335, | |
| "grad_norm": 0.40234375, | |
| "learning_rate": 5.352768279562315e-06, | |
| "loss": 0.9277, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 100, | |
| "tokens_per_second_per_gpu": 10657.64, | |
| "total_tokens": 19136164 | |
| }, | |
| { | |
| "epoch": 2.1041666666666665, | |
| "grad_norm": 0.435546875, | |
| "learning_rate": 5.14016586757394e-06, | |
| "loss": 1.0391, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 101, | |
| "tokens_per_second_per_gpu": 9843.32, | |
| "total_tokens": 19324813 | |
| }, | |
| { | |
| "epoch": 2.125, | |
| "grad_norm": 0.4453125, | |
| "learning_rate": 4.930401461864099e-06, | |
| "loss": 1.0273, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 102, | |
| "tokens_per_second_per_gpu": 10233.37, | |
| "total_tokens": 19509949 | |
| }, | |
| { | |
| "epoch": 2.1458333333333335, | |
| "grad_norm": 0.431640625, | |
| "learning_rate": 4.7235975589386715e-06, | |
| "loss": 0.9805, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 103, | |
| "tokens_per_second_per_gpu": 10815.24, | |
| "total_tokens": 19703191 | |
| }, | |
| { | |
| "epoch": 2.1666666666666665, | |
| "grad_norm": 0.447265625, | |
| "learning_rate": 4.519874926453303e-06, | |
| "loss": 1.0, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 104, | |
| "tokens_per_second_per_gpu": 10097.8, | |
| "total_tokens": 19897478 | |
| }, | |
| { | |
| "epoch": 2.1875, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 4.319352532688444e-06, | |
| "loss": 1.0059, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 105, | |
| "tokens_per_second_per_gpu": 9860.18, | |
| "total_tokens": 20079605 | |
| }, | |
| { | |
| "epoch": 2.2083333333333335, | |
| "grad_norm": 0.439453125, | |
| "learning_rate": 4.12214747707527e-06, | |
| "loss": 1.0039, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 106, | |
| "tokens_per_second_per_gpu": 9991.56, | |
| "total_tokens": 20263849 | |
| }, | |
| { | |
| "epoch": 2.2291666666666665, | |
| "grad_norm": 0.43359375, | |
| "learning_rate": 3.9283749218128885e-06, | |
| "loss": 0.9863, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 107, | |
| "tokens_per_second_per_gpu": 10478.45, | |
| "total_tokens": 20454416 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "grad_norm": 0.45703125, | |
| "learning_rate": 3.738148024616863e-06, | |
| "loss": 0.998, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 108, | |
| "tokens_per_second_per_gpu": 10475.15, | |
| "total_tokens": 20641862 | |
| }, | |
| { | |
| "epoch": 2.2708333333333335, | |
| "grad_norm": 0.443359375, | |
| "learning_rate": 3.5515778726382967e-06, | |
| "loss": 0.9727, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 109, | |
| "tokens_per_second_per_gpu": 11180.87, | |
| "total_tokens": 20840096 | |
| }, | |
| { | |
| "epoch": 2.2916666666666665, | |
| "grad_norm": 0.41796875, | |
| "learning_rate": 3.3687734175920505e-06, | |
| "loss": 0.916, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 110, | |
| "tokens_per_second_per_gpu": 10457.13, | |
| "total_tokens": 21041565 | |
| }, | |
| { | |
| "epoch": 2.3125, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 3.1898414121320277e-06, | |
| "loss": 0.959, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 111, | |
| "tokens_per_second_per_gpu": 10793.07, | |
| "total_tokens": 21240749 | |
| }, | |
| { | |
| "epoch": 2.3333333333333335, | |
| "grad_norm": 0.439453125, | |
| "learning_rate": 3.0148863475106315e-06, | |
| "loss": 0.9453, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 112, | |
| "tokens_per_second_per_gpu": 10139.81, | |
| "total_tokens": 21429576 | |
| }, | |
| { | |
| "epoch": 2.3541666666666665, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 2.8440103925587904e-06, | |
| "loss": 0.9199, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 113, | |
| "tokens_per_second_per_gpu": 10567.3, | |
| "total_tokens": 21624286 | |
| }, | |
| { | |
| "epoch": 2.375, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 2.6773133340222677e-06, | |
| "loss": 0.9746, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 114, | |
| "tokens_per_second_per_gpu": 9746.37, | |
| "total_tokens": 21806885 | |
| }, | |
| { | |
| "epoch": 2.3958333333333335, | |
| "grad_norm": 0.4453125, | |
| "learning_rate": 2.514892518288988e-06, | |
| "loss": 0.9121, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 115, | |
| "tokens_per_second_per_gpu": 10113.82, | |
| "total_tokens": 22002533 | |
| }, | |
| { | |
| "epoch": 2.4166666666666665, | |
| "grad_norm": 0.56640625, | |
| "learning_rate": 2.3568427945415163e-06, | |
| "loss": 0.9434, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 116, | |
| "tokens_per_second_per_gpu": 10289.14, | |
| "total_tokens": 22190926 | |
| }, | |
| { | |
| "epoch": 2.4375, | |
| "grad_norm": 0.5703125, | |
| "learning_rate": 2.2032564593677773e-06, | |
| "loss": 0.9434, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 117, | |
| "tokens_per_second_per_gpu": 10024.32, | |
| "total_tokens": 22379128 | |
| }, | |
| { | |
| "epoch": 2.4583333333333335, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 2.0542232028624585e-06, | |
| "loss": 0.9707, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 118, | |
| "tokens_per_second_per_gpu": 10599.9, | |
| "total_tokens": 22572492 | |
| }, | |
| { | |
| "epoch": 2.4791666666666665, | |
| "grad_norm": 0.4375, | |
| "learning_rate": 1.9098300562505266e-06, | |
| "loss": 1.0039, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 119, | |
| "tokens_per_second_per_gpu": 10064.53, | |
| "total_tokens": 22758307 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "grad_norm": 0.421875, | |
| "learning_rate": 1.7701613410634367e-06, | |
| "loss": 0.9551, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 120, | |
| "tokens_per_second_per_gpu": 10637.75, | |
| "total_tokens": 22953079 | |
| }, | |
| { | |
| "epoch": 2.5208333333333335, | |
| "grad_norm": 0.43359375, | |
| "learning_rate": 1.6352986198977327e-06, | |
| "loss": 0.9844, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 121, | |
| "tokens_per_second_per_gpu": 10333.95, | |
| "total_tokens": 23145784 | |
| }, | |
| { | |
| "epoch": 2.5416666666666665, | |
| "grad_norm": 0.453125, | |
| "learning_rate": 1.5053206487847916e-06, | |
| "loss": 0.9707, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 122, | |
| "tokens_per_second_per_gpu": 9911.59, | |
| "total_tokens": 23327111 | |
| }, | |
| { | |
| "epoch": 2.5625, | |
| "grad_norm": 0.421875, | |
| "learning_rate": 1.3803033311995072e-06, | |
| "loss": 0.9688, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 123, | |
| "tokens_per_second_per_gpu": 10171.63, | |
| "total_tokens": 23518340 | |
| }, | |
| { | |
| "epoch": 2.5833333333333335, | |
| "grad_norm": 0.41796875, | |
| "learning_rate": 1.2603196737348211e-06, | |
| "loss": 0.9434, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 124, | |
| "tokens_per_second_per_gpu": 10639.7, | |
| "total_tokens": 23710497 | |
| }, | |
| { | |
| "epoch": 2.6041666666666665, | |
| "grad_norm": 0.41796875, | |
| "learning_rate": 1.1454397434679022e-06, | |
| "loss": 0.9805, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 125, | |
| "tokens_per_second_per_gpu": 10197.71, | |
| "total_tokens": 23901748 | |
| }, | |
| { | |
| "epoch": 2.625, | |
| "grad_norm": 0.419921875, | |
| "learning_rate": 1.0357306270429623e-06, | |
| "loss": 0.9785, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 126, | |
| "tokens_per_second_per_gpu": 9804.54, | |
| "total_tokens": 24084572 | |
| }, | |
| { | |
| "epoch": 2.6458333333333335, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 9.312563914945461e-07, | |
| "loss": 0.9766, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 127, | |
| "tokens_per_second_per_gpu": 10687.65, | |
| "total_tokens": 24271146 | |
| }, | |
| { | |
| "epoch": 2.6666666666666665, | |
| "grad_norm": 0.41796875, | |
| "learning_rate": 8.320780468341761e-07, | |
| "loss": 0.9961, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 128, | |
| "tokens_per_second_per_gpu": 10789.28, | |
| "total_tokens": 24468402 | |
| }, | |
| { | |
| "epoch": 2.6875, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 7.382535104222366e-07, | |
| "loss": 1.0117, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 129, | |
| "tokens_per_second_per_gpu": 10137.94, | |
| "total_tokens": 24655755 | |
| }, | |
| { | |
| "epoch": 2.7083333333333335, | |
| "grad_norm": 0.4296875, | |
| "learning_rate": 6.498375731458529e-07, | |
| "loss": 0.9883, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 130, | |
| "tokens_per_second_per_gpu": 10640.27, | |
| "total_tokens": 24840534 | |
| }, | |
| { | |
| "epoch": 2.7291666666666665, | |
| "grad_norm": 0.439453125, | |
| "learning_rate": 5.668818674225684e-07, | |
| "loss": 0.9941, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 131, | |
| "tokens_per_second_per_gpu": 10018.08, | |
| "total_tokens": 25024617 | |
| }, | |
| { | |
| "epoch": 2.75, | |
| "grad_norm": 0.427734375, | |
| "learning_rate": 4.894348370484648e-07, | |
| "loss": 0.9746, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 132, | |
| "tokens_per_second_per_gpu": 10544.96, | |
| "total_tokens": 25217845 | |
| }, | |
| { | |
| "epoch": 2.7708333333333335, | |
| "grad_norm": 0.455078125, | |
| "learning_rate": 4.1754170890833777e-07, | |
| "loss": 1.0234, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 133, | |
| "tokens_per_second_per_gpu": 10490.58, | |
| "total_tokens": 25406071 | |
| }, | |
| { | |
| "epoch": 2.7916666666666665, | |
| "grad_norm": 0.419921875, | |
| "learning_rate": 3.5124446656448654e-07, | |
| "loss": 0.9414, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 134, | |
| "tokens_per_second_per_gpu": 10903.67, | |
| "total_tokens": 25606313 | |
| }, | |
| { | |
| "epoch": 2.8125, | |
| "grad_norm": 0.416015625, | |
| "learning_rate": 2.905818257394799e-07, | |
| "loss": 0.9688, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 135, | |
| "tokens_per_second_per_gpu": 10824.22, | |
| "total_tokens": 25806285 | |
| }, | |
| { | |
| "epoch": 2.8333333333333335, | |
| "grad_norm": 0.421875, | |
| "learning_rate": 2.355892117072789e-07, | |
| "loss": 0.9277, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 136, | |
| "tokens_per_second_per_gpu": 9550.25, | |
| "total_tokens": 26001634 | |
| }, | |
| { | |
| "epoch": 2.8541666666666665, | |
| "grad_norm": 0.56640625, | |
| "learning_rate": 1.8629873860586567e-07, | |
| "loss": 1.0156, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 137, | |
| "tokens_per_second_per_gpu": 10503.94, | |
| "total_tokens": 26194751 | |
| }, | |
| { | |
| "epoch": 2.875, | |
| "grad_norm": 0.42578125, | |
| "learning_rate": 1.4273919068349184e-07, | |
| "loss": 1.0176, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 138, | |
| "tokens_per_second_per_gpu": 10613.72, | |
| "total_tokens": 26389840 | |
| }, | |
| { | |
| "epoch": 2.8958333333333335, | |
| "grad_norm": 0.55078125, | |
| "learning_rate": 1.0493600548948879e-07, | |
| "loss": 0.959, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 139, | |
| "tokens_per_second_per_gpu": 10081.76, | |
| "total_tokens": 26576949 | |
| }, | |
| { | |
| "epoch": 2.9166666666666665, | |
| "grad_norm": 0.421875, | |
| "learning_rate": 7.291125901946027e-08, | |
| "loss": 0.9727, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 140, | |
| "tokens_per_second_per_gpu": 10169.04, | |
| "total_tokens": 26773080 | |
| }, | |
| { | |
| "epoch": 2.9375, | |
| "grad_norm": 0.40625, | |
| "learning_rate": 4.6683652823513725e-08, | |
| "loss": 0.9434, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 141, | |
| "tokens_per_second_per_gpu": 10390.82, | |
| "total_tokens": 26972460 | |
| }, | |
| { | |
| "epoch": 2.9583333333333335, | |
| "grad_norm": 0.423828125, | |
| "learning_rate": 2.6268503085089547e-08, | |
| "loss": 1.0469, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 142, | |
| "tokens_per_second_per_gpu": 10107.1, | |
| "total_tokens": 27164036 | |
| }, | |
| { | |
| "epoch": 2.9791666666666665, | |
| "grad_norm": 0.400390625, | |
| "learning_rate": 1.1677731676733584e-08, | |
| "loss": 0.916, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 143, | |
| "tokens_per_second_per_gpu": 9845.86, | |
| "total_tokens": 27355465 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "grad_norm": 0.421875, | |
| "learning_rate": 2.9198591980705847e-09, | |
| "loss": 0.9844, | |
| "memory/device_reserved (GiB)": 77.12, | |
| "memory/max_active (GiB)": 69.81, | |
| "memory/max_allocated (GiB)": 69.81, | |
| "step": 144, | |
| "tokens_per_second_per_gpu": 10249.72, | |
| "total_tokens": 27546276 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 144, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 48, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 8.229628582947717e+17, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |