| { |
| "config": { |
| "epochs": 10, |
| "lr": 0.0003, |
| "warmup_steps": 300, |
| "grad_accum_steps": 8, |
| "grad_clip": 1.0 |
| }, |
| "train": [ |
| { |
| "epoch": 1, |
| "avg_loss": 7.070409, |
| "lr": 0.00027, |
| "step": 270, |
| "time_s": 971.0 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 6.138723, |
| "lr": 0.0002986039232296601, |
| "step": 540, |
| "time_s": 969.5 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 5.511719, |
| "lr": 0.00029373201732051774, |
| "step": 810, |
| "time_s": 970.5 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 5.033876, |
| "lr": 0.00028548319801432657, |
| "step": 1080, |
| "time_s": 966.4 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 4.711938, |
| "lr": 0.00027406193668993577, |
| "step": 1350, |
| "time_s": 977.1 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 4.464801, |
| "lr": 0.00027245653264858043, |
| "step": 1620, |
| "time_s": 973.7 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 4.292098, |
| "lr": 0.0002676801152109725, |
| "step": 1890, |
| "time_s": 977.7 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 4.150779, |
| "lr": 0.00025985108200750904, |
| "step": 2160, |
| "time_s": 975.7 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 4.048912, |
| "lr": 0.0002595061312929626, |
| "step": 2486, |
| "time_s": 1655.5 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 3.914366, |
| "lr": 0.00025878265404557577, |
| "step": 2809, |
| "time_s": 1653.9 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 3.793121, |
| "lr": 0.00025758003153357115, |
| "step": 3132, |
| "time_s": 1677.3 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 3.684527, |
| "lr": 0.00025590300995857436, |
| "step": 3455, |
| "time_s": 1674.5 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 3.584779, |
| "lr": 0.0002557817233080656, |
| "step": 3778, |
| "time_s": 1717.0 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 3.478163, |
| "lr": 0.0002554181076117055, |
| "step": 4101, |
| "time_s": 1719.1 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 3.377975, |
| "lr": 0.0002548128951431141, |
| "step": 4424, |
| "time_s": 1718.5 |
| }, |
| { |
| "epoch": 16, |
| "avg_loss": 3.283389, |
| "lr": 0.00025396730471965995, |
| "step": 4747, |
| "time_s": 1719.1 |
| }, |
| { |
| "epoch": 17, |
| "avg_loss": 3.198379, |
| "lr": 0.0002528830392479232, |
| "step": 5070, |
| "time_s": 1718.4 |
| }, |
| { |
| "epoch": 18, |
| "avg_loss": 3.123276, |
| "lr": 0.0002515622822942667, |
| "step": 5393, |
| "time_s": 1716.5 |
| }, |
| { |
| "epoch": 19, |
| "avg_loss": 3.054019, |
| "lr": 0.00025000769368742505, |
| "step": 5716, |
| "time_s": 1714.5 |
| }, |
| { |
| "epoch": 20, |
| "avg_loss": 2.990954, |
| "lr": 0.00024822240416196397, |
| "step": 6039, |
| "time_s": 1717.5 |
| }, |
| { |
| "epoch": 21, |
| "avg_loss": 2.932507, |
| "lr": 0.0002462100090533974, |
| "step": 6362, |
| "time_s": 1720.0 |
| }, |
| { |
| "epoch": 22, |
| "avg_loss": 2.879704, |
| "lr": 0.0002439745610576592, |
| "step": 6685, |
| "time_s": 1717.7 |
| }, |
| { |
| "epoch": 23, |
| "avg_loss": 2.830638, |
| "lr": 0.00024152056206951165, |
| "step": 7008, |
| "time_s": 1716.3 |
| }, |
| { |
| "epoch": 24, |
| "avg_loss": 2.785015, |
| "lr": 0.00023885295411633396, |
| "step": 7331, |
| "time_s": 1718.6 |
| }, |
| { |
| "epoch": 25, |
| "avg_loss": 2.743108, |
| "lr": 0.00023597710940553054, |
| "step": 7654, |
| "time_s": 1719.6 |
| }, |
| { |
| "epoch": 26, |
| "avg_loss": 2.72249, |
| "lr": 0.00023586585478300336, |
| "step": 7977, |
| "time_s": 1699.7 |
| }, |
| { |
| "epoch": 27, |
| "avg_loss": 2.681254, |
| "lr": 0.00023553231496741537, |
| "step": 8300, |
| "time_s": 1697.8 |
| }, |
| { |
| "epoch": 28, |
| "avg_loss": 2.644416, |
| "lr": 0.0002349771616635675, |
| "step": 8623, |
| "time_s": 1702.3 |
| }, |
| { |
| "epoch": 29, |
| "avg_loss": 2.610499, |
| "lr": 0.00023420151287633192, |
| "step": 8946, |
| "time_s": 1700.0 |
| }, |
| { |
| "epoch": 30, |
| "avg_loss": 2.578076, |
| "lr": 0.00023320693065914135, |
| "step": 9269, |
| "time_s": 1706.3 |
| } |
| ], |
| "val": [ |
| { |
| "epoch": 1, |
| "avg_loss": 6.459693 |
| }, |
| { |
| "epoch": 2, |
| "avg_loss": 5.804065 |
| }, |
| { |
| "epoch": 3, |
| "avg_loss": 5.179338 |
| }, |
| { |
| "epoch": 4, |
| "avg_loss": 4.774045 |
| }, |
| { |
| "epoch": 5, |
| "avg_loss": 4.511194 |
| }, |
| { |
| "epoch": 6, |
| "avg_loss": 4.305547 |
| }, |
| { |
| "epoch": 7, |
| "avg_loss": 4.154681 |
| }, |
| { |
| "epoch": 8, |
| "avg_loss": 4.037178 |
| }, |
| { |
| "epoch": 9, |
| "avg_loss": 3.829285 |
| }, |
| { |
| "epoch": 10, |
| "avg_loss": 3.708552 |
| }, |
| { |
| "epoch": 11, |
| "avg_loss": 3.594035 |
| }, |
| { |
| "epoch": 12, |
| "avg_loss": 3.495301 |
| }, |
| { |
| "epoch": 13, |
| "avg_loss": 3.318829 |
| }, |
| { |
| "epoch": 14, |
| "avg_loss": 3.237618 |
| }, |
| { |
| "epoch": 15, |
| "avg_loss": 3.171485 |
| }, |
| { |
| "epoch": 16, |
| "avg_loss": 3.082833 |
| }, |
| { |
| "epoch": 17, |
| "avg_loss": 3.013669 |
| }, |
| { |
| "epoch": 18, |
| "avg_loss": 2.949275 |
| }, |
| { |
| "epoch": 19, |
| "avg_loss": 2.898899 |
| }, |
| { |
| "epoch": 20, |
| "avg_loss": 2.855805 |
| }, |
| { |
| "epoch": 21, |
| "avg_loss": 2.804574 |
| }, |
| { |
| "epoch": 22, |
| "avg_loss": 2.76583 |
| }, |
| { |
| "epoch": 23, |
| "avg_loss": 2.720584 |
| }, |
| { |
| "epoch": 24, |
| "avg_loss": 2.692678 |
| }, |
| { |
| "epoch": 25, |
| "avg_loss": 2.660146 |
| }, |
| { |
| "epoch": 26, |
| "avg_loss": 2.477178 |
| }, |
| { |
| "epoch": 27, |
| "avg_loss": 2.463243 |
| }, |
| { |
| "epoch": 28, |
| "avg_loss": 2.459879 |
| }, |
| { |
| "epoch": 29, |
| "avg_loss": 2.455403 |
| }, |
| { |
| "epoch": 30, |
| "avg_loss": 2.439061 |
| } |
| ], |
| "sanity": [] |
| } |
