Upload folder using huggingface_hub
Browse files- 40M_121M_12e3_10B.csv +0 -0
- 40M_121M_12e3_10B/blumodel_muon_step00000953.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00001906.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00002859.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00003812.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00004765.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00005718.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00006671.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00007624.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00008577.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00009530.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00010483.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00011436.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00012389.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00013342.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00014295.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00015248.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00016201.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00017154.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00018107.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00019060.pt +3 -0
- 40M_121M_12e3_10B/blumodel_muon_step00019074.pt +3 -0
- 40M_121M_15e4_10B_half.csv +0 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00000953.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00001906.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00002859.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00003812.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00004765.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00005718.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00006671.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00007624.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00008577.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00009472.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00009530.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00010483.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00011436.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00012389.pt +3 -0
- 40M_121M_15e4_10B_half/blumodel_muon_step00012561.pt +3 -0
- train_muon_40M_121M.py +6 -4
40M_121M_12e3_10B.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
40M_121M_12e3_10B/blumodel_muon_step00000953.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7023e87ad346da92f43f809029cb5f8f7a34018e4f21c0fc1f3b9a5761e974de
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00001906.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:024b0cdb822abc37af81af46503653d0a9e92df76b441a70cf2a40d190566a60
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00002859.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ebdf793279d2e667b1031c84e1b9a946f1a69bc15544e2aadf87f3ebf2381ef9
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00003812.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c0c59e1cc35793516af3d385924f8a9c359c644e30c10dc696d4848fac8b778
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00004765.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:02f1680d344746846118ac959e95e18ea7b38a58c6321021bb1738ab0702620c
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00005718.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:30b024d80fbbb6af3e708a89c4d4f2974aee4fd515564e439ddd77adb5762789
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00006671.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:975d1bc6a707a76270c296b06ab6243e8220ebf647e4cb04cf9fe3328bbad15e
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00007624.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c07f4a1ad205088d63d00550703f247f5b19d47ac1047dda0755966ffa44bcfa
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00008577.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a4f57e1e23b4b499c622759ede3eb7b70174b0d2456a4895972b75aa0facca92
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00009530.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c28d33e1583ca639cfb544e385909a5d39e789a7f56bf3e130036d94c4008737
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00010483.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:38739953a5a7a5b1f8bf1296854109c7068abdd821a677e1e822d8adf9b583e5
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00011436.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:00095f5458f6ec025a13846fc1cff11d76c2442ca72d52cdf6f9dfb0aa9f5c31
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00012389.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:70037094f573ed2fb371f801fa5f34294dacc9029c06668141712fcf5012034c
|
| 3 |
+
size 1390854167
|
40M_121M_12e3_10B/blumodel_muon_step00013342.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f04325d91731dda25082e0442461e72b53cd25fdef48714a547e715a25a82653
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00014295.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9f62fd8b98c3d49acce8adaa7f8b2db87c5b62016a601f470175e3f253faae9c
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00015248.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eb917d9e89c50faa1b3c7be90a0d59c8bdd4581a1d8e9ce57fe149f50146fa02
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00016201.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5509afe86bbdf8ff3dec0c7c0cd45f91724ff1ec57be4b3401c6a1c8cde3af4
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00017154.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c0e3e51d5f375ed4935d38ec28998fdeb604880b5061b78fdc8962d777243cc
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00018107.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6eb836f3b012e79ec06281deb8b0d5bb4d7ef379109843eb14413d2a0fcd8e31
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00019060.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1511e1c34beda4c21b9df1c811aa99bfcdbfdd330fa073782f68bc7e04b4425b
|
| 3 |
+
size 1390854359
|
40M_121M_12e3_10B/blumodel_muon_step00019074.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d2df289c8a615240fbc63a112524ceed945c30ec744aeed8fbb42c6b0ae8eb7e
|
| 3 |
+
size 1390854359
|
40M_121M_15e4_10B_half.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
40M_121M_15e4_10B_half/blumodel_muon_step00000953.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cd28f86f2d54d49465b6e2ce54a94d0e10ed892bbbdba45099024d6abcfff279
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00001906.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7547afd16f9da6f8088fda6f7a77713bcd7d78a036ba8ab652036669ac7500ed
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00002859.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ddd72a910f5193b6188b0325a128ff2bcb6e2c28d83b8ce136eaccd90fc6ee08
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00003812.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2359d3f89c2f6e33a8d40957e5716dd4fbd7dd6cb22234699c7b3c8a5e94ee60
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00004765.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3eb51edba26156e6006acd25d4a7bc40db5d00bc5a26d50aedb3d6296d0110c4
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00005718.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dfc63632d803657fc389e2b43f04944ffd177ae69881bde27189da65bf258c7c
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00006671.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:045666e7b9cf91c8e6b47d6fccf422fdfc5db2cdf633ebb51a155540bc06ade0
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00007624.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f2af442bbf058277aa227d32857674fdb6db8932517aa5041f8b387d90f177f
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00008577.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:446152e130a27a9dbf9b38474b73c15ad15f75401dae2da51c5eb9124986e3b7
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00009472.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dd45086b3f03269986861c1165b308afe6ecec165c9cfb44f637d50adc7b86ef
|
| 3 |
+
size 1390854167
|
40M_121M_15e4_10B_half/blumodel_muon_step00009530.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7669df868493732e06beaab855ea534243991afb90cdd743f80d2ed29e1fa583
|
| 3 |
+
size 1390854359
|
40M_121M_15e4_10B_half/blumodel_muon_step00010483.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:63d820065ac285914a8de9d81e68b8c0f37961c0694b101f8e18284855f8912b
|
| 3 |
+
size 1390854359
|
40M_121M_15e4_10B_half/blumodel_muon_step00011436.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f00489f260da72c811ba5427b02d2535b22d8338723303791e382bfde1546130
|
| 3 |
+
size 1390854359
|
40M_121M_15e4_10B_half/blumodel_muon_step00012389.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6ea2e91c1068284ae4172fd1b2ee665ac5b2a05982d5927ce425e87e179a061c
|
| 3 |
+
size 1390854359
|
40M_121M_15e4_10B_half/blumodel_muon_step00012561.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0feccda8e9f75c1c046e11a37b25e1f03c4f095bdf81056cf0b167408c6e3e36
|
| 3 |
+
size 1390854359
|
train_muon_40M_121M.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
# =============================================================================
|
| 3 |
# train_muon.py — the MUON arm of train.py. Same model, same data order, same
|
|
@@ -457,11 +458,11 @@ class ModelConfig:
|
|
| 457 |
# WSD: linear warmup -> constant at max_lr -> cosine decay to min_lr.
|
| 458 |
# Requires warmup_steps + stable_steps < max_steps - 1 (the remainder is
|
| 459 |
# the decay tail).
|
| 460 |
-
max_lr: float = 0.
|
| 461 |
-
min_lr: float = 0.
|
| 462 |
|
| 463 |
-
muon_lr: float = 0.
|
| 464 |
-
muon_min_lr: float = 0.
|
| 465 |
|
| 466 |
max_steps: int = 19074
|
| 467 |
warmup_steps: int = 500
|
|
@@ -1806,3 +1807,4 @@ if __name__ == "__main__":
|
|
| 1806 |
|
| 1807 |
|
| 1808 |
|
|
|
|
|
|
| 1 |
+
|
| 2 |
#!/usr/bin/env python3
|
| 3 |
# =============================================================================
|
| 4 |
# train_muon.py — the MUON arm of train.py. Same model, same data order, same
|
|
|
|
| 458 |
# WSD: linear warmup -> constant at max_lr -> cosine decay to min_lr.
|
| 459 |
# Requires warmup_steps + stable_steps < max_steps - 1 (the remainder is
|
| 460 |
# the decay tail).
|
| 461 |
+
max_lr: float = 0.0015
|
| 462 |
+
min_lr: float = 0.00015 # 10% of max_lr
|
| 463 |
|
| 464 |
+
muon_lr: float = 0.0015
|
| 465 |
+
muon_min_lr: float = 0.00015
|
| 466 |
|
| 467 |
max_steps: int = 19074
|
| 468 |
warmup_steps: int = 500
|
|
|
|
| 1807 |
|
| 1808 |
|
| 1809 |
|
| 1810 |
+
|