| { |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 13.038, |
| "global_step": 3000, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.0, |
| "learning_rate": 2e-06, |
| "loss": 11.0098, |
| "step": 1 |
| }, |
| { |
| "epoch": 0.02, |
| "learning_rate": 9.999999999999999e-05, |
| "loss": 8.5348, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.03, |
| "learning_rate": 0.00019999999999999998, |
| "loss": 6.0584, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.05, |
| "learning_rate": 0.0003, |
| "loss": 5.5248, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.07, |
| "learning_rate": 0.00039999999999999996, |
| "loss": 5.0954, |
| "step": 200 |
| }, |
| { |
| "epoch": 1.01, |
| "learning_rate": 0.0005, |
| "loss": 4.8186, |
| "step": 250 |
| }, |
| { |
| "epoch": 1.03, |
| "learning_rate": 0.0006, |
| "loss": 4.6442, |
| "step": 300 |
| }, |
| { |
| "epoch": 1.04, |
| "learning_rate": 0.0005888888888888889, |
| "loss": 4.4369, |
| "step": 350 |
| }, |
| { |
| "epoch": 1.06, |
| "learning_rate": 0.0005777777777777776, |
| "loss": 4.1967, |
| "step": 400 |
| }, |
| { |
| "epoch": 2.0, |
| "learning_rate": 0.0005666666666666666, |
| "loss": 4.0663, |
| "step": 450 |
| }, |
| { |
| "epoch": 2.02, |
| "learning_rate": 0.0005555555555555556, |
| "loss": 3.9725, |
| "step": 500 |
| }, |
| { |
| "epoch": 2.04, |
| "learning_rate": 0.0005444444444444444, |
| "loss": 3.9152, |
| "step": 550 |
| }, |
| { |
| "epoch": 2.05, |
| "learning_rate": 0.0005333333333333333, |
| "loss": 3.8671, |
| "step": 600 |
| }, |
| { |
| "epoch": 2.07, |
| "learning_rate": 0.0005222222222222222, |
| "loss": 3.8332, |
| "step": 650 |
| }, |
| { |
| "epoch": 3.01, |
| "learning_rate": 0.0005111111111111111, |
| "loss": 3.8044, |
| "step": 700 |
| }, |
| { |
| "epoch": 3.03, |
| "learning_rate": 0.0005, |
| "loss": 3.7774, |
| "step": 750 |
| }, |
| { |
| "epoch": 3.04, |
| "learning_rate": 0.0004888888888888889, |
| "loss": 3.7509, |
| "step": 800 |
| }, |
| { |
| "epoch": 3.06, |
| "learning_rate": 0.0004777777777777777, |
| "loss": 3.7289, |
| "step": 850 |
| }, |
| { |
| "epoch": 4.0, |
| "learning_rate": 0.0004666666666666666, |
| "loss": 3.7133, |
| "step": 900 |
| }, |
| { |
| "epoch": 4.02, |
| "learning_rate": 0.00045555555555555556, |
| "loss": 3.6884, |
| "step": 950 |
| }, |
| { |
| "epoch": 4.04, |
| "learning_rate": 0.00044444444444444436, |
| "loss": 3.6722, |
| "step": 1000 |
| }, |
| { |
| "epoch": 4.04, |
| "eval_loss": 4.149817943572998, |
| "eval_runtime": 95.0109, |
| "eval_samples_per_second": 131.711, |
| "eval_steps_per_second": 8.241, |
| "step": 1000 |
| }, |
| { |
| "epoch": 4.04, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_loss": 4.149817943572998, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_ppl": 63.42245278186028, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_runtime": 95.0109, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_samples_per_second": 131.711, |
| "step": 1000 |
| }, |
| { |
| "epoch": 4.05, |
| "learning_rate": 0.0004333333333333333, |
| "loss": 3.6561, |
| "step": 1050 |
| }, |
| { |
| "epoch": 4.07, |
| "learning_rate": 0.0004222222222222222, |
| "loss": 3.6445, |
| "step": 1100 |
| }, |
| { |
| "epoch": 5.01, |
| "learning_rate": 0.0004111111111111111, |
| "loss": 3.6333, |
| "step": 1150 |
| }, |
| { |
| "epoch": 5.03, |
| "learning_rate": 0.00039999999999999996, |
| "loss": 3.6214, |
| "step": 1200 |
| }, |
| { |
| "epoch": 5.05, |
| "learning_rate": 0.00038888888888888887, |
| "loss": 3.6075, |
| "step": 1250 |
| }, |
| { |
| "epoch": 5.06, |
| "learning_rate": 0.00037777777777777777, |
| "loss": 3.5992, |
| "step": 1300 |
| }, |
| { |
| "epoch": 6.01, |
| "learning_rate": 0.00036666666666666667, |
| "loss": 3.5956, |
| "step": 1350 |
| }, |
| { |
| "epoch": 6.02, |
| "learning_rate": 0.0003555555555555555, |
| "loss": 3.5829, |
| "step": 1400 |
| }, |
| { |
| "epoch": 6.04, |
| "learning_rate": 0.0003444444444444444, |
| "loss": 3.5736, |
| "step": 1450 |
| }, |
| { |
| "epoch": 6.06, |
| "learning_rate": 0.0003333333333333333, |
| "loss": 3.566, |
| "step": 1500 |
| }, |
| { |
| "epoch": 6.07, |
| "learning_rate": 0.0003222222222222222, |
| "loss": 3.5615, |
| "step": 1550 |
| }, |
| { |
| "epoch": 7.02, |
| "learning_rate": 0.00031111111111111107, |
| "loss": 3.5553, |
| "step": 1600 |
| }, |
| { |
| "epoch": 7.03, |
| "learning_rate": 0.0003, |
| "loss": 3.5475, |
| "step": 1650 |
| }, |
| { |
| "epoch": 7.05, |
| "learning_rate": 0.0002888888888888888, |
| "loss": 3.54, |
| "step": 1700 |
| }, |
| { |
| "epoch": 7.07, |
| "learning_rate": 0.0002777777777777778, |
| "loss": 3.5355, |
| "step": 1750 |
| }, |
| { |
| "epoch": 8.01, |
| "learning_rate": 0.0002666666666666666, |
| "loss": 3.5329, |
| "step": 1800 |
| }, |
| { |
| "epoch": 8.02, |
| "learning_rate": 0.00025555555555555553, |
| "loss": 3.5247, |
| "step": 1850 |
| }, |
| { |
| "epoch": 8.04, |
| "learning_rate": 0.00024444444444444443, |
| "loss": 3.5188, |
| "step": 1900 |
| }, |
| { |
| "epoch": 8.06, |
| "learning_rate": 0.0002333333333333333, |
| "loss": 3.5129, |
| "step": 1950 |
| }, |
| { |
| "epoch": 9.0, |
| "learning_rate": 0.00022222222222222218, |
| "loss": 3.5148, |
| "step": 2000 |
| }, |
| { |
| "epoch": 9.0, |
| "eval_loss": 4.061845302581787, |
| "eval_runtime": 95.5917, |
| "eval_samples_per_second": 130.911, |
| "eval_steps_per_second": 8.191, |
| "step": 2000 |
| }, |
| { |
| "epoch": 9.0, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_loss": 4.061845302581787, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_ppl": 58.081389990982125, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_runtime": 95.5917, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_samples_per_second": 130.911, |
| "step": 2000 |
| }, |
| { |
| "epoch": 9.02, |
| "learning_rate": 0.0002111111111111111, |
| "loss": 3.5045, |
| "step": 2050 |
| }, |
| { |
| "epoch": 9.03, |
| "learning_rate": 0.00019999999999999998, |
| "loss": 3.5013, |
| "step": 2100 |
| }, |
| { |
| "epoch": 9.05, |
| "learning_rate": 0.00018888888888888888, |
| "loss": 3.4959, |
| "step": 2150 |
| }, |
| { |
| "epoch": 9.07, |
| "learning_rate": 0.00017777777777777776, |
| "loss": 3.4938, |
| "step": 2200 |
| }, |
| { |
| "epoch": 10.01, |
| "learning_rate": 0.00016666666666666666, |
| "loss": 3.4916, |
| "step": 2250 |
| }, |
| { |
| "epoch": 10.03, |
| "learning_rate": 0.00015555555555555554, |
| "loss": 3.4859, |
| "step": 2300 |
| }, |
| { |
| "epoch": 10.04, |
| "learning_rate": 0.0001444444444444444, |
| "loss": 3.4813, |
| "step": 2350 |
| }, |
| { |
| "epoch": 10.06, |
| "learning_rate": 0.0001333333333333333, |
| "loss": 3.4765, |
| "step": 2400 |
| }, |
| { |
| "epoch": 11.0, |
| "learning_rate": 0.00012222222222222221, |
| "loss": 3.4802, |
| "step": 2450 |
| }, |
| { |
| "epoch": 11.02, |
| "learning_rate": 0.00011111111111111109, |
| "loss": 3.4708, |
| "step": 2500 |
| }, |
| { |
| "epoch": 11.04, |
| "learning_rate": 9.999999999999999e-05, |
| "loss": 3.4683, |
| "step": 2550 |
| }, |
| { |
| "epoch": 11.05, |
| "learning_rate": 8.888888888888888e-05, |
| "loss": 3.4653, |
| "step": 2600 |
| }, |
| { |
| "epoch": 11.07, |
| "learning_rate": 7.777777777777777e-05, |
| "loss": 3.4637, |
| "step": 2650 |
| }, |
| { |
| "epoch": 12.01, |
| "learning_rate": 6.666666666666666e-05, |
| "loss": 3.4623, |
| "step": 2700 |
| }, |
| { |
| "epoch": 12.03, |
| "learning_rate": 5.5555555555555545e-05, |
| "loss": 3.4586, |
| "step": 2750 |
| }, |
| { |
| "epoch": 12.05, |
| "learning_rate": 4.444444444444444e-05, |
| "loss": 3.4533, |
| "step": 2800 |
| }, |
| { |
| "epoch": 12.06, |
| "learning_rate": 3.333333333333333e-05, |
| "loss": 3.4516, |
| "step": 2850 |
| }, |
| { |
| "epoch": 13.0, |
| "learning_rate": 2.222222222222222e-05, |
| "loss": 3.455, |
| "step": 2900 |
| }, |
| { |
| "epoch": 13.02, |
| "learning_rate": 1.111111111111111e-05, |
| "loss": 3.4478, |
| "step": 2950 |
| }, |
| { |
| "epoch": 13.04, |
| "learning_rate": 0.0, |
| "loss": 3.4458, |
| "step": 3000 |
| }, |
| { |
| "epoch": 13.04, |
| "eval_loss": 4.029599189758301, |
| "eval_runtime": 94.0962, |
| "eval_samples_per_second": 132.991, |
| "eval_steps_per_second": 8.321, |
| "step": 3000 |
| }, |
| { |
| "epoch": 13.04, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_loss": 4.029599189758301, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_ppl": 56.23836581621429, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_runtime": 94.0962, |
| "eval_/home/s2678328/mission-impossible-language-models/training/babylm_dataset.py_samples_per_second": 132.991, |
| "step": 3000 |
| } |
| ], |
| "max_steps": 3000, |
| "num_train_epochs": 9223372036854775807, |
| "total_flos": 1.3382077680451584e+17, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
