diff --git a/.gitattributes b/.gitattributes index 99a00d5c56ff049ce6f738b6fc8ef48b54f31aad..f12474519cda9711b01bd3e89b7dcff50bd24065 100644 --- a/.gitattributes +++ b/.gitattributes @@ -71,3 +71,12 @@ babylm_hop_control_10M_seed0/runs/checkpoint-1800/tokenizer.json filter=lfs diff babylm_hop_control_10M_seed0/runs/checkpoint-200/tokenizer.json filter=lfs diff=lfs merge=lfs -text babylm_hop_control_10M_seed0/runs/checkpoint-2400/tokenizer.json filter=lfs diff=lfs merge=lfs -text babylm_hop_control_10M_seed0/runs/checkpoint-2409/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-900/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-800/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-2000/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-1500/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-500/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-1900/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-2300/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text +babylm_hop_control_10M_seed0/runs/checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1100/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-1100/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b0a59b3d69f7c73d7ad02040ef9ddf0af64846ba --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1100/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:500aa1e2983f7f455a3e8210712a5fb824056d0fb864457f35db87cedc4f15d7 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..3259c6605b4fec3d2abdf15c7c3ccbdef85f5d96 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6b6aefb2eb9cf046f058c1e4bd20c70066eedd4535bf2e9f2b3f178ddbfbfd9e +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..9f4723485ff30930b59224e8afe0cb0b2b3c51bc --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f34c8972483cae48cde438f54c14179a5393e31298d634393f2e166c6a9bfb3e +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..6f245457df6ae96d9dae4e4269752c2f036e6ee3 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d0068d515416f5010564895d620c337c347a04bf8d5382015b5833708d52850 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..ca7b87736ad96984e06c88dc58b2e5efcdb20f2d --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a5d60eb151191f050cb4b3590ac23ee7ff247807e742892633ea85518fbb160a +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1200/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1200/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..491d7aa6873ac345ad7d2d8238b3751df06748aa --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc77a0f3d732b76deec0b9f9ee77f2ad22c4c8a301af0167ff41a26d22d5beb3 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..0c1653c04de59e53b4d05673f04ed4b2c7039c73 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a2973146fd35e5e1a515f6332b8a2b33728c4360853ce2c03e1a7a90b79c2cc +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..4cf5bfa682e62dbe5c2e496536a84208dc8433d0 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8d1658c107e9585fdf02f4eb4b846349794600f9e1c2f5151d959824c6021559 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a94b34e1aa5087ed1129ca130601f722e8d7959d --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d470a1566fade71dbfe3de9e5fe45feb7e321a52bff45349224c90061e1661d9 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1500/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1500/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..0b4ae0642e5f92f74c658999b194b5ef1fe11845 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8eeb05b2464265890d03970b06f167c82600ae380c5af274feadb43b529af8da +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..7f17a148b6fb83bbdca37275527834750b348cea --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0ba564af3daac2ee124ab61d7e40b0c6a31626027ac994a15a45b4c2ff14ad78 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..af9aae0e9d9650fb5e117bc169005317e2f3c5b1 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b7e9ef8a3b92a4a920eec9610cc2d3ae5914e60cee13c62730935b1fecc7ec2d +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..f70248c467716c25b5bb51158c6e363983a8b666 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f9e9dac98de75514ef8a2170470529bcca9086f59ac0a6a3251d108795f3e16a +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-1900/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-1900/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..3fde7d7c49c81b00a28223393cebc50bfe541e0b --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5419fdbc76df1033d8bbd5e67a29bea177e0db060d06136380184babd6724a83 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..b32943dd52f24c79c4a09bfcf59fbfbb94eea725 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0fe86e559cde3904e483c35c110a6ec5956237290f913253e911d4bf45d20674 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..028025132ce3c773ea373f1d990200a8d2f08a37 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c41d639def0ca99b5b4aedc81d4d3f65c59ee96f4fe0b4cb39a3f8d1ba5d4c0e +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c55d1fc7408e89387d12b601adb1e24691c6fb06 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:049dbfd63edf1c4b69fb1683cc6d1b2e977990cdc5a3b7199a3fdc77ce4a2c47 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2000/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2000/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..cf9f7cc0f22e38cdfea7c36a09f25ba146158728 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b93ade60213134037a6e2a03060d84a67c5041245aff2ad374a5c97f337ee4c2 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..22044561af392bd37b8a514ffc971741f1a025c2 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e17eca23ec632d21ab0115d8db1a9360104fa7b08d928954eee2eed9adf46b8a +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..24228040a3cbfaaf0d0e62b9e12dbef736514695 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6614221a631cfb519fe3c961fa724d0b1c62628b10e8afa55fbb30d5380c88e3 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..26faa3073c1eb0fc2bc47348851da9349eacad02 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:92e4e58a248bf0af7c3e00e00322cb1b7fb7e630ff6a437f6ca349ad76b9160a +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-2300/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-2300/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..083e16a85544a9d01e5efb17ecd7c530a20c1b69 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ed293d350d442fe903b83c290dfec9d04cc68797bbec9e4ab1262101ce270aee +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..d97c0db530cd50432564e335d6dc51190a59f735 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:efbadc27db14dd5750ddb1f5bfc9c98b2855855aca979de62032d04c6e2a9dcd +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..ad81578e26b1383d2f0f46a9c9301aa7b88146b1 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dcb6841be147129e79b182e07cbd5b0891ea133c78c669856fbc306a088e7867 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-500/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c8addd0c7032329830494eab37e85ff2f70fbb7d --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a4b69bc627a7734863da81864ffe77cfa2dc5c5516d8c34505dbac3b9dc1ec8 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-500/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-500/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-500/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-500/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..d5f37008b65b2b6b774ba8ca3e08ea7f2886beeb --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:898e9b2b9cf458e184c0b3debd303a4fd7c10dad3b82242d58682e89cace8b1e +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..43698f33ac8bdb91266f12645eda8b36787da828 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:050132a910188198ac7405dcc27a0ec731bdadfd7fd9c07e7468ac838c90fae0 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..90d0b3f139f4cb1368a680fad6f12df4c66f078f --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21ab24991874894fe74f6d6882102cbe2b931a423ab076862a8f74e33fa92f52 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-600/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..aeba18b3f74be376a645e91e43000677785c02d2 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:030ba726537291b026d46325eab79e8e9916823e9059bf1870c5c7ab595e53e3 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-600/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-600/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-600/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-600/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..10b2f5e4f6c9f32023941f5b8cf39645a86289fd --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2ccffad8722b9db65bab96085aca5b7293a1ef8d534c7b8817b948837c87f6b8 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..1f46eec58ec6affe22de910a8d659fe43a6018f4 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:633d50e430585944b70d0f16b52abb22f10fe2d185c48b1ba52f7c6a7004014f +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..89cef672a6b7edc3804190d221d1a77cd6a2d146 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c200394d27cbd1e2e417e51ec9d687fba67a2bda4d806d7a86762d3cf2c3f77 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/scheduler.pt b/babylm_hop_control_10M_seed0/runs/checkpoint-800/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..69bf883712a5fc210130703c570147f01420987e --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b764de8f0a7386632694b8b1ee7ba7aeae3190be43916b4b86dbeaf03deefaa2 +size 627 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-800/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-800/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-800/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-800/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_0.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..5c35fc80c7e9ebb0c9c57cecbf768fa6c69b78fe --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:06902b0190fdd2aea8d8d5c17aba1fd1f0db21bbf358b00fd19eb3353cb07576 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_1.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..8aa3d05bad492ad7e0b65d4c65cf7978c4c3e4fe --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2926fbc21fd903ac19866cf5bf9960dd1204045592eb7c1d4e5e0a050fbb3dfa +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_2.pth b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..8477ec8475f11de41748f67901272044f0210140 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-900/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2a05888d563eb0fa6710fb0fe530ff883a284844e09d19f9a2f6f51963234895 +size 16567 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-900/tokenizer.json b/babylm_hop_control_10M_seed0/runs/checkpoint-900/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..f563348f854ed1af8a8e9ff3391872f8d994ff83 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-900/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c2119a48316b293375cf214819b7f053f5f12171eb3ee8dccfce26510f91e2c +size 17210462 diff --git a/babylm_hop_control_10M_seed0/runs/checkpoint-900/training_args.bin b/babylm_hop_control_10M_seed0/runs/checkpoint-900/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..d28b7c159ad5567421a03464d70e00a1e2d43361 --- /dev/null +++ b/babylm_hop_control_10M_seed0/runs/checkpoint-900/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b717b63c99e15a7061bfb9ad0d8db9718415d88d6419d3c3dc4217194b1c01 +size 6011 diff --git a/babylm_hop_tokens4_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e362b81bf93cef8c6d96b08171bab63416808c70 --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:584d8d3e3f82f7964955174dfe5e3b1cf117a9d859f022cfdf7fcb884856e002 +size 4965799096 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e3ea8adc42bd3947b051b722ba0edc54a47f7501 --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd5a3b9b8a59c99d90270c8335ebd5f0ca86f0ed4988966b56011ec2405deeb6 +size 4965798912 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1400/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1400/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9dac8f2c54f04fe6ce90b1637ba884155fbdbb82 --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1400/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:83f12a0450661ade358600c69866a3679b1ce566be053f5070d763215e5bfc3d +size 4965798912 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1600/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1600/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..776d5be0784e7e58b81966c693a98fa53e0a3300 --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-1600/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:12227213680a39e184106d815f030d37d88d71fb8ea13d1e9f3000ddcbe593a1 +size 4965798912 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..8293a1cd7f49c9584eed368441ea9ce8b0b59cbf --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6ef2df93ecc9e6145d992793a85f57697d585c4cba8a609cf2e75d8cb2b1b702 +size 4965798912 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-300/global_step300/mp_rank_00_model_states.pt b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-300/global_step300/mp_rank_00_model_states.pt new file mode 100644 index 0000000000000000000000000000000000000000..ca9705d3670a59388152d4e242a9a608f0950b5b --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-300/global_step300/mp_rank_00_model_states.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e749a6aaf21655386d85514aeee02b70879be54cb0cdd7677cf8f0bc2cb39f7 +size 6425576219 diff --git a/babylm_hop_tokens4_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..c09e765e7df18670f7435daded61a25ab187f35e --- /dev/null +++ b/babylm_hop_tokens4_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c905229cbe88c4f377ca90cafe1ff929ab43124f8283eedfc970627a00ccc964 +size 4965798912 diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/latest new file mode 100644 index 0000000000000000000000000000000000000000..22cd5c3402316b70299aed2025d7943595f5d495 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/latest @@ -0,0 +1 @@ +global_step1100 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..3135ffc5e070d534f8035b6aee716ceae15b5c09 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/trainer_state.json @@ -0,0 +1,8613 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 1.3888888888888888, + "eval_steps": 10, + "global_step": 1100, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 3.429040350298112e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1100/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/latest new file mode 100644 index 0000000000000000000000000000000000000000..873b08355ae55e303606c474ab5fba25f9e8262a --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/latest @@ -0,0 +1 @@ +global_step1200 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..cc6ebf5f984e8c3f4419ec7a16712f6c18b01050 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/trainer_state.json @@ -0,0 +1,9393 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 1.5151515151515151, + "eval_steps": 10, + "global_step": 1200, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 3.740771291234304e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1200/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/latest new file mode 100644 index 0000000000000000000000000000000000000000..c2a0b5b17b996514a9bea0debe9403a641844bce --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/latest @@ -0,0 +1 @@ +global_step1400 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..442841752861e78d938b5d89e199c532f1b495ed --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/trainer_state.json @@ -0,0 +1,10953 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 1.7676767676767677, + "eval_steps": 10, + "global_step": 1400, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + }, + { + "epoch": 1.5164141414141414, + "grad_norm": 3.041095733642578, + "learning_rate": 2.782974742750234e-06, + "loss": 1.4736, + "step": 1201 + }, + { + "epoch": 1.5176767676767677, + "grad_norm": 2.8400142192840576, + "learning_rate": 2.780636108512629e-06, + "loss": 1.4169, + "step": 1202 + }, + { + "epoch": 1.518939393939394, + "grad_norm": 2.785592555999756, + "learning_rate": 2.7782974742750236e-06, + "loss": 1.4553, + "step": 1203 + }, + { + "epoch": 1.5202020202020203, + "grad_norm": 2.75209641456604, + "learning_rate": 2.775958840037418e-06, + "loss": 1.4267, + "step": 1204 + }, + { + "epoch": 1.5214646464646466, + "grad_norm": 2.927558183670044, + "learning_rate": 2.773620205799813e-06, + "loss": 1.4377, + "step": 1205 + }, + { + "epoch": 1.5227272727272727, + "grad_norm": 2.9335508346557617, + "learning_rate": 2.7712815715622077e-06, + "loss": 1.4295, + "step": 1206 + }, + { + "epoch": 1.523989898989899, + "grad_norm": 2.7812745571136475, + "learning_rate": 2.7689429373246026e-06, + "loss": 1.4029, + "step": 1207 + }, + { + "epoch": 1.5252525252525253, + "grad_norm": 3.230062961578369, + "learning_rate": 2.766604303086998e-06, + "loss": 1.539, + "step": 1208 + }, + { + "epoch": 1.5265151515151514, + "grad_norm": 3.1244332790374756, + "learning_rate": 2.7642656688493923e-06, + "loss": 1.4829, + "step": 1209 + }, + { + "epoch": 1.5277777777777777, + "grad_norm": 3.2391650676727295, + "learning_rate": 2.761927034611787e-06, + "loss": 1.4677, + "step": 1210 + }, + { + "epoch": 1.5277777777777777, + "eval_loss": 1.6849907636642456, + "eval_runtime": 36.8638, + "eval_samples_per_second": 27.127, + "eval_steps_per_second": 1.139, + "step": 1210 + }, + { + "epoch": 1.529040404040404, + "grad_norm": 3.1884987354278564, + "learning_rate": 2.759588400374182e-06, + "loss": 1.4185, + "step": 1211 + }, + { + "epoch": 1.5303030303030303, + "grad_norm": 2.959118366241455, + "learning_rate": 2.7572497661365764e-06, + "loss": 1.4536, + "step": 1212 + }, + { + "epoch": 1.5315656565656566, + "grad_norm": 2.882380247116089, + "learning_rate": 2.7549111318989712e-06, + "loss": 1.4872, + "step": 1213 + }, + { + "epoch": 1.5328282828282829, + "grad_norm": 3.0574429035186768, + "learning_rate": 2.752572497661366e-06, + "loss": 1.4343, + "step": 1214 + }, + { + "epoch": 1.5340909090909092, + "grad_norm": 3.1645233631134033, + "learning_rate": 2.750233863423761e-06, + "loss": 1.5101, + "step": 1215 + }, + { + "epoch": 1.5353535353535355, + "grad_norm": 2.83034086227417, + "learning_rate": 2.7478952291861553e-06, + "loss": 1.45, + "step": 1216 + }, + { + "epoch": 1.5366161616161618, + "grad_norm": 2.964780330657959, + "learning_rate": 2.74555659494855e-06, + "loss": 1.453, + "step": 1217 + }, + { + "epoch": 1.5378787878787878, + "grad_norm": 2.9896774291992188, + "learning_rate": 2.743217960710945e-06, + "loss": 1.4648, + "step": 1218 + }, + { + "epoch": 1.5391414141414141, + "grad_norm": 3.013664484024048, + "learning_rate": 2.74087932647334e-06, + "loss": 1.4828, + "step": 1219 + }, + { + "epoch": 1.5404040404040404, + "grad_norm": 3.0011773109436035, + "learning_rate": 2.7385406922357343e-06, + "loss": 1.4705, + "step": 1220 + }, + { + "epoch": 1.5404040404040404, + "eval_loss": 1.6872731447219849, + "eval_runtime": 36.9864, + "eval_samples_per_second": 27.037, + "eval_steps_per_second": 1.136, + "step": 1220 + }, + { + "epoch": 1.5416666666666665, + "grad_norm": 3.239349842071533, + "learning_rate": 2.736202057998129e-06, + "loss": 1.484, + "step": 1221 + }, + { + "epoch": 1.5429292929292928, + "grad_norm": 2.9267003536224365, + "learning_rate": 2.733863423760524e-06, + "loss": 1.4587, + "step": 1222 + }, + { + "epoch": 1.5441919191919191, + "grad_norm": 2.890094518661499, + "learning_rate": 2.7315247895229192e-06, + "loss": 1.4456, + "step": 1223 + }, + { + "epoch": 1.5454545454545454, + "grad_norm": 3.0287511348724365, + "learning_rate": 2.7291861552853136e-06, + "loss": 1.5059, + "step": 1224 + }, + { + "epoch": 1.5467171717171717, + "grad_norm": 3.041609525680542, + "learning_rate": 2.7268475210477085e-06, + "loss": 1.4341, + "step": 1225 + }, + { + "epoch": 1.547979797979798, + "grad_norm": 2.8771862983703613, + "learning_rate": 2.7245088868101033e-06, + "loss": 1.4308, + "step": 1226 + }, + { + "epoch": 1.5492424242424243, + "grad_norm": 3.1173794269561768, + "learning_rate": 2.722170252572498e-06, + "loss": 1.514, + "step": 1227 + }, + { + "epoch": 1.5505050505050506, + "grad_norm": 2.838275671005249, + "learning_rate": 2.7198316183348926e-06, + "loss": 1.4798, + "step": 1228 + }, + { + "epoch": 1.551767676767677, + "grad_norm": 2.9917166233062744, + "learning_rate": 2.7174929840972874e-06, + "loss": 1.5256, + "step": 1229 + }, + { + "epoch": 1.553030303030303, + "grad_norm": 3.071840763092041, + "learning_rate": 2.7151543498596823e-06, + "loss": 1.5187, + "step": 1230 + }, + { + "epoch": 1.553030303030303, + "eval_loss": 1.6850661039352417, + "eval_runtime": 37.0094, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 1230 + }, + { + "epoch": 1.5542929292929293, + "grad_norm": 3.253169536590576, + "learning_rate": 2.7128157156220767e-06, + "loss": 1.5522, + "step": 1231 + }, + { + "epoch": 1.5555555555555556, + "grad_norm": 3.037151336669922, + "learning_rate": 2.7104770813844715e-06, + "loss": 1.5125, + "step": 1232 + }, + { + "epoch": 1.5568181818181817, + "grad_norm": 2.8486485481262207, + "learning_rate": 2.7081384471468664e-06, + "loss": 1.4169, + "step": 1233 + }, + { + "epoch": 1.558080808080808, + "grad_norm": 2.953824996948242, + "learning_rate": 2.7057998129092612e-06, + "loss": 1.4505, + "step": 1234 + }, + { + "epoch": 1.5593434343434343, + "grad_norm": 3.012303590774536, + "learning_rate": 2.7034611786716557e-06, + "loss": 1.4816, + "step": 1235 + }, + { + "epoch": 1.5606060606060606, + "grad_norm": 3.1798768043518066, + "learning_rate": 2.7011225444340505e-06, + "loss": 1.5116, + "step": 1236 + }, + { + "epoch": 1.5618686868686869, + "grad_norm": 3.11370587348938, + "learning_rate": 2.6987839101964453e-06, + "loss": 1.5035, + "step": 1237 + }, + { + "epoch": 1.5631313131313131, + "grad_norm": 3.0563342571258545, + "learning_rate": 2.69644527595884e-06, + "loss": 1.4387, + "step": 1238 + }, + { + "epoch": 1.5643939393939394, + "grad_norm": 2.9884026050567627, + "learning_rate": 2.6941066417212354e-06, + "loss": 1.428, + "step": 1239 + }, + { + "epoch": 1.5656565656565657, + "grad_norm": 2.955773115158081, + "learning_rate": 2.69176800748363e-06, + "loss": 1.4208, + "step": 1240 + }, + { + "epoch": 1.5656565656565657, + "eval_loss": 1.685523509979248, + "eval_runtime": 37.0248, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1240 + }, + { + "epoch": 1.566919191919192, + "grad_norm": 2.9858005046844482, + "learning_rate": 2.6894293732460247e-06, + "loss": 1.4706, + "step": 1241 + }, + { + "epoch": 1.5681818181818183, + "grad_norm": 3.122340440750122, + "learning_rate": 2.6870907390084196e-06, + "loss": 1.4892, + "step": 1242 + }, + { + "epoch": 1.5694444444444444, + "grad_norm": 2.9828262329101562, + "learning_rate": 2.684752104770814e-06, + "loss": 1.4255, + "step": 1243 + }, + { + "epoch": 1.5707070707070707, + "grad_norm": 2.916653633117676, + "learning_rate": 2.682413470533209e-06, + "loss": 1.4241, + "step": 1244 + }, + { + "epoch": 1.571969696969697, + "grad_norm": 3.1615540981292725, + "learning_rate": 2.6800748362956037e-06, + "loss": 1.4719, + "step": 1245 + }, + { + "epoch": 1.573232323232323, + "grad_norm": 3.154130697250366, + "learning_rate": 2.6777362020579985e-06, + "loss": 1.4731, + "step": 1246 + }, + { + "epoch": 1.5744949494949494, + "grad_norm": 3.040848731994629, + "learning_rate": 2.675397567820393e-06, + "loss": 1.5149, + "step": 1247 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.9808318614959717, + "learning_rate": 2.6730589335827878e-06, + "loss": 1.431, + "step": 1248 + }, + { + "epoch": 1.577020202020202, + "grad_norm": 2.8538382053375244, + "learning_rate": 2.6707202993451826e-06, + "loss": 1.4458, + "step": 1249 + }, + { + "epoch": 1.5782828282828283, + "grad_norm": 3.113945960998535, + "learning_rate": 2.6683816651075774e-06, + "loss": 1.4873, + "step": 1250 + }, + { + "epoch": 1.5782828282828283, + "eval_loss": 1.6823841333389282, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1250 + }, + { + "epoch": 1.5795454545454546, + "grad_norm": 3.0044517517089844, + "learning_rate": 2.666043030869972e-06, + "loss": 1.4904, + "step": 1251 + }, + { + "epoch": 1.5808080808080809, + "grad_norm": 2.845820665359497, + "learning_rate": 2.6637043966323667e-06, + "loss": 1.4174, + "step": 1252 + }, + { + "epoch": 1.5820707070707072, + "grad_norm": 2.9123542308807373, + "learning_rate": 2.6613657623947616e-06, + "loss": 1.4896, + "step": 1253 + }, + { + "epoch": 1.5833333333333335, + "grad_norm": 3.1593222618103027, + "learning_rate": 2.659027128157157e-06, + "loss": 1.5208, + "step": 1254 + }, + { + "epoch": 1.5845959595959596, + "grad_norm": 3.0782854557037354, + "learning_rate": 2.6566884939195512e-06, + "loss": 1.4911, + "step": 1255 + }, + { + "epoch": 1.5858585858585859, + "grad_norm": 2.77309513092041, + "learning_rate": 2.654349859681946e-06, + "loss": 1.4475, + "step": 1256 + }, + { + "epoch": 1.5871212121212122, + "grad_norm": 2.9154622554779053, + "learning_rate": 2.652011225444341e-06, + "loss": 1.4324, + "step": 1257 + }, + { + "epoch": 1.5883838383838382, + "grad_norm": 3.107395887374878, + "learning_rate": 2.6496725912067358e-06, + "loss": 1.5126, + "step": 1258 + }, + { + "epoch": 1.5896464646464645, + "grad_norm": 2.825025796890259, + "learning_rate": 2.64733395696913e-06, + "loss": 1.4723, + "step": 1259 + }, + { + "epoch": 1.5909090909090908, + "grad_norm": 2.833296060562134, + "learning_rate": 2.644995322731525e-06, + "loss": 1.4516, + "step": 1260 + }, + { + "epoch": 1.5909090909090908, + "eval_loss": 1.6950432062149048, + "eval_runtime": 37.0343, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1260 + }, + { + "epoch": 1.5921717171717171, + "grad_norm": 2.9508421421051025, + "learning_rate": 2.64265668849392e-06, + "loss": 1.4002, + "step": 1261 + }, + { + "epoch": 1.5934343434343434, + "grad_norm": 2.8778491020202637, + "learning_rate": 2.6403180542563143e-06, + "loss": 1.4258, + "step": 1262 + }, + { + "epoch": 1.5946969696969697, + "grad_norm": 2.81261944770813, + "learning_rate": 2.637979420018709e-06, + "loss": 1.435, + "step": 1263 + }, + { + "epoch": 1.595959595959596, + "grad_norm": 3.1811511516571045, + "learning_rate": 2.635640785781104e-06, + "loss": 1.4792, + "step": 1264 + }, + { + "epoch": 1.5972222222222223, + "grad_norm": 3.0168304443359375, + "learning_rate": 2.633302151543499e-06, + "loss": 1.5394, + "step": 1265 + }, + { + "epoch": 1.5984848484848486, + "grad_norm": 2.9531311988830566, + "learning_rate": 2.6309635173058932e-06, + "loss": 1.4775, + "step": 1266 + }, + { + "epoch": 1.5997474747474747, + "grad_norm": 2.835839033126831, + "learning_rate": 2.628624883068288e-06, + "loss": 1.478, + "step": 1267 + }, + { + "epoch": 1.601010101010101, + "grad_norm": 2.9014039039611816, + "learning_rate": 2.626286248830683e-06, + "loss": 1.4926, + "step": 1268 + }, + { + "epoch": 1.6022727272727273, + "grad_norm": 2.89461088180542, + "learning_rate": 2.623947614593078e-06, + "loss": 1.448, + "step": 1269 + }, + { + "epoch": 1.6035353535353534, + "grad_norm": 2.8005120754241943, + "learning_rate": 2.621608980355473e-06, + "loss": 1.4733, + "step": 1270 + }, + { + "epoch": 1.6035353535353534, + "eval_loss": 1.6889415979385376, + "eval_runtime": 37.0345, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1270 + }, + { + "epoch": 1.6047979797979797, + "grad_norm": 2.9576175212860107, + "learning_rate": 2.6192703461178675e-06, + "loss": 1.5116, + "step": 1271 + }, + { + "epoch": 1.606060606060606, + "grad_norm": 2.926638603210449, + "learning_rate": 2.6169317118802623e-06, + "loss": 1.4744, + "step": 1272 + }, + { + "epoch": 1.6073232323232323, + "grad_norm": 2.911587715148926, + "learning_rate": 2.614593077642657e-06, + "loss": 1.4695, + "step": 1273 + }, + { + "epoch": 1.6085858585858586, + "grad_norm": 2.8878750801086426, + "learning_rate": 2.6122544434050516e-06, + "loss": 1.4574, + "step": 1274 + }, + { + "epoch": 1.6098484848484849, + "grad_norm": 2.8489067554473877, + "learning_rate": 2.6099158091674464e-06, + "loss": 1.4599, + "step": 1275 + }, + { + "epoch": 1.6111111111111112, + "grad_norm": 2.8223719596862793, + "learning_rate": 2.6075771749298412e-06, + "loss": 1.4431, + "step": 1276 + }, + { + "epoch": 1.6123737373737375, + "grad_norm": 2.869910955429077, + "learning_rate": 2.605238540692236e-06, + "loss": 1.48, + "step": 1277 + }, + { + "epoch": 1.6136363636363638, + "grad_norm": 2.8435168266296387, + "learning_rate": 2.6028999064546305e-06, + "loss": 1.4471, + "step": 1278 + }, + { + "epoch": 1.61489898989899, + "grad_norm": 2.9384665489196777, + "learning_rate": 2.6005612722170253e-06, + "loss": 1.5176, + "step": 1279 + }, + { + "epoch": 1.6161616161616161, + "grad_norm": 3.021785020828247, + "learning_rate": 2.59822263797942e-06, + "loss": 1.4789, + "step": 1280 + }, + { + "epoch": 1.6161616161616161, + "eval_loss": 1.6864516735076904, + "eval_runtime": 37.0249, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1280 + }, + { + "epoch": 1.6174242424242424, + "grad_norm": 3.007183313369751, + "learning_rate": 2.5958840037418146e-06, + "loss": 1.4686, + "step": 1281 + }, + { + "epoch": 1.6186868686868687, + "grad_norm": 3.115910053253174, + "learning_rate": 2.5935453695042095e-06, + "loss": 1.5382, + "step": 1282 + }, + { + "epoch": 1.6199494949494948, + "grad_norm": 2.98108172416687, + "learning_rate": 2.5912067352666043e-06, + "loss": 1.4569, + "step": 1283 + }, + { + "epoch": 1.621212121212121, + "grad_norm": 2.926832437515259, + "learning_rate": 2.5888681010289996e-06, + "loss": 1.4515, + "step": 1284 + }, + { + "epoch": 1.6224747474747474, + "grad_norm": 2.9045534133911133, + "learning_rate": 2.5865294667913944e-06, + "loss": 1.4259, + "step": 1285 + }, + { + "epoch": 1.6237373737373737, + "grad_norm": 3.0757768154144287, + "learning_rate": 2.584190832553789e-06, + "loss": 1.5436, + "step": 1286 + }, + { + "epoch": 1.625, + "grad_norm": 3.045257806777954, + "learning_rate": 2.5818521983161837e-06, + "loss": 1.4378, + "step": 1287 + }, + { + "epoch": 1.6262626262626263, + "grad_norm": 2.8131601810455322, + "learning_rate": 2.5795135640785785e-06, + "loss": 1.4224, + "step": 1288 + }, + { + "epoch": 1.6275252525252526, + "grad_norm": 2.953326940536499, + "learning_rate": 2.5771749298409734e-06, + "loss": 1.4137, + "step": 1289 + }, + { + "epoch": 1.628787878787879, + "grad_norm": 2.946232318878174, + "learning_rate": 2.5748362956033678e-06, + "loss": 1.4575, + "step": 1290 + }, + { + "epoch": 1.628787878787879, + "eval_loss": 1.6896334886550903, + "eval_runtime": 37.317, + "eval_samples_per_second": 26.797, + "eval_steps_per_second": 1.125, + "step": 1290 + }, + { + "epoch": 1.6300505050505052, + "grad_norm": 2.860867738723755, + "learning_rate": 2.5724976613657626e-06, + "loss": 1.4412, + "step": 1291 + }, + { + "epoch": 1.6313131313131313, + "grad_norm": 3.0417261123657227, + "learning_rate": 2.5701590271281575e-06, + "loss": 1.4919, + "step": 1292 + }, + { + "epoch": 1.6325757575757576, + "grad_norm": 3.21425199508667, + "learning_rate": 2.567820392890552e-06, + "loss": 1.5243, + "step": 1293 + }, + { + "epoch": 1.6338383838383839, + "grad_norm": 3.000302314758301, + "learning_rate": 2.5654817586529467e-06, + "loss": 1.4571, + "step": 1294 + }, + { + "epoch": 1.63510101010101, + "grad_norm": 2.88415789604187, + "learning_rate": 2.5631431244153416e-06, + "loss": 1.4527, + "step": 1295 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.7747156620025635, + "learning_rate": 2.5608044901777364e-06, + "loss": 1.3984, + "step": 1296 + }, + { + "epoch": 1.6376262626262625, + "grad_norm": 2.9298319816589355, + "learning_rate": 2.558465855940131e-06, + "loss": 1.4587, + "step": 1297 + }, + { + "epoch": 1.6388888888888888, + "grad_norm": 3.0199968814849854, + "learning_rate": 2.5561272217025257e-06, + "loss": 1.4539, + "step": 1298 + }, + { + "epoch": 1.6401515151515151, + "grad_norm": 3.09503436088562, + "learning_rate": 2.5537885874649205e-06, + "loss": 1.4551, + "step": 1299 + }, + { + "epoch": 1.6414141414141414, + "grad_norm": 2.972161293029785, + "learning_rate": 2.5514499532273158e-06, + "loss": 1.4123, + "step": 1300 + }, + { + "epoch": 1.6414141414141414, + "eval_loss": 1.6877377033233643, + "eval_runtime": 37.4088, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1300 + }, + { + "epoch": 1.6426767676767677, + "grad_norm": 3.081058979034424, + "learning_rate": 2.5491113189897106e-06, + "loss": 1.457, + "step": 1301 + }, + { + "epoch": 1.643939393939394, + "grad_norm": 3.040106773376465, + "learning_rate": 2.546772684752105e-06, + "loss": 1.4465, + "step": 1302 + }, + { + "epoch": 1.6452020202020203, + "grad_norm": 3.133812427520752, + "learning_rate": 2.5444340505145e-06, + "loss": 1.4382, + "step": 1303 + }, + { + "epoch": 1.6464646464646466, + "grad_norm": 2.9701125621795654, + "learning_rate": 2.5420954162768947e-06, + "loss": 1.457, + "step": 1304 + }, + { + "epoch": 1.6477272727272727, + "grad_norm": 2.9272050857543945, + "learning_rate": 2.539756782039289e-06, + "loss": 1.464, + "step": 1305 + }, + { + "epoch": 1.648989898989899, + "grad_norm": 3.048401117324829, + "learning_rate": 2.537418147801684e-06, + "loss": 1.4688, + "step": 1306 + }, + { + "epoch": 1.6502525252525253, + "grad_norm": 2.9103102684020996, + "learning_rate": 2.535079513564079e-06, + "loss": 1.5025, + "step": 1307 + }, + { + "epoch": 1.6515151515151514, + "grad_norm": 3.0841548442840576, + "learning_rate": 2.5327408793264737e-06, + "loss": 1.4938, + "step": 1308 + }, + { + "epoch": 1.6527777777777777, + "grad_norm": 3.022184371948242, + "learning_rate": 2.530402245088868e-06, + "loss": 1.4332, + "step": 1309 + }, + { + "epoch": 1.654040404040404, + "grad_norm": 2.881479024887085, + "learning_rate": 2.528063610851263e-06, + "loss": 1.4541, + "step": 1310 + }, + { + "epoch": 1.654040404040404, + "eval_loss": 1.6814675331115723, + "eval_runtime": 36.9446, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.6553030303030303, + "grad_norm": 2.906719446182251, + "learning_rate": 2.5257249766136578e-06, + "loss": 1.4968, + "step": 1311 + }, + { + "epoch": 1.6565656565656566, + "grad_norm": 3.0566461086273193, + "learning_rate": 2.523386342376052e-06, + "loss": 1.4513, + "step": 1312 + }, + { + "epoch": 1.6578282828282829, + "grad_norm": 2.895026206970215, + "learning_rate": 2.521047708138447e-06, + "loss": 1.4217, + "step": 1313 + }, + { + "epoch": 1.6590909090909092, + "grad_norm": 3.0006465911865234, + "learning_rate": 2.518709073900842e-06, + "loss": 1.4793, + "step": 1314 + }, + { + "epoch": 1.6603535353535355, + "grad_norm": 2.9853575229644775, + "learning_rate": 2.516370439663237e-06, + "loss": 1.4642, + "step": 1315 + }, + { + "epoch": 1.6616161616161618, + "grad_norm": 2.9729392528533936, + "learning_rate": 2.514031805425632e-06, + "loss": 1.4644, + "step": 1316 + }, + { + "epoch": 1.6628787878787878, + "grad_norm": 3.0997586250305176, + "learning_rate": 2.5116931711880264e-06, + "loss": 1.4392, + "step": 1317 + }, + { + "epoch": 1.6641414141414141, + "grad_norm": 2.9621379375457764, + "learning_rate": 2.5093545369504213e-06, + "loss": 1.4387, + "step": 1318 + }, + { + "epoch": 1.6654040404040404, + "grad_norm": 3.0571911334991455, + "learning_rate": 2.507015902712816e-06, + "loss": 1.4638, + "step": 1319 + }, + { + "epoch": 1.6666666666666665, + "grad_norm": 2.9580321311950684, + "learning_rate": 2.504677268475211e-06, + "loss": 1.4985, + "step": 1320 + }, + { + "epoch": 1.6666666666666665, + "eval_loss": 1.6790977716445923, + "eval_runtime": 37.0242, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1320 + }, + { + "epoch": 1.6679292929292928, + "grad_norm": 2.87432599067688, + "learning_rate": 2.5023386342376054e-06, + "loss": 1.4058, + "step": 1321 + }, + { + "epoch": 1.6691919191919191, + "grad_norm": 2.925546407699585, + "learning_rate": 2.5e-06, + "loss": 1.4986, + "step": 1322 + }, + { + "epoch": 1.6704545454545454, + "grad_norm": 2.837743043899536, + "learning_rate": 2.497661365762395e-06, + "loss": 1.455, + "step": 1323 + }, + { + "epoch": 1.6717171717171717, + "grad_norm": 3.111581563949585, + "learning_rate": 2.4953227315247895e-06, + "loss": 1.5005, + "step": 1324 + }, + { + "epoch": 1.672979797979798, + "grad_norm": 2.851076126098633, + "learning_rate": 2.4929840972871843e-06, + "loss": 1.4365, + "step": 1325 + }, + { + "epoch": 1.6742424242424243, + "grad_norm": 2.966965675354004, + "learning_rate": 2.4906454630495796e-06, + "loss": 1.4697, + "step": 1326 + }, + { + "epoch": 1.6755050505050506, + "grad_norm": 3.015193223953247, + "learning_rate": 2.488306828811974e-06, + "loss": 1.4486, + "step": 1327 + }, + { + "epoch": 1.676767676767677, + "grad_norm": 3.017681121826172, + "learning_rate": 2.485968194574369e-06, + "loss": 1.4017, + "step": 1328 + }, + { + "epoch": 1.678030303030303, + "grad_norm": 3.071981906890869, + "learning_rate": 2.4836295603367637e-06, + "loss": 1.4572, + "step": 1329 + }, + { + "epoch": 1.6792929292929293, + "grad_norm": 3.051741123199463, + "learning_rate": 2.481290926099158e-06, + "loss": 1.4498, + "step": 1330 + }, + { + "epoch": 1.6792929292929293, + "eval_loss": 1.6771777868270874, + "eval_runtime": 37.0747, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1330 + }, + { + "epoch": 1.6805555555555556, + "grad_norm": 2.9561071395874023, + "learning_rate": 2.478952291861553e-06, + "loss": 1.4428, + "step": 1331 + }, + { + "epoch": 1.6818181818181817, + "grad_norm": 2.992584466934204, + "learning_rate": 2.4766136576239478e-06, + "loss": 1.461, + "step": 1332 + }, + { + "epoch": 1.683080808080808, + "grad_norm": 3.0473921298980713, + "learning_rate": 2.4742750233863426e-06, + "loss": 1.489, + "step": 1333 + }, + { + "epoch": 1.6843434343434343, + "grad_norm": 2.8721725940704346, + "learning_rate": 2.4719363891487375e-06, + "loss": 1.487, + "step": 1334 + }, + { + "epoch": 1.6856060606060606, + "grad_norm": 2.8650765419006348, + "learning_rate": 2.4695977549111323e-06, + "loss": 1.4504, + "step": 1335 + }, + { + "epoch": 1.6868686868686869, + "grad_norm": 2.998887538909912, + "learning_rate": 2.4672591206735267e-06, + "loss": 1.4586, + "step": 1336 + }, + { + "epoch": 1.6881313131313131, + "grad_norm": 2.915188789367676, + "learning_rate": 2.4649204864359216e-06, + "loss": 1.4333, + "step": 1337 + }, + { + "epoch": 1.6893939393939394, + "grad_norm": 2.9660091400146484, + "learning_rate": 2.4625818521983164e-06, + "loss": 1.4711, + "step": 1338 + }, + { + "epoch": 1.6906565656565657, + "grad_norm": 2.916274070739746, + "learning_rate": 2.4602432179607113e-06, + "loss": 1.4642, + "step": 1339 + }, + { + "epoch": 1.691919191919192, + "grad_norm": 3.04154896736145, + "learning_rate": 2.4579045837231057e-06, + "loss": 1.4233, + "step": 1340 + }, + { + "epoch": 1.691919191919192, + "eval_loss": 1.6764302253723145, + "eval_runtime": 37.0686, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1340 + }, + { + "epoch": 1.6931818181818183, + "grad_norm": 2.9976465702056885, + "learning_rate": 2.4555659494855005e-06, + "loss": 1.4775, + "step": 1341 + }, + { + "epoch": 1.6944444444444444, + "grad_norm": 2.9140090942382812, + "learning_rate": 2.4532273152478954e-06, + "loss": 1.4532, + "step": 1342 + }, + { + "epoch": 1.6957070707070707, + "grad_norm": 3.0224454402923584, + "learning_rate": 2.45088868101029e-06, + "loss": 1.5091, + "step": 1343 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.8979439735412598, + "learning_rate": 2.448550046772685e-06, + "loss": 1.4634, + "step": 1344 + }, + { + "epoch": 1.698232323232323, + "grad_norm": 2.913433313369751, + "learning_rate": 2.44621141253508e-06, + "loss": 1.4483, + "step": 1345 + }, + { + "epoch": 1.6994949494949494, + "grad_norm": 2.813455820083618, + "learning_rate": 2.4438727782974743e-06, + "loss": 1.446, + "step": 1346 + }, + { + "epoch": 1.7007575757575757, + "grad_norm": 2.988722324371338, + "learning_rate": 2.441534144059869e-06, + "loss": 1.4908, + "step": 1347 + }, + { + "epoch": 1.702020202020202, + "grad_norm": 2.850446939468384, + "learning_rate": 2.439195509822264e-06, + "loss": 1.4221, + "step": 1348 + }, + { + "epoch": 1.7032828282828283, + "grad_norm": 2.8405508995056152, + "learning_rate": 2.436856875584659e-06, + "loss": 1.4496, + "step": 1349 + }, + { + "epoch": 1.7045454545454546, + "grad_norm": 3.0188651084899902, + "learning_rate": 2.4345182413470537e-06, + "loss": 1.489, + "step": 1350 + }, + { + "epoch": 1.7045454545454546, + "eval_loss": 1.6780083179473877, + "eval_runtime": 37.0724, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1350 + }, + { + "epoch": 1.7058080808080809, + "grad_norm": 2.905088424682617, + "learning_rate": 2.4321796071094485e-06, + "loss": 1.4283, + "step": 1351 + }, + { + "epoch": 1.7070707070707072, + "grad_norm": 2.878998279571533, + "learning_rate": 2.429840972871843e-06, + "loss": 1.4357, + "step": 1352 + }, + { + "epoch": 1.7083333333333335, + "grad_norm": 3.1707355976104736, + "learning_rate": 2.427502338634238e-06, + "loss": 1.496, + "step": 1353 + }, + { + "epoch": 1.7095959595959596, + "grad_norm": 2.9610493183135986, + "learning_rate": 2.4251637043966326e-06, + "loss": 1.4905, + "step": 1354 + }, + { + "epoch": 1.7108585858585859, + "grad_norm": 2.8372201919555664, + "learning_rate": 2.422825070159027e-06, + "loss": 1.4563, + "step": 1355 + }, + { + "epoch": 1.7121212121212122, + "grad_norm": 2.8741753101348877, + "learning_rate": 2.420486435921422e-06, + "loss": 1.4525, + "step": 1356 + }, + { + "epoch": 1.7133838383838382, + "grad_norm": 2.973726749420166, + "learning_rate": 2.418147801683817e-06, + "loss": 1.4878, + "step": 1357 + }, + { + "epoch": 1.7146464646464645, + "grad_norm": 3.05717396736145, + "learning_rate": 2.4158091674462116e-06, + "loss": 1.5402, + "step": 1358 + }, + { + "epoch": 1.7159090909090908, + "grad_norm": 2.883772373199463, + "learning_rate": 2.4134705332086064e-06, + "loss": 1.4329, + "step": 1359 + }, + { + "epoch": 1.7171717171717171, + "grad_norm": 2.8246653079986572, + "learning_rate": 2.4111318989710013e-06, + "loss": 1.4821, + "step": 1360 + }, + { + "epoch": 1.7171717171717171, + "eval_loss": 1.6737746000289917, + "eval_runtime": 37.0666, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 1360 + }, + { + "epoch": 1.7184343434343434, + "grad_norm": 2.905977964401245, + "learning_rate": 2.4087932647333957e-06, + "loss": 1.4378, + "step": 1361 + }, + { + "epoch": 1.7196969696969697, + "grad_norm": 2.9268195629119873, + "learning_rate": 2.4064546304957905e-06, + "loss": 1.4765, + "step": 1362 + }, + { + "epoch": 1.720959595959596, + "grad_norm": 3.1369476318359375, + "learning_rate": 2.4041159962581854e-06, + "loss": 1.4532, + "step": 1363 + }, + { + "epoch": 1.7222222222222223, + "grad_norm": 3.04546856880188, + "learning_rate": 2.4017773620205802e-06, + "loss": 1.4921, + "step": 1364 + }, + { + "epoch": 1.7234848484848486, + "grad_norm": 2.872852325439453, + "learning_rate": 2.399438727782975e-06, + "loss": 1.4323, + "step": 1365 + }, + { + "epoch": 1.7247474747474747, + "grad_norm": 2.961681604385376, + "learning_rate": 2.39710009354537e-06, + "loss": 1.5181, + "step": 1366 + }, + { + "epoch": 1.726010101010101, + "grad_norm": 2.997729778289795, + "learning_rate": 2.3947614593077643e-06, + "loss": 1.4817, + "step": 1367 + }, + { + "epoch": 1.7272727272727273, + "grad_norm": 2.9097466468811035, + "learning_rate": 2.392422825070159e-06, + "loss": 1.4478, + "step": 1368 + }, + { + "epoch": 1.7285353535353534, + "grad_norm": 2.82433819770813, + "learning_rate": 2.390084190832554e-06, + "loss": 1.4065, + "step": 1369 + }, + { + "epoch": 1.7297979797979797, + "grad_norm": 2.940910577774048, + "learning_rate": 2.387745556594949e-06, + "loss": 1.4735, + "step": 1370 + }, + { + "epoch": 1.7297979797979797, + "eval_loss": 1.6717814207077026, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1370 + }, + { + "epoch": 1.731060606060606, + "grad_norm": 2.79101300239563, + "learning_rate": 2.3854069223573433e-06, + "loss": 1.4545, + "step": 1371 + }, + { + "epoch": 1.7323232323232323, + "grad_norm": 2.916954517364502, + "learning_rate": 2.3830682881197385e-06, + "loss": 1.4293, + "step": 1372 + }, + { + "epoch": 1.7335858585858586, + "grad_norm": 2.9609909057617188, + "learning_rate": 2.380729653882133e-06, + "loss": 1.4578, + "step": 1373 + }, + { + "epoch": 1.7348484848484849, + "grad_norm": 2.935925245285034, + "learning_rate": 2.378391019644528e-06, + "loss": 1.4577, + "step": 1374 + }, + { + "epoch": 1.7361111111111112, + "grad_norm": 2.9803526401519775, + "learning_rate": 2.3760523854069226e-06, + "loss": 1.4449, + "step": 1375 + }, + { + "epoch": 1.7373737373737375, + "grad_norm": 3.0292913913726807, + "learning_rate": 2.3737137511693175e-06, + "loss": 1.4308, + "step": 1376 + }, + { + "epoch": 1.7386363636363638, + "grad_norm": 3.0247879028320312, + "learning_rate": 2.371375116931712e-06, + "loss": 1.5406, + "step": 1377 + }, + { + "epoch": 1.73989898989899, + "grad_norm": 2.86775541305542, + "learning_rate": 2.3690364826941067e-06, + "loss": 1.4324, + "step": 1378 + }, + { + "epoch": 1.7411616161616161, + "grad_norm": 2.8593552112579346, + "learning_rate": 2.3666978484565016e-06, + "loss": 1.4526, + "step": 1379 + }, + { + "epoch": 1.7424242424242424, + "grad_norm": 2.8735320568084717, + "learning_rate": 2.3643592142188964e-06, + "loss": 1.4097, + "step": 1380 + }, + { + "epoch": 1.7424242424242424, + "eval_loss": 1.675827145576477, + "eval_runtime": 37.1206, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1380 + }, + { + "epoch": 1.7436868686868687, + "grad_norm": 2.948153495788574, + "learning_rate": 2.3620205799812913e-06, + "loss": 1.4234, + "step": 1381 + }, + { + "epoch": 1.7449494949494948, + "grad_norm": 2.9344282150268555, + "learning_rate": 2.359681945743686e-06, + "loss": 1.4653, + "step": 1382 + }, + { + "epoch": 1.746212121212121, + "grad_norm": 2.9309439659118652, + "learning_rate": 2.3573433115060805e-06, + "loss": 1.4676, + "step": 1383 + }, + { + "epoch": 1.7474747474747474, + "grad_norm": 3.0762970447540283, + "learning_rate": 2.3550046772684754e-06, + "loss": 1.4407, + "step": 1384 + }, + { + "epoch": 1.7487373737373737, + "grad_norm": 3.149824857711792, + "learning_rate": 2.3526660430308702e-06, + "loss": 1.4629, + "step": 1385 + }, + { + "epoch": 1.75, + "grad_norm": 3.109571695327759, + "learning_rate": 2.3503274087932646e-06, + "loss": 1.494, + "step": 1386 + }, + { + "epoch": 1.7512626262626263, + "grad_norm": 2.8359758853912354, + "learning_rate": 2.34798877455566e-06, + "loss": 1.4056, + "step": 1387 + }, + { + "epoch": 1.7525252525252526, + "grad_norm": 2.8523781299591064, + "learning_rate": 2.3456501403180548e-06, + "loss": 1.4389, + "step": 1388 + }, + { + "epoch": 1.753787878787879, + "grad_norm": 3.0218093395233154, + "learning_rate": 2.343311506080449e-06, + "loss": 1.4587, + "step": 1389 + }, + { + "epoch": 1.7550505050505052, + "grad_norm": 3.0975496768951416, + "learning_rate": 2.340972871842844e-06, + "loss": 1.5133, + "step": 1390 + }, + { + "epoch": 1.7550505050505052, + "eval_loss": 1.671931266784668, + "eval_runtime": 37.2749, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 1390 + }, + { + "epoch": 1.7563131313131313, + "grad_norm": 3.0177836418151855, + "learning_rate": 2.338634237605239e-06, + "loss": 1.4716, + "step": 1391 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 3.049105167388916, + "learning_rate": 2.3362956033676333e-06, + "loss": 1.4858, + "step": 1392 + }, + { + "epoch": 1.7588383838383839, + "grad_norm": 3.0062026977539062, + "learning_rate": 2.333956969130028e-06, + "loss": 1.4239, + "step": 1393 + }, + { + "epoch": 1.76010101010101, + "grad_norm": 3.071772813796997, + "learning_rate": 2.331618334892423e-06, + "loss": 1.4654, + "step": 1394 + }, + { + "epoch": 1.7613636363636362, + "grad_norm": 3.140510320663452, + "learning_rate": 2.329279700654818e-06, + "loss": 1.4854, + "step": 1395 + }, + { + "epoch": 1.7626262626262625, + "grad_norm": 2.9903907775878906, + "learning_rate": 2.3269410664172126e-06, + "loss": 1.4406, + "step": 1396 + }, + { + "epoch": 1.7638888888888888, + "grad_norm": 2.839891195297241, + "learning_rate": 2.3246024321796075e-06, + "loss": 1.4498, + "step": 1397 + }, + { + "epoch": 1.7651515151515151, + "grad_norm": 3.0252437591552734, + "learning_rate": 2.322263797942002e-06, + "loss": 1.472, + "step": 1398 + }, + { + "epoch": 1.7664141414141414, + "grad_norm": 3.0214762687683105, + "learning_rate": 2.3199251637043968e-06, + "loss": 1.4031, + "step": 1399 + }, + { + "epoch": 1.7676767676767677, + "grad_norm": 3.190088987350464, + "learning_rate": 2.3175865294667916e-06, + "loss": 1.4497, + "step": 1400 + }, + { + "epoch": 1.7676767676767677, + "eval_loss": 1.673120379447937, + "eval_runtime": 37.4414, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1400 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 4.364233173106688e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1400/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/latest new file mode 100644 index 0000000000000000000000000000000000000000..cb24338af451be6d3ccddc18a950b58a70589761 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/latest @@ -0,0 +1 @@ +global_step1700 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..b9efe4d3710651a7fc1e0d36a832d96359df8487 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/trainer_state.json @@ -0,0 +1,13293 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.1464646464646466, + "eval_steps": 10, + "global_step": 1700, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + }, + { + "epoch": 1.5164141414141414, + "grad_norm": 3.041095733642578, + "learning_rate": 2.782974742750234e-06, + "loss": 1.4736, + "step": 1201 + }, + { + "epoch": 1.5176767676767677, + "grad_norm": 2.8400142192840576, + "learning_rate": 2.780636108512629e-06, + "loss": 1.4169, + "step": 1202 + }, + { + "epoch": 1.518939393939394, + "grad_norm": 2.785592555999756, + "learning_rate": 2.7782974742750236e-06, + "loss": 1.4553, + "step": 1203 + }, + { + "epoch": 1.5202020202020203, + "grad_norm": 2.75209641456604, + "learning_rate": 2.775958840037418e-06, + "loss": 1.4267, + "step": 1204 + }, + { + "epoch": 1.5214646464646466, + "grad_norm": 2.927558183670044, + "learning_rate": 2.773620205799813e-06, + "loss": 1.4377, + "step": 1205 + }, + { + "epoch": 1.5227272727272727, + "grad_norm": 2.9335508346557617, + "learning_rate": 2.7712815715622077e-06, + "loss": 1.4295, + "step": 1206 + }, + { + "epoch": 1.523989898989899, + "grad_norm": 2.7812745571136475, + "learning_rate": 2.7689429373246026e-06, + "loss": 1.4029, + "step": 1207 + }, + { + "epoch": 1.5252525252525253, + "grad_norm": 3.230062961578369, + "learning_rate": 2.766604303086998e-06, + "loss": 1.539, + "step": 1208 + }, + { + "epoch": 1.5265151515151514, + "grad_norm": 3.1244332790374756, + "learning_rate": 2.7642656688493923e-06, + "loss": 1.4829, + "step": 1209 + }, + { + "epoch": 1.5277777777777777, + "grad_norm": 3.2391650676727295, + "learning_rate": 2.761927034611787e-06, + "loss": 1.4677, + "step": 1210 + }, + { + "epoch": 1.5277777777777777, + "eval_loss": 1.6849907636642456, + "eval_runtime": 36.8638, + "eval_samples_per_second": 27.127, + "eval_steps_per_second": 1.139, + "step": 1210 + }, + { + "epoch": 1.529040404040404, + "grad_norm": 3.1884987354278564, + "learning_rate": 2.759588400374182e-06, + "loss": 1.4185, + "step": 1211 + }, + { + "epoch": 1.5303030303030303, + "grad_norm": 2.959118366241455, + "learning_rate": 2.7572497661365764e-06, + "loss": 1.4536, + "step": 1212 + }, + { + "epoch": 1.5315656565656566, + "grad_norm": 2.882380247116089, + "learning_rate": 2.7549111318989712e-06, + "loss": 1.4872, + "step": 1213 + }, + { + "epoch": 1.5328282828282829, + "grad_norm": 3.0574429035186768, + "learning_rate": 2.752572497661366e-06, + "loss": 1.4343, + "step": 1214 + }, + { + "epoch": 1.5340909090909092, + "grad_norm": 3.1645233631134033, + "learning_rate": 2.750233863423761e-06, + "loss": 1.5101, + "step": 1215 + }, + { + "epoch": 1.5353535353535355, + "grad_norm": 2.83034086227417, + "learning_rate": 2.7478952291861553e-06, + "loss": 1.45, + "step": 1216 + }, + { + "epoch": 1.5366161616161618, + "grad_norm": 2.964780330657959, + "learning_rate": 2.74555659494855e-06, + "loss": 1.453, + "step": 1217 + }, + { + "epoch": 1.5378787878787878, + "grad_norm": 2.9896774291992188, + "learning_rate": 2.743217960710945e-06, + "loss": 1.4648, + "step": 1218 + }, + { + "epoch": 1.5391414141414141, + "grad_norm": 3.013664484024048, + "learning_rate": 2.74087932647334e-06, + "loss": 1.4828, + "step": 1219 + }, + { + "epoch": 1.5404040404040404, + "grad_norm": 3.0011773109436035, + "learning_rate": 2.7385406922357343e-06, + "loss": 1.4705, + "step": 1220 + }, + { + "epoch": 1.5404040404040404, + "eval_loss": 1.6872731447219849, + "eval_runtime": 36.9864, + "eval_samples_per_second": 27.037, + "eval_steps_per_second": 1.136, + "step": 1220 + }, + { + "epoch": 1.5416666666666665, + "grad_norm": 3.239349842071533, + "learning_rate": 2.736202057998129e-06, + "loss": 1.484, + "step": 1221 + }, + { + "epoch": 1.5429292929292928, + "grad_norm": 2.9267003536224365, + "learning_rate": 2.733863423760524e-06, + "loss": 1.4587, + "step": 1222 + }, + { + "epoch": 1.5441919191919191, + "grad_norm": 2.890094518661499, + "learning_rate": 2.7315247895229192e-06, + "loss": 1.4456, + "step": 1223 + }, + { + "epoch": 1.5454545454545454, + "grad_norm": 3.0287511348724365, + "learning_rate": 2.7291861552853136e-06, + "loss": 1.5059, + "step": 1224 + }, + { + "epoch": 1.5467171717171717, + "grad_norm": 3.041609525680542, + "learning_rate": 2.7268475210477085e-06, + "loss": 1.4341, + "step": 1225 + }, + { + "epoch": 1.547979797979798, + "grad_norm": 2.8771862983703613, + "learning_rate": 2.7245088868101033e-06, + "loss": 1.4308, + "step": 1226 + }, + { + "epoch": 1.5492424242424243, + "grad_norm": 3.1173794269561768, + "learning_rate": 2.722170252572498e-06, + "loss": 1.514, + "step": 1227 + }, + { + "epoch": 1.5505050505050506, + "grad_norm": 2.838275671005249, + "learning_rate": 2.7198316183348926e-06, + "loss": 1.4798, + "step": 1228 + }, + { + "epoch": 1.551767676767677, + "grad_norm": 2.9917166233062744, + "learning_rate": 2.7174929840972874e-06, + "loss": 1.5256, + "step": 1229 + }, + { + "epoch": 1.553030303030303, + "grad_norm": 3.071840763092041, + "learning_rate": 2.7151543498596823e-06, + "loss": 1.5187, + "step": 1230 + }, + { + "epoch": 1.553030303030303, + "eval_loss": 1.6850661039352417, + "eval_runtime": 37.0094, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 1230 + }, + { + "epoch": 1.5542929292929293, + "grad_norm": 3.253169536590576, + "learning_rate": 2.7128157156220767e-06, + "loss": 1.5522, + "step": 1231 + }, + { + "epoch": 1.5555555555555556, + "grad_norm": 3.037151336669922, + "learning_rate": 2.7104770813844715e-06, + "loss": 1.5125, + "step": 1232 + }, + { + "epoch": 1.5568181818181817, + "grad_norm": 2.8486485481262207, + "learning_rate": 2.7081384471468664e-06, + "loss": 1.4169, + "step": 1233 + }, + { + "epoch": 1.558080808080808, + "grad_norm": 2.953824996948242, + "learning_rate": 2.7057998129092612e-06, + "loss": 1.4505, + "step": 1234 + }, + { + "epoch": 1.5593434343434343, + "grad_norm": 3.012303590774536, + "learning_rate": 2.7034611786716557e-06, + "loss": 1.4816, + "step": 1235 + }, + { + "epoch": 1.5606060606060606, + "grad_norm": 3.1798768043518066, + "learning_rate": 2.7011225444340505e-06, + "loss": 1.5116, + "step": 1236 + }, + { + "epoch": 1.5618686868686869, + "grad_norm": 3.11370587348938, + "learning_rate": 2.6987839101964453e-06, + "loss": 1.5035, + "step": 1237 + }, + { + "epoch": 1.5631313131313131, + "grad_norm": 3.0563342571258545, + "learning_rate": 2.69644527595884e-06, + "loss": 1.4387, + "step": 1238 + }, + { + "epoch": 1.5643939393939394, + "grad_norm": 2.9884026050567627, + "learning_rate": 2.6941066417212354e-06, + "loss": 1.428, + "step": 1239 + }, + { + "epoch": 1.5656565656565657, + "grad_norm": 2.955773115158081, + "learning_rate": 2.69176800748363e-06, + "loss": 1.4208, + "step": 1240 + }, + { + "epoch": 1.5656565656565657, + "eval_loss": 1.685523509979248, + "eval_runtime": 37.0248, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1240 + }, + { + "epoch": 1.566919191919192, + "grad_norm": 2.9858005046844482, + "learning_rate": 2.6894293732460247e-06, + "loss": 1.4706, + "step": 1241 + }, + { + "epoch": 1.5681818181818183, + "grad_norm": 3.122340440750122, + "learning_rate": 2.6870907390084196e-06, + "loss": 1.4892, + "step": 1242 + }, + { + "epoch": 1.5694444444444444, + "grad_norm": 2.9828262329101562, + "learning_rate": 2.684752104770814e-06, + "loss": 1.4255, + "step": 1243 + }, + { + "epoch": 1.5707070707070707, + "grad_norm": 2.916653633117676, + "learning_rate": 2.682413470533209e-06, + "loss": 1.4241, + "step": 1244 + }, + { + "epoch": 1.571969696969697, + "grad_norm": 3.1615540981292725, + "learning_rate": 2.6800748362956037e-06, + "loss": 1.4719, + "step": 1245 + }, + { + "epoch": 1.573232323232323, + "grad_norm": 3.154130697250366, + "learning_rate": 2.6777362020579985e-06, + "loss": 1.4731, + "step": 1246 + }, + { + "epoch": 1.5744949494949494, + "grad_norm": 3.040848731994629, + "learning_rate": 2.675397567820393e-06, + "loss": 1.5149, + "step": 1247 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.9808318614959717, + "learning_rate": 2.6730589335827878e-06, + "loss": 1.431, + "step": 1248 + }, + { + "epoch": 1.577020202020202, + "grad_norm": 2.8538382053375244, + "learning_rate": 2.6707202993451826e-06, + "loss": 1.4458, + "step": 1249 + }, + { + "epoch": 1.5782828282828283, + "grad_norm": 3.113945960998535, + "learning_rate": 2.6683816651075774e-06, + "loss": 1.4873, + "step": 1250 + }, + { + "epoch": 1.5782828282828283, + "eval_loss": 1.6823841333389282, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1250 + }, + { + "epoch": 1.5795454545454546, + "grad_norm": 3.0044517517089844, + "learning_rate": 2.666043030869972e-06, + "loss": 1.4904, + "step": 1251 + }, + { + "epoch": 1.5808080808080809, + "grad_norm": 2.845820665359497, + "learning_rate": 2.6637043966323667e-06, + "loss": 1.4174, + "step": 1252 + }, + { + "epoch": 1.5820707070707072, + "grad_norm": 2.9123542308807373, + "learning_rate": 2.6613657623947616e-06, + "loss": 1.4896, + "step": 1253 + }, + { + "epoch": 1.5833333333333335, + "grad_norm": 3.1593222618103027, + "learning_rate": 2.659027128157157e-06, + "loss": 1.5208, + "step": 1254 + }, + { + "epoch": 1.5845959595959596, + "grad_norm": 3.0782854557037354, + "learning_rate": 2.6566884939195512e-06, + "loss": 1.4911, + "step": 1255 + }, + { + "epoch": 1.5858585858585859, + "grad_norm": 2.77309513092041, + "learning_rate": 2.654349859681946e-06, + "loss": 1.4475, + "step": 1256 + }, + { + "epoch": 1.5871212121212122, + "grad_norm": 2.9154622554779053, + "learning_rate": 2.652011225444341e-06, + "loss": 1.4324, + "step": 1257 + }, + { + "epoch": 1.5883838383838382, + "grad_norm": 3.107395887374878, + "learning_rate": 2.6496725912067358e-06, + "loss": 1.5126, + "step": 1258 + }, + { + "epoch": 1.5896464646464645, + "grad_norm": 2.825025796890259, + "learning_rate": 2.64733395696913e-06, + "loss": 1.4723, + "step": 1259 + }, + { + "epoch": 1.5909090909090908, + "grad_norm": 2.833296060562134, + "learning_rate": 2.644995322731525e-06, + "loss": 1.4516, + "step": 1260 + }, + { + "epoch": 1.5909090909090908, + "eval_loss": 1.6950432062149048, + "eval_runtime": 37.0343, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1260 + }, + { + "epoch": 1.5921717171717171, + "grad_norm": 2.9508421421051025, + "learning_rate": 2.64265668849392e-06, + "loss": 1.4002, + "step": 1261 + }, + { + "epoch": 1.5934343434343434, + "grad_norm": 2.8778491020202637, + "learning_rate": 2.6403180542563143e-06, + "loss": 1.4258, + "step": 1262 + }, + { + "epoch": 1.5946969696969697, + "grad_norm": 2.81261944770813, + "learning_rate": 2.637979420018709e-06, + "loss": 1.435, + "step": 1263 + }, + { + "epoch": 1.595959595959596, + "grad_norm": 3.1811511516571045, + "learning_rate": 2.635640785781104e-06, + "loss": 1.4792, + "step": 1264 + }, + { + "epoch": 1.5972222222222223, + "grad_norm": 3.0168304443359375, + "learning_rate": 2.633302151543499e-06, + "loss": 1.5394, + "step": 1265 + }, + { + "epoch": 1.5984848484848486, + "grad_norm": 2.9531311988830566, + "learning_rate": 2.6309635173058932e-06, + "loss": 1.4775, + "step": 1266 + }, + { + "epoch": 1.5997474747474747, + "grad_norm": 2.835839033126831, + "learning_rate": 2.628624883068288e-06, + "loss": 1.478, + "step": 1267 + }, + { + "epoch": 1.601010101010101, + "grad_norm": 2.9014039039611816, + "learning_rate": 2.626286248830683e-06, + "loss": 1.4926, + "step": 1268 + }, + { + "epoch": 1.6022727272727273, + "grad_norm": 2.89461088180542, + "learning_rate": 2.623947614593078e-06, + "loss": 1.448, + "step": 1269 + }, + { + "epoch": 1.6035353535353534, + "grad_norm": 2.8005120754241943, + "learning_rate": 2.621608980355473e-06, + "loss": 1.4733, + "step": 1270 + }, + { + "epoch": 1.6035353535353534, + "eval_loss": 1.6889415979385376, + "eval_runtime": 37.0345, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1270 + }, + { + "epoch": 1.6047979797979797, + "grad_norm": 2.9576175212860107, + "learning_rate": 2.6192703461178675e-06, + "loss": 1.5116, + "step": 1271 + }, + { + "epoch": 1.606060606060606, + "grad_norm": 2.926638603210449, + "learning_rate": 2.6169317118802623e-06, + "loss": 1.4744, + "step": 1272 + }, + { + "epoch": 1.6073232323232323, + "grad_norm": 2.911587715148926, + "learning_rate": 2.614593077642657e-06, + "loss": 1.4695, + "step": 1273 + }, + { + "epoch": 1.6085858585858586, + "grad_norm": 2.8878750801086426, + "learning_rate": 2.6122544434050516e-06, + "loss": 1.4574, + "step": 1274 + }, + { + "epoch": 1.6098484848484849, + "grad_norm": 2.8489067554473877, + "learning_rate": 2.6099158091674464e-06, + "loss": 1.4599, + "step": 1275 + }, + { + "epoch": 1.6111111111111112, + "grad_norm": 2.8223719596862793, + "learning_rate": 2.6075771749298412e-06, + "loss": 1.4431, + "step": 1276 + }, + { + "epoch": 1.6123737373737375, + "grad_norm": 2.869910955429077, + "learning_rate": 2.605238540692236e-06, + "loss": 1.48, + "step": 1277 + }, + { + "epoch": 1.6136363636363638, + "grad_norm": 2.8435168266296387, + "learning_rate": 2.6028999064546305e-06, + "loss": 1.4471, + "step": 1278 + }, + { + "epoch": 1.61489898989899, + "grad_norm": 2.9384665489196777, + "learning_rate": 2.6005612722170253e-06, + "loss": 1.5176, + "step": 1279 + }, + { + "epoch": 1.6161616161616161, + "grad_norm": 3.021785020828247, + "learning_rate": 2.59822263797942e-06, + "loss": 1.4789, + "step": 1280 + }, + { + "epoch": 1.6161616161616161, + "eval_loss": 1.6864516735076904, + "eval_runtime": 37.0249, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1280 + }, + { + "epoch": 1.6174242424242424, + "grad_norm": 3.007183313369751, + "learning_rate": 2.5958840037418146e-06, + "loss": 1.4686, + "step": 1281 + }, + { + "epoch": 1.6186868686868687, + "grad_norm": 3.115910053253174, + "learning_rate": 2.5935453695042095e-06, + "loss": 1.5382, + "step": 1282 + }, + { + "epoch": 1.6199494949494948, + "grad_norm": 2.98108172416687, + "learning_rate": 2.5912067352666043e-06, + "loss": 1.4569, + "step": 1283 + }, + { + "epoch": 1.621212121212121, + "grad_norm": 2.926832437515259, + "learning_rate": 2.5888681010289996e-06, + "loss": 1.4515, + "step": 1284 + }, + { + "epoch": 1.6224747474747474, + "grad_norm": 2.9045534133911133, + "learning_rate": 2.5865294667913944e-06, + "loss": 1.4259, + "step": 1285 + }, + { + "epoch": 1.6237373737373737, + "grad_norm": 3.0757768154144287, + "learning_rate": 2.584190832553789e-06, + "loss": 1.5436, + "step": 1286 + }, + { + "epoch": 1.625, + "grad_norm": 3.045257806777954, + "learning_rate": 2.5818521983161837e-06, + "loss": 1.4378, + "step": 1287 + }, + { + "epoch": 1.6262626262626263, + "grad_norm": 2.8131601810455322, + "learning_rate": 2.5795135640785785e-06, + "loss": 1.4224, + "step": 1288 + }, + { + "epoch": 1.6275252525252526, + "grad_norm": 2.953326940536499, + "learning_rate": 2.5771749298409734e-06, + "loss": 1.4137, + "step": 1289 + }, + { + "epoch": 1.628787878787879, + "grad_norm": 2.946232318878174, + "learning_rate": 2.5748362956033678e-06, + "loss": 1.4575, + "step": 1290 + }, + { + "epoch": 1.628787878787879, + "eval_loss": 1.6896334886550903, + "eval_runtime": 37.317, + "eval_samples_per_second": 26.797, + "eval_steps_per_second": 1.125, + "step": 1290 + }, + { + "epoch": 1.6300505050505052, + "grad_norm": 2.860867738723755, + "learning_rate": 2.5724976613657626e-06, + "loss": 1.4412, + "step": 1291 + }, + { + "epoch": 1.6313131313131313, + "grad_norm": 3.0417261123657227, + "learning_rate": 2.5701590271281575e-06, + "loss": 1.4919, + "step": 1292 + }, + { + "epoch": 1.6325757575757576, + "grad_norm": 3.21425199508667, + "learning_rate": 2.567820392890552e-06, + "loss": 1.5243, + "step": 1293 + }, + { + "epoch": 1.6338383838383839, + "grad_norm": 3.000302314758301, + "learning_rate": 2.5654817586529467e-06, + "loss": 1.4571, + "step": 1294 + }, + { + "epoch": 1.63510101010101, + "grad_norm": 2.88415789604187, + "learning_rate": 2.5631431244153416e-06, + "loss": 1.4527, + "step": 1295 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.7747156620025635, + "learning_rate": 2.5608044901777364e-06, + "loss": 1.3984, + "step": 1296 + }, + { + "epoch": 1.6376262626262625, + "grad_norm": 2.9298319816589355, + "learning_rate": 2.558465855940131e-06, + "loss": 1.4587, + "step": 1297 + }, + { + "epoch": 1.6388888888888888, + "grad_norm": 3.0199968814849854, + "learning_rate": 2.5561272217025257e-06, + "loss": 1.4539, + "step": 1298 + }, + { + "epoch": 1.6401515151515151, + "grad_norm": 3.09503436088562, + "learning_rate": 2.5537885874649205e-06, + "loss": 1.4551, + "step": 1299 + }, + { + "epoch": 1.6414141414141414, + "grad_norm": 2.972161293029785, + "learning_rate": 2.5514499532273158e-06, + "loss": 1.4123, + "step": 1300 + }, + { + "epoch": 1.6414141414141414, + "eval_loss": 1.6877377033233643, + "eval_runtime": 37.4088, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1300 + }, + { + "epoch": 1.6426767676767677, + "grad_norm": 3.081058979034424, + "learning_rate": 2.5491113189897106e-06, + "loss": 1.457, + "step": 1301 + }, + { + "epoch": 1.643939393939394, + "grad_norm": 3.040106773376465, + "learning_rate": 2.546772684752105e-06, + "loss": 1.4465, + "step": 1302 + }, + { + "epoch": 1.6452020202020203, + "grad_norm": 3.133812427520752, + "learning_rate": 2.5444340505145e-06, + "loss": 1.4382, + "step": 1303 + }, + { + "epoch": 1.6464646464646466, + "grad_norm": 2.9701125621795654, + "learning_rate": 2.5420954162768947e-06, + "loss": 1.457, + "step": 1304 + }, + { + "epoch": 1.6477272727272727, + "grad_norm": 2.9272050857543945, + "learning_rate": 2.539756782039289e-06, + "loss": 1.464, + "step": 1305 + }, + { + "epoch": 1.648989898989899, + "grad_norm": 3.048401117324829, + "learning_rate": 2.537418147801684e-06, + "loss": 1.4688, + "step": 1306 + }, + { + "epoch": 1.6502525252525253, + "grad_norm": 2.9103102684020996, + "learning_rate": 2.535079513564079e-06, + "loss": 1.5025, + "step": 1307 + }, + { + "epoch": 1.6515151515151514, + "grad_norm": 3.0841548442840576, + "learning_rate": 2.5327408793264737e-06, + "loss": 1.4938, + "step": 1308 + }, + { + "epoch": 1.6527777777777777, + "grad_norm": 3.022184371948242, + "learning_rate": 2.530402245088868e-06, + "loss": 1.4332, + "step": 1309 + }, + { + "epoch": 1.654040404040404, + "grad_norm": 2.881479024887085, + "learning_rate": 2.528063610851263e-06, + "loss": 1.4541, + "step": 1310 + }, + { + "epoch": 1.654040404040404, + "eval_loss": 1.6814675331115723, + "eval_runtime": 36.9446, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.6553030303030303, + "grad_norm": 2.906719446182251, + "learning_rate": 2.5257249766136578e-06, + "loss": 1.4968, + "step": 1311 + }, + { + "epoch": 1.6565656565656566, + "grad_norm": 3.0566461086273193, + "learning_rate": 2.523386342376052e-06, + "loss": 1.4513, + "step": 1312 + }, + { + "epoch": 1.6578282828282829, + "grad_norm": 2.895026206970215, + "learning_rate": 2.521047708138447e-06, + "loss": 1.4217, + "step": 1313 + }, + { + "epoch": 1.6590909090909092, + "grad_norm": 3.0006465911865234, + "learning_rate": 2.518709073900842e-06, + "loss": 1.4793, + "step": 1314 + }, + { + "epoch": 1.6603535353535355, + "grad_norm": 2.9853575229644775, + "learning_rate": 2.516370439663237e-06, + "loss": 1.4642, + "step": 1315 + }, + { + "epoch": 1.6616161616161618, + "grad_norm": 2.9729392528533936, + "learning_rate": 2.514031805425632e-06, + "loss": 1.4644, + "step": 1316 + }, + { + "epoch": 1.6628787878787878, + "grad_norm": 3.0997586250305176, + "learning_rate": 2.5116931711880264e-06, + "loss": 1.4392, + "step": 1317 + }, + { + "epoch": 1.6641414141414141, + "grad_norm": 2.9621379375457764, + "learning_rate": 2.5093545369504213e-06, + "loss": 1.4387, + "step": 1318 + }, + { + "epoch": 1.6654040404040404, + "grad_norm": 3.0571911334991455, + "learning_rate": 2.507015902712816e-06, + "loss": 1.4638, + "step": 1319 + }, + { + "epoch": 1.6666666666666665, + "grad_norm": 2.9580321311950684, + "learning_rate": 2.504677268475211e-06, + "loss": 1.4985, + "step": 1320 + }, + { + "epoch": 1.6666666666666665, + "eval_loss": 1.6790977716445923, + "eval_runtime": 37.0242, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1320 + }, + { + "epoch": 1.6679292929292928, + "grad_norm": 2.87432599067688, + "learning_rate": 2.5023386342376054e-06, + "loss": 1.4058, + "step": 1321 + }, + { + "epoch": 1.6691919191919191, + "grad_norm": 2.925546407699585, + "learning_rate": 2.5e-06, + "loss": 1.4986, + "step": 1322 + }, + { + "epoch": 1.6704545454545454, + "grad_norm": 2.837743043899536, + "learning_rate": 2.497661365762395e-06, + "loss": 1.455, + "step": 1323 + }, + { + "epoch": 1.6717171717171717, + "grad_norm": 3.111581563949585, + "learning_rate": 2.4953227315247895e-06, + "loss": 1.5005, + "step": 1324 + }, + { + "epoch": 1.672979797979798, + "grad_norm": 2.851076126098633, + "learning_rate": 2.4929840972871843e-06, + "loss": 1.4365, + "step": 1325 + }, + { + "epoch": 1.6742424242424243, + "grad_norm": 2.966965675354004, + "learning_rate": 2.4906454630495796e-06, + "loss": 1.4697, + "step": 1326 + }, + { + "epoch": 1.6755050505050506, + "grad_norm": 3.015193223953247, + "learning_rate": 2.488306828811974e-06, + "loss": 1.4486, + "step": 1327 + }, + { + "epoch": 1.676767676767677, + "grad_norm": 3.017681121826172, + "learning_rate": 2.485968194574369e-06, + "loss": 1.4017, + "step": 1328 + }, + { + "epoch": 1.678030303030303, + "grad_norm": 3.071981906890869, + "learning_rate": 2.4836295603367637e-06, + "loss": 1.4572, + "step": 1329 + }, + { + "epoch": 1.6792929292929293, + "grad_norm": 3.051741123199463, + "learning_rate": 2.481290926099158e-06, + "loss": 1.4498, + "step": 1330 + }, + { + "epoch": 1.6792929292929293, + "eval_loss": 1.6771777868270874, + "eval_runtime": 37.0747, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1330 + }, + { + "epoch": 1.6805555555555556, + "grad_norm": 2.9561071395874023, + "learning_rate": 2.478952291861553e-06, + "loss": 1.4428, + "step": 1331 + }, + { + "epoch": 1.6818181818181817, + "grad_norm": 2.992584466934204, + "learning_rate": 2.4766136576239478e-06, + "loss": 1.461, + "step": 1332 + }, + { + "epoch": 1.683080808080808, + "grad_norm": 3.0473921298980713, + "learning_rate": 2.4742750233863426e-06, + "loss": 1.489, + "step": 1333 + }, + { + "epoch": 1.6843434343434343, + "grad_norm": 2.8721725940704346, + "learning_rate": 2.4719363891487375e-06, + "loss": 1.487, + "step": 1334 + }, + { + "epoch": 1.6856060606060606, + "grad_norm": 2.8650765419006348, + "learning_rate": 2.4695977549111323e-06, + "loss": 1.4504, + "step": 1335 + }, + { + "epoch": 1.6868686868686869, + "grad_norm": 2.998887538909912, + "learning_rate": 2.4672591206735267e-06, + "loss": 1.4586, + "step": 1336 + }, + { + "epoch": 1.6881313131313131, + "grad_norm": 2.915188789367676, + "learning_rate": 2.4649204864359216e-06, + "loss": 1.4333, + "step": 1337 + }, + { + "epoch": 1.6893939393939394, + "grad_norm": 2.9660091400146484, + "learning_rate": 2.4625818521983164e-06, + "loss": 1.4711, + "step": 1338 + }, + { + "epoch": 1.6906565656565657, + "grad_norm": 2.916274070739746, + "learning_rate": 2.4602432179607113e-06, + "loss": 1.4642, + "step": 1339 + }, + { + "epoch": 1.691919191919192, + "grad_norm": 3.04154896736145, + "learning_rate": 2.4579045837231057e-06, + "loss": 1.4233, + "step": 1340 + }, + { + "epoch": 1.691919191919192, + "eval_loss": 1.6764302253723145, + "eval_runtime": 37.0686, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1340 + }, + { + "epoch": 1.6931818181818183, + "grad_norm": 2.9976465702056885, + "learning_rate": 2.4555659494855005e-06, + "loss": 1.4775, + "step": 1341 + }, + { + "epoch": 1.6944444444444444, + "grad_norm": 2.9140090942382812, + "learning_rate": 2.4532273152478954e-06, + "loss": 1.4532, + "step": 1342 + }, + { + "epoch": 1.6957070707070707, + "grad_norm": 3.0224454402923584, + "learning_rate": 2.45088868101029e-06, + "loss": 1.5091, + "step": 1343 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.8979439735412598, + "learning_rate": 2.448550046772685e-06, + "loss": 1.4634, + "step": 1344 + }, + { + "epoch": 1.698232323232323, + "grad_norm": 2.913433313369751, + "learning_rate": 2.44621141253508e-06, + "loss": 1.4483, + "step": 1345 + }, + { + "epoch": 1.6994949494949494, + "grad_norm": 2.813455820083618, + "learning_rate": 2.4438727782974743e-06, + "loss": 1.446, + "step": 1346 + }, + { + "epoch": 1.7007575757575757, + "grad_norm": 2.988722324371338, + "learning_rate": 2.441534144059869e-06, + "loss": 1.4908, + "step": 1347 + }, + { + "epoch": 1.702020202020202, + "grad_norm": 2.850446939468384, + "learning_rate": 2.439195509822264e-06, + "loss": 1.4221, + "step": 1348 + }, + { + "epoch": 1.7032828282828283, + "grad_norm": 2.8405508995056152, + "learning_rate": 2.436856875584659e-06, + "loss": 1.4496, + "step": 1349 + }, + { + "epoch": 1.7045454545454546, + "grad_norm": 3.0188651084899902, + "learning_rate": 2.4345182413470537e-06, + "loss": 1.489, + "step": 1350 + }, + { + "epoch": 1.7045454545454546, + "eval_loss": 1.6780083179473877, + "eval_runtime": 37.0724, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1350 + }, + { + "epoch": 1.7058080808080809, + "grad_norm": 2.905088424682617, + "learning_rate": 2.4321796071094485e-06, + "loss": 1.4283, + "step": 1351 + }, + { + "epoch": 1.7070707070707072, + "grad_norm": 2.878998279571533, + "learning_rate": 2.429840972871843e-06, + "loss": 1.4357, + "step": 1352 + }, + { + "epoch": 1.7083333333333335, + "grad_norm": 3.1707355976104736, + "learning_rate": 2.427502338634238e-06, + "loss": 1.496, + "step": 1353 + }, + { + "epoch": 1.7095959595959596, + "grad_norm": 2.9610493183135986, + "learning_rate": 2.4251637043966326e-06, + "loss": 1.4905, + "step": 1354 + }, + { + "epoch": 1.7108585858585859, + "grad_norm": 2.8372201919555664, + "learning_rate": 2.422825070159027e-06, + "loss": 1.4563, + "step": 1355 + }, + { + "epoch": 1.7121212121212122, + "grad_norm": 2.8741753101348877, + "learning_rate": 2.420486435921422e-06, + "loss": 1.4525, + "step": 1356 + }, + { + "epoch": 1.7133838383838382, + "grad_norm": 2.973726749420166, + "learning_rate": 2.418147801683817e-06, + "loss": 1.4878, + "step": 1357 + }, + { + "epoch": 1.7146464646464645, + "grad_norm": 3.05717396736145, + "learning_rate": 2.4158091674462116e-06, + "loss": 1.5402, + "step": 1358 + }, + { + "epoch": 1.7159090909090908, + "grad_norm": 2.883772373199463, + "learning_rate": 2.4134705332086064e-06, + "loss": 1.4329, + "step": 1359 + }, + { + "epoch": 1.7171717171717171, + "grad_norm": 2.8246653079986572, + "learning_rate": 2.4111318989710013e-06, + "loss": 1.4821, + "step": 1360 + }, + { + "epoch": 1.7171717171717171, + "eval_loss": 1.6737746000289917, + "eval_runtime": 37.0666, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 1360 + }, + { + "epoch": 1.7184343434343434, + "grad_norm": 2.905977964401245, + "learning_rate": 2.4087932647333957e-06, + "loss": 1.4378, + "step": 1361 + }, + { + "epoch": 1.7196969696969697, + "grad_norm": 2.9268195629119873, + "learning_rate": 2.4064546304957905e-06, + "loss": 1.4765, + "step": 1362 + }, + { + "epoch": 1.720959595959596, + "grad_norm": 3.1369476318359375, + "learning_rate": 2.4041159962581854e-06, + "loss": 1.4532, + "step": 1363 + }, + { + "epoch": 1.7222222222222223, + "grad_norm": 3.04546856880188, + "learning_rate": 2.4017773620205802e-06, + "loss": 1.4921, + "step": 1364 + }, + { + "epoch": 1.7234848484848486, + "grad_norm": 2.872852325439453, + "learning_rate": 2.399438727782975e-06, + "loss": 1.4323, + "step": 1365 + }, + { + "epoch": 1.7247474747474747, + "grad_norm": 2.961681604385376, + "learning_rate": 2.39710009354537e-06, + "loss": 1.5181, + "step": 1366 + }, + { + "epoch": 1.726010101010101, + "grad_norm": 2.997729778289795, + "learning_rate": 2.3947614593077643e-06, + "loss": 1.4817, + "step": 1367 + }, + { + "epoch": 1.7272727272727273, + "grad_norm": 2.9097466468811035, + "learning_rate": 2.392422825070159e-06, + "loss": 1.4478, + "step": 1368 + }, + { + "epoch": 1.7285353535353534, + "grad_norm": 2.82433819770813, + "learning_rate": 2.390084190832554e-06, + "loss": 1.4065, + "step": 1369 + }, + { + "epoch": 1.7297979797979797, + "grad_norm": 2.940910577774048, + "learning_rate": 2.387745556594949e-06, + "loss": 1.4735, + "step": 1370 + }, + { + "epoch": 1.7297979797979797, + "eval_loss": 1.6717814207077026, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1370 + }, + { + "epoch": 1.731060606060606, + "grad_norm": 2.79101300239563, + "learning_rate": 2.3854069223573433e-06, + "loss": 1.4545, + "step": 1371 + }, + { + "epoch": 1.7323232323232323, + "grad_norm": 2.916954517364502, + "learning_rate": 2.3830682881197385e-06, + "loss": 1.4293, + "step": 1372 + }, + { + "epoch": 1.7335858585858586, + "grad_norm": 2.9609909057617188, + "learning_rate": 2.380729653882133e-06, + "loss": 1.4578, + "step": 1373 + }, + { + "epoch": 1.7348484848484849, + "grad_norm": 2.935925245285034, + "learning_rate": 2.378391019644528e-06, + "loss": 1.4577, + "step": 1374 + }, + { + "epoch": 1.7361111111111112, + "grad_norm": 2.9803526401519775, + "learning_rate": 2.3760523854069226e-06, + "loss": 1.4449, + "step": 1375 + }, + { + "epoch": 1.7373737373737375, + "grad_norm": 3.0292913913726807, + "learning_rate": 2.3737137511693175e-06, + "loss": 1.4308, + "step": 1376 + }, + { + "epoch": 1.7386363636363638, + "grad_norm": 3.0247879028320312, + "learning_rate": 2.371375116931712e-06, + "loss": 1.5406, + "step": 1377 + }, + { + "epoch": 1.73989898989899, + "grad_norm": 2.86775541305542, + "learning_rate": 2.3690364826941067e-06, + "loss": 1.4324, + "step": 1378 + }, + { + "epoch": 1.7411616161616161, + "grad_norm": 2.8593552112579346, + "learning_rate": 2.3666978484565016e-06, + "loss": 1.4526, + "step": 1379 + }, + { + "epoch": 1.7424242424242424, + "grad_norm": 2.8735320568084717, + "learning_rate": 2.3643592142188964e-06, + "loss": 1.4097, + "step": 1380 + }, + { + "epoch": 1.7424242424242424, + "eval_loss": 1.675827145576477, + "eval_runtime": 37.1206, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1380 + }, + { + "epoch": 1.7436868686868687, + "grad_norm": 2.948153495788574, + "learning_rate": 2.3620205799812913e-06, + "loss": 1.4234, + "step": 1381 + }, + { + "epoch": 1.7449494949494948, + "grad_norm": 2.9344282150268555, + "learning_rate": 2.359681945743686e-06, + "loss": 1.4653, + "step": 1382 + }, + { + "epoch": 1.746212121212121, + "grad_norm": 2.9309439659118652, + "learning_rate": 2.3573433115060805e-06, + "loss": 1.4676, + "step": 1383 + }, + { + "epoch": 1.7474747474747474, + "grad_norm": 3.0762970447540283, + "learning_rate": 2.3550046772684754e-06, + "loss": 1.4407, + "step": 1384 + }, + { + "epoch": 1.7487373737373737, + "grad_norm": 3.149824857711792, + "learning_rate": 2.3526660430308702e-06, + "loss": 1.4629, + "step": 1385 + }, + { + "epoch": 1.75, + "grad_norm": 3.109571695327759, + "learning_rate": 2.3503274087932646e-06, + "loss": 1.494, + "step": 1386 + }, + { + "epoch": 1.7512626262626263, + "grad_norm": 2.8359758853912354, + "learning_rate": 2.34798877455566e-06, + "loss": 1.4056, + "step": 1387 + }, + { + "epoch": 1.7525252525252526, + "grad_norm": 2.8523781299591064, + "learning_rate": 2.3456501403180548e-06, + "loss": 1.4389, + "step": 1388 + }, + { + "epoch": 1.753787878787879, + "grad_norm": 3.0218093395233154, + "learning_rate": 2.343311506080449e-06, + "loss": 1.4587, + "step": 1389 + }, + { + "epoch": 1.7550505050505052, + "grad_norm": 3.0975496768951416, + "learning_rate": 2.340972871842844e-06, + "loss": 1.5133, + "step": 1390 + }, + { + "epoch": 1.7550505050505052, + "eval_loss": 1.671931266784668, + "eval_runtime": 37.2749, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 1390 + }, + { + "epoch": 1.7563131313131313, + "grad_norm": 3.0177836418151855, + "learning_rate": 2.338634237605239e-06, + "loss": 1.4716, + "step": 1391 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 3.049105167388916, + "learning_rate": 2.3362956033676333e-06, + "loss": 1.4858, + "step": 1392 + }, + { + "epoch": 1.7588383838383839, + "grad_norm": 3.0062026977539062, + "learning_rate": 2.333956969130028e-06, + "loss": 1.4239, + "step": 1393 + }, + { + "epoch": 1.76010101010101, + "grad_norm": 3.071772813796997, + "learning_rate": 2.331618334892423e-06, + "loss": 1.4654, + "step": 1394 + }, + { + "epoch": 1.7613636363636362, + "grad_norm": 3.140510320663452, + "learning_rate": 2.329279700654818e-06, + "loss": 1.4854, + "step": 1395 + }, + { + "epoch": 1.7626262626262625, + "grad_norm": 2.9903907775878906, + "learning_rate": 2.3269410664172126e-06, + "loss": 1.4406, + "step": 1396 + }, + { + "epoch": 1.7638888888888888, + "grad_norm": 2.839891195297241, + "learning_rate": 2.3246024321796075e-06, + "loss": 1.4498, + "step": 1397 + }, + { + "epoch": 1.7651515151515151, + "grad_norm": 3.0252437591552734, + "learning_rate": 2.322263797942002e-06, + "loss": 1.472, + "step": 1398 + }, + { + "epoch": 1.7664141414141414, + "grad_norm": 3.0214762687683105, + "learning_rate": 2.3199251637043968e-06, + "loss": 1.4031, + "step": 1399 + }, + { + "epoch": 1.7676767676767677, + "grad_norm": 3.190088987350464, + "learning_rate": 2.3175865294667916e-06, + "loss": 1.4497, + "step": 1400 + }, + { + "epoch": 1.7676767676767677, + "eval_loss": 1.673120379447937, + "eval_runtime": 37.4414, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1400 + }, + { + "epoch": 1.768939393939394, + "grad_norm": 2.9620201587677, + "learning_rate": 2.3152478952291864e-06, + "loss": 1.4149, + "step": 1401 + }, + { + "epoch": 1.7702020202020203, + "grad_norm": 3.112372875213623, + "learning_rate": 2.312909260991581e-06, + "loss": 1.4569, + "step": 1402 + }, + { + "epoch": 1.7714646464646466, + "grad_norm": 3.138857364654541, + "learning_rate": 2.310570626753976e-06, + "loss": 1.4819, + "step": 1403 + }, + { + "epoch": 1.7727272727272727, + "grad_norm": 3.0957937240600586, + "learning_rate": 2.3082319925163705e-06, + "loss": 1.4374, + "step": 1404 + }, + { + "epoch": 1.773989898989899, + "grad_norm": 3.014039993286133, + "learning_rate": 2.3058933582787654e-06, + "loss": 1.4262, + "step": 1405 + }, + { + "epoch": 1.7752525252525253, + "grad_norm": 2.8127005100250244, + "learning_rate": 2.3035547240411602e-06, + "loss": 1.4468, + "step": 1406 + }, + { + "epoch": 1.7765151515151514, + "grad_norm": 2.9180614948272705, + "learning_rate": 2.301216089803555e-06, + "loss": 1.4259, + "step": 1407 + }, + { + "epoch": 1.7777777777777777, + "grad_norm": 3.0376124382019043, + "learning_rate": 2.2988774555659495e-06, + "loss": 1.4628, + "step": 1408 + }, + { + "epoch": 1.779040404040404, + "grad_norm": 3.031740665435791, + "learning_rate": 2.2965388213283443e-06, + "loss": 1.5058, + "step": 1409 + }, + { + "epoch": 1.7803030303030303, + "grad_norm": 2.9876646995544434, + "learning_rate": 2.294200187090739e-06, + "loss": 1.4605, + "step": 1410 + }, + { + "epoch": 1.7803030303030303, + "eval_loss": 1.668077826499939, + "eval_runtime": 36.9734, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 1410 + }, + { + "epoch": 1.7815656565656566, + "grad_norm": 2.953744888305664, + "learning_rate": 2.291861552853134e-06, + "loss": 1.4257, + "step": 1411 + }, + { + "epoch": 1.7828282828282829, + "grad_norm": 3.049036741256714, + "learning_rate": 2.289522918615529e-06, + "loss": 1.4476, + "step": 1412 + }, + { + "epoch": 1.7840909090909092, + "grad_norm": 3.102116584777832, + "learning_rate": 2.2871842843779237e-06, + "loss": 1.4701, + "step": 1413 + }, + { + "epoch": 1.7853535353535355, + "grad_norm": 2.842970132827759, + "learning_rate": 2.284845650140318e-06, + "loss": 1.415, + "step": 1414 + }, + { + "epoch": 1.7866161616161618, + "grad_norm": 2.9386725425720215, + "learning_rate": 2.282507015902713e-06, + "loss": 1.4561, + "step": 1415 + }, + { + "epoch": 1.7878787878787878, + "grad_norm": 2.942713499069214, + "learning_rate": 2.280168381665108e-06, + "loss": 1.435, + "step": 1416 + }, + { + "epoch": 1.7891414141414141, + "grad_norm": 2.943105936050415, + "learning_rate": 2.2778297474275022e-06, + "loss": 1.4923, + "step": 1417 + }, + { + "epoch": 1.7904040404040404, + "grad_norm": 2.7094717025756836, + "learning_rate": 2.2754911131898975e-06, + "loss": 1.4457, + "step": 1418 + }, + { + "epoch": 1.7916666666666665, + "grad_norm": 2.7859625816345215, + "learning_rate": 2.2731524789522923e-06, + "loss": 1.3993, + "step": 1419 + }, + { + "epoch": 1.7929292929292928, + "grad_norm": 3.0923988819122314, + "learning_rate": 2.2708138447146868e-06, + "loss": 1.5054, + "step": 1420 + }, + { + "epoch": 1.7929292929292928, + "eval_loss": 1.6687692403793335, + "eval_runtime": 37.0567, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.7941919191919191, + "grad_norm": 2.849726676940918, + "learning_rate": 2.2684752104770816e-06, + "loss": 1.3978, + "step": 1421 + }, + { + "epoch": 1.7954545454545454, + "grad_norm": 3.028932809829712, + "learning_rate": 2.2661365762394764e-06, + "loss": 1.4786, + "step": 1422 + }, + { + "epoch": 1.7967171717171717, + "grad_norm": 2.973188638687134, + "learning_rate": 2.263797942001871e-06, + "loss": 1.4393, + "step": 1423 + }, + { + "epoch": 1.797979797979798, + "grad_norm": 2.884032964706421, + "learning_rate": 2.2614593077642657e-06, + "loss": 1.4197, + "step": 1424 + }, + { + "epoch": 1.7992424242424243, + "grad_norm": 2.9058549404144287, + "learning_rate": 2.2591206735266606e-06, + "loss": 1.4313, + "step": 1425 + }, + { + "epoch": 1.8005050505050506, + "grad_norm": 2.905841827392578, + "learning_rate": 2.2567820392890554e-06, + "loss": 1.439, + "step": 1426 + }, + { + "epoch": 1.801767676767677, + "grad_norm": 2.927028179168701, + "learning_rate": 2.2544434050514502e-06, + "loss": 1.4296, + "step": 1427 + }, + { + "epoch": 1.803030303030303, + "grad_norm": 2.965529203414917, + "learning_rate": 2.252104770813845e-06, + "loss": 1.4185, + "step": 1428 + }, + { + "epoch": 1.8042929292929293, + "grad_norm": 2.9494433403015137, + "learning_rate": 2.2497661365762395e-06, + "loss": 1.5087, + "step": 1429 + }, + { + "epoch": 1.8055555555555556, + "grad_norm": 3.0137765407562256, + "learning_rate": 2.2474275023386343e-06, + "loss": 1.4383, + "step": 1430 + }, + { + "epoch": 1.8055555555555556, + "eval_loss": 1.6728461980819702, + "eval_runtime": 37.1033, + "eval_samples_per_second": 26.952, + "eval_steps_per_second": 1.132, + "step": 1430 + }, + { + "epoch": 1.8068181818181817, + "grad_norm": 2.980062484741211, + "learning_rate": 2.245088868101029e-06, + "loss": 1.4333, + "step": 1431 + }, + { + "epoch": 1.808080808080808, + "grad_norm": 3.143759250640869, + "learning_rate": 2.242750233863424e-06, + "loss": 1.4676, + "step": 1432 + }, + { + "epoch": 1.8093434343434343, + "grad_norm": 3.0134518146514893, + "learning_rate": 2.240411599625819e-06, + "loss": 1.4223, + "step": 1433 + }, + { + "epoch": 1.8106060606060606, + "grad_norm": 3.000018358230591, + "learning_rate": 2.2380729653882137e-06, + "loss": 1.4537, + "step": 1434 + }, + { + "epoch": 1.8118686868686869, + "grad_norm": 2.9828288555145264, + "learning_rate": 2.235734331150608e-06, + "loss": 1.4379, + "step": 1435 + }, + { + "epoch": 1.8131313131313131, + "grad_norm": 2.965486764907837, + "learning_rate": 2.233395696913003e-06, + "loss": 1.4244, + "step": 1436 + }, + { + "epoch": 1.8143939393939394, + "grad_norm": 2.9787070751190186, + "learning_rate": 2.231057062675398e-06, + "loss": 1.4117, + "step": 1437 + }, + { + "epoch": 1.8156565656565657, + "grad_norm": 3.0893537998199463, + "learning_rate": 2.2287184284377927e-06, + "loss": 1.4809, + "step": 1438 + }, + { + "epoch": 1.816919191919192, + "grad_norm": 2.8568801879882812, + "learning_rate": 2.226379794200187e-06, + "loss": 1.4232, + "step": 1439 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 3.1005046367645264, + "learning_rate": 2.224041159962582e-06, + "loss": 1.482, + "step": 1440 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.6738661527633667, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1440 + }, + { + "epoch": 1.8194444444444444, + "grad_norm": 2.9064700603485107, + "learning_rate": 2.2217025257249768e-06, + "loss": 1.4179, + "step": 1441 + }, + { + "epoch": 1.8207070707070707, + "grad_norm": 3.0111782550811768, + "learning_rate": 2.2193638914873716e-06, + "loss": 1.4084, + "step": 1442 + }, + { + "epoch": 1.821969696969697, + "grad_norm": 2.9269816875457764, + "learning_rate": 2.2170252572497665e-06, + "loss": 1.4361, + "step": 1443 + }, + { + "epoch": 1.823232323232323, + "grad_norm": 2.9985902309417725, + "learning_rate": 2.2146866230121613e-06, + "loss": 1.4345, + "step": 1444 + }, + { + "epoch": 1.8244949494949494, + "grad_norm": 3.1306066513061523, + "learning_rate": 2.2123479887745557e-06, + "loss": 1.4728, + "step": 1445 + }, + { + "epoch": 1.8257575757575757, + "grad_norm": 2.9876086711883545, + "learning_rate": 2.2100093545369506e-06, + "loss": 1.378, + "step": 1446 + }, + { + "epoch": 1.827020202020202, + "grad_norm": 3.018763780593872, + "learning_rate": 2.2076707202993454e-06, + "loss": 1.4506, + "step": 1447 + }, + { + "epoch": 1.8282828282828283, + "grad_norm": 2.960573673248291, + "learning_rate": 2.2053320860617402e-06, + "loss": 1.4674, + "step": 1448 + }, + { + "epoch": 1.8295454545454546, + "grad_norm": 2.849881887435913, + "learning_rate": 2.202993451824135e-06, + "loss": 1.4687, + "step": 1449 + }, + { + "epoch": 1.8308080808080809, + "grad_norm": 2.8798882961273193, + "learning_rate": 2.2006548175865295e-06, + "loss": 1.3868, + "step": 1450 + }, + { + "epoch": 1.8308080808080809, + "eval_loss": 1.6657519340515137, + "eval_runtime": 37.0993, + "eval_samples_per_second": 26.955, + "eval_steps_per_second": 1.132, + "step": 1450 + }, + { + "epoch": 1.8320707070707072, + "grad_norm": 2.937471628189087, + "learning_rate": 2.1983161833489243e-06, + "loss": 1.4085, + "step": 1451 + }, + { + "epoch": 1.8333333333333335, + "grad_norm": 3.0815699100494385, + "learning_rate": 2.195977549111319e-06, + "loss": 1.4055, + "step": 1452 + }, + { + "epoch": 1.8345959595959596, + "grad_norm": 2.927917718887329, + "learning_rate": 2.193638914873714e-06, + "loss": 1.4014, + "step": 1453 + }, + { + "epoch": 1.8358585858585859, + "grad_norm": 2.9576570987701416, + "learning_rate": 2.1913002806361085e-06, + "loss": 1.4394, + "step": 1454 + }, + { + "epoch": 1.8371212121212122, + "grad_norm": 2.9505863189697266, + "learning_rate": 2.1889616463985033e-06, + "loss": 1.4018, + "step": 1455 + }, + { + "epoch": 1.8383838383838382, + "grad_norm": 3.014965772628784, + "learning_rate": 2.186623012160898e-06, + "loss": 1.3735, + "step": 1456 + }, + { + "epoch": 1.8396464646464645, + "grad_norm": 3.121122121810913, + "learning_rate": 2.184284377923293e-06, + "loss": 1.4289, + "step": 1457 + }, + { + "epoch": 1.8409090909090908, + "grad_norm": 2.9710142612457275, + "learning_rate": 2.181945743685688e-06, + "loss": 1.4494, + "step": 1458 + }, + { + "epoch": 1.8421717171717171, + "grad_norm": 3.028554916381836, + "learning_rate": 2.1796071094480827e-06, + "loss": 1.5151, + "step": 1459 + }, + { + "epoch": 1.8434343434343434, + "grad_norm": 2.8973751068115234, + "learning_rate": 2.177268475210477e-06, + "loss": 1.4171, + "step": 1460 + }, + { + "epoch": 1.8434343434343434, + "eval_loss": 1.6670986413955688, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1460 + }, + { + "epoch": 1.8446969696969697, + "grad_norm": 3.046865701675415, + "learning_rate": 2.174929840972872e-06, + "loss": 1.4016, + "step": 1461 + }, + { + "epoch": 1.845959595959596, + "grad_norm": 3.0509603023529053, + "learning_rate": 2.1725912067352668e-06, + "loss": 1.4196, + "step": 1462 + }, + { + "epoch": 1.8472222222222223, + "grad_norm": 2.951892137527466, + "learning_rate": 2.1702525724976616e-06, + "loss": 1.4063, + "step": 1463 + }, + { + "epoch": 1.8484848484848486, + "grad_norm": 3.2126667499542236, + "learning_rate": 2.1679139382600565e-06, + "loss": 1.5106, + "step": 1464 + }, + { + "epoch": 1.8497474747474747, + "grad_norm": 2.965576648712158, + "learning_rate": 2.1655753040224513e-06, + "loss": 1.4096, + "step": 1465 + }, + { + "epoch": 1.851010101010101, + "grad_norm": 2.8640592098236084, + "learning_rate": 2.1632366697848457e-06, + "loss": 1.389, + "step": 1466 + }, + { + "epoch": 1.8522727272727273, + "grad_norm": 3.1476943492889404, + "learning_rate": 2.1608980355472406e-06, + "loss": 1.4793, + "step": 1467 + }, + { + "epoch": 1.8535353535353534, + "grad_norm": 2.814695119857788, + "learning_rate": 2.1585594013096354e-06, + "loss": 1.4265, + "step": 1468 + }, + { + "epoch": 1.8547979797979797, + "grad_norm": 2.9728431701660156, + "learning_rate": 2.1562207670720302e-06, + "loss": 1.386, + "step": 1469 + }, + { + "epoch": 1.856060606060606, + "grad_norm": 2.9062652587890625, + "learning_rate": 2.1538821328344247e-06, + "loss": 1.3986, + "step": 1470 + }, + { + "epoch": 1.856060606060606, + "eval_loss": 1.6698763370513916, + "eval_runtime": 37.1129, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1470 + }, + { + "epoch": 1.8573232323232323, + "grad_norm": 2.950857162475586, + "learning_rate": 2.15154349859682e-06, + "loss": 1.4193, + "step": 1471 + }, + { + "epoch": 1.8585858585858586, + "grad_norm": 3.017594337463379, + "learning_rate": 2.1492048643592144e-06, + "loss": 1.4318, + "step": 1472 + }, + { + "epoch": 1.8598484848484849, + "grad_norm": 2.991514205932617, + "learning_rate": 2.146866230121609e-06, + "loss": 1.4462, + "step": 1473 + }, + { + "epoch": 1.8611111111111112, + "grad_norm": 3.149475336074829, + "learning_rate": 2.144527595884004e-06, + "loss": 1.4486, + "step": 1474 + }, + { + "epoch": 1.8623737373737375, + "grad_norm": 2.948828935623169, + "learning_rate": 2.1421889616463985e-06, + "loss": 1.4522, + "step": 1475 + }, + { + "epoch": 1.8636363636363638, + "grad_norm": 3.0383694171905518, + "learning_rate": 2.1398503274087933e-06, + "loss": 1.45, + "step": 1476 + }, + { + "epoch": 1.86489898989899, + "grad_norm": 3.093449831008911, + "learning_rate": 2.137511693171188e-06, + "loss": 1.4266, + "step": 1477 + }, + { + "epoch": 1.8661616161616161, + "grad_norm": 3.017549753189087, + "learning_rate": 2.135173058933583e-06, + "loss": 1.4067, + "step": 1478 + }, + { + "epoch": 1.8674242424242424, + "grad_norm": 2.985044002532959, + "learning_rate": 2.132834424695978e-06, + "loss": 1.4904, + "step": 1479 + }, + { + "epoch": 1.8686868686868687, + "grad_norm": 2.9600346088409424, + "learning_rate": 2.1304957904583727e-06, + "loss": 1.4006, + "step": 1480 + }, + { + "epoch": 1.8686868686868687, + "eval_loss": 1.6648480892181396, + "eval_runtime": 37.1225, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1480 + }, + { + "epoch": 1.8699494949494948, + "grad_norm": 2.8905625343322754, + "learning_rate": 2.128157156220767e-06, + "loss": 1.4096, + "step": 1481 + }, + { + "epoch": 1.871212121212121, + "grad_norm": 2.95342755317688, + "learning_rate": 2.125818521983162e-06, + "loss": 1.3888, + "step": 1482 + }, + { + "epoch": 1.8724747474747474, + "grad_norm": 2.9438536167144775, + "learning_rate": 2.1234798877455568e-06, + "loss": 1.413, + "step": 1483 + }, + { + "epoch": 1.8737373737373737, + "grad_norm": 3.1257364749908447, + "learning_rate": 2.1211412535079516e-06, + "loss": 1.4902, + "step": 1484 + }, + { + "epoch": 1.875, + "grad_norm": 3.1377382278442383, + "learning_rate": 2.118802619270346e-06, + "loss": 1.4815, + "step": 1485 + }, + { + "epoch": 1.8762626262626263, + "grad_norm": 3.122253656387329, + "learning_rate": 2.116463985032741e-06, + "loss": 1.4695, + "step": 1486 + }, + { + "epoch": 1.8775252525252526, + "grad_norm": 3.0128729343414307, + "learning_rate": 2.1141253507951357e-06, + "loss": 1.4673, + "step": 1487 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.921835422515869, + "learning_rate": 2.1117867165575306e-06, + "loss": 1.4325, + "step": 1488 + }, + { + "epoch": 1.8800505050505052, + "grad_norm": 3.19289231300354, + "learning_rate": 2.1094480823199254e-06, + "loss": 1.4504, + "step": 1489 + }, + { + "epoch": 1.8813131313131313, + "grad_norm": 2.8579299449920654, + "learning_rate": 2.1071094480823203e-06, + "loss": 1.4042, + "step": 1490 + }, + { + "epoch": 1.8813131313131313, + "eval_loss": 1.6603271961212158, + "eval_runtime": 37.2997, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 1490 + }, + { + "epoch": 1.8825757575757576, + "grad_norm": 2.9113969802856445, + "learning_rate": 2.1047708138447147e-06, + "loss": 1.4128, + "step": 1491 + }, + { + "epoch": 1.8838383838383839, + "grad_norm": 3.058133363723755, + "learning_rate": 2.1024321796071095e-06, + "loss": 1.4574, + "step": 1492 + }, + { + "epoch": 1.88510101010101, + "grad_norm": 3.0066583156585693, + "learning_rate": 2.1000935453695044e-06, + "loss": 1.4521, + "step": 1493 + }, + { + "epoch": 1.8863636363636362, + "grad_norm": 2.9154908657073975, + "learning_rate": 2.097754911131899e-06, + "loss": 1.4103, + "step": 1494 + }, + { + "epoch": 1.8876262626262625, + "grad_norm": 2.977062702178955, + "learning_rate": 2.095416276894294e-06, + "loss": 1.4003, + "step": 1495 + }, + { + "epoch": 1.8888888888888888, + "grad_norm": 2.9048168659210205, + "learning_rate": 2.093077642656689e-06, + "loss": 1.4216, + "step": 1496 + }, + { + "epoch": 1.8901515151515151, + "grad_norm": 3.008733034133911, + "learning_rate": 2.0907390084190833e-06, + "loss": 1.4516, + "step": 1497 + }, + { + "epoch": 1.8914141414141414, + "grad_norm": 3.0832529067993164, + "learning_rate": 2.088400374181478e-06, + "loss": 1.4248, + "step": 1498 + }, + { + "epoch": 1.8926767676767677, + "grad_norm": 2.848066568374634, + "learning_rate": 2.086061739943873e-06, + "loss": 1.4042, + "step": 1499 + }, + { + "epoch": 1.893939393939394, + "grad_norm": 2.846782684326172, + "learning_rate": 2.0837231057062674e-06, + "loss": 1.409, + "step": 1500 + }, + { + "epoch": 1.893939393939394, + "eval_loss": 1.6672260761260986, + "eval_runtime": 37.4695, + "eval_samples_per_second": 26.688, + "eval_steps_per_second": 1.121, + "step": 1500 + }, + { + "epoch": 1.8952020202020203, + "grad_norm": 3.026818037033081, + "learning_rate": 2.0813844714686623e-06, + "loss": 1.4842, + "step": 1501 + }, + { + "epoch": 1.8964646464646466, + "grad_norm": 2.977665662765503, + "learning_rate": 2.0790458372310575e-06, + "loss": 1.4635, + "step": 1502 + }, + { + "epoch": 1.8977272727272727, + "grad_norm": 2.9093048572540283, + "learning_rate": 2.076707202993452e-06, + "loss": 1.4303, + "step": 1503 + }, + { + "epoch": 1.898989898989899, + "grad_norm": 3.0777814388275146, + "learning_rate": 2.0743685687558468e-06, + "loss": 1.4581, + "step": 1504 + }, + { + "epoch": 1.9002525252525253, + "grad_norm": 3.0283970832824707, + "learning_rate": 2.0720299345182416e-06, + "loss": 1.4665, + "step": 1505 + }, + { + "epoch": 1.9015151515151514, + "grad_norm": 2.852412223815918, + "learning_rate": 2.069691300280636e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.9027777777777777, + "grad_norm": 3.1163125038146973, + "learning_rate": 2.067352666043031e-06, + "loss": 1.4907, + "step": 1507 + }, + { + "epoch": 1.904040404040404, + "grad_norm": 2.9857897758483887, + "learning_rate": 2.0650140318054257e-06, + "loss": 1.4262, + "step": 1508 + }, + { + "epoch": 1.9053030303030303, + "grad_norm": 2.957388401031494, + "learning_rate": 2.0626753975678206e-06, + "loss": 1.4525, + "step": 1509 + }, + { + "epoch": 1.9065656565656566, + "grad_norm": 3.0107505321502686, + "learning_rate": 2.0603367633302154e-06, + "loss": 1.4461, + "step": 1510 + }, + { + "epoch": 1.9065656565656566, + "eval_loss": 1.6621215343475342, + "eval_runtime": 36.9336, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1510 + }, + { + "epoch": 1.9078282828282829, + "grad_norm": 3.112478494644165, + "learning_rate": 2.0579981290926103e-06, + "loss": 1.479, + "step": 1511 + }, + { + "epoch": 1.9090909090909092, + "grad_norm": 3.083494186401367, + "learning_rate": 2.0556594948550047e-06, + "loss": 1.4777, + "step": 1512 + }, + { + "epoch": 1.9103535353535355, + "grad_norm": 2.932652711868286, + "learning_rate": 2.0533208606173995e-06, + "loss": 1.3937, + "step": 1513 + }, + { + "epoch": 1.9116161616161618, + "grad_norm": 2.8349449634552, + "learning_rate": 2.0509822263797944e-06, + "loss": 1.4096, + "step": 1514 + }, + { + "epoch": 1.9128787878787878, + "grad_norm": 2.9760472774505615, + "learning_rate": 2.048643592142189e-06, + "loss": 1.4312, + "step": 1515 + }, + { + "epoch": 1.9141414141414141, + "grad_norm": 3.1012730598449707, + "learning_rate": 2.0463049579045836e-06, + "loss": 1.5189, + "step": 1516 + }, + { + "epoch": 1.9154040404040404, + "grad_norm": 2.9072659015655518, + "learning_rate": 2.043966323666979e-06, + "loss": 1.4051, + "step": 1517 + }, + { + "epoch": 1.9166666666666665, + "grad_norm": 2.840203285217285, + "learning_rate": 2.0416276894293733e-06, + "loss": 1.3899, + "step": 1518 + }, + { + "epoch": 1.9179292929292928, + "grad_norm": 3.130812406539917, + "learning_rate": 2.039289055191768e-06, + "loss": 1.4834, + "step": 1519 + }, + { + "epoch": 1.9191919191919191, + "grad_norm": 3.1507933139801025, + "learning_rate": 2.036950420954163e-06, + "loss": 1.402, + "step": 1520 + }, + { + "epoch": 1.9191919191919191, + "eval_loss": 1.6648352146148682, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1520 + }, + { + "epoch": 1.9204545454545454, + "grad_norm": 3.099269151687622, + "learning_rate": 2.034611786716558e-06, + "loss": 1.4189, + "step": 1521 + }, + { + "epoch": 1.9217171717171717, + "grad_norm": 3.0046563148498535, + "learning_rate": 2.0322731524789523e-06, + "loss": 1.405, + "step": 1522 + }, + { + "epoch": 1.922979797979798, + "grad_norm": 3.229635715484619, + "learning_rate": 2.029934518241347e-06, + "loss": 1.5031, + "step": 1523 + }, + { + "epoch": 1.9242424242424243, + "grad_norm": 3.0336225032806396, + "learning_rate": 2.027595884003742e-06, + "loss": 1.4685, + "step": 1524 + }, + { + "epoch": 1.9255050505050506, + "grad_norm": 2.9998443126678467, + "learning_rate": 2.025257249766137e-06, + "loss": 1.4558, + "step": 1525 + }, + { + "epoch": 1.926767676767677, + "grad_norm": 3.1587722301483154, + "learning_rate": 2.0229186155285316e-06, + "loss": 1.4916, + "step": 1526 + }, + { + "epoch": 1.928030303030303, + "grad_norm": 3.2228925228118896, + "learning_rate": 2.0205799812909265e-06, + "loss": 1.4373, + "step": 1527 + }, + { + "epoch": 1.9292929292929293, + "grad_norm": 3.06015944480896, + "learning_rate": 2.018241347053321e-06, + "loss": 1.4147, + "step": 1528 + }, + { + "epoch": 1.9305555555555556, + "grad_norm": 3.02284574508667, + "learning_rate": 2.0159027128157157e-06, + "loss": 1.4574, + "step": 1529 + }, + { + "epoch": 1.9318181818181817, + "grad_norm": 2.9933414459228516, + "learning_rate": 2.0135640785781106e-06, + "loss": 1.4283, + "step": 1530 + }, + { + "epoch": 1.9318181818181817, + "eval_loss": 1.6711069345474243, + "eval_runtime": 37.1182, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1530 + }, + { + "epoch": 1.933080808080808, + "grad_norm": 3.325230121612549, + "learning_rate": 2.011225444340505e-06, + "loss": 1.4532, + "step": 1531 + }, + { + "epoch": 1.9343434343434343, + "grad_norm": 3.0051827430725098, + "learning_rate": 2.0088868101029003e-06, + "loss": 1.3965, + "step": 1532 + }, + { + "epoch": 1.9356060606060606, + "grad_norm": 3.1392757892608643, + "learning_rate": 2.006548175865295e-06, + "loss": 1.4678, + "step": 1533 + }, + { + "epoch": 1.9368686868686869, + "grad_norm": 3.0719001293182373, + "learning_rate": 2.0042095416276895e-06, + "loss": 1.4374, + "step": 1534 + }, + { + "epoch": 1.9381313131313131, + "grad_norm": 3.0406386852264404, + "learning_rate": 2.0018709073900844e-06, + "loss": 1.4199, + "step": 1535 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9481236934661865, + "learning_rate": 1.9995322731524792e-06, + "loss": 1.4599, + "step": 1536 + }, + { + "epoch": 1.9406565656565657, + "grad_norm": 2.901155948638916, + "learning_rate": 1.9971936389148736e-06, + "loss": 1.381, + "step": 1537 + }, + { + "epoch": 1.941919191919192, + "grad_norm": 2.9623661041259766, + "learning_rate": 1.9948550046772685e-06, + "loss": 1.4769, + "step": 1538 + }, + { + "epoch": 1.9431818181818183, + "grad_norm": 2.8457608222961426, + "learning_rate": 1.9925163704396633e-06, + "loss": 1.4117, + "step": 1539 + }, + { + "epoch": 1.9444444444444444, + "grad_norm": 2.9782702922821045, + "learning_rate": 1.990177736202058e-06, + "loss": 1.4824, + "step": 1540 + }, + { + "epoch": 1.9444444444444444, + "eval_loss": 1.6663525104522705, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1540 + }, + { + "epoch": 1.9457070707070707, + "grad_norm": 2.8265817165374756, + "learning_rate": 1.987839101964453e-06, + "loss": 1.4375, + "step": 1541 + }, + { + "epoch": 1.946969696969697, + "grad_norm": 2.9103686809539795, + "learning_rate": 1.985500467726848e-06, + "loss": 1.474, + "step": 1542 + }, + { + "epoch": 1.948232323232323, + "grad_norm": 2.837301254272461, + "learning_rate": 1.9831618334892423e-06, + "loss": 1.4171, + "step": 1543 + }, + { + "epoch": 1.9494949494949494, + "grad_norm": 3.0468413829803467, + "learning_rate": 1.980823199251637e-06, + "loss": 1.5095, + "step": 1544 + }, + { + "epoch": 1.9507575757575757, + "grad_norm": 2.968480348587036, + "learning_rate": 1.978484565014032e-06, + "loss": 1.4203, + "step": 1545 + }, + { + "epoch": 1.952020202020202, + "grad_norm": 2.9865822792053223, + "learning_rate": 1.976145930776427e-06, + "loss": 1.3916, + "step": 1546 + }, + { + "epoch": 1.9532828282828283, + "grad_norm": 2.9733808040618896, + "learning_rate": 1.9738072965388212e-06, + "loss": 1.4216, + "step": 1547 + }, + { + "epoch": 1.9545454545454546, + "grad_norm": 2.941718101501465, + "learning_rate": 1.9714686623012165e-06, + "loss": 1.426, + "step": 1548 + }, + { + "epoch": 1.9558080808080809, + "grad_norm": 3.0980522632598877, + "learning_rate": 1.969130028063611e-06, + "loss": 1.471, + "step": 1549 + }, + { + "epoch": 1.9570707070707072, + "grad_norm": 2.797102689743042, + "learning_rate": 1.9667913938260057e-06, + "loss": 1.3897, + "step": 1550 + }, + { + "epoch": 1.9570707070707072, + "eval_loss": 1.661893367767334, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 1550 + }, + { + "epoch": 1.9583333333333335, + "grad_norm": 2.9562623500823975, + "learning_rate": 1.9644527595884006e-06, + "loss": 1.4481, + "step": 1551 + }, + { + "epoch": 1.9595959595959596, + "grad_norm": 2.9586915969848633, + "learning_rate": 1.9621141253507954e-06, + "loss": 1.4068, + "step": 1552 + }, + { + "epoch": 1.9608585858585859, + "grad_norm": 3.061110496520996, + "learning_rate": 1.95977549111319e-06, + "loss": 1.4524, + "step": 1553 + }, + { + "epoch": 1.9621212121212122, + "grad_norm": 3.080491542816162, + "learning_rate": 1.9574368568755847e-06, + "loss": 1.4457, + "step": 1554 + }, + { + "epoch": 1.9633838383838382, + "grad_norm": 3.083254814147949, + "learning_rate": 1.9550982226379795e-06, + "loss": 1.4775, + "step": 1555 + }, + { + "epoch": 1.9646464646464645, + "grad_norm": 3.008596420288086, + "learning_rate": 1.9527595884003744e-06, + "loss": 1.425, + "step": 1556 + }, + { + "epoch": 1.9659090909090908, + "grad_norm": 2.999195098876953, + "learning_rate": 1.9504209541627692e-06, + "loss": 1.4114, + "step": 1557 + }, + { + "epoch": 1.9671717171717171, + "grad_norm": 3.055265188217163, + "learning_rate": 1.948082319925164e-06, + "loss": 1.4767, + "step": 1558 + }, + { + "epoch": 1.9684343434343434, + "grad_norm": 3.0922064781188965, + "learning_rate": 1.9457436856875585e-06, + "loss": 1.4548, + "step": 1559 + }, + { + "epoch": 1.9696969696969697, + "grad_norm": 2.8740789890289307, + "learning_rate": 1.9434050514499533e-06, + "loss": 1.4582, + "step": 1560 + }, + { + "epoch": 1.9696969696969697, + "eval_loss": 1.6608474254608154, + "eval_runtime": 37.1201, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1560 + }, + { + "epoch": 1.970959595959596, + "grad_norm": 2.983570098876953, + "learning_rate": 1.941066417212348e-06, + "loss": 1.4602, + "step": 1561 + }, + { + "epoch": 1.9722222222222223, + "grad_norm": 3.132474184036255, + "learning_rate": 1.9387277829747426e-06, + "loss": 1.4689, + "step": 1562 + }, + { + "epoch": 1.9734848484848486, + "grad_norm": 2.9685440063476562, + "learning_rate": 1.936389148737138e-06, + "loss": 1.39, + "step": 1563 + }, + { + "epoch": 1.9747474747474747, + "grad_norm": 3.123037815093994, + "learning_rate": 1.9340505144995327e-06, + "loss": 1.4151, + "step": 1564 + }, + { + "epoch": 1.976010101010101, + "grad_norm": 3.135232925415039, + "learning_rate": 1.931711880261927e-06, + "loss": 1.4166, + "step": 1565 + }, + { + "epoch": 1.9772727272727273, + "grad_norm": 2.9131808280944824, + "learning_rate": 1.929373246024322e-06, + "loss": 1.3952, + "step": 1566 + }, + { + "epoch": 1.9785353535353534, + "grad_norm": 3.1187453269958496, + "learning_rate": 1.927034611786717e-06, + "loss": 1.4132, + "step": 1567 + }, + { + "epoch": 1.9797979797979797, + "grad_norm": 2.983447551727295, + "learning_rate": 1.9246959775491112e-06, + "loss": 1.4517, + "step": 1568 + }, + { + "epoch": 1.981060606060606, + "grad_norm": 2.9731497764587402, + "learning_rate": 1.922357343311506e-06, + "loss": 1.432, + "step": 1569 + }, + { + "epoch": 1.9823232323232323, + "grad_norm": 2.8526554107666016, + "learning_rate": 1.920018709073901e-06, + "loss": 1.4141, + "step": 1570 + }, + { + "epoch": 1.9823232323232323, + "eval_loss": 1.6579400300979614, + "eval_runtime": 37.1216, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1570 + }, + { + "epoch": 1.9835858585858586, + "grad_norm": 2.900019407272339, + "learning_rate": 1.9176800748362958e-06, + "loss": 1.4008, + "step": 1571 + }, + { + "epoch": 1.9848484848484849, + "grad_norm": 2.96079683303833, + "learning_rate": 1.9153414405986906e-06, + "loss": 1.4586, + "step": 1572 + }, + { + "epoch": 1.9861111111111112, + "grad_norm": 3.1275715827941895, + "learning_rate": 1.9130028063610854e-06, + "loss": 1.4768, + "step": 1573 + }, + { + "epoch": 1.9873737373737375, + "grad_norm": 3.044041395187378, + "learning_rate": 1.91066417212348e-06, + "loss": 1.4428, + "step": 1574 + }, + { + "epoch": 1.9886363636363638, + "grad_norm": 3.0206832885742188, + "learning_rate": 1.9083255378858747e-06, + "loss": 1.4492, + "step": 1575 + }, + { + "epoch": 1.98989898989899, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.398, + "step": 1576 + }, + { + "epoch": 1.9911616161616161, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.4373, + "step": 1577 + }, + { + "epoch": 1.9924242424242424, + "grad_norm": 3.129822254180908, + "learning_rate": 1.9036482694106642e-06, + "loss": 1.4599, + "step": 1578 + }, + { + "epoch": 1.9936868686868687, + "grad_norm": 2.978015661239624, + "learning_rate": 1.9013096351730592e-06, + "loss": 1.409, + "step": 1579 + }, + { + "epoch": 1.9949494949494948, + "grad_norm": 2.980295181274414, + "learning_rate": 1.8989710009354539e-06, + "loss": 1.4251, + "step": 1580 + }, + { + "epoch": 1.9949494949494948, + "eval_loss": 1.661036491394043, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.996212121212121, + "grad_norm": 2.878361225128174, + "learning_rate": 1.8966323666978487e-06, + "loss": 1.4289, + "step": 1581 + }, + { + "epoch": 1.9974747474747474, + "grad_norm": 2.967414379119873, + "learning_rate": 1.8942937324602433e-06, + "loss": 1.5151, + "step": 1582 + }, + { + "epoch": 1.9987373737373737, + "grad_norm": 2.9448516368865967, + "learning_rate": 1.8919550982226382e-06, + "loss": 1.4327, + "step": 1583 + }, + { + "epoch": 2.0, + "grad_norm": 2.9000067710876465, + "learning_rate": 1.8896164639850328e-06, + "loss": 1.4113, + "step": 1584 + }, + { + "epoch": 2.0012626262626263, + "grad_norm": 3.1630377769470215, + "learning_rate": 1.8872778297474277e-06, + "loss": 1.4179, + "step": 1585 + }, + { + "epoch": 2.0025252525252526, + "grad_norm": 3.2642927169799805, + "learning_rate": 1.8849391955098223e-06, + "loss": 1.3873, + "step": 1586 + }, + { + "epoch": 2.003787878787879, + "grad_norm": 3.1669039726257324, + "learning_rate": 1.8826005612722173e-06, + "loss": 1.3834, + "step": 1587 + }, + { + "epoch": 2.005050505050505, + "grad_norm": 3.145040988922119, + "learning_rate": 1.880261927034612e-06, + "loss": 1.3334, + "step": 1588 + }, + { + "epoch": 2.0063131313131315, + "grad_norm": 3.707200050354004, + "learning_rate": 1.8779232927970068e-06, + "loss": 1.3979, + "step": 1589 + }, + { + "epoch": 2.007575757575758, + "grad_norm": 3.936523199081421, + "learning_rate": 1.8755846585594014e-06, + "loss": 1.4221, + "step": 1590 + }, + { + "epoch": 2.007575757575758, + "eval_loss": 1.6722478866577148, + "eval_runtime": 37.2129, + "eval_samples_per_second": 26.872, + "eval_steps_per_second": 1.129, + "step": 1590 + }, + { + "epoch": 2.0088383838383836, + "grad_norm": 3.4221086502075195, + "learning_rate": 1.8732460243217963e-06, + "loss": 1.3894, + "step": 1591 + }, + { + "epoch": 2.01010101010101, + "grad_norm": 3.3115553855895996, + "learning_rate": 1.870907390084191e-06, + "loss": 1.4016, + "step": 1592 + }, + { + "epoch": 2.0113636363636362, + "grad_norm": 3.055779457092285, + "learning_rate": 1.8685687558465855e-06, + "loss": 1.3676, + "step": 1593 + }, + { + "epoch": 2.0126262626262625, + "grad_norm": 3.4615232944488525, + "learning_rate": 1.8662301216089804e-06, + "loss": 1.3624, + "step": 1594 + }, + { + "epoch": 2.013888888888889, + "grad_norm": 3.3298869132995605, + "learning_rate": 1.8638914873713754e-06, + "loss": 1.3939, + "step": 1595 + }, + { + "epoch": 2.015151515151515, + "grad_norm": 3.271921396255493, + "learning_rate": 1.86155285313377e-06, + "loss": 1.3994, + "step": 1596 + }, + { + "epoch": 2.0164141414141414, + "grad_norm": 3.272677421569824, + "learning_rate": 1.859214218896165e-06, + "loss": 1.3934, + "step": 1597 + }, + { + "epoch": 2.0176767676767677, + "grad_norm": 3.3485944271087646, + "learning_rate": 1.8568755846585596e-06, + "loss": 1.3626, + "step": 1598 + }, + { + "epoch": 2.018939393939394, + "grad_norm": 3.426981210708618, + "learning_rate": 1.8545369504209542e-06, + "loss": 1.4231, + "step": 1599 + }, + { + "epoch": 2.0202020202020203, + "grad_norm": 3.3859283924102783, + "learning_rate": 1.852198316183349e-06, + "loss": 1.3587, + "step": 1600 + }, + { + "epoch": 2.0202020202020203, + "eval_loss": 1.6732399463653564, + "eval_runtime": 37.3484, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 1600 + }, + { + "epoch": 2.0214646464646466, + "grad_norm": 3.537806987762451, + "learning_rate": 1.8498596819457437e-06, + "loss": 1.3729, + "step": 1601 + }, + { + "epoch": 2.022727272727273, + "grad_norm": 3.5417566299438477, + "learning_rate": 1.8475210477081387e-06, + "loss": 1.4272, + "step": 1602 + }, + { + "epoch": 2.023989898989899, + "grad_norm": 3.287195920944214, + "learning_rate": 1.8451824134705336e-06, + "loss": 1.3647, + "step": 1603 + }, + { + "epoch": 2.025252525252525, + "grad_norm": 3.233365297317505, + "learning_rate": 1.8428437792329282e-06, + "loss": 1.4062, + "step": 1604 + }, + { + "epoch": 2.0265151515151514, + "grad_norm": 3.296578884124756, + "learning_rate": 1.8405051449953228e-06, + "loss": 1.3841, + "step": 1605 + }, + { + "epoch": 2.0277777777777777, + "grad_norm": 3.2697315216064453, + "learning_rate": 1.8381665107577177e-06, + "loss": 1.3957, + "step": 1606 + }, + { + "epoch": 2.029040404040404, + "grad_norm": 3.1117050647735596, + "learning_rate": 1.8358278765201123e-06, + "loss": 1.3643, + "step": 1607 + }, + { + "epoch": 2.0303030303030303, + "grad_norm": 3.14170241355896, + "learning_rate": 1.8334892422825071e-06, + "loss": 1.3483, + "step": 1608 + }, + { + "epoch": 2.0315656565656566, + "grad_norm": 3.2749483585357666, + "learning_rate": 1.8311506080449018e-06, + "loss": 1.4007, + "step": 1609 + }, + { + "epoch": 2.032828282828283, + "grad_norm": 3.266631603240967, + "learning_rate": 1.8288119738072968e-06, + "loss": 1.3943, + "step": 1610 + }, + { + "epoch": 2.032828282828283, + "eval_loss": 1.6679413318634033, + "eval_runtime": 36.8768, + "eval_samples_per_second": 27.117, + "eval_steps_per_second": 1.139, + "step": 1610 + }, + { + "epoch": 2.034090909090909, + "grad_norm": 3.535625696182251, + "learning_rate": 1.8264733395696914e-06, + "loss": 1.4027, + "step": 1611 + }, + { + "epoch": 2.0353535353535355, + "grad_norm": 3.2741401195526123, + "learning_rate": 1.8241347053320863e-06, + "loss": 1.4075, + "step": 1612 + }, + { + "epoch": 2.0366161616161618, + "grad_norm": 3.3183717727661133, + "learning_rate": 1.821796071094481e-06, + "loss": 1.4223, + "step": 1613 + }, + { + "epoch": 2.037878787878788, + "grad_norm": 3.323057174682617, + "learning_rate": 1.8194574368568758e-06, + "loss": 1.3595, + "step": 1614 + }, + { + "epoch": 2.0391414141414144, + "grad_norm": 3.202054738998413, + "learning_rate": 1.8171188026192704e-06, + "loss": 1.3399, + "step": 1615 + }, + { + "epoch": 2.04040404040404, + "grad_norm": 3.3273448944091797, + "learning_rate": 1.8147801683816652e-06, + "loss": 1.3848, + "step": 1616 + }, + { + "epoch": 2.0416666666666665, + "grad_norm": 3.396843671798706, + "learning_rate": 1.81244153414406e-06, + "loss": 1.3968, + "step": 1617 + }, + { + "epoch": 2.042929292929293, + "grad_norm": 3.2145836353302, + "learning_rate": 1.810102899906455e-06, + "loss": 1.3571, + "step": 1618 + }, + { + "epoch": 2.044191919191919, + "grad_norm": 3.156754732131958, + "learning_rate": 1.8077642656688496e-06, + "loss": 1.3439, + "step": 1619 + }, + { + "epoch": 2.0454545454545454, + "grad_norm": 3.4259471893310547, + "learning_rate": 1.8054256314312444e-06, + "loss": 1.4123, + "step": 1620 + }, + { + "epoch": 2.0454545454545454, + "eval_loss": 1.6703134775161743, + "eval_runtime": 37.002, + "eval_samples_per_second": 27.026, + "eval_steps_per_second": 1.135, + "step": 1620 + }, + { + "epoch": 2.0467171717171717, + "grad_norm": 3.3556408882141113, + "learning_rate": 1.803086997193639e-06, + "loss": 1.4252, + "step": 1621 + }, + { + "epoch": 2.047979797979798, + "grad_norm": 3.6319949626922607, + "learning_rate": 1.8007483629560339e-06, + "loss": 1.4634, + "step": 1622 + }, + { + "epoch": 2.0492424242424243, + "grad_norm": 3.4495089054107666, + "learning_rate": 1.7984097287184285e-06, + "loss": 1.4066, + "step": 1623 + }, + { + "epoch": 2.0505050505050506, + "grad_norm": 3.182297945022583, + "learning_rate": 1.7960710944808231e-06, + "loss": 1.3899, + "step": 1624 + }, + { + "epoch": 2.051767676767677, + "grad_norm": 3.097158193588257, + "learning_rate": 1.7937324602432182e-06, + "loss": 1.293, + "step": 1625 + }, + { + "epoch": 2.053030303030303, + "grad_norm": 3.429065704345703, + "learning_rate": 1.791393826005613e-06, + "loss": 1.3905, + "step": 1626 + }, + { + "epoch": 2.0542929292929295, + "grad_norm": 3.183366298675537, + "learning_rate": 1.7890551917680077e-06, + "loss": 1.3444, + "step": 1627 + }, + { + "epoch": 2.0555555555555554, + "grad_norm": 3.175612449645996, + "learning_rate": 1.7867165575304025e-06, + "loss": 1.3985, + "step": 1628 + }, + { + "epoch": 2.0568181818181817, + "grad_norm": 3.26491641998291, + "learning_rate": 1.7843779232927971e-06, + "loss": 1.3607, + "step": 1629 + }, + { + "epoch": 2.058080808080808, + "grad_norm": 3.2279412746429443, + "learning_rate": 1.7820392890551918e-06, + "loss": 1.3545, + "step": 1630 + }, + { + "epoch": 2.058080808080808, + "eval_loss": 1.6729198694229126, + "eval_runtime": 37.0369, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 1630 + }, + { + "epoch": 2.0593434343434343, + "grad_norm": 3.3255503177642822, + "learning_rate": 1.7797006548175866e-06, + "loss": 1.3938, + "step": 1631 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.2912404537200928, + "learning_rate": 1.7773620205799812e-06, + "loss": 1.366, + "step": 1632 + }, + { + "epoch": 2.061868686868687, + "grad_norm": 3.2314395904541016, + "learning_rate": 1.7750233863423763e-06, + "loss": 1.3298, + "step": 1633 + }, + { + "epoch": 2.063131313131313, + "grad_norm": 3.3304178714752197, + "learning_rate": 1.772684752104771e-06, + "loss": 1.3592, + "step": 1634 + }, + { + "epoch": 2.0643939393939394, + "grad_norm": 3.3113889694213867, + "learning_rate": 1.7703461178671658e-06, + "loss": 1.3865, + "step": 1635 + }, + { + "epoch": 2.0656565656565657, + "grad_norm": 3.260434150695801, + "learning_rate": 1.7680074836295604e-06, + "loss": 1.3306, + "step": 1636 + }, + { + "epoch": 2.066919191919192, + "grad_norm": 3.2890963554382324, + "learning_rate": 1.7656688493919552e-06, + "loss": 1.3533, + "step": 1637 + }, + { + "epoch": 2.0681818181818183, + "grad_norm": 3.3027193546295166, + "learning_rate": 1.7633302151543499e-06, + "loss": 1.3746, + "step": 1638 + }, + { + "epoch": 2.0694444444444446, + "grad_norm": 3.3163208961486816, + "learning_rate": 1.7609915809167447e-06, + "loss": 1.4113, + "step": 1639 + }, + { + "epoch": 2.0707070707070705, + "grad_norm": 3.300557851791382, + "learning_rate": 1.7586529466791396e-06, + "loss": 1.3669, + "step": 1640 + }, + { + "epoch": 2.0707070707070705, + "eval_loss": 1.6676570177078247, + "eval_runtime": 37.0589, + "eval_samples_per_second": 26.984, + "eval_steps_per_second": 1.133, + "step": 1640 + }, + { + "epoch": 2.071969696969697, + "grad_norm": 3.305338144302368, + "learning_rate": 1.7563143124415344e-06, + "loss": 1.3969, + "step": 1641 + }, + { + "epoch": 2.073232323232323, + "grad_norm": 3.2047572135925293, + "learning_rate": 1.753975678203929e-06, + "loss": 1.3856, + "step": 1642 + }, + { + "epoch": 2.0744949494949494, + "grad_norm": 3.575038433074951, + "learning_rate": 1.7516370439663239e-06, + "loss": 1.3947, + "step": 1643 + }, + { + "epoch": 2.0757575757575757, + "grad_norm": 3.44581937789917, + "learning_rate": 1.7492984097287185e-06, + "loss": 1.4037, + "step": 1644 + }, + { + "epoch": 2.077020202020202, + "grad_norm": 3.574326753616333, + "learning_rate": 1.7469597754911134e-06, + "loss": 1.4216, + "step": 1645 + }, + { + "epoch": 2.0782828282828283, + "grad_norm": 3.2762558460235596, + "learning_rate": 1.744621141253508e-06, + "loss": 1.3621, + "step": 1646 + }, + { + "epoch": 2.0795454545454546, + "grad_norm": 3.3749659061431885, + "learning_rate": 1.7422825070159028e-06, + "loss": 1.3762, + "step": 1647 + }, + { + "epoch": 2.080808080808081, + "grad_norm": 3.287447214126587, + "learning_rate": 1.7399438727782977e-06, + "loss": 1.3565, + "step": 1648 + }, + { + "epoch": 2.082070707070707, + "grad_norm": 3.5107264518737793, + "learning_rate": 1.7376052385406925e-06, + "loss": 1.392, + "step": 1649 + }, + { + "epoch": 2.0833333333333335, + "grad_norm": 3.319650173187256, + "learning_rate": 1.7352666043030871e-06, + "loss": 1.3835, + "step": 1650 + }, + { + "epoch": 2.0833333333333335, + "eval_loss": 1.6727185249328613, + "eval_runtime": 37.0658, + "eval_samples_per_second": 26.979, + "eval_steps_per_second": 1.133, + "step": 1650 + }, + { + "epoch": 2.08459595959596, + "grad_norm": 3.4447925090789795, + "learning_rate": 1.732927970065482e-06, + "loss": 1.4186, + "step": 1651 + }, + { + "epoch": 2.0858585858585856, + "grad_norm": 3.5944249629974365, + "learning_rate": 1.7305893358278766e-06, + "loss": 1.4461, + "step": 1652 + }, + { + "epoch": 2.087121212121212, + "grad_norm": 3.5437049865722656, + "learning_rate": 1.7282507015902715e-06, + "loss": 1.3343, + "step": 1653 + }, + { + "epoch": 2.0883838383838382, + "grad_norm": 3.361407518386841, + "learning_rate": 1.725912067352666e-06, + "loss": 1.3653, + "step": 1654 + }, + { + "epoch": 2.0896464646464645, + "grad_norm": 3.4728963375091553, + "learning_rate": 1.7235734331150607e-06, + "loss": 1.4297, + "step": 1655 + }, + { + "epoch": 2.090909090909091, + "grad_norm": 3.3794610500335693, + "learning_rate": 1.7212347988774558e-06, + "loss": 1.3941, + "step": 1656 + }, + { + "epoch": 2.092171717171717, + "grad_norm": 3.280857801437378, + "learning_rate": 1.7188961646398506e-06, + "loss": 1.3445, + "step": 1657 + }, + { + "epoch": 2.0934343434343434, + "grad_norm": 3.29036283493042, + "learning_rate": 1.7165575304022453e-06, + "loss": 1.3581, + "step": 1658 + }, + { + "epoch": 2.0946969696969697, + "grad_norm": 3.1731486320495605, + "learning_rate": 1.7142188961646399e-06, + "loss": 1.3873, + "step": 1659 + }, + { + "epoch": 2.095959595959596, + "grad_norm": 3.2283008098602295, + "learning_rate": 1.7118802619270347e-06, + "loss": 1.3462, + "step": 1660 + }, + { + "epoch": 2.095959595959596, + "eval_loss": 1.6726486682891846, + "eval_runtime": 37.0617, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1660 + }, + { + "epoch": 2.0972222222222223, + "grad_norm": 3.329171657562256, + "learning_rate": 1.7095416276894294e-06, + "loss": 1.3974, + "step": 1661 + }, + { + "epoch": 2.0984848484848486, + "grad_norm": 3.4113657474517822, + "learning_rate": 1.7072029934518242e-06, + "loss": 1.4237, + "step": 1662 + }, + { + "epoch": 2.099747474747475, + "grad_norm": 3.431184768676758, + "learning_rate": 1.7048643592142193e-06, + "loss": 1.38, + "step": 1663 + }, + { + "epoch": 2.101010101010101, + "grad_norm": 3.5471761226654053, + "learning_rate": 1.7025257249766139e-06, + "loss": 1.3762, + "step": 1664 + }, + { + "epoch": 2.102272727272727, + "grad_norm": 3.43166446685791, + "learning_rate": 1.7001870907390085e-06, + "loss": 1.4371, + "step": 1665 + }, + { + "epoch": 2.1035353535353534, + "grad_norm": 3.28086256980896, + "learning_rate": 1.6978484565014034e-06, + "loss": 1.4062, + "step": 1666 + }, + { + "epoch": 2.1047979797979797, + "grad_norm": 3.166083335876465, + "learning_rate": 1.695509822263798e-06, + "loss": 1.3535, + "step": 1667 + }, + { + "epoch": 2.106060606060606, + "grad_norm": 3.4707589149475098, + "learning_rate": 1.6931711880261928e-06, + "loss": 1.4237, + "step": 1668 + }, + { + "epoch": 2.1073232323232323, + "grad_norm": 3.5033648014068604, + "learning_rate": 1.6908325537885875e-06, + "loss": 1.4331, + "step": 1669 + }, + { + "epoch": 2.1085858585858586, + "grad_norm": 3.3915514945983887, + "learning_rate": 1.6884939195509823e-06, + "loss": 1.3783, + "step": 1670 + }, + { + "epoch": 2.1085858585858586, + "eval_loss": 1.6657648086547852, + "eval_runtime": 37.0742, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1670 + }, + { + "epoch": 2.109848484848485, + "grad_norm": 3.3420865535736084, + "learning_rate": 1.6861552853133772e-06, + "loss": 1.4016, + "step": 1671 + }, + { + "epoch": 2.111111111111111, + "grad_norm": 3.382840394973755, + "learning_rate": 1.683816651075772e-06, + "loss": 1.4209, + "step": 1672 + }, + { + "epoch": 2.1123737373737375, + "grad_norm": 3.3109424114227295, + "learning_rate": 1.6814780168381666e-06, + "loss": 1.3716, + "step": 1673 + }, + { + "epoch": 2.1136363636363638, + "grad_norm": 3.3781373500823975, + "learning_rate": 1.6791393826005615e-06, + "loss": 1.4106, + "step": 1674 + }, + { + "epoch": 2.11489898989899, + "grad_norm": 3.29980731010437, + "learning_rate": 1.676800748362956e-06, + "loss": 1.3628, + "step": 1675 + }, + { + "epoch": 2.1161616161616164, + "grad_norm": 3.38468599319458, + "learning_rate": 1.674462114125351e-06, + "loss": 1.3478, + "step": 1676 + }, + { + "epoch": 2.117424242424242, + "grad_norm": 3.4106788635253906, + "learning_rate": 1.6721234798877456e-06, + "loss": 1.391, + "step": 1677 + }, + { + "epoch": 2.1186868686868685, + "grad_norm": 3.412184476852417, + "learning_rate": 1.6697848456501404e-06, + "loss": 1.4245, + "step": 1678 + }, + { + "epoch": 2.119949494949495, + "grad_norm": 3.3659188747406006, + "learning_rate": 1.6674462114125353e-06, + "loss": 1.3358, + "step": 1679 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.241384744644165, + "learning_rate": 1.66510757717493e-06, + "loss": 1.3855, + "step": 1680 + }, + { + "epoch": 2.121212121212121, + "eval_loss": 1.6706457138061523, + "eval_runtime": 37.0651, + "eval_samples_per_second": 26.98, + "eval_steps_per_second": 1.133, + "step": 1680 + }, + { + "epoch": 2.1224747474747474, + "grad_norm": 3.3625874519348145, + "learning_rate": 1.6627689429373247e-06, + "loss": 1.3955, + "step": 1681 + }, + { + "epoch": 2.1237373737373737, + "grad_norm": 3.378143787384033, + "learning_rate": 1.6604303086997196e-06, + "loss": 1.3635, + "step": 1682 + }, + { + "epoch": 2.125, + "grad_norm": 3.4872324466705322, + "learning_rate": 1.6580916744621142e-06, + "loss": 1.4037, + "step": 1683 + }, + { + "epoch": 2.1262626262626263, + "grad_norm": 3.4733831882476807, + "learning_rate": 1.655753040224509e-06, + "loss": 1.3846, + "step": 1684 + }, + { + "epoch": 2.1275252525252526, + "grad_norm": 3.4518625736236572, + "learning_rate": 1.6534144059869037e-06, + "loss": 1.3984, + "step": 1685 + }, + { + "epoch": 2.128787878787879, + "grad_norm": 3.3702666759490967, + "learning_rate": 1.6510757717492987e-06, + "loss": 1.3712, + "step": 1686 + }, + { + "epoch": 2.130050505050505, + "grad_norm": 3.357708215713501, + "learning_rate": 1.6487371375116934e-06, + "loss": 1.4204, + "step": 1687 + }, + { + "epoch": 2.1313131313131315, + "grad_norm": 3.3902523517608643, + "learning_rate": 1.6463985032740882e-06, + "loss": 1.439, + "step": 1688 + }, + { + "epoch": 2.132575757575758, + "grad_norm": 3.246450424194336, + "learning_rate": 1.6440598690364828e-06, + "loss": 1.4061, + "step": 1689 + }, + { + "epoch": 2.1338383838383836, + "grad_norm": 3.5247316360473633, + "learning_rate": 1.6417212347988775e-06, + "loss": 1.4148, + "step": 1690 + }, + { + "epoch": 2.1338383838383836, + "eval_loss": 1.6684828996658325, + "eval_runtime": 37.3778, + "eval_samples_per_second": 26.754, + "eval_steps_per_second": 1.124, + "step": 1690 + }, + { + "epoch": 2.13510101010101, + "grad_norm": 3.3932440280914307, + "learning_rate": 1.6393826005612723e-06, + "loss": 1.3756, + "step": 1691 + }, + { + "epoch": 2.1363636363636362, + "grad_norm": 3.5242092609405518, + "learning_rate": 1.637043966323667e-06, + "loss": 1.4009, + "step": 1692 + }, + { + "epoch": 2.1376262626262625, + "grad_norm": 3.464235544204712, + "learning_rate": 1.6347053320860618e-06, + "loss": 1.4129, + "step": 1693 + }, + { + "epoch": 2.138888888888889, + "grad_norm": 3.4409964084625244, + "learning_rate": 1.6323666978484568e-06, + "loss": 1.3969, + "step": 1694 + }, + { + "epoch": 2.140151515151515, + "grad_norm": 3.329281806945801, + "learning_rate": 1.6300280636108515e-06, + "loss": 1.367, + "step": 1695 + }, + { + "epoch": 2.1414141414141414, + "grad_norm": 3.28047251701355, + "learning_rate": 1.6276894293732461e-06, + "loss": 1.3994, + "step": 1696 + }, + { + "epoch": 2.1426767676767677, + "grad_norm": 3.1937954425811768, + "learning_rate": 1.625350795135641e-06, + "loss": 1.3626, + "step": 1697 + }, + { + "epoch": 2.143939393939394, + "grad_norm": 3.5084168910980225, + "learning_rate": 1.6230121608980356e-06, + "loss": 1.4067, + "step": 1698 + }, + { + "epoch": 2.1452020202020203, + "grad_norm": 3.3637681007385254, + "learning_rate": 1.6206735266604304e-06, + "loss": 1.3619, + "step": 1699 + }, + { + "epoch": 2.1464646464646466, + "grad_norm": 3.2855823040008545, + "learning_rate": 1.618334892422825e-06, + "loss": 1.4227, + "step": 1700 + }, + { + "epoch": 2.1464646464646466, + "eval_loss": 1.6681337356567383, + "eval_runtime": 37.4375, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1700 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 5.299425995915264e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1700/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/latest new file mode 100644 index 0000000000000000000000000000000000000000..987364c74ed1983839a73116dad31651fadca561 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/latest @@ -0,0 +1 @@ +global_step1900 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..5656c2f89aa4e174075f551bd2bf019bab1d7eba --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-1900/trainer_state.json @@ -0,0 +1,14853 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.398989898989899, + "eval_steps": 10, + "global_step": 1900, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + }, + { + "epoch": 1.5164141414141414, + "grad_norm": 3.041095733642578, + "learning_rate": 2.782974742750234e-06, + "loss": 1.4736, + "step": 1201 + }, + { + "epoch": 1.5176767676767677, + "grad_norm": 2.8400142192840576, + "learning_rate": 2.780636108512629e-06, + "loss": 1.4169, + "step": 1202 + }, + { + "epoch": 1.518939393939394, + "grad_norm": 2.785592555999756, + "learning_rate": 2.7782974742750236e-06, + "loss": 1.4553, + "step": 1203 + }, + { + "epoch": 1.5202020202020203, + "grad_norm": 2.75209641456604, + "learning_rate": 2.775958840037418e-06, + "loss": 1.4267, + "step": 1204 + }, + { + "epoch": 1.5214646464646466, + "grad_norm": 2.927558183670044, + "learning_rate": 2.773620205799813e-06, + "loss": 1.4377, + "step": 1205 + }, + { + "epoch": 1.5227272727272727, + "grad_norm": 2.9335508346557617, + "learning_rate": 2.7712815715622077e-06, + "loss": 1.4295, + "step": 1206 + }, + { + "epoch": 1.523989898989899, + "grad_norm": 2.7812745571136475, + "learning_rate": 2.7689429373246026e-06, + "loss": 1.4029, + "step": 1207 + }, + { + "epoch": 1.5252525252525253, + "grad_norm": 3.230062961578369, + "learning_rate": 2.766604303086998e-06, + "loss": 1.539, + "step": 1208 + }, + { + "epoch": 1.5265151515151514, + "grad_norm": 3.1244332790374756, + "learning_rate": 2.7642656688493923e-06, + "loss": 1.4829, + "step": 1209 + }, + { + "epoch": 1.5277777777777777, + "grad_norm": 3.2391650676727295, + "learning_rate": 2.761927034611787e-06, + "loss": 1.4677, + "step": 1210 + }, + { + "epoch": 1.5277777777777777, + "eval_loss": 1.6849907636642456, + "eval_runtime": 36.8638, + "eval_samples_per_second": 27.127, + "eval_steps_per_second": 1.139, + "step": 1210 + }, + { + "epoch": 1.529040404040404, + "grad_norm": 3.1884987354278564, + "learning_rate": 2.759588400374182e-06, + "loss": 1.4185, + "step": 1211 + }, + { + "epoch": 1.5303030303030303, + "grad_norm": 2.959118366241455, + "learning_rate": 2.7572497661365764e-06, + "loss": 1.4536, + "step": 1212 + }, + { + "epoch": 1.5315656565656566, + "grad_norm": 2.882380247116089, + "learning_rate": 2.7549111318989712e-06, + "loss": 1.4872, + "step": 1213 + }, + { + "epoch": 1.5328282828282829, + "grad_norm": 3.0574429035186768, + "learning_rate": 2.752572497661366e-06, + "loss": 1.4343, + "step": 1214 + }, + { + "epoch": 1.5340909090909092, + "grad_norm": 3.1645233631134033, + "learning_rate": 2.750233863423761e-06, + "loss": 1.5101, + "step": 1215 + }, + { + "epoch": 1.5353535353535355, + "grad_norm": 2.83034086227417, + "learning_rate": 2.7478952291861553e-06, + "loss": 1.45, + "step": 1216 + }, + { + "epoch": 1.5366161616161618, + "grad_norm": 2.964780330657959, + "learning_rate": 2.74555659494855e-06, + "loss": 1.453, + "step": 1217 + }, + { + "epoch": 1.5378787878787878, + "grad_norm": 2.9896774291992188, + "learning_rate": 2.743217960710945e-06, + "loss": 1.4648, + "step": 1218 + }, + { + "epoch": 1.5391414141414141, + "grad_norm": 3.013664484024048, + "learning_rate": 2.74087932647334e-06, + "loss": 1.4828, + "step": 1219 + }, + { + "epoch": 1.5404040404040404, + "grad_norm": 3.0011773109436035, + "learning_rate": 2.7385406922357343e-06, + "loss": 1.4705, + "step": 1220 + }, + { + "epoch": 1.5404040404040404, + "eval_loss": 1.6872731447219849, + "eval_runtime": 36.9864, + "eval_samples_per_second": 27.037, + "eval_steps_per_second": 1.136, + "step": 1220 + }, + { + "epoch": 1.5416666666666665, + "grad_norm": 3.239349842071533, + "learning_rate": 2.736202057998129e-06, + "loss": 1.484, + "step": 1221 + }, + { + "epoch": 1.5429292929292928, + "grad_norm": 2.9267003536224365, + "learning_rate": 2.733863423760524e-06, + "loss": 1.4587, + "step": 1222 + }, + { + "epoch": 1.5441919191919191, + "grad_norm": 2.890094518661499, + "learning_rate": 2.7315247895229192e-06, + "loss": 1.4456, + "step": 1223 + }, + { + "epoch": 1.5454545454545454, + "grad_norm": 3.0287511348724365, + "learning_rate": 2.7291861552853136e-06, + "loss": 1.5059, + "step": 1224 + }, + { + "epoch": 1.5467171717171717, + "grad_norm": 3.041609525680542, + "learning_rate": 2.7268475210477085e-06, + "loss": 1.4341, + "step": 1225 + }, + { + "epoch": 1.547979797979798, + "grad_norm": 2.8771862983703613, + "learning_rate": 2.7245088868101033e-06, + "loss": 1.4308, + "step": 1226 + }, + { + "epoch": 1.5492424242424243, + "grad_norm": 3.1173794269561768, + "learning_rate": 2.722170252572498e-06, + "loss": 1.514, + "step": 1227 + }, + { + "epoch": 1.5505050505050506, + "grad_norm": 2.838275671005249, + "learning_rate": 2.7198316183348926e-06, + "loss": 1.4798, + "step": 1228 + }, + { + "epoch": 1.551767676767677, + "grad_norm": 2.9917166233062744, + "learning_rate": 2.7174929840972874e-06, + "loss": 1.5256, + "step": 1229 + }, + { + "epoch": 1.553030303030303, + "grad_norm": 3.071840763092041, + "learning_rate": 2.7151543498596823e-06, + "loss": 1.5187, + "step": 1230 + }, + { + "epoch": 1.553030303030303, + "eval_loss": 1.6850661039352417, + "eval_runtime": 37.0094, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 1230 + }, + { + "epoch": 1.5542929292929293, + "grad_norm": 3.253169536590576, + "learning_rate": 2.7128157156220767e-06, + "loss": 1.5522, + "step": 1231 + }, + { + "epoch": 1.5555555555555556, + "grad_norm": 3.037151336669922, + "learning_rate": 2.7104770813844715e-06, + "loss": 1.5125, + "step": 1232 + }, + { + "epoch": 1.5568181818181817, + "grad_norm": 2.8486485481262207, + "learning_rate": 2.7081384471468664e-06, + "loss": 1.4169, + "step": 1233 + }, + { + "epoch": 1.558080808080808, + "grad_norm": 2.953824996948242, + "learning_rate": 2.7057998129092612e-06, + "loss": 1.4505, + "step": 1234 + }, + { + "epoch": 1.5593434343434343, + "grad_norm": 3.012303590774536, + "learning_rate": 2.7034611786716557e-06, + "loss": 1.4816, + "step": 1235 + }, + { + "epoch": 1.5606060606060606, + "grad_norm": 3.1798768043518066, + "learning_rate": 2.7011225444340505e-06, + "loss": 1.5116, + "step": 1236 + }, + { + "epoch": 1.5618686868686869, + "grad_norm": 3.11370587348938, + "learning_rate": 2.6987839101964453e-06, + "loss": 1.5035, + "step": 1237 + }, + { + "epoch": 1.5631313131313131, + "grad_norm": 3.0563342571258545, + "learning_rate": 2.69644527595884e-06, + "loss": 1.4387, + "step": 1238 + }, + { + "epoch": 1.5643939393939394, + "grad_norm": 2.9884026050567627, + "learning_rate": 2.6941066417212354e-06, + "loss": 1.428, + "step": 1239 + }, + { + "epoch": 1.5656565656565657, + "grad_norm": 2.955773115158081, + "learning_rate": 2.69176800748363e-06, + "loss": 1.4208, + "step": 1240 + }, + { + "epoch": 1.5656565656565657, + "eval_loss": 1.685523509979248, + "eval_runtime": 37.0248, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1240 + }, + { + "epoch": 1.566919191919192, + "grad_norm": 2.9858005046844482, + "learning_rate": 2.6894293732460247e-06, + "loss": 1.4706, + "step": 1241 + }, + { + "epoch": 1.5681818181818183, + "grad_norm": 3.122340440750122, + "learning_rate": 2.6870907390084196e-06, + "loss": 1.4892, + "step": 1242 + }, + { + "epoch": 1.5694444444444444, + "grad_norm": 2.9828262329101562, + "learning_rate": 2.684752104770814e-06, + "loss": 1.4255, + "step": 1243 + }, + { + "epoch": 1.5707070707070707, + "grad_norm": 2.916653633117676, + "learning_rate": 2.682413470533209e-06, + "loss": 1.4241, + "step": 1244 + }, + { + "epoch": 1.571969696969697, + "grad_norm": 3.1615540981292725, + "learning_rate": 2.6800748362956037e-06, + "loss": 1.4719, + "step": 1245 + }, + { + "epoch": 1.573232323232323, + "grad_norm": 3.154130697250366, + "learning_rate": 2.6777362020579985e-06, + "loss": 1.4731, + "step": 1246 + }, + { + "epoch": 1.5744949494949494, + "grad_norm": 3.040848731994629, + "learning_rate": 2.675397567820393e-06, + "loss": 1.5149, + "step": 1247 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.9808318614959717, + "learning_rate": 2.6730589335827878e-06, + "loss": 1.431, + "step": 1248 + }, + { + "epoch": 1.577020202020202, + "grad_norm": 2.8538382053375244, + "learning_rate": 2.6707202993451826e-06, + "loss": 1.4458, + "step": 1249 + }, + { + "epoch": 1.5782828282828283, + "grad_norm": 3.113945960998535, + "learning_rate": 2.6683816651075774e-06, + "loss": 1.4873, + "step": 1250 + }, + { + "epoch": 1.5782828282828283, + "eval_loss": 1.6823841333389282, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1250 + }, + { + "epoch": 1.5795454545454546, + "grad_norm": 3.0044517517089844, + "learning_rate": 2.666043030869972e-06, + "loss": 1.4904, + "step": 1251 + }, + { + "epoch": 1.5808080808080809, + "grad_norm": 2.845820665359497, + "learning_rate": 2.6637043966323667e-06, + "loss": 1.4174, + "step": 1252 + }, + { + "epoch": 1.5820707070707072, + "grad_norm": 2.9123542308807373, + "learning_rate": 2.6613657623947616e-06, + "loss": 1.4896, + "step": 1253 + }, + { + "epoch": 1.5833333333333335, + "grad_norm": 3.1593222618103027, + "learning_rate": 2.659027128157157e-06, + "loss": 1.5208, + "step": 1254 + }, + { + "epoch": 1.5845959595959596, + "grad_norm": 3.0782854557037354, + "learning_rate": 2.6566884939195512e-06, + "loss": 1.4911, + "step": 1255 + }, + { + "epoch": 1.5858585858585859, + "grad_norm": 2.77309513092041, + "learning_rate": 2.654349859681946e-06, + "loss": 1.4475, + "step": 1256 + }, + { + "epoch": 1.5871212121212122, + "grad_norm": 2.9154622554779053, + "learning_rate": 2.652011225444341e-06, + "loss": 1.4324, + "step": 1257 + }, + { + "epoch": 1.5883838383838382, + "grad_norm": 3.107395887374878, + "learning_rate": 2.6496725912067358e-06, + "loss": 1.5126, + "step": 1258 + }, + { + "epoch": 1.5896464646464645, + "grad_norm": 2.825025796890259, + "learning_rate": 2.64733395696913e-06, + "loss": 1.4723, + "step": 1259 + }, + { + "epoch": 1.5909090909090908, + "grad_norm": 2.833296060562134, + "learning_rate": 2.644995322731525e-06, + "loss": 1.4516, + "step": 1260 + }, + { + "epoch": 1.5909090909090908, + "eval_loss": 1.6950432062149048, + "eval_runtime": 37.0343, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1260 + }, + { + "epoch": 1.5921717171717171, + "grad_norm": 2.9508421421051025, + "learning_rate": 2.64265668849392e-06, + "loss": 1.4002, + "step": 1261 + }, + { + "epoch": 1.5934343434343434, + "grad_norm": 2.8778491020202637, + "learning_rate": 2.6403180542563143e-06, + "loss": 1.4258, + "step": 1262 + }, + { + "epoch": 1.5946969696969697, + "grad_norm": 2.81261944770813, + "learning_rate": 2.637979420018709e-06, + "loss": 1.435, + "step": 1263 + }, + { + "epoch": 1.595959595959596, + "grad_norm": 3.1811511516571045, + "learning_rate": 2.635640785781104e-06, + "loss": 1.4792, + "step": 1264 + }, + { + "epoch": 1.5972222222222223, + "grad_norm": 3.0168304443359375, + "learning_rate": 2.633302151543499e-06, + "loss": 1.5394, + "step": 1265 + }, + { + "epoch": 1.5984848484848486, + "grad_norm": 2.9531311988830566, + "learning_rate": 2.6309635173058932e-06, + "loss": 1.4775, + "step": 1266 + }, + { + "epoch": 1.5997474747474747, + "grad_norm": 2.835839033126831, + "learning_rate": 2.628624883068288e-06, + "loss": 1.478, + "step": 1267 + }, + { + "epoch": 1.601010101010101, + "grad_norm": 2.9014039039611816, + "learning_rate": 2.626286248830683e-06, + "loss": 1.4926, + "step": 1268 + }, + { + "epoch": 1.6022727272727273, + "grad_norm": 2.89461088180542, + "learning_rate": 2.623947614593078e-06, + "loss": 1.448, + "step": 1269 + }, + { + "epoch": 1.6035353535353534, + "grad_norm": 2.8005120754241943, + "learning_rate": 2.621608980355473e-06, + "loss": 1.4733, + "step": 1270 + }, + { + "epoch": 1.6035353535353534, + "eval_loss": 1.6889415979385376, + "eval_runtime": 37.0345, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1270 + }, + { + "epoch": 1.6047979797979797, + "grad_norm": 2.9576175212860107, + "learning_rate": 2.6192703461178675e-06, + "loss": 1.5116, + "step": 1271 + }, + { + "epoch": 1.606060606060606, + "grad_norm": 2.926638603210449, + "learning_rate": 2.6169317118802623e-06, + "loss": 1.4744, + "step": 1272 + }, + { + "epoch": 1.6073232323232323, + "grad_norm": 2.911587715148926, + "learning_rate": 2.614593077642657e-06, + "loss": 1.4695, + "step": 1273 + }, + { + "epoch": 1.6085858585858586, + "grad_norm": 2.8878750801086426, + "learning_rate": 2.6122544434050516e-06, + "loss": 1.4574, + "step": 1274 + }, + { + "epoch": 1.6098484848484849, + "grad_norm": 2.8489067554473877, + "learning_rate": 2.6099158091674464e-06, + "loss": 1.4599, + "step": 1275 + }, + { + "epoch": 1.6111111111111112, + "grad_norm": 2.8223719596862793, + "learning_rate": 2.6075771749298412e-06, + "loss": 1.4431, + "step": 1276 + }, + { + "epoch": 1.6123737373737375, + "grad_norm": 2.869910955429077, + "learning_rate": 2.605238540692236e-06, + "loss": 1.48, + "step": 1277 + }, + { + "epoch": 1.6136363636363638, + "grad_norm": 2.8435168266296387, + "learning_rate": 2.6028999064546305e-06, + "loss": 1.4471, + "step": 1278 + }, + { + "epoch": 1.61489898989899, + "grad_norm": 2.9384665489196777, + "learning_rate": 2.6005612722170253e-06, + "loss": 1.5176, + "step": 1279 + }, + { + "epoch": 1.6161616161616161, + "grad_norm": 3.021785020828247, + "learning_rate": 2.59822263797942e-06, + "loss": 1.4789, + "step": 1280 + }, + { + "epoch": 1.6161616161616161, + "eval_loss": 1.6864516735076904, + "eval_runtime": 37.0249, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1280 + }, + { + "epoch": 1.6174242424242424, + "grad_norm": 3.007183313369751, + "learning_rate": 2.5958840037418146e-06, + "loss": 1.4686, + "step": 1281 + }, + { + "epoch": 1.6186868686868687, + "grad_norm": 3.115910053253174, + "learning_rate": 2.5935453695042095e-06, + "loss": 1.5382, + "step": 1282 + }, + { + "epoch": 1.6199494949494948, + "grad_norm": 2.98108172416687, + "learning_rate": 2.5912067352666043e-06, + "loss": 1.4569, + "step": 1283 + }, + { + "epoch": 1.621212121212121, + "grad_norm": 2.926832437515259, + "learning_rate": 2.5888681010289996e-06, + "loss": 1.4515, + "step": 1284 + }, + { + "epoch": 1.6224747474747474, + "grad_norm": 2.9045534133911133, + "learning_rate": 2.5865294667913944e-06, + "loss": 1.4259, + "step": 1285 + }, + { + "epoch": 1.6237373737373737, + "grad_norm": 3.0757768154144287, + "learning_rate": 2.584190832553789e-06, + "loss": 1.5436, + "step": 1286 + }, + { + "epoch": 1.625, + "grad_norm": 3.045257806777954, + "learning_rate": 2.5818521983161837e-06, + "loss": 1.4378, + "step": 1287 + }, + { + "epoch": 1.6262626262626263, + "grad_norm": 2.8131601810455322, + "learning_rate": 2.5795135640785785e-06, + "loss": 1.4224, + "step": 1288 + }, + { + "epoch": 1.6275252525252526, + "grad_norm": 2.953326940536499, + "learning_rate": 2.5771749298409734e-06, + "loss": 1.4137, + "step": 1289 + }, + { + "epoch": 1.628787878787879, + "grad_norm": 2.946232318878174, + "learning_rate": 2.5748362956033678e-06, + "loss": 1.4575, + "step": 1290 + }, + { + "epoch": 1.628787878787879, + "eval_loss": 1.6896334886550903, + "eval_runtime": 37.317, + "eval_samples_per_second": 26.797, + "eval_steps_per_second": 1.125, + "step": 1290 + }, + { + "epoch": 1.6300505050505052, + "grad_norm": 2.860867738723755, + "learning_rate": 2.5724976613657626e-06, + "loss": 1.4412, + "step": 1291 + }, + { + "epoch": 1.6313131313131313, + "grad_norm": 3.0417261123657227, + "learning_rate": 2.5701590271281575e-06, + "loss": 1.4919, + "step": 1292 + }, + { + "epoch": 1.6325757575757576, + "grad_norm": 3.21425199508667, + "learning_rate": 2.567820392890552e-06, + "loss": 1.5243, + "step": 1293 + }, + { + "epoch": 1.6338383838383839, + "grad_norm": 3.000302314758301, + "learning_rate": 2.5654817586529467e-06, + "loss": 1.4571, + "step": 1294 + }, + { + "epoch": 1.63510101010101, + "grad_norm": 2.88415789604187, + "learning_rate": 2.5631431244153416e-06, + "loss": 1.4527, + "step": 1295 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.7747156620025635, + "learning_rate": 2.5608044901777364e-06, + "loss": 1.3984, + "step": 1296 + }, + { + "epoch": 1.6376262626262625, + "grad_norm": 2.9298319816589355, + "learning_rate": 2.558465855940131e-06, + "loss": 1.4587, + "step": 1297 + }, + { + "epoch": 1.6388888888888888, + "grad_norm": 3.0199968814849854, + "learning_rate": 2.5561272217025257e-06, + "loss": 1.4539, + "step": 1298 + }, + { + "epoch": 1.6401515151515151, + "grad_norm": 3.09503436088562, + "learning_rate": 2.5537885874649205e-06, + "loss": 1.4551, + "step": 1299 + }, + { + "epoch": 1.6414141414141414, + "grad_norm": 2.972161293029785, + "learning_rate": 2.5514499532273158e-06, + "loss": 1.4123, + "step": 1300 + }, + { + "epoch": 1.6414141414141414, + "eval_loss": 1.6877377033233643, + "eval_runtime": 37.4088, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1300 + }, + { + "epoch": 1.6426767676767677, + "grad_norm": 3.081058979034424, + "learning_rate": 2.5491113189897106e-06, + "loss": 1.457, + "step": 1301 + }, + { + "epoch": 1.643939393939394, + "grad_norm": 3.040106773376465, + "learning_rate": 2.546772684752105e-06, + "loss": 1.4465, + "step": 1302 + }, + { + "epoch": 1.6452020202020203, + "grad_norm": 3.133812427520752, + "learning_rate": 2.5444340505145e-06, + "loss": 1.4382, + "step": 1303 + }, + { + "epoch": 1.6464646464646466, + "grad_norm": 2.9701125621795654, + "learning_rate": 2.5420954162768947e-06, + "loss": 1.457, + "step": 1304 + }, + { + "epoch": 1.6477272727272727, + "grad_norm": 2.9272050857543945, + "learning_rate": 2.539756782039289e-06, + "loss": 1.464, + "step": 1305 + }, + { + "epoch": 1.648989898989899, + "grad_norm": 3.048401117324829, + "learning_rate": 2.537418147801684e-06, + "loss": 1.4688, + "step": 1306 + }, + { + "epoch": 1.6502525252525253, + "grad_norm": 2.9103102684020996, + "learning_rate": 2.535079513564079e-06, + "loss": 1.5025, + "step": 1307 + }, + { + "epoch": 1.6515151515151514, + "grad_norm": 3.0841548442840576, + "learning_rate": 2.5327408793264737e-06, + "loss": 1.4938, + "step": 1308 + }, + { + "epoch": 1.6527777777777777, + "grad_norm": 3.022184371948242, + "learning_rate": 2.530402245088868e-06, + "loss": 1.4332, + "step": 1309 + }, + { + "epoch": 1.654040404040404, + "grad_norm": 2.881479024887085, + "learning_rate": 2.528063610851263e-06, + "loss": 1.4541, + "step": 1310 + }, + { + "epoch": 1.654040404040404, + "eval_loss": 1.6814675331115723, + "eval_runtime": 36.9446, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.6553030303030303, + "grad_norm": 2.906719446182251, + "learning_rate": 2.5257249766136578e-06, + "loss": 1.4968, + "step": 1311 + }, + { + "epoch": 1.6565656565656566, + "grad_norm": 3.0566461086273193, + "learning_rate": 2.523386342376052e-06, + "loss": 1.4513, + "step": 1312 + }, + { + "epoch": 1.6578282828282829, + "grad_norm": 2.895026206970215, + "learning_rate": 2.521047708138447e-06, + "loss": 1.4217, + "step": 1313 + }, + { + "epoch": 1.6590909090909092, + "grad_norm": 3.0006465911865234, + "learning_rate": 2.518709073900842e-06, + "loss": 1.4793, + "step": 1314 + }, + { + "epoch": 1.6603535353535355, + "grad_norm": 2.9853575229644775, + "learning_rate": 2.516370439663237e-06, + "loss": 1.4642, + "step": 1315 + }, + { + "epoch": 1.6616161616161618, + "grad_norm": 2.9729392528533936, + "learning_rate": 2.514031805425632e-06, + "loss": 1.4644, + "step": 1316 + }, + { + "epoch": 1.6628787878787878, + "grad_norm": 3.0997586250305176, + "learning_rate": 2.5116931711880264e-06, + "loss": 1.4392, + "step": 1317 + }, + { + "epoch": 1.6641414141414141, + "grad_norm": 2.9621379375457764, + "learning_rate": 2.5093545369504213e-06, + "loss": 1.4387, + "step": 1318 + }, + { + "epoch": 1.6654040404040404, + "grad_norm": 3.0571911334991455, + "learning_rate": 2.507015902712816e-06, + "loss": 1.4638, + "step": 1319 + }, + { + "epoch": 1.6666666666666665, + "grad_norm": 2.9580321311950684, + "learning_rate": 2.504677268475211e-06, + "loss": 1.4985, + "step": 1320 + }, + { + "epoch": 1.6666666666666665, + "eval_loss": 1.6790977716445923, + "eval_runtime": 37.0242, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1320 + }, + { + "epoch": 1.6679292929292928, + "grad_norm": 2.87432599067688, + "learning_rate": 2.5023386342376054e-06, + "loss": 1.4058, + "step": 1321 + }, + { + "epoch": 1.6691919191919191, + "grad_norm": 2.925546407699585, + "learning_rate": 2.5e-06, + "loss": 1.4986, + "step": 1322 + }, + { + "epoch": 1.6704545454545454, + "grad_norm": 2.837743043899536, + "learning_rate": 2.497661365762395e-06, + "loss": 1.455, + "step": 1323 + }, + { + "epoch": 1.6717171717171717, + "grad_norm": 3.111581563949585, + "learning_rate": 2.4953227315247895e-06, + "loss": 1.5005, + "step": 1324 + }, + { + "epoch": 1.672979797979798, + "grad_norm": 2.851076126098633, + "learning_rate": 2.4929840972871843e-06, + "loss": 1.4365, + "step": 1325 + }, + { + "epoch": 1.6742424242424243, + "grad_norm": 2.966965675354004, + "learning_rate": 2.4906454630495796e-06, + "loss": 1.4697, + "step": 1326 + }, + { + "epoch": 1.6755050505050506, + "grad_norm": 3.015193223953247, + "learning_rate": 2.488306828811974e-06, + "loss": 1.4486, + "step": 1327 + }, + { + "epoch": 1.676767676767677, + "grad_norm": 3.017681121826172, + "learning_rate": 2.485968194574369e-06, + "loss": 1.4017, + "step": 1328 + }, + { + "epoch": 1.678030303030303, + "grad_norm": 3.071981906890869, + "learning_rate": 2.4836295603367637e-06, + "loss": 1.4572, + "step": 1329 + }, + { + "epoch": 1.6792929292929293, + "grad_norm": 3.051741123199463, + "learning_rate": 2.481290926099158e-06, + "loss": 1.4498, + "step": 1330 + }, + { + "epoch": 1.6792929292929293, + "eval_loss": 1.6771777868270874, + "eval_runtime": 37.0747, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1330 + }, + { + "epoch": 1.6805555555555556, + "grad_norm": 2.9561071395874023, + "learning_rate": 2.478952291861553e-06, + "loss": 1.4428, + "step": 1331 + }, + { + "epoch": 1.6818181818181817, + "grad_norm": 2.992584466934204, + "learning_rate": 2.4766136576239478e-06, + "loss": 1.461, + "step": 1332 + }, + { + "epoch": 1.683080808080808, + "grad_norm": 3.0473921298980713, + "learning_rate": 2.4742750233863426e-06, + "loss": 1.489, + "step": 1333 + }, + { + "epoch": 1.6843434343434343, + "grad_norm": 2.8721725940704346, + "learning_rate": 2.4719363891487375e-06, + "loss": 1.487, + "step": 1334 + }, + { + "epoch": 1.6856060606060606, + "grad_norm": 2.8650765419006348, + "learning_rate": 2.4695977549111323e-06, + "loss": 1.4504, + "step": 1335 + }, + { + "epoch": 1.6868686868686869, + "grad_norm": 2.998887538909912, + "learning_rate": 2.4672591206735267e-06, + "loss": 1.4586, + "step": 1336 + }, + { + "epoch": 1.6881313131313131, + "grad_norm": 2.915188789367676, + "learning_rate": 2.4649204864359216e-06, + "loss": 1.4333, + "step": 1337 + }, + { + "epoch": 1.6893939393939394, + "grad_norm": 2.9660091400146484, + "learning_rate": 2.4625818521983164e-06, + "loss": 1.4711, + "step": 1338 + }, + { + "epoch": 1.6906565656565657, + "grad_norm": 2.916274070739746, + "learning_rate": 2.4602432179607113e-06, + "loss": 1.4642, + "step": 1339 + }, + { + "epoch": 1.691919191919192, + "grad_norm": 3.04154896736145, + "learning_rate": 2.4579045837231057e-06, + "loss": 1.4233, + "step": 1340 + }, + { + "epoch": 1.691919191919192, + "eval_loss": 1.6764302253723145, + "eval_runtime": 37.0686, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1340 + }, + { + "epoch": 1.6931818181818183, + "grad_norm": 2.9976465702056885, + "learning_rate": 2.4555659494855005e-06, + "loss": 1.4775, + "step": 1341 + }, + { + "epoch": 1.6944444444444444, + "grad_norm": 2.9140090942382812, + "learning_rate": 2.4532273152478954e-06, + "loss": 1.4532, + "step": 1342 + }, + { + "epoch": 1.6957070707070707, + "grad_norm": 3.0224454402923584, + "learning_rate": 2.45088868101029e-06, + "loss": 1.5091, + "step": 1343 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.8979439735412598, + "learning_rate": 2.448550046772685e-06, + "loss": 1.4634, + "step": 1344 + }, + { + "epoch": 1.698232323232323, + "grad_norm": 2.913433313369751, + "learning_rate": 2.44621141253508e-06, + "loss": 1.4483, + "step": 1345 + }, + { + "epoch": 1.6994949494949494, + "grad_norm": 2.813455820083618, + "learning_rate": 2.4438727782974743e-06, + "loss": 1.446, + "step": 1346 + }, + { + "epoch": 1.7007575757575757, + "grad_norm": 2.988722324371338, + "learning_rate": 2.441534144059869e-06, + "loss": 1.4908, + "step": 1347 + }, + { + "epoch": 1.702020202020202, + "grad_norm": 2.850446939468384, + "learning_rate": 2.439195509822264e-06, + "loss": 1.4221, + "step": 1348 + }, + { + "epoch": 1.7032828282828283, + "grad_norm": 2.8405508995056152, + "learning_rate": 2.436856875584659e-06, + "loss": 1.4496, + "step": 1349 + }, + { + "epoch": 1.7045454545454546, + "grad_norm": 3.0188651084899902, + "learning_rate": 2.4345182413470537e-06, + "loss": 1.489, + "step": 1350 + }, + { + "epoch": 1.7045454545454546, + "eval_loss": 1.6780083179473877, + "eval_runtime": 37.0724, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1350 + }, + { + "epoch": 1.7058080808080809, + "grad_norm": 2.905088424682617, + "learning_rate": 2.4321796071094485e-06, + "loss": 1.4283, + "step": 1351 + }, + { + "epoch": 1.7070707070707072, + "grad_norm": 2.878998279571533, + "learning_rate": 2.429840972871843e-06, + "loss": 1.4357, + "step": 1352 + }, + { + "epoch": 1.7083333333333335, + "grad_norm": 3.1707355976104736, + "learning_rate": 2.427502338634238e-06, + "loss": 1.496, + "step": 1353 + }, + { + "epoch": 1.7095959595959596, + "grad_norm": 2.9610493183135986, + "learning_rate": 2.4251637043966326e-06, + "loss": 1.4905, + "step": 1354 + }, + { + "epoch": 1.7108585858585859, + "grad_norm": 2.8372201919555664, + "learning_rate": 2.422825070159027e-06, + "loss": 1.4563, + "step": 1355 + }, + { + "epoch": 1.7121212121212122, + "grad_norm": 2.8741753101348877, + "learning_rate": 2.420486435921422e-06, + "loss": 1.4525, + "step": 1356 + }, + { + "epoch": 1.7133838383838382, + "grad_norm": 2.973726749420166, + "learning_rate": 2.418147801683817e-06, + "loss": 1.4878, + "step": 1357 + }, + { + "epoch": 1.7146464646464645, + "grad_norm": 3.05717396736145, + "learning_rate": 2.4158091674462116e-06, + "loss": 1.5402, + "step": 1358 + }, + { + "epoch": 1.7159090909090908, + "grad_norm": 2.883772373199463, + "learning_rate": 2.4134705332086064e-06, + "loss": 1.4329, + "step": 1359 + }, + { + "epoch": 1.7171717171717171, + "grad_norm": 2.8246653079986572, + "learning_rate": 2.4111318989710013e-06, + "loss": 1.4821, + "step": 1360 + }, + { + "epoch": 1.7171717171717171, + "eval_loss": 1.6737746000289917, + "eval_runtime": 37.0666, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 1360 + }, + { + "epoch": 1.7184343434343434, + "grad_norm": 2.905977964401245, + "learning_rate": 2.4087932647333957e-06, + "loss": 1.4378, + "step": 1361 + }, + { + "epoch": 1.7196969696969697, + "grad_norm": 2.9268195629119873, + "learning_rate": 2.4064546304957905e-06, + "loss": 1.4765, + "step": 1362 + }, + { + "epoch": 1.720959595959596, + "grad_norm": 3.1369476318359375, + "learning_rate": 2.4041159962581854e-06, + "loss": 1.4532, + "step": 1363 + }, + { + "epoch": 1.7222222222222223, + "grad_norm": 3.04546856880188, + "learning_rate": 2.4017773620205802e-06, + "loss": 1.4921, + "step": 1364 + }, + { + "epoch": 1.7234848484848486, + "grad_norm": 2.872852325439453, + "learning_rate": 2.399438727782975e-06, + "loss": 1.4323, + "step": 1365 + }, + { + "epoch": 1.7247474747474747, + "grad_norm": 2.961681604385376, + "learning_rate": 2.39710009354537e-06, + "loss": 1.5181, + "step": 1366 + }, + { + "epoch": 1.726010101010101, + "grad_norm": 2.997729778289795, + "learning_rate": 2.3947614593077643e-06, + "loss": 1.4817, + "step": 1367 + }, + { + "epoch": 1.7272727272727273, + "grad_norm": 2.9097466468811035, + "learning_rate": 2.392422825070159e-06, + "loss": 1.4478, + "step": 1368 + }, + { + "epoch": 1.7285353535353534, + "grad_norm": 2.82433819770813, + "learning_rate": 2.390084190832554e-06, + "loss": 1.4065, + "step": 1369 + }, + { + "epoch": 1.7297979797979797, + "grad_norm": 2.940910577774048, + "learning_rate": 2.387745556594949e-06, + "loss": 1.4735, + "step": 1370 + }, + { + "epoch": 1.7297979797979797, + "eval_loss": 1.6717814207077026, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1370 + }, + { + "epoch": 1.731060606060606, + "grad_norm": 2.79101300239563, + "learning_rate": 2.3854069223573433e-06, + "loss": 1.4545, + "step": 1371 + }, + { + "epoch": 1.7323232323232323, + "grad_norm": 2.916954517364502, + "learning_rate": 2.3830682881197385e-06, + "loss": 1.4293, + "step": 1372 + }, + { + "epoch": 1.7335858585858586, + "grad_norm": 2.9609909057617188, + "learning_rate": 2.380729653882133e-06, + "loss": 1.4578, + "step": 1373 + }, + { + "epoch": 1.7348484848484849, + "grad_norm": 2.935925245285034, + "learning_rate": 2.378391019644528e-06, + "loss": 1.4577, + "step": 1374 + }, + { + "epoch": 1.7361111111111112, + "grad_norm": 2.9803526401519775, + "learning_rate": 2.3760523854069226e-06, + "loss": 1.4449, + "step": 1375 + }, + { + "epoch": 1.7373737373737375, + "grad_norm": 3.0292913913726807, + "learning_rate": 2.3737137511693175e-06, + "loss": 1.4308, + "step": 1376 + }, + { + "epoch": 1.7386363636363638, + "grad_norm": 3.0247879028320312, + "learning_rate": 2.371375116931712e-06, + "loss": 1.5406, + "step": 1377 + }, + { + "epoch": 1.73989898989899, + "grad_norm": 2.86775541305542, + "learning_rate": 2.3690364826941067e-06, + "loss": 1.4324, + "step": 1378 + }, + { + "epoch": 1.7411616161616161, + "grad_norm": 2.8593552112579346, + "learning_rate": 2.3666978484565016e-06, + "loss": 1.4526, + "step": 1379 + }, + { + "epoch": 1.7424242424242424, + "grad_norm": 2.8735320568084717, + "learning_rate": 2.3643592142188964e-06, + "loss": 1.4097, + "step": 1380 + }, + { + "epoch": 1.7424242424242424, + "eval_loss": 1.675827145576477, + "eval_runtime": 37.1206, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1380 + }, + { + "epoch": 1.7436868686868687, + "grad_norm": 2.948153495788574, + "learning_rate": 2.3620205799812913e-06, + "loss": 1.4234, + "step": 1381 + }, + { + "epoch": 1.7449494949494948, + "grad_norm": 2.9344282150268555, + "learning_rate": 2.359681945743686e-06, + "loss": 1.4653, + "step": 1382 + }, + { + "epoch": 1.746212121212121, + "grad_norm": 2.9309439659118652, + "learning_rate": 2.3573433115060805e-06, + "loss": 1.4676, + "step": 1383 + }, + { + "epoch": 1.7474747474747474, + "grad_norm": 3.0762970447540283, + "learning_rate": 2.3550046772684754e-06, + "loss": 1.4407, + "step": 1384 + }, + { + "epoch": 1.7487373737373737, + "grad_norm": 3.149824857711792, + "learning_rate": 2.3526660430308702e-06, + "loss": 1.4629, + "step": 1385 + }, + { + "epoch": 1.75, + "grad_norm": 3.109571695327759, + "learning_rate": 2.3503274087932646e-06, + "loss": 1.494, + "step": 1386 + }, + { + "epoch": 1.7512626262626263, + "grad_norm": 2.8359758853912354, + "learning_rate": 2.34798877455566e-06, + "loss": 1.4056, + "step": 1387 + }, + { + "epoch": 1.7525252525252526, + "grad_norm": 2.8523781299591064, + "learning_rate": 2.3456501403180548e-06, + "loss": 1.4389, + "step": 1388 + }, + { + "epoch": 1.753787878787879, + "grad_norm": 3.0218093395233154, + "learning_rate": 2.343311506080449e-06, + "loss": 1.4587, + "step": 1389 + }, + { + "epoch": 1.7550505050505052, + "grad_norm": 3.0975496768951416, + "learning_rate": 2.340972871842844e-06, + "loss": 1.5133, + "step": 1390 + }, + { + "epoch": 1.7550505050505052, + "eval_loss": 1.671931266784668, + "eval_runtime": 37.2749, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 1390 + }, + { + "epoch": 1.7563131313131313, + "grad_norm": 3.0177836418151855, + "learning_rate": 2.338634237605239e-06, + "loss": 1.4716, + "step": 1391 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 3.049105167388916, + "learning_rate": 2.3362956033676333e-06, + "loss": 1.4858, + "step": 1392 + }, + { + "epoch": 1.7588383838383839, + "grad_norm": 3.0062026977539062, + "learning_rate": 2.333956969130028e-06, + "loss": 1.4239, + "step": 1393 + }, + { + "epoch": 1.76010101010101, + "grad_norm": 3.071772813796997, + "learning_rate": 2.331618334892423e-06, + "loss": 1.4654, + "step": 1394 + }, + { + "epoch": 1.7613636363636362, + "grad_norm": 3.140510320663452, + "learning_rate": 2.329279700654818e-06, + "loss": 1.4854, + "step": 1395 + }, + { + "epoch": 1.7626262626262625, + "grad_norm": 2.9903907775878906, + "learning_rate": 2.3269410664172126e-06, + "loss": 1.4406, + "step": 1396 + }, + { + "epoch": 1.7638888888888888, + "grad_norm": 2.839891195297241, + "learning_rate": 2.3246024321796075e-06, + "loss": 1.4498, + "step": 1397 + }, + { + "epoch": 1.7651515151515151, + "grad_norm": 3.0252437591552734, + "learning_rate": 2.322263797942002e-06, + "loss": 1.472, + "step": 1398 + }, + { + "epoch": 1.7664141414141414, + "grad_norm": 3.0214762687683105, + "learning_rate": 2.3199251637043968e-06, + "loss": 1.4031, + "step": 1399 + }, + { + "epoch": 1.7676767676767677, + "grad_norm": 3.190088987350464, + "learning_rate": 2.3175865294667916e-06, + "loss": 1.4497, + "step": 1400 + }, + { + "epoch": 1.7676767676767677, + "eval_loss": 1.673120379447937, + "eval_runtime": 37.4414, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1400 + }, + { + "epoch": 1.768939393939394, + "grad_norm": 2.9620201587677, + "learning_rate": 2.3152478952291864e-06, + "loss": 1.4149, + "step": 1401 + }, + { + "epoch": 1.7702020202020203, + "grad_norm": 3.112372875213623, + "learning_rate": 2.312909260991581e-06, + "loss": 1.4569, + "step": 1402 + }, + { + "epoch": 1.7714646464646466, + "grad_norm": 3.138857364654541, + "learning_rate": 2.310570626753976e-06, + "loss": 1.4819, + "step": 1403 + }, + { + "epoch": 1.7727272727272727, + "grad_norm": 3.0957937240600586, + "learning_rate": 2.3082319925163705e-06, + "loss": 1.4374, + "step": 1404 + }, + { + "epoch": 1.773989898989899, + "grad_norm": 3.014039993286133, + "learning_rate": 2.3058933582787654e-06, + "loss": 1.4262, + "step": 1405 + }, + { + "epoch": 1.7752525252525253, + "grad_norm": 2.8127005100250244, + "learning_rate": 2.3035547240411602e-06, + "loss": 1.4468, + "step": 1406 + }, + { + "epoch": 1.7765151515151514, + "grad_norm": 2.9180614948272705, + "learning_rate": 2.301216089803555e-06, + "loss": 1.4259, + "step": 1407 + }, + { + "epoch": 1.7777777777777777, + "grad_norm": 3.0376124382019043, + "learning_rate": 2.2988774555659495e-06, + "loss": 1.4628, + "step": 1408 + }, + { + "epoch": 1.779040404040404, + "grad_norm": 3.031740665435791, + "learning_rate": 2.2965388213283443e-06, + "loss": 1.5058, + "step": 1409 + }, + { + "epoch": 1.7803030303030303, + "grad_norm": 2.9876646995544434, + "learning_rate": 2.294200187090739e-06, + "loss": 1.4605, + "step": 1410 + }, + { + "epoch": 1.7803030303030303, + "eval_loss": 1.668077826499939, + "eval_runtime": 36.9734, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 1410 + }, + { + "epoch": 1.7815656565656566, + "grad_norm": 2.953744888305664, + "learning_rate": 2.291861552853134e-06, + "loss": 1.4257, + "step": 1411 + }, + { + "epoch": 1.7828282828282829, + "grad_norm": 3.049036741256714, + "learning_rate": 2.289522918615529e-06, + "loss": 1.4476, + "step": 1412 + }, + { + "epoch": 1.7840909090909092, + "grad_norm": 3.102116584777832, + "learning_rate": 2.2871842843779237e-06, + "loss": 1.4701, + "step": 1413 + }, + { + "epoch": 1.7853535353535355, + "grad_norm": 2.842970132827759, + "learning_rate": 2.284845650140318e-06, + "loss": 1.415, + "step": 1414 + }, + { + "epoch": 1.7866161616161618, + "grad_norm": 2.9386725425720215, + "learning_rate": 2.282507015902713e-06, + "loss": 1.4561, + "step": 1415 + }, + { + "epoch": 1.7878787878787878, + "grad_norm": 2.942713499069214, + "learning_rate": 2.280168381665108e-06, + "loss": 1.435, + "step": 1416 + }, + { + "epoch": 1.7891414141414141, + "grad_norm": 2.943105936050415, + "learning_rate": 2.2778297474275022e-06, + "loss": 1.4923, + "step": 1417 + }, + { + "epoch": 1.7904040404040404, + "grad_norm": 2.7094717025756836, + "learning_rate": 2.2754911131898975e-06, + "loss": 1.4457, + "step": 1418 + }, + { + "epoch": 1.7916666666666665, + "grad_norm": 2.7859625816345215, + "learning_rate": 2.2731524789522923e-06, + "loss": 1.3993, + "step": 1419 + }, + { + "epoch": 1.7929292929292928, + "grad_norm": 3.0923988819122314, + "learning_rate": 2.2708138447146868e-06, + "loss": 1.5054, + "step": 1420 + }, + { + "epoch": 1.7929292929292928, + "eval_loss": 1.6687692403793335, + "eval_runtime": 37.0567, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.7941919191919191, + "grad_norm": 2.849726676940918, + "learning_rate": 2.2684752104770816e-06, + "loss": 1.3978, + "step": 1421 + }, + { + "epoch": 1.7954545454545454, + "grad_norm": 3.028932809829712, + "learning_rate": 2.2661365762394764e-06, + "loss": 1.4786, + "step": 1422 + }, + { + "epoch": 1.7967171717171717, + "grad_norm": 2.973188638687134, + "learning_rate": 2.263797942001871e-06, + "loss": 1.4393, + "step": 1423 + }, + { + "epoch": 1.797979797979798, + "grad_norm": 2.884032964706421, + "learning_rate": 2.2614593077642657e-06, + "loss": 1.4197, + "step": 1424 + }, + { + "epoch": 1.7992424242424243, + "grad_norm": 2.9058549404144287, + "learning_rate": 2.2591206735266606e-06, + "loss": 1.4313, + "step": 1425 + }, + { + "epoch": 1.8005050505050506, + "grad_norm": 2.905841827392578, + "learning_rate": 2.2567820392890554e-06, + "loss": 1.439, + "step": 1426 + }, + { + "epoch": 1.801767676767677, + "grad_norm": 2.927028179168701, + "learning_rate": 2.2544434050514502e-06, + "loss": 1.4296, + "step": 1427 + }, + { + "epoch": 1.803030303030303, + "grad_norm": 2.965529203414917, + "learning_rate": 2.252104770813845e-06, + "loss": 1.4185, + "step": 1428 + }, + { + "epoch": 1.8042929292929293, + "grad_norm": 2.9494433403015137, + "learning_rate": 2.2497661365762395e-06, + "loss": 1.5087, + "step": 1429 + }, + { + "epoch": 1.8055555555555556, + "grad_norm": 3.0137765407562256, + "learning_rate": 2.2474275023386343e-06, + "loss": 1.4383, + "step": 1430 + }, + { + "epoch": 1.8055555555555556, + "eval_loss": 1.6728461980819702, + "eval_runtime": 37.1033, + "eval_samples_per_second": 26.952, + "eval_steps_per_second": 1.132, + "step": 1430 + }, + { + "epoch": 1.8068181818181817, + "grad_norm": 2.980062484741211, + "learning_rate": 2.245088868101029e-06, + "loss": 1.4333, + "step": 1431 + }, + { + "epoch": 1.808080808080808, + "grad_norm": 3.143759250640869, + "learning_rate": 2.242750233863424e-06, + "loss": 1.4676, + "step": 1432 + }, + { + "epoch": 1.8093434343434343, + "grad_norm": 3.0134518146514893, + "learning_rate": 2.240411599625819e-06, + "loss": 1.4223, + "step": 1433 + }, + { + "epoch": 1.8106060606060606, + "grad_norm": 3.000018358230591, + "learning_rate": 2.2380729653882137e-06, + "loss": 1.4537, + "step": 1434 + }, + { + "epoch": 1.8118686868686869, + "grad_norm": 2.9828288555145264, + "learning_rate": 2.235734331150608e-06, + "loss": 1.4379, + "step": 1435 + }, + { + "epoch": 1.8131313131313131, + "grad_norm": 2.965486764907837, + "learning_rate": 2.233395696913003e-06, + "loss": 1.4244, + "step": 1436 + }, + { + "epoch": 1.8143939393939394, + "grad_norm": 2.9787070751190186, + "learning_rate": 2.231057062675398e-06, + "loss": 1.4117, + "step": 1437 + }, + { + "epoch": 1.8156565656565657, + "grad_norm": 3.0893537998199463, + "learning_rate": 2.2287184284377927e-06, + "loss": 1.4809, + "step": 1438 + }, + { + "epoch": 1.816919191919192, + "grad_norm": 2.8568801879882812, + "learning_rate": 2.226379794200187e-06, + "loss": 1.4232, + "step": 1439 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 3.1005046367645264, + "learning_rate": 2.224041159962582e-06, + "loss": 1.482, + "step": 1440 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.6738661527633667, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1440 + }, + { + "epoch": 1.8194444444444444, + "grad_norm": 2.9064700603485107, + "learning_rate": 2.2217025257249768e-06, + "loss": 1.4179, + "step": 1441 + }, + { + "epoch": 1.8207070707070707, + "grad_norm": 3.0111782550811768, + "learning_rate": 2.2193638914873716e-06, + "loss": 1.4084, + "step": 1442 + }, + { + "epoch": 1.821969696969697, + "grad_norm": 2.9269816875457764, + "learning_rate": 2.2170252572497665e-06, + "loss": 1.4361, + "step": 1443 + }, + { + "epoch": 1.823232323232323, + "grad_norm": 2.9985902309417725, + "learning_rate": 2.2146866230121613e-06, + "loss": 1.4345, + "step": 1444 + }, + { + "epoch": 1.8244949494949494, + "grad_norm": 3.1306066513061523, + "learning_rate": 2.2123479887745557e-06, + "loss": 1.4728, + "step": 1445 + }, + { + "epoch": 1.8257575757575757, + "grad_norm": 2.9876086711883545, + "learning_rate": 2.2100093545369506e-06, + "loss": 1.378, + "step": 1446 + }, + { + "epoch": 1.827020202020202, + "grad_norm": 3.018763780593872, + "learning_rate": 2.2076707202993454e-06, + "loss": 1.4506, + "step": 1447 + }, + { + "epoch": 1.8282828282828283, + "grad_norm": 2.960573673248291, + "learning_rate": 2.2053320860617402e-06, + "loss": 1.4674, + "step": 1448 + }, + { + "epoch": 1.8295454545454546, + "grad_norm": 2.849881887435913, + "learning_rate": 2.202993451824135e-06, + "loss": 1.4687, + "step": 1449 + }, + { + "epoch": 1.8308080808080809, + "grad_norm": 2.8798882961273193, + "learning_rate": 2.2006548175865295e-06, + "loss": 1.3868, + "step": 1450 + }, + { + "epoch": 1.8308080808080809, + "eval_loss": 1.6657519340515137, + "eval_runtime": 37.0993, + "eval_samples_per_second": 26.955, + "eval_steps_per_second": 1.132, + "step": 1450 + }, + { + "epoch": 1.8320707070707072, + "grad_norm": 2.937471628189087, + "learning_rate": 2.1983161833489243e-06, + "loss": 1.4085, + "step": 1451 + }, + { + "epoch": 1.8333333333333335, + "grad_norm": 3.0815699100494385, + "learning_rate": 2.195977549111319e-06, + "loss": 1.4055, + "step": 1452 + }, + { + "epoch": 1.8345959595959596, + "grad_norm": 2.927917718887329, + "learning_rate": 2.193638914873714e-06, + "loss": 1.4014, + "step": 1453 + }, + { + "epoch": 1.8358585858585859, + "grad_norm": 2.9576570987701416, + "learning_rate": 2.1913002806361085e-06, + "loss": 1.4394, + "step": 1454 + }, + { + "epoch": 1.8371212121212122, + "grad_norm": 2.9505863189697266, + "learning_rate": 2.1889616463985033e-06, + "loss": 1.4018, + "step": 1455 + }, + { + "epoch": 1.8383838383838382, + "grad_norm": 3.014965772628784, + "learning_rate": 2.186623012160898e-06, + "loss": 1.3735, + "step": 1456 + }, + { + "epoch": 1.8396464646464645, + "grad_norm": 3.121122121810913, + "learning_rate": 2.184284377923293e-06, + "loss": 1.4289, + "step": 1457 + }, + { + "epoch": 1.8409090909090908, + "grad_norm": 2.9710142612457275, + "learning_rate": 2.181945743685688e-06, + "loss": 1.4494, + "step": 1458 + }, + { + "epoch": 1.8421717171717171, + "grad_norm": 3.028554916381836, + "learning_rate": 2.1796071094480827e-06, + "loss": 1.5151, + "step": 1459 + }, + { + "epoch": 1.8434343434343434, + "grad_norm": 2.8973751068115234, + "learning_rate": 2.177268475210477e-06, + "loss": 1.4171, + "step": 1460 + }, + { + "epoch": 1.8434343434343434, + "eval_loss": 1.6670986413955688, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1460 + }, + { + "epoch": 1.8446969696969697, + "grad_norm": 3.046865701675415, + "learning_rate": 2.174929840972872e-06, + "loss": 1.4016, + "step": 1461 + }, + { + "epoch": 1.845959595959596, + "grad_norm": 3.0509603023529053, + "learning_rate": 2.1725912067352668e-06, + "loss": 1.4196, + "step": 1462 + }, + { + "epoch": 1.8472222222222223, + "grad_norm": 2.951892137527466, + "learning_rate": 2.1702525724976616e-06, + "loss": 1.4063, + "step": 1463 + }, + { + "epoch": 1.8484848484848486, + "grad_norm": 3.2126667499542236, + "learning_rate": 2.1679139382600565e-06, + "loss": 1.5106, + "step": 1464 + }, + { + "epoch": 1.8497474747474747, + "grad_norm": 2.965576648712158, + "learning_rate": 2.1655753040224513e-06, + "loss": 1.4096, + "step": 1465 + }, + { + "epoch": 1.851010101010101, + "grad_norm": 2.8640592098236084, + "learning_rate": 2.1632366697848457e-06, + "loss": 1.389, + "step": 1466 + }, + { + "epoch": 1.8522727272727273, + "grad_norm": 3.1476943492889404, + "learning_rate": 2.1608980355472406e-06, + "loss": 1.4793, + "step": 1467 + }, + { + "epoch": 1.8535353535353534, + "grad_norm": 2.814695119857788, + "learning_rate": 2.1585594013096354e-06, + "loss": 1.4265, + "step": 1468 + }, + { + "epoch": 1.8547979797979797, + "grad_norm": 2.9728431701660156, + "learning_rate": 2.1562207670720302e-06, + "loss": 1.386, + "step": 1469 + }, + { + "epoch": 1.856060606060606, + "grad_norm": 2.9062652587890625, + "learning_rate": 2.1538821328344247e-06, + "loss": 1.3986, + "step": 1470 + }, + { + "epoch": 1.856060606060606, + "eval_loss": 1.6698763370513916, + "eval_runtime": 37.1129, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1470 + }, + { + "epoch": 1.8573232323232323, + "grad_norm": 2.950857162475586, + "learning_rate": 2.15154349859682e-06, + "loss": 1.4193, + "step": 1471 + }, + { + "epoch": 1.8585858585858586, + "grad_norm": 3.017594337463379, + "learning_rate": 2.1492048643592144e-06, + "loss": 1.4318, + "step": 1472 + }, + { + "epoch": 1.8598484848484849, + "grad_norm": 2.991514205932617, + "learning_rate": 2.146866230121609e-06, + "loss": 1.4462, + "step": 1473 + }, + { + "epoch": 1.8611111111111112, + "grad_norm": 3.149475336074829, + "learning_rate": 2.144527595884004e-06, + "loss": 1.4486, + "step": 1474 + }, + { + "epoch": 1.8623737373737375, + "grad_norm": 2.948828935623169, + "learning_rate": 2.1421889616463985e-06, + "loss": 1.4522, + "step": 1475 + }, + { + "epoch": 1.8636363636363638, + "grad_norm": 3.0383694171905518, + "learning_rate": 2.1398503274087933e-06, + "loss": 1.45, + "step": 1476 + }, + { + "epoch": 1.86489898989899, + "grad_norm": 3.093449831008911, + "learning_rate": 2.137511693171188e-06, + "loss": 1.4266, + "step": 1477 + }, + { + "epoch": 1.8661616161616161, + "grad_norm": 3.017549753189087, + "learning_rate": 2.135173058933583e-06, + "loss": 1.4067, + "step": 1478 + }, + { + "epoch": 1.8674242424242424, + "grad_norm": 2.985044002532959, + "learning_rate": 2.132834424695978e-06, + "loss": 1.4904, + "step": 1479 + }, + { + "epoch": 1.8686868686868687, + "grad_norm": 2.9600346088409424, + "learning_rate": 2.1304957904583727e-06, + "loss": 1.4006, + "step": 1480 + }, + { + "epoch": 1.8686868686868687, + "eval_loss": 1.6648480892181396, + "eval_runtime": 37.1225, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1480 + }, + { + "epoch": 1.8699494949494948, + "grad_norm": 2.8905625343322754, + "learning_rate": 2.128157156220767e-06, + "loss": 1.4096, + "step": 1481 + }, + { + "epoch": 1.871212121212121, + "grad_norm": 2.95342755317688, + "learning_rate": 2.125818521983162e-06, + "loss": 1.3888, + "step": 1482 + }, + { + "epoch": 1.8724747474747474, + "grad_norm": 2.9438536167144775, + "learning_rate": 2.1234798877455568e-06, + "loss": 1.413, + "step": 1483 + }, + { + "epoch": 1.8737373737373737, + "grad_norm": 3.1257364749908447, + "learning_rate": 2.1211412535079516e-06, + "loss": 1.4902, + "step": 1484 + }, + { + "epoch": 1.875, + "grad_norm": 3.1377382278442383, + "learning_rate": 2.118802619270346e-06, + "loss": 1.4815, + "step": 1485 + }, + { + "epoch": 1.8762626262626263, + "grad_norm": 3.122253656387329, + "learning_rate": 2.116463985032741e-06, + "loss": 1.4695, + "step": 1486 + }, + { + "epoch": 1.8775252525252526, + "grad_norm": 3.0128729343414307, + "learning_rate": 2.1141253507951357e-06, + "loss": 1.4673, + "step": 1487 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.921835422515869, + "learning_rate": 2.1117867165575306e-06, + "loss": 1.4325, + "step": 1488 + }, + { + "epoch": 1.8800505050505052, + "grad_norm": 3.19289231300354, + "learning_rate": 2.1094480823199254e-06, + "loss": 1.4504, + "step": 1489 + }, + { + "epoch": 1.8813131313131313, + "grad_norm": 2.8579299449920654, + "learning_rate": 2.1071094480823203e-06, + "loss": 1.4042, + "step": 1490 + }, + { + "epoch": 1.8813131313131313, + "eval_loss": 1.6603271961212158, + "eval_runtime": 37.2997, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 1490 + }, + { + "epoch": 1.8825757575757576, + "grad_norm": 2.9113969802856445, + "learning_rate": 2.1047708138447147e-06, + "loss": 1.4128, + "step": 1491 + }, + { + "epoch": 1.8838383838383839, + "grad_norm": 3.058133363723755, + "learning_rate": 2.1024321796071095e-06, + "loss": 1.4574, + "step": 1492 + }, + { + "epoch": 1.88510101010101, + "grad_norm": 3.0066583156585693, + "learning_rate": 2.1000935453695044e-06, + "loss": 1.4521, + "step": 1493 + }, + { + "epoch": 1.8863636363636362, + "grad_norm": 2.9154908657073975, + "learning_rate": 2.097754911131899e-06, + "loss": 1.4103, + "step": 1494 + }, + { + "epoch": 1.8876262626262625, + "grad_norm": 2.977062702178955, + "learning_rate": 2.095416276894294e-06, + "loss": 1.4003, + "step": 1495 + }, + { + "epoch": 1.8888888888888888, + "grad_norm": 2.9048168659210205, + "learning_rate": 2.093077642656689e-06, + "loss": 1.4216, + "step": 1496 + }, + { + "epoch": 1.8901515151515151, + "grad_norm": 3.008733034133911, + "learning_rate": 2.0907390084190833e-06, + "loss": 1.4516, + "step": 1497 + }, + { + "epoch": 1.8914141414141414, + "grad_norm": 3.0832529067993164, + "learning_rate": 2.088400374181478e-06, + "loss": 1.4248, + "step": 1498 + }, + { + "epoch": 1.8926767676767677, + "grad_norm": 2.848066568374634, + "learning_rate": 2.086061739943873e-06, + "loss": 1.4042, + "step": 1499 + }, + { + "epoch": 1.893939393939394, + "grad_norm": 2.846782684326172, + "learning_rate": 2.0837231057062674e-06, + "loss": 1.409, + "step": 1500 + }, + { + "epoch": 1.893939393939394, + "eval_loss": 1.6672260761260986, + "eval_runtime": 37.4695, + "eval_samples_per_second": 26.688, + "eval_steps_per_second": 1.121, + "step": 1500 + }, + { + "epoch": 1.8952020202020203, + "grad_norm": 3.026818037033081, + "learning_rate": 2.0813844714686623e-06, + "loss": 1.4842, + "step": 1501 + }, + { + "epoch": 1.8964646464646466, + "grad_norm": 2.977665662765503, + "learning_rate": 2.0790458372310575e-06, + "loss": 1.4635, + "step": 1502 + }, + { + "epoch": 1.8977272727272727, + "grad_norm": 2.9093048572540283, + "learning_rate": 2.076707202993452e-06, + "loss": 1.4303, + "step": 1503 + }, + { + "epoch": 1.898989898989899, + "grad_norm": 3.0777814388275146, + "learning_rate": 2.0743685687558468e-06, + "loss": 1.4581, + "step": 1504 + }, + { + "epoch": 1.9002525252525253, + "grad_norm": 3.0283970832824707, + "learning_rate": 2.0720299345182416e-06, + "loss": 1.4665, + "step": 1505 + }, + { + "epoch": 1.9015151515151514, + "grad_norm": 2.852412223815918, + "learning_rate": 2.069691300280636e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.9027777777777777, + "grad_norm": 3.1163125038146973, + "learning_rate": 2.067352666043031e-06, + "loss": 1.4907, + "step": 1507 + }, + { + "epoch": 1.904040404040404, + "grad_norm": 2.9857897758483887, + "learning_rate": 2.0650140318054257e-06, + "loss": 1.4262, + "step": 1508 + }, + { + "epoch": 1.9053030303030303, + "grad_norm": 2.957388401031494, + "learning_rate": 2.0626753975678206e-06, + "loss": 1.4525, + "step": 1509 + }, + { + "epoch": 1.9065656565656566, + "grad_norm": 3.0107505321502686, + "learning_rate": 2.0603367633302154e-06, + "loss": 1.4461, + "step": 1510 + }, + { + "epoch": 1.9065656565656566, + "eval_loss": 1.6621215343475342, + "eval_runtime": 36.9336, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1510 + }, + { + "epoch": 1.9078282828282829, + "grad_norm": 3.112478494644165, + "learning_rate": 2.0579981290926103e-06, + "loss": 1.479, + "step": 1511 + }, + { + "epoch": 1.9090909090909092, + "grad_norm": 3.083494186401367, + "learning_rate": 2.0556594948550047e-06, + "loss": 1.4777, + "step": 1512 + }, + { + "epoch": 1.9103535353535355, + "grad_norm": 2.932652711868286, + "learning_rate": 2.0533208606173995e-06, + "loss": 1.3937, + "step": 1513 + }, + { + "epoch": 1.9116161616161618, + "grad_norm": 2.8349449634552, + "learning_rate": 2.0509822263797944e-06, + "loss": 1.4096, + "step": 1514 + }, + { + "epoch": 1.9128787878787878, + "grad_norm": 2.9760472774505615, + "learning_rate": 2.048643592142189e-06, + "loss": 1.4312, + "step": 1515 + }, + { + "epoch": 1.9141414141414141, + "grad_norm": 3.1012730598449707, + "learning_rate": 2.0463049579045836e-06, + "loss": 1.5189, + "step": 1516 + }, + { + "epoch": 1.9154040404040404, + "grad_norm": 2.9072659015655518, + "learning_rate": 2.043966323666979e-06, + "loss": 1.4051, + "step": 1517 + }, + { + "epoch": 1.9166666666666665, + "grad_norm": 2.840203285217285, + "learning_rate": 2.0416276894293733e-06, + "loss": 1.3899, + "step": 1518 + }, + { + "epoch": 1.9179292929292928, + "grad_norm": 3.130812406539917, + "learning_rate": 2.039289055191768e-06, + "loss": 1.4834, + "step": 1519 + }, + { + "epoch": 1.9191919191919191, + "grad_norm": 3.1507933139801025, + "learning_rate": 2.036950420954163e-06, + "loss": 1.402, + "step": 1520 + }, + { + "epoch": 1.9191919191919191, + "eval_loss": 1.6648352146148682, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1520 + }, + { + "epoch": 1.9204545454545454, + "grad_norm": 3.099269151687622, + "learning_rate": 2.034611786716558e-06, + "loss": 1.4189, + "step": 1521 + }, + { + "epoch": 1.9217171717171717, + "grad_norm": 3.0046563148498535, + "learning_rate": 2.0322731524789523e-06, + "loss": 1.405, + "step": 1522 + }, + { + "epoch": 1.922979797979798, + "grad_norm": 3.229635715484619, + "learning_rate": 2.029934518241347e-06, + "loss": 1.5031, + "step": 1523 + }, + { + "epoch": 1.9242424242424243, + "grad_norm": 3.0336225032806396, + "learning_rate": 2.027595884003742e-06, + "loss": 1.4685, + "step": 1524 + }, + { + "epoch": 1.9255050505050506, + "grad_norm": 2.9998443126678467, + "learning_rate": 2.025257249766137e-06, + "loss": 1.4558, + "step": 1525 + }, + { + "epoch": 1.926767676767677, + "grad_norm": 3.1587722301483154, + "learning_rate": 2.0229186155285316e-06, + "loss": 1.4916, + "step": 1526 + }, + { + "epoch": 1.928030303030303, + "grad_norm": 3.2228925228118896, + "learning_rate": 2.0205799812909265e-06, + "loss": 1.4373, + "step": 1527 + }, + { + "epoch": 1.9292929292929293, + "grad_norm": 3.06015944480896, + "learning_rate": 2.018241347053321e-06, + "loss": 1.4147, + "step": 1528 + }, + { + "epoch": 1.9305555555555556, + "grad_norm": 3.02284574508667, + "learning_rate": 2.0159027128157157e-06, + "loss": 1.4574, + "step": 1529 + }, + { + "epoch": 1.9318181818181817, + "grad_norm": 2.9933414459228516, + "learning_rate": 2.0135640785781106e-06, + "loss": 1.4283, + "step": 1530 + }, + { + "epoch": 1.9318181818181817, + "eval_loss": 1.6711069345474243, + "eval_runtime": 37.1182, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1530 + }, + { + "epoch": 1.933080808080808, + "grad_norm": 3.325230121612549, + "learning_rate": 2.011225444340505e-06, + "loss": 1.4532, + "step": 1531 + }, + { + "epoch": 1.9343434343434343, + "grad_norm": 3.0051827430725098, + "learning_rate": 2.0088868101029003e-06, + "loss": 1.3965, + "step": 1532 + }, + { + "epoch": 1.9356060606060606, + "grad_norm": 3.1392757892608643, + "learning_rate": 2.006548175865295e-06, + "loss": 1.4678, + "step": 1533 + }, + { + "epoch": 1.9368686868686869, + "grad_norm": 3.0719001293182373, + "learning_rate": 2.0042095416276895e-06, + "loss": 1.4374, + "step": 1534 + }, + { + "epoch": 1.9381313131313131, + "grad_norm": 3.0406386852264404, + "learning_rate": 2.0018709073900844e-06, + "loss": 1.4199, + "step": 1535 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9481236934661865, + "learning_rate": 1.9995322731524792e-06, + "loss": 1.4599, + "step": 1536 + }, + { + "epoch": 1.9406565656565657, + "grad_norm": 2.901155948638916, + "learning_rate": 1.9971936389148736e-06, + "loss": 1.381, + "step": 1537 + }, + { + "epoch": 1.941919191919192, + "grad_norm": 2.9623661041259766, + "learning_rate": 1.9948550046772685e-06, + "loss": 1.4769, + "step": 1538 + }, + { + "epoch": 1.9431818181818183, + "grad_norm": 2.8457608222961426, + "learning_rate": 1.9925163704396633e-06, + "loss": 1.4117, + "step": 1539 + }, + { + "epoch": 1.9444444444444444, + "grad_norm": 2.9782702922821045, + "learning_rate": 1.990177736202058e-06, + "loss": 1.4824, + "step": 1540 + }, + { + "epoch": 1.9444444444444444, + "eval_loss": 1.6663525104522705, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1540 + }, + { + "epoch": 1.9457070707070707, + "grad_norm": 2.8265817165374756, + "learning_rate": 1.987839101964453e-06, + "loss": 1.4375, + "step": 1541 + }, + { + "epoch": 1.946969696969697, + "grad_norm": 2.9103686809539795, + "learning_rate": 1.985500467726848e-06, + "loss": 1.474, + "step": 1542 + }, + { + "epoch": 1.948232323232323, + "grad_norm": 2.837301254272461, + "learning_rate": 1.9831618334892423e-06, + "loss": 1.4171, + "step": 1543 + }, + { + "epoch": 1.9494949494949494, + "grad_norm": 3.0468413829803467, + "learning_rate": 1.980823199251637e-06, + "loss": 1.5095, + "step": 1544 + }, + { + "epoch": 1.9507575757575757, + "grad_norm": 2.968480348587036, + "learning_rate": 1.978484565014032e-06, + "loss": 1.4203, + "step": 1545 + }, + { + "epoch": 1.952020202020202, + "grad_norm": 2.9865822792053223, + "learning_rate": 1.976145930776427e-06, + "loss": 1.3916, + "step": 1546 + }, + { + "epoch": 1.9532828282828283, + "grad_norm": 2.9733808040618896, + "learning_rate": 1.9738072965388212e-06, + "loss": 1.4216, + "step": 1547 + }, + { + "epoch": 1.9545454545454546, + "grad_norm": 2.941718101501465, + "learning_rate": 1.9714686623012165e-06, + "loss": 1.426, + "step": 1548 + }, + { + "epoch": 1.9558080808080809, + "grad_norm": 3.0980522632598877, + "learning_rate": 1.969130028063611e-06, + "loss": 1.471, + "step": 1549 + }, + { + "epoch": 1.9570707070707072, + "grad_norm": 2.797102689743042, + "learning_rate": 1.9667913938260057e-06, + "loss": 1.3897, + "step": 1550 + }, + { + "epoch": 1.9570707070707072, + "eval_loss": 1.661893367767334, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 1550 + }, + { + "epoch": 1.9583333333333335, + "grad_norm": 2.9562623500823975, + "learning_rate": 1.9644527595884006e-06, + "loss": 1.4481, + "step": 1551 + }, + { + "epoch": 1.9595959595959596, + "grad_norm": 2.9586915969848633, + "learning_rate": 1.9621141253507954e-06, + "loss": 1.4068, + "step": 1552 + }, + { + "epoch": 1.9608585858585859, + "grad_norm": 3.061110496520996, + "learning_rate": 1.95977549111319e-06, + "loss": 1.4524, + "step": 1553 + }, + { + "epoch": 1.9621212121212122, + "grad_norm": 3.080491542816162, + "learning_rate": 1.9574368568755847e-06, + "loss": 1.4457, + "step": 1554 + }, + { + "epoch": 1.9633838383838382, + "grad_norm": 3.083254814147949, + "learning_rate": 1.9550982226379795e-06, + "loss": 1.4775, + "step": 1555 + }, + { + "epoch": 1.9646464646464645, + "grad_norm": 3.008596420288086, + "learning_rate": 1.9527595884003744e-06, + "loss": 1.425, + "step": 1556 + }, + { + "epoch": 1.9659090909090908, + "grad_norm": 2.999195098876953, + "learning_rate": 1.9504209541627692e-06, + "loss": 1.4114, + "step": 1557 + }, + { + "epoch": 1.9671717171717171, + "grad_norm": 3.055265188217163, + "learning_rate": 1.948082319925164e-06, + "loss": 1.4767, + "step": 1558 + }, + { + "epoch": 1.9684343434343434, + "grad_norm": 3.0922064781188965, + "learning_rate": 1.9457436856875585e-06, + "loss": 1.4548, + "step": 1559 + }, + { + "epoch": 1.9696969696969697, + "grad_norm": 2.8740789890289307, + "learning_rate": 1.9434050514499533e-06, + "loss": 1.4582, + "step": 1560 + }, + { + "epoch": 1.9696969696969697, + "eval_loss": 1.6608474254608154, + "eval_runtime": 37.1201, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1560 + }, + { + "epoch": 1.970959595959596, + "grad_norm": 2.983570098876953, + "learning_rate": 1.941066417212348e-06, + "loss": 1.4602, + "step": 1561 + }, + { + "epoch": 1.9722222222222223, + "grad_norm": 3.132474184036255, + "learning_rate": 1.9387277829747426e-06, + "loss": 1.4689, + "step": 1562 + }, + { + "epoch": 1.9734848484848486, + "grad_norm": 2.9685440063476562, + "learning_rate": 1.936389148737138e-06, + "loss": 1.39, + "step": 1563 + }, + { + "epoch": 1.9747474747474747, + "grad_norm": 3.123037815093994, + "learning_rate": 1.9340505144995327e-06, + "loss": 1.4151, + "step": 1564 + }, + { + "epoch": 1.976010101010101, + "grad_norm": 3.135232925415039, + "learning_rate": 1.931711880261927e-06, + "loss": 1.4166, + "step": 1565 + }, + { + "epoch": 1.9772727272727273, + "grad_norm": 2.9131808280944824, + "learning_rate": 1.929373246024322e-06, + "loss": 1.3952, + "step": 1566 + }, + { + "epoch": 1.9785353535353534, + "grad_norm": 3.1187453269958496, + "learning_rate": 1.927034611786717e-06, + "loss": 1.4132, + "step": 1567 + }, + { + "epoch": 1.9797979797979797, + "grad_norm": 2.983447551727295, + "learning_rate": 1.9246959775491112e-06, + "loss": 1.4517, + "step": 1568 + }, + { + "epoch": 1.981060606060606, + "grad_norm": 2.9731497764587402, + "learning_rate": 1.922357343311506e-06, + "loss": 1.432, + "step": 1569 + }, + { + "epoch": 1.9823232323232323, + "grad_norm": 2.8526554107666016, + "learning_rate": 1.920018709073901e-06, + "loss": 1.4141, + "step": 1570 + }, + { + "epoch": 1.9823232323232323, + "eval_loss": 1.6579400300979614, + "eval_runtime": 37.1216, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1570 + }, + { + "epoch": 1.9835858585858586, + "grad_norm": 2.900019407272339, + "learning_rate": 1.9176800748362958e-06, + "loss": 1.4008, + "step": 1571 + }, + { + "epoch": 1.9848484848484849, + "grad_norm": 2.96079683303833, + "learning_rate": 1.9153414405986906e-06, + "loss": 1.4586, + "step": 1572 + }, + { + "epoch": 1.9861111111111112, + "grad_norm": 3.1275715827941895, + "learning_rate": 1.9130028063610854e-06, + "loss": 1.4768, + "step": 1573 + }, + { + "epoch": 1.9873737373737375, + "grad_norm": 3.044041395187378, + "learning_rate": 1.91066417212348e-06, + "loss": 1.4428, + "step": 1574 + }, + { + "epoch": 1.9886363636363638, + "grad_norm": 3.0206832885742188, + "learning_rate": 1.9083255378858747e-06, + "loss": 1.4492, + "step": 1575 + }, + { + "epoch": 1.98989898989899, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.398, + "step": 1576 + }, + { + "epoch": 1.9911616161616161, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.4373, + "step": 1577 + }, + { + "epoch": 1.9924242424242424, + "grad_norm": 3.129822254180908, + "learning_rate": 1.9036482694106642e-06, + "loss": 1.4599, + "step": 1578 + }, + { + "epoch": 1.9936868686868687, + "grad_norm": 2.978015661239624, + "learning_rate": 1.9013096351730592e-06, + "loss": 1.409, + "step": 1579 + }, + { + "epoch": 1.9949494949494948, + "grad_norm": 2.980295181274414, + "learning_rate": 1.8989710009354539e-06, + "loss": 1.4251, + "step": 1580 + }, + { + "epoch": 1.9949494949494948, + "eval_loss": 1.661036491394043, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.996212121212121, + "grad_norm": 2.878361225128174, + "learning_rate": 1.8966323666978487e-06, + "loss": 1.4289, + "step": 1581 + }, + { + "epoch": 1.9974747474747474, + "grad_norm": 2.967414379119873, + "learning_rate": 1.8942937324602433e-06, + "loss": 1.5151, + "step": 1582 + }, + { + "epoch": 1.9987373737373737, + "grad_norm": 2.9448516368865967, + "learning_rate": 1.8919550982226382e-06, + "loss": 1.4327, + "step": 1583 + }, + { + "epoch": 2.0, + "grad_norm": 2.9000067710876465, + "learning_rate": 1.8896164639850328e-06, + "loss": 1.4113, + "step": 1584 + }, + { + "epoch": 2.0012626262626263, + "grad_norm": 3.1630377769470215, + "learning_rate": 1.8872778297474277e-06, + "loss": 1.4179, + "step": 1585 + }, + { + "epoch": 2.0025252525252526, + "grad_norm": 3.2642927169799805, + "learning_rate": 1.8849391955098223e-06, + "loss": 1.3873, + "step": 1586 + }, + { + "epoch": 2.003787878787879, + "grad_norm": 3.1669039726257324, + "learning_rate": 1.8826005612722173e-06, + "loss": 1.3834, + "step": 1587 + }, + { + "epoch": 2.005050505050505, + "grad_norm": 3.145040988922119, + "learning_rate": 1.880261927034612e-06, + "loss": 1.3334, + "step": 1588 + }, + { + "epoch": 2.0063131313131315, + "grad_norm": 3.707200050354004, + "learning_rate": 1.8779232927970068e-06, + "loss": 1.3979, + "step": 1589 + }, + { + "epoch": 2.007575757575758, + "grad_norm": 3.936523199081421, + "learning_rate": 1.8755846585594014e-06, + "loss": 1.4221, + "step": 1590 + }, + { + "epoch": 2.007575757575758, + "eval_loss": 1.6722478866577148, + "eval_runtime": 37.2129, + "eval_samples_per_second": 26.872, + "eval_steps_per_second": 1.129, + "step": 1590 + }, + { + "epoch": 2.0088383838383836, + "grad_norm": 3.4221086502075195, + "learning_rate": 1.8732460243217963e-06, + "loss": 1.3894, + "step": 1591 + }, + { + "epoch": 2.01010101010101, + "grad_norm": 3.3115553855895996, + "learning_rate": 1.870907390084191e-06, + "loss": 1.4016, + "step": 1592 + }, + { + "epoch": 2.0113636363636362, + "grad_norm": 3.055779457092285, + "learning_rate": 1.8685687558465855e-06, + "loss": 1.3676, + "step": 1593 + }, + { + "epoch": 2.0126262626262625, + "grad_norm": 3.4615232944488525, + "learning_rate": 1.8662301216089804e-06, + "loss": 1.3624, + "step": 1594 + }, + { + "epoch": 2.013888888888889, + "grad_norm": 3.3298869132995605, + "learning_rate": 1.8638914873713754e-06, + "loss": 1.3939, + "step": 1595 + }, + { + "epoch": 2.015151515151515, + "grad_norm": 3.271921396255493, + "learning_rate": 1.86155285313377e-06, + "loss": 1.3994, + "step": 1596 + }, + { + "epoch": 2.0164141414141414, + "grad_norm": 3.272677421569824, + "learning_rate": 1.859214218896165e-06, + "loss": 1.3934, + "step": 1597 + }, + { + "epoch": 2.0176767676767677, + "grad_norm": 3.3485944271087646, + "learning_rate": 1.8568755846585596e-06, + "loss": 1.3626, + "step": 1598 + }, + { + "epoch": 2.018939393939394, + "grad_norm": 3.426981210708618, + "learning_rate": 1.8545369504209542e-06, + "loss": 1.4231, + "step": 1599 + }, + { + "epoch": 2.0202020202020203, + "grad_norm": 3.3859283924102783, + "learning_rate": 1.852198316183349e-06, + "loss": 1.3587, + "step": 1600 + }, + { + "epoch": 2.0202020202020203, + "eval_loss": 1.6732399463653564, + "eval_runtime": 37.3484, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 1600 + }, + { + "epoch": 2.0214646464646466, + "grad_norm": 3.537806987762451, + "learning_rate": 1.8498596819457437e-06, + "loss": 1.3729, + "step": 1601 + }, + { + "epoch": 2.022727272727273, + "grad_norm": 3.5417566299438477, + "learning_rate": 1.8475210477081387e-06, + "loss": 1.4272, + "step": 1602 + }, + { + "epoch": 2.023989898989899, + "grad_norm": 3.287195920944214, + "learning_rate": 1.8451824134705336e-06, + "loss": 1.3647, + "step": 1603 + }, + { + "epoch": 2.025252525252525, + "grad_norm": 3.233365297317505, + "learning_rate": 1.8428437792329282e-06, + "loss": 1.4062, + "step": 1604 + }, + { + "epoch": 2.0265151515151514, + "grad_norm": 3.296578884124756, + "learning_rate": 1.8405051449953228e-06, + "loss": 1.3841, + "step": 1605 + }, + { + "epoch": 2.0277777777777777, + "grad_norm": 3.2697315216064453, + "learning_rate": 1.8381665107577177e-06, + "loss": 1.3957, + "step": 1606 + }, + { + "epoch": 2.029040404040404, + "grad_norm": 3.1117050647735596, + "learning_rate": 1.8358278765201123e-06, + "loss": 1.3643, + "step": 1607 + }, + { + "epoch": 2.0303030303030303, + "grad_norm": 3.14170241355896, + "learning_rate": 1.8334892422825071e-06, + "loss": 1.3483, + "step": 1608 + }, + { + "epoch": 2.0315656565656566, + "grad_norm": 3.2749483585357666, + "learning_rate": 1.8311506080449018e-06, + "loss": 1.4007, + "step": 1609 + }, + { + "epoch": 2.032828282828283, + "grad_norm": 3.266631603240967, + "learning_rate": 1.8288119738072968e-06, + "loss": 1.3943, + "step": 1610 + }, + { + "epoch": 2.032828282828283, + "eval_loss": 1.6679413318634033, + "eval_runtime": 36.8768, + "eval_samples_per_second": 27.117, + "eval_steps_per_second": 1.139, + "step": 1610 + }, + { + "epoch": 2.034090909090909, + "grad_norm": 3.535625696182251, + "learning_rate": 1.8264733395696914e-06, + "loss": 1.4027, + "step": 1611 + }, + { + "epoch": 2.0353535353535355, + "grad_norm": 3.2741401195526123, + "learning_rate": 1.8241347053320863e-06, + "loss": 1.4075, + "step": 1612 + }, + { + "epoch": 2.0366161616161618, + "grad_norm": 3.3183717727661133, + "learning_rate": 1.821796071094481e-06, + "loss": 1.4223, + "step": 1613 + }, + { + "epoch": 2.037878787878788, + "grad_norm": 3.323057174682617, + "learning_rate": 1.8194574368568758e-06, + "loss": 1.3595, + "step": 1614 + }, + { + "epoch": 2.0391414141414144, + "grad_norm": 3.202054738998413, + "learning_rate": 1.8171188026192704e-06, + "loss": 1.3399, + "step": 1615 + }, + { + "epoch": 2.04040404040404, + "grad_norm": 3.3273448944091797, + "learning_rate": 1.8147801683816652e-06, + "loss": 1.3848, + "step": 1616 + }, + { + "epoch": 2.0416666666666665, + "grad_norm": 3.396843671798706, + "learning_rate": 1.81244153414406e-06, + "loss": 1.3968, + "step": 1617 + }, + { + "epoch": 2.042929292929293, + "grad_norm": 3.2145836353302, + "learning_rate": 1.810102899906455e-06, + "loss": 1.3571, + "step": 1618 + }, + { + "epoch": 2.044191919191919, + "grad_norm": 3.156754732131958, + "learning_rate": 1.8077642656688496e-06, + "loss": 1.3439, + "step": 1619 + }, + { + "epoch": 2.0454545454545454, + "grad_norm": 3.4259471893310547, + "learning_rate": 1.8054256314312444e-06, + "loss": 1.4123, + "step": 1620 + }, + { + "epoch": 2.0454545454545454, + "eval_loss": 1.6703134775161743, + "eval_runtime": 37.002, + "eval_samples_per_second": 27.026, + "eval_steps_per_second": 1.135, + "step": 1620 + }, + { + "epoch": 2.0467171717171717, + "grad_norm": 3.3556408882141113, + "learning_rate": 1.803086997193639e-06, + "loss": 1.4252, + "step": 1621 + }, + { + "epoch": 2.047979797979798, + "grad_norm": 3.6319949626922607, + "learning_rate": 1.8007483629560339e-06, + "loss": 1.4634, + "step": 1622 + }, + { + "epoch": 2.0492424242424243, + "grad_norm": 3.4495089054107666, + "learning_rate": 1.7984097287184285e-06, + "loss": 1.4066, + "step": 1623 + }, + { + "epoch": 2.0505050505050506, + "grad_norm": 3.182297945022583, + "learning_rate": 1.7960710944808231e-06, + "loss": 1.3899, + "step": 1624 + }, + { + "epoch": 2.051767676767677, + "grad_norm": 3.097158193588257, + "learning_rate": 1.7937324602432182e-06, + "loss": 1.293, + "step": 1625 + }, + { + "epoch": 2.053030303030303, + "grad_norm": 3.429065704345703, + "learning_rate": 1.791393826005613e-06, + "loss": 1.3905, + "step": 1626 + }, + { + "epoch": 2.0542929292929295, + "grad_norm": 3.183366298675537, + "learning_rate": 1.7890551917680077e-06, + "loss": 1.3444, + "step": 1627 + }, + { + "epoch": 2.0555555555555554, + "grad_norm": 3.175612449645996, + "learning_rate": 1.7867165575304025e-06, + "loss": 1.3985, + "step": 1628 + }, + { + "epoch": 2.0568181818181817, + "grad_norm": 3.26491641998291, + "learning_rate": 1.7843779232927971e-06, + "loss": 1.3607, + "step": 1629 + }, + { + "epoch": 2.058080808080808, + "grad_norm": 3.2279412746429443, + "learning_rate": 1.7820392890551918e-06, + "loss": 1.3545, + "step": 1630 + }, + { + "epoch": 2.058080808080808, + "eval_loss": 1.6729198694229126, + "eval_runtime": 37.0369, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 1630 + }, + { + "epoch": 2.0593434343434343, + "grad_norm": 3.3255503177642822, + "learning_rate": 1.7797006548175866e-06, + "loss": 1.3938, + "step": 1631 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.2912404537200928, + "learning_rate": 1.7773620205799812e-06, + "loss": 1.366, + "step": 1632 + }, + { + "epoch": 2.061868686868687, + "grad_norm": 3.2314395904541016, + "learning_rate": 1.7750233863423763e-06, + "loss": 1.3298, + "step": 1633 + }, + { + "epoch": 2.063131313131313, + "grad_norm": 3.3304178714752197, + "learning_rate": 1.772684752104771e-06, + "loss": 1.3592, + "step": 1634 + }, + { + "epoch": 2.0643939393939394, + "grad_norm": 3.3113889694213867, + "learning_rate": 1.7703461178671658e-06, + "loss": 1.3865, + "step": 1635 + }, + { + "epoch": 2.0656565656565657, + "grad_norm": 3.260434150695801, + "learning_rate": 1.7680074836295604e-06, + "loss": 1.3306, + "step": 1636 + }, + { + "epoch": 2.066919191919192, + "grad_norm": 3.2890963554382324, + "learning_rate": 1.7656688493919552e-06, + "loss": 1.3533, + "step": 1637 + }, + { + "epoch": 2.0681818181818183, + "grad_norm": 3.3027193546295166, + "learning_rate": 1.7633302151543499e-06, + "loss": 1.3746, + "step": 1638 + }, + { + "epoch": 2.0694444444444446, + "grad_norm": 3.3163208961486816, + "learning_rate": 1.7609915809167447e-06, + "loss": 1.4113, + "step": 1639 + }, + { + "epoch": 2.0707070707070705, + "grad_norm": 3.300557851791382, + "learning_rate": 1.7586529466791396e-06, + "loss": 1.3669, + "step": 1640 + }, + { + "epoch": 2.0707070707070705, + "eval_loss": 1.6676570177078247, + "eval_runtime": 37.0589, + "eval_samples_per_second": 26.984, + "eval_steps_per_second": 1.133, + "step": 1640 + }, + { + "epoch": 2.071969696969697, + "grad_norm": 3.305338144302368, + "learning_rate": 1.7563143124415344e-06, + "loss": 1.3969, + "step": 1641 + }, + { + "epoch": 2.073232323232323, + "grad_norm": 3.2047572135925293, + "learning_rate": 1.753975678203929e-06, + "loss": 1.3856, + "step": 1642 + }, + { + "epoch": 2.0744949494949494, + "grad_norm": 3.575038433074951, + "learning_rate": 1.7516370439663239e-06, + "loss": 1.3947, + "step": 1643 + }, + { + "epoch": 2.0757575757575757, + "grad_norm": 3.44581937789917, + "learning_rate": 1.7492984097287185e-06, + "loss": 1.4037, + "step": 1644 + }, + { + "epoch": 2.077020202020202, + "grad_norm": 3.574326753616333, + "learning_rate": 1.7469597754911134e-06, + "loss": 1.4216, + "step": 1645 + }, + { + "epoch": 2.0782828282828283, + "grad_norm": 3.2762558460235596, + "learning_rate": 1.744621141253508e-06, + "loss": 1.3621, + "step": 1646 + }, + { + "epoch": 2.0795454545454546, + "grad_norm": 3.3749659061431885, + "learning_rate": 1.7422825070159028e-06, + "loss": 1.3762, + "step": 1647 + }, + { + "epoch": 2.080808080808081, + "grad_norm": 3.287447214126587, + "learning_rate": 1.7399438727782977e-06, + "loss": 1.3565, + "step": 1648 + }, + { + "epoch": 2.082070707070707, + "grad_norm": 3.5107264518737793, + "learning_rate": 1.7376052385406925e-06, + "loss": 1.392, + "step": 1649 + }, + { + "epoch": 2.0833333333333335, + "grad_norm": 3.319650173187256, + "learning_rate": 1.7352666043030871e-06, + "loss": 1.3835, + "step": 1650 + }, + { + "epoch": 2.0833333333333335, + "eval_loss": 1.6727185249328613, + "eval_runtime": 37.0658, + "eval_samples_per_second": 26.979, + "eval_steps_per_second": 1.133, + "step": 1650 + }, + { + "epoch": 2.08459595959596, + "grad_norm": 3.4447925090789795, + "learning_rate": 1.732927970065482e-06, + "loss": 1.4186, + "step": 1651 + }, + { + "epoch": 2.0858585858585856, + "grad_norm": 3.5944249629974365, + "learning_rate": 1.7305893358278766e-06, + "loss": 1.4461, + "step": 1652 + }, + { + "epoch": 2.087121212121212, + "grad_norm": 3.5437049865722656, + "learning_rate": 1.7282507015902715e-06, + "loss": 1.3343, + "step": 1653 + }, + { + "epoch": 2.0883838383838382, + "grad_norm": 3.361407518386841, + "learning_rate": 1.725912067352666e-06, + "loss": 1.3653, + "step": 1654 + }, + { + "epoch": 2.0896464646464645, + "grad_norm": 3.4728963375091553, + "learning_rate": 1.7235734331150607e-06, + "loss": 1.4297, + "step": 1655 + }, + { + "epoch": 2.090909090909091, + "grad_norm": 3.3794610500335693, + "learning_rate": 1.7212347988774558e-06, + "loss": 1.3941, + "step": 1656 + }, + { + "epoch": 2.092171717171717, + "grad_norm": 3.280857801437378, + "learning_rate": 1.7188961646398506e-06, + "loss": 1.3445, + "step": 1657 + }, + { + "epoch": 2.0934343434343434, + "grad_norm": 3.29036283493042, + "learning_rate": 1.7165575304022453e-06, + "loss": 1.3581, + "step": 1658 + }, + { + "epoch": 2.0946969696969697, + "grad_norm": 3.1731486320495605, + "learning_rate": 1.7142188961646399e-06, + "loss": 1.3873, + "step": 1659 + }, + { + "epoch": 2.095959595959596, + "grad_norm": 3.2283008098602295, + "learning_rate": 1.7118802619270347e-06, + "loss": 1.3462, + "step": 1660 + }, + { + "epoch": 2.095959595959596, + "eval_loss": 1.6726486682891846, + "eval_runtime": 37.0617, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1660 + }, + { + "epoch": 2.0972222222222223, + "grad_norm": 3.329171657562256, + "learning_rate": 1.7095416276894294e-06, + "loss": 1.3974, + "step": 1661 + }, + { + "epoch": 2.0984848484848486, + "grad_norm": 3.4113657474517822, + "learning_rate": 1.7072029934518242e-06, + "loss": 1.4237, + "step": 1662 + }, + { + "epoch": 2.099747474747475, + "grad_norm": 3.431184768676758, + "learning_rate": 1.7048643592142193e-06, + "loss": 1.38, + "step": 1663 + }, + { + "epoch": 2.101010101010101, + "grad_norm": 3.5471761226654053, + "learning_rate": 1.7025257249766139e-06, + "loss": 1.3762, + "step": 1664 + }, + { + "epoch": 2.102272727272727, + "grad_norm": 3.43166446685791, + "learning_rate": 1.7001870907390085e-06, + "loss": 1.4371, + "step": 1665 + }, + { + "epoch": 2.1035353535353534, + "grad_norm": 3.28086256980896, + "learning_rate": 1.6978484565014034e-06, + "loss": 1.4062, + "step": 1666 + }, + { + "epoch": 2.1047979797979797, + "grad_norm": 3.166083335876465, + "learning_rate": 1.695509822263798e-06, + "loss": 1.3535, + "step": 1667 + }, + { + "epoch": 2.106060606060606, + "grad_norm": 3.4707589149475098, + "learning_rate": 1.6931711880261928e-06, + "loss": 1.4237, + "step": 1668 + }, + { + "epoch": 2.1073232323232323, + "grad_norm": 3.5033648014068604, + "learning_rate": 1.6908325537885875e-06, + "loss": 1.4331, + "step": 1669 + }, + { + "epoch": 2.1085858585858586, + "grad_norm": 3.3915514945983887, + "learning_rate": 1.6884939195509823e-06, + "loss": 1.3783, + "step": 1670 + }, + { + "epoch": 2.1085858585858586, + "eval_loss": 1.6657648086547852, + "eval_runtime": 37.0742, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1670 + }, + { + "epoch": 2.109848484848485, + "grad_norm": 3.3420865535736084, + "learning_rate": 1.6861552853133772e-06, + "loss": 1.4016, + "step": 1671 + }, + { + "epoch": 2.111111111111111, + "grad_norm": 3.382840394973755, + "learning_rate": 1.683816651075772e-06, + "loss": 1.4209, + "step": 1672 + }, + { + "epoch": 2.1123737373737375, + "grad_norm": 3.3109424114227295, + "learning_rate": 1.6814780168381666e-06, + "loss": 1.3716, + "step": 1673 + }, + { + "epoch": 2.1136363636363638, + "grad_norm": 3.3781373500823975, + "learning_rate": 1.6791393826005615e-06, + "loss": 1.4106, + "step": 1674 + }, + { + "epoch": 2.11489898989899, + "grad_norm": 3.29980731010437, + "learning_rate": 1.676800748362956e-06, + "loss": 1.3628, + "step": 1675 + }, + { + "epoch": 2.1161616161616164, + "grad_norm": 3.38468599319458, + "learning_rate": 1.674462114125351e-06, + "loss": 1.3478, + "step": 1676 + }, + { + "epoch": 2.117424242424242, + "grad_norm": 3.4106788635253906, + "learning_rate": 1.6721234798877456e-06, + "loss": 1.391, + "step": 1677 + }, + { + "epoch": 2.1186868686868685, + "grad_norm": 3.412184476852417, + "learning_rate": 1.6697848456501404e-06, + "loss": 1.4245, + "step": 1678 + }, + { + "epoch": 2.119949494949495, + "grad_norm": 3.3659188747406006, + "learning_rate": 1.6674462114125353e-06, + "loss": 1.3358, + "step": 1679 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.241384744644165, + "learning_rate": 1.66510757717493e-06, + "loss": 1.3855, + "step": 1680 + }, + { + "epoch": 2.121212121212121, + "eval_loss": 1.6706457138061523, + "eval_runtime": 37.0651, + "eval_samples_per_second": 26.98, + "eval_steps_per_second": 1.133, + "step": 1680 + }, + { + "epoch": 2.1224747474747474, + "grad_norm": 3.3625874519348145, + "learning_rate": 1.6627689429373247e-06, + "loss": 1.3955, + "step": 1681 + }, + { + "epoch": 2.1237373737373737, + "grad_norm": 3.378143787384033, + "learning_rate": 1.6604303086997196e-06, + "loss": 1.3635, + "step": 1682 + }, + { + "epoch": 2.125, + "grad_norm": 3.4872324466705322, + "learning_rate": 1.6580916744621142e-06, + "loss": 1.4037, + "step": 1683 + }, + { + "epoch": 2.1262626262626263, + "grad_norm": 3.4733831882476807, + "learning_rate": 1.655753040224509e-06, + "loss": 1.3846, + "step": 1684 + }, + { + "epoch": 2.1275252525252526, + "grad_norm": 3.4518625736236572, + "learning_rate": 1.6534144059869037e-06, + "loss": 1.3984, + "step": 1685 + }, + { + "epoch": 2.128787878787879, + "grad_norm": 3.3702666759490967, + "learning_rate": 1.6510757717492987e-06, + "loss": 1.3712, + "step": 1686 + }, + { + "epoch": 2.130050505050505, + "grad_norm": 3.357708215713501, + "learning_rate": 1.6487371375116934e-06, + "loss": 1.4204, + "step": 1687 + }, + { + "epoch": 2.1313131313131315, + "grad_norm": 3.3902523517608643, + "learning_rate": 1.6463985032740882e-06, + "loss": 1.439, + "step": 1688 + }, + { + "epoch": 2.132575757575758, + "grad_norm": 3.246450424194336, + "learning_rate": 1.6440598690364828e-06, + "loss": 1.4061, + "step": 1689 + }, + { + "epoch": 2.1338383838383836, + "grad_norm": 3.5247316360473633, + "learning_rate": 1.6417212347988775e-06, + "loss": 1.4148, + "step": 1690 + }, + { + "epoch": 2.1338383838383836, + "eval_loss": 1.6684828996658325, + "eval_runtime": 37.3778, + "eval_samples_per_second": 26.754, + "eval_steps_per_second": 1.124, + "step": 1690 + }, + { + "epoch": 2.13510101010101, + "grad_norm": 3.3932440280914307, + "learning_rate": 1.6393826005612723e-06, + "loss": 1.3756, + "step": 1691 + }, + { + "epoch": 2.1363636363636362, + "grad_norm": 3.5242092609405518, + "learning_rate": 1.637043966323667e-06, + "loss": 1.4009, + "step": 1692 + }, + { + "epoch": 2.1376262626262625, + "grad_norm": 3.464235544204712, + "learning_rate": 1.6347053320860618e-06, + "loss": 1.4129, + "step": 1693 + }, + { + "epoch": 2.138888888888889, + "grad_norm": 3.4409964084625244, + "learning_rate": 1.6323666978484568e-06, + "loss": 1.3969, + "step": 1694 + }, + { + "epoch": 2.140151515151515, + "grad_norm": 3.329281806945801, + "learning_rate": 1.6300280636108515e-06, + "loss": 1.367, + "step": 1695 + }, + { + "epoch": 2.1414141414141414, + "grad_norm": 3.28047251701355, + "learning_rate": 1.6276894293732461e-06, + "loss": 1.3994, + "step": 1696 + }, + { + "epoch": 2.1426767676767677, + "grad_norm": 3.1937954425811768, + "learning_rate": 1.625350795135641e-06, + "loss": 1.3626, + "step": 1697 + }, + { + "epoch": 2.143939393939394, + "grad_norm": 3.5084168910980225, + "learning_rate": 1.6230121608980356e-06, + "loss": 1.4067, + "step": 1698 + }, + { + "epoch": 2.1452020202020203, + "grad_norm": 3.3637681007385254, + "learning_rate": 1.6206735266604304e-06, + "loss": 1.3619, + "step": 1699 + }, + { + "epoch": 2.1464646464646466, + "grad_norm": 3.2855823040008545, + "learning_rate": 1.618334892422825e-06, + "loss": 1.4227, + "step": 1700 + }, + { + "epoch": 2.1464646464646466, + "eval_loss": 1.6681337356567383, + "eval_runtime": 37.4375, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1700 + }, + { + "epoch": 2.147727272727273, + "grad_norm": 3.4610795974731445, + "learning_rate": 1.6159962581852201e-06, + "loss": 1.3842, + "step": 1701 + }, + { + "epoch": 2.148989898989899, + "grad_norm": 3.420252561569214, + "learning_rate": 1.6136576239476147e-06, + "loss": 1.3951, + "step": 1702 + }, + { + "epoch": 2.150252525252525, + "grad_norm": 3.4725613594055176, + "learning_rate": 1.6113189897100096e-06, + "loss": 1.3697, + "step": 1703 + }, + { + "epoch": 2.1515151515151514, + "grad_norm": 3.3183791637420654, + "learning_rate": 1.6089803554724042e-06, + "loss": 1.3813, + "step": 1704 + }, + { + "epoch": 2.1527777777777777, + "grad_norm": 3.3727011680603027, + "learning_rate": 1.606641721234799e-06, + "loss": 1.3654, + "step": 1705 + }, + { + "epoch": 2.154040404040404, + "grad_norm": 3.420855760574341, + "learning_rate": 1.6043030869971937e-06, + "loss": 1.3399, + "step": 1706 + }, + { + "epoch": 2.1553030303030303, + "grad_norm": 3.3293979167938232, + "learning_rate": 1.6019644527595885e-06, + "loss": 1.3694, + "step": 1707 + }, + { + "epoch": 2.1565656565656566, + "grad_norm": 3.414578437805176, + "learning_rate": 1.5996258185219832e-06, + "loss": 1.4002, + "step": 1708 + }, + { + "epoch": 2.157828282828283, + "grad_norm": 3.3196682929992676, + "learning_rate": 1.5972871842843782e-06, + "loss": 1.379, + "step": 1709 + }, + { + "epoch": 2.159090909090909, + "grad_norm": 3.2492880821228027, + "learning_rate": 1.5949485500467728e-06, + "loss": 1.3694, + "step": 1710 + }, + { + "epoch": 2.159090909090909, + "eval_loss": 1.6655367612838745, + "eval_runtime": 36.9249, + "eval_samples_per_second": 27.082, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 2.1603535353535355, + "grad_norm": 3.3484954833984375, + "learning_rate": 1.5926099158091677e-06, + "loss": 1.374, + "step": 1711 + }, + { + "epoch": 2.1616161616161618, + "grad_norm": 3.2919235229492188, + "learning_rate": 1.5902712815715623e-06, + "loss": 1.3524, + "step": 1712 + }, + { + "epoch": 2.162878787878788, + "grad_norm": 3.395267963409424, + "learning_rate": 1.5879326473339572e-06, + "loss": 1.3848, + "step": 1713 + }, + { + "epoch": 2.1641414141414144, + "grad_norm": 3.5136897563934326, + "learning_rate": 1.5855940130963518e-06, + "loss": 1.3465, + "step": 1714 + }, + { + "epoch": 2.16540404040404, + "grad_norm": 3.5640273094177246, + "learning_rate": 1.5832553788587464e-06, + "loss": 1.3924, + "step": 1715 + }, + { + "epoch": 2.1666666666666665, + "grad_norm": 3.761011838912964, + "learning_rate": 1.5809167446211413e-06, + "loss": 1.4401, + "step": 1716 + }, + { + "epoch": 2.167929292929293, + "grad_norm": 3.259582042694092, + "learning_rate": 1.5785781103835363e-06, + "loss": 1.3808, + "step": 1717 + }, + { + "epoch": 2.169191919191919, + "grad_norm": 3.365035057067871, + "learning_rate": 1.576239476145931e-06, + "loss": 1.3928, + "step": 1718 + }, + { + "epoch": 2.1704545454545454, + "grad_norm": 3.466689109802246, + "learning_rate": 1.5739008419083258e-06, + "loss": 1.3797, + "step": 1719 + }, + { + "epoch": 2.1717171717171717, + "grad_norm": 3.5278398990631104, + "learning_rate": 1.5715622076707204e-06, + "loss": 1.3752, + "step": 1720 + }, + { + "epoch": 2.1717171717171717, + "eval_loss": 1.6711663007736206, + "eval_runtime": 37.0568, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1720 + }, + { + "epoch": 2.172979797979798, + "grad_norm": 3.43217134475708, + "learning_rate": 1.569223573433115e-06, + "loss": 1.4222, + "step": 1721 + }, + { + "epoch": 2.1742424242424243, + "grad_norm": 3.295238733291626, + "learning_rate": 1.56688493919551e-06, + "loss": 1.3537, + "step": 1722 + }, + { + "epoch": 2.1755050505050506, + "grad_norm": 3.4373223781585693, + "learning_rate": 1.5645463049579045e-06, + "loss": 1.3813, + "step": 1723 + }, + { + "epoch": 2.176767676767677, + "grad_norm": 3.4872143268585205, + "learning_rate": 1.5622076707202996e-06, + "loss": 1.4018, + "step": 1724 + }, + { + "epoch": 2.178030303030303, + "grad_norm": 3.3276984691619873, + "learning_rate": 1.5598690364826944e-06, + "loss": 1.3644, + "step": 1725 + }, + { + "epoch": 2.179292929292929, + "grad_norm": 3.441479444503784, + "learning_rate": 1.557530402245089e-06, + "loss": 1.4105, + "step": 1726 + }, + { + "epoch": 2.1805555555555554, + "grad_norm": 3.3938825130462646, + "learning_rate": 1.5551917680074837e-06, + "loss": 1.3478, + "step": 1727 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.428677797317505, + "learning_rate": 1.5528531337698785e-06, + "loss": 1.3977, + "step": 1728 + }, + { + "epoch": 2.183080808080808, + "grad_norm": 3.3802578449249268, + "learning_rate": 1.5505144995322732e-06, + "loss": 1.3441, + "step": 1729 + }, + { + "epoch": 2.1843434343434343, + "grad_norm": 3.419437885284424, + "learning_rate": 1.548175865294668e-06, + "loss": 1.3811, + "step": 1730 + }, + { + "epoch": 2.1843434343434343, + "eval_loss": 1.669846773147583, + "eval_runtime": 37.0868, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1730 + }, + { + "epoch": 2.1856060606060606, + "grad_norm": 3.31225848197937, + "learning_rate": 1.5458372310570626e-06, + "loss": 1.3923, + "step": 1731 + }, + { + "epoch": 2.186868686868687, + "grad_norm": 3.5382823944091797, + "learning_rate": 1.5434985968194577e-06, + "loss": 1.4213, + "step": 1732 + }, + { + "epoch": 2.188131313131313, + "grad_norm": 3.207977533340454, + "learning_rate": 1.5411599625818523e-06, + "loss": 1.3175, + "step": 1733 + }, + { + "epoch": 2.1893939393939394, + "grad_norm": 3.3290693759918213, + "learning_rate": 1.5388213283442472e-06, + "loss": 1.4157, + "step": 1734 + }, + { + "epoch": 2.1906565656565657, + "grad_norm": 3.3432180881500244, + "learning_rate": 1.5364826941066418e-06, + "loss": 1.3844, + "step": 1735 + }, + { + "epoch": 2.191919191919192, + "grad_norm": 3.6149415969848633, + "learning_rate": 1.5341440598690366e-06, + "loss": 1.4243, + "step": 1736 + }, + { + "epoch": 2.1931818181818183, + "grad_norm": 3.3316938877105713, + "learning_rate": 1.5318054256314313e-06, + "loss": 1.372, + "step": 1737 + }, + { + "epoch": 2.1944444444444446, + "grad_norm": 3.4198312759399414, + "learning_rate": 1.5294667913938261e-06, + "loss": 1.3177, + "step": 1738 + }, + { + "epoch": 2.195707070707071, + "grad_norm": 3.450404167175293, + "learning_rate": 1.5271281571562207e-06, + "loss": 1.3596, + "step": 1739 + }, + { + "epoch": 2.196969696969697, + "grad_norm": 3.261406898498535, + "learning_rate": 1.5247895229186158e-06, + "loss": 1.3639, + "step": 1740 + }, + { + "epoch": 2.196969696969697, + "eval_loss": 1.6702104806900024, + "eval_runtime": 37.1009, + "eval_samples_per_second": 26.953, + "eval_steps_per_second": 1.132, + "step": 1740 + }, + { + "epoch": 2.198232323232323, + "grad_norm": 3.5527029037475586, + "learning_rate": 1.5224508886810104e-06, + "loss": 1.4129, + "step": 1741 + }, + { + "epoch": 2.1994949494949494, + "grad_norm": 3.498812437057495, + "learning_rate": 1.5201122544434053e-06, + "loss": 1.396, + "step": 1742 + }, + { + "epoch": 2.2007575757575757, + "grad_norm": 3.509518623352051, + "learning_rate": 1.5177736202058e-06, + "loss": 1.4506, + "step": 1743 + }, + { + "epoch": 2.202020202020202, + "grad_norm": 3.2482805252075195, + "learning_rate": 1.5154349859681948e-06, + "loss": 1.3793, + "step": 1744 + }, + { + "epoch": 2.2032828282828283, + "grad_norm": 3.299140214920044, + "learning_rate": 1.5130963517305894e-06, + "loss": 1.4011, + "step": 1745 + }, + { + "epoch": 2.2045454545454546, + "grad_norm": 3.4633231163024902, + "learning_rate": 1.510757717492984e-06, + "loss": 1.3904, + "step": 1746 + }, + { + "epoch": 2.205808080808081, + "grad_norm": 3.2724618911743164, + "learning_rate": 1.508419083255379e-06, + "loss": 1.3699, + "step": 1747 + }, + { + "epoch": 2.207070707070707, + "grad_norm": 3.361274242401123, + "learning_rate": 1.506080449017774e-06, + "loss": 1.3759, + "step": 1748 + }, + { + "epoch": 2.2083333333333335, + "grad_norm": 3.2770802974700928, + "learning_rate": 1.5037418147801685e-06, + "loss": 1.3497, + "step": 1749 + }, + { + "epoch": 2.20959595959596, + "grad_norm": 3.3818726539611816, + "learning_rate": 1.5014031805425634e-06, + "loss": 1.3277, + "step": 1750 + }, + { + "epoch": 2.20959595959596, + "eval_loss": 1.660167932510376, + "eval_runtime": 37.1049, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1750 + }, + { + "epoch": 2.2108585858585856, + "grad_norm": 3.296377420425415, + "learning_rate": 1.499064546304958e-06, + "loss": 1.3584, + "step": 1751 + }, + { + "epoch": 2.212121212121212, + "grad_norm": 3.4132192134857178, + "learning_rate": 1.4967259120673526e-06, + "loss": 1.367, + "step": 1752 + }, + { + "epoch": 2.2133838383838382, + "grad_norm": 3.3144986629486084, + "learning_rate": 1.4943872778297475e-06, + "loss": 1.3406, + "step": 1753 + }, + { + "epoch": 2.2146464646464645, + "grad_norm": 3.4283008575439453, + "learning_rate": 1.4920486435921421e-06, + "loss": 1.355, + "step": 1754 + }, + { + "epoch": 2.215909090909091, + "grad_norm": 3.319946050643921, + "learning_rate": 1.4897100093545372e-06, + "loss": 1.3665, + "step": 1755 + }, + { + "epoch": 2.217171717171717, + "grad_norm": 3.5585882663726807, + "learning_rate": 1.487371375116932e-06, + "loss": 1.4407, + "step": 1756 + }, + { + "epoch": 2.2184343434343434, + "grad_norm": 3.4216599464416504, + "learning_rate": 1.4850327408793267e-06, + "loss": 1.4105, + "step": 1757 + }, + { + "epoch": 2.2196969696969697, + "grad_norm": 3.3320260047912598, + "learning_rate": 1.4826941066417213e-06, + "loss": 1.3822, + "step": 1758 + }, + { + "epoch": 2.220959595959596, + "grad_norm": 3.2759056091308594, + "learning_rate": 1.4803554724041161e-06, + "loss": 1.4091, + "step": 1759 + }, + { + "epoch": 2.2222222222222223, + "grad_norm": 3.2865848541259766, + "learning_rate": 1.4780168381665108e-06, + "loss": 1.3643, + "step": 1760 + }, + { + "epoch": 2.2222222222222223, + "eval_loss": 1.662533164024353, + "eval_runtime": 37.1194, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1760 + }, + { + "epoch": 2.2234848484848486, + "grad_norm": 3.2485909461975098, + "learning_rate": 1.4756782039289056e-06, + "loss": 1.3986, + "step": 1761 + }, + { + "epoch": 2.224747474747475, + "grad_norm": 3.367898941040039, + "learning_rate": 1.4733395696913002e-06, + "loss": 1.37, + "step": 1762 + }, + { + "epoch": 2.226010101010101, + "grad_norm": 3.3919382095336914, + "learning_rate": 1.4710009354536953e-06, + "loss": 1.3517, + "step": 1763 + }, + { + "epoch": 2.227272727272727, + "grad_norm": 3.388627052307129, + "learning_rate": 1.46866230121609e-06, + "loss": 1.394, + "step": 1764 + }, + { + "epoch": 2.2285353535353534, + "grad_norm": 3.470121383666992, + "learning_rate": 1.4663236669784848e-06, + "loss": 1.3705, + "step": 1765 + }, + { + "epoch": 2.2297979797979797, + "grad_norm": 3.3515899181365967, + "learning_rate": 1.4639850327408794e-06, + "loss": 1.3736, + "step": 1766 + }, + { + "epoch": 2.231060606060606, + "grad_norm": 3.427877426147461, + "learning_rate": 1.4616463985032742e-06, + "loss": 1.3819, + "step": 1767 + }, + { + "epoch": 2.2323232323232323, + "grad_norm": 3.4387621879577637, + "learning_rate": 1.4593077642656689e-06, + "loss": 1.4043, + "step": 1768 + }, + { + "epoch": 2.2335858585858586, + "grad_norm": 3.4503068923950195, + "learning_rate": 1.4569691300280637e-06, + "loss": 1.3784, + "step": 1769 + }, + { + "epoch": 2.234848484848485, + "grad_norm": 3.5118091106414795, + "learning_rate": 1.4546304957904585e-06, + "loss": 1.319, + "step": 1770 + }, + { + "epoch": 2.234848484848485, + "eval_loss": 1.6638855934143066, + "eval_runtime": 37.1208, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 2.236111111111111, + "grad_norm": 3.3450958728790283, + "learning_rate": 1.4522918615528534e-06, + "loss": 1.3969, + "step": 1771 + }, + { + "epoch": 2.2373737373737375, + "grad_norm": 3.249265193939209, + "learning_rate": 1.449953227315248e-06, + "loss": 1.3543, + "step": 1772 + }, + { + "epoch": 2.2386363636363638, + "grad_norm": 3.33396315574646, + "learning_rate": 1.4476145930776429e-06, + "loss": 1.3595, + "step": 1773 + }, + { + "epoch": 2.23989898989899, + "grad_norm": 3.2275466918945312, + "learning_rate": 1.4452759588400375e-06, + "loss": 1.387, + "step": 1774 + }, + { + "epoch": 2.2411616161616164, + "grad_norm": 3.1885244846343994, + "learning_rate": 1.4429373246024323e-06, + "loss": 1.3815, + "step": 1775 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.41682767868042, + "learning_rate": 1.440598690364827e-06, + "loss": 1.398, + "step": 1776 + }, + { + "epoch": 2.2436868686868685, + "grad_norm": 3.3739681243896484, + "learning_rate": 1.4382600561272216e-06, + "loss": 1.4076, + "step": 1777 + }, + { + "epoch": 2.244949494949495, + "grad_norm": 3.3971354961395264, + "learning_rate": 1.4359214218896167e-06, + "loss": 1.3665, + "step": 1778 + }, + { + "epoch": 2.246212121212121, + "grad_norm": 3.375438690185547, + "learning_rate": 1.4335827876520115e-06, + "loss": 1.3841, + "step": 1779 + }, + { + "epoch": 2.2474747474747474, + "grad_norm": 3.4381961822509766, + "learning_rate": 1.4312441534144061e-06, + "loss": 1.3725, + "step": 1780 + }, + { + "epoch": 2.2474747474747474, + "eval_loss": 1.664459228515625, + "eval_runtime": 37.1364, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 1780 + }, + { + "epoch": 2.2487373737373737, + "grad_norm": 3.3988287448883057, + "learning_rate": 1.428905519176801e-06, + "loss": 1.3677, + "step": 1781 + }, + { + "epoch": 2.25, + "grad_norm": 3.3283984661102295, + "learning_rate": 1.4265668849391956e-06, + "loss": 1.4028, + "step": 1782 + }, + { + "epoch": 2.2512626262626263, + "grad_norm": 3.377519369125366, + "learning_rate": 1.4242282507015902e-06, + "loss": 1.4629, + "step": 1783 + }, + { + "epoch": 2.2525252525252526, + "grad_norm": 3.3736071586608887, + "learning_rate": 1.421889616463985e-06, + "loss": 1.3599, + "step": 1784 + }, + { + "epoch": 2.253787878787879, + "grad_norm": 3.4322469234466553, + "learning_rate": 1.4195509822263801e-06, + "loss": 1.3742, + "step": 1785 + }, + { + "epoch": 2.255050505050505, + "grad_norm": 3.2117607593536377, + "learning_rate": 1.4172123479887748e-06, + "loss": 1.3642, + "step": 1786 + }, + { + "epoch": 2.2563131313131315, + "grad_norm": 3.3926846981048584, + "learning_rate": 1.4148737137511694e-06, + "loss": 1.3988, + "step": 1787 + }, + { + "epoch": 2.257575757575758, + "grad_norm": 3.371910810470581, + "learning_rate": 1.4125350795135642e-06, + "loss": 1.3552, + "step": 1788 + }, + { + "epoch": 2.2588383838383836, + "grad_norm": 3.447720766067505, + "learning_rate": 1.4101964452759589e-06, + "loss": 1.3906, + "step": 1789 + }, + { + "epoch": 2.26010101010101, + "grad_norm": 3.396575450897217, + "learning_rate": 1.4078578110383537e-06, + "loss": 1.3641, + "step": 1790 + }, + { + "epoch": 2.26010101010101, + "eval_loss": 1.6667912006378174, + "eval_runtime": 37.3275, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 1790 + }, + { + "epoch": 2.2613636363636362, + "grad_norm": 3.532827138900757, + "learning_rate": 1.4055191768007483e-06, + "loss": 1.4078, + "step": 1791 + }, + { + "epoch": 2.2626262626262625, + "grad_norm": 3.438328981399536, + "learning_rate": 1.4031805425631432e-06, + "loss": 1.3589, + "step": 1792 + }, + { + "epoch": 2.263888888888889, + "grad_norm": 3.3978145122528076, + "learning_rate": 1.400841908325538e-06, + "loss": 1.4299, + "step": 1793 + }, + { + "epoch": 2.265151515151515, + "grad_norm": 3.457463502883911, + "learning_rate": 1.3985032740879329e-06, + "loss": 1.3814, + "step": 1794 + }, + { + "epoch": 2.2664141414141414, + "grad_norm": 3.2421817779541016, + "learning_rate": 1.3961646398503275e-06, + "loss": 1.3732, + "step": 1795 + }, + { + "epoch": 2.2676767676767677, + "grad_norm": 3.420693874359131, + "learning_rate": 1.3938260056127223e-06, + "loss": 1.431, + "step": 1796 + }, + { + "epoch": 2.268939393939394, + "grad_norm": 3.3793931007385254, + "learning_rate": 1.391487371375117e-06, + "loss": 1.3338, + "step": 1797 + }, + { + "epoch": 2.2702020202020203, + "grad_norm": 3.4106879234313965, + "learning_rate": 1.3891487371375118e-06, + "loss": 1.366, + "step": 1798 + }, + { + "epoch": 2.2714646464646466, + "grad_norm": 3.391782760620117, + "learning_rate": 1.3868101028999065e-06, + "loss": 1.3898, + "step": 1799 + }, + { + "epoch": 2.2727272727272725, + "grad_norm": 3.329619884490967, + "learning_rate": 1.3844714686623013e-06, + "loss": 1.3494, + "step": 1800 + }, + { + "epoch": 2.2727272727272725, + "eval_loss": 1.6635981798171997, + "eval_runtime": 37.4748, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 2.273989898989899, + "grad_norm": 3.383571147918701, + "learning_rate": 1.3821328344246961e-06, + "loss": 1.3603, + "step": 1801 + }, + { + "epoch": 2.275252525252525, + "grad_norm": 3.4224114418029785, + "learning_rate": 1.379794200187091e-06, + "loss": 1.4003, + "step": 1802 + }, + { + "epoch": 2.2765151515151514, + "grad_norm": 3.4758248329162598, + "learning_rate": 1.3774555659494856e-06, + "loss": 1.357, + "step": 1803 + }, + { + "epoch": 2.2777777777777777, + "grad_norm": 3.410489320755005, + "learning_rate": 1.3751169317118805e-06, + "loss": 1.3507, + "step": 1804 + }, + { + "epoch": 2.279040404040404, + "grad_norm": 3.4158382415771484, + "learning_rate": 1.372778297474275e-06, + "loss": 1.3822, + "step": 1805 + }, + { + "epoch": 2.2803030303030303, + "grad_norm": 3.562300443649292, + "learning_rate": 1.37043966323667e-06, + "loss": 1.3898, + "step": 1806 + }, + { + "epoch": 2.2815656565656566, + "grad_norm": 3.5485477447509766, + "learning_rate": 1.3681010289990646e-06, + "loss": 1.3341, + "step": 1807 + }, + { + "epoch": 2.282828282828283, + "grad_norm": 3.624049186706543, + "learning_rate": 1.3657623947614596e-06, + "loss": 1.4034, + "step": 1808 + }, + { + "epoch": 2.284090909090909, + "grad_norm": 3.37754225730896, + "learning_rate": 1.3634237605238542e-06, + "loss": 1.4328, + "step": 1809 + }, + { + "epoch": 2.2853535353535355, + "grad_norm": 3.5212652683258057, + "learning_rate": 1.361085126286249e-06, + "loss": 1.4038, + "step": 1810 + }, + { + "epoch": 2.2853535353535355, + "eval_loss": 1.6660585403442383, + "eval_runtime": 36.944, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 2.2866161616161618, + "grad_norm": 3.5564818382263184, + "learning_rate": 1.3587464920486437e-06, + "loss": 1.4132, + "step": 1811 + }, + { + "epoch": 2.287878787878788, + "grad_norm": 3.613368272781372, + "learning_rate": 1.3564078578110384e-06, + "loss": 1.4147, + "step": 1812 + }, + { + "epoch": 2.2891414141414144, + "grad_norm": 3.3956761360168457, + "learning_rate": 1.3540692235734332e-06, + "loss": 1.3691, + "step": 1813 + }, + { + "epoch": 2.29040404040404, + "grad_norm": 3.4510204792022705, + "learning_rate": 1.3517305893358278e-06, + "loss": 1.4152, + "step": 1814 + }, + { + "epoch": 2.2916666666666665, + "grad_norm": 3.475883960723877, + "learning_rate": 1.3493919550982227e-06, + "loss": 1.3786, + "step": 1815 + }, + { + "epoch": 2.292929292929293, + "grad_norm": 3.4566831588745117, + "learning_rate": 1.3470533208606177e-06, + "loss": 1.3737, + "step": 1816 + }, + { + "epoch": 2.294191919191919, + "grad_norm": 3.2443597316741943, + "learning_rate": 1.3447146866230124e-06, + "loss": 1.3435, + "step": 1817 + }, + { + "epoch": 2.2954545454545454, + "grad_norm": 3.4844541549682617, + "learning_rate": 1.342376052385407e-06, + "loss": 1.3759, + "step": 1818 + }, + { + "epoch": 2.2967171717171717, + "grad_norm": 3.3630642890930176, + "learning_rate": 1.3400374181478018e-06, + "loss": 1.363, + "step": 1819 + }, + { + "epoch": 2.297979797979798, + "grad_norm": 3.382788896560669, + "learning_rate": 1.3376987839101965e-06, + "loss": 1.4096, + "step": 1820 + }, + { + "epoch": 2.297979797979798, + "eval_loss": 1.664247751235962, + "eval_runtime": 37.0799, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1820 + }, + { + "epoch": 2.2992424242424243, + "grad_norm": 3.287827491760254, + "learning_rate": 1.3353601496725913e-06, + "loss": 1.348, + "step": 1821 + }, + { + "epoch": 2.3005050505050506, + "grad_norm": 3.388911008834839, + "learning_rate": 1.333021515434986e-06, + "loss": 1.3901, + "step": 1822 + }, + { + "epoch": 2.301767676767677, + "grad_norm": 3.3604469299316406, + "learning_rate": 1.3306828811973808e-06, + "loss": 1.3533, + "step": 1823 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.4020297527313232, + "learning_rate": 1.3283442469597756e-06, + "loss": 1.3469, + "step": 1824 + }, + { + "epoch": 2.304292929292929, + "grad_norm": 3.449728488922119, + "learning_rate": 1.3260056127221705e-06, + "loss": 1.3622, + "step": 1825 + }, + { + "epoch": 2.3055555555555554, + "grad_norm": 3.531832218170166, + "learning_rate": 1.323666978484565e-06, + "loss": 1.354, + "step": 1826 + }, + { + "epoch": 2.3068181818181817, + "grad_norm": 3.3406412601470947, + "learning_rate": 1.32132834424696e-06, + "loss": 1.3699, + "step": 1827 + }, + { + "epoch": 2.308080808080808, + "grad_norm": 3.5115885734558105, + "learning_rate": 1.3189897100093546e-06, + "loss": 1.4186, + "step": 1828 + }, + { + "epoch": 2.3093434343434343, + "grad_norm": 3.5668814182281494, + "learning_rate": 1.3166510757717494e-06, + "loss": 1.3894, + "step": 1829 + }, + { + "epoch": 2.3106060606060606, + "grad_norm": 3.5255587100982666, + "learning_rate": 1.314312441534144e-06, + "loss": 1.3567, + "step": 1830 + }, + { + "epoch": 2.3106060606060606, + "eval_loss": 1.6662371158599854, + "eval_runtime": 37.1254, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 1830 + }, + { + "epoch": 2.311868686868687, + "grad_norm": 3.4452052116394043, + "learning_rate": 1.311973807296539e-06, + "loss": 1.331, + "step": 1831 + }, + { + "epoch": 2.313131313131313, + "grad_norm": 3.466440439224243, + "learning_rate": 1.3096351730589337e-06, + "loss": 1.4204, + "step": 1832 + }, + { + "epoch": 2.3143939393939394, + "grad_norm": 3.2339625358581543, + "learning_rate": 1.3072965388213286e-06, + "loss": 1.3663, + "step": 1833 + }, + { + "epoch": 2.3156565656565657, + "grad_norm": 3.4634203910827637, + "learning_rate": 1.3049579045837232e-06, + "loss": 1.3805, + "step": 1834 + }, + { + "epoch": 2.316919191919192, + "grad_norm": 3.4900529384613037, + "learning_rate": 1.302619270346118e-06, + "loss": 1.3851, + "step": 1835 + }, + { + "epoch": 2.3181818181818183, + "grad_norm": 3.314610242843628, + "learning_rate": 1.3002806361085127e-06, + "loss": 1.3971, + "step": 1836 + }, + { + "epoch": 2.3194444444444446, + "grad_norm": 3.3920958042144775, + "learning_rate": 1.2979420018709073e-06, + "loss": 1.404, + "step": 1837 + }, + { + "epoch": 2.320707070707071, + "grad_norm": 3.3682339191436768, + "learning_rate": 1.2956033676333021e-06, + "loss": 1.3803, + "step": 1838 + }, + { + "epoch": 2.321969696969697, + "grad_norm": 3.332442283630371, + "learning_rate": 1.2932647333956972e-06, + "loss": 1.4214, + "step": 1839 + }, + { + "epoch": 2.323232323232323, + "grad_norm": 3.3797237873077393, + "learning_rate": 1.2909260991580918e-06, + "loss": 1.3323, + "step": 1840 + }, + { + "epoch": 2.323232323232323, + "eval_loss": 1.664433479309082, + "eval_runtime": 37.1313, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 2.3244949494949494, + "grad_norm": 3.454676866531372, + "learning_rate": 1.2885874649204867e-06, + "loss": 1.4014, + "step": 1841 + }, + { + "epoch": 2.3257575757575757, + "grad_norm": 3.4918594360351562, + "learning_rate": 1.2862488306828813e-06, + "loss": 1.3926, + "step": 1842 + }, + { + "epoch": 2.327020202020202, + "grad_norm": 3.291851043701172, + "learning_rate": 1.283910196445276e-06, + "loss": 1.3957, + "step": 1843 + }, + { + "epoch": 2.3282828282828283, + "grad_norm": 3.308403491973877, + "learning_rate": 1.2815715622076708e-06, + "loss": 1.3866, + "step": 1844 + }, + { + "epoch": 2.3295454545454546, + "grad_norm": 3.4691977500915527, + "learning_rate": 1.2792329279700654e-06, + "loss": 1.3944, + "step": 1845 + }, + { + "epoch": 2.330808080808081, + "grad_norm": 3.3912553787231445, + "learning_rate": 1.2768942937324603e-06, + "loss": 1.3521, + "step": 1846 + }, + { + "epoch": 2.332070707070707, + "grad_norm": 3.399017810821533, + "learning_rate": 1.2745556594948553e-06, + "loss": 1.3798, + "step": 1847 + }, + { + "epoch": 2.3333333333333335, + "grad_norm": 3.4829556941986084, + "learning_rate": 1.27221702525725e-06, + "loss": 1.3854, + "step": 1848 + }, + { + "epoch": 2.33459595959596, + "grad_norm": 3.363145351409912, + "learning_rate": 1.2698783910196446e-06, + "loss": 1.3581, + "step": 1849 + }, + { + "epoch": 2.3358585858585856, + "grad_norm": 3.3383405208587646, + "learning_rate": 1.2675397567820394e-06, + "loss": 1.3314, + "step": 1850 + }, + { + "epoch": 2.3358585858585856, + "eval_loss": 1.6598613262176514, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1850 + }, + { + "epoch": 2.337121212121212, + "grad_norm": 3.54443359375, + "learning_rate": 1.265201122544434e-06, + "loss": 1.3831, + "step": 1851 + }, + { + "epoch": 2.3383838383838382, + "grad_norm": 3.373283624649048, + "learning_rate": 1.2628624883068289e-06, + "loss": 1.4073, + "step": 1852 + }, + { + "epoch": 2.3396464646464645, + "grad_norm": 3.5655102729797363, + "learning_rate": 1.2605238540692235e-06, + "loss": 1.4081, + "step": 1853 + }, + { + "epoch": 2.340909090909091, + "grad_norm": 3.5852856636047363, + "learning_rate": 1.2581852198316186e-06, + "loss": 1.4566, + "step": 1854 + }, + { + "epoch": 2.342171717171717, + "grad_norm": 3.370293378829956, + "learning_rate": 1.2558465855940132e-06, + "loss": 1.3959, + "step": 1855 + }, + { + "epoch": 2.3434343434343434, + "grad_norm": 3.556837558746338, + "learning_rate": 1.253507951356408e-06, + "loss": 1.3989, + "step": 1856 + }, + { + "epoch": 2.3446969696969697, + "grad_norm": 3.3351283073425293, + "learning_rate": 1.2511693171188027e-06, + "loss": 1.3425, + "step": 1857 + }, + { + "epoch": 2.345959595959596, + "grad_norm": 3.432695150375366, + "learning_rate": 1.2488306828811975e-06, + "loss": 1.3849, + "step": 1858 + }, + { + "epoch": 2.3472222222222223, + "grad_norm": 3.6309726238250732, + "learning_rate": 1.2464920486435922e-06, + "loss": 1.4181, + "step": 1859 + }, + { + "epoch": 2.3484848484848486, + "grad_norm": 3.465672016143799, + "learning_rate": 1.244153414405987e-06, + "loss": 1.3992, + "step": 1860 + }, + { + "epoch": 2.3484848484848486, + "eval_loss": 1.6598951816558838, + "eval_runtime": 37.1338, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 1860 + }, + { + "epoch": 2.349747474747475, + "grad_norm": 3.333996534347534, + "learning_rate": 1.2418147801683818e-06, + "loss": 1.3765, + "step": 1861 + }, + { + "epoch": 2.351010101010101, + "grad_norm": 3.3039495944976807, + "learning_rate": 1.2394761459307765e-06, + "loss": 1.4055, + "step": 1862 + }, + { + "epoch": 2.3522727272727275, + "grad_norm": 3.42413067817688, + "learning_rate": 1.2371375116931713e-06, + "loss": 1.4089, + "step": 1863 + }, + { + "epoch": 2.3535353535353534, + "grad_norm": 3.626856565475464, + "learning_rate": 1.2347988774555662e-06, + "loss": 1.3382, + "step": 1864 + }, + { + "epoch": 2.3547979797979797, + "grad_norm": 3.677388906478882, + "learning_rate": 1.2324602432179608e-06, + "loss": 1.4251, + "step": 1865 + }, + { + "epoch": 2.356060606060606, + "grad_norm": 3.7962775230407715, + "learning_rate": 1.2301216089803556e-06, + "loss": 1.3808, + "step": 1866 + }, + { + "epoch": 2.3573232323232323, + "grad_norm": 3.4381752014160156, + "learning_rate": 1.2277829747427503e-06, + "loss": 1.3561, + "step": 1867 + }, + { + "epoch": 2.3585858585858586, + "grad_norm": 3.4678566455841064, + "learning_rate": 1.225444340505145e-06, + "loss": 1.3542, + "step": 1868 + }, + { + "epoch": 2.359848484848485, + "grad_norm": 3.4447884559631348, + "learning_rate": 1.22310570626754e-06, + "loss": 1.3809, + "step": 1869 + }, + { + "epoch": 2.361111111111111, + "grad_norm": 3.3679399490356445, + "learning_rate": 1.2207670720299346e-06, + "loss": 1.3825, + "step": 1870 + }, + { + "epoch": 2.361111111111111, + "eval_loss": 1.6562567949295044, + "eval_runtime": 37.1451, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 1870 + }, + { + "epoch": 2.3623737373737375, + "grad_norm": 3.3724935054779053, + "learning_rate": 1.2184284377923294e-06, + "loss": 1.3394, + "step": 1871 + }, + { + "epoch": 2.3636363636363638, + "grad_norm": 3.426579475402832, + "learning_rate": 1.2160898035547243e-06, + "loss": 1.4326, + "step": 1872 + }, + { + "epoch": 2.36489898989899, + "grad_norm": 3.4336345195770264, + "learning_rate": 1.213751169317119e-06, + "loss": 1.3575, + "step": 1873 + }, + { + "epoch": 2.3661616161616164, + "grad_norm": 3.4542808532714844, + "learning_rate": 1.2114125350795135e-06, + "loss": 1.3449, + "step": 1874 + }, + { + "epoch": 2.367424242424242, + "grad_norm": 3.362525701522827, + "learning_rate": 1.2090739008419086e-06, + "loss": 1.3964, + "step": 1875 + }, + { + "epoch": 2.3686868686868685, + "grad_norm": 3.392415761947632, + "learning_rate": 1.2067352666043032e-06, + "loss": 1.4272, + "step": 1876 + }, + { + "epoch": 2.369949494949495, + "grad_norm": 3.4062328338623047, + "learning_rate": 1.2043966323666978e-06, + "loss": 1.3407, + "step": 1877 + }, + { + "epoch": 2.371212121212121, + "grad_norm": 3.5435354709625244, + "learning_rate": 1.2020579981290927e-06, + "loss": 1.3671, + "step": 1878 + }, + { + "epoch": 2.3724747474747474, + "grad_norm": 3.743882656097412, + "learning_rate": 1.1997193638914875e-06, + "loss": 1.4046, + "step": 1879 + }, + { + "epoch": 2.3737373737373737, + "grad_norm": 3.752319574356079, + "learning_rate": 1.1973807296538822e-06, + "loss": 1.3631, + "step": 1880 + }, + { + "epoch": 2.3737373737373737, + "eval_loss": 1.6618274450302124, + "eval_runtime": 37.1614, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1880 + }, + { + "epoch": 2.375, + "grad_norm": 3.57468318939209, + "learning_rate": 1.195042095416277e-06, + "loss": 1.3584, + "step": 1881 + }, + { + "epoch": 2.3762626262626263, + "grad_norm": 3.544557809829712, + "learning_rate": 1.1927034611786716e-06, + "loss": 1.3759, + "step": 1882 + }, + { + "epoch": 2.3775252525252526, + "grad_norm": 3.313106060028076, + "learning_rate": 1.1903648269410665e-06, + "loss": 1.36, + "step": 1883 + }, + { + "epoch": 2.378787878787879, + "grad_norm": 3.347764730453491, + "learning_rate": 1.1880261927034613e-06, + "loss": 1.3789, + "step": 1884 + }, + { + "epoch": 2.380050505050505, + "grad_norm": 3.260697364807129, + "learning_rate": 1.185687558465856e-06, + "loss": 1.3456, + "step": 1885 + }, + { + "epoch": 2.3813131313131315, + "grad_norm": 3.287977695465088, + "learning_rate": 1.1833489242282508e-06, + "loss": 1.3739, + "step": 1886 + }, + { + "epoch": 2.382575757575758, + "grad_norm": 3.27716064453125, + "learning_rate": 1.1810102899906456e-06, + "loss": 1.3792, + "step": 1887 + }, + { + "epoch": 2.3838383838383836, + "grad_norm": 3.4265832901000977, + "learning_rate": 1.1786716557530403e-06, + "loss": 1.3604, + "step": 1888 + }, + { + "epoch": 2.38510101010101, + "grad_norm": 3.5619351863861084, + "learning_rate": 1.1763330215154351e-06, + "loss": 1.3786, + "step": 1889 + }, + { + "epoch": 2.3863636363636362, + "grad_norm": 3.482348680496216, + "learning_rate": 1.17399438727783e-06, + "loss": 1.415, + "step": 1890 + }, + { + "epoch": 2.3863636363636362, + "eval_loss": 1.6625829935073853, + "eval_runtime": 37.2791, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 2.3876262626262625, + "grad_norm": 3.375915288925171, + "learning_rate": 1.1716557530402246e-06, + "loss": 1.335, + "step": 1891 + }, + { + "epoch": 2.388888888888889, + "grad_norm": 3.5826265811920166, + "learning_rate": 1.1693171188026194e-06, + "loss": 1.3612, + "step": 1892 + }, + { + "epoch": 2.390151515151515, + "grad_norm": 3.787109851837158, + "learning_rate": 1.166978484565014e-06, + "loss": 1.386, + "step": 1893 + }, + { + "epoch": 2.3914141414141414, + "grad_norm": 3.5518991947174072, + "learning_rate": 1.164639850327409e-06, + "loss": 1.3372, + "step": 1894 + }, + { + "epoch": 2.3926767676767677, + "grad_norm": 3.4609735012054443, + "learning_rate": 1.1623012160898037e-06, + "loss": 1.362, + "step": 1895 + }, + { + "epoch": 2.393939393939394, + "grad_norm": 3.578777551651001, + "learning_rate": 1.1599625818521984e-06, + "loss": 1.3754, + "step": 1896 + }, + { + "epoch": 2.3952020202020203, + "grad_norm": 3.5557262897491455, + "learning_rate": 1.1576239476145932e-06, + "loss": 1.4048, + "step": 1897 + }, + { + "epoch": 2.3964646464646466, + "grad_norm": 3.3413872718811035, + "learning_rate": 1.155285313376988e-06, + "loss": 1.3531, + "step": 1898 + }, + { + "epoch": 2.3977272727272725, + "grad_norm": 3.207866668701172, + "learning_rate": 1.1529466791393827e-06, + "loss": 1.3479, + "step": 1899 + }, + { + "epoch": 2.398989898989899, + "grad_norm": 3.3909525871276855, + "learning_rate": 1.1506080449017775e-06, + "loss": 1.3663, + "step": 1900 + }, + { + "epoch": 2.398989898989899, + "eval_loss": 1.661224126815796, + "eval_runtime": 37.4772, + "eval_samples_per_second": 26.683, + "eval_steps_per_second": 1.121, + "step": 1900 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 5.922887877787648e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/latest new file mode 100644 index 0000000000000000000000000000000000000000..adbdf3f70e1e0db88a47bff3edcb0592ed0a2e61 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/latest @@ -0,0 +1 @@ +global_step2200 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..fd60b72712d5289becc0bc00f7441a5254cf6acc --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/trainer_state.json @@ -0,0 +1,17193 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.7777777777777777, + "eval_steps": 10, + "global_step": 2200, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + }, + { + "epoch": 1.5164141414141414, + "grad_norm": 3.041095733642578, + "learning_rate": 2.782974742750234e-06, + "loss": 1.4736, + "step": 1201 + }, + { + "epoch": 1.5176767676767677, + "grad_norm": 2.8400142192840576, + "learning_rate": 2.780636108512629e-06, + "loss": 1.4169, + "step": 1202 + }, + { + "epoch": 1.518939393939394, + "grad_norm": 2.785592555999756, + "learning_rate": 2.7782974742750236e-06, + "loss": 1.4553, + "step": 1203 + }, + { + "epoch": 1.5202020202020203, + "grad_norm": 2.75209641456604, + "learning_rate": 2.775958840037418e-06, + "loss": 1.4267, + "step": 1204 + }, + { + "epoch": 1.5214646464646466, + "grad_norm": 2.927558183670044, + "learning_rate": 2.773620205799813e-06, + "loss": 1.4377, + "step": 1205 + }, + { + "epoch": 1.5227272727272727, + "grad_norm": 2.9335508346557617, + "learning_rate": 2.7712815715622077e-06, + "loss": 1.4295, + "step": 1206 + }, + { + "epoch": 1.523989898989899, + "grad_norm": 2.7812745571136475, + "learning_rate": 2.7689429373246026e-06, + "loss": 1.4029, + "step": 1207 + }, + { + "epoch": 1.5252525252525253, + "grad_norm": 3.230062961578369, + "learning_rate": 2.766604303086998e-06, + "loss": 1.539, + "step": 1208 + }, + { + "epoch": 1.5265151515151514, + "grad_norm": 3.1244332790374756, + "learning_rate": 2.7642656688493923e-06, + "loss": 1.4829, + "step": 1209 + }, + { + "epoch": 1.5277777777777777, + "grad_norm": 3.2391650676727295, + "learning_rate": 2.761927034611787e-06, + "loss": 1.4677, + "step": 1210 + }, + { + "epoch": 1.5277777777777777, + "eval_loss": 1.6849907636642456, + "eval_runtime": 36.8638, + "eval_samples_per_second": 27.127, + "eval_steps_per_second": 1.139, + "step": 1210 + }, + { + "epoch": 1.529040404040404, + "grad_norm": 3.1884987354278564, + "learning_rate": 2.759588400374182e-06, + "loss": 1.4185, + "step": 1211 + }, + { + "epoch": 1.5303030303030303, + "grad_norm": 2.959118366241455, + "learning_rate": 2.7572497661365764e-06, + "loss": 1.4536, + "step": 1212 + }, + { + "epoch": 1.5315656565656566, + "grad_norm": 2.882380247116089, + "learning_rate": 2.7549111318989712e-06, + "loss": 1.4872, + "step": 1213 + }, + { + "epoch": 1.5328282828282829, + "grad_norm": 3.0574429035186768, + "learning_rate": 2.752572497661366e-06, + "loss": 1.4343, + "step": 1214 + }, + { + "epoch": 1.5340909090909092, + "grad_norm": 3.1645233631134033, + "learning_rate": 2.750233863423761e-06, + "loss": 1.5101, + "step": 1215 + }, + { + "epoch": 1.5353535353535355, + "grad_norm": 2.83034086227417, + "learning_rate": 2.7478952291861553e-06, + "loss": 1.45, + "step": 1216 + }, + { + "epoch": 1.5366161616161618, + "grad_norm": 2.964780330657959, + "learning_rate": 2.74555659494855e-06, + "loss": 1.453, + "step": 1217 + }, + { + "epoch": 1.5378787878787878, + "grad_norm": 2.9896774291992188, + "learning_rate": 2.743217960710945e-06, + "loss": 1.4648, + "step": 1218 + }, + { + "epoch": 1.5391414141414141, + "grad_norm": 3.013664484024048, + "learning_rate": 2.74087932647334e-06, + "loss": 1.4828, + "step": 1219 + }, + { + "epoch": 1.5404040404040404, + "grad_norm": 3.0011773109436035, + "learning_rate": 2.7385406922357343e-06, + "loss": 1.4705, + "step": 1220 + }, + { + "epoch": 1.5404040404040404, + "eval_loss": 1.6872731447219849, + "eval_runtime": 36.9864, + "eval_samples_per_second": 27.037, + "eval_steps_per_second": 1.136, + "step": 1220 + }, + { + "epoch": 1.5416666666666665, + "grad_norm": 3.239349842071533, + "learning_rate": 2.736202057998129e-06, + "loss": 1.484, + "step": 1221 + }, + { + "epoch": 1.5429292929292928, + "grad_norm": 2.9267003536224365, + "learning_rate": 2.733863423760524e-06, + "loss": 1.4587, + "step": 1222 + }, + { + "epoch": 1.5441919191919191, + "grad_norm": 2.890094518661499, + "learning_rate": 2.7315247895229192e-06, + "loss": 1.4456, + "step": 1223 + }, + { + "epoch": 1.5454545454545454, + "grad_norm": 3.0287511348724365, + "learning_rate": 2.7291861552853136e-06, + "loss": 1.5059, + "step": 1224 + }, + { + "epoch": 1.5467171717171717, + "grad_norm": 3.041609525680542, + "learning_rate": 2.7268475210477085e-06, + "loss": 1.4341, + "step": 1225 + }, + { + "epoch": 1.547979797979798, + "grad_norm": 2.8771862983703613, + "learning_rate": 2.7245088868101033e-06, + "loss": 1.4308, + "step": 1226 + }, + { + "epoch": 1.5492424242424243, + "grad_norm": 3.1173794269561768, + "learning_rate": 2.722170252572498e-06, + "loss": 1.514, + "step": 1227 + }, + { + "epoch": 1.5505050505050506, + "grad_norm": 2.838275671005249, + "learning_rate": 2.7198316183348926e-06, + "loss": 1.4798, + "step": 1228 + }, + { + "epoch": 1.551767676767677, + "grad_norm": 2.9917166233062744, + "learning_rate": 2.7174929840972874e-06, + "loss": 1.5256, + "step": 1229 + }, + { + "epoch": 1.553030303030303, + "grad_norm": 3.071840763092041, + "learning_rate": 2.7151543498596823e-06, + "loss": 1.5187, + "step": 1230 + }, + { + "epoch": 1.553030303030303, + "eval_loss": 1.6850661039352417, + "eval_runtime": 37.0094, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 1230 + }, + { + "epoch": 1.5542929292929293, + "grad_norm": 3.253169536590576, + "learning_rate": 2.7128157156220767e-06, + "loss": 1.5522, + "step": 1231 + }, + { + "epoch": 1.5555555555555556, + "grad_norm": 3.037151336669922, + "learning_rate": 2.7104770813844715e-06, + "loss": 1.5125, + "step": 1232 + }, + { + "epoch": 1.5568181818181817, + "grad_norm": 2.8486485481262207, + "learning_rate": 2.7081384471468664e-06, + "loss": 1.4169, + "step": 1233 + }, + { + "epoch": 1.558080808080808, + "grad_norm": 2.953824996948242, + "learning_rate": 2.7057998129092612e-06, + "loss": 1.4505, + "step": 1234 + }, + { + "epoch": 1.5593434343434343, + "grad_norm": 3.012303590774536, + "learning_rate": 2.7034611786716557e-06, + "loss": 1.4816, + "step": 1235 + }, + { + "epoch": 1.5606060606060606, + "grad_norm": 3.1798768043518066, + "learning_rate": 2.7011225444340505e-06, + "loss": 1.5116, + "step": 1236 + }, + { + "epoch": 1.5618686868686869, + "grad_norm": 3.11370587348938, + "learning_rate": 2.6987839101964453e-06, + "loss": 1.5035, + "step": 1237 + }, + { + "epoch": 1.5631313131313131, + "grad_norm": 3.0563342571258545, + "learning_rate": 2.69644527595884e-06, + "loss": 1.4387, + "step": 1238 + }, + { + "epoch": 1.5643939393939394, + "grad_norm": 2.9884026050567627, + "learning_rate": 2.6941066417212354e-06, + "loss": 1.428, + "step": 1239 + }, + { + "epoch": 1.5656565656565657, + "grad_norm": 2.955773115158081, + "learning_rate": 2.69176800748363e-06, + "loss": 1.4208, + "step": 1240 + }, + { + "epoch": 1.5656565656565657, + "eval_loss": 1.685523509979248, + "eval_runtime": 37.0248, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1240 + }, + { + "epoch": 1.566919191919192, + "grad_norm": 2.9858005046844482, + "learning_rate": 2.6894293732460247e-06, + "loss": 1.4706, + "step": 1241 + }, + { + "epoch": 1.5681818181818183, + "grad_norm": 3.122340440750122, + "learning_rate": 2.6870907390084196e-06, + "loss": 1.4892, + "step": 1242 + }, + { + "epoch": 1.5694444444444444, + "grad_norm": 2.9828262329101562, + "learning_rate": 2.684752104770814e-06, + "loss": 1.4255, + "step": 1243 + }, + { + "epoch": 1.5707070707070707, + "grad_norm": 2.916653633117676, + "learning_rate": 2.682413470533209e-06, + "loss": 1.4241, + "step": 1244 + }, + { + "epoch": 1.571969696969697, + "grad_norm": 3.1615540981292725, + "learning_rate": 2.6800748362956037e-06, + "loss": 1.4719, + "step": 1245 + }, + { + "epoch": 1.573232323232323, + "grad_norm": 3.154130697250366, + "learning_rate": 2.6777362020579985e-06, + "loss": 1.4731, + "step": 1246 + }, + { + "epoch": 1.5744949494949494, + "grad_norm": 3.040848731994629, + "learning_rate": 2.675397567820393e-06, + "loss": 1.5149, + "step": 1247 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.9808318614959717, + "learning_rate": 2.6730589335827878e-06, + "loss": 1.431, + "step": 1248 + }, + { + "epoch": 1.577020202020202, + "grad_norm": 2.8538382053375244, + "learning_rate": 2.6707202993451826e-06, + "loss": 1.4458, + "step": 1249 + }, + { + "epoch": 1.5782828282828283, + "grad_norm": 3.113945960998535, + "learning_rate": 2.6683816651075774e-06, + "loss": 1.4873, + "step": 1250 + }, + { + "epoch": 1.5782828282828283, + "eval_loss": 1.6823841333389282, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1250 + }, + { + "epoch": 1.5795454545454546, + "grad_norm": 3.0044517517089844, + "learning_rate": 2.666043030869972e-06, + "loss": 1.4904, + "step": 1251 + }, + { + "epoch": 1.5808080808080809, + "grad_norm": 2.845820665359497, + "learning_rate": 2.6637043966323667e-06, + "loss": 1.4174, + "step": 1252 + }, + { + "epoch": 1.5820707070707072, + "grad_norm": 2.9123542308807373, + "learning_rate": 2.6613657623947616e-06, + "loss": 1.4896, + "step": 1253 + }, + { + "epoch": 1.5833333333333335, + "grad_norm": 3.1593222618103027, + "learning_rate": 2.659027128157157e-06, + "loss": 1.5208, + "step": 1254 + }, + { + "epoch": 1.5845959595959596, + "grad_norm": 3.0782854557037354, + "learning_rate": 2.6566884939195512e-06, + "loss": 1.4911, + "step": 1255 + }, + { + "epoch": 1.5858585858585859, + "grad_norm": 2.77309513092041, + "learning_rate": 2.654349859681946e-06, + "loss": 1.4475, + "step": 1256 + }, + { + "epoch": 1.5871212121212122, + "grad_norm": 2.9154622554779053, + "learning_rate": 2.652011225444341e-06, + "loss": 1.4324, + "step": 1257 + }, + { + "epoch": 1.5883838383838382, + "grad_norm": 3.107395887374878, + "learning_rate": 2.6496725912067358e-06, + "loss": 1.5126, + "step": 1258 + }, + { + "epoch": 1.5896464646464645, + "grad_norm": 2.825025796890259, + "learning_rate": 2.64733395696913e-06, + "loss": 1.4723, + "step": 1259 + }, + { + "epoch": 1.5909090909090908, + "grad_norm": 2.833296060562134, + "learning_rate": 2.644995322731525e-06, + "loss": 1.4516, + "step": 1260 + }, + { + "epoch": 1.5909090909090908, + "eval_loss": 1.6950432062149048, + "eval_runtime": 37.0343, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1260 + }, + { + "epoch": 1.5921717171717171, + "grad_norm": 2.9508421421051025, + "learning_rate": 2.64265668849392e-06, + "loss": 1.4002, + "step": 1261 + }, + { + "epoch": 1.5934343434343434, + "grad_norm": 2.8778491020202637, + "learning_rate": 2.6403180542563143e-06, + "loss": 1.4258, + "step": 1262 + }, + { + "epoch": 1.5946969696969697, + "grad_norm": 2.81261944770813, + "learning_rate": 2.637979420018709e-06, + "loss": 1.435, + "step": 1263 + }, + { + "epoch": 1.595959595959596, + "grad_norm": 3.1811511516571045, + "learning_rate": 2.635640785781104e-06, + "loss": 1.4792, + "step": 1264 + }, + { + "epoch": 1.5972222222222223, + "grad_norm": 3.0168304443359375, + "learning_rate": 2.633302151543499e-06, + "loss": 1.5394, + "step": 1265 + }, + { + "epoch": 1.5984848484848486, + "grad_norm": 2.9531311988830566, + "learning_rate": 2.6309635173058932e-06, + "loss": 1.4775, + "step": 1266 + }, + { + "epoch": 1.5997474747474747, + "grad_norm": 2.835839033126831, + "learning_rate": 2.628624883068288e-06, + "loss": 1.478, + "step": 1267 + }, + { + "epoch": 1.601010101010101, + "grad_norm": 2.9014039039611816, + "learning_rate": 2.626286248830683e-06, + "loss": 1.4926, + "step": 1268 + }, + { + "epoch": 1.6022727272727273, + "grad_norm": 2.89461088180542, + "learning_rate": 2.623947614593078e-06, + "loss": 1.448, + "step": 1269 + }, + { + "epoch": 1.6035353535353534, + "grad_norm": 2.8005120754241943, + "learning_rate": 2.621608980355473e-06, + "loss": 1.4733, + "step": 1270 + }, + { + "epoch": 1.6035353535353534, + "eval_loss": 1.6889415979385376, + "eval_runtime": 37.0345, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1270 + }, + { + "epoch": 1.6047979797979797, + "grad_norm": 2.9576175212860107, + "learning_rate": 2.6192703461178675e-06, + "loss": 1.5116, + "step": 1271 + }, + { + "epoch": 1.606060606060606, + "grad_norm": 2.926638603210449, + "learning_rate": 2.6169317118802623e-06, + "loss": 1.4744, + "step": 1272 + }, + { + "epoch": 1.6073232323232323, + "grad_norm": 2.911587715148926, + "learning_rate": 2.614593077642657e-06, + "loss": 1.4695, + "step": 1273 + }, + { + "epoch": 1.6085858585858586, + "grad_norm": 2.8878750801086426, + "learning_rate": 2.6122544434050516e-06, + "loss": 1.4574, + "step": 1274 + }, + { + "epoch": 1.6098484848484849, + "grad_norm": 2.8489067554473877, + "learning_rate": 2.6099158091674464e-06, + "loss": 1.4599, + "step": 1275 + }, + { + "epoch": 1.6111111111111112, + "grad_norm": 2.8223719596862793, + "learning_rate": 2.6075771749298412e-06, + "loss": 1.4431, + "step": 1276 + }, + { + "epoch": 1.6123737373737375, + "grad_norm": 2.869910955429077, + "learning_rate": 2.605238540692236e-06, + "loss": 1.48, + "step": 1277 + }, + { + "epoch": 1.6136363636363638, + "grad_norm": 2.8435168266296387, + "learning_rate": 2.6028999064546305e-06, + "loss": 1.4471, + "step": 1278 + }, + { + "epoch": 1.61489898989899, + "grad_norm": 2.9384665489196777, + "learning_rate": 2.6005612722170253e-06, + "loss": 1.5176, + "step": 1279 + }, + { + "epoch": 1.6161616161616161, + "grad_norm": 3.021785020828247, + "learning_rate": 2.59822263797942e-06, + "loss": 1.4789, + "step": 1280 + }, + { + "epoch": 1.6161616161616161, + "eval_loss": 1.6864516735076904, + "eval_runtime": 37.0249, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1280 + }, + { + "epoch": 1.6174242424242424, + "grad_norm": 3.007183313369751, + "learning_rate": 2.5958840037418146e-06, + "loss": 1.4686, + "step": 1281 + }, + { + "epoch": 1.6186868686868687, + "grad_norm": 3.115910053253174, + "learning_rate": 2.5935453695042095e-06, + "loss": 1.5382, + "step": 1282 + }, + { + "epoch": 1.6199494949494948, + "grad_norm": 2.98108172416687, + "learning_rate": 2.5912067352666043e-06, + "loss": 1.4569, + "step": 1283 + }, + { + "epoch": 1.621212121212121, + "grad_norm": 2.926832437515259, + "learning_rate": 2.5888681010289996e-06, + "loss": 1.4515, + "step": 1284 + }, + { + "epoch": 1.6224747474747474, + "grad_norm": 2.9045534133911133, + "learning_rate": 2.5865294667913944e-06, + "loss": 1.4259, + "step": 1285 + }, + { + "epoch": 1.6237373737373737, + "grad_norm": 3.0757768154144287, + "learning_rate": 2.584190832553789e-06, + "loss": 1.5436, + "step": 1286 + }, + { + "epoch": 1.625, + "grad_norm": 3.045257806777954, + "learning_rate": 2.5818521983161837e-06, + "loss": 1.4378, + "step": 1287 + }, + { + "epoch": 1.6262626262626263, + "grad_norm": 2.8131601810455322, + "learning_rate": 2.5795135640785785e-06, + "loss": 1.4224, + "step": 1288 + }, + { + "epoch": 1.6275252525252526, + "grad_norm": 2.953326940536499, + "learning_rate": 2.5771749298409734e-06, + "loss": 1.4137, + "step": 1289 + }, + { + "epoch": 1.628787878787879, + "grad_norm": 2.946232318878174, + "learning_rate": 2.5748362956033678e-06, + "loss": 1.4575, + "step": 1290 + }, + { + "epoch": 1.628787878787879, + "eval_loss": 1.6896334886550903, + "eval_runtime": 37.317, + "eval_samples_per_second": 26.797, + "eval_steps_per_second": 1.125, + "step": 1290 + }, + { + "epoch": 1.6300505050505052, + "grad_norm": 2.860867738723755, + "learning_rate": 2.5724976613657626e-06, + "loss": 1.4412, + "step": 1291 + }, + { + "epoch": 1.6313131313131313, + "grad_norm": 3.0417261123657227, + "learning_rate": 2.5701590271281575e-06, + "loss": 1.4919, + "step": 1292 + }, + { + "epoch": 1.6325757575757576, + "grad_norm": 3.21425199508667, + "learning_rate": 2.567820392890552e-06, + "loss": 1.5243, + "step": 1293 + }, + { + "epoch": 1.6338383838383839, + "grad_norm": 3.000302314758301, + "learning_rate": 2.5654817586529467e-06, + "loss": 1.4571, + "step": 1294 + }, + { + "epoch": 1.63510101010101, + "grad_norm": 2.88415789604187, + "learning_rate": 2.5631431244153416e-06, + "loss": 1.4527, + "step": 1295 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.7747156620025635, + "learning_rate": 2.5608044901777364e-06, + "loss": 1.3984, + "step": 1296 + }, + { + "epoch": 1.6376262626262625, + "grad_norm": 2.9298319816589355, + "learning_rate": 2.558465855940131e-06, + "loss": 1.4587, + "step": 1297 + }, + { + "epoch": 1.6388888888888888, + "grad_norm": 3.0199968814849854, + "learning_rate": 2.5561272217025257e-06, + "loss": 1.4539, + "step": 1298 + }, + { + "epoch": 1.6401515151515151, + "grad_norm": 3.09503436088562, + "learning_rate": 2.5537885874649205e-06, + "loss": 1.4551, + "step": 1299 + }, + { + "epoch": 1.6414141414141414, + "grad_norm": 2.972161293029785, + "learning_rate": 2.5514499532273158e-06, + "loss": 1.4123, + "step": 1300 + }, + { + "epoch": 1.6414141414141414, + "eval_loss": 1.6877377033233643, + "eval_runtime": 37.4088, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1300 + }, + { + "epoch": 1.6426767676767677, + "grad_norm": 3.081058979034424, + "learning_rate": 2.5491113189897106e-06, + "loss": 1.457, + "step": 1301 + }, + { + "epoch": 1.643939393939394, + "grad_norm": 3.040106773376465, + "learning_rate": 2.546772684752105e-06, + "loss": 1.4465, + "step": 1302 + }, + { + "epoch": 1.6452020202020203, + "grad_norm": 3.133812427520752, + "learning_rate": 2.5444340505145e-06, + "loss": 1.4382, + "step": 1303 + }, + { + "epoch": 1.6464646464646466, + "grad_norm": 2.9701125621795654, + "learning_rate": 2.5420954162768947e-06, + "loss": 1.457, + "step": 1304 + }, + { + "epoch": 1.6477272727272727, + "grad_norm": 2.9272050857543945, + "learning_rate": 2.539756782039289e-06, + "loss": 1.464, + "step": 1305 + }, + { + "epoch": 1.648989898989899, + "grad_norm": 3.048401117324829, + "learning_rate": 2.537418147801684e-06, + "loss": 1.4688, + "step": 1306 + }, + { + "epoch": 1.6502525252525253, + "grad_norm": 2.9103102684020996, + "learning_rate": 2.535079513564079e-06, + "loss": 1.5025, + "step": 1307 + }, + { + "epoch": 1.6515151515151514, + "grad_norm": 3.0841548442840576, + "learning_rate": 2.5327408793264737e-06, + "loss": 1.4938, + "step": 1308 + }, + { + "epoch": 1.6527777777777777, + "grad_norm": 3.022184371948242, + "learning_rate": 2.530402245088868e-06, + "loss": 1.4332, + "step": 1309 + }, + { + "epoch": 1.654040404040404, + "grad_norm": 2.881479024887085, + "learning_rate": 2.528063610851263e-06, + "loss": 1.4541, + "step": 1310 + }, + { + "epoch": 1.654040404040404, + "eval_loss": 1.6814675331115723, + "eval_runtime": 36.9446, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.6553030303030303, + "grad_norm": 2.906719446182251, + "learning_rate": 2.5257249766136578e-06, + "loss": 1.4968, + "step": 1311 + }, + { + "epoch": 1.6565656565656566, + "grad_norm": 3.0566461086273193, + "learning_rate": 2.523386342376052e-06, + "loss": 1.4513, + "step": 1312 + }, + { + "epoch": 1.6578282828282829, + "grad_norm": 2.895026206970215, + "learning_rate": 2.521047708138447e-06, + "loss": 1.4217, + "step": 1313 + }, + { + "epoch": 1.6590909090909092, + "grad_norm": 3.0006465911865234, + "learning_rate": 2.518709073900842e-06, + "loss": 1.4793, + "step": 1314 + }, + { + "epoch": 1.6603535353535355, + "grad_norm": 2.9853575229644775, + "learning_rate": 2.516370439663237e-06, + "loss": 1.4642, + "step": 1315 + }, + { + "epoch": 1.6616161616161618, + "grad_norm": 2.9729392528533936, + "learning_rate": 2.514031805425632e-06, + "loss": 1.4644, + "step": 1316 + }, + { + "epoch": 1.6628787878787878, + "grad_norm": 3.0997586250305176, + "learning_rate": 2.5116931711880264e-06, + "loss": 1.4392, + "step": 1317 + }, + { + "epoch": 1.6641414141414141, + "grad_norm": 2.9621379375457764, + "learning_rate": 2.5093545369504213e-06, + "loss": 1.4387, + "step": 1318 + }, + { + "epoch": 1.6654040404040404, + "grad_norm": 3.0571911334991455, + "learning_rate": 2.507015902712816e-06, + "loss": 1.4638, + "step": 1319 + }, + { + "epoch": 1.6666666666666665, + "grad_norm": 2.9580321311950684, + "learning_rate": 2.504677268475211e-06, + "loss": 1.4985, + "step": 1320 + }, + { + "epoch": 1.6666666666666665, + "eval_loss": 1.6790977716445923, + "eval_runtime": 37.0242, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1320 + }, + { + "epoch": 1.6679292929292928, + "grad_norm": 2.87432599067688, + "learning_rate": 2.5023386342376054e-06, + "loss": 1.4058, + "step": 1321 + }, + { + "epoch": 1.6691919191919191, + "grad_norm": 2.925546407699585, + "learning_rate": 2.5e-06, + "loss": 1.4986, + "step": 1322 + }, + { + "epoch": 1.6704545454545454, + "grad_norm": 2.837743043899536, + "learning_rate": 2.497661365762395e-06, + "loss": 1.455, + "step": 1323 + }, + { + "epoch": 1.6717171717171717, + "grad_norm": 3.111581563949585, + "learning_rate": 2.4953227315247895e-06, + "loss": 1.5005, + "step": 1324 + }, + { + "epoch": 1.672979797979798, + "grad_norm": 2.851076126098633, + "learning_rate": 2.4929840972871843e-06, + "loss": 1.4365, + "step": 1325 + }, + { + "epoch": 1.6742424242424243, + "grad_norm": 2.966965675354004, + "learning_rate": 2.4906454630495796e-06, + "loss": 1.4697, + "step": 1326 + }, + { + "epoch": 1.6755050505050506, + "grad_norm": 3.015193223953247, + "learning_rate": 2.488306828811974e-06, + "loss": 1.4486, + "step": 1327 + }, + { + "epoch": 1.676767676767677, + "grad_norm": 3.017681121826172, + "learning_rate": 2.485968194574369e-06, + "loss": 1.4017, + "step": 1328 + }, + { + "epoch": 1.678030303030303, + "grad_norm": 3.071981906890869, + "learning_rate": 2.4836295603367637e-06, + "loss": 1.4572, + "step": 1329 + }, + { + "epoch": 1.6792929292929293, + "grad_norm": 3.051741123199463, + "learning_rate": 2.481290926099158e-06, + "loss": 1.4498, + "step": 1330 + }, + { + "epoch": 1.6792929292929293, + "eval_loss": 1.6771777868270874, + "eval_runtime": 37.0747, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1330 + }, + { + "epoch": 1.6805555555555556, + "grad_norm": 2.9561071395874023, + "learning_rate": 2.478952291861553e-06, + "loss": 1.4428, + "step": 1331 + }, + { + "epoch": 1.6818181818181817, + "grad_norm": 2.992584466934204, + "learning_rate": 2.4766136576239478e-06, + "loss": 1.461, + "step": 1332 + }, + { + "epoch": 1.683080808080808, + "grad_norm": 3.0473921298980713, + "learning_rate": 2.4742750233863426e-06, + "loss": 1.489, + "step": 1333 + }, + { + "epoch": 1.6843434343434343, + "grad_norm": 2.8721725940704346, + "learning_rate": 2.4719363891487375e-06, + "loss": 1.487, + "step": 1334 + }, + { + "epoch": 1.6856060606060606, + "grad_norm": 2.8650765419006348, + "learning_rate": 2.4695977549111323e-06, + "loss": 1.4504, + "step": 1335 + }, + { + "epoch": 1.6868686868686869, + "grad_norm": 2.998887538909912, + "learning_rate": 2.4672591206735267e-06, + "loss": 1.4586, + "step": 1336 + }, + { + "epoch": 1.6881313131313131, + "grad_norm": 2.915188789367676, + "learning_rate": 2.4649204864359216e-06, + "loss": 1.4333, + "step": 1337 + }, + { + "epoch": 1.6893939393939394, + "grad_norm": 2.9660091400146484, + "learning_rate": 2.4625818521983164e-06, + "loss": 1.4711, + "step": 1338 + }, + { + "epoch": 1.6906565656565657, + "grad_norm": 2.916274070739746, + "learning_rate": 2.4602432179607113e-06, + "loss": 1.4642, + "step": 1339 + }, + { + "epoch": 1.691919191919192, + "grad_norm": 3.04154896736145, + "learning_rate": 2.4579045837231057e-06, + "loss": 1.4233, + "step": 1340 + }, + { + "epoch": 1.691919191919192, + "eval_loss": 1.6764302253723145, + "eval_runtime": 37.0686, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1340 + }, + { + "epoch": 1.6931818181818183, + "grad_norm": 2.9976465702056885, + "learning_rate": 2.4555659494855005e-06, + "loss": 1.4775, + "step": 1341 + }, + { + "epoch": 1.6944444444444444, + "grad_norm": 2.9140090942382812, + "learning_rate": 2.4532273152478954e-06, + "loss": 1.4532, + "step": 1342 + }, + { + "epoch": 1.6957070707070707, + "grad_norm": 3.0224454402923584, + "learning_rate": 2.45088868101029e-06, + "loss": 1.5091, + "step": 1343 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.8979439735412598, + "learning_rate": 2.448550046772685e-06, + "loss": 1.4634, + "step": 1344 + }, + { + "epoch": 1.698232323232323, + "grad_norm": 2.913433313369751, + "learning_rate": 2.44621141253508e-06, + "loss": 1.4483, + "step": 1345 + }, + { + "epoch": 1.6994949494949494, + "grad_norm": 2.813455820083618, + "learning_rate": 2.4438727782974743e-06, + "loss": 1.446, + "step": 1346 + }, + { + "epoch": 1.7007575757575757, + "grad_norm": 2.988722324371338, + "learning_rate": 2.441534144059869e-06, + "loss": 1.4908, + "step": 1347 + }, + { + "epoch": 1.702020202020202, + "grad_norm": 2.850446939468384, + "learning_rate": 2.439195509822264e-06, + "loss": 1.4221, + "step": 1348 + }, + { + "epoch": 1.7032828282828283, + "grad_norm": 2.8405508995056152, + "learning_rate": 2.436856875584659e-06, + "loss": 1.4496, + "step": 1349 + }, + { + "epoch": 1.7045454545454546, + "grad_norm": 3.0188651084899902, + "learning_rate": 2.4345182413470537e-06, + "loss": 1.489, + "step": 1350 + }, + { + "epoch": 1.7045454545454546, + "eval_loss": 1.6780083179473877, + "eval_runtime": 37.0724, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1350 + }, + { + "epoch": 1.7058080808080809, + "grad_norm": 2.905088424682617, + "learning_rate": 2.4321796071094485e-06, + "loss": 1.4283, + "step": 1351 + }, + { + "epoch": 1.7070707070707072, + "grad_norm": 2.878998279571533, + "learning_rate": 2.429840972871843e-06, + "loss": 1.4357, + "step": 1352 + }, + { + "epoch": 1.7083333333333335, + "grad_norm": 3.1707355976104736, + "learning_rate": 2.427502338634238e-06, + "loss": 1.496, + "step": 1353 + }, + { + "epoch": 1.7095959595959596, + "grad_norm": 2.9610493183135986, + "learning_rate": 2.4251637043966326e-06, + "loss": 1.4905, + "step": 1354 + }, + { + "epoch": 1.7108585858585859, + "grad_norm": 2.8372201919555664, + "learning_rate": 2.422825070159027e-06, + "loss": 1.4563, + "step": 1355 + }, + { + "epoch": 1.7121212121212122, + "grad_norm": 2.8741753101348877, + "learning_rate": 2.420486435921422e-06, + "loss": 1.4525, + "step": 1356 + }, + { + "epoch": 1.7133838383838382, + "grad_norm": 2.973726749420166, + "learning_rate": 2.418147801683817e-06, + "loss": 1.4878, + "step": 1357 + }, + { + "epoch": 1.7146464646464645, + "grad_norm": 3.05717396736145, + "learning_rate": 2.4158091674462116e-06, + "loss": 1.5402, + "step": 1358 + }, + { + "epoch": 1.7159090909090908, + "grad_norm": 2.883772373199463, + "learning_rate": 2.4134705332086064e-06, + "loss": 1.4329, + "step": 1359 + }, + { + "epoch": 1.7171717171717171, + "grad_norm": 2.8246653079986572, + "learning_rate": 2.4111318989710013e-06, + "loss": 1.4821, + "step": 1360 + }, + { + "epoch": 1.7171717171717171, + "eval_loss": 1.6737746000289917, + "eval_runtime": 37.0666, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 1360 + }, + { + "epoch": 1.7184343434343434, + "grad_norm": 2.905977964401245, + "learning_rate": 2.4087932647333957e-06, + "loss": 1.4378, + "step": 1361 + }, + { + "epoch": 1.7196969696969697, + "grad_norm": 2.9268195629119873, + "learning_rate": 2.4064546304957905e-06, + "loss": 1.4765, + "step": 1362 + }, + { + "epoch": 1.720959595959596, + "grad_norm": 3.1369476318359375, + "learning_rate": 2.4041159962581854e-06, + "loss": 1.4532, + "step": 1363 + }, + { + "epoch": 1.7222222222222223, + "grad_norm": 3.04546856880188, + "learning_rate": 2.4017773620205802e-06, + "loss": 1.4921, + "step": 1364 + }, + { + "epoch": 1.7234848484848486, + "grad_norm": 2.872852325439453, + "learning_rate": 2.399438727782975e-06, + "loss": 1.4323, + "step": 1365 + }, + { + "epoch": 1.7247474747474747, + "grad_norm": 2.961681604385376, + "learning_rate": 2.39710009354537e-06, + "loss": 1.5181, + "step": 1366 + }, + { + "epoch": 1.726010101010101, + "grad_norm": 2.997729778289795, + "learning_rate": 2.3947614593077643e-06, + "loss": 1.4817, + "step": 1367 + }, + { + "epoch": 1.7272727272727273, + "grad_norm": 2.9097466468811035, + "learning_rate": 2.392422825070159e-06, + "loss": 1.4478, + "step": 1368 + }, + { + "epoch": 1.7285353535353534, + "grad_norm": 2.82433819770813, + "learning_rate": 2.390084190832554e-06, + "loss": 1.4065, + "step": 1369 + }, + { + "epoch": 1.7297979797979797, + "grad_norm": 2.940910577774048, + "learning_rate": 2.387745556594949e-06, + "loss": 1.4735, + "step": 1370 + }, + { + "epoch": 1.7297979797979797, + "eval_loss": 1.6717814207077026, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1370 + }, + { + "epoch": 1.731060606060606, + "grad_norm": 2.79101300239563, + "learning_rate": 2.3854069223573433e-06, + "loss": 1.4545, + "step": 1371 + }, + { + "epoch": 1.7323232323232323, + "grad_norm": 2.916954517364502, + "learning_rate": 2.3830682881197385e-06, + "loss": 1.4293, + "step": 1372 + }, + { + "epoch": 1.7335858585858586, + "grad_norm": 2.9609909057617188, + "learning_rate": 2.380729653882133e-06, + "loss": 1.4578, + "step": 1373 + }, + { + "epoch": 1.7348484848484849, + "grad_norm": 2.935925245285034, + "learning_rate": 2.378391019644528e-06, + "loss": 1.4577, + "step": 1374 + }, + { + "epoch": 1.7361111111111112, + "grad_norm": 2.9803526401519775, + "learning_rate": 2.3760523854069226e-06, + "loss": 1.4449, + "step": 1375 + }, + { + "epoch": 1.7373737373737375, + "grad_norm": 3.0292913913726807, + "learning_rate": 2.3737137511693175e-06, + "loss": 1.4308, + "step": 1376 + }, + { + "epoch": 1.7386363636363638, + "grad_norm": 3.0247879028320312, + "learning_rate": 2.371375116931712e-06, + "loss": 1.5406, + "step": 1377 + }, + { + "epoch": 1.73989898989899, + "grad_norm": 2.86775541305542, + "learning_rate": 2.3690364826941067e-06, + "loss": 1.4324, + "step": 1378 + }, + { + "epoch": 1.7411616161616161, + "grad_norm": 2.8593552112579346, + "learning_rate": 2.3666978484565016e-06, + "loss": 1.4526, + "step": 1379 + }, + { + "epoch": 1.7424242424242424, + "grad_norm": 2.8735320568084717, + "learning_rate": 2.3643592142188964e-06, + "loss": 1.4097, + "step": 1380 + }, + { + "epoch": 1.7424242424242424, + "eval_loss": 1.675827145576477, + "eval_runtime": 37.1206, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1380 + }, + { + "epoch": 1.7436868686868687, + "grad_norm": 2.948153495788574, + "learning_rate": 2.3620205799812913e-06, + "loss": 1.4234, + "step": 1381 + }, + { + "epoch": 1.7449494949494948, + "grad_norm": 2.9344282150268555, + "learning_rate": 2.359681945743686e-06, + "loss": 1.4653, + "step": 1382 + }, + { + "epoch": 1.746212121212121, + "grad_norm": 2.9309439659118652, + "learning_rate": 2.3573433115060805e-06, + "loss": 1.4676, + "step": 1383 + }, + { + "epoch": 1.7474747474747474, + "grad_norm": 3.0762970447540283, + "learning_rate": 2.3550046772684754e-06, + "loss": 1.4407, + "step": 1384 + }, + { + "epoch": 1.7487373737373737, + "grad_norm": 3.149824857711792, + "learning_rate": 2.3526660430308702e-06, + "loss": 1.4629, + "step": 1385 + }, + { + "epoch": 1.75, + "grad_norm": 3.109571695327759, + "learning_rate": 2.3503274087932646e-06, + "loss": 1.494, + "step": 1386 + }, + { + "epoch": 1.7512626262626263, + "grad_norm": 2.8359758853912354, + "learning_rate": 2.34798877455566e-06, + "loss": 1.4056, + "step": 1387 + }, + { + "epoch": 1.7525252525252526, + "grad_norm": 2.8523781299591064, + "learning_rate": 2.3456501403180548e-06, + "loss": 1.4389, + "step": 1388 + }, + { + "epoch": 1.753787878787879, + "grad_norm": 3.0218093395233154, + "learning_rate": 2.343311506080449e-06, + "loss": 1.4587, + "step": 1389 + }, + { + "epoch": 1.7550505050505052, + "grad_norm": 3.0975496768951416, + "learning_rate": 2.340972871842844e-06, + "loss": 1.5133, + "step": 1390 + }, + { + "epoch": 1.7550505050505052, + "eval_loss": 1.671931266784668, + "eval_runtime": 37.2749, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 1390 + }, + { + "epoch": 1.7563131313131313, + "grad_norm": 3.0177836418151855, + "learning_rate": 2.338634237605239e-06, + "loss": 1.4716, + "step": 1391 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 3.049105167388916, + "learning_rate": 2.3362956033676333e-06, + "loss": 1.4858, + "step": 1392 + }, + { + "epoch": 1.7588383838383839, + "grad_norm": 3.0062026977539062, + "learning_rate": 2.333956969130028e-06, + "loss": 1.4239, + "step": 1393 + }, + { + "epoch": 1.76010101010101, + "grad_norm": 3.071772813796997, + "learning_rate": 2.331618334892423e-06, + "loss": 1.4654, + "step": 1394 + }, + { + "epoch": 1.7613636363636362, + "grad_norm": 3.140510320663452, + "learning_rate": 2.329279700654818e-06, + "loss": 1.4854, + "step": 1395 + }, + { + "epoch": 1.7626262626262625, + "grad_norm": 2.9903907775878906, + "learning_rate": 2.3269410664172126e-06, + "loss": 1.4406, + "step": 1396 + }, + { + "epoch": 1.7638888888888888, + "grad_norm": 2.839891195297241, + "learning_rate": 2.3246024321796075e-06, + "loss": 1.4498, + "step": 1397 + }, + { + "epoch": 1.7651515151515151, + "grad_norm": 3.0252437591552734, + "learning_rate": 2.322263797942002e-06, + "loss": 1.472, + "step": 1398 + }, + { + "epoch": 1.7664141414141414, + "grad_norm": 3.0214762687683105, + "learning_rate": 2.3199251637043968e-06, + "loss": 1.4031, + "step": 1399 + }, + { + "epoch": 1.7676767676767677, + "grad_norm": 3.190088987350464, + "learning_rate": 2.3175865294667916e-06, + "loss": 1.4497, + "step": 1400 + }, + { + "epoch": 1.7676767676767677, + "eval_loss": 1.673120379447937, + "eval_runtime": 37.4414, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1400 + }, + { + "epoch": 1.768939393939394, + "grad_norm": 2.9620201587677, + "learning_rate": 2.3152478952291864e-06, + "loss": 1.4149, + "step": 1401 + }, + { + "epoch": 1.7702020202020203, + "grad_norm": 3.112372875213623, + "learning_rate": 2.312909260991581e-06, + "loss": 1.4569, + "step": 1402 + }, + { + "epoch": 1.7714646464646466, + "grad_norm": 3.138857364654541, + "learning_rate": 2.310570626753976e-06, + "loss": 1.4819, + "step": 1403 + }, + { + "epoch": 1.7727272727272727, + "grad_norm": 3.0957937240600586, + "learning_rate": 2.3082319925163705e-06, + "loss": 1.4374, + "step": 1404 + }, + { + "epoch": 1.773989898989899, + "grad_norm": 3.014039993286133, + "learning_rate": 2.3058933582787654e-06, + "loss": 1.4262, + "step": 1405 + }, + { + "epoch": 1.7752525252525253, + "grad_norm": 2.8127005100250244, + "learning_rate": 2.3035547240411602e-06, + "loss": 1.4468, + "step": 1406 + }, + { + "epoch": 1.7765151515151514, + "grad_norm": 2.9180614948272705, + "learning_rate": 2.301216089803555e-06, + "loss": 1.4259, + "step": 1407 + }, + { + "epoch": 1.7777777777777777, + "grad_norm": 3.0376124382019043, + "learning_rate": 2.2988774555659495e-06, + "loss": 1.4628, + "step": 1408 + }, + { + "epoch": 1.779040404040404, + "grad_norm": 3.031740665435791, + "learning_rate": 2.2965388213283443e-06, + "loss": 1.5058, + "step": 1409 + }, + { + "epoch": 1.7803030303030303, + "grad_norm": 2.9876646995544434, + "learning_rate": 2.294200187090739e-06, + "loss": 1.4605, + "step": 1410 + }, + { + "epoch": 1.7803030303030303, + "eval_loss": 1.668077826499939, + "eval_runtime": 36.9734, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 1410 + }, + { + "epoch": 1.7815656565656566, + "grad_norm": 2.953744888305664, + "learning_rate": 2.291861552853134e-06, + "loss": 1.4257, + "step": 1411 + }, + { + "epoch": 1.7828282828282829, + "grad_norm": 3.049036741256714, + "learning_rate": 2.289522918615529e-06, + "loss": 1.4476, + "step": 1412 + }, + { + "epoch": 1.7840909090909092, + "grad_norm": 3.102116584777832, + "learning_rate": 2.2871842843779237e-06, + "loss": 1.4701, + "step": 1413 + }, + { + "epoch": 1.7853535353535355, + "grad_norm": 2.842970132827759, + "learning_rate": 2.284845650140318e-06, + "loss": 1.415, + "step": 1414 + }, + { + "epoch": 1.7866161616161618, + "grad_norm": 2.9386725425720215, + "learning_rate": 2.282507015902713e-06, + "loss": 1.4561, + "step": 1415 + }, + { + "epoch": 1.7878787878787878, + "grad_norm": 2.942713499069214, + "learning_rate": 2.280168381665108e-06, + "loss": 1.435, + "step": 1416 + }, + { + "epoch": 1.7891414141414141, + "grad_norm": 2.943105936050415, + "learning_rate": 2.2778297474275022e-06, + "loss": 1.4923, + "step": 1417 + }, + { + "epoch": 1.7904040404040404, + "grad_norm": 2.7094717025756836, + "learning_rate": 2.2754911131898975e-06, + "loss": 1.4457, + "step": 1418 + }, + { + "epoch": 1.7916666666666665, + "grad_norm": 2.7859625816345215, + "learning_rate": 2.2731524789522923e-06, + "loss": 1.3993, + "step": 1419 + }, + { + "epoch": 1.7929292929292928, + "grad_norm": 3.0923988819122314, + "learning_rate": 2.2708138447146868e-06, + "loss": 1.5054, + "step": 1420 + }, + { + "epoch": 1.7929292929292928, + "eval_loss": 1.6687692403793335, + "eval_runtime": 37.0567, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.7941919191919191, + "grad_norm": 2.849726676940918, + "learning_rate": 2.2684752104770816e-06, + "loss": 1.3978, + "step": 1421 + }, + { + "epoch": 1.7954545454545454, + "grad_norm": 3.028932809829712, + "learning_rate": 2.2661365762394764e-06, + "loss": 1.4786, + "step": 1422 + }, + { + "epoch": 1.7967171717171717, + "grad_norm": 2.973188638687134, + "learning_rate": 2.263797942001871e-06, + "loss": 1.4393, + "step": 1423 + }, + { + "epoch": 1.797979797979798, + "grad_norm": 2.884032964706421, + "learning_rate": 2.2614593077642657e-06, + "loss": 1.4197, + "step": 1424 + }, + { + "epoch": 1.7992424242424243, + "grad_norm": 2.9058549404144287, + "learning_rate": 2.2591206735266606e-06, + "loss": 1.4313, + "step": 1425 + }, + { + "epoch": 1.8005050505050506, + "grad_norm": 2.905841827392578, + "learning_rate": 2.2567820392890554e-06, + "loss": 1.439, + "step": 1426 + }, + { + "epoch": 1.801767676767677, + "grad_norm": 2.927028179168701, + "learning_rate": 2.2544434050514502e-06, + "loss": 1.4296, + "step": 1427 + }, + { + "epoch": 1.803030303030303, + "grad_norm": 2.965529203414917, + "learning_rate": 2.252104770813845e-06, + "loss": 1.4185, + "step": 1428 + }, + { + "epoch": 1.8042929292929293, + "grad_norm": 2.9494433403015137, + "learning_rate": 2.2497661365762395e-06, + "loss": 1.5087, + "step": 1429 + }, + { + "epoch": 1.8055555555555556, + "grad_norm": 3.0137765407562256, + "learning_rate": 2.2474275023386343e-06, + "loss": 1.4383, + "step": 1430 + }, + { + "epoch": 1.8055555555555556, + "eval_loss": 1.6728461980819702, + "eval_runtime": 37.1033, + "eval_samples_per_second": 26.952, + "eval_steps_per_second": 1.132, + "step": 1430 + }, + { + "epoch": 1.8068181818181817, + "grad_norm": 2.980062484741211, + "learning_rate": 2.245088868101029e-06, + "loss": 1.4333, + "step": 1431 + }, + { + "epoch": 1.808080808080808, + "grad_norm": 3.143759250640869, + "learning_rate": 2.242750233863424e-06, + "loss": 1.4676, + "step": 1432 + }, + { + "epoch": 1.8093434343434343, + "grad_norm": 3.0134518146514893, + "learning_rate": 2.240411599625819e-06, + "loss": 1.4223, + "step": 1433 + }, + { + "epoch": 1.8106060606060606, + "grad_norm": 3.000018358230591, + "learning_rate": 2.2380729653882137e-06, + "loss": 1.4537, + "step": 1434 + }, + { + "epoch": 1.8118686868686869, + "grad_norm": 2.9828288555145264, + "learning_rate": 2.235734331150608e-06, + "loss": 1.4379, + "step": 1435 + }, + { + "epoch": 1.8131313131313131, + "grad_norm": 2.965486764907837, + "learning_rate": 2.233395696913003e-06, + "loss": 1.4244, + "step": 1436 + }, + { + "epoch": 1.8143939393939394, + "grad_norm": 2.9787070751190186, + "learning_rate": 2.231057062675398e-06, + "loss": 1.4117, + "step": 1437 + }, + { + "epoch": 1.8156565656565657, + "grad_norm": 3.0893537998199463, + "learning_rate": 2.2287184284377927e-06, + "loss": 1.4809, + "step": 1438 + }, + { + "epoch": 1.816919191919192, + "grad_norm": 2.8568801879882812, + "learning_rate": 2.226379794200187e-06, + "loss": 1.4232, + "step": 1439 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 3.1005046367645264, + "learning_rate": 2.224041159962582e-06, + "loss": 1.482, + "step": 1440 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.6738661527633667, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1440 + }, + { + "epoch": 1.8194444444444444, + "grad_norm": 2.9064700603485107, + "learning_rate": 2.2217025257249768e-06, + "loss": 1.4179, + "step": 1441 + }, + { + "epoch": 1.8207070707070707, + "grad_norm": 3.0111782550811768, + "learning_rate": 2.2193638914873716e-06, + "loss": 1.4084, + "step": 1442 + }, + { + "epoch": 1.821969696969697, + "grad_norm": 2.9269816875457764, + "learning_rate": 2.2170252572497665e-06, + "loss": 1.4361, + "step": 1443 + }, + { + "epoch": 1.823232323232323, + "grad_norm": 2.9985902309417725, + "learning_rate": 2.2146866230121613e-06, + "loss": 1.4345, + "step": 1444 + }, + { + "epoch": 1.8244949494949494, + "grad_norm": 3.1306066513061523, + "learning_rate": 2.2123479887745557e-06, + "loss": 1.4728, + "step": 1445 + }, + { + "epoch": 1.8257575757575757, + "grad_norm": 2.9876086711883545, + "learning_rate": 2.2100093545369506e-06, + "loss": 1.378, + "step": 1446 + }, + { + "epoch": 1.827020202020202, + "grad_norm": 3.018763780593872, + "learning_rate": 2.2076707202993454e-06, + "loss": 1.4506, + "step": 1447 + }, + { + "epoch": 1.8282828282828283, + "grad_norm": 2.960573673248291, + "learning_rate": 2.2053320860617402e-06, + "loss": 1.4674, + "step": 1448 + }, + { + "epoch": 1.8295454545454546, + "grad_norm": 2.849881887435913, + "learning_rate": 2.202993451824135e-06, + "loss": 1.4687, + "step": 1449 + }, + { + "epoch": 1.8308080808080809, + "grad_norm": 2.8798882961273193, + "learning_rate": 2.2006548175865295e-06, + "loss": 1.3868, + "step": 1450 + }, + { + "epoch": 1.8308080808080809, + "eval_loss": 1.6657519340515137, + "eval_runtime": 37.0993, + "eval_samples_per_second": 26.955, + "eval_steps_per_second": 1.132, + "step": 1450 + }, + { + "epoch": 1.8320707070707072, + "grad_norm": 2.937471628189087, + "learning_rate": 2.1983161833489243e-06, + "loss": 1.4085, + "step": 1451 + }, + { + "epoch": 1.8333333333333335, + "grad_norm": 3.0815699100494385, + "learning_rate": 2.195977549111319e-06, + "loss": 1.4055, + "step": 1452 + }, + { + "epoch": 1.8345959595959596, + "grad_norm": 2.927917718887329, + "learning_rate": 2.193638914873714e-06, + "loss": 1.4014, + "step": 1453 + }, + { + "epoch": 1.8358585858585859, + "grad_norm": 2.9576570987701416, + "learning_rate": 2.1913002806361085e-06, + "loss": 1.4394, + "step": 1454 + }, + { + "epoch": 1.8371212121212122, + "grad_norm": 2.9505863189697266, + "learning_rate": 2.1889616463985033e-06, + "loss": 1.4018, + "step": 1455 + }, + { + "epoch": 1.8383838383838382, + "grad_norm": 3.014965772628784, + "learning_rate": 2.186623012160898e-06, + "loss": 1.3735, + "step": 1456 + }, + { + "epoch": 1.8396464646464645, + "grad_norm": 3.121122121810913, + "learning_rate": 2.184284377923293e-06, + "loss": 1.4289, + "step": 1457 + }, + { + "epoch": 1.8409090909090908, + "grad_norm": 2.9710142612457275, + "learning_rate": 2.181945743685688e-06, + "loss": 1.4494, + "step": 1458 + }, + { + "epoch": 1.8421717171717171, + "grad_norm": 3.028554916381836, + "learning_rate": 2.1796071094480827e-06, + "loss": 1.5151, + "step": 1459 + }, + { + "epoch": 1.8434343434343434, + "grad_norm": 2.8973751068115234, + "learning_rate": 2.177268475210477e-06, + "loss": 1.4171, + "step": 1460 + }, + { + "epoch": 1.8434343434343434, + "eval_loss": 1.6670986413955688, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1460 + }, + { + "epoch": 1.8446969696969697, + "grad_norm": 3.046865701675415, + "learning_rate": 2.174929840972872e-06, + "loss": 1.4016, + "step": 1461 + }, + { + "epoch": 1.845959595959596, + "grad_norm": 3.0509603023529053, + "learning_rate": 2.1725912067352668e-06, + "loss": 1.4196, + "step": 1462 + }, + { + "epoch": 1.8472222222222223, + "grad_norm": 2.951892137527466, + "learning_rate": 2.1702525724976616e-06, + "loss": 1.4063, + "step": 1463 + }, + { + "epoch": 1.8484848484848486, + "grad_norm": 3.2126667499542236, + "learning_rate": 2.1679139382600565e-06, + "loss": 1.5106, + "step": 1464 + }, + { + "epoch": 1.8497474747474747, + "grad_norm": 2.965576648712158, + "learning_rate": 2.1655753040224513e-06, + "loss": 1.4096, + "step": 1465 + }, + { + "epoch": 1.851010101010101, + "grad_norm": 2.8640592098236084, + "learning_rate": 2.1632366697848457e-06, + "loss": 1.389, + "step": 1466 + }, + { + "epoch": 1.8522727272727273, + "grad_norm": 3.1476943492889404, + "learning_rate": 2.1608980355472406e-06, + "loss": 1.4793, + "step": 1467 + }, + { + "epoch": 1.8535353535353534, + "grad_norm": 2.814695119857788, + "learning_rate": 2.1585594013096354e-06, + "loss": 1.4265, + "step": 1468 + }, + { + "epoch": 1.8547979797979797, + "grad_norm": 2.9728431701660156, + "learning_rate": 2.1562207670720302e-06, + "loss": 1.386, + "step": 1469 + }, + { + "epoch": 1.856060606060606, + "grad_norm": 2.9062652587890625, + "learning_rate": 2.1538821328344247e-06, + "loss": 1.3986, + "step": 1470 + }, + { + "epoch": 1.856060606060606, + "eval_loss": 1.6698763370513916, + "eval_runtime": 37.1129, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1470 + }, + { + "epoch": 1.8573232323232323, + "grad_norm": 2.950857162475586, + "learning_rate": 2.15154349859682e-06, + "loss": 1.4193, + "step": 1471 + }, + { + "epoch": 1.8585858585858586, + "grad_norm": 3.017594337463379, + "learning_rate": 2.1492048643592144e-06, + "loss": 1.4318, + "step": 1472 + }, + { + "epoch": 1.8598484848484849, + "grad_norm": 2.991514205932617, + "learning_rate": 2.146866230121609e-06, + "loss": 1.4462, + "step": 1473 + }, + { + "epoch": 1.8611111111111112, + "grad_norm": 3.149475336074829, + "learning_rate": 2.144527595884004e-06, + "loss": 1.4486, + "step": 1474 + }, + { + "epoch": 1.8623737373737375, + "grad_norm": 2.948828935623169, + "learning_rate": 2.1421889616463985e-06, + "loss": 1.4522, + "step": 1475 + }, + { + "epoch": 1.8636363636363638, + "grad_norm": 3.0383694171905518, + "learning_rate": 2.1398503274087933e-06, + "loss": 1.45, + "step": 1476 + }, + { + "epoch": 1.86489898989899, + "grad_norm": 3.093449831008911, + "learning_rate": 2.137511693171188e-06, + "loss": 1.4266, + "step": 1477 + }, + { + "epoch": 1.8661616161616161, + "grad_norm": 3.017549753189087, + "learning_rate": 2.135173058933583e-06, + "loss": 1.4067, + "step": 1478 + }, + { + "epoch": 1.8674242424242424, + "grad_norm": 2.985044002532959, + "learning_rate": 2.132834424695978e-06, + "loss": 1.4904, + "step": 1479 + }, + { + "epoch": 1.8686868686868687, + "grad_norm": 2.9600346088409424, + "learning_rate": 2.1304957904583727e-06, + "loss": 1.4006, + "step": 1480 + }, + { + "epoch": 1.8686868686868687, + "eval_loss": 1.6648480892181396, + "eval_runtime": 37.1225, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1480 + }, + { + "epoch": 1.8699494949494948, + "grad_norm": 2.8905625343322754, + "learning_rate": 2.128157156220767e-06, + "loss": 1.4096, + "step": 1481 + }, + { + "epoch": 1.871212121212121, + "grad_norm": 2.95342755317688, + "learning_rate": 2.125818521983162e-06, + "loss": 1.3888, + "step": 1482 + }, + { + "epoch": 1.8724747474747474, + "grad_norm": 2.9438536167144775, + "learning_rate": 2.1234798877455568e-06, + "loss": 1.413, + "step": 1483 + }, + { + "epoch": 1.8737373737373737, + "grad_norm": 3.1257364749908447, + "learning_rate": 2.1211412535079516e-06, + "loss": 1.4902, + "step": 1484 + }, + { + "epoch": 1.875, + "grad_norm": 3.1377382278442383, + "learning_rate": 2.118802619270346e-06, + "loss": 1.4815, + "step": 1485 + }, + { + "epoch": 1.8762626262626263, + "grad_norm": 3.122253656387329, + "learning_rate": 2.116463985032741e-06, + "loss": 1.4695, + "step": 1486 + }, + { + "epoch": 1.8775252525252526, + "grad_norm": 3.0128729343414307, + "learning_rate": 2.1141253507951357e-06, + "loss": 1.4673, + "step": 1487 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.921835422515869, + "learning_rate": 2.1117867165575306e-06, + "loss": 1.4325, + "step": 1488 + }, + { + "epoch": 1.8800505050505052, + "grad_norm": 3.19289231300354, + "learning_rate": 2.1094480823199254e-06, + "loss": 1.4504, + "step": 1489 + }, + { + "epoch": 1.8813131313131313, + "grad_norm": 2.8579299449920654, + "learning_rate": 2.1071094480823203e-06, + "loss": 1.4042, + "step": 1490 + }, + { + "epoch": 1.8813131313131313, + "eval_loss": 1.6603271961212158, + "eval_runtime": 37.2997, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 1490 + }, + { + "epoch": 1.8825757575757576, + "grad_norm": 2.9113969802856445, + "learning_rate": 2.1047708138447147e-06, + "loss": 1.4128, + "step": 1491 + }, + { + "epoch": 1.8838383838383839, + "grad_norm": 3.058133363723755, + "learning_rate": 2.1024321796071095e-06, + "loss": 1.4574, + "step": 1492 + }, + { + "epoch": 1.88510101010101, + "grad_norm": 3.0066583156585693, + "learning_rate": 2.1000935453695044e-06, + "loss": 1.4521, + "step": 1493 + }, + { + "epoch": 1.8863636363636362, + "grad_norm": 2.9154908657073975, + "learning_rate": 2.097754911131899e-06, + "loss": 1.4103, + "step": 1494 + }, + { + "epoch": 1.8876262626262625, + "grad_norm": 2.977062702178955, + "learning_rate": 2.095416276894294e-06, + "loss": 1.4003, + "step": 1495 + }, + { + "epoch": 1.8888888888888888, + "grad_norm": 2.9048168659210205, + "learning_rate": 2.093077642656689e-06, + "loss": 1.4216, + "step": 1496 + }, + { + "epoch": 1.8901515151515151, + "grad_norm": 3.008733034133911, + "learning_rate": 2.0907390084190833e-06, + "loss": 1.4516, + "step": 1497 + }, + { + "epoch": 1.8914141414141414, + "grad_norm": 3.0832529067993164, + "learning_rate": 2.088400374181478e-06, + "loss": 1.4248, + "step": 1498 + }, + { + "epoch": 1.8926767676767677, + "grad_norm": 2.848066568374634, + "learning_rate": 2.086061739943873e-06, + "loss": 1.4042, + "step": 1499 + }, + { + "epoch": 1.893939393939394, + "grad_norm": 2.846782684326172, + "learning_rate": 2.0837231057062674e-06, + "loss": 1.409, + "step": 1500 + }, + { + "epoch": 1.893939393939394, + "eval_loss": 1.6672260761260986, + "eval_runtime": 37.4695, + "eval_samples_per_second": 26.688, + "eval_steps_per_second": 1.121, + "step": 1500 + }, + { + "epoch": 1.8952020202020203, + "grad_norm": 3.026818037033081, + "learning_rate": 2.0813844714686623e-06, + "loss": 1.4842, + "step": 1501 + }, + { + "epoch": 1.8964646464646466, + "grad_norm": 2.977665662765503, + "learning_rate": 2.0790458372310575e-06, + "loss": 1.4635, + "step": 1502 + }, + { + "epoch": 1.8977272727272727, + "grad_norm": 2.9093048572540283, + "learning_rate": 2.076707202993452e-06, + "loss": 1.4303, + "step": 1503 + }, + { + "epoch": 1.898989898989899, + "grad_norm": 3.0777814388275146, + "learning_rate": 2.0743685687558468e-06, + "loss": 1.4581, + "step": 1504 + }, + { + "epoch": 1.9002525252525253, + "grad_norm": 3.0283970832824707, + "learning_rate": 2.0720299345182416e-06, + "loss": 1.4665, + "step": 1505 + }, + { + "epoch": 1.9015151515151514, + "grad_norm": 2.852412223815918, + "learning_rate": 2.069691300280636e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.9027777777777777, + "grad_norm": 3.1163125038146973, + "learning_rate": 2.067352666043031e-06, + "loss": 1.4907, + "step": 1507 + }, + { + "epoch": 1.904040404040404, + "grad_norm": 2.9857897758483887, + "learning_rate": 2.0650140318054257e-06, + "loss": 1.4262, + "step": 1508 + }, + { + "epoch": 1.9053030303030303, + "grad_norm": 2.957388401031494, + "learning_rate": 2.0626753975678206e-06, + "loss": 1.4525, + "step": 1509 + }, + { + "epoch": 1.9065656565656566, + "grad_norm": 3.0107505321502686, + "learning_rate": 2.0603367633302154e-06, + "loss": 1.4461, + "step": 1510 + }, + { + "epoch": 1.9065656565656566, + "eval_loss": 1.6621215343475342, + "eval_runtime": 36.9336, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1510 + }, + { + "epoch": 1.9078282828282829, + "grad_norm": 3.112478494644165, + "learning_rate": 2.0579981290926103e-06, + "loss": 1.479, + "step": 1511 + }, + { + "epoch": 1.9090909090909092, + "grad_norm": 3.083494186401367, + "learning_rate": 2.0556594948550047e-06, + "loss": 1.4777, + "step": 1512 + }, + { + "epoch": 1.9103535353535355, + "grad_norm": 2.932652711868286, + "learning_rate": 2.0533208606173995e-06, + "loss": 1.3937, + "step": 1513 + }, + { + "epoch": 1.9116161616161618, + "grad_norm": 2.8349449634552, + "learning_rate": 2.0509822263797944e-06, + "loss": 1.4096, + "step": 1514 + }, + { + "epoch": 1.9128787878787878, + "grad_norm": 2.9760472774505615, + "learning_rate": 2.048643592142189e-06, + "loss": 1.4312, + "step": 1515 + }, + { + "epoch": 1.9141414141414141, + "grad_norm": 3.1012730598449707, + "learning_rate": 2.0463049579045836e-06, + "loss": 1.5189, + "step": 1516 + }, + { + "epoch": 1.9154040404040404, + "grad_norm": 2.9072659015655518, + "learning_rate": 2.043966323666979e-06, + "loss": 1.4051, + "step": 1517 + }, + { + "epoch": 1.9166666666666665, + "grad_norm": 2.840203285217285, + "learning_rate": 2.0416276894293733e-06, + "loss": 1.3899, + "step": 1518 + }, + { + "epoch": 1.9179292929292928, + "grad_norm": 3.130812406539917, + "learning_rate": 2.039289055191768e-06, + "loss": 1.4834, + "step": 1519 + }, + { + "epoch": 1.9191919191919191, + "grad_norm": 3.1507933139801025, + "learning_rate": 2.036950420954163e-06, + "loss": 1.402, + "step": 1520 + }, + { + "epoch": 1.9191919191919191, + "eval_loss": 1.6648352146148682, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1520 + }, + { + "epoch": 1.9204545454545454, + "grad_norm": 3.099269151687622, + "learning_rate": 2.034611786716558e-06, + "loss": 1.4189, + "step": 1521 + }, + { + "epoch": 1.9217171717171717, + "grad_norm": 3.0046563148498535, + "learning_rate": 2.0322731524789523e-06, + "loss": 1.405, + "step": 1522 + }, + { + "epoch": 1.922979797979798, + "grad_norm": 3.229635715484619, + "learning_rate": 2.029934518241347e-06, + "loss": 1.5031, + "step": 1523 + }, + { + "epoch": 1.9242424242424243, + "grad_norm": 3.0336225032806396, + "learning_rate": 2.027595884003742e-06, + "loss": 1.4685, + "step": 1524 + }, + { + "epoch": 1.9255050505050506, + "grad_norm": 2.9998443126678467, + "learning_rate": 2.025257249766137e-06, + "loss": 1.4558, + "step": 1525 + }, + { + "epoch": 1.926767676767677, + "grad_norm": 3.1587722301483154, + "learning_rate": 2.0229186155285316e-06, + "loss": 1.4916, + "step": 1526 + }, + { + "epoch": 1.928030303030303, + "grad_norm": 3.2228925228118896, + "learning_rate": 2.0205799812909265e-06, + "loss": 1.4373, + "step": 1527 + }, + { + "epoch": 1.9292929292929293, + "grad_norm": 3.06015944480896, + "learning_rate": 2.018241347053321e-06, + "loss": 1.4147, + "step": 1528 + }, + { + "epoch": 1.9305555555555556, + "grad_norm": 3.02284574508667, + "learning_rate": 2.0159027128157157e-06, + "loss": 1.4574, + "step": 1529 + }, + { + "epoch": 1.9318181818181817, + "grad_norm": 2.9933414459228516, + "learning_rate": 2.0135640785781106e-06, + "loss": 1.4283, + "step": 1530 + }, + { + "epoch": 1.9318181818181817, + "eval_loss": 1.6711069345474243, + "eval_runtime": 37.1182, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1530 + }, + { + "epoch": 1.933080808080808, + "grad_norm": 3.325230121612549, + "learning_rate": 2.011225444340505e-06, + "loss": 1.4532, + "step": 1531 + }, + { + "epoch": 1.9343434343434343, + "grad_norm": 3.0051827430725098, + "learning_rate": 2.0088868101029003e-06, + "loss": 1.3965, + "step": 1532 + }, + { + "epoch": 1.9356060606060606, + "grad_norm": 3.1392757892608643, + "learning_rate": 2.006548175865295e-06, + "loss": 1.4678, + "step": 1533 + }, + { + "epoch": 1.9368686868686869, + "grad_norm": 3.0719001293182373, + "learning_rate": 2.0042095416276895e-06, + "loss": 1.4374, + "step": 1534 + }, + { + "epoch": 1.9381313131313131, + "grad_norm": 3.0406386852264404, + "learning_rate": 2.0018709073900844e-06, + "loss": 1.4199, + "step": 1535 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9481236934661865, + "learning_rate": 1.9995322731524792e-06, + "loss": 1.4599, + "step": 1536 + }, + { + "epoch": 1.9406565656565657, + "grad_norm": 2.901155948638916, + "learning_rate": 1.9971936389148736e-06, + "loss": 1.381, + "step": 1537 + }, + { + "epoch": 1.941919191919192, + "grad_norm": 2.9623661041259766, + "learning_rate": 1.9948550046772685e-06, + "loss": 1.4769, + "step": 1538 + }, + { + "epoch": 1.9431818181818183, + "grad_norm": 2.8457608222961426, + "learning_rate": 1.9925163704396633e-06, + "loss": 1.4117, + "step": 1539 + }, + { + "epoch": 1.9444444444444444, + "grad_norm": 2.9782702922821045, + "learning_rate": 1.990177736202058e-06, + "loss": 1.4824, + "step": 1540 + }, + { + "epoch": 1.9444444444444444, + "eval_loss": 1.6663525104522705, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1540 + }, + { + "epoch": 1.9457070707070707, + "grad_norm": 2.8265817165374756, + "learning_rate": 1.987839101964453e-06, + "loss": 1.4375, + "step": 1541 + }, + { + "epoch": 1.946969696969697, + "grad_norm": 2.9103686809539795, + "learning_rate": 1.985500467726848e-06, + "loss": 1.474, + "step": 1542 + }, + { + "epoch": 1.948232323232323, + "grad_norm": 2.837301254272461, + "learning_rate": 1.9831618334892423e-06, + "loss": 1.4171, + "step": 1543 + }, + { + "epoch": 1.9494949494949494, + "grad_norm": 3.0468413829803467, + "learning_rate": 1.980823199251637e-06, + "loss": 1.5095, + "step": 1544 + }, + { + "epoch": 1.9507575757575757, + "grad_norm": 2.968480348587036, + "learning_rate": 1.978484565014032e-06, + "loss": 1.4203, + "step": 1545 + }, + { + "epoch": 1.952020202020202, + "grad_norm": 2.9865822792053223, + "learning_rate": 1.976145930776427e-06, + "loss": 1.3916, + "step": 1546 + }, + { + "epoch": 1.9532828282828283, + "grad_norm": 2.9733808040618896, + "learning_rate": 1.9738072965388212e-06, + "loss": 1.4216, + "step": 1547 + }, + { + "epoch": 1.9545454545454546, + "grad_norm": 2.941718101501465, + "learning_rate": 1.9714686623012165e-06, + "loss": 1.426, + "step": 1548 + }, + { + "epoch": 1.9558080808080809, + "grad_norm": 3.0980522632598877, + "learning_rate": 1.969130028063611e-06, + "loss": 1.471, + "step": 1549 + }, + { + "epoch": 1.9570707070707072, + "grad_norm": 2.797102689743042, + "learning_rate": 1.9667913938260057e-06, + "loss": 1.3897, + "step": 1550 + }, + { + "epoch": 1.9570707070707072, + "eval_loss": 1.661893367767334, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 1550 + }, + { + "epoch": 1.9583333333333335, + "grad_norm": 2.9562623500823975, + "learning_rate": 1.9644527595884006e-06, + "loss": 1.4481, + "step": 1551 + }, + { + "epoch": 1.9595959595959596, + "grad_norm": 2.9586915969848633, + "learning_rate": 1.9621141253507954e-06, + "loss": 1.4068, + "step": 1552 + }, + { + "epoch": 1.9608585858585859, + "grad_norm": 3.061110496520996, + "learning_rate": 1.95977549111319e-06, + "loss": 1.4524, + "step": 1553 + }, + { + "epoch": 1.9621212121212122, + "grad_norm": 3.080491542816162, + "learning_rate": 1.9574368568755847e-06, + "loss": 1.4457, + "step": 1554 + }, + { + "epoch": 1.9633838383838382, + "grad_norm": 3.083254814147949, + "learning_rate": 1.9550982226379795e-06, + "loss": 1.4775, + "step": 1555 + }, + { + "epoch": 1.9646464646464645, + "grad_norm": 3.008596420288086, + "learning_rate": 1.9527595884003744e-06, + "loss": 1.425, + "step": 1556 + }, + { + "epoch": 1.9659090909090908, + "grad_norm": 2.999195098876953, + "learning_rate": 1.9504209541627692e-06, + "loss": 1.4114, + "step": 1557 + }, + { + "epoch": 1.9671717171717171, + "grad_norm": 3.055265188217163, + "learning_rate": 1.948082319925164e-06, + "loss": 1.4767, + "step": 1558 + }, + { + "epoch": 1.9684343434343434, + "grad_norm": 3.0922064781188965, + "learning_rate": 1.9457436856875585e-06, + "loss": 1.4548, + "step": 1559 + }, + { + "epoch": 1.9696969696969697, + "grad_norm": 2.8740789890289307, + "learning_rate": 1.9434050514499533e-06, + "loss": 1.4582, + "step": 1560 + }, + { + "epoch": 1.9696969696969697, + "eval_loss": 1.6608474254608154, + "eval_runtime": 37.1201, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1560 + }, + { + "epoch": 1.970959595959596, + "grad_norm": 2.983570098876953, + "learning_rate": 1.941066417212348e-06, + "loss": 1.4602, + "step": 1561 + }, + { + "epoch": 1.9722222222222223, + "grad_norm": 3.132474184036255, + "learning_rate": 1.9387277829747426e-06, + "loss": 1.4689, + "step": 1562 + }, + { + "epoch": 1.9734848484848486, + "grad_norm": 2.9685440063476562, + "learning_rate": 1.936389148737138e-06, + "loss": 1.39, + "step": 1563 + }, + { + "epoch": 1.9747474747474747, + "grad_norm": 3.123037815093994, + "learning_rate": 1.9340505144995327e-06, + "loss": 1.4151, + "step": 1564 + }, + { + "epoch": 1.976010101010101, + "grad_norm": 3.135232925415039, + "learning_rate": 1.931711880261927e-06, + "loss": 1.4166, + "step": 1565 + }, + { + "epoch": 1.9772727272727273, + "grad_norm": 2.9131808280944824, + "learning_rate": 1.929373246024322e-06, + "loss": 1.3952, + "step": 1566 + }, + { + "epoch": 1.9785353535353534, + "grad_norm": 3.1187453269958496, + "learning_rate": 1.927034611786717e-06, + "loss": 1.4132, + "step": 1567 + }, + { + "epoch": 1.9797979797979797, + "grad_norm": 2.983447551727295, + "learning_rate": 1.9246959775491112e-06, + "loss": 1.4517, + "step": 1568 + }, + { + "epoch": 1.981060606060606, + "grad_norm": 2.9731497764587402, + "learning_rate": 1.922357343311506e-06, + "loss": 1.432, + "step": 1569 + }, + { + "epoch": 1.9823232323232323, + "grad_norm": 2.8526554107666016, + "learning_rate": 1.920018709073901e-06, + "loss": 1.4141, + "step": 1570 + }, + { + "epoch": 1.9823232323232323, + "eval_loss": 1.6579400300979614, + "eval_runtime": 37.1216, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1570 + }, + { + "epoch": 1.9835858585858586, + "grad_norm": 2.900019407272339, + "learning_rate": 1.9176800748362958e-06, + "loss": 1.4008, + "step": 1571 + }, + { + "epoch": 1.9848484848484849, + "grad_norm": 2.96079683303833, + "learning_rate": 1.9153414405986906e-06, + "loss": 1.4586, + "step": 1572 + }, + { + "epoch": 1.9861111111111112, + "grad_norm": 3.1275715827941895, + "learning_rate": 1.9130028063610854e-06, + "loss": 1.4768, + "step": 1573 + }, + { + "epoch": 1.9873737373737375, + "grad_norm": 3.044041395187378, + "learning_rate": 1.91066417212348e-06, + "loss": 1.4428, + "step": 1574 + }, + { + "epoch": 1.9886363636363638, + "grad_norm": 3.0206832885742188, + "learning_rate": 1.9083255378858747e-06, + "loss": 1.4492, + "step": 1575 + }, + { + "epoch": 1.98989898989899, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.398, + "step": 1576 + }, + { + "epoch": 1.9911616161616161, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.4373, + "step": 1577 + }, + { + "epoch": 1.9924242424242424, + "grad_norm": 3.129822254180908, + "learning_rate": 1.9036482694106642e-06, + "loss": 1.4599, + "step": 1578 + }, + { + "epoch": 1.9936868686868687, + "grad_norm": 2.978015661239624, + "learning_rate": 1.9013096351730592e-06, + "loss": 1.409, + "step": 1579 + }, + { + "epoch": 1.9949494949494948, + "grad_norm": 2.980295181274414, + "learning_rate": 1.8989710009354539e-06, + "loss": 1.4251, + "step": 1580 + }, + { + "epoch": 1.9949494949494948, + "eval_loss": 1.661036491394043, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.996212121212121, + "grad_norm": 2.878361225128174, + "learning_rate": 1.8966323666978487e-06, + "loss": 1.4289, + "step": 1581 + }, + { + "epoch": 1.9974747474747474, + "grad_norm": 2.967414379119873, + "learning_rate": 1.8942937324602433e-06, + "loss": 1.5151, + "step": 1582 + }, + { + "epoch": 1.9987373737373737, + "grad_norm": 2.9448516368865967, + "learning_rate": 1.8919550982226382e-06, + "loss": 1.4327, + "step": 1583 + }, + { + "epoch": 2.0, + "grad_norm": 2.9000067710876465, + "learning_rate": 1.8896164639850328e-06, + "loss": 1.4113, + "step": 1584 + }, + { + "epoch": 2.0012626262626263, + "grad_norm": 3.1630377769470215, + "learning_rate": 1.8872778297474277e-06, + "loss": 1.4179, + "step": 1585 + }, + { + "epoch": 2.0025252525252526, + "grad_norm": 3.2642927169799805, + "learning_rate": 1.8849391955098223e-06, + "loss": 1.3873, + "step": 1586 + }, + { + "epoch": 2.003787878787879, + "grad_norm": 3.1669039726257324, + "learning_rate": 1.8826005612722173e-06, + "loss": 1.3834, + "step": 1587 + }, + { + "epoch": 2.005050505050505, + "grad_norm": 3.145040988922119, + "learning_rate": 1.880261927034612e-06, + "loss": 1.3334, + "step": 1588 + }, + { + "epoch": 2.0063131313131315, + "grad_norm": 3.707200050354004, + "learning_rate": 1.8779232927970068e-06, + "loss": 1.3979, + "step": 1589 + }, + { + "epoch": 2.007575757575758, + "grad_norm": 3.936523199081421, + "learning_rate": 1.8755846585594014e-06, + "loss": 1.4221, + "step": 1590 + }, + { + "epoch": 2.007575757575758, + "eval_loss": 1.6722478866577148, + "eval_runtime": 37.2129, + "eval_samples_per_second": 26.872, + "eval_steps_per_second": 1.129, + "step": 1590 + }, + { + "epoch": 2.0088383838383836, + "grad_norm": 3.4221086502075195, + "learning_rate": 1.8732460243217963e-06, + "loss": 1.3894, + "step": 1591 + }, + { + "epoch": 2.01010101010101, + "grad_norm": 3.3115553855895996, + "learning_rate": 1.870907390084191e-06, + "loss": 1.4016, + "step": 1592 + }, + { + "epoch": 2.0113636363636362, + "grad_norm": 3.055779457092285, + "learning_rate": 1.8685687558465855e-06, + "loss": 1.3676, + "step": 1593 + }, + { + "epoch": 2.0126262626262625, + "grad_norm": 3.4615232944488525, + "learning_rate": 1.8662301216089804e-06, + "loss": 1.3624, + "step": 1594 + }, + { + "epoch": 2.013888888888889, + "grad_norm": 3.3298869132995605, + "learning_rate": 1.8638914873713754e-06, + "loss": 1.3939, + "step": 1595 + }, + { + "epoch": 2.015151515151515, + "grad_norm": 3.271921396255493, + "learning_rate": 1.86155285313377e-06, + "loss": 1.3994, + "step": 1596 + }, + { + "epoch": 2.0164141414141414, + "grad_norm": 3.272677421569824, + "learning_rate": 1.859214218896165e-06, + "loss": 1.3934, + "step": 1597 + }, + { + "epoch": 2.0176767676767677, + "grad_norm": 3.3485944271087646, + "learning_rate": 1.8568755846585596e-06, + "loss": 1.3626, + "step": 1598 + }, + { + "epoch": 2.018939393939394, + "grad_norm": 3.426981210708618, + "learning_rate": 1.8545369504209542e-06, + "loss": 1.4231, + "step": 1599 + }, + { + "epoch": 2.0202020202020203, + "grad_norm": 3.3859283924102783, + "learning_rate": 1.852198316183349e-06, + "loss": 1.3587, + "step": 1600 + }, + { + "epoch": 2.0202020202020203, + "eval_loss": 1.6732399463653564, + "eval_runtime": 37.3484, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 1600 + }, + { + "epoch": 2.0214646464646466, + "grad_norm": 3.537806987762451, + "learning_rate": 1.8498596819457437e-06, + "loss": 1.3729, + "step": 1601 + }, + { + "epoch": 2.022727272727273, + "grad_norm": 3.5417566299438477, + "learning_rate": 1.8475210477081387e-06, + "loss": 1.4272, + "step": 1602 + }, + { + "epoch": 2.023989898989899, + "grad_norm": 3.287195920944214, + "learning_rate": 1.8451824134705336e-06, + "loss": 1.3647, + "step": 1603 + }, + { + "epoch": 2.025252525252525, + "grad_norm": 3.233365297317505, + "learning_rate": 1.8428437792329282e-06, + "loss": 1.4062, + "step": 1604 + }, + { + "epoch": 2.0265151515151514, + "grad_norm": 3.296578884124756, + "learning_rate": 1.8405051449953228e-06, + "loss": 1.3841, + "step": 1605 + }, + { + "epoch": 2.0277777777777777, + "grad_norm": 3.2697315216064453, + "learning_rate": 1.8381665107577177e-06, + "loss": 1.3957, + "step": 1606 + }, + { + "epoch": 2.029040404040404, + "grad_norm": 3.1117050647735596, + "learning_rate": 1.8358278765201123e-06, + "loss": 1.3643, + "step": 1607 + }, + { + "epoch": 2.0303030303030303, + "grad_norm": 3.14170241355896, + "learning_rate": 1.8334892422825071e-06, + "loss": 1.3483, + "step": 1608 + }, + { + "epoch": 2.0315656565656566, + "grad_norm": 3.2749483585357666, + "learning_rate": 1.8311506080449018e-06, + "loss": 1.4007, + "step": 1609 + }, + { + "epoch": 2.032828282828283, + "grad_norm": 3.266631603240967, + "learning_rate": 1.8288119738072968e-06, + "loss": 1.3943, + "step": 1610 + }, + { + "epoch": 2.032828282828283, + "eval_loss": 1.6679413318634033, + "eval_runtime": 36.8768, + "eval_samples_per_second": 27.117, + "eval_steps_per_second": 1.139, + "step": 1610 + }, + { + "epoch": 2.034090909090909, + "grad_norm": 3.535625696182251, + "learning_rate": 1.8264733395696914e-06, + "loss": 1.4027, + "step": 1611 + }, + { + "epoch": 2.0353535353535355, + "grad_norm": 3.2741401195526123, + "learning_rate": 1.8241347053320863e-06, + "loss": 1.4075, + "step": 1612 + }, + { + "epoch": 2.0366161616161618, + "grad_norm": 3.3183717727661133, + "learning_rate": 1.821796071094481e-06, + "loss": 1.4223, + "step": 1613 + }, + { + "epoch": 2.037878787878788, + "grad_norm": 3.323057174682617, + "learning_rate": 1.8194574368568758e-06, + "loss": 1.3595, + "step": 1614 + }, + { + "epoch": 2.0391414141414144, + "grad_norm": 3.202054738998413, + "learning_rate": 1.8171188026192704e-06, + "loss": 1.3399, + "step": 1615 + }, + { + "epoch": 2.04040404040404, + "grad_norm": 3.3273448944091797, + "learning_rate": 1.8147801683816652e-06, + "loss": 1.3848, + "step": 1616 + }, + { + "epoch": 2.0416666666666665, + "grad_norm": 3.396843671798706, + "learning_rate": 1.81244153414406e-06, + "loss": 1.3968, + "step": 1617 + }, + { + "epoch": 2.042929292929293, + "grad_norm": 3.2145836353302, + "learning_rate": 1.810102899906455e-06, + "loss": 1.3571, + "step": 1618 + }, + { + "epoch": 2.044191919191919, + "grad_norm": 3.156754732131958, + "learning_rate": 1.8077642656688496e-06, + "loss": 1.3439, + "step": 1619 + }, + { + "epoch": 2.0454545454545454, + "grad_norm": 3.4259471893310547, + "learning_rate": 1.8054256314312444e-06, + "loss": 1.4123, + "step": 1620 + }, + { + "epoch": 2.0454545454545454, + "eval_loss": 1.6703134775161743, + "eval_runtime": 37.002, + "eval_samples_per_second": 27.026, + "eval_steps_per_second": 1.135, + "step": 1620 + }, + { + "epoch": 2.0467171717171717, + "grad_norm": 3.3556408882141113, + "learning_rate": 1.803086997193639e-06, + "loss": 1.4252, + "step": 1621 + }, + { + "epoch": 2.047979797979798, + "grad_norm": 3.6319949626922607, + "learning_rate": 1.8007483629560339e-06, + "loss": 1.4634, + "step": 1622 + }, + { + "epoch": 2.0492424242424243, + "grad_norm": 3.4495089054107666, + "learning_rate": 1.7984097287184285e-06, + "loss": 1.4066, + "step": 1623 + }, + { + "epoch": 2.0505050505050506, + "grad_norm": 3.182297945022583, + "learning_rate": 1.7960710944808231e-06, + "loss": 1.3899, + "step": 1624 + }, + { + "epoch": 2.051767676767677, + "grad_norm": 3.097158193588257, + "learning_rate": 1.7937324602432182e-06, + "loss": 1.293, + "step": 1625 + }, + { + "epoch": 2.053030303030303, + "grad_norm": 3.429065704345703, + "learning_rate": 1.791393826005613e-06, + "loss": 1.3905, + "step": 1626 + }, + { + "epoch": 2.0542929292929295, + "grad_norm": 3.183366298675537, + "learning_rate": 1.7890551917680077e-06, + "loss": 1.3444, + "step": 1627 + }, + { + "epoch": 2.0555555555555554, + "grad_norm": 3.175612449645996, + "learning_rate": 1.7867165575304025e-06, + "loss": 1.3985, + "step": 1628 + }, + { + "epoch": 2.0568181818181817, + "grad_norm": 3.26491641998291, + "learning_rate": 1.7843779232927971e-06, + "loss": 1.3607, + "step": 1629 + }, + { + "epoch": 2.058080808080808, + "grad_norm": 3.2279412746429443, + "learning_rate": 1.7820392890551918e-06, + "loss": 1.3545, + "step": 1630 + }, + { + "epoch": 2.058080808080808, + "eval_loss": 1.6729198694229126, + "eval_runtime": 37.0369, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 1630 + }, + { + "epoch": 2.0593434343434343, + "grad_norm": 3.3255503177642822, + "learning_rate": 1.7797006548175866e-06, + "loss": 1.3938, + "step": 1631 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.2912404537200928, + "learning_rate": 1.7773620205799812e-06, + "loss": 1.366, + "step": 1632 + }, + { + "epoch": 2.061868686868687, + "grad_norm": 3.2314395904541016, + "learning_rate": 1.7750233863423763e-06, + "loss": 1.3298, + "step": 1633 + }, + { + "epoch": 2.063131313131313, + "grad_norm": 3.3304178714752197, + "learning_rate": 1.772684752104771e-06, + "loss": 1.3592, + "step": 1634 + }, + { + "epoch": 2.0643939393939394, + "grad_norm": 3.3113889694213867, + "learning_rate": 1.7703461178671658e-06, + "loss": 1.3865, + "step": 1635 + }, + { + "epoch": 2.0656565656565657, + "grad_norm": 3.260434150695801, + "learning_rate": 1.7680074836295604e-06, + "loss": 1.3306, + "step": 1636 + }, + { + "epoch": 2.066919191919192, + "grad_norm": 3.2890963554382324, + "learning_rate": 1.7656688493919552e-06, + "loss": 1.3533, + "step": 1637 + }, + { + "epoch": 2.0681818181818183, + "grad_norm": 3.3027193546295166, + "learning_rate": 1.7633302151543499e-06, + "loss": 1.3746, + "step": 1638 + }, + { + "epoch": 2.0694444444444446, + "grad_norm": 3.3163208961486816, + "learning_rate": 1.7609915809167447e-06, + "loss": 1.4113, + "step": 1639 + }, + { + "epoch": 2.0707070707070705, + "grad_norm": 3.300557851791382, + "learning_rate": 1.7586529466791396e-06, + "loss": 1.3669, + "step": 1640 + }, + { + "epoch": 2.0707070707070705, + "eval_loss": 1.6676570177078247, + "eval_runtime": 37.0589, + "eval_samples_per_second": 26.984, + "eval_steps_per_second": 1.133, + "step": 1640 + }, + { + "epoch": 2.071969696969697, + "grad_norm": 3.305338144302368, + "learning_rate": 1.7563143124415344e-06, + "loss": 1.3969, + "step": 1641 + }, + { + "epoch": 2.073232323232323, + "grad_norm": 3.2047572135925293, + "learning_rate": 1.753975678203929e-06, + "loss": 1.3856, + "step": 1642 + }, + { + "epoch": 2.0744949494949494, + "grad_norm": 3.575038433074951, + "learning_rate": 1.7516370439663239e-06, + "loss": 1.3947, + "step": 1643 + }, + { + "epoch": 2.0757575757575757, + "grad_norm": 3.44581937789917, + "learning_rate": 1.7492984097287185e-06, + "loss": 1.4037, + "step": 1644 + }, + { + "epoch": 2.077020202020202, + "grad_norm": 3.574326753616333, + "learning_rate": 1.7469597754911134e-06, + "loss": 1.4216, + "step": 1645 + }, + { + "epoch": 2.0782828282828283, + "grad_norm": 3.2762558460235596, + "learning_rate": 1.744621141253508e-06, + "loss": 1.3621, + "step": 1646 + }, + { + "epoch": 2.0795454545454546, + "grad_norm": 3.3749659061431885, + "learning_rate": 1.7422825070159028e-06, + "loss": 1.3762, + "step": 1647 + }, + { + "epoch": 2.080808080808081, + "grad_norm": 3.287447214126587, + "learning_rate": 1.7399438727782977e-06, + "loss": 1.3565, + "step": 1648 + }, + { + "epoch": 2.082070707070707, + "grad_norm": 3.5107264518737793, + "learning_rate": 1.7376052385406925e-06, + "loss": 1.392, + "step": 1649 + }, + { + "epoch": 2.0833333333333335, + "grad_norm": 3.319650173187256, + "learning_rate": 1.7352666043030871e-06, + "loss": 1.3835, + "step": 1650 + }, + { + "epoch": 2.0833333333333335, + "eval_loss": 1.6727185249328613, + "eval_runtime": 37.0658, + "eval_samples_per_second": 26.979, + "eval_steps_per_second": 1.133, + "step": 1650 + }, + { + "epoch": 2.08459595959596, + "grad_norm": 3.4447925090789795, + "learning_rate": 1.732927970065482e-06, + "loss": 1.4186, + "step": 1651 + }, + { + "epoch": 2.0858585858585856, + "grad_norm": 3.5944249629974365, + "learning_rate": 1.7305893358278766e-06, + "loss": 1.4461, + "step": 1652 + }, + { + "epoch": 2.087121212121212, + "grad_norm": 3.5437049865722656, + "learning_rate": 1.7282507015902715e-06, + "loss": 1.3343, + "step": 1653 + }, + { + "epoch": 2.0883838383838382, + "grad_norm": 3.361407518386841, + "learning_rate": 1.725912067352666e-06, + "loss": 1.3653, + "step": 1654 + }, + { + "epoch": 2.0896464646464645, + "grad_norm": 3.4728963375091553, + "learning_rate": 1.7235734331150607e-06, + "loss": 1.4297, + "step": 1655 + }, + { + "epoch": 2.090909090909091, + "grad_norm": 3.3794610500335693, + "learning_rate": 1.7212347988774558e-06, + "loss": 1.3941, + "step": 1656 + }, + { + "epoch": 2.092171717171717, + "grad_norm": 3.280857801437378, + "learning_rate": 1.7188961646398506e-06, + "loss": 1.3445, + "step": 1657 + }, + { + "epoch": 2.0934343434343434, + "grad_norm": 3.29036283493042, + "learning_rate": 1.7165575304022453e-06, + "loss": 1.3581, + "step": 1658 + }, + { + "epoch": 2.0946969696969697, + "grad_norm": 3.1731486320495605, + "learning_rate": 1.7142188961646399e-06, + "loss": 1.3873, + "step": 1659 + }, + { + "epoch": 2.095959595959596, + "grad_norm": 3.2283008098602295, + "learning_rate": 1.7118802619270347e-06, + "loss": 1.3462, + "step": 1660 + }, + { + "epoch": 2.095959595959596, + "eval_loss": 1.6726486682891846, + "eval_runtime": 37.0617, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1660 + }, + { + "epoch": 2.0972222222222223, + "grad_norm": 3.329171657562256, + "learning_rate": 1.7095416276894294e-06, + "loss": 1.3974, + "step": 1661 + }, + { + "epoch": 2.0984848484848486, + "grad_norm": 3.4113657474517822, + "learning_rate": 1.7072029934518242e-06, + "loss": 1.4237, + "step": 1662 + }, + { + "epoch": 2.099747474747475, + "grad_norm": 3.431184768676758, + "learning_rate": 1.7048643592142193e-06, + "loss": 1.38, + "step": 1663 + }, + { + "epoch": 2.101010101010101, + "grad_norm": 3.5471761226654053, + "learning_rate": 1.7025257249766139e-06, + "loss": 1.3762, + "step": 1664 + }, + { + "epoch": 2.102272727272727, + "grad_norm": 3.43166446685791, + "learning_rate": 1.7001870907390085e-06, + "loss": 1.4371, + "step": 1665 + }, + { + "epoch": 2.1035353535353534, + "grad_norm": 3.28086256980896, + "learning_rate": 1.6978484565014034e-06, + "loss": 1.4062, + "step": 1666 + }, + { + "epoch": 2.1047979797979797, + "grad_norm": 3.166083335876465, + "learning_rate": 1.695509822263798e-06, + "loss": 1.3535, + "step": 1667 + }, + { + "epoch": 2.106060606060606, + "grad_norm": 3.4707589149475098, + "learning_rate": 1.6931711880261928e-06, + "loss": 1.4237, + "step": 1668 + }, + { + "epoch": 2.1073232323232323, + "grad_norm": 3.5033648014068604, + "learning_rate": 1.6908325537885875e-06, + "loss": 1.4331, + "step": 1669 + }, + { + "epoch": 2.1085858585858586, + "grad_norm": 3.3915514945983887, + "learning_rate": 1.6884939195509823e-06, + "loss": 1.3783, + "step": 1670 + }, + { + "epoch": 2.1085858585858586, + "eval_loss": 1.6657648086547852, + "eval_runtime": 37.0742, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1670 + }, + { + "epoch": 2.109848484848485, + "grad_norm": 3.3420865535736084, + "learning_rate": 1.6861552853133772e-06, + "loss": 1.4016, + "step": 1671 + }, + { + "epoch": 2.111111111111111, + "grad_norm": 3.382840394973755, + "learning_rate": 1.683816651075772e-06, + "loss": 1.4209, + "step": 1672 + }, + { + "epoch": 2.1123737373737375, + "grad_norm": 3.3109424114227295, + "learning_rate": 1.6814780168381666e-06, + "loss": 1.3716, + "step": 1673 + }, + { + "epoch": 2.1136363636363638, + "grad_norm": 3.3781373500823975, + "learning_rate": 1.6791393826005615e-06, + "loss": 1.4106, + "step": 1674 + }, + { + "epoch": 2.11489898989899, + "grad_norm": 3.29980731010437, + "learning_rate": 1.676800748362956e-06, + "loss": 1.3628, + "step": 1675 + }, + { + "epoch": 2.1161616161616164, + "grad_norm": 3.38468599319458, + "learning_rate": 1.674462114125351e-06, + "loss": 1.3478, + "step": 1676 + }, + { + "epoch": 2.117424242424242, + "grad_norm": 3.4106788635253906, + "learning_rate": 1.6721234798877456e-06, + "loss": 1.391, + "step": 1677 + }, + { + "epoch": 2.1186868686868685, + "grad_norm": 3.412184476852417, + "learning_rate": 1.6697848456501404e-06, + "loss": 1.4245, + "step": 1678 + }, + { + "epoch": 2.119949494949495, + "grad_norm": 3.3659188747406006, + "learning_rate": 1.6674462114125353e-06, + "loss": 1.3358, + "step": 1679 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.241384744644165, + "learning_rate": 1.66510757717493e-06, + "loss": 1.3855, + "step": 1680 + }, + { + "epoch": 2.121212121212121, + "eval_loss": 1.6706457138061523, + "eval_runtime": 37.0651, + "eval_samples_per_second": 26.98, + "eval_steps_per_second": 1.133, + "step": 1680 + }, + { + "epoch": 2.1224747474747474, + "grad_norm": 3.3625874519348145, + "learning_rate": 1.6627689429373247e-06, + "loss": 1.3955, + "step": 1681 + }, + { + "epoch": 2.1237373737373737, + "grad_norm": 3.378143787384033, + "learning_rate": 1.6604303086997196e-06, + "loss": 1.3635, + "step": 1682 + }, + { + "epoch": 2.125, + "grad_norm": 3.4872324466705322, + "learning_rate": 1.6580916744621142e-06, + "loss": 1.4037, + "step": 1683 + }, + { + "epoch": 2.1262626262626263, + "grad_norm": 3.4733831882476807, + "learning_rate": 1.655753040224509e-06, + "loss": 1.3846, + "step": 1684 + }, + { + "epoch": 2.1275252525252526, + "grad_norm": 3.4518625736236572, + "learning_rate": 1.6534144059869037e-06, + "loss": 1.3984, + "step": 1685 + }, + { + "epoch": 2.128787878787879, + "grad_norm": 3.3702666759490967, + "learning_rate": 1.6510757717492987e-06, + "loss": 1.3712, + "step": 1686 + }, + { + "epoch": 2.130050505050505, + "grad_norm": 3.357708215713501, + "learning_rate": 1.6487371375116934e-06, + "loss": 1.4204, + "step": 1687 + }, + { + "epoch": 2.1313131313131315, + "grad_norm": 3.3902523517608643, + "learning_rate": 1.6463985032740882e-06, + "loss": 1.439, + "step": 1688 + }, + { + "epoch": 2.132575757575758, + "grad_norm": 3.246450424194336, + "learning_rate": 1.6440598690364828e-06, + "loss": 1.4061, + "step": 1689 + }, + { + "epoch": 2.1338383838383836, + "grad_norm": 3.5247316360473633, + "learning_rate": 1.6417212347988775e-06, + "loss": 1.4148, + "step": 1690 + }, + { + "epoch": 2.1338383838383836, + "eval_loss": 1.6684828996658325, + "eval_runtime": 37.3778, + "eval_samples_per_second": 26.754, + "eval_steps_per_second": 1.124, + "step": 1690 + }, + { + "epoch": 2.13510101010101, + "grad_norm": 3.3932440280914307, + "learning_rate": 1.6393826005612723e-06, + "loss": 1.3756, + "step": 1691 + }, + { + "epoch": 2.1363636363636362, + "grad_norm": 3.5242092609405518, + "learning_rate": 1.637043966323667e-06, + "loss": 1.4009, + "step": 1692 + }, + { + "epoch": 2.1376262626262625, + "grad_norm": 3.464235544204712, + "learning_rate": 1.6347053320860618e-06, + "loss": 1.4129, + "step": 1693 + }, + { + "epoch": 2.138888888888889, + "grad_norm": 3.4409964084625244, + "learning_rate": 1.6323666978484568e-06, + "loss": 1.3969, + "step": 1694 + }, + { + "epoch": 2.140151515151515, + "grad_norm": 3.329281806945801, + "learning_rate": 1.6300280636108515e-06, + "loss": 1.367, + "step": 1695 + }, + { + "epoch": 2.1414141414141414, + "grad_norm": 3.28047251701355, + "learning_rate": 1.6276894293732461e-06, + "loss": 1.3994, + "step": 1696 + }, + { + "epoch": 2.1426767676767677, + "grad_norm": 3.1937954425811768, + "learning_rate": 1.625350795135641e-06, + "loss": 1.3626, + "step": 1697 + }, + { + "epoch": 2.143939393939394, + "grad_norm": 3.5084168910980225, + "learning_rate": 1.6230121608980356e-06, + "loss": 1.4067, + "step": 1698 + }, + { + "epoch": 2.1452020202020203, + "grad_norm": 3.3637681007385254, + "learning_rate": 1.6206735266604304e-06, + "loss": 1.3619, + "step": 1699 + }, + { + "epoch": 2.1464646464646466, + "grad_norm": 3.2855823040008545, + "learning_rate": 1.618334892422825e-06, + "loss": 1.4227, + "step": 1700 + }, + { + "epoch": 2.1464646464646466, + "eval_loss": 1.6681337356567383, + "eval_runtime": 37.4375, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1700 + }, + { + "epoch": 2.147727272727273, + "grad_norm": 3.4610795974731445, + "learning_rate": 1.6159962581852201e-06, + "loss": 1.3842, + "step": 1701 + }, + { + "epoch": 2.148989898989899, + "grad_norm": 3.420252561569214, + "learning_rate": 1.6136576239476147e-06, + "loss": 1.3951, + "step": 1702 + }, + { + "epoch": 2.150252525252525, + "grad_norm": 3.4725613594055176, + "learning_rate": 1.6113189897100096e-06, + "loss": 1.3697, + "step": 1703 + }, + { + "epoch": 2.1515151515151514, + "grad_norm": 3.3183791637420654, + "learning_rate": 1.6089803554724042e-06, + "loss": 1.3813, + "step": 1704 + }, + { + "epoch": 2.1527777777777777, + "grad_norm": 3.3727011680603027, + "learning_rate": 1.606641721234799e-06, + "loss": 1.3654, + "step": 1705 + }, + { + "epoch": 2.154040404040404, + "grad_norm": 3.420855760574341, + "learning_rate": 1.6043030869971937e-06, + "loss": 1.3399, + "step": 1706 + }, + { + "epoch": 2.1553030303030303, + "grad_norm": 3.3293979167938232, + "learning_rate": 1.6019644527595885e-06, + "loss": 1.3694, + "step": 1707 + }, + { + "epoch": 2.1565656565656566, + "grad_norm": 3.414578437805176, + "learning_rate": 1.5996258185219832e-06, + "loss": 1.4002, + "step": 1708 + }, + { + "epoch": 2.157828282828283, + "grad_norm": 3.3196682929992676, + "learning_rate": 1.5972871842843782e-06, + "loss": 1.379, + "step": 1709 + }, + { + "epoch": 2.159090909090909, + "grad_norm": 3.2492880821228027, + "learning_rate": 1.5949485500467728e-06, + "loss": 1.3694, + "step": 1710 + }, + { + "epoch": 2.159090909090909, + "eval_loss": 1.6655367612838745, + "eval_runtime": 36.9249, + "eval_samples_per_second": 27.082, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 2.1603535353535355, + "grad_norm": 3.3484954833984375, + "learning_rate": 1.5926099158091677e-06, + "loss": 1.374, + "step": 1711 + }, + { + "epoch": 2.1616161616161618, + "grad_norm": 3.2919235229492188, + "learning_rate": 1.5902712815715623e-06, + "loss": 1.3524, + "step": 1712 + }, + { + "epoch": 2.162878787878788, + "grad_norm": 3.395267963409424, + "learning_rate": 1.5879326473339572e-06, + "loss": 1.3848, + "step": 1713 + }, + { + "epoch": 2.1641414141414144, + "grad_norm": 3.5136897563934326, + "learning_rate": 1.5855940130963518e-06, + "loss": 1.3465, + "step": 1714 + }, + { + "epoch": 2.16540404040404, + "grad_norm": 3.5640273094177246, + "learning_rate": 1.5832553788587464e-06, + "loss": 1.3924, + "step": 1715 + }, + { + "epoch": 2.1666666666666665, + "grad_norm": 3.761011838912964, + "learning_rate": 1.5809167446211413e-06, + "loss": 1.4401, + "step": 1716 + }, + { + "epoch": 2.167929292929293, + "grad_norm": 3.259582042694092, + "learning_rate": 1.5785781103835363e-06, + "loss": 1.3808, + "step": 1717 + }, + { + "epoch": 2.169191919191919, + "grad_norm": 3.365035057067871, + "learning_rate": 1.576239476145931e-06, + "loss": 1.3928, + "step": 1718 + }, + { + "epoch": 2.1704545454545454, + "grad_norm": 3.466689109802246, + "learning_rate": 1.5739008419083258e-06, + "loss": 1.3797, + "step": 1719 + }, + { + "epoch": 2.1717171717171717, + "grad_norm": 3.5278398990631104, + "learning_rate": 1.5715622076707204e-06, + "loss": 1.3752, + "step": 1720 + }, + { + "epoch": 2.1717171717171717, + "eval_loss": 1.6711663007736206, + "eval_runtime": 37.0568, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1720 + }, + { + "epoch": 2.172979797979798, + "grad_norm": 3.43217134475708, + "learning_rate": 1.569223573433115e-06, + "loss": 1.4222, + "step": 1721 + }, + { + "epoch": 2.1742424242424243, + "grad_norm": 3.295238733291626, + "learning_rate": 1.56688493919551e-06, + "loss": 1.3537, + "step": 1722 + }, + { + "epoch": 2.1755050505050506, + "grad_norm": 3.4373223781585693, + "learning_rate": 1.5645463049579045e-06, + "loss": 1.3813, + "step": 1723 + }, + { + "epoch": 2.176767676767677, + "grad_norm": 3.4872143268585205, + "learning_rate": 1.5622076707202996e-06, + "loss": 1.4018, + "step": 1724 + }, + { + "epoch": 2.178030303030303, + "grad_norm": 3.3276984691619873, + "learning_rate": 1.5598690364826944e-06, + "loss": 1.3644, + "step": 1725 + }, + { + "epoch": 2.179292929292929, + "grad_norm": 3.441479444503784, + "learning_rate": 1.557530402245089e-06, + "loss": 1.4105, + "step": 1726 + }, + { + "epoch": 2.1805555555555554, + "grad_norm": 3.3938825130462646, + "learning_rate": 1.5551917680074837e-06, + "loss": 1.3478, + "step": 1727 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.428677797317505, + "learning_rate": 1.5528531337698785e-06, + "loss": 1.3977, + "step": 1728 + }, + { + "epoch": 2.183080808080808, + "grad_norm": 3.3802578449249268, + "learning_rate": 1.5505144995322732e-06, + "loss": 1.3441, + "step": 1729 + }, + { + "epoch": 2.1843434343434343, + "grad_norm": 3.419437885284424, + "learning_rate": 1.548175865294668e-06, + "loss": 1.3811, + "step": 1730 + }, + { + "epoch": 2.1843434343434343, + "eval_loss": 1.669846773147583, + "eval_runtime": 37.0868, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1730 + }, + { + "epoch": 2.1856060606060606, + "grad_norm": 3.31225848197937, + "learning_rate": 1.5458372310570626e-06, + "loss": 1.3923, + "step": 1731 + }, + { + "epoch": 2.186868686868687, + "grad_norm": 3.5382823944091797, + "learning_rate": 1.5434985968194577e-06, + "loss": 1.4213, + "step": 1732 + }, + { + "epoch": 2.188131313131313, + "grad_norm": 3.207977533340454, + "learning_rate": 1.5411599625818523e-06, + "loss": 1.3175, + "step": 1733 + }, + { + "epoch": 2.1893939393939394, + "grad_norm": 3.3290693759918213, + "learning_rate": 1.5388213283442472e-06, + "loss": 1.4157, + "step": 1734 + }, + { + "epoch": 2.1906565656565657, + "grad_norm": 3.3432180881500244, + "learning_rate": 1.5364826941066418e-06, + "loss": 1.3844, + "step": 1735 + }, + { + "epoch": 2.191919191919192, + "grad_norm": 3.6149415969848633, + "learning_rate": 1.5341440598690366e-06, + "loss": 1.4243, + "step": 1736 + }, + { + "epoch": 2.1931818181818183, + "grad_norm": 3.3316938877105713, + "learning_rate": 1.5318054256314313e-06, + "loss": 1.372, + "step": 1737 + }, + { + "epoch": 2.1944444444444446, + "grad_norm": 3.4198312759399414, + "learning_rate": 1.5294667913938261e-06, + "loss": 1.3177, + "step": 1738 + }, + { + "epoch": 2.195707070707071, + "grad_norm": 3.450404167175293, + "learning_rate": 1.5271281571562207e-06, + "loss": 1.3596, + "step": 1739 + }, + { + "epoch": 2.196969696969697, + "grad_norm": 3.261406898498535, + "learning_rate": 1.5247895229186158e-06, + "loss": 1.3639, + "step": 1740 + }, + { + "epoch": 2.196969696969697, + "eval_loss": 1.6702104806900024, + "eval_runtime": 37.1009, + "eval_samples_per_second": 26.953, + "eval_steps_per_second": 1.132, + "step": 1740 + }, + { + "epoch": 2.198232323232323, + "grad_norm": 3.5527029037475586, + "learning_rate": 1.5224508886810104e-06, + "loss": 1.4129, + "step": 1741 + }, + { + "epoch": 2.1994949494949494, + "grad_norm": 3.498812437057495, + "learning_rate": 1.5201122544434053e-06, + "loss": 1.396, + "step": 1742 + }, + { + "epoch": 2.2007575757575757, + "grad_norm": 3.509518623352051, + "learning_rate": 1.5177736202058e-06, + "loss": 1.4506, + "step": 1743 + }, + { + "epoch": 2.202020202020202, + "grad_norm": 3.2482805252075195, + "learning_rate": 1.5154349859681948e-06, + "loss": 1.3793, + "step": 1744 + }, + { + "epoch": 2.2032828282828283, + "grad_norm": 3.299140214920044, + "learning_rate": 1.5130963517305894e-06, + "loss": 1.4011, + "step": 1745 + }, + { + "epoch": 2.2045454545454546, + "grad_norm": 3.4633231163024902, + "learning_rate": 1.510757717492984e-06, + "loss": 1.3904, + "step": 1746 + }, + { + "epoch": 2.205808080808081, + "grad_norm": 3.2724618911743164, + "learning_rate": 1.508419083255379e-06, + "loss": 1.3699, + "step": 1747 + }, + { + "epoch": 2.207070707070707, + "grad_norm": 3.361274242401123, + "learning_rate": 1.506080449017774e-06, + "loss": 1.3759, + "step": 1748 + }, + { + "epoch": 2.2083333333333335, + "grad_norm": 3.2770802974700928, + "learning_rate": 1.5037418147801685e-06, + "loss": 1.3497, + "step": 1749 + }, + { + "epoch": 2.20959595959596, + "grad_norm": 3.3818726539611816, + "learning_rate": 1.5014031805425634e-06, + "loss": 1.3277, + "step": 1750 + }, + { + "epoch": 2.20959595959596, + "eval_loss": 1.660167932510376, + "eval_runtime": 37.1049, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1750 + }, + { + "epoch": 2.2108585858585856, + "grad_norm": 3.296377420425415, + "learning_rate": 1.499064546304958e-06, + "loss": 1.3584, + "step": 1751 + }, + { + "epoch": 2.212121212121212, + "grad_norm": 3.4132192134857178, + "learning_rate": 1.4967259120673526e-06, + "loss": 1.367, + "step": 1752 + }, + { + "epoch": 2.2133838383838382, + "grad_norm": 3.3144986629486084, + "learning_rate": 1.4943872778297475e-06, + "loss": 1.3406, + "step": 1753 + }, + { + "epoch": 2.2146464646464645, + "grad_norm": 3.4283008575439453, + "learning_rate": 1.4920486435921421e-06, + "loss": 1.355, + "step": 1754 + }, + { + "epoch": 2.215909090909091, + "grad_norm": 3.319946050643921, + "learning_rate": 1.4897100093545372e-06, + "loss": 1.3665, + "step": 1755 + }, + { + "epoch": 2.217171717171717, + "grad_norm": 3.5585882663726807, + "learning_rate": 1.487371375116932e-06, + "loss": 1.4407, + "step": 1756 + }, + { + "epoch": 2.2184343434343434, + "grad_norm": 3.4216599464416504, + "learning_rate": 1.4850327408793267e-06, + "loss": 1.4105, + "step": 1757 + }, + { + "epoch": 2.2196969696969697, + "grad_norm": 3.3320260047912598, + "learning_rate": 1.4826941066417213e-06, + "loss": 1.3822, + "step": 1758 + }, + { + "epoch": 2.220959595959596, + "grad_norm": 3.2759056091308594, + "learning_rate": 1.4803554724041161e-06, + "loss": 1.4091, + "step": 1759 + }, + { + "epoch": 2.2222222222222223, + "grad_norm": 3.2865848541259766, + "learning_rate": 1.4780168381665108e-06, + "loss": 1.3643, + "step": 1760 + }, + { + "epoch": 2.2222222222222223, + "eval_loss": 1.662533164024353, + "eval_runtime": 37.1194, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1760 + }, + { + "epoch": 2.2234848484848486, + "grad_norm": 3.2485909461975098, + "learning_rate": 1.4756782039289056e-06, + "loss": 1.3986, + "step": 1761 + }, + { + "epoch": 2.224747474747475, + "grad_norm": 3.367898941040039, + "learning_rate": 1.4733395696913002e-06, + "loss": 1.37, + "step": 1762 + }, + { + "epoch": 2.226010101010101, + "grad_norm": 3.3919382095336914, + "learning_rate": 1.4710009354536953e-06, + "loss": 1.3517, + "step": 1763 + }, + { + "epoch": 2.227272727272727, + "grad_norm": 3.388627052307129, + "learning_rate": 1.46866230121609e-06, + "loss": 1.394, + "step": 1764 + }, + { + "epoch": 2.2285353535353534, + "grad_norm": 3.470121383666992, + "learning_rate": 1.4663236669784848e-06, + "loss": 1.3705, + "step": 1765 + }, + { + "epoch": 2.2297979797979797, + "grad_norm": 3.3515899181365967, + "learning_rate": 1.4639850327408794e-06, + "loss": 1.3736, + "step": 1766 + }, + { + "epoch": 2.231060606060606, + "grad_norm": 3.427877426147461, + "learning_rate": 1.4616463985032742e-06, + "loss": 1.3819, + "step": 1767 + }, + { + "epoch": 2.2323232323232323, + "grad_norm": 3.4387621879577637, + "learning_rate": 1.4593077642656689e-06, + "loss": 1.4043, + "step": 1768 + }, + { + "epoch": 2.2335858585858586, + "grad_norm": 3.4503068923950195, + "learning_rate": 1.4569691300280637e-06, + "loss": 1.3784, + "step": 1769 + }, + { + "epoch": 2.234848484848485, + "grad_norm": 3.5118091106414795, + "learning_rate": 1.4546304957904585e-06, + "loss": 1.319, + "step": 1770 + }, + { + "epoch": 2.234848484848485, + "eval_loss": 1.6638855934143066, + "eval_runtime": 37.1208, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 2.236111111111111, + "grad_norm": 3.3450958728790283, + "learning_rate": 1.4522918615528534e-06, + "loss": 1.3969, + "step": 1771 + }, + { + "epoch": 2.2373737373737375, + "grad_norm": 3.249265193939209, + "learning_rate": 1.449953227315248e-06, + "loss": 1.3543, + "step": 1772 + }, + { + "epoch": 2.2386363636363638, + "grad_norm": 3.33396315574646, + "learning_rate": 1.4476145930776429e-06, + "loss": 1.3595, + "step": 1773 + }, + { + "epoch": 2.23989898989899, + "grad_norm": 3.2275466918945312, + "learning_rate": 1.4452759588400375e-06, + "loss": 1.387, + "step": 1774 + }, + { + "epoch": 2.2411616161616164, + "grad_norm": 3.1885244846343994, + "learning_rate": 1.4429373246024323e-06, + "loss": 1.3815, + "step": 1775 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.41682767868042, + "learning_rate": 1.440598690364827e-06, + "loss": 1.398, + "step": 1776 + }, + { + "epoch": 2.2436868686868685, + "grad_norm": 3.3739681243896484, + "learning_rate": 1.4382600561272216e-06, + "loss": 1.4076, + "step": 1777 + }, + { + "epoch": 2.244949494949495, + "grad_norm": 3.3971354961395264, + "learning_rate": 1.4359214218896167e-06, + "loss": 1.3665, + "step": 1778 + }, + { + "epoch": 2.246212121212121, + "grad_norm": 3.375438690185547, + "learning_rate": 1.4335827876520115e-06, + "loss": 1.3841, + "step": 1779 + }, + { + "epoch": 2.2474747474747474, + "grad_norm": 3.4381961822509766, + "learning_rate": 1.4312441534144061e-06, + "loss": 1.3725, + "step": 1780 + }, + { + "epoch": 2.2474747474747474, + "eval_loss": 1.664459228515625, + "eval_runtime": 37.1364, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 1780 + }, + { + "epoch": 2.2487373737373737, + "grad_norm": 3.3988287448883057, + "learning_rate": 1.428905519176801e-06, + "loss": 1.3677, + "step": 1781 + }, + { + "epoch": 2.25, + "grad_norm": 3.3283984661102295, + "learning_rate": 1.4265668849391956e-06, + "loss": 1.4028, + "step": 1782 + }, + { + "epoch": 2.2512626262626263, + "grad_norm": 3.377519369125366, + "learning_rate": 1.4242282507015902e-06, + "loss": 1.4629, + "step": 1783 + }, + { + "epoch": 2.2525252525252526, + "grad_norm": 3.3736071586608887, + "learning_rate": 1.421889616463985e-06, + "loss": 1.3599, + "step": 1784 + }, + { + "epoch": 2.253787878787879, + "grad_norm": 3.4322469234466553, + "learning_rate": 1.4195509822263801e-06, + "loss": 1.3742, + "step": 1785 + }, + { + "epoch": 2.255050505050505, + "grad_norm": 3.2117607593536377, + "learning_rate": 1.4172123479887748e-06, + "loss": 1.3642, + "step": 1786 + }, + { + "epoch": 2.2563131313131315, + "grad_norm": 3.3926846981048584, + "learning_rate": 1.4148737137511694e-06, + "loss": 1.3988, + "step": 1787 + }, + { + "epoch": 2.257575757575758, + "grad_norm": 3.371910810470581, + "learning_rate": 1.4125350795135642e-06, + "loss": 1.3552, + "step": 1788 + }, + { + "epoch": 2.2588383838383836, + "grad_norm": 3.447720766067505, + "learning_rate": 1.4101964452759589e-06, + "loss": 1.3906, + "step": 1789 + }, + { + "epoch": 2.26010101010101, + "grad_norm": 3.396575450897217, + "learning_rate": 1.4078578110383537e-06, + "loss": 1.3641, + "step": 1790 + }, + { + "epoch": 2.26010101010101, + "eval_loss": 1.6667912006378174, + "eval_runtime": 37.3275, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 1790 + }, + { + "epoch": 2.2613636363636362, + "grad_norm": 3.532827138900757, + "learning_rate": 1.4055191768007483e-06, + "loss": 1.4078, + "step": 1791 + }, + { + "epoch": 2.2626262626262625, + "grad_norm": 3.438328981399536, + "learning_rate": 1.4031805425631432e-06, + "loss": 1.3589, + "step": 1792 + }, + { + "epoch": 2.263888888888889, + "grad_norm": 3.3978145122528076, + "learning_rate": 1.400841908325538e-06, + "loss": 1.4299, + "step": 1793 + }, + { + "epoch": 2.265151515151515, + "grad_norm": 3.457463502883911, + "learning_rate": 1.3985032740879329e-06, + "loss": 1.3814, + "step": 1794 + }, + { + "epoch": 2.2664141414141414, + "grad_norm": 3.2421817779541016, + "learning_rate": 1.3961646398503275e-06, + "loss": 1.3732, + "step": 1795 + }, + { + "epoch": 2.2676767676767677, + "grad_norm": 3.420693874359131, + "learning_rate": 1.3938260056127223e-06, + "loss": 1.431, + "step": 1796 + }, + { + "epoch": 2.268939393939394, + "grad_norm": 3.3793931007385254, + "learning_rate": 1.391487371375117e-06, + "loss": 1.3338, + "step": 1797 + }, + { + "epoch": 2.2702020202020203, + "grad_norm": 3.4106879234313965, + "learning_rate": 1.3891487371375118e-06, + "loss": 1.366, + "step": 1798 + }, + { + "epoch": 2.2714646464646466, + "grad_norm": 3.391782760620117, + "learning_rate": 1.3868101028999065e-06, + "loss": 1.3898, + "step": 1799 + }, + { + "epoch": 2.2727272727272725, + "grad_norm": 3.329619884490967, + "learning_rate": 1.3844714686623013e-06, + "loss": 1.3494, + "step": 1800 + }, + { + "epoch": 2.2727272727272725, + "eval_loss": 1.6635981798171997, + "eval_runtime": 37.4748, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 2.273989898989899, + "grad_norm": 3.383571147918701, + "learning_rate": 1.3821328344246961e-06, + "loss": 1.3603, + "step": 1801 + }, + { + "epoch": 2.275252525252525, + "grad_norm": 3.4224114418029785, + "learning_rate": 1.379794200187091e-06, + "loss": 1.4003, + "step": 1802 + }, + { + "epoch": 2.2765151515151514, + "grad_norm": 3.4758248329162598, + "learning_rate": 1.3774555659494856e-06, + "loss": 1.357, + "step": 1803 + }, + { + "epoch": 2.2777777777777777, + "grad_norm": 3.410489320755005, + "learning_rate": 1.3751169317118805e-06, + "loss": 1.3507, + "step": 1804 + }, + { + "epoch": 2.279040404040404, + "grad_norm": 3.4158382415771484, + "learning_rate": 1.372778297474275e-06, + "loss": 1.3822, + "step": 1805 + }, + { + "epoch": 2.2803030303030303, + "grad_norm": 3.562300443649292, + "learning_rate": 1.37043966323667e-06, + "loss": 1.3898, + "step": 1806 + }, + { + "epoch": 2.2815656565656566, + "grad_norm": 3.5485477447509766, + "learning_rate": 1.3681010289990646e-06, + "loss": 1.3341, + "step": 1807 + }, + { + "epoch": 2.282828282828283, + "grad_norm": 3.624049186706543, + "learning_rate": 1.3657623947614596e-06, + "loss": 1.4034, + "step": 1808 + }, + { + "epoch": 2.284090909090909, + "grad_norm": 3.37754225730896, + "learning_rate": 1.3634237605238542e-06, + "loss": 1.4328, + "step": 1809 + }, + { + "epoch": 2.2853535353535355, + "grad_norm": 3.5212652683258057, + "learning_rate": 1.361085126286249e-06, + "loss": 1.4038, + "step": 1810 + }, + { + "epoch": 2.2853535353535355, + "eval_loss": 1.6660585403442383, + "eval_runtime": 36.944, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 2.2866161616161618, + "grad_norm": 3.5564818382263184, + "learning_rate": 1.3587464920486437e-06, + "loss": 1.4132, + "step": 1811 + }, + { + "epoch": 2.287878787878788, + "grad_norm": 3.613368272781372, + "learning_rate": 1.3564078578110384e-06, + "loss": 1.4147, + "step": 1812 + }, + { + "epoch": 2.2891414141414144, + "grad_norm": 3.3956761360168457, + "learning_rate": 1.3540692235734332e-06, + "loss": 1.3691, + "step": 1813 + }, + { + "epoch": 2.29040404040404, + "grad_norm": 3.4510204792022705, + "learning_rate": 1.3517305893358278e-06, + "loss": 1.4152, + "step": 1814 + }, + { + "epoch": 2.2916666666666665, + "grad_norm": 3.475883960723877, + "learning_rate": 1.3493919550982227e-06, + "loss": 1.3786, + "step": 1815 + }, + { + "epoch": 2.292929292929293, + "grad_norm": 3.4566831588745117, + "learning_rate": 1.3470533208606177e-06, + "loss": 1.3737, + "step": 1816 + }, + { + "epoch": 2.294191919191919, + "grad_norm": 3.2443597316741943, + "learning_rate": 1.3447146866230124e-06, + "loss": 1.3435, + "step": 1817 + }, + { + "epoch": 2.2954545454545454, + "grad_norm": 3.4844541549682617, + "learning_rate": 1.342376052385407e-06, + "loss": 1.3759, + "step": 1818 + }, + { + "epoch": 2.2967171717171717, + "grad_norm": 3.3630642890930176, + "learning_rate": 1.3400374181478018e-06, + "loss": 1.363, + "step": 1819 + }, + { + "epoch": 2.297979797979798, + "grad_norm": 3.382788896560669, + "learning_rate": 1.3376987839101965e-06, + "loss": 1.4096, + "step": 1820 + }, + { + "epoch": 2.297979797979798, + "eval_loss": 1.664247751235962, + "eval_runtime": 37.0799, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1820 + }, + { + "epoch": 2.2992424242424243, + "grad_norm": 3.287827491760254, + "learning_rate": 1.3353601496725913e-06, + "loss": 1.348, + "step": 1821 + }, + { + "epoch": 2.3005050505050506, + "grad_norm": 3.388911008834839, + "learning_rate": 1.333021515434986e-06, + "loss": 1.3901, + "step": 1822 + }, + { + "epoch": 2.301767676767677, + "grad_norm": 3.3604469299316406, + "learning_rate": 1.3306828811973808e-06, + "loss": 1.3533, + "step": 1823 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.4020297527313232, + "learning_rate": 1.3283442469597756e-06, + "loss": 1.3469, + "step": 1824 + }, + { + "epoch": 2.304292929292929, + "grad_norm": 3.449728488922119, + "learning_rate": 1.3260056127221705e-06, + "loss": 1.3622, + "step": 1825 + }, + { + "epoch": 2.3055555555555554, + "grad_norm": 3.531832218170166, + "learning_rate": 1.323666978484565e-06, + "loss": 1.354, + "step": 1826 + }, + { + "epoch": 2.3068181818181817, + "grad_norm": 3.3406412601470947, + "learning_rate": 1.32132834424696e-06, + "loss": 1.3699, + "step": 1827 + }, + { + "epoch": 2.308080808080808, + "grad_norm": 3.5115885734558105, + "learning_rate": 1.3189897100093546e-06, + "loss": 1.4186, + "step": 1828 + }, + { + "epoch": 2.3093434343434343, + "grad_norm": 3.5668814182281494, + "learning_rate": 1.3166510757717494e-06, + "loss": 1.3894, + "step": 1829 + }, + { + "epoch": 2.3106060606060606, + "grad_norm": 3.5255587100982666, + "learning_rate": 1.314312441534144e-06, + "loss": 1.3567, + "step": 1830 + }, + { + "epoch": 2.3106060606060606, + "eval_loss": 1.6662371158599854, + "eval_runtime": 37.1254, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 1830 + }, + { + "epoch": 2.311868686868687, + "grad_norm": 3.4452052116394043, + "learning_rate": 1.311973807296539e-06, + "loss": 1.331, + "step": 1831 + }, + { + "epoch": 2.313131313131313, + "grad_norm": 3.466440439224243, + "learning_rate": 1.3096351730589337e-06, + "loss": 1.4204, + "step": 1832 + }, + { + "epoch": 2.3143939393939394, + "grad_norm": 3.2339625358581543, + "learning_rate": 1.3072965388213286e-06, + "loss": 1.3663, + "step": 1833 + }, + { + "epoch": 2.3156565656565657, + "grad_norm": 3.4634203910827637, + "learning_rate": 1.3049579045837232e-06, + "loss": 1.3805, + "step": 1834 + }, + { + "epoch": 2.316919191919192, + "grad_norm": 3.4900529384613037, + "learning_rate": 1.302619270346118e-06, + "loss": 1.3851, + "step": 1835 + }, + { + "epoch": 2.3181818181818183, + "grad_norm": 3.314610242843628, + "learning_rate": 1.3002806361085127e-06, + "loss": 1.3971, + "step": 1836 + }, + { + "epoch": 2.3194444444444446, + "grad_norm": 3.3920958042144775, + "learning_rate": 1.2979420018709073e-06, + "loss": 1.404, + "step": 1837 + }, + { + "epoch": 2.320707070707071, + "grad_norm": 3.3682339191436768, + "learning_rate": 1.2956033676333021e-06, + "loss": 1.3803, + "step": 1838 + }, + { + "epoch": 2.321969696969697, + "grad_norm": 3.332442283630371, + "learning_rate": 1.2932647333956972e-06, + "loss": 1.4214, + "step": 1839 + }, + { + "epoch": 2.323232323232323, + "grad_norm": 3.3797237873077393, + "learning_rate": 1.2909260991580918e-06, + "loss": 1.3323, + "step": 1840 + }, + { + "epoch": 2.323232323232323, + "eval_loss": 1.664433479309082, + "eval_runtime": 37.1313, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 2.3244949494949494, + "grad_norm": 3.454676866531372, + "learning_rate": 1.2885874649204867e-06, + "loss": 1.4014, + "step": 1841 + }, + { + "epoch": 2.3257575757575757, + "grad_norm": 3.4918594360351562, + "learning_rate": 1.2862488306828813e-06, + "loss": 1.3926, + "step": 1842 + }, + { + "epoch": 2.327020202020202, + "grad_norm": 3.291851043701172, + "learning_rate": 1.283910196445276e-06, + "loss": 1.3957, + "step": 1843 + }, + { + "epoch": 2.3282828282828283, + "grad_norm": 3.308403491973877, + "learning_rate": 1.2815715622076708e-06, + "loss": 1.3866, + "step": 1844 + }, + { + "epoch": 2.3295454545454546, + "grad_norm": 3.4691977500915527, + "learning_rate": 1.2792329279700654e-06, + "loss": 1.3944, + "step": 1845 + }, + { + "epoch": 2.330808080808081, + "grad_norm": 3.3912553787231445, + "learning_rate": 1.2768942937324603e-06, + "loss": 1.3521, + "step": 1846 + }, + { + "epoch": 2.332070707070707, + "grad_norm": 3.399017810821533, + "learning_rate": 1.2745556594948553e-06, + "loss": 1.3798, + "step": 1847 + }, + { + "epoch": 2.3333333333333335, + "grad_norm": 3.4829556941986084, + "learning_rate": 1.27221702525725e-06, + "loss": 1.3854, + "step": 1848 + }, + { + "epoch": 2.33459595959596, + "grad_norm": 3.363145351409912, + "learning_rate": 1.2698783910196446e-06, + "loss": 1.3581, + "step": 1849 + }, + { + "epoch": 2.3358585858585856, + "grad_norm": 3.3383405208587646, + "learning_rate": 1.2675397567820394e-06, + "loss": 1.3314, + "step": 1850 + }, + { + "epoch": 2.3358585858585856, + "eval_loss": 1.6598613262176514, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1850 + }, + { + "epoch": 2.337121212121212, + "grad_norm": 3.54443359375, + "learning_rate": 1.265201122544434e-06, + "loss": 1.3831, + "step": 1851 + }, + { + "epoch": 2.3383838383838382, + "grad_norm": 3.373283624649048, + "learning_rate": 1.2628624883068289e-06, + "loss": 1.4073, + "step": 1852 + }, + { + "epoch": 2.3396464646464645, + "grad_norm": 3.5655102729797363, + "learning_rate": 1.2605238540692235e-06, + "loss": 1.4081, + "step": 1853 + }, + { + "epoch": 2.340909090909091, + "grad_norm": 3.5852856636047363, + "learning_rate": 1.2581852198316186e-06, + "loss": 1.4566, + "step": 1854 + }, + { + "epoch": 2.342171717171717, + "grad_norm": 3.370293378829956, + "learning_rate": 1.2558465855940132e-06, + "loss": 1.3959, + "step": 1855 + }, + { + "epoch": 2.3434343434343434, + "grad_norm": 3.556837558746338, + "learning_rate": 1.253507951356408e-06, + "loss": 1.3989, + "step": 1856 + }, + { + "epoch": 2.3446969696969697, + "grad_norm": 3.3351283073425293, + "learning_rate": 1.2511693171188027e-06, + "loss": 1.3425, + "step": 1857 + }, + { + "epoch": 2.345959595959596, + "grad_norm": 3.432695150375366, + "learning_rate": 1.2488306828811975e-06, + "loss": 1.3849, + "step": 1858 + }, + { + "epoch": 2.3472222222222223, + "grad_norm": 3.6309726238250732, + "learning_rate": 1.2464920486435922e-06, + "loss": 1.4181, + "step": 1859 + }, + { + "epoch": 2.3484848484848486, + "grad_norm": 3.465672016143799, + "learning_rate": 1.244153414405987e-06, + "loss": 1.3992, + "step": 1860 + }, + { + "epoch": 2.3484848484848486, + "eval_loss": 1.6598951816558838, + "eval_runtime": 37.1338, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 1860 + }, + { + "epoch": 2.349747474747475, + "grad_norm": 3.333996534347534, + "learning_rate": 1.2418147801683818e-06, + "loss": 1.3765, + "step": 1861 + }, + { + "epoch": 2.351010101010101, + "grad_norm": 3.3039495944976807, + "learning_rate": 1.2394761459307765e-06, + "loss": 1.4055, + "step": 1862 + }, + { + "epoch": 2.3522727272727275, + "grad_norm": 3.42413067817688, + "learning_rate": 1.2371375116931713e-06, + "loss": 1.4089, + "step": 1863 + }, + { + "epoch": 2.3535353535353534, + "grad_norm": 3.626856565475464, + "learning_rate": 1.2347988774555662e-06, + "loss": 1.3382, + "step": 1864 + }, + { + "epoch": 2.3547979797979797, + "grad_norm": 3.677388906478882, + "learning_rate": 1.2324602432179608e-06, + "loss": 1.4251, + "step": 1865 + }, + { + "epoch": 2.356060606060606, + "grad_norm": 3.7962775230407715, + "learning_rate": 1.2301216089803556e-06, + "loss": 1.3808, + "step": 1866 + }, + { + "epoch": 2.3573232323232323, + "grad_norm": 3.4381752014160156, + "learning_rate": 1.2277829747427503e-06, + "loss": 1.3561, + "step": 1867 + }, + { + "epoch": 2.3585858585858586, + "grad_norm": 3.4678566455841064, + "learning_rate": 1.225444340505145e-06, + "loss": 1.3542, + "step": 1868 + }, + { + "epoch": 2.359848484848485, + "grad_norm": 3.4447884559631348, + "learning_rate": 1.22310570626754e-06, + "loss": 1.3809, + "step": 1869 + }, + { + "epoch": 2.361111111111111, + "grad_norm": 3.3679399490356445, + "learning_rate": 1.2207670720299346e-06, + "loss": 1.3825, + "step": 1870 + }, + { + "epoch": 2.361111111111111, + "eval_loss": 1.6562567949295044, + "eval_runtime": 37.1451, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 1870 + }, + { + "epoch": 2.3623737373737375, + "grad_norm": 3.3724935054779053, + "learning_rate": 1.2184284377923294e-06, + "loss": 1.3394, + "step": 1871 + }, + { + "epoch": 2.3636363636363638, + "grad_norm": 3.426579475402832, + "learning_rate": 1.2160898035547243e-06, + "loss": 1.4326, + "step": 1872 + }, + { + "epoch": 2.36489898989899, + "grad_norm": 3.4336345195770264, + "learning_rate": 1.213751169317119e-06, + "loss": 1.3575, + "step": 1873 + }, + { + "epoch": 2.3661616161616164, + "grad_norm": 3.4542808532714844, + "learning_rate": 1.2114125350795135e-06, + "loss": 1.3449, + "step": 1874 + }, + { + "epoch": 2.367424242424242, + "grad_norm": 3.362525701522827, + "learning_rate": 1.2090739008419086e-06, + "loss": 1.3964, + "step": 1875 + }, + { + "epoch": 2.3686868686868685, + "grad_norm": 3.392415761947632, + "learning_rate": 1.2067352666043032e-06, + "loss": 1.4272, + "step": 1876 + }, + { + "epoch": 2.369949494949495, + "grad_norm": 3.4062328338623047, + "learning_rate": 1.2043966323666978e-06, + "loss": 1.3407, + "step": 1877 + }, + { + "epoch": 2.371212121212121, + "grad_norm": 3.5435354709625244, + "learning_rate": 1.2020579981290927e-06, + "loss": 1.3671, + "step": 1878 + }, + { + "epoch": 2.3724747474747474, + "grad_norm": 3.743882656097412, + "learning_rate": 1.1997193638914875e-06, + "loss": 1.4046, + "step": 1879 + }, + { + "epoch": 2.3737373737373737, + "grad_norm": 3.752319574356079, + "learning_rate": 1.1973807296538822e-06, + "loss": 1.3631, + "step": 1880 + }, + { + "epoch": 2.3737373737373737, + "eval_loss": 1.6618274450302124, + "eval_runtime": 37.1614, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1880 + }, + { + "epoch": 2.375, + "grad_norm": 3.57468318939209, + "learning_rate": 1.195042095416277e-06, + "loss": 1.3584, + "step": 1881 + }, + { + "epoch": 2.3762626262626263, + "grad_norm": 3.544557809829712, + "learning_rate": 1.1927034611786716e-06, + "loss": 1.3759, + "step": 1882 + }, + { + "epoch": 2.3775252525252526, + "grad_norm": 3.313106060028076, + "learning_rate": 1.1903648269410665e-06, + "loss": 1.36, + "step": 1883 + }, + { + "epoch": 2.378787878787879, + "grad_norm": 3.347764730453491, + "learning_rate": 1.1880261927034613e-06, + "loss": 1.3789, + "step": 1884 + }, + { + "epoch": 2.380050505050505, + "grad_norm": 3.260697364807129, + "learning_rate": 1.185687558465856e-06, + "loss": 1.3456, + "step": 1885 + }, + { + "epoch": 2.3813131313131315, + "grad_norm": 3.287977695465088, + "learning_rate": 1.1833489242282508e-06, + "loss": 1.3739, + "step": 1886 + }, + { + "epoch": 2.382575757575758, + "grad_norm": 3.27716064453125, + "learning_rate": 1.1810102899906456e-06, + "loss": 1.3792, + "step": 1887 + }, + { + "epoch": 2.3838383838383836, + "grad_norm": 3.4265832901000977, + "learning_rate": 1.1786716557530403e-06, + "loss": 1.3604, + "step": 1888 + }, + { + "epoch": 2.38510101010101, + "grad_norm": 3.5619351863861084, + "learning_rate": 1.1763330215154351e-06, + "loss": 1.3786, + "step": 1889 + }, + { + "epoch": 2.3863636363636362, + "grad_norm": 3.482348680496216, + "learning_rate": 1.17399438727783e-06, + "loss": 1.415, + "step": 1890 + }, + { + "epoch": 2.3863636363636362, + "eval_loss": 1.6625829935073853, + "eval_runtime": 37.2791, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 2.3876262626262625, + "grad_norm": 3.375915288925171, + "learning_rate": 1.1716557530402246e-06, + "loss": 1.335, + "step": 1891 + }, + { + "epoch": 2.388888888888889, + "grad_norm": 3.5826265811920166, + "learning_rate": 1.1693171188026194e-06, + "loss": 1.3612, + "step": 1892 + }, + { + "epoch": 2.390151515151515, + "grad_norm": 3.787109851837158, + "learning_rate": 1.166978484565014e-06, + "loss": 1.386, + "step": 1893 + }, + { + "epoch": 2.3914141414141414, + "grad_norm": 3.5518991947174072, + "learning_rate": 1.164639850327409e-06, + "loss": 1.3372, + "step": 1894 + }, + { + "epoch": 2.3926767676767677, + "grad_norm": 3.4609735012054443, + "learning_rate": 1.1623012160898037e-06, + "loss": 1.362, + "step": 1895 + }, + { + "epoch": 2.393939393939394, + "grad_norm": 3.578777551651001, + "learning_rate": 1.1599625818521984e-06, + "loss": 1.3754, + "step": 1896 + }, + { + "epoch": 2.3952020202020203, + "grad_norm": 3.5557262897491455, + "learning_rate": 1.1576239476145932e-06, + "loss": 1.4048, + "step": 1897 + }, + { + "epoch": 2.3964646464646466, + "grad_norm": 3.3413872718811035, + "learning_rate": 1.155285313376988e-06, + "loss": 1.3531, + "step": 1898 + }, + { + "epoch": 2.3977272727272725, + "grad_norm": 3.207866668701172, + "learning_rate": 1.1529466791393827e-06, + "loss": 1.3479, + "step": 1899 + }, + { + "epoch": 2.398989898989899, + "grad_norm": 3.3909525871276855, + "learning_rate": 1.1506080449017775e-06, + "loss": 1.3663, + "step": 1900 + }, + { + "epoch": 2.398989898989899, + "eval_loss": 1.661224126815796, + "eval_runtime": 37.4772, + "eval_samples_per_second": 26.683, + "eval_steps_per_second": 1.121, + "step": 1900 + }, + { + "epoch": 2.400252525252525, + "grad_norm": 3.4364285469055176, + "learning_rate": 1.1482694106641722e-06, + "loss": 1.3541, + "step": 1901 + }, + { + "epoch": 2.4015151515151514, + "grad_norm": 3.5648012161254883, + "learning_rate": 1.145930776426567e-06, + "loss": 1.3976, + "step": 1902 + }, + { + "epoch": 2.4027777777777777, + "grad_norm": 3.4142239093780518, + "learning_rate": 1.1435921421889619e-06, + "loss": 1.4217, + "step": 1903 + }, + { + "epoch": 2.404040404040404, + "grad_norm": 3.415621757507324, + "learning_rate": 1.1412535079513565e-06, + "loss": 1.4152, + "step": 1904 + }, + { + "epoch": 2.4053030303030303, + "grad_norm": 3.401431083679199, + "learning_rate": 1.1389148737137511e-06, + "loss": 1.3588, + "step": 1905 + }, + { + "epoch": 2.4065656565656566, + "grad_norm": 3.528501272201538, + "learning_rate": 1.1365762394761462e-06, + "loss": 1.4113, + "step": 1906 + }, + { + "epoch": 2.407828282828283, + "grad_norm": 3.5070385932922363, + "learning_rate": 1.1342376052385408e-06, + "loss": 1.4051, + "step": 1907 + }, + { + "epoch": 2.409090909090909, + "grad_norm": 3.2571558952331543, + "learning_rate": 1.1318989710009354e-06, + "loss": 1.3471, + "step": 1908 + }, + { + "epoch": 2.4103535353535355, + "grad_norm": 3.4780328273773193, + "learning_rate": 1.1295603367633303e-06, + "loss": 1.3895, + "step": 1909 + }, + { + "epoch": 2.4116161616161618, + "grad_norm": 3.431661367416382, + "learning_rate": 1.1272217025257251e-06, + "loss": 1.366, + "step": 1910 + }, + { + "epoch": 2.4116161616161618, + "eval_loss": 1.663731575012207, + "eval_runtime": 36.9118, + "eval_samples_per_second": 27.092, + "eval_steps_per_second": 1.138, + "step": 1910 + }, + { + "epoch": 2.412878787878788, + "grad_norm": 3.4981627464294434, + "learning_rate": 1.1248830682881197e-06, + "loss": 1.3476, + "step": 1911 + }, + { + "epoch": 2.4141414141414144, + "grad_norm": 3.1849045753479004, + "learning_rate": 1.1225444340505146e-06, + "loss": 1.32, + "step": 1912 + }, + { + "epoch": 2.41540404040404, + "grad_norm": 3.4398815631866455, + "learning_rate": 1.1202057998129094e-06, + "loss": 1.4065, + "step": 1913 + }, + { + "epoch": 2.4166666666666665, + "grad_norm": 3.5676138401031494, + "learning_rate": 1.117867165575304e-06, + "loss": 1.3973, + "step": 1914 + }, + { + "epoch": 2.417929292929293, + "grad_norm": 3.5278069972991943, + "learning_rate": 1.115528531337699e-06, + "loss": 1.3802, + "step": 1915 + }, + { + "epoch": 2.419191919191919, + "grad_norm": 3.4156792163848877, + "learning_rate": 1.1131898971000935e-06, + "loss": 1.3605, + "step": 1916 + }, + { + "epoch": 2.4204545454545454, + "grad_norm": 3.3457775115966797, + "learning_rate": 1.1108512628624884e-06, + "loss": 1.3688, + "step": 1917 + }, + { + "epoch": 2.4217171717171717, + "grad_norm": 3.3902130126953125, + "learning_rate": 1.1085126286248832e-06, + "loss": 1.3829, + "step": 1918 + }, + { + "epoch": 2.422979797979798, + "grad_norm": 3.361523389816284, + "learning_rate": 1.1061739943872779e-06, + "loss": 1.3495, + "step": 1919 + }, + { + "epoch": 2.4242424242424243, + "grad_norm": 3.2795472145080566, + "learning_rate": 1.1038353601496727e-06, + "loss": 1.3165, + "step": 1920 + }, + { + "epoch": 2.4242424242424243, + "eval_loss": 1.6594749689102173, + "eval_runtime": 37.0393, + "eval_samples_per_second": 26.998, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 2.4255050505050506, + "grad_norm": 3.5426580905914307, + "learning_rate": 1.1014967259120675e-06, + "loss": 1.403, + "step": 1921 + }, + { + "epoch": 2.426767676767677, + "grad_norm": 3.329042434692383, + "learning_rate": 1.0991580916744622e-06, + "loss": 1.3731, + "step": 1922 + }, + { + "epoch": 2.428030303030303, + "grad_norm": 3.419233798980713, + "learning_rate": 1.096819457436857e-06, + "loss": 1.3784, + "step": 1923 + }, + { + "epoch": 2.429292929292929, + "grad_norm": 3.6780271530151367, + "learning_rate": 1.0944808231992516e-06, + "loss": 1.4061, + "step": 1924 + }, + { + "epoch": 2.4305555555555554, + "grad_norm": 3.790707588195801, + "learning_rate": 1.0921421889616465e-06, + "loss": 1.4125, + "step": 1925 + }, + { + "epoch": 2.4318181818181817, + "grad_norm": 3.4600441455841064, + "learning_rate": 1.0898035547240413e-06, + "loss": 1.4084, + "step": 1926 + }, + { + "epoch": 2.433080808080808, + "grad_norm": 3.502103805541992, + "learning_rate": 1.087464920486436e-06, + "loss": 1.3882, + "step": 1927 + }, + { + "epoch": 2.4343434343434343, + "grad_norm": 3.3180925846099854, + "learning_rate": 1.0851262862488308e-06, + "loss": 1.3669, + "step": 1928 + }, + { + "epoch": 2.4356060606060606, + "grad_norm": 3.447608709335327, + "learning_rate": 1.0827876520112257e-06, + "loss": 1.2922, + "step": 1929 + }, + { + "epoch": 2.436868686868687, + "grad_norm": 3.4833571910858154, + "learning_rate": 1.0804490177736203e-06, + "loss": 1.3754, + "step": 1930 + }, + { + "epoch": 2.436868686868687, + "eval_loss": 1.665055751800537, + "eval_runtime": 37.0614, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1930 + }, + { + "epoch": 2.438131313131313, + "grad_norm": 3.426940679550171, + "learning_rate": 1.0781103835360151e-06, + "loss": 1.3659, + "step": 1931 + }, + { + "epoch": 2.4393939393939394, + "grad_norm": 3.4161434173583984, + "learning_rate": 1.07577174929841e-06, + "loss": 1.4161, + "step": 1932 + }, + { + "epoch": 2.4406565656565657, + "grad_norm": 3.5886597633361816, + "learning_rate": 1.0734331150608046e-06, + "loss": 1.448, + "step": 1933 + }, + { + "epoch": 2.441919191919192, + "grad_norm": 3.4271745681762695, + "learning_rate": 1.0710944808231992e-06, + "loss": 1.3608, + "step": 1934 + }, + { + "epoch": 2.4431818181818183, + "grad_norm": 3.357823133468628, + "learning_rate": 1.068755846585594e-06, + "loss": 1.3477, + "step": 1935 + }, + { + "epoch": 2.4444444444444446, + "grad_norm": 3.604701519012451, + "learning_rate": 1.066417212347989e-06, + "loss": 1.4402, + "step": 1936 + }, + { + "epoch": 2.445707070707071, + "grad_norm": 3.3494462966918945, + "learning_rate": 1.0640785781103835e-06, + "loss": 1.3878, + "step": 1937 + }, + { + "epoch": 2.446969696969697, + "grad_norm": 3.3025968074798584, + "learning_rate": 1.0617399438727784e-06, + "loss": 1.3671, + "step": 1938 + }, + { + "epoch": 2.448232323232323, + "grad_norm": 3.430521249771118, + "learning_rate": 1.059401309635173e-06, + "loss": 1.3795, + "step": 1939 + }, + { + "epoch": 2.4494949494949494, + "grad_norm": 3.5415585041046143, + "learning_rate": 1.0570626753975679e-06, + "loss": 1.3867, + "step": 1940 + }, + { + "epoch": 2.4494949494949494, + "eval_loss": 1.6607574224472046, + "eval_runtime": 37.0458, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1940 + }, + { + "epoch": 2.4507575757575757, + "grad_norm": 3.527456521987915, + "learning_rate": 1.0547240411599627e-06, + "loss": 1.3641, + "step": 1941 + }, + { + "epoch": 2.452020202020202, + "grad_norm": 3.5055441856384277, + "learning_rate": 1.0523854069223573e-06, + "loss": 1.3905, + "step": 1942 + }, + { + "epoch": 2.4532828282828283, + "grad_norm": 3.512960910797119, + "learning_rate": 1.0500467726847522e-06, + "loss": 1.3824, + "step": 1943 + }, + { + "epoch": 2.4545454545454546, + "grad_norm": 3.4122636318206787, + "learning_rate": 1.047708138447147e-06, + "loss": 1.3857, + "step": 1944 + }, + { + "epoch": 2.455808080808081, + "grad_norm": 3.665520668029785, + "learning_rate": 1.0453695042095417e-06, + "loss": 1.3866, + "step": 1945 + }, + { + "epoch": 2.457070707070707, + "grad_norm": 3.7062342166900635, + "learning_rate": 1.0430308699719365e-06, + "loss": 1.3968, + "step": 1946 + }, + { + "epoch": 2.4583333333333335, + "grad_norm": 3.558770179748535, + "learning_rate": 1.0406922357343311e-06, + "loss": 1.4005, + "step": 1947 + }, + { + "epoch": 2.45959595959596, + "grad_norm": 3.3678979873657227, + "learning_rate": 1.038353601496726e-06, + "loss": 1.3258, + "step": 1948 + }, + { + "epoch": 2.4608585858585856, + "grad_norm": 3.4766786098480225, + "learning_rate": 1.0360149672591208e-06, + "loss": 1.3634, + "step": 1949 + }, + { + "epoch": 2.462121212121212, + "grad_norm": 3.3427937030792236, + "learning_rate": 1.0336763330215154e-06, + "loss": 1.3419, + "step": 1950 + }, + { + "epoch": 2.462121212121212, + "eval_loss": 1.660905361175537, + "eval_runtime": 37.0483, + "eval_samples_per_second": 26.992, + "eval_steps_per_second": 1.134, + "step": 1950 + }, + { + "epoch": 2.4633838383838382, + "grad_norm": 3.436858654022217, + "learning_rate": 1.0313376987839103e-06, + "loss": 1.377, + "step": 1951 + }, + { + "epoch": 2.4646464646464645, + "grad_norm": 3.4107327461242676, + "learning_rate": 1.0289990645463051e-06, + "loss": 1.3484, + "step": 1952 + }, + { + "epoch": 2.465909090909091, + "grad_norm": 3.3843321800231934, + "learning_rate": 1.0266604303086998e-06, + "loss": 1.3373, + "step": 1953 + }, + { + "epoch": 2.467171717171717, + "grad_norm": 3.368032455444336, + "learning_rate": 1.0243217960710946e-06, + "loss": 1.3245, + "step": 1954 + }, + { + "epoch": 2.4684343434343434, + "grad_norm": 3.3483028411865234, + "learning_rate": 1.0219831618334894e-06, + "loss": 1.3498, + "step": 1955 + }, + { + "epoch": 2.4696969696969697, + "grad_norm": 3.5274465084075928, + "learning_rate": 1.019644527595884e-06, + "loss": 1.4074, + "step": 1956 + }, + { + "epoch": 2.470959595959596, + "grad_norm": 3.4556262493133545, + "learning_rate": 1.017305893358279e-06, + "loss": 1.3778, + "step": 1957 + }, + { + "epoch": 2.4722222222222223, + "grad_norm": 3.4345815181732178, + "learning_rate": 1.0149672591206736e-06, + "loss": 1.3847, + "step": 1958 + }, + { + "epoch": 2.4734848484848486, + "grad_norm": 3.437321662902832, + "learning_rate": 1.0126286248830684e-06, + "loss": 1.3784, + "step": 1959 + }, + { + "epoch": 2.474747474747475, + "grad_norm": 3.4794695377349854, + "learning_rate": 1.0102899906454632e-06, + "loss": 1.3646, + "step": 1960 + }, + { + "epoch": 2.474747474747475, + "eval_loss": 1.6639539003372192, + "eval_runtime": 37.0499, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1960 + }, + { + "epoch": 2.476010101010101, + "grad_norm": 3.457455635070801, + "learning_rate": 1.0079513564078579e-06, + "loss": 1.3654, + "step": 1961 + }, + { + "epoch": 2.4772727272727275, + "grad_norm": 3.432436227798462, + "learning_rate": 1.0056127221702525e-06, + "loss": 1.3224, + "step": 1962 + }, + { + "epoch": 2.4785353535353534, + "grad_norm": 3.454395055770874, + "learning_rate": 1.0032740879326476e-06, + "loss": 1.3513, + "step": 1963 + }, + { + "epoch": 2.4797979797979797, + "grad_norm": 3.5861761569976807, + "learning_rate": 1.0009354536950422e-06, + "loss": 1.3855, + "step": 1964 + }, + { + "epoch": 2.481060606060606, + "grad_norm": 3.2933616638183594, + "learning_rate": 9.985968194574368e-07, + "loss": 1.3303, + "step": 1965 + }, + { + "epoch": 2.4823232323232323, + "grad_norm": 3.368325710296631, + "learning_rate": 9.962581852198317e-07, + "loss": 1.3584, + "step": 1966 + }, + { + "epoch": 2.4835858585858586, + "grad_norm": 3.539789915084839, + "learning_rate": 9.939195509822265e-07, + "loss": 1.4043, + "step": 1967 + }, + { + "epoch": 2.484848484848485, + "grad_norm": 3.4011685848236084, + "learning_rate": 9.915809167446211e-07, + "loss": 1.4291, + "step": 1968 + }, + { + "epoch": 2.486111111111111, + "grad_norm": 3.239996910095215, + "learning_rate": 9.89242282507016e-07, + "loss": 1.3769, + "step": 1969 + }, + { + "epoch": 2.4873737373737375, + "grad_norm": 3.311413288116455, + "learning_rate": 9.869036482694106e-07, + "loss": 1.3636, + "step": 1970 + }, + { + "epoch": 2.4873737373737375, + "eval_loss": 1.6609822511672974, + "eval_runtime": 37.0636, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1970 + }, + { + "epoch": 2.4886363636363638, + "grad_norm": 3.514861583709717, + "learning_rate": 9.845650140318055e-07, + "loss": 1.3879, + "step": 1971 + }, + { + "epoch": 2.48989898989899, + "grad_norm": 3.385402202606201, + "learning_rate": 9.822263797942003e-07, + "loss": 1.336, + "step": 1972 + }, + { + "epoch": 2.4911616161616164, + "grad_norm": 3.3489670753479004, + "learning_rate": 9.79887745556595e-07, + "loss": 1.3734, + "step": 1973 + }, + { + "epoch": 2.492424242424242, + "grad_norm": 3.3082852363586426, + "learning_rate": 9.775491113189898e-07, + "loss": 1.3461, + "step": 1974 + }, + { + "epoch": 2.4936868686868685, + "grad_norm": 3.348860740661621, + "learning_rate": 9.752104770813846e-07, + "loss": 1.3502, + "step": 1975 + }, + { + "epoch": 2.494949494949495, + "grad_norm": 3.4895060062408447, + "learning_rate": 9.728718428437792e-07, + "loss": 1.3889, + "step": 1976 + }, + { + "epoch": 2.496212121212121, + "grad_norm": 3.5898680686950684, + "learning_rate": 9.70533208606174e-07, + "loss": 1.4076, + "step": 1977 + }, + { + "epoch": 2.4974747474747474, + "grad_norm": 3.306730270385742, + "learning_rate": 9.68194574368569e-07, + "loss": 1.3627, + "step": 1978 + }, + { + "epoch": 2.4987373737373737, + "grad_norm": 3.375053882598877, + "learning_rate": 9.658559401309636e-07, + "loss": 1.3422, + "step": 1979 + }, + { + "epoch": 2.5, + "grad_norm": 3.5166397094726562, + "learning_rate": 9.635173058933584e-07, + "loss": 1.3922, + "step": 1980 + }, + { + "epoch": 2.5, + "eval_loss": 1.6624678373336792, + "eval_runtime": 37.0699, + "eval_samples_per_second": 26.976, + "eval_steps_per_second": 1.133, + "step": 1980 + }, + { + "epoch": 2.5012626262626263, + "grad_norm": 3.405553102493286, + "learning_rate": 9.61178671655753e-07, + "loss": 1.3411, + "step": 1981 + }, + { + "epoch": 2.5025252525252526, + "grad_norm": 3.56683349609375, + "learning_rate": 9.588400374181479e-07, + "loss": 1.3832, + "step": 1982 + }, + { + "epoch": 2.503787878787879, + "grad_norm": 3.387821912765503, + "learning_rate": 9.565014031805427e-07, + "loss": 1.3407, + "step": 1983 + }, + { + "epoch": 2.505050505050505, + "grad_norm": 3.5401318073272705, + "learning_rate": 9.541627689429374e-07, + "loss": 1.3903, + "step": 1984 + }, + { + "epoch": 2.5063131313131315, + "grad_norm": 3.5890989303588867, + "learning_rate": 9.518241347053321e-07, + "loss": 1.3839, + "step": 1985 + }, + { + "epoch": 2.507575757575758, + "grad_norm": 3.3846449851989746, + "learning_rate": 9.494855004677269e-07, + "loss": 1.3808, + "step": 1986 + }, + { + "epoch": 2.508838383838384, + "grad_norm": 3.7580108642578125, + "learning_rate": 9.471468662301217e-07, + "loss": 1.403, + "step": 1987 + }, + { + "epoch": 2.51010101010101, + "grad_norm": 3.317741870880127, + "learning_rate": 9.448082319925164e-07, + "loss": 1.315, + "step": 1988 + }, + { + "epoch": 2.5113636363636362, + "grad_norm": 3.453925848007202, + "learning_rate": 9.424695977549111e-07, + "loss": 1.3763, + "step": 1989 + }, + { + "epoch": 2.5126262626262625, + "grad_norm": 3.5050547122955322, + "learning_rate": 9.40130963517306e-07, + "loss": 1.3237, + "step": 1990 + }, + { + "epoch": 2.5126262626262625, + "eval_loss": 1.6650363206863403, + "eval_runtime": 37.2939, + "eval_samples_per_second": 26.814, + "eval_steps_per_second": 1.126, + "step": 1990 + }, + { + "epoch": 2.513888888888889, + "grad_norm": 3.484497308731079, + "learning_rate": 9.377923292797007e-07, + "loss": 1.371, + "step": 1991 + }, + { + "epoch": 2.515151515151515, + "grad_norm": 3.4720635414123535, + "learning_rate": 9.354536950420955e-07, + "loss": 1.3496, + "step": 1992 + }, + { + "epoch": 2.5164141414141414, + "grad_norm": 3.511245012283325, + "learning_rate": 9.331150608044902e-07, + "loss": 1.3208, + "step": 1993 + }, + { + "epoch": 2.5176767676767677, + "grad_norm": 3.4779229164123535, + "learning_rate": 9.30776426566885e-07, + "loss": 1.3743, + "step": 1994 + }, + { + "epoch": 2.518939393939394, + "grad_norm": 3.5714519023895264, + "learning_rate": 9.284377923292798e-07, + "loss": 1.3659, + "step": 1995 + }, + { + "epoch": 2.5202020202020203, + "grad_norm": 3.501225709915161, + "learning_rate": 9.260991580916745e-07, + "loss": 1.4598, + "step": 1996 + }, + { + "epoch": 2.5214646464646466, + "grad_norm": 3.281121253967285, + "learning_rate": 9.237605238540694e-07, + "loss": 1.3342, + "step": 1997 + }, + { + "epoch": 2.5227272727272725, + "grad_norm": 3.4490602016448975, + "learning_rate": 9.214218896164641e-07, + "loss": 1.3958, + "step": 1998 + }, + { + "epoch": 2.523989898989899, + "grad_norm": 3.4208924770355225, + "learning_rate": 9.190832553788588e-07, + "loss": 1.3943, + "step": 1999 + }, + { + "epoch": 2.525252525252525, + "grad_norm": 3.2830605506896973, + "learning_rate": 9.167446211412536e-07, + "loss": 1.3649, + "step": 2000 + }, + { + "epoch": 2.525252525252525, + "eval_loss": 1.6609365940093994, + "eval_runtime": 37.39, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 2000 + }, + { + "epoch": 2.5265151515151514, + "grad_norm": 3.5467946529388428, + "learning_rate": 9.144059869036484e-07, + "loss": 1.4129, + "step": 2001 + }, + { + "epoch": 2.5277777777777777, + "grad_norm": 3.435167074203491, + "learning_rate": 9.120673526660431e-07, + "loss": 1.3497, + "step": 2002 + }, + { + "epoch": 2.529040404040404, + "grad_norm": 3.4507105350494385, + "learning_rate": 9.097287184284379e-07, + "loss": 1.4018, + "step": 2003 + }, + { + "epoch": 2.5303030303030303, + "grad_norm": 3.4729814529418945, + "learning_rate": 9.073900841908326e-07, + "loss": 1.4069, + "step": 2004 + }, + { + "epoch": 2.5315656565656566, + "grad_norm": 3.3762307167053223, + "learning_rate": 9.050514499532275e-07, + "loss": 1.3379, + "step": 2005 + }, + { + "epoch": 2.532828282828283, + "grad_norm": 3.4308269023895264, + "learning_rate": 9.027128157156222e-07, + "loss": 1.385, + "step": 2006 + }, + { + "epoch": 2.534090909090909, + "grad_norm": 3.455213785171509, + "learning_rate": 9.003741814780169e-07, + "loss": 1.3543, + "step": 2007 + }, + { + "epoch": 2.5353535353535355, + "grad_norm": 3.360630989074707, + "learning_rate": 8.980355472404116e-07, + "loss": 1.3399, + "step": 2008 + }, + { + "epoch": 2.5366161616161618, + "grad_norm": 3.4755003452301025, + "learning_rate": 8.956969130028065e-07, + "loss": 1.4, + "step": 2009 + }, + { + "epoch": 2.537878787878788, + "grad_norm": 3.473783016204834, + "learning_rate": 8.933582787652013e-07, + "loss": 1.3265, + "step": 2010 + }, + { + "epoch": 2.537878787878788, + "eval_loss": 1.6632015705108643, + "eval_runtime": 36.9441, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.5391414141414144, + "grad_norm": 3.3548569679260254, + "learning_rate": 8.910196445275959e-07, + "loss": 1.3642, + "step": 2011 + }, + { + "epoch": 2.5404040404040407, + "grad_norm": 3.4288694858551025, + "learning_rate": 8.886810102899906e-07, + "loss": 1.3571, + "step": 2012 + }, + { + "epoch": 2.5416666666666665, + "grad_norm": 3.5140700340270996, + "learning_rate": 8.863423760523855e-07, + "loss": 1.3791, + "step": 2013 + }, + { + "epoch": 2.542929292929293, + "grad_norm": 3.4733200073242188, + "learning_rate": 8.840037418147802e-07, + "loss": 1.4035, + "step": 2014 + }, + { + "epoch": 2.544191919191919, + "grad_norm": 3.2244248390197754, + "learning_rate": 8.816651075771749e-07, + "loss": 1.3167, + "step": 2015 + }, + { + "epoch": 2.5454545454545454, + "grad_norm": 3.3097362518310547, + "learning_rate": 8.793264733395698e-07, + "loss": 1.3032, + "step": 2016 + }, + { + "epoch": 2.5467171717171717, + "grad_norm": 3.2589709758758545, + "learning_rate": 8.769878391019645e-07, + "loss": 1.343, + "step": 2017 + }, + { + "epoch": 2.547979797979798, + "grad_norm": 3.3932478427886963, + "learning_rate": 8.746492048643593e-07, + "loss": 1.3539, + "step": 2018 + }, + { + "epoch": 2.5492424242424243, + "grad_norm": 3.305260181427002, + "learning_rate": 8.72310570626754e-07, + "loss": 1.3596, + "step": 2019 + }, + { + "epoch": 2.5505050505050506, + "grad_norm": 3.381182909011841, + "learning_rate": 8.699719363891488e-07, + "loss": 1.3562, + "step": 2020 + }, + { + "epoch": 2.5505050505050506, + "eval_loss": 1.6605889797210693, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.551767676767677, + "grad_norm": 3.406937599182129, + "learning_rate": 8.676333021515436e-07, + "loss": 1.3388, + "step": 2021 + }, + { + "epoch": 2.5530303030303028, + "grad_norm": 3.587719678878784, + "learning_rate": 8.652946679139383e-07, + "loss": 1.4079, + "step": 2022 + }, + { + "epoch": 2.554292929292929, + "grad_norm": 3.5528013706207275, + "learning_rate": 8.62956033676333e-07, + "loss": 1.3852, + "step": 2023 + }, + { + "epoch": 2.5555555555555554, + "grad_norm": 3.4368793964385986, + "learning_rate": 8.606173994387279e-07, + "loss": 1.3738, + "step": 2024 + }, + { + "epoch": 2.5568181818181817, + "grad_norm": 3.517357587814331, + "learning_rate": 8.582787652011226e-07, + "loss": 1.4152, + "step": 2025 + }, + { + "epoch": 2.558080808080808, + "grad_norm": 3.5744385719299316, + "learning_rate": 8.559401309635174e-07, + "loss": 1.381, + "step": 2026 + }, + { + "epoch": 2.5593434343434343, + "grad_norm": 3.3060715198516846, + "learning_rate": 8.536014967259121e-07, + "loss": 1.3174, + "step": 2027 + }, + { + "epoch": 2.5606060606060606, + "grad_norm": 3.3646938800811768, + "learning_rate": 8.512628624883069e-07, + "loss": 1.3647, + "step": 2028 + }, + { + "epoch": 2.561868686868687, + "grad_norm": 3.4408724308013916, + "learning_rate": 8.489242282507017e-07, + "loss": 1.3381, + "step": 2029 + }, + { + "epoch": 2.563131313131313, + "grad_norm": 3.381373405456543, + "learning_rate": 8.465855940130964e-07, + "loss": 1.3228, + "step": 2030 + }, + { + "epoch": 2.563131313131313, + "eval_loss": 1.6585171222686768, + "eval_runtime": 37.1174, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.5643939393939394, + "grad_norm": 3.440295696258545, + "learning_rate": 8.442469597754912e-07, + "loss": 1.4118, + "step": 2031 + }, + { + "epoch": 2.5656565656565657, + "grad_norm": 3.497279644012451, + "learning_rate": 8.41908325537886e-07, + "loss": 1.3721, + "step": 2032 + }, + { + "epoch": 2.566919191919192, + "grad_norm": 3.302016496658325, + "learning_rate": 8.395696913002807e-07, + "loss": 1.3179, + "step": 2033 + }, + { + "epoch": 2.5681818181818183, + "grad_norm": 3.311852216720581, + "learning_rate": 8.372310570626755e-07, + "loss": 1.3641, + "step": 2034 + }, + { + "epoch": 2.5694444444444446, + "grad_norm": 3.566032648086548, + "learning_rate": 8.348924228250702e-07, + "loss": 1.3996, + "step": 2035 + }, + { + "epoch": 2.570707070707071, + "grad_norm": 3.6114914417266846, + "learning_rate": 8.32553788587465e-07, + "loss": 1.3694, + "step": 2036 + }, + { + "epoch": 2.5719696969696972, + "grad_norm": 3.52956223487854, + "learning_rate": 8.302151543498598e-07, + "loss": 1.3811, + "step": 2037 + }, + { + "epoch": 2.573232323232323, + "grad_norm": 3.372699022293091, + "learning_rate": 8.278765201122545e-07, + "loss": 1.3762, + "step": 2038 + }, + { + "epoch": 2.5744949494949494, + "grad_norm": 3.3569037914276123, + "learning_rate": 8.255378858746494e-07, + "loss": 1.3126, + "step": 2039 + }, + { + "epoch": 2.5757575757575757, + "grad_norm": 3.764512062072754, + "learning_rate": 8.231992516370441e-07, + "loss": 1.397, + "step": 2040 + }, + { + "epoch": 2.5757575757575757, + "eval_loss": 1.6595747470855713, + "eval_runtime": 37.1249, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2040 + }, + { + "epoch": 2.577020202020202, + "grad_norm": 3.3517415523529053, + "learning_rate": 8.208606173994387e-07, + "loss": 1.2947, + "step": 2041 + }, + { + "epoch": 2.5782828282828283, + "grad_norm": 3.386699676513672, + "learning_rate": 8.185219831618335e-07, + "loss": 1.3313, + "step": 2042 + }, + { + "epoch": 2.5795454545454546, + "grad_norm": 3.480076313018799, + "learning_rate": 8.161833489242284e-07, + "loss": 1.3852, + "step": 2043 + }, + { + "epoch": 2.580808080808081, + "grad_norm": 3.2437856197357178, + "learning_rate": 8.138447146866231e-07, + "loss": 1.3265, + "step": 2044 + }, + { + "epoch": 2.582070707070707, + "grad_norm": 3.411377191543579, + "learning_rate": 8.115060804490178e-07, + "loss": 1.3815, + "step": 2045 + }, + { + "epoch": 2.5833333333333335, + "grad_norm": 3.3562614917755127, + "learning_rate": 8.091674462114125e-07, + "loss": 1.3599, + "step": 2046 + }, + { + "epoch": 2.5845959595959593, + "grad_norm": 3.4753265380859375, + "learning_rate": 8.068288119738074e-07, + "loss": 1.3585, + "step": 2047 + }, + { + "epoch": 2.5858585858585856, + "grad_norm": 3.36663556098938, + "learning_rate": 8.044901777362021e-07, + "loss": 1.3942, + "step": 2048 + }, + { + "epoch": 2.587121212121212, + "grad_norm": 3.3582539558410645, + "learning_rate": 8.021515434985968e-07, + "loss": 1.3131, + "step": 2049 + }, + { + "epoch": 2.5883838383838382, + "grad_norm": 3.4529905319213867, + "learning_rate": 7.998129092609916e-07, + "loss": 1.3659, + "step": 2050 + }, + { + "epoch": 2.5883838383838382, + "eval_loss": 1.6579172611236572, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.5896464646464645, + "grad_norm": 3.482149124145508, + "learning_rate": 7.974742750233864e-07, + "loss": 1.3699, + "step": 2051 + }, + { + "epoch": 2.590909090909091, + "grad_norm": 3.3723981380462646, + "learning_rate": 7.951356407857812e-07, + "loss": 1.3659, + "step": 2052 + }, + { + "epoch": 2.592171717171717, + "grad_norm": 3.3570878505706787, + "learning_rate": 7.927970065481759e-07, + "loss": 1.3449, + "step": 2053 + }, + { + "epoch": 2.5934343434343434, + "grad_norm": 3.4706034660339355, + "learning_rate": 7.904583723105706e-07, + "loss": 1.3752, + "step": 2054 + }, + { + "epoch": 2.5946969696969697, + "grad_norm": 3.4540836811065674, + "learning_rate": 7.881197380729655e-07, + "loss": 1.3561, + "step": 2055 + }, + { + "epoch": 2.595959595959596, + "grad_norm": 3.3945505619049072, + "learning_rate": 7.857811038353602e-07, + "loss": 1.3815, + "step": 2056 + }, + { + "epoch": 2.5972222222222223, + "grad_norm": 3.5257701873779297, + "learning_rate": 7.83442469597755e-07, + "loss": 1.3682, + "step": 2057 + }, + { + "epoch": 2.5984848484848486, + "grad_norm": 3.3582730293273926, + "learning_rate": 7.811038353601498e-07, + "loss": 1.3481, + "step": 2058 + }, + { + "epoch": 2.599747474747475, + "grad_norm": 3.277299404144287, + "learning_rate": 7.787652011225445e-07, + "loss": 1.2955, + "step": 2059 + }, + { + "epoch": 2.601010101010101, + "grad_norm": 3.414294719696045, + "learning_rate": 7.764265668849393e-07, + "loss": 1.3627, + "step": 2060 + }, + { + "epoch": 2.601010101010101, + "eval_loss": 1.6582155227661133, + "eval_runtime": 37.1197, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 2060 + }, + { + "epoch": 2.6022727272727275, + "grad_norm": 3.399158000946045, + "learning_rate": 7.74087932647334e-07, + "loss": 1.2828, + "step": 2061 + }, + { + "epoch": 2.6035353535353534, + "grad_norm": 3.4865469932556152, + "learning_rate": 7.717492984097288e-07, + "loss": 1.3871, + "step": 2062 + }, + { + "epoch": 2.6047979797979797, + "grad_norm": 3.5197713375091553, + "learning_rate": 7.694106641721236e-07, + "loss": 1.3994, + "step": 2063 + }, + { + "epoch": 2.606060606060606, + "grad_norm": 3.4030301570892334, + "learning_rate": 7.670720299345183e-07, + "loss": 1.3309, + "step": 2064 + }, + { + "epoch": 2.6073232323232323, + "grad_norm": 3.514847755432129, + "learning_rate": 7.647333956969131e-07, + "loss": 1.3955, + "step": 2065 + }, + { + "epoch": 2.6085858585858586, + "grad_norm": 3.4094550609588623, + "learning_rate": 7.623947614593079e-07, + "loss": 1.3486, + "step": 2066 + }, + { + "epoch": 2.609848484848485, + "grad_norm": 3.335812568664551, + "learning_rate": 7.600561272217026e-07, + "loss": 1.3556, + "step": 2067 + }, + { + "epoch": 2.611111111111111, + "grad_norm": 3.3734757900238037, + "learning_rate": 7.577174929840974e-07, + "loss": 1.3443, + "step": 2068 + }, + { + "epoch": 2.6123737373737375, + "grad_norm": 3.4860036373138428, + "learning_rate": 7.55378858746492e-07, + "loss": 1.3652, + "step": 2069 + }, + { + "epoch": 2.6136363636363638, + "grad_norm": 3.323676347732544, + "learning_rate": 7.53040224508887e-07, + "loss": 1.3547, + "step": 2070 + }, + { + "epoch": 2.6136363636363638, + "eval_loss": 1.6536892652511597, + "eval_runtime": 37.1179, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.61489898989899, + "grad_norm": 3.372002363204956, + "learning_rate": 7.507015902712817e-07, + "loss": 1.3369, + "step": 2071 + }, + { + "epoch": 2.616161616161616, + "grad_norm": 3.329336643218994, + "learning_rate": 7.483629560336763e-07, + "loss": 1.3463, + "step": 2072 + }, + { + "epoch": 2.617424242424242, + "grad_norm": 3.4763405323028564, + "learning_rate": 7.460243217960711e-07, + "loss": 1.3634, + "step": 2073 + }, + { + "epoch": 2.6186868686868685, + "grad_norm": 3.662146806716919, + "learning_rate": 7.43685687558466e-07, + "loss": 1.3864, + "step": 2074 + }, + { + "epoch": 2.619949494949495, + "grad_norm": 3.4778242111206055, + "learning_rate": 7.413470533208606e-07, + "loss": 1.4011, + "step": 2075 + }, + { + "epoch": 2.621212121212121, + "grad_norm": 3.494809627532959, + "learning_rate": 7.390084190832554e-07, + "loss": 1.3807, + "step": 2076 + }, + { + "epoch": 2.6224747474747474, + "grad_norm": 3.383918523788452, + "learning_rate": 7.366697848456501e-07, + "loss": 1.3477, + "step": 2077 + }, + { + "epoch": 2.6237373737373737, + "grad_norm": 3.415273666381836, + "learning_rate": 7.34331150608045e-07, + "loss": 1.3611, + "step": 2078 + }, + { + "epoch": 2.625, + "grad_norm": 3.4107258319854736, + "learning_rate": 7.319925163704397e-07, + "loss": 1.3545, + "step": 2079 + }, + { + "epoch": 2.6262626262626263, + "grad_norm": 3.4525718688964844, + "learning_rate": 7.296538821328344e-07, + "loss": 1.3402, + "step": 2080 + }, + { + "epoch": 2.6262626262626263, + "eval_loss": 1.651517629623413, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.6275252525252526, + "grad_norm": 3.373176336288452, + "learning_rate": 7.273152478952293e-07, + "loss": 1.3789, + "step": 2081 + }, + { + "epoch": 2.628787878787879, + "grad_norm": 3.379733085632324, + "learning_rate": 7.24976613657624e-07, + "loss": 1.3595, + "step": 2082 + }, + { + "epoch": 2.630050505050505, + "grad_norm": 3.2968251705169678, + "learning_rate": 7.226379794200187e-07, + "loss": 1.354, + "step": 2083 + }, + { + "epoch": 2.6313131313131315, + "grad_norm": 3.4465534687042236, + "learning_rate": 7.202993451824135e-07, + "loss": 1.3776, + "step": 2084 + }, + { + "epoch": 2.632575757575758, + "grad_norm": 3.4734654426574707, + "learning_rate": 7.179607109448083e-07, + "loss": 1.3893, + "step": 2085 + }, + { + "epoch": 2.633838383838384, + "grad_norm": 3.4402577877044678, + "learning_rate": 7.156220767072031e-07, + "loss": 1.3412, + "step": 2086 + }, + { + "epoch": 2.63510101010101, + "grad_norm": 3.5570950508117676, + "learning_rate": 7.132834424695978e-07, + "loss": 1.38, + "step": 2087 + }, + { + "epoch": 2.6363636363636362, + "grad_norm": 3.4646284580230713, + "learning_rate": 7.109448082319925e-07, + "loss": 1.3816, + "step": 2088 + }, + { + "epoch": 2.6376262626262625, + "grad_norm": 3.5913166999816895, + "learning_rate": 7.086061739943874e-07, + "loss": 1.3754, + "step": 2089 + }, + { + "epoch": 2.638888888888889, + "grad_norm": 3.4469869136810303, + "learning_rate": 7.062675397567821e-07, + "loss": 1.3582, + "step": 2090 + }, + { + "epoch": 2.638888888888889, + "eval_loss": 1.6559001207351685, + "eval_runtime": 37.3478, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.640151515151515, + "grad_norm": 3.5393295288085938, + "learning_rate": 7.039289055191769e-07, + "loss": 1.3602, + "step": 2091 + }, + { + "epoch": 2.6414141414141414, + "grad_norm": 3.431546211242676, + "learning_rate": 7.015902712815716e-07, + "loss": 1.3074, + "step": 2092 + }, + { + "epoch": 2.6426767676767677, + "grad_norm": 3.3025810718536377, + "learning_rate": 6.992516370439664e-07, + "loss": 1.2884, + "step": 2093 + }, + { + "epoch": 2.643939393939394, + "grad_norm": 3.4054274559020996, + "learning_rate": 6.969130028063612e-07, + "loss": 1.3283, + "step": 2094 + }, + { + "epoch": 2.6452020202020203, + "grad_norm": 3.5449748039245605, + "learning_rate": 6.945743685687559e-07, + "loss": 1.3896, + "step": 2095 + }, + { + "epoch": 2.6464646464646466, + "grad_norm": 3.528944253921509, + "learning_rate": 6.922357343311506e-07, + "loss": 1.3487, + "step": 2096 + }, + { + "epoch": 2.6477272727272725, + "grad_norm": 3.5601792335510254, + "learning_rate": 6.898971000935455e-07, + "loss": 1.3995, + "step": 2097 + }, + { + "epoch": 2.648989898989899, + "grad_norm": 3.358328104019165, + "learning_rate": 6.875584658559402e-07, + "loss": 1.3395, + "step": 2098 + }, + { + "epoch": 2.650252525252525, + "grad_norm": 3.392995595932007, + "learning_rate": 6.85219831618335e-07, + "loss": 1.3806, + "step": 2099 + }, + { + "epoch": 2.6515151515151514, + "grad_norm": 3.4869582653045654, + "learning_rate": 6.828811973807298e-07, + "loss": 1.3696, + "step": 2100 + }, + { + "epoch": 2.6515151515151514, + "eval_loss": 1.6575806140899658, + "eval_runtime": 37.4666, + "eval_samples_per_second": 26.69, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.6527777777777777, + "grad_norm": 3.29205060005188, + "learning_rate": 6.805425631431245e-07, + "loss": 1.3119, + "step": 2101 + }, + { + "epoch": 2.654040404040404, + "grad_norm": 3.309109687805176, + "learning_rate": 6.782039289055192e-07, + "loss": 1.326, + "step": 2102 + }, + { + "epoch": 2.6553030303030303, + "grad_norm": 3.3156912326812744, + "learning_rate": 6.758652946679139e-07, + "loss": 1.3296, + "step": 2103 + }, + { + "epoch": 2.6565656565656566, + "grad_norm": 3.45403790473938, + "learning_rate": 6.735266604303089e-07, + "loss": 1.4157, + "step": 2104 + }, + { + "epoch": 2.657828282828283, + "grad_norm": 3.540456533432007, + "learning_rate": 6.711880261927035e-07, + "loss": 1.3798, + "step": 2105 + }, + { + "epoch": 2.659090909090909, + "grad_norm": 3.4654109477996826, + "learning_rate": 6.688493919550982e-07, + "loss": 1.3992, + "step": 2106 + }, + { + "epoch": 2.6603535353535355, + "grad_norm": 3.514021873474121, + "learning_rate": 6.66510757717493e-07, + "loss": 1.3868, + "step": 2107 + }, + { + "epoch": 2.6616161616161618, + "grad_norm": 3.4709298610687256, + "learning_rate": 6.641721234798878e-07, + "loss": 1.3414, + "step": 2108 + }, + { + "epoch": 2.662878787878788, + "grad_norm": 3.4472742080688477, + "learning_rate": 6.618334892422825e-07, + "loss": 1.2879, + "step": 2109 + }, + { + "epoch": 2.6641414141414144, + "grad_norm": 3.7136893272399902, + "learning_rate": 6.594948550046773e-07, + "loss": 1.4012, + "step": 2110 + }, + { + "epoch": 2.6641414141414144, + "eval_loss": 1.658721923828125, + "eval_runtime": 36.9489, + "eval_samples_per_second": 27.064, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.6654040404040407, + "grad_norm": 3.3402535915374756, + "learning_rate": 6.57156220767072e-07, + "loss": 1.3412, + "step": 2111 + }, + { + "epoch": 2.6666666666666665, + "grad_norm": 3.3103139400482178, + "learning_rate": 6.548175865294669e-07, + "loss": 1.3207, + "step": 2112 + }, + { + "epoch": 2.667929292929293, + "grad_norm": 3.3308048248291016, + "learning_rate": 6.524789522918616e-07, + "loss": 1.3047, + "step": 2113 + }, + { + "epoch": 2.669191919191919, + "grad_norm": 3.4672188758850098, + "learning_rate": 6.501403180542563e-07, + "loss": 1.4158, + "step": 2114 + }, + { + "epoch": 2.6704545454545454, + "grad_norm": 3.522867202758789, + "learning_rate": 6.478016838166511e-07, + "loss": 1.3464, + "step": 2115 + }, + { + "epoch": 2.6717171717171717, + "grad_norm": 3.4110591411590576, + "learning_rate": 6.454630495790459e-07, + "loss": 1.3651, + "step": 2116 + }, + { + "epoch": 2.672979797979798, + "grad_norm": 3.4079062938690186, + "learning_rate": 6.431244153414407e-07, + "loss": 1.3554, + "step": 2117 + }, + { + "epoch": 2.6742424242424243, + "grad_norm": 3.4102203845977783, + "learning_rate": 6.407857811038354e-07, + "loss": 1.3953, + "step": 2118 + }, + { + "epoch": 2.6755050505050506, + "grad_norm": 3.381063222885132, + "learning_rate": 6.384471468662301e-07, + "loss": 1.3461, + "step": 2119 + }, + { + "epoch": 2.676767676767677, + "grad_norm": 3.3743748664855957, + "learning_rate": 6.36108512628625e-07, + "loss": 1.3559, + "step": 2120 + }, + { + "epoch": 2.676767676767677, + "eval_loss": 1.6545666456222534, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.6780303030303028, + "grad_norm": 3.450329542160034, + "learning_rate": 6.337698783910197e-07, + "loss": 1.4058, + "step": 2121 + }, + { + "epoch": 2.679292929292929, + "grad_norm": 3.490386486053467, + "learning_rate": 6.314312441534144e-07, + "loss": 1.3704, + "step": 2122 + }, + { + "epoch": 2.6805555555555554, + "grad_norm": 3.689189910888672, + "learning_rate": 6.290926099158093e-07, + "loss": 1.3778, + "step": 2123 + }, + { + "epoch": 2.6818181818181817, + "grad_norm": 3.437407970428467, + "learning_rate": 6.26753975678204e-07, + "loss": 1.3673, + "step": 2124 + }, + { + "epoch": 2.683080808080808, + "grad_norm": 3.6720778942108154, + "learning_rate": 6.244153414405988e-07, + "loss": 1.4601, + "step": 2125 + }, + { + "epoch": 2.6843434343434343, + "grad_norm": 3.5332326889038086, + "learning_rate": 6.220767072029935e-07, + "loss": 1.3204, + "step": 2126 + }, + { + "epoch": 2.6856060606060606, + "grad_norm": 3.474308967590332, + "learning_rate": 6.197380729653882e-07, + "loss": 1.357, + "step": 2127 + }, + { + "epoch": 2.686868686868687, + "grad_norm": 3.4824960231781006, + "learning_rate": 6.173994387277831e-07, + "loss": 1.3412, + "step": 2128 + }, + { + "epoch": 2.688131313131313, + "grad_norm": 3.622313976287842, + "learning_rate": 6.150608044901778e-07, + "loss": 1.4076, + "step": 2129 + }, + { + "epoch": 2.6893939393939394, + "grad_norm": 3.520381212234497, + "learning_rate": 6.127221702525726e-07, + "loss": 1.3341, + "step": 2130 + }, + { + "epoch": 2.6893939393939394, + "eval_loss": 1.6562730073928833, + "eval_runtime": 37.1249, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.6906565656565657, + "grad_norm": 3.5275325775146484, + "learning_rate": 6.103835360149673e-07, + "loss": 1.3876, + "step": 2131 + }, + { + "epoch": 2.691919191919192, + "grad_norm": 3.5301501750946045, + "learning_rate": 6.080449017773621e-07, + "loss": 1.3745, + "step": 2132 + }, + { + "epoch": 2.6931818181818183, + "grad_norm": 3.4208099842071533, + "learning_rate": 6.057062675397568e-07, + "loss": 1.3435, + "step": 2133 + }, + { + "epoch": 2.6944444444444446, + "grad_norm": 3.3709633350372314, + "learning_rate": 6.033676333021516e-07, + "loss": 1.3356, + "step": 2134 + }, + { + "epoch": 2.695707070707071, + "grad_norm": 3.427328109741211, + "learning_rate": 6.010289990645463e-07, + "loss": 1.3395, + "step": 2135 + }, + { + "epoch": 2.6969696969696972, + "grad_norm": 3.543050527572632, + "learning_rate": 5.986903648269411e-07, + "loss": 1.364, + "step": 2136 + }, + { + "epoch": 2.698232323232323, + "grad_norm": 3.422373056411743, + "learning_rate": 5.963517305893358e-07, + "loss": 1.3119, + "step": 2137 + }, + { + "epoch": 2.6994949494949494, + "grad_norm": 3.4447410106658936, + "learning_rate": 5.940130963517307e-07, + "loss": 1.2964, + "step": 2138 + }, + { + "epoch": 2.7007575757575757, + "grad_norm": 3.498711585998535, + "learning_rate": 5.916744621141254e-07, + "loss": 1.3655, + "step": 2139 + }, + { + "epoch": 2.702020202020202, + "grad_norm": 3.433220148086548, + "learning_rate": 5.893358278765201e-07, + "loss": 1.3559, + "step": 2140 + }, + { + "epoch": 2.702020202020202, + "eval_loss": 1.6559937000274658, + "eval_runtime": 37.1318, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.7032828282828283, + "grad_norm": 3.3882076740264893, + "learning_rate": 5.86997193638915e-07, + "loss": 1.3324, + "step": 2141 + }, + { + "epoch": 2.7045454545454546, + "grad_norm": 3.4268569946289062, + "learning_rate": 5.846585594013097e-07, + "loss": 1.3666, + "step": 2142 + }, + { + "epoch": 2.705808080808081, + "grad_norm": 3.507413625717163, + "learning_rate": 5.823199251637045e-07, + "loss": 1.3435, + "step": 2143 + }, + { + "epoch": 2.707070707070707, + "grad_norm": 3.5061936378479004, + "learning_rate": 5.799812909260992e-07, + "loss": 1.3656, + "step": 2144 + }, + { + "epoch": 2.7083333333333335, + "grad_norm": 3.4619619846343994, + "learning_rate": 5.77642656688494e-07, + "loss": 1.3643, + "step": 2145 + }, + { + "epoch": 2.7095959595959593, + "grad_norm": 3.463290214538574, + "learning_rate": 5.753040224508888e-07, + "loss": 1.337, + "step": 2146 + }, + { + "epoch": 2.7108585858585856, + "grad_norm": 3.4522857666015625, + "learning_rate": 5.729653882132835e-07, + "loss": 1.4072, + "step": 2147 + }, + { + "epoch": 2.712121212121212, + "grad_norm": 3.468033790588379, + "learning_rate": 5.706267539756782e-07, + "loss": 1.3417, + "step": 2148 + }, + { + "epoch": 2.7133838383838382, + "grad_norm": 3.556464910507202, + "learning_rate": 5.682881197380731e-07, + "loss": 1.3936, + "step": 2149 + }, + { + "epoch": 2.7146464646464645, + "grad_norm": 3.4759442806243896, + "learning_rate": 5.659494855004677e-07, + "loss": 1.3114, + "step": 2150 + }, + { + "epoch": 2.7146464646464645, + "eval_loss": 1.6544948816299438, + "eval_runtime": 37.1258, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 2150 + }, + { + "epoch": 2.715909090909091, + "grad_norm": 3.5131609439849854, + "learning_rate": 5.636108512628626e-07, + "loss": 1.3116, + "step": 2151 + }, + { + "epoch": 2.717171717171717, + "grad_norm": 3.492794990539551, + "learning_rate": 5.612722170252573e-07, + "loss": 1.3297, + "step": 2152 + }, + { + "epoch": 2.7184343434343434, + "grad_norm": 3.3546905517578125, + "learning_rate": 5.58933582787652e-07, + "loss": 1.3348, + "step": 2153 + }, + { + "epoch": 2.7196969696969697, + "grad_norm": 3.3591794967651367, + "learning_rate": 5.565949485500468e-07, + "loss": 1.3355, + "step": 2154 + }, + { + "epoch": 2.720959595959596, + "grad_norm": 3.4083364009857178, + "learning_rate": 5.542563143124416e-07, + "loss": 1.329, + "step": 2155 + }, + { + "epoch": 2.7222222222222223, + "grad_norm": 3.625196933746338, + "learning_rate": 5.519176800748364e-07, + "loss": 1.4063, + "step": 2156 + }, + { + "epoch": 2.7234848484848486, + "grad_norm": 3.534701347351074, + "learning_rate": 5.495790458372311e-07, + "loss": 1.4223, + "step": 2157 + }, + { + "epoch": 2.724747474747475, + "grad_norm": 3.4186694622039795, + "learning_rate": 5.472404115996258e-07, + "loss": 1.373, + "step": 2158 + }, + { + "epoch": 2.726010101010101, + "grad_norm": 3.31740140914917, + "learning_rate": 5.449017773620207e-07, + "loss": 1.3691, + "step": 2159 + }, + { + "epoch": 2.7272727272727275, + "grad_norm": 3.4586780071258545, + "learning_rate": 5.425631431244154e-07, + "loss": 1.3948, + "step": 2160 + }, + { + "epoch": 2.7272727272727275, + "eval_loss": 1.65037202835083, + "eval_runtime": 37.1256, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2160 + }, + { + "epoch": 2.7285353535353534, + "grad_norm": 3.497927188873291, + "learning_rate": 5.402245088868101e-07, + "loss": 1.3483, + "step": 2161 + }, + { + "epoch": 2.7297979797979797, + "grad_norm": 3.4493894577026367, + "learning_rate": 5.37885874649205e-07, + "loss": 1.365, + "step": 2162 + }, + { + "epoch": 2.731060606060606, + "grad_norm": 3.4639272689819336, + "learning_rate": 5.355472404115996e-07, + "loss": 1.3822, + "step": 2163 + }, + { + "epoch": 2.7323232323232323, + "grad_norm": 3.5272274017333984, + "learning_rate": 5.332086061739945e-07, + "loss": 1.3722, + "step": 2164 + }, + { + "epoch": 2.7335858585858586, + "grad_norm": 3.3754897117614746, + "learning_rate": 5.308699719363892e-07, + "loss": 1.3444, + "step": 2165 + }, + { + "epoch": 2.734848484848485, + "grad_norm": 3.374189615249634, + "learning_rate": 5.285313376987839e-07, + "loss": 1.3459, + "step": 2166 + }, + { + "epoch": 2.736111111111111, + "grad_norm": 3.2838594913482666, + "learning_rate": 5.261927034611787e-07, + "loss": 1.3476, + "step": 2167 + }, + { + "epoch": 2.7373737373737375, + "grad_norm": 3.6404688358306885, + "learning_rate": 5.238540692235735e-07, + "loss": 1.4402, + "step": 2168 + }, + { + "epoch": 2.7386363636363638, + "grad_norm": 3.395650625228882, + "learning_rate": 5.215154349859682e-07, + "loss": 1.3657, + "step": 2169 + }, + { + "epoch": 2.73989898989899, + "grad_norm": 3.5969934463500977, + "learning_rate": 5.19176800748363e-07, + "loss": 1.3823, + "step": 2170 + }, + { + "epoch": 2.73989898989899, + "eval_loss": 1.6518349647521973, + "eval_runtime": 37.1587, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.741161616161616, + "grad_norm": 3.495293617248535, + "learning_rate": 5.168381665107577e-07, + "loss": 1.3524, + "step": 2171 + }, + { + "epoch": 2.742424242424242, + "grad_norm": 3.486459255218506, + "learning_rate": 5.144995322731526e-07, + "loss": 1.3554, + "step": 2172 + }, + { + "epoch": 2.7436868686868685, + "grad_norm": 3.5103650093078613, + "learning_rate": 5.121608980355473e-07, + "loss": 1.3699, + "step": 2173 + }, + { + "epoch": 2.744949494949495, + "grad_norm": 3.401812791824341, + "learning_rate": 5.09822263797942e-07, + "loss": 1.4116, + "step": 2174 + }, + { + "epoch": 2.746212121212121, + "grad_norm": 3.5474627017974854, + "learning_rate": 5.074836295603368e-07, + "loss": 1.3821, + "step": 2175 + }, + { + "epoch": 2.7474747474747474, + "grad_norm": 3.347783327102661, + "learning_rate": 5.051449953227316e-07, + "loss": 1.3235, + "step": 2176 + }, + { + "epoch": 2.7487373737373737, + "grad_norm": 3.385366916656494, + "learning_rate": 5.028063610851263e-07, + "loss": 1.379, + "step": 2177 + }, + { + "epoch": 2.75, + "grad_norm": 3.563000202178955, + "learning_rate": 5.004677268475211e-07, + "loss": 1.411, + "step": 2178 + }, + { + "epoch": 2.7512626262626263, + "grad_norm": 3.539781093597412, + "learning_rate": 4.981290926099158e-07, + "loss": 1.3351, + "step": 2179 + }, + { + "epoch": 2.7525252525252526, + "grad_norm": 3.477810859680176, + "learning_rate": 4.957904583723106e-07, + "loss": 1.3385, + "step": 2180 + }, + { + "epoch": 2.7525252525252526, + "eval_loss": 1.6517527103424072, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.753787878787879, + "grad_norm": 3.6181442737579346, + "learning_rate": 4.934518241347053e-07, + "loss": 1.4082, + "step": 2181 + }, + { + "epoch": 2.755050505050505, + "grad_norm": 3.5040857791900635, + "learning_rate": 4.911131898971001e-07, + "loss": 1.3886, + "step": 2182 + }, + { + "epoch": 2.7563131313131315, + "grad_norm": 3.616530656814575, + "learning_rate": 4.887745556594949e-07, + "loss": 1.4308, + "step": 2183 + }, + { + "epoch": 2.757575757575758, + "grad_norm": 3.689122200012207, + "learning_rate": 4.864359214218896e-07, + "loss": 1.3394, + "step": 2184 + }, + { + "epoch": 2.758838383838384, + "grad_norm": 3.432558298110962, + "learning_rate": 4.840972871842845e-07, + "loss": 1.3169, + "step": 2185 + }, + { + "epoch": 2.76010101010101, + "grad_norm": 3.4511947631835938, + "learning_rate": 4.817586529466792e-07, + "loss": 1.3488, + "step": 2186 + }, + { + "epoch": 2.7613636363636362, + "grad_norm": 3.4622750282287598, + "learning_rate": 4.794200187090739e-07, + "loss": 1.3744, + "step": 2187 + }, + { + "epoch": 2.7626262626262625, + "grad_norm": 3.3733813762664795, + "learning_rate": 4.770813844714687e-07, + "loss": 1.3744, + "step": 2188 + }, + { + "epoch": 2.763888888888889, + "grad_norm": 3.4657459259033203, + "learning_rate": 4.7474275023386347e-07, + "loss": 1.3686, + "step": 2189 + }, + { + "epoch": 2.765151515151515, + "grad_norm": 3.5236029624938965, + "learning_rate": 4.724041159962582e-07, + "loss": 1.3809, + "step": 2190 + }, + { + "epoch": 2.765151515151515, + "eval_loss": 1.653585433959961, + "eval_runtime": 37.2341, + "eval_samples_per_second": 26.857, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.7664141414141414, + "grad_norm": 3.5313808917999268, + "learning_rate": 4.70065481758653e-07, + "loss": 1.3772, + "step": 2191 + }, + { + "epoch": 2.7676767676767677, + "grad_norm": 3.4364354610443115, + "learning_rate": 4.6772684752104773e-07, + "loss": 1.3641, + "step": 2192 + }, + { + "epoch": 2.768939393939394, + "grad_norm": 3.5184450149536133, + "learning_rate": 4.653882132834425e-07, + "loss": 1.3629, + "step": 2193 + }, + { + "epoch": 2.7702020202020203, + "grad_norm": 3.4177098274230957, + "learning_rate": 4.6304957904583726e-07, + "loss": 1.3624, + "step": 2194 + }, + { + "epoch": 2.7714646464646466, + "grad_norm": 3.365241527557373, + "learning_rate": 4.6071094480823205e-07, + "loss": 1.3077, + "step": 2195 + }, + { + "epoch": 2.7727272727272725, + "grad_norm": 3.479496955871582, + "learning_rate": 4.583723105706268e-07, + "loss": 1.3509, + "step": 2196 + }, + { + "epoch": 2.773989898989899, + "grad_norm": 3.5142934322357178, + "learning_rate": 4.5603367633302157e-07, + "loss": 1.3876, + "step": 2197 + }, + { + "epoch": 2.775252525252525, + "grad_norm": 3.4927608966827393, + "learning_rate": 4.536950420954163e-07, + "loss": 1.3403, + "step": 2198 + }, + { + "epoch": 2.7765151515151514, + "grad_norm": 3.4133102893829346, + "learning_rate": 4.513564078578111e-07, + "loss": 1.3748, + "step": 2199 + }, + { + "epoch": 2.7777777777777777, + "grad_norm": 3.4805235862731934, + "learning_rate": 4.490177736202058e-07, + "loss": 1.3569, + "step": 2200 + }, + { + "epoch": 2.7777777777777777, + "eval_loss": 1.6538234949111938, + "eval_runtime": 37.4443, + "eval_samples_per_second": 26.706, + "eval_steps_per_second": 1.122, + "step": 2200 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 6.858080700596224e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2200/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/latest new file mode 100644 index 0000000000000000000000000000000000000000..78dc01ce4ce1498b753887dfb7af92409705a69b --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/latest @@ -0,0 +1 @@ +global_step2300 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..118cc80ce7ae798b591e047a1e311991b2cad18c --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/trainer_state.json @@ -0,0 +1,17973 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.904040404040404, + "eval_steps": 10, + "global_step": 2300, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + }, + { + "epoch": 0.88510101010101, + "grad_norm": 2.7135095596313477, + "learning_rate": 3.952291861552854e-06, + "loss": 1.6013, + "step": 701 + }, + { + "epoch": 0.8863636363636364, + "grad_norm": 2.812061071395874, + "learning_rate": 3.949953227315248e-06, + "loss": 1.5518, + "step": 702 + }, + { + "epoch": 0.8876262626262627, + "grad_norm": 3.0532422065734863, + "learning_rate": 3.9476145930776424e-06, + "loss": 1.5579, + "step": 703 + }, + { + "epoch": 0.8888888888888888, + "grad_norm": 2.8306567668914795, + "learning_rate": 3.945275958840038e-06, + "loss": 1.5478, + "step": 704 + }, + { + "epoch": 0.8901515151515151, + "grad_norm": 2.737921953201294, + "learning_rate": 3.942937324602433e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.8914141414141414, + "grad_norm": 2.902211904525757, + "learning_rate": 3.940598690364827e-06, + "loss": 1.5715, + "step": 706 + }, + { + "epoch": 0.8926767676767676, + "grad_norm": 2.725818395614624, + "learning_rate": 3.938260056127222e-06, + "loss": 1.5502, + "step": 707 + }, + { + "epoch": 0.8939393939393939, + "grad_norm": 2.9530506134033203, + "learning_rate": 3.935921421889617e-06, + "loss": 1.564, + "step": 708 + }, + { + "epoch": 0.8952020202020202, + "grad_norm": 2.798231363296509, + "learning_rate": 3.9335827876520115e-06, + "loss": 1.5445, + "step": 709 + }, + { + "epoch": 0.8964646464646465, + "grad_norm": 3.0233471393585205, + "learning_rate": 3.931244153414407e-06, + "loss": 1.6193, + "step": 710 + }, + { + "epoch": 0.8964646464646465, + "eval_loss": 1.742077350616455, + "eval_runtime": 37.0255, + "eval_samples_per_second": 27.008, + "eval_steps_per_second": 1.134, + "step": 710 + }, + { + "epoch": 0.8977272727272727, + "grad_norm": 2.84309458732605, + "learning_rate": 3.928905519176801e-06, + "loss": 1.5379, + "step": 711 + }, + { + "epoch": 0.898989898989899, + "grad_norm": 2.7412185668945312, + "learning_rate": 3.926566884939196e-06, + "loss": 1.5837, + "step": 712 + }, + { + "epoch": 0.9002525252525253, + "grad_norm": 3.1094822883605957, + "learning_rate": 3.924228250701591e-06, + "loss": 1.5621, + "step": 713 + }, + { + "epoch": 0.9015151515151515, + "grad_norm": 2.776059627532959, + "learning_rate": 3.921889616463985e-06, + "loss": 1.5714, + "step": 714 + }, + { + "epoch": 0.9027777777777778, + "grad_norm": 2.7650489807128906, + "learning_rate": 3.91955098222638e-06, + "loss": 1.5418, + "step": 715 + }, + { + "epoch": 0.9040404040404041, + "grad_norm": 2.8586974143981934, + "learning_rate": 3.917212347988775e-06, + "loss": 1.5891, + "step": 716 + }, + { + "epoch": 0.9053030303030303, + "grad_norm": 2.74332594871521, + "learning_rate": 3.914873713751169e-06, + "loss": 1.5492, + "step": 717 + }, + { + "epoch": 0.9065656565656566, + "grad_norm": 2.931682586669922, + "learning_rate": 3.912535079513564e-06, + "loss": 1.54, + "step": 718 + }, + { + "epoch": 0.9078282828282829, + "grad_norm": 2.678812026977539, + "learning_rate": 3.910196445275959e-06, + "loss": 1.5677, + "step": 719 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.7282893657684326, + "learning_rate": 3.907857811038354e-06, + "loss": 1.5413, + "step": 720 + }, + { + "epoch": 0.9090909090909091, + "eval_loss": 1.7409753799438477, + "eval_runtime": 37.1883, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 720 + }, + { + "epoch": 0.9103535353535354, + "grad_norm": 3.073997735977173, + "learning_rate": 3.905519176800749e-06, + "loss": 1.5734, + "step": 721 + }, + { + "epoch": 0.9116161616161617, + "grad_norm": 2.6249935626983643, + "learning_rate": 3.903180542563144e-06, + "loss": 1.5063, + "step": 722 + }, + { + "epoch": 0.9128787878787878, + "grad_norm": 2.7758615016937256, + "learning_rate": 3.9008419083255384e-06, + "loss": 1.5112, + "step": 723 + }, + { + "epoch": 0.9141414141414141, + "grad_norm": 2.8954596519470215, + "learning_rate": 3.898503274087933e-06, + "loss": 1.5508, + "step": 724 + }, + { + "epoch": 0.9154040404040404, + "grad_norm": 2.79073166847229, + "learning_rate": 3.896164639850328e-06, + "loss": 1.5773, + "step": 725 + }, + { + "epoch": 0.9166666666666666, + "grad_norm": 2.687027931213379, + "learning_rate": 3.8938260056127226e-06, + "loss": 1.5243, + "step": 726 + }, + { + "epoch": 0.9179292929292929, + "grad_norm": 2.5704851150512695, + "learning_rate": 3.891487371375117e-06, + "loss": 1.5075, + "step": 727 + }, + { + "epoch": 0.9191919191919192, + "grad_norm": 2.8684048652648926, + "learning_rate": 3.889148737137512e-06, + "loss": 1.6471, + "step": 728 + }, + { + "epoch": 0.9204545454545454, + "grad_norm": 2.8995590209960938, + "learning_rate": 3.886810102899907e-06, + "loss": 1.5429, + "step": 729 + }, + { + "epoch": 0.9217171717171717, + "grad_norm": 2.7575254440307617, + "learning_rate": 3.884471468662301e-06, + "loss": 1.5241, + "step": 730 + }, + { + "epoch": 0.9217171717171717, + "eval_loss": 1.7294232845306396, + "eval_runtime": 37.1941, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 730 + }, + { + "epoch": 0.922979797979798, + "grad_norm": 2.7057321071624756, + "learning_rate": 3.882132834424696e-06, + "loss": 1.5498, + "step": 731 + }, + { + "epoch": 0.9242424242424242, + "grad_norm": 2.7679436206817627, + "learning_rate": 3.879794200187091e-06, + "loss": 1.563, + "step": 732 + }, + { + "epoch": 0.9255050505050505, + "grad_norm": 2.7258687019348145, + "learning_rate": 3.877455565949485e-06, + "loss": 1.5016, + "step": 733 + }, + { + "epoch": 0.9267676767676768, + "grad_norm": 2.8474221229553223, + "learning_rate": 3.8751169317118804e-06, + "loss": 1.5318, + "step": 734 + }, + { + "epoch": 0.928030303030303, + "grad_norm": 2.7718396186828613, + "learning_rate": 3.872778297474276e-06, + "loss": 1.5919, + "step": 735 + }, + { + "epoch": 0.9292929292929293, + "grad_norm": 2.7785022258758545, + "learning_rate": 3.87043966323667e-06, + "loss": 1.5399, + "step": 736 + }, + { + "epoch": 0.9305555555555556, + "grad_norm": 2.742476463317871, + "learning_rate": 3.868101028999065e-06, + "loss": 1.5428, + "step": 737 + }, + { + "epoch": 0.9318181818181818, + "grad_norm": 2.7050726413726807, + "learning_rate": 3.86576239476146e-06, + "loss": 1.5604, + "step": 738 + }, + { + "epoch": 0.9330808080808081, + "grad_norm": 2.6755242347717285, + "learning_rate": 3.863423760523854e-06, + "loss": 1.5409, + "step": 739 + }, + { + "epoch": 0.9343434343434344, + "grad_norm": 2.6947829723358154, + "learning_rate": 3.8610851262862495e-06, + "loss": 1.5537, + "step": 740 + }, + { + "epoch": 0.9343434343434344, + "eval_loss": 1.7396037578582764, + "eval_runtime": 37.1912, + "eval_samples_per_second": 26.888, + "eval_steps_per_second": 1.129, + "step": 740 + }, + { + "epoch": 0.9356060606060606, + "grad_norm": 2.729816436767578, + "learning_rate": 3.858746492048644e-06, + "loss": 1.5581, + "step": 741 + }, + { + "epoch": 0.9368686868686869, + "grad_norm": 2.6951887607574463, + "learning_rate": 3.856407857811038e-06, + "loss": 1.5929, + "step": 742 + }, + { + "epoch": 0.9381313131313131, + "grad_norm": 2.6601064205169678, + "learning_rate": 3.854069223573434e-06, + "loss": 1.4881, + "step": 743 + }, + { + "epoch": 0.9393939393939394, + "grad_norm": 2.623898983001709, + "learning_rate": 3.851730589335828e-06, + "loss": 1.5499, + "step": 744 + }, + { + "epoch": 0.9406565656565656, + "grad_norm": 2.703871250152588, + "learning_rate": 3.8493919550982224e-06, + "loss": 1.5585, + "step": 745 + }, + { + "epoch": 0.9419191919191919, + "grad_norm": 2.674893379211426, + "learning_rate": 3.847053320860618e-06, + "loss": 1.5586, + "step": 746 + }, + { + "epoch": 0.9431818181818182, + "grad_norm": 2.8353004455566406, + "learning_rate": 3.844714686623012e-06, + "loss": 1.5937, + "step": 747 + }, + { + "epoch": 0.9444444444444444, + "grad_norm": 2.68009614944458, + "learning_rate": 3.842376052385407e-06, + "loss": 1.5192, + "step": 748 + }, + { + "epoch": 0.9457070707070707, + "grad_norm": 2.6847407817840576, + "learning_rate": 3.840037418147802e-06, + "loss": 1.512, + "step": 749 + }, + { + "epoch": 0.946969696969697, + "grad_norm": 2.624290943145752, + "learning_rate": 3.837698783910197e-06, + "loss": 1.5348, + "step": 750 + }, + { + "epoch": 0.946969696969697, + "eval_loss": 1.738332748413086, + "eval_runtime": 37.1782, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 750 + }, + { + "epoch": 0.9482323232323232, + "grad_norm": 2.7467777729034424, + "learning_rate": 3.8353601496725915e-06, + "loss": 1.5603, + "step": 751 + }, + { + "epoch": 0.9494949494949495, + "grad_norm": 2.851987600326538, + "learning_rate": 3.833021515434987e-06, + "loss": 1.5974, + "step": 752 + }, + { + "epoch": 0.9507575757575758, + "grad_norm": 2.6642227172851562, + "learning_rate": 3.830682881197381e-06, + "loss": 1.5138, + "step": 753 + }, + { + "epoch": 0.952020202020202, + "grad_norm": 2.7351491451263428, + "learning_rate": 3.828344246959776e-06, + "loss": 1.5656, + "step": 754 + }, + { + "epoch": 0.9532828282828283, + "grad_norm": 2.593748092651367, + "learning_rate": 3.826005612722171e-06, + "loss": 1.5395, + "step": 755 + }, + { + "epoch": 0.9545454545454546, + "grad_norm": 2.639180898666382, + "learning_rate": 3.823666978484565e-06, + "loss": 1.5349, + "step": 756 + }, + { + "epoch": 0.9558080808080808, + "grad_norm": 2.805148124694824, + "learning_rate": 3.82132834424696e-06, + "loss": 1.5805, + "step": 757 + }, + { + "epoch": 0.9570707070707071, + "grad_norm": 2.5619425773620605, + "learning_rate": 3.818989710009355e-06, + "loss": 1.5389, + "step": 758 + }, + { + "epoch": 0.9583333333333334, + "grad_norm": 2.7224011421203613, + "learning_rate": 3.816651075771749e-06, + "loss": 1.5316, + "step": 759 + }, + { + "epoch": 0.9595959595959596, + "grad_norm": 2.4667656421661377, + "learning_rate": 3.8143124415341442e-06, + "loss": 1.4942, + "step": 760 + }, + { + "epoch": 0.9595959595959596, + "eval_loss": 1.7205606698989868, + "eval_runtime": 37.1654, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.9608585858585859, + "grad_norm": 2.456643581390381, + "learning_rate": 3.811973807296539e-06, + "loss": 1.538, + "step": 761 + }, + { + "epoch": 0.9621212121212122, + "grad_norm": 2.696720838546753, + "learning_rate": 3.8096351730589335e-06, + "loss": 1.5462, + "step": 762 + }, + { + "epoch": 0.9633838383838383, + "grad_norm": 2.7314131259918213, + "learning_rate": 3.8072965388213284e-06, + "loss": 1.5931, + "step": 763 + }, + { + "epoch": 0.9646464646464646, + "grad_norm": 2.616532325744629, + "learning_rate": 3.804957904583723e-06, + "loss": 1.5469, + "step": 764 + }, + { + "epoch": 0.9659090909090909, + "grad_norm": 2.7471885681152344, + "learning_rate": 3.8026192703461185e-06, + "loss": 1.5431, + "step": 765 + }, + { + "epoch": 0.9671717171717171, + "grad_norm": 2.7478253841400146, + "learning_rate": 3.8002806361085133e-06, + "loss": 1.5452, + "step": 766 + }, + { + "epoch": 0.9684343434343434, + "grad_norm": 2.543790340423584, + "learning_rate": 3.7979420018709077e-06, + "loss": 1.6091, + "step": 767 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.6169509887695312, + "learning_rate": 3.7956033676333026e-06, + "loss": 1.5557, + "step": 768 + }, + { + "epoch": 0.9709595959595959, + "grad_norm": 2.4768588542938232, + "learning_rate": 3.7932647333956974e-06, + "loss": 1.5247, + "step": 769 + }, + { + "epoch": 0.9722222222222222, + "grad_norm": 2.654648780822754, + "learning_rate": 3.7909260991580923e-06, + "loss": 1.5281, + "step": 770 + }, + { + "epoch": 0.9722222222222222, + "eval_loss": 1.7289395332336426, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 770 + }, + { + "epoch": 0.9734848484848485, + "grad_norm": 2.893073081970215, + "learning_rate": 3.7885874649204867e-06, + "loss": 1.5749, + "step": 771 + }, + { + "epoch": 0.9747474747474747, + "grad_norm": 2.696341037750244, + "learning_rate": 3.7862488306828815e-06, + "loss": 1.5025, + "step": 772 + }, + { + "epoch": 0.976010101010101, + "grad_norm": 2.678222417831421, + "learning_rate": 3.7839101964452764e-06, + "loss": 1.538, + "step": 773 + }, + { + "epoch": 0.9772727272727273, + "grad_norm": 2.7772746086120605, + "learning_rate": 3.7815715622076708e-06, + "loss": 1.5648, + "step": 774 + }, + { + "epoch": 0.9785353535353535, + "grad_norm": 3.076620101928711, + "learning_rate": 3.7792329279700656e-06, + "loss": 1.6114, + "step": 775 + }, + { + "epoch": 0.9797979797979798, + "grad_norm": 2.700056791305542, + "learning_rate": 3.7768942937324605e-06, + "loss": 1.5881, + "step": 776 + }, + { + "epoch": 0.9810606060606061, + "grad_norm": 2.601419687271118, + "learning_rate": 3.7745556594948553e-06, + "loss": 1.5319, + "step": 777 + }, + { + "epoch": 0.9823232323232324, + "grad_norm": 2.6664950847625732, + "learning_rate": 3.7722170252572497e-06, + "loss": 1.5497, + "step": 778 + }, + { + "epoch": 0.9835858585858586, + "grad_norm": 2.7374589443206787, + "learning_rate": 3.7698783910196446e-06, + "loss": 1.5488, + "step": 779 + }, + { + "epoch": 0.9848484848484849, + "grad_norm": 2.797938823699951, + "learning_rate": 3.76753975678204e-06, + "loss": 1.5376, + "step": 780 + }, + { + "epoch": 0.9848484848484849, + "eval_loss": 1.7334709167480469, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 780 + }, + { + "epoch": 0.9861111111111112, + "grad_norm": 2.881591796875, + "learning_rate": 3.7652011225444347e-06, + "loss": 1.5405, + "step": 781 + }, + { + "epoch": 0.9873737373737373, + "grad_norm": 2.6965644359588623, + "learning_rate": 3.7628624883068295e-06, + "loss": 1.5066, + "step": 782 + }, + { + "epoch": 0.9886363636363636, + "grad_norm": 2.7286899089813232, + "learning_rate": 3.760523854069224e-06, + "loss": 1.5277, + "step": 783 + }, + { + "epoch": 0.98989898989899, + "grad_norm": 2.795644998550415, + "learning_rate": 3.7581852198316188e-06, + "loss": 1.5335, + "step": 784 + }, + { + "epoch": 0.9911616161616161, + "grad_norm": 2.8160440921783447, + "learning_rate": 3.7558465855940136e-06, + "loss": 1.585, + "step": 785 + }, + { + "epoch": 0.9924242424242424, + "grad_norm": 2.7984936237335205, + "learning_rate": 3.753507951356408e-06, + "loss": 1.5569, + "step": 786 + }, + { + "epoch": 0.9936868686868687, + "grad_norm": 2.7596278190612793, + "learning_rate": 3.751169317118803e-06, + "loss": 1.4838, + "step": 787 + }, + { + "epoch": 0.9949494949494949, + "grad_norm": 2.666383743286133, + "learning_rate": 3.7488306828811977e-06, + "loss": 1.5656, + "step": 788 + }, + { + "epoch": 0.9962121212121212, + "grad_norm": 2.8139331340789795, + "learning_rate": 3.7464920486435926e-06, + "loss": 1.5753, + "step": 789 + }, + { + "epoch": 0.9974747474747475, + "grad_norm": 2.7714269161224365, + "learning_rate": 3.744153414405987e-06, + "loss": 1.5187, + "step": 790 + }, + { + "epoch": 0.9974747474747475, + "eval_loss": 1.7257990837097168, + "eval_runtime": 37.1065, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 790 + }, + { + "epoch": 0.9987373737373737, + "grad_norm": 2.697606325149536, + "learning_rate": 3.741814780168382e-06, + "loss": 1.56, + "step": 791 + }, + { + "epoch": 1.0, + "grad_norm": 2.7685229778289795, + "learning_rate": 3.7394761459307767e-06, + "loss": 1.5705, + "step": 792 + }, + { + "epoch": 1.0012626262626263, + "grad_norm": 2.778200626373291, + "learning_rate": 3.737137511693171e-06, + "loss": 1.5288, + "step": 793 + }, + { + "epoch": 1.0025252525252526, + "grad_norm": 2.7211337089538574, + "learning_rate": 3.734798877455566e-06, + "loss": 1.5348, + "step": 794 + }, + { + "epoch": 1.003787878787879, + "grad_norm": 2.690739154815674, + "learning_rate": 3.7324602432179608e-06, + "loss": 1.4734, + "step": 795 + }, + { + "epoch": 1.005050505050505, + "grad_norm": 2.6809420585632324, + "learning_rate": 3.730121608980356e-06, + "loss": 1.4852, + "step": 796 + }, + { + "epoch": 1.0063131313131313, + "grad_norm": 2.8550117015838623, + "learning_rate": 3.727782974742751e-06, + "loss": 1.4562, + "step": 797 + }, + { + "epoch": 1.0075757575757576, + "grad_norm": 2.713935613632202, + "learning_rate": 3.7254443405051453e-06, + "loss": 1.5005, + "step": 798 + }, + { + "epoch": 1.0088383838383839, + "grad_norm": 2.668363332748413, + "learning_rate": 3.72310570626754e-06, + "loss": 1.485, + "step": 799 + }, + { + "epoch": 1.0101010101010102, + "grad_norm": 2.840769052505493, + "learning_rate": 3.720767072029935e-06, + "loss": 1.4531, + "step": 800 + }, + { + "epoch": 1.0101010101010102, + "eval_loss": 1.725937008857727, + "eval_runtime": 37.2471, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 800 + }, + { + "epoch": 1.0113636363636365, + "grad_norm": 2.7345798015594482, + "learning_rate": 3.71842843779233e-06, + "loss": 1.5091, + "step": 801 + }, + { + "epoch": 1.0126262626262625, + "grad_norm": 2.9052624702453613, + "learning_rate": 3.7160898035547243e-06, + "loss": 1.5039, + "step": 802 + }, + { + "epoch": 1.0138888888888888, + "grad_norm": 2.893073558807373, + "learning_rate": 3.713751169317119e-06, + "loss": 1.4613, + "step": 803 + }, + { + "epoch": 1.0151515151515151, + "grad_norm": 2.8505427837371826, + "learning_rate": 3.711412535079514e-06, + "loss": 1.4932, + "step": 804 + }, + { + "epoch": 1.0164141414141414, + "grad_norm": 2.9065170288085938, + "learning_rate": 3.7090739008419084e-06, + "loss": 1.5082, + "step": 805 + }, + { + "epoch": 1.0176767676767677, + "grad_norm": 2.888068675994873, + "learning_rate": 3.706735266604303e-06, + "loss": 1.5675, + "step": 806 + }, + { + "epoch": 1.018939393939394, + "grad_norm": 2.7325260639190674, + "learning_rate": 3.704396632366698e-06, + "loss": 1.4649, + "step": 807 + }, + { + "epoch": 1.02020202020202, + "grad_norm": 2.8861136436462402, + "learning_rate": 3.702057998129093e-06, + "loss": 1.4517, + "step": 808 + }, + { + "epoch": 1.0214646464646464, + "grad_norm": 2.8837692737579346, + "learning_rate": 3.6997193638914873e-06, + "loss": 1.5198, + "step": 809 + }, + { + "epoch": 1.0227272727272727, + "grad_norm": 2.790330410003662, + "learning_rate": 3.697380729653882e-06, + "loss": 1.5066, + "step": 810 + }, + { + "epoch": 1.0227272727272727, + "eval_loss": 1.7269084453582764, + "eval_runtime": 36.89, + "eval_samples_per_second": 27.108, + "eval_steps_per_second": 1.139, + "step": 810 + }, + { + "epoch": 1.023989898989899, + "grad_norm": 3.03643536567688, + "learning_rate": 3.6950420954162774e-06, + "loss": 1.4928, + "step": 811 + }, + { + "epoch": 1.0252525252525253, + "grad_norm": 2.841055154800415, + "learning_rate": 3.6927034611786723e-06, + "loss": 1.5273, + "step": 812 + }, + { + "epoch": 1.0265151515151516, + "grad_norm": 2.84134578704834, + "learning_rate": 3.690364826941067e-06, + "loss": 1.5443, + "step": 813 + }, + { + "epoch": 1.0277777777777777, + "grad_norm": 2.837758779525757, + "learning_rate": 3.6880261927034615e-06, + "loss": 1.5586, + "step": 814 + }, + { + "epoch": 1.029040404040404, + "grad_norm": 2.731412410736084, + "learning_rate": 3.6856875584658564e-06, + "loss": 1.4465, + "step": 815 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.8979599475860596, + "learning_rate": 3.683348924228251e-06, + "loss": 1.5663, + "step": 816 + }, + { + "epoch": 1.0315656565656566, + "grad_norm": 2.7740373611450195, + "learning_rate": 3.6810102899906456e-06, + "loss": 1.4578, + "step": 817 + }, + { + "epoch": 1.0328282828282829, + "grad_norm": 2.8984005451202393, + "learning_rate": 3.6786716557530405e-06, + "loss": 1.493, + "step": 818 + }, + { + "epoch": 1.0340909090909092, + "grad_norm": 2.880680561065674, + "learning_rate": 3.6763330215154353e-06, + "loss": 1.4858, + "step": 819 + }, + { + "epoch": 1.0353535353535352, + "grad_norm": 3.0220961570739746, + "learning_rate": 3.67399438727783e-06, + "loss": 1.5009, + "step": 820 + }, + { + "epoch": 1.0353535353535352, + "eval_loss": 1.7245674133300781, + "eval_runtime": 37.0235, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 1.0366161616161615, + "grad_norm": 2.7905962467193604, + "learning_rate": 3.6716557530402246e-06, + "loss": 1.4728, + "step": 821 + }, + { + "epoch": 1.0378787878787878, + "grad_norm": 2.856935501098633, + "learning_rate": 3.6693171188026194e-06, + "loss": 1.521, + "step": 822 + }, + { + "epoch": 1.0391414141414141, + "grad_norm": 2.956584930419922, + "learning_rate": 3.6669784845650143e-06, + "loss": 1.4725, + "step": 823 + }, + { + "epoch": 1.0404040404040404, + "grad_norm": 3.130960702896118, + "learning_rate": 3.6646398503274087e-06, + "loss": 1.6127, + "step": 824 + }, + { + "epoch": 1.0416666666666667, + "grad_norm": 2.8855528831481934, + "learning_rate": 3.6623012160898035e-06, + "loss": 1.5078, + "step": 825 + }, + { + "epoch": 1.0429292929292928, + "grad_norm": 2.8030972480773926, + "learning_rate": 3.659962581852199e-06, + "loss": 1.4635, + "step": 826 + }, + { + "epoch": 1.0441919191919191, + "grad_norm": 3.061593770980835, + "learning_rate": 3.6576239476145936e-06, + "loss": 1.4944, + "step": 827 + }, + { + "epoch": 1.0454545454545454, + "grad_norm": 2.9983572959899902, + "learning_rate": 3.6552853133769885e-06, + "loss": 1.505, + "step": 828 + }, + { + "epoch": 1.0467171717171717, + "grad_norm": 2.7681095600128174, + "learning_rate": 3.652946679139383e-06, + "loss": 1.4774, + "step": 829 + }, + { + "epoch": 1.047979797979798, + "grad_norm": 3.0530946254730225, + "learning_rate": 3.6506080449017777e-06, + "loss": 1.4676, + "step": 830 + }, + { + "epoch": 1.047979797979798, + "eval_loss": 1.7242811918258667, + "eval_runtime": 37.0752, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 830 + }, + { + "epoch": 1.0492424242424243, + "grad_norm": 2.8383126258850098, + "learning_rate": 3.6482694106641726e-06, + "loss": 1.486, + "step": 831 + }, + { + "epoch": 1.0505050505050506, + "grad_norm": 3.095928907394409, + "learning_rate": 3.6459307764265674e-06, + "loss": 1.5331, + "step": 832 + }, + { + "epoch": 1.0517676767676767, + "grad_norm": 2.9225456714630127, + "learning_rate": 3.643592142188962e-06, + "loss": 1.556, + "step": 833 + }, + { + "epoch": 1.053030303030303, + "grad_norm": 2.9037210941314697, + "learning_rate": 3.6412535079513567e-06, + "loss": 1.4834, + "step": 834 + }, + { + "epoch": 1.0542929292929293, + "grad_norm": 2.915022373199463, + "learning_rate": 3.6389148737137515e-06, + "loss": 1.5346, + "step": 835 + }, + { + "epoch": 1.0555555555555556, + "grad_norm": 3.250411033630371, + "learning_rate": 3.636576239476146e-06, + "loss": 1.4885, + "step": 836 + }, + { + "epoch": 1.0568181818181819, + "grad_norm": 2.759061336517334, + "learning_rate": 3.634237605238541e-06, + "loss": 1.456, + "step": 837 + }, + { + "epoch": 1.0580808080808082, + "grad_norm": 2.955981731414795, + "learning_rate": 3.6318989710009356e-06, + "loss": 1.4889, + "step": 838 + }, + { + "epoch": 1.0593434343434343, + "grad_norm": 3.243338108062744, + "learning_rate": 3.6295603367633305e-06, + "loss": 1.5207, + "step": 839 + }, + { + "epoch": 1.0606060606060606, + "grad_norm": 3.075840950012207, + "learning_rate": 3.627221702525725e-06, + "loss": 1.4906, + "step": 840 + }, + { + "epoch": 1.0606060606060606, + "eval_loss": 1.7290290594100952, + "eval_runtime": 37.097, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 840 + }, + { + "epoch": 1.0618686868686869, + "grad_norm": 3.149125099182129, + "learning_rate": 3.62488306828812e-06, + "loss": 1.4755, + "step": 841 + }, + { + "epoch": 1.0631313131313131, + "grad_norm": 3.099318504333496, + "learning_rate": 3.622544434050515e-06, + "loss": 1.534, + "step": 842 + }, + { + "epoch": 1.0643939393939394, + "grad_norm": 2.8942148685455322, + "learning_rate": 3.62020579981291e-06, + "loss": 1.4844, + "step": 843 + }, + { + "epoch": 1.0656565656565657, + "grad_norm": 3.142664670944214, + "learning_rate": 3.6178671655753047e-06, + "loss": 1.5179, + "step": 844 + }, + { + "epoch": 1.0669191919191918, + "grad_norm": 3.106552839279175, + "learning_rate": 3.615528531337699e-06, + "loss": 1.4667, + "step": 845 + }, + { + "epoch": 1.0681818181818181, + "grad_norm": 2.8507936000823975, + "learning_rate": 3.613189897100094e-06, + "loss": 1.4463, + "step": 846 + }, + { + "epoch": 1.0694444444444444, + "grad_norm": 3.2198901176452637, + "learning_rate": 3.610851262862489e-06, + "loss": 1.5146, + "step": 847 + }, + { + "epoch": 1.0707070707070707, + "grad_norm": 3.146388292312622, + "learning_rate": 3.6085126286248832e-06, + "loss": 1.4927, + "step": 848 + }, + { + "epoch": 1.071969696969697, + "grad_norm": 2.8304760456085205, + "learning_rate": 3.606173994387278e-06, + "loss": 1.4971, + "step": 849 + }, + { + "epoch": 1.0732323232323233, + "grad_norm": 3.1714529991149902, + "learning_rate": 3.603835360149673e-06, + "loss": 1.5418, + "step": 850 + }, + { + "epoch": 1.0732323232323233, + "eval_loss": 1.725188970565796, + "eval_runtime": 37.117, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 850 + }, + { + "epoch": 1.0744949494949494, + "grad_norm": 3.06162428855896, + "learning_rate": 3.6014967259120677e-06, + "loss": 1.4959, + "step": 851 + }, + { + "epoch": 1.0757575757575757, + "grad_norm": 2.812206268310547, + "learning_rate": 3.599158091674462e-06, + "loss": 1.5249, + "step": 852 + }, + { + "epoch": 1.077020202020202, + "grad_norm": 3.236210584640503, + "learning_rate": 3.596819457436857e-06, + "loss": 1.5166, + "step": 853 + }, + { + "epoch": 1.0782828282828283, + "grad_norm": 2.8397765159606934, + "learning_rate": 3.594480823199252e-06, + "loss": 1.5043, + "step": 854 + }, + { + "epoch": 1.0795454545454546, + "grad_norm": 2.9215919971466064, + "learning_rate": 3.5921421889616463e-06, + "loss": 1.5235, + "step": 855 + }, + { + "epoch": 1.0808080808080809, + "grad_norm": 2.8219058513641357, + "learning_rate": 3.589803554724041e-06, + "loss": 1.5091, + "step": 856 + }, + { + "epoch": 1.0820707070707072, + "grad_norm": 2.9344675540924072, + "learning_rate": 3.5874649204864364e-06, + "loss": 1.5355, + "step": 857 + }, + { + "epoch": 1.0833333333333333, + "grad_norm": 2.949094533920288, + "learning_rate": 3.5851262862488312e-06, + "loss": 1.5273, + "step": 858 + }, + { + "epoch": 1.0845959595959596, + "grad_norm": 2.9801411628723145, + "learning_rate": 3.582787652011226e-06, + "loss": 1.5078, + "step": 859 + }, + { + "epoch": 1.0858585858585859, + "grad_norm": 2.802182912826538, + "learning_rate": 3.5804490177736205e-06, + "loss": 1.5015, + "step": 860 + }, + { + "epoch": 1.0858585858585859, + "eval_loss": 1.7209502458572388, + "eval_runtime": 37.1238, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 1.0871212121212122, + "grad_norm": 2.857457160949707, + "learning_rate": 3.5781103835360153e-06, + "loss": 1.5057, + "step": 861 + }, + { + "epoch": 1.0883838383838385, + "grad_norm": 2.794478178024292, + "learning_rate": 3.57577174929841e-06, + "loss": 1.4594, + "step": 862 + }, + { + "epoch": 1.0896464646464645, + "grad_norm": 3.0028598308563232, + "learning_rate": 3.573433115060805e-06, + "loss": 1.4666, + "step": 863 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.7701406478881836, + "learning_rate": 3.5710944808231994e-06, + "loss": 1.4596, + "step": 864 + }, + { + "epoch": 1.0921717171717171, + "grad_norm": 2.9705960750579834, + "learning_rate": 3.5687558465855943e-06, + "loss": 1.5092, + "step": 865 + }, + { + "epoch": 1.0934343434343434, + "grad_norm": 2.9820821285247803, + "learning_rate": 3.566417212347989e-06, + "loss": 1.5246, + "step": 866 + }, + { + "epoch": 1.0946969696969697, + "grad_norm": 2.988356113433838, + "learning_rate": 3.5640785781103835e-06, + "loss": 1.5035, + "step": 867 + }, + { + "epoch": 1.095959595959596, + "grad_norm": 2.901599884033203, + "learning_rate": 3.5617399438727784e-06, + "loss": 1.5097, + "step": 868 + }, + { + "epoch": 1.0972222222222223, + "grad_norm": 2.9128799438476562, + "learning_rate": 3.5594013096351732e-06, + "loss": 1.5424, + "step": 869 + }, + { + "epoch": 1.0984848484848484, + "grad_norm": 2.719289779663086, + "learning_rate": 3.557062675397568e-06, + "loss": 1.4313, + "step": 870 + }, + { + "epoch": 1.0984848484848484, + "eval_loss": 1.7159698009490967, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 870 + }, + { + "epoch": 1.0997474747474747, + "grad_norm": 3.0080976486206055, + "learning_rate": 3.5547240411599625e-06, + "loss": 1.4483, + "step": 871 + }, + { + "epoch": 1.101010101010101, + "grad_norm": 2.856523036956787, + "learning_rate": 3.5523854069223578e-06, + "loss": 1.5463, + "step": 872 + }, + { + "epoch": 1.1022727272727273, + "grad_norm": 2.8585214614868164, + "learning_rate": 3.5500467726847526e-06, + "loss": 1.5391, + "step": 873 + }, + { + "epoch": 1.1035353535353536, + "grad_norm": 2.8440098762512207, + "learning_rate": 3.5477081384471474e-06, + "loss": 1.499, + "step": 874 + }, + { + "epoch": 1.10479797979798, + "grad_norm": 2.8114542961120605, + "learning_rate": 3.545369504209542e-06, + "loss": 1.4691, + "step": 875 + }, + { + "epoch": 1.106060606060606, + "grad_norm": 2.980273962020874, + "learning_rate": 3.5430308699719367e-06, + "loss": 1.4541, + "step": 876 + }, + { + "epoch": 1.1073232323232323, + "grad_norm": 3.0053417682647705, + "learning_rate": 3.5406922357343315e-06, + "loss": 1.4895, + "step": 877 + }, + { + "epoch": 1.1085858585858586, + "grad_norm": 2.924046754837036, + "learning_rate": 3.5383536014967264e-06, + "loss": 1.5186, + "step": 878 + }, + { + "epoch": 1.1098484848484849, + "grad_norm": 2.969461679458618, + "learning_rate": 3.536014967259121e-06, + "loss": 1.5213, + "step": 879 + }, + { + "epoch": 1.1111111111111112, + "grad_norm": 2.8936445713043213, + "learning_rate": 3.5336763330215156e-06, + "loss": 1.5585, + "step": 880 + }, + { + "epoch": 1.1111111111111112, + "eval_loss": 1.7150688171386719, + "eval_runtime": 37.1105, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 1.1123737373737375, + "grad_norm": 2.992034435272217, + "learning_rate": 3.5313376987839105e-06, + "loss": 1.5264, + "step": 881 + }, + { + "epoch": 1.1136363636363635, + "grad_norm": 3.0453898906707764, + "learning_rate": 3.5289990645463053e-06, + "loss": 1.5482, + "step": 882 + }, + { + "epoch": 1.1148989898989898, + "grad_norm": 2.7480897903442383, + "learning_rate": 3.5266604303086998e-06, + "loss": 1.4851, + "step": 883 + }, + { + "epoch": 1.1161616161616161, + "grad_norm": 2.9408013820648193, + "learning_rate": 3.5243217960710946e-06, + "loss": 1.536, + "step": 884 + }, + { + "epoch": 1.1174242424242424, + "grad_norm": 2.854438543319702, + "learning_rate": 3.5219831618334894e-06, + "loss": 1.5814, + "step": 885 + }, + { + "epoch": 1.1186868686868687, + "grad_norm": 2.9060397148132324, + "learning_rate": 3.519644527595884e-06, + "loss": 1.5019, + "step": 886 + }, + { + "epoch": 1.119949494949495, + "grad_norm": 2.8741681575775146, + "learning_rate": 3.517305893358279e-06, + "loss": 1.4713, + "step": 887 + }, + { + "epoch": 1.121212121212121, + "grad_norm": 2.972510576248169, + "learning_rate": 3.514967259120674e-06, + "loss": 1.5104, + "step": 888 + }, + { + "epoch": 1.1224747474747474, + "grad_norm": 2.8113558292388916, + "learning_rate": 3.512628624883069e-06, + "loss": 1.4973, + "step": 889 + }, + { + "epoch": 1.1237373737373737, + "grad_norm": 2.9687416553497314, + "learning_rate": 3.5102899906454637e-06, + "loss": 1.4774, + "step": 890 + }, + { + "epoch": 1.1237373737373737, + "eval_loss": 1.7169743776321411, + "eval_runtime": 37.0911, + "eval_samples_per_second": 26.961, + "eval_steps_per_second": 1.132, + "step": 890 + }, + { + "epoch": 1.125, + "grad_norm": 2.911409854888916, + "learning_rate": 3.507951356407858e-06, + "loss": 1.4536, + "step": 891 + }, + { + "epoch": 1.1262626262626263, + "grad_norm": 2.8209526538848877, + "learning_rate": 3.505612722170253e-06, + "loss": 1.5357, + "step": 892 + }, + { + "epoch": 1.1275252525252526, + "grad_norm": 2.811971426010132, + "learning_rate": 3.5032740879326478e-06, + "loss": 1.4889, + "step": 893 + }, + { + "epoch": 1.128787878787879, + "grad_norm": 2.969878673553467, + "learning_rate": 3.5009354536950426e-06, + "loss": 1.4758, + "step": 894 + }, + { + "epoch": 1.130050505050505, + "grad_norm": 2.858635425567627, + "learning_rate": 3.498596819457437e-06, + "loss": 1.5355, + "step": 895 + }, + { + "epoch": 1.1313131313131313, + "grad_norm": 3.018164873123169, + "learning_rate": 3.496258185219832e-06, + "loss": 1.5421, + "step": 896 + }, + { + "epoch": 1.1325757575757576, + "grad_norm": 2.846277952194214, + "learning_rate": 3.4939195509822267e-06, + "loss": 1.4936, + "step": 897 + }, + { + "epoch": 1.1338383838383839, + "grad_norm": 2.965245246887207, + "learning_rate": 3.491580916744621e-06, + "loss": 1.4903, + "step": 898 + }, + { + "epoch": 1.1351010101010102, + "grad_norm": 2.8500895500183105, + "learning_rate": 3.489242282507016e-06, + "loss": 1.5152, + "step": 899 + }, + { + "epoch": 1.1363636363636362, + "grad_norm": 2.8242156505584717, + "learning_rate": 3.486903648269411e-06, + "loss": 1.5531, + "step": 900 + }, + { + "epoch": 1.1363636363636362, + "eval_loss": 1.7180876731872559, + "eval_runtime": 37.4145, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 900 + }, + { + "epoch": 1.1376262626262625, + "grad_norm": 2.8846635818481445, + "learning_rate": 3.4845650140318057e-06, + "loss": 1.5314, + "step": 901 + }, + { + "epoch": 1.1388888888888888, + "grad_norm": 2.9443984031677246, + "learning_rate": 3.4822263797942e-06, + "loss": 1.4699, + "step": 902 + }, + { + "epoch": 1.1401515151515151, + "grad_norm": 2.7435739040374756, + "learning_rate": 3.4798877455565953e-06, + "loss": 1.4949, + "step": 903 + }, + { + "epoch": 1.1414141414141414, + "grad_norm": 2.979102373123169, + "learning_rate": 3.47754911131899e-06, + "loss": 1.5689, + "step": 904 + }, + { + "epoch": 1.1426767676767677, + "grad_norm": 2.9814586639404297, + "learning_rate": 3.475210477081385e-06, + "loss": 1.5183, + "step": 905 + }, + { + "epoch": 1.143939393939394, + "grad_norm": 2.8380966186523438, + "learning_rate": 3.4728718428437794e-06, + "loss": 1.5177, + "step": 906 + }, + { + "epoch": 1.14520202020202, + "grad_norm": 2.8478410243988037, + "learning_rate": 3.4705332086061743e-06, + "loss": 1.4787, + "step": 907 + }, + { + "epoch": 1.1464646464646464, + "grad_norm": 2.8247780799865723, + "learning_rate": 3.468194574368569e-06, + "loss": 1.425, + "step": 908 + }, + { + "epoch": 1.1477272727272727, + "grad_norm": 2.869649887084961, + "learning_rate": 3.465855940130964e-06, + "loss": 1.4619, + "step": 909 + }, + { + "epoch": 1.148989898989899, + "grad_norm": 2.9439175128936768, + "learning_rate": 3.4635173058933584e-06, + "loss": 1.5194, + "step": 910 + }, + { + "epoch": 1.148989898989899, + "eval_loss": 1.7227450609207153, + "eval_runtime": 36.9723, + "eval_samples_per_second": 27.047, + "eval_steps_per_second": 1.136, + "step": 910 + }, + { + "epoch": 1.1502525252525253, + "grad_norm": 3.0219292640686035, + "learning_rate": 3.4611786716557532e-06, + "loss": 1.5878, + "step": 911 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.988475799560547, + "learning_rate": 3.458840037418148e-06, + "loss": 1.5349, + "step": 912 + }, + { + "epoch": 1.1527777777777777, + "grad_norm": 2.708740234375, + "learning_rate": 3.456501403180543e-06, + "loss": 1.4637, + "step": 913 + }, + { + "epoch": 1.154040404040404, + "grad_norm": 3.0844545364379883, + "learning_rate": 3.4541627689429373e-06, + "loss": 1.546, + "step": 914 + }, + { + "epoch": 1.1553030303030303, + "grad_norm": 2.800487518310547, + "learning_rate": 3.451824134705332e-06, + "loss": 1.5374, + "step": 915 + }, + { + "epoch": 1.1565656565656566, + "grad_norm": 2.8701722621917725, + "learning_rate": 3.449485500467727e-06, + "loss": 1.4549, + "step": 916 + }, + { + "epoch": 1.1578282828282829, + "grad_norm": 3.036418914794922, + "learning_rate": 3.4471468662301214e-06, + "loss": 1.5253, + "step": 917 + }, + { + "epoch": 1.1590909090909092, + "grad_norm": 3.0300986766815186, + "learning_rate": 3.4448082319925167e-06, + "loss": 1.4727, + "step": 918 + }, + { + "epoch": 1.1603535353535355, + "grad_norm": 2.8652496337890625, + "learning_rate": 3.4424695977549116e-06, + "loss": 1.4842, + "step": 919 + }, + { + "epoch": 1.1616161616161615, + "grad_norm": 2.9140491485595703, + "learning_rate": 3.4401309635173064e-06, + "loss": 1.4968, + "step": 920 + }, + { + "epoch": 1.1616161616161615, + "eval_loss": 1.7184255123138428, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 920 + }, + { + "epoch": 1.1628787878787878, + "grad_norm": 2.9708127975463867, + "learning_rate": 3.4377923292797012e-06, + "loss": 1.4817, + "step": 921 + }, + { + "epoch": 1.1641414141414141, + "grad_norm": 2.954814910888672, + "learning_rate": 3.4354536950420957e-06, + "loss": 1.4937, + "step": 922 + }, + { + "epoch": 1.1654040404040404, + "grad_norm": 2.803945779800415, + "learning_rate": 3.4331150608044905e-06, + "loss": 1.527, + "step": 923 + }, + { + "epoch": 1.1666666666666667, + "grad_norm": 2.863250255584717, + "learning_rate": 3.4307764265668853e-06, + "loss": 1.4922, + "step": 924 + }, + { + "epoch": 1.1679292929292928, + "grad_norm": 2.8030712604522705, + "learning_rate": 3.4284377923292798e-06, + "loss": 1.4427, + "step": 925 + }, + { + "epoch": 1.1691919191919191, + "grad_norm": 2.944110155105591, + "learning_rate": 3.4260991580916746e-06, + "loss": 1.4435, + "step": 926 + }, + { + "epoch": 1.1704545454545454, + "grad_norm": 2.9672467708587646, + "learning_rate": 3.4237605238540695e-06, + "loss": 1.5005, + "step": 927 + }, + { + "epoch": 1.1717171717171717, + "grad_norm": 2.920205593109131, + "learning_rate": 3.4214218896164643e-06, + "loss": 1.5303, + "step": 928 + }, + { + "epoch": 1.172979797979798, + "grad_norm": 3.1062121391296387, + "learning_rate": 3.4190832553788587e-06, + "loss": 1.5395, + "step": 929 + }, + { + "epoch": 1.1742424242424243, + "grad_norm": 3.1106555461883545, + "learning_rate": 3.4167446211412536e-06, + "loss": 1.5038, + "step": 930 + }, + { + "epoch": 1.1742424242424243, + "eval_loss": 1.7127314805984497, + "eval_runtime": 37.1193, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 930 + }, + { + "epoch": 1.1755050505050506, + "grad_norm": 2.7898054122924805, + "learning_rate": 3.4144059869036484e-06, + "loss": 1.5096, + "step": 931 + }, + { + "epoch": 1.1767676767676767, + "grad_norm": 2.881694793701172, + "learning_rate": 3.4120673526660432e-06, + "loss": 1.4556, + "step": 932 + }, + { + "epoch": 1.178030303030303, + "grad_norm": 3.131551742553711, + "learning_rate": 3.4097287184284385e-06, + "loss": 1.5487, + "step": 933 + }, + { + "epoch": 1.1792929292929293, + "grad_norm": 2.9734153747558594, + "learning_rate": 3.407390084190833e-06, + "loss": 1.439, + "step": 934 + }, + { + "epoch": 1.1805555555555556, + "grad_norm": 3.009657144546509, + "learning_rate": 3.4050514499532278e-06, + "loss": 1.462, + "step": 935 + }, + { + "epoch": 1.1818181818181819, + "grad_norm": 2.8378758430480957, + "learning_rate": 3.4027128157156226e-06, + "loss": 1.5083, + "step": 936 + }, + { + "epoch": 1.1830808080808082, + "grad_norm": 2.923969268798828, + "learning_rate": 3.400374181478017e-06, + "loss": 1.5844, + "step": 937 + }, + { + "epoch": 1.1843434343434343, + "grad_norm": 3.2384307384490967, + "learning_rate": 3.398035547240412e-06, + "loss": 1.5175, + "step": 938 + }, + { + "epoch": 1.1856060606060606, + "grad_norm": 3.024385929107666, + "learning_rate": 3.3956969130028067e-06, + "loss": 1.4756, + "step": 939 + }, + { + "epoch": 1.1868686868686869, + "grad_norm": 3.088745594024658, + "learning_rate": 3.3933582787652016e-06, + "loss": 1.4914, + "step": 940 + }, + { + "epoch": 1.1868686868686869, + "eval_loss": 1.705490231513977, + "eval_runtime": 37.1291, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 940 + }, + { + "epoch": 1.1881313131313131, + "grad_norm": 2.898893117904663, + "learning_rate": 3.391019644527596e-06, + "loss": 1.5002, + "step": 941 + }, + { + "epoch": 1.1893939393939394, + "grad_norm": 2.800103187561035, + "learning_rate": 3.388681010289991e-06, + "loss": 1.4842, + "step": 942 + }, + { + "epoch": 1.1906565656565657, + "grad_norm": 3.0287230014801025, + "learning_rate": 3.3863423760523857e-06, + "loss": 1.5454, + "step": 943 + }, + { + "epoch": 1.1919191919191918, + "grad_norm": 2.9379546642303467, + "learning_rate": 3.3840037418147805e-06, + "loss": 1.5427, + "step": 944 + }, + { + "epoch": 1.1931818181818181, + "grad_norm": 2.879699468612671, + "learning_rate": 3.381665107577175e-06, + "loss": 1.4804, + "step": 945 + }, + { + "epoch": 1.1944444444444444, + "grad_norm": 2.8708608150482178, + "learning_rate": 3.3793264733395698e-06, + "loss": 1.479, + "step": 946 + }, + { + "epoch": 1.1957070707070707, + "grad_norm": 2.8513312339782715, + "learning_rate": 3.3769878391019646e-06, + "loss": 1.5343, + "step": 947 + }, + { + "epoch": 1.196969696969697, + "grad_norm": 2.8715100288391113, + "learning_rate": 3.37464920486436e-06, + "loss": 1.5114, + "step": 948 + }, + { + "epoch": 1.1982323232323233, + "grad_norm": 2.819213628768921, + "learning_rate": 3.3723105706267543e-06, + "loss": 1.4873, + "step": 949 + }, + { + "epoch": 1.1994949494949494, + "grad_norm": 2.946578025817871, + "learning_rate": 3.369971936389149e-06, + "loss": 1.5378, + "step": 950 + }, + { + "epoch": 1.1994949494949494, + "eval_loss": 1.7094330787658691, + "eval_runtime": 37.1453, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.2007575757575757, + "grad_norm": 2.8170437812805176, + "learning_rate": 3.367633302151544e-06, + "loss": 1.4416, + "step": 951 + }, + { + "epoch": 1.202020202020202, + "grad_norm": 2.9654042720794678, + "learning_rate": 3.365294667913939e-06, + "loss": 1.5115, + "step": 952 + }, + { + "epoch": 1.2032828282828283, + "grad_norm": 2.9138922691345215, + "learning_rate": 3.3629560336763333e-06, + "loss": 1.5106, + "step": 953 + }, + { + "epoch": 1.2045454545454546, + "grad_norm": 2.989961862564087, + "learning_rate": 3.360617399438728e-06, + "loss": 1.5278, + "step": 954 + }, + { + "epoch": 1.2058080808080809, + "grad_norm": 2.9620680809020996, + "learning_rate": 3.358278765201123e-06, + "loss": 1.4949, + "step": 955 + }, + { + "epoch": 1.2070707070707072, + "grad_norm": 2.945505380630493, + "learning_rate": 3.3559401309635174e-06, + "loss": 1.5018, + "step": 956 + }, + { + "epoch": 1.2083333333333333, + "grad_norm": 2.806464433670044, + "learning_rate": 3.353601496725912e-06, + "loss": 1.4951, + "step": 957 + }, + { + "epoch": 1.2095959595959596, + "grad_norm": 2.7847559452056885, + "learning_rate": 3.351262862488307e-06, + "loss": 1.4259, + "step": 958 + }, + { + "epoch": 1.2108585858585859, + "grad_norm": 2.760003089904785, + "learning_rate": 3.348924228250702e-06, + "loss": 1.4757, + "step": 959 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 3.037522792816162, + "learning_rate": 3.3465855940130963e-06, + "loss": 1.5462, + "step": 960 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.7071956396102905, + "eval_runtime": 37.1351, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.2133838383838385, + "grad_norm": 2.7755539417266846, + "learning_rate": 3.344246959775491e-06, + "loss": 1.453, + "step": 961 + }, + { + "epoch": 1.2146464646464645, + "grad_norm": 2.910369634628296, + "learning_rate": 3.341908325537886e-06, + "loss": 1.4914, + "step": 962 + }, + { + "epoch": 1.2159090909090908, + "grad_norm": 2.8999392986297607, + "learning_rate": 3.339569691300281e-06, + "loss": 1.5389, + "step": 963 + }, + { + "epoch": 1.2171717171717171, + "grad_norm": 2.891995906829834, + "learning_rate": 3.337231057062676e-06, + "loss": 1.4915, + "step": 964 + }, + { + "epoch": 1.2184343434343434, + "grad_norm": 2.814624309539795, + "learning_rate": 3.3348924228250705e-06, + "loss": 1.4936, + "step": 965 + }, + { + "epoch": 1.2196969696969697, + "grad_norm": 2.971163034439087, + "learning_rate": 3.3325537885874654e-06, + "loss": 1.4813, + "step": 966 + }, + { + "epoch": 1.220959595959596, + "grad_norm": 2.934155225753784, + "learning_rate": 3.33021515434986e-06, + "loss": 1.5162, + "step": 967 + }, + { + "epoch": 1.2222222222222223, + "grad_norm": 3.007847309112549, + "learning_rate": 3.3278765201122546e-06, + "loss": 1.4741, + "step": 968 + }, + { + "epoch": 1.2234848484848484, + "grad_norm": 2.977207660675049, + "learning_rate": 3.3255378858746495e-06, + "loss": 1.5186, + "step": 969 + }, + { + "epoch": 1.2247474747474747, + "grad_norm": 2.926567316055298, + "learning_rate": 3.3231992516370443e-06, + "loss": 1.5111, + "step": 970 + }, + { + "epoch": 1.2247474747474747, + "eval_loss": 1.7068836688995361, + "eval_runtime": 37.1419, + "eval_samples_per_second": 26.924, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 1.226010101010101, + "grad_norm": 3.063671350479126, + "learning_rate": 3.320860617399439e-06, + "loss": 1.4376, + "step": 971 + }, + { + "epoch": 1.2272727272727273, + "grad_norm": 2.7263271808624268, + "learning_rate": 3.3185219831618336e-06, + "loss": 1.4555, + "step": 972 + }, + { + "epoch": 1.2285353535353536, + "grad_norm": 2.912658929824829, + "learning_rate": 3.3161833489242284e-06, + "loss": 1.5116, + "step": 973 + }, + { + "epoch": 1.22979797979798, + "grad_norm": 3.2216858863830566, + "learning_rate": 3.3138447146866233e-06, + "loss": 1.5124, + "step": 974 + }, + { + "epoch": 1.231060606060606, + "grad_norm": 2.9138946533203125, + "learning_rate": 3.311506080449018e-06, + "loss": 1.4917, + "step": 975 + }, + { + "epoch": 1.2323232323232323, + "grad_norm": 3.0519120693206787, + "learning_rate": 3.3091674462114125e-06, + "loss": 1.4986, + "step": 976 + }, + { + "epoch": 1.2335858585858586, + "grad_norm": 2.8186402320861816, + "learning_rate": 3.3068288119738074e-06, + "loss": 1.4835, + "step": 977 + }, + { + "epoch": 1.2348484848484849, + "grad_norm": 2.799189329147339, + "learning_rate": 3.304490177736202e-06, + "loss": 1.5069, + "step": 978 + }, + { + "epoch": 1.2361111111111112, + "grad_norm": 3.010010004043579, + "learning_rate": 3.3021515434985975e-06, + "loss": 1.4276, + "step": 979 + }, + { + "epoch": 1.2373737373737375, + "grad_norm": 3.153592109680176, + "learning_rate": 3.299812909260992e-06, + "loss": 1.5459, + "step": 980 + }, + { + "epoch": 1.2373737373737375, + "eval_loss": 1.7070198059082031, + "eval_runtime": 37.1397, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 980 + }, + { + "epoch": 1.2386363636363638, + "grad_norm": 2.7818424701690674, + "learning_rate": 3.2974742750233867e-06, + "loss": 1.5045, + "step": 981 + }, + { + "epoch": 1.2398989898989898, + "grad_norm": 2.9211390018463135, + "learning_rate": 3.2951356407857816e-06, + "loss": 1.4966, + "step": 982 + }, + { + "epoch": 1.2411616161616161, + "grad_norm": 2.979743242263794, + "learning_rate": 3.2927970065481764e-06, + "loss": 1.4754, + "step": 983 + }, + { + "epoch": 1.2424242424242424, + "grad_norm": 2.8476738929748535, + "learning_rate": 3.290458372310571e-06, + "loss": 1.4619, + "step": 984 + }, + { + "epoch": 1.2436868686868687, + "grad_norm": 2.8304896354675293, + "learning_rate": 3.2881197380729657e-06, + "loss": 1.4192, + "step": 985 + }, + { + "epoch": 1.244949494949495, + "grad_norm": 3.3150880336761475, + "learning_rate": 3.2857811038353605e-06, + "loss": 1.515, + "step": 986 + }, + { + "epoch": 1.246212121212121, + "grad_norm": 3.049398422241211, + "learning_rate": 3.283442469597755e-06, + "loss": 1.5382, + "step": 987 + }, + { + "epoch": 1.2474747474747474, + "grad_norm": 3.0910539627075195, + "learning_rate": 3.2811038353601498e-06, + "loss": 1.5458, + "step": 988 + }, + { + "epoch": 1.2487373737373737, + "grad_norm": 3.002788782119751, + "learning_rate": 3.2787652011225446e-06, + "loss": 1.4668, + "step": 989 + }, + { + "epoch": 1.25, + "grad_norm": 2.9342103004455566, + "learning_rate": 3.2764265668849395e-06, + "loss": 1.5187, + "step": 990 + }, + { + "epoch": 1.25, + "eval_loss": 1.7082287073135376, + "eval_runtime": 37.1399, + "eval_samples_per_second": 26.925, + "eval_steps_per_second": 1.131, + "step": 990 + }, + { + "epoch": 1.2512626262626263, + "grad_norm": 2.7596492767333984, + "learning_rate": 3.274087932647334e-06, + "loss": 1.5191, + "step": 991 + }, + { + "epoch": 1.2525252525252526, + "grad_norm": 2.949368953704834, + "learning_rate": 3.2717492984097287e-06, + "loss": 1.4776, + "step": 992 + }, + { + "epoch": 1.253787878787879, + "grad_norm": 2.7099363803863525, + "learning_rate": 3.2694106641721236e-06, + "loss": 1.455, + "step": 993 + }, + { + "epoch": 1.255050505050505, + "grad_norm": 2.8631885051727295, + "learning_rate": 3.267072029934519e-06, + "loss": 1.4394, + "step": 994 + }, + { + "epoch": 1.2563131313131313, + "grad_norm": 3.09389591217041, + "learning_rate": 3.2647333956969137e-06, + "loss": 1.4958, + "step": 995 + }, + { + "epoch": 1.2575757575757576, + "grad_norm": 3.108271360397339, + "learning_rate": 3.262394761459308e-06, + "loss": 1.5108, + "step": 996 + }, + { + "epoch": 1.2588383838383839, + "grad_norm": 3.079176902770996, + "learning_rate": 3.260056127221703e-06, + "loss": 1.4749, + "step": 997 + }, + { + "epoch": 1.2601010101010102, + "grad_norm": 3.0458056926727295, + "learning_rate": 3.257717492984098e-06, + "loss": 1.4911, + "step": 998 + }, + { + "epoch": 1.2613636363636362, + "grad_norm": 2.952151298522949, + "learning_rate": 3.2553788587464922e-06, + "loss": 1.4842, + "step": 999 + }, + { + "epoch": 1.2626262626262625, + "grad_norm": 2.7918665409088135, + "learning_rate": 3.253040224508887e-06, + "loss": 1.4684, + "step": 1000 + }, + { + "epoch": 1.2626262626262625, + "eval_loss": 1.7100834846496582, + "eval_runtime": 37.3594, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1000 + }, + { + "epoch": 1.2638888888888888, + "grad_norm": 2.798064947128296, + "learning_rate": 3.250701590271282e-06, + "loss": 1.4295, + "step": 1001 + }, + { + "epoch": 1.2651515151515151, + "grad_norm": 2.852234363555908, + "learning_rate": 3.2483629560336767e-06, + "loss": 1.5039, + "step": 1002 + }, + { + "epoch": 1.2664141414141414, + "grad_norm": 2.762645721435547, + "learning_rate": 3.246024321796071e-06, + "loss": 1.4365, + "step": 1003 + }, + { + "epoch": 1.2676767676767677, + "grad_norm": 2.944326162338257, + "learning_rate": 3.243685687558466e-06, + "loss": 1.4849, + "step": 1004 + }, + { + "epoch": 1.268939393939394, + "grad_norm": 2.9298877716064453, + "learning_rate": 3.241347053320861e-06, + "loss": 1.4397, + "step": 1005 + }, + { + "epoch": 1.2702020202020203, + "grad_norm": 3.1259615421295166, + "learning_rate": 3.2390084190832553e-06, + "loss": 1.4819, + "step": 1006 + }, + { + "epoch": 1.2714646464646464, + "grad_norm": 3.114792823791504, + "learning_rate": 3.23666978484565e-06, + "loss": 1.4728, + "step": 1007 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.9800360202789307, + "learning_rate": 3.234331150608045e-06, + "loss": 1.4426, + "step": 1008 + }, + { + "epoch": 1.273989898989899, + "grad_norm": 2.8487257957458496, + "learning_rate": 3.2319925163704402e-06, + "loss": 1.461, + "step": 1009 + }, + { + "epoch": 1.2752525252525253, + "grad_norm": 2.800760507583618, + "learning_rate": 3.229653882132835e-06, + "loss": 1.4764, + "step": 1010 + }, + { + "epoch": 1.2752525252525253, + "eval_loss": 1.701302409172058, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1010 + }, + { + "epoch": 1.2765151515151514, + "grad_norm": 2.851012706756592, + "learning_rate": 3.2273152478952295e-06, + "loss": 1.5124, + "step": 1011 + }, + { + "epoch": 1.2777777777777777, + "grad_norm": 2.9498190879821777, + "learning_rate": 3.2249766136576243e-06, + "loss": 1.4789, + "step": 1012 + }, + { + "epoch": 1.279040404040404, + "grad_norm": 2.80710768699646, + "learning_rate": 3.222637979420019e-06, + "loss": 1.5592, + "step": 1013 + }, + { + "epoch": 1.2803030303030303, + "grad_norm": 2.9525952339172363, + "learning_rate": 3.220299345182414e-06, + "loss": 1.4941, + "step": 1014 + }, + { + "epoch": 1.2815656565656566, + "grad_norm": 2.9770407676696777, + "learning_rate": 3.2179607109448084e-06, + "loss": 1.4437, + "step": 1015 + }, + { + "epoch": 1.2828282828282829, + "grad_norm": 3.2553279399871826, + "learning_rate": 3.2156220767072033e-06, + "loss": 1.5248, + "step": 1016 + }, + { + "epoch": 1.2840909090909092, + "grad_norm": 3.0591554641723633, + "learning_rate": 3.213283442469598e-06, + "loss": 1.5704, + "step": 1017 + }, + { + "epoch": 1.2853535353535355, + "grad_norm": 2.776968002319336, + "learning_rate": 3.2109448082319925e-06, + "loss": 1.4239, + "step": 1018 + }, + { + "epoch": 1.2866161616161615, + "grad_norm": 2.8728222846984863, + "learning_rate": 3.2086061739943874e-06, + "loss": 1.5115, + "step": 1019 + }, + { + "epoch": 1.2878787878787878, + "grad_norm": 2.907308340072632, + "learning_rate": 3.2062675397567822e-06, + "loss": 1.4764, + "step": 1020 + }, + { + "epoch": 1.2878787878787878, + "eval_loss": 1.7016223669052124, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.2891414141414141, + "grad_norm": 3.0686309337615967, + "learning_rate": 3.203928905519177e-06, + "loss": 1.5175, + "step": 1021 + }, + { + "epoch": 1.2904040404040404, + "grad_norm": 2.8636863231658936, + "learning_rate": 3.2015902712815715e-06, + "loss": 1.5025, + "step": 1022 + }, + { + "epoch": 1.2916666666666667, + "grad_norm": 2.8539068698883057, + "learning_rate": 3.1992516370439663e-06, + "loss": 1.4774, + "step": 1023 + }, + { + "epoch": 1.2929292929292928, + "grad_norm": 2.9830520153045654, + "learning_rate": 3.196913002806361e-06, + "loss": 1.533, + "step": 1024 + }, + { + "epoch": 1.2941919191919191, + "grad_norm": 2.8806374073028564, + "learning_rate": 3.1945743685687564e-06, + "loss": 1.5298, + "step": 1025 + }, + { + "epoch": 1.2954545454545454, + "grad_norm": 2.932870626449585, + "learning_rate": 3.1922357343311513e-06, + "loss": 1.5028, + "step": 1026 + }, + { + "epoch": 1.2967171717171717, + "grad_norm": 2.889601945877075, + "learning_rate": 3.1898971000935457e-06, + "loss": 1.4534, + "step": 1027 + }, + { + "epoch": 1.297979797979798, + "grad_norm": 2.930546522140503, + "learning_rate": 3.1875584658559405e-06, + "loss": 1.5047, + "step": 1028 + }, + { + "epoch": 1.2992424242424243, + "grad_norm": 3.0944340229034424, + "learning_rate": 3.1852198316183354e-06, + "loss": 1.5198, + "step": 1029 + }, + { + "epoch": 1.3005050505050506, + "grad_norm": 2.8930089473724365, + "learning_rate": 3.18288119738073e-06, + "loss": 1.4655, + "step": 1030 + }, + { + "epoch": 1.3005050505050506, + "eval_loss": 1.7008370161056519, + "eval_runtime": 37.1263, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.3017676767676767, + "grad_norm": 2.888303518295288, + "learning_rate": 3.1805425631431246e-06, + "loss": 1.4249, + "step": 1031 + }, + { + "epoch": 1.303030303030303, + "grad_norm": 3.0144314765930176, + "learning_rate": 3.1782039289055195e-06, + "loss": 1.4963, + "step": 1032 + }, + { + "epoch": 1.3042929292929293, + "grad_norm": 3.0118296146392822, + "learning_rate": 3.1758652946679143e-06, + "loss": 1.5119, + "step": 1033 + }, + { + "epoch": 1.3055555555555556, + "grad_norm": 2.8148205280303955, + "learning_rate": 3.1735266604303087e-06, + "loss": 1.4921, + "step": 1034 + }, + { + "epoch": 1.3068181818181819, + "grad_norm": 2.811166286468506, + "learning_rate": 3.1711880261927036e-06, + "loss": 1.4689, + "step": 1035 + }, + { + "epoch": 1.308080808080808, + "grad_norm": 2.924088478088379, + "learning_rate": 3.1688493919550984e-06, + "loss": 1.5682, + "step": 1036 + }, + { + "epoch": 1.3093434343434343, + "grad_norm": 2.905988931655884, + "learning_rate": 3.166510757717493e-06, + "loss": 1.5104, + "step": 1037 + }, + { + "epoch": 1.3106060606060606, + "grad_norm": 2.972963571548462, + "learning_rate": 3.1641721234798877e-06, + "loss": 1.4543, + "step": 1038 + }, + { + "epoch": 1.3118686868686869, + "grad_norm": 2.8034229278564453, + "learning_rate": 3.1618334892422825e-06, + "loss": 1.45, + "step": 1039 + }, + { + "epoch": 1.3131313131313131, + "grad_norm": 2.8977267742156982, + "learning_rate": 3.159494855004678e-06, + "loss": 1.4531, + "step": 1040 + }, + { + "epoch": 1.3131313131313131, + "eval_loss": 1.7027597427368164, + "eval_runtime": 37.0629, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1040 + }, + { + "epoch": 1.3143939393939394, + "grad_norm": 3.106973886489868, + "learning_rate": 3.1571562207670726e-06, + "loss": 1.4723, + "step": 1041 + }, + { + "epoch": 1.3156565656565657, + "grad_norm": 2.7628490924835205, + "learning_rate": 3.154817586529467e-06, + "loss": 1.4902, + "step": 1042 + }, + { + "epoch": 1.316919191919192, + "grad_norm": 3.119751453399658, + "learning_rate": 3.152478952291862e-06, + "loss": 1.5326, + "step": 1043 + }, + { + "epoch": 1.3181818181818181, + "grad_norm": 3.1470835208892822, + "learning_rate": 3.1501403180542568e-06, + "loss": 1.4626, + "step": 1044 + }, + { + "epoch": 1.3194444444444444, + "grad_norm": 2.866586208343506, + "learning_rate": 3.1478016838166516e-06, + "loss": 1.5006, + "step": 1045 + }, + { + "epoch": 1.3207070707070707, + "grad_norm": 2.878464698791504, + "learning_rate": 3.145463049579046e-06, + "loss": 1.4727, + "step": 1046 + }, + { + "epoch": 1.321969696969697, + "grad_norm": 2.8567826747894287, + "learning_rate": 3.143124415341441e-06, + "loss": 1.4202, + "step": 1047 + }, + { + "epoch": 1.3232323232323233, + "grad_norm": 2.8718442916870117, + "learning_rate": 3.1407857811038357e-06, + "loss": 1.473, + "step": 1048 + }, + { + "epoch": 1.3244949494949494, + "grad_norm": 3.003040313720703, + "learning_rate": 3.13844714686623e-06, + "loss": 1.4784, + "step": 1049 + }, + { + "epoch": 1.3257575757575757, + "grad_norm": 3.080288887023926, + "learning_rate": 3.136108512628625e-06, + "loss": 1.4514, + "step": 1050 + }, + { + "epoch": 1.3257575757575757, + "eval_loss": 1.704396367073059, + "eval_runtime": 37.0757, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1050 + }, + { + "epoch": 1.327020202020202, + "grad_norm": 2.8699166774749756, + "learning_rate": 3.13376987839102e-06, + "loss": 1.4966, + "step": 1051 + }, + { + "epoch": 1.3282828282828283, + "grad_norm": 2.979238271713257, + "learning_rate": 3.1314312441534146e-06, + "loss": 1.4422, + "step": 1052 + }, + { + "epoch": 1.3295454545454546, + "grad_norm": 2.8612024784088135, + "learning_rate": 3.129092609915809e-06, + "loss": 1.4982, + "step": 1053 + }, + { + "epoch": 1.3308080808080809, + "grad_norm": 2.7759032249450684, + "learning_rate": 3.126753975678204e-06, + "loss": 1.413, + "step": 1054 + }, + { + "epoch": 1.3320707070707072, + "grad_norm": 2.9772675037384033, + "learning_rate": 3.124415341440599e-06, + "loss": 1.4789, + "step": 1055 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.9382526874542236, + "learning_rate": 3.122076707202994e-06, + "loss": 1.4547, + "step": 1056 + }, + { + "epoch": 1.3345959595959596, + "grad_norm": 2.85050630569458, + "learning_rate": 3.119738072965389e-06, + "loss": 1.4193, + "step": 1057 + }, + { + "epoch": 1.3358585858585859, + "grad_norm": 3.1148695945739746, + "learning_rate": 3.1173994387277833e-06, + "loss": 1.4527, + "step": 1058 + }, + { + "epoch": 1.3371212121212122, + "grad_norm": 3.3906028270721436, + "learning_rate": 3.115060804490178e-06, + "loss": 1.5273, + "step": 1059 + }, + { + "epoch": 1.3383838383838385, + "grad_norm": 2.9665908813476562, + "learning_rate": 3.112722170252573e-06, + "loss": 1.5237, + "step": 1060 + }, + { + "epoch": 1.3383838383838385, + "eval_loss": 1.7004541158676147, + "eval_runtime": 37.081, + "eval_samples_per_second": 26.968, + "eval_steps_per_second": 1.133, + "step": 1060 + }, + { + "epoch": 1.3396464646464645, + "grad_norm": 2.9809224605560303, + "learning_rate": 3.1103835360149674e-06, + "loss": 1.498, + "step": 1061 + }, + { + "epoch": 1.3409090909090908, + "grad_norm": 2.95265793800354, + "learning_rate": 3.1080449017773622e-06, + "loss": 1.511, + "step": 1062 + }, + { + "epoch": 1.3421717171717171, + "grad_norm": 3.0814359188079834, + "learning_rate": 3.105706267539757e-06, + "loss": 1.4448, + "step": 1063 + }, + { + "epoch": 1.3434343434343434, + "grad_norm": 2.8267405033111572, + "learning_rate": 3.103367633302152e-06, + "loss": 1.4648, + "step": 1064 + }, + { + "epoch": 1.3446969696969697, + "grad_norm": 3.08182954788208, + "learning_rate": 3.1010289990645463e-06, + "loss": 1.5245, + "step": 1065 + }, + { + "epoch": 1.345959595959596, + "grad_norm": 3.183209180831909, + "learning_rate": 3.098690364826941e-06, + "loss": 1.4388, + "step": 1066 + }, + { + "epoch": 1.3472222222222223, + "grad_norm": 3.2121284008026123, + "learning_rate": 3.096351730589336e-06, + "loss": 1.4328, + "step": 1067 + }, + { + "epoch": 1.3484848484848486, + "grad_norm": 3.093921661376953, + "learning_rate": 3.0940130963517304e-06, + "loss": 1.5176, + "step": 1068 + }, + { + "epoch": 1.3497474747474747, + "grad_norm": 2.977710008621216, + "learning_rate": 3.0916744621141253e-06, + "loss": 1.4153, + "step": 1069 + }, + { + "epoch": 1.351010101010101, + "grad_norm": 3.119001865386963, + "learning_rate": 3.08933582787652e-06, + "loss": 1.4879, + "step": 1070 + }, + { + "epoch": 1.351010101010101, + "eval_loss": 1.7006300687789917, + "eval_runtime": 37.0959, + "eval_samples_per_second": 26.957, + "eval_steps_per_second": 1.132, + "step": 1070 + }, + { + "epoch": 1.3522727272727273, + "grad_norm": 3.026405096054077, + "learning_rate": 3.0869971936389154e-06, + "loss": 1.483, + "step": 1071 + }, + { + "epoch": 1.3535353535353536, + "grad_norm": 2.770728826522827, + "learning_rate": 3.0846585594013102e-06, + "loss": 1.5175, + "step": 1072 + }, + { + "epoch": 1.3547979797979797, + "grad_norm": 2.7930657863616943, + "learning_rate": 3.0823199251637047e-06, + "loss": 1.4833, + "step": 1073 + }, + { + "epoch": 1.356060606060606, + "grad_norm": 2.9128804206848145, + "learning_rate": 3.0799812909260995e-06, + "loss": 1.5016, + "step": 1074 + }, + { + "epoch": 1.3573232323232323, + "grad_norm": 3.0432205200195312, + "learning_rate": 3.0776426566884943e-06, + "loss": 1.5211, + "step": 1075 + }, + { + "epoch": 1.3585858585858586, + "grad_norm": 2.9197583198547363, + "learning_rate": 3.075304022450889e-06, + "loss": 1.5065, + "step": 1076 + }, + { + "epoch": 1.3598484848484849, + "grad_norm": 2.944986581802368, + "learning_rate": 3.0729653882132836e-06, + "loss": 1.5015, + "step": 1077 + }, + { + "epoch": 1.3611111111111112, + "grad_norm": 2.7553489208221436, + "learning_rate": 3.0706267539756784e-06, + "loss": 1.4692, + "step": 1078 + }, + { + "epoch": 1.3623737373737375, + "grad_norm": 2.7607858180999756, + "learning_rate": 3.0682881197380733e-06, + "loss": 1.4494, + "step": 1079 + }, + { + "epoch": 1.3636363636363638, + "grad_norm": 2.7896173000335693, + "learning_rate": 3.0659494855004677e-06, + "loss": 1.4625, + "step": 1080 + }, + { + "epoch": 1.3636363636363638, + "eval_loss": 1.6998732089996338, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1080 + }, + { + "epoch": 1.3648989898989898, + "grad_norm": 3.020059823989868, + "learning_rate": 3.0636108512628626e-06, + "loss": 1.5331, + "step": 1081 + }, + { + "epoch": 1.3661616161616161, + "grad_norm": 2.805593729019165, + "learning_rate": 3.0612722170252574e-06, + "loss": 1.4867, + "step": 1082 + }, + { + "epoch": 1.3674242424242424, + "grad_norm": 2.965871810913086, + "learning_rate": 3.0589335827876522e-06, + "loss": 1.4494, + "step": 1083 + }, + { + "epoch": 1.3686868686868687, + "grad_norm": 2.8010783195495605, + "learning_rate": 3.0565949485500467e-06, + "loss": 1.4484, + "step": 1084 + }, + { + "epoch": 1.369949494949495, + "grad_norm": 2.985258102416992, + "learning_rate": 3.0542563143124415e-06, + "loss": 1.5029, + "step": 1085 + }, + { + "epoch": 1.371212121212121, + "grad_norm": 2.7528188228607178, + "learning_rate": 3.0519176800748368e-06, + "loss": 1.4695, + "step": 1086 + }, + { + "epoch": 1.3724747474747474, + "grad_norm": 2.9123780727386475, + "learning_rate": 3.0495790458372316e-06, + "loss": 1.5527, + "step": 1087 + }, + { + "epoch": 1.3737373737373737, + "grad_norm": 2.8589160442352295, + "learning_rate": 3.0472404115996265e-06, + "loss": 1.5104, + "step": 1088 + }, + { + "epoch": 1.375, + "grad_norm": 2.853546142578125, + "learning_rate": 3.044901777362021e-06, + "loss": 1.4498, + "step": 1089 + }, + { + "epoch": 1.3762626262626263, + "grad_norm": 2.7989797592163086, + "learning_rate": 3.0425631431244157e-06, + "loss": 1.4494, + "step": 1090 + }, + { + "epoch": 1.3762626262626263, + "eval_loss": 1.6923459768295288, + "eval_runtime": 37.2917, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 1090 + }, + { + "epoch": 1.3775252525252526, + "grad_norm": 2.8750412464141846, + "learning_rate": 3.0402245088868106e-06, + "loss": 1.5026, + "step": 1091 + }, + { + "epoch": 1.378787878787879, + "grad_norm": 2.8185842037200928, + "learning_rate": 3.037885874649205e-06, + "loss": 1.4859, + "step": 1092 + }, + { + "epoch": 1.380050505050505, + "grad_norm": 2.736332893371582, + "learning_rate": 3.0355472404116e-06, + "loss": 1.4944, + "step": 1093 + }, + { + "epoch": 1.3813131313131313, + "grad_norm": 2.93930983543396, + "learning_rate": 3.0332086061739947e-06, + "loss": 1.4975, + "step": 1094 + }, + { + "epoch": 1.3825757575757576, + "grad_norm": 2.964146852493286, + "learning_rate": 3.0308699719363895e-06, + "loss": 1.4861, + "step": 1095 + }, + { + "epoch": 1.3838383838383839, + "grad_norm": 2.8321545124053955, + "learning_rate": 3.028531337698784e-06, + "loss": 1.4535, + "step": 1096 + }, + { + "epoch": 1.3851010101010102, + "grad_norm": 2.921372413635254, + "learning_rate": 3.0261927034611788e-06, + "loss": 1.5242, + "step": 1097 + }, + { + "epoch": 1.3863636363636362, + "grad_norm": 2.7309911251068115, + "learning_rate": 3.0238540692235736e-06, + "loss": 1.4105, + "step": 1098 + }, + { + "epoch": 1.3876262626262625, + "grad_norm": 2.768899917602539, + "learning_rate": 3.021515434985968e-06, + "loss": 1.4183, + "step": 1099 + }, + { + "epoch": 1.3888888888888888, + "grad_norm": 2.883399248123169, + "learning_rate": 3.019176800748363e-06, + "loss": 1.5307, + "step": 1100 + }, + { + "epoch": 1.3888888888888888, + "eval_loss": 1.6948082447052002, + "eval_runtime": 37.488, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1100 + }, + { + "epoch": 1.3901515151515151, + "grad_norm": 2.8610544204711914, + "learning_rate": 3.016838166510758e-06, + "loss": 1.5473, + "step": 1101 + }, + { + "epoch": 1.3914141414141414, + "grad_norm": 2.762694835662842, + "learning_rate": 3.014499532273153e-06, + "loss": 1.4179, + "step": 1102 + }, + { + "epoch": 1.3926767676767677, + "grad_norm": 2.975828170776367, + "learning_rate": 3.012160898035548e-06, + "loss": 1.4959, + "step": 1103 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 3.0633926391601562, + "learning_rate": 3.0098222637979422e-06, + "loss": 1.4979, + "step": 1104 + }, + { + "epoch": 1.3952020202020203, + "grad_norm": 2.972900867462158, + "learning_rate": 3.007483629560337e-06, + "loss": 1.5311, + "step": 1105 + }, + { + "epoch": 1.3964646464646464, + "grad_norm": 2.936033248901367, + "learning_rate": 3.005144995322732e-06, + "loss": 1.4856, + "step": 1106 + }, + { + "epoch": 1.3977272727272727, + "grad_norm": 3.0375452041625977, + "learning_rate": 3.0028063610851268e-06, + "loss": 1.5001, + "step": 1107 + }, + { + "epoch": 1.398989898989899, + "grad_norm": 2.998283624649048, + "learning_rate": 3.000467726847521e-06, + "loss": 1.5219, + "step": 1108 + }, + { + "epoch": 1.4002525252525253, + "grad_norm": 2.831354856491089, + "learning_rate": 2.998129092609916e-06, + "loss": 1.4498, + "step": 1109 + }, + { + "epoch": 1.4015151515151514, + "grad_norm": 2.975390911102295, + "learning_rate": 2.995790458372311e-06, + "loss": 1.4655, + "step": 1110 + }, + { + "epoch": 1.4015151515151514, + "eval_loss": 1.6940916776657104, + "eval_runtime": 36.9469, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.4027777777777777, + "grad_norm": 2.9645442962646484, + "learning_rate": 2.9934518241347053e-06, + "loss": 1.4568, + "step": 1111 + }, + { + "epoch": 1.404040404040404, + "grad_norm": 2.873746633529663, + "learning_rate": 2.9911131898971e-06, + "loss": 1.4483, + "step": 1112 + }, + { + "epoch": 1.4053030303030303, + "grad_norm": 3.1041767597198486, + "learning_rate": 2.988774555659495e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.4065656565656566, + "grad_norm": 3.0122971534729004, + "learning_rate": 2.98643592142189e-06, + "loss": 1.4494, + "step": 1114 + }, + { + "epoch": 1.4078282828282829, + "grad_norm": 2.842233180999756, + "learning_rate": 2.9840972871842842e-06, + "loss": 1.4274, + "step": 1115 + }, + { + "epoch": 1.4090909090909092, + "grad_norm": 3.058265447616577, + "learning_rate": 2.9817586529466795e-06, + "loss": 1.4637, + "step": 1116 + }, + { + "epoch": 1.4103535353535355, + "grad_norm": 3.06136155128479, + "learning_rate": 2.9794200187090744e-06, + "loss": 1.4791, + "step": 1117 + }, + { + "epoch": 1.4116161616161615, + "grad_norm": 2.8931167125701904, + "learning_rate": 2.977081384471469e-06, + "loss": 1.4476, + "step": 1118 + }, + { + "epoch": 1.4128787878787878, + "grad_norm": 2.8025107383728027, + "learning_rate": 2.974742750233864e-06, + "loss": 1.4891, + "step": 1119 + }, + { + "epoch": 1.4141414141414141, + "grad_norm": 3.1213603019714355, + "learning_rate": 2.9724041159962585e-06, + "loss": 1.5513, + "step": 1120 + }, + { + "epoch": 1.4141414141414141, + "eval_loss": 1.6972154378890991, + "eval_runtime": 37.0863, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1120 + }, + { + "epoch": 1.4154040404040404, + "grad_norm": 3.126791477203369, + "learning_rate": 2.9700654817586533e-06, + "loss": 1.4952, + "step": 1121 + }, + { + "epoch": 1.4166666666666667, + "grad_norm": 2.959385395050049, + "learning_rate": 2.967726847521048e-06, + "loss": 1.5098, + "step": 1122 + }, + { + "epoch": 1.4179292929292928, + "grad_norm": 2.907566547393799, + "learning_rate": 2.9653882132834426e-06, + "loss": 1.4959, + "step": 1123 + }, + { + "epoch": 1.4191919191919191, + "grad_norm": 2.9181833267211914, + "learning_rate": 2.9630495790458374e-06, + "loss": 1.4607, + "step": 1124 + }, + { + "epoch": 1.4204545454545454, + "grad_norm": 2.8703668117523193, + "learning_rate": 2.9607109448082323e-06, + "loss": 1.5215, + "step": 1125 + }, + { + "epoch": 1.4217171717171717, + "grad_norm": 3.166255235671997, + "learning_rate": 2.958372310570627e-06, + "loss": 1.4451, + "step": 1126 + }, + { + "epoch": 1.422979797979798, + "grad_norm": 2.863943576812744, + "learning_rate": 2.9560336763330215e-06, + "loss": 1.4481, + "step": 1127 + }, + { + "epoch": 1.4242424242424243, + "grad_norm": 2.975482702255249, + "learning_rate": 2.9536950420954164e-06, + "loss": 1.5118, + "step": 1128 + }, + { + "epoch": 1.4255050505050506, + "grad_norm": 3.077357769012451, + "learning_rate": 2.951356407857811e-06, + "loss": 1.4678, + "step": 1129 + }, + { + "epoch": 1.4267676767676767, + "grad_norm": 3.0375332832336426, + "learning_rate": 2.9490177736202056e-06, + "loss": 1.5036, + "step": 1130 + }, + { + "epoch": 1.4267676767676767, + "eval_loss": 1.6973206996917725, + "eval_runtime": 37.1448, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1130 + }, + { + "epoch": 1.428030303030303, + "grad_norm": 3.022068500518799, + "learning_rate": 2.9466791393826005e-06, + "loss": 1.442, + "step": 1131 + }, + { + "epoch": 1.4292929292929293, + "grad_norm": 2.9089183807373047, + "learning_rate": 2.9443405051449957e-06, + "loss": 1.4627, + "step": 1132 + }, + { + "epoch": 1.4305555555555556, + "grad_norm": 2.840149402618408, + "learning_rate": 2.9420018709073906e-06, + "loss": 1.4745, + "step": 1133 + }, + { + "epoch": 1.4318181818181819, + "grad_norm": 2.938650608062744, + "learning_rate": 2.9396632366697854e-06, + "loss": 1.4795, + "step": 1134 + }, + { + "epoch": 1.433080808080808, + "grad_norm": 2.9602437019348145, + "learning_rate": 2.93732460243218e-06, + "loss": 1.4228, + "step": 1135 + }, + { + "epoch": 1.4343434343434343, + "grad_norm": 2.929257869720459, + "learning_rate": 2.9349859681945747e-06, + "loss": 1.5016, + "step": 1136 + }, + { + "epoch": 1.4356060606060606, + "grad_norm": 2.8120505809783936, + "learning_rate": 2.9326473339569695e-06, + "loss": 1.474, + "step": 1137 + }, + { + "epoch": 1.4368686868686869, + "grad_norm": 3.0380513668060303, + "learning_rate": 2.9303086997193644e-06, + "loss": 1.4714, + "step": 1138 + }, + { + "epoch": 1.4381313131313131, + "grad_norm": 2.91106915473938, + "learning_rate": 2.9279700654817588e-06, + "loss": 1.4558, + "step": 1139 + }, + { + "epoch": 1.4393939393939394, + "grad_norm": 2.9970154762268066, + "learning_rate": 2.9256314312441536e-06, + "loss": 1.4895, + "step": 1140 + }, + { + "epoch": 1.4393939393939394, + "eval_loss": 1.6916719675064087, + "eval_runtime": 37.1538, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 1140 + }, + { + "epoch": 1.4406565656565657, + "grad_norm": 3.041616201400757, + "learning_rate": 2.9232927970065485e-06, + "loss": 1.5177, + "step": 1141 + }, + { + "epoch": 1.441919191919192, + "grad_norm": 2.875385046005249, + "learning_rate": 2.920954162768943e-06, + "loss": 1.4697, + "step": 1142 + }, + { + "epoch": 1.4431818181818181, + "grad_norm": 2.7117881774902344, + "learning_rate": 2.9186155285313377e-06, + "loss": 1.4809, + "step": 1143 + }, + { + "epoch": 1.4444444444444444, + "grad_norm": 2.912123203277588, + "learning_rate": 2.9162768942937326e-06, + "loss": 1.4032, + "step": 1144 + }, + { + "epoch": 1.4457070707070707, + "grad_norm": 2.798440933227539, + "learning_rate": 2.9139382600561274e-06, + "loss": 1.4442, + "step": 1145 + }, + { + "epoch": 1.446969696969697, + "grad_norm": 2.9299135208129883, + "learning_rate": 2.911599625818522e-06, + "loss": 1.4412, + "step": 1146 + }, + { + "epoch": 1.4482323232323233, + "grad_norm": 2.988434076309204, + "learning_rate": 2.909260991580917e-06, + "loss": 1.4473, + "step": 1147 + }, + { + "epoch": 1.4494949494949494, + "grad_norm": 3.0697057247161865, + "learning_rate": 2.906922357343312e-06, + "loss": 1.5151, + "step": 1148 + }, + { + "epoch": 1.4507575757575757, + "grad_norm": 2.9009311199188232, + "learning_rate": 2.9045837231057068e-06, + "loss": 1.4434, + "step": 1149 + }, + { + "epoch": 1.452020202020202, + "grad_norm": 2.9949986934661865, + "learning_rate": 2.902245088868101e-06, + "loss": 1.5436, + "step": 1150 + }, + { + "epoch": 1.452020202020202, + "eval_loss": 1.6880143880844116, + "eval_runtime": 37.1212, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.4532828282828283, + "grad_norm": 2.7827775478363037, + "learning_rate": 2.899906454630496e-06, + "loss": 1.4284, + "step": 1151 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.8879382610321045, + "learning_rate": 2.897567820392891e-06, + "loss": 1.4772, + "step": 1152 + }, + { + "epoch": 1.4558080808080809, + "grad_norm": 3.0119130611419678, + "learning_rate": 2.8952291861552857e-06, + "loss": 1.4546, + "step": 1153 + }, + { + "epoch": 1.4570707070707072, + "grad_norm": 2.9002952575683594, + "learning_rate": 2.89289055191768e-06, + "loss": 1.4898, + "step": 1154 + }, + { + "epoch": 1.4583333333333333, + "grad_norm": 2.8384580612182617, + "learning_rate": 2.890551917680075e-06, + "loss": 1.4685, + "step": 1155 + }, + { + "epoch": 1.4595959595959596, + "grad_norm": 2.9149580001831055, + "learning_rate": 2.88821328344247e-06, + "loss": 1.5133, + "step": 1156 + }, + { + "epoch": 1.4608585858585859, + "grad_norm": 3.041875123977661, + "learning_rate": 2.8858746492048647e-06, + "loss": 1.5038, + "step": 1157 + }, + { + "epoch": 1.4621212121212122, + "grad_norm": 2.9623730182647705, + "learning_rate": 2.883536014967259e-06, + "loss": 1.4776, + "step": 1158 + }, + { + "epoch": 1.4633838383838385, + "grad_norm": 2.935295343399048, + "learning_rate": 2.881197380729654e-06, + "loss": 1.4166, + "step": 1159 + }, + { + "epoch": 1.4646464646464645, + "grad_norm": 2.9556751251220703, + "learning_rate": 2.8788587464920488e-06, + "loss": 1.5132, + "step": 1160 + }, + { + "epoch": 1.4646464646464645, + "eval_loss": 1.69142746925354, + "eval_runtime": 37.1425, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1160 + }, + { + "epoch": 1.4659090909090908, + "grad_norm": 2.992587089538574, + "learning_rate": 2.876520112254443e-06, + "loss": 1.4695, + "step": 1161 + }, + { + "epoch": 1.4671717171717171, + "grad_norm": 3.005990982055664, + "learning_rate": 2.8741814780168385e-06, + "loss": 1.5122, + "step": 1162 + }, + { + "epoch": 1.4684343434343434, + "grad_norm": 2.9341115951538086, + "learning_rate": 2.8718428437792333e-06, + "loss": 1.4916, + "step": 1163 + }, + { + "epoch": 1.4696969696969697, + "grad_norm": 2.9665164947509766, + "learning_rate": 2.869504209541628e-06, + "loss": 1.4827, + "step": 1164 + }, + { + "epoch": 1.470959595959596, + "grad_norm": 2.9972221851348877, + "learning_rate": 2.867165575304023e-06, + "loss": 1.4845, + "step": 1165 + }, + { + "epoch": 1.4722222222222223, + "grad_norm": 2.981387138366699, + "learning_rate": 2.8648269410664174e-06, + "loss": 1.4374, + "step": 1166 + }, + { + "epoch": 1.4734848484848486, + "grad_norm": 3.04587459564209, + "learning_rate": 2.8624883068288123e-06, + "loss": 1.431, + "step": 1167 + }, + { + "epoch": 1.4747474747474747, + "grad_norm": 3.194389820098877, + "learning_rate": 2.860149672591207e-06, + "loss": 1.4847, + "step": 1168 + }, + { + "epoch": 1.476010101010101, + "grad_norm": 2.949721574783325, + "learning_rate": 2.857811038353602e-06, + "loss": 1.4596, + "step": 1169 + }, + { + "epoch": 1.4772727272727273, + "grad_norm": 3.0032384395599365, + "learning_rate": 2.8554724041159964e-06, + "loss": 1.4821, + "step": 1170 + }, + { + "epoch": 1.4772727272727273, + "eval_loss": 1.6863130331039429, + "eval_runtime": 37.1386, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.4785353535353536, + "grad_norm": 2.943141460418701, + "learning_rate": 2.853133769878391e-06, + "loss": 1.4998, + "step": 1171 + }, + { + "epoch": 1.4797979797979797, + "grad_norm": 3.0163350105285645, + "learning_rate": 2.850795135640786e-06, + "loss": 1.4565, + "step": 1172 + }, + { + "epoch": 1.481060606060606, + "grad_norm": 2.9713706970214844, + "learning_rate": 2.8484565014031805e-06, + "loss": 1.4663, + "step": 1173 + }, + { + "epoch": 1.4823232323232323, + "grad_norm": 2.9195785522460938, + "learning_rate": 2.8461178671655753e-06, + "loss": 1.4594, + "step": 1174 + }, + { + "epoch": 1.4835858585858586, + "grad_norm": 3.127619504928589, + "learning_rate": 2.84377923292797e-06, + "loss": 1.4742, + "step": 1175 + }, + { + "epoch": 1.4848484848484849, + "grad_norm": 3.0264782905578613, + "learning_rate": 2.841440598690365e-06, + "loss": 1.4366, + "step": 1176 + }, + { + "epoch": 1.4861111111111112, + "grad_norm": 3.0541677474975586, + "learning_rate": 2.8391019644527603e-06, + "loss": 1.4629, + "step": 1177 + }, + { + "epoch": 1.4873737373737375, + "grad_norm": 2.8967537879943848, + "learning_rate": 2.8367633302151547e-06, + "loss": 1.4281, + "step": 1178 + }, + { + "epoch": 1.4886363636363638, + "grad_norm": 2.975773811340332, + "learning_rate": 2.8344246959775495e-06, + "loss": 1.4272, + "step": 1179 + }, + { + "epoch": 1.4898989898989898, + "grad_norm": 2.9768712520599365, + "learning_rate": 2.8320860617399444e-06, + "loss": 1.453, + "step": 1180 + }, + { + "epoch": 1.4898989898989898, + "eval_loss": 1.6810754537582397, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1180 + }, + { + "epoch": 1.4911616161616161, + "grad_norm": 2.8897762298583984, + "learning_rate": 2.829747427502339e-06, + "loss": 1.4577, + "step": 1181 + }, + { + "epoch": 1.4924242424242424, + "grad_norm": 2.848259925842285, + "learning_rate": 2.8274087932647336e-06, + "loss": 1.4323, + "step": 1182 + }, + { + "epoch": 1.4936868686868687, + "grad_norm": 2.9275119304656982, + "learning_rate": 2.8250701590271285e-06, + "loss": 1.4712, + "step": 1183 + }, + { + "epoch": 1.494949494949495, + "grad_norm": 2.8643906116485596, + "learning_rate": 2.8227315247895233e-06, + "loss": 1.4375, + "step": 1184 + }, + { + "epoch": 1.496212121212121, + "grad_norm": 3.0250704288482666, + "learning_rate": 2.8203928905519177e-06, + "loss": 1.5482, + "step": 1185 + }, + { + "epoch": 1.4974747474747474, + "grad_norm": 3.1019556522369385, + "learning_rate": 2.8180542563143126e-06, + "loss": 1.4618, + "step": 1186 + }, + { + "epoch": 1.4987373737373737, + "grad_norm": 3.1480300426483154, + "learning_rate": 2.8157156220767074e-06, + "loss": 1.4152, + "step": 1187 + }, + { + "epoch": 1.5, + "grad_norm": 2.82671856880188, + "learning_rate": 2.8133769878391023e-06, + "loss": 1.4692, + "step": 1188 + }, + { + "epoch": 1.5012626262626263, + "grad_norm": 3.2128899097442627, + "learning_rate": 2.8110383536014967e-06, + "loss": 1.4087, + "step": 1189 + }, + { + "epoch": 1.5025252525252526, + "grad_norm": 3.1939210891723633, + "learning_rate": 2.8086997193638915e-06, + "loss": 1.4919, + "step": 1190 + }, + { + "epoch": 1.5025252525252526, + "eval_loss": 1.6861544847488403, + "eval_runtime": 37.3328, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.503787878787879, + "grad_norm": 3.022059679031372, + "learning_rate": 2.8063610851262864e-06, + "loss": 1.4865, + "step": 1191 + }, + { + "epoch": 1.5050505050505052, + "grad_norm": 3.0068860054016113, + "learning_rate": 2.804022450888681e-06, + "loss": 1.478, + "step": 1192 + }, + { + "epoch": 1.5063131313131313, + "grad_norm": 3.2898383140563965, + "learning_rate": 2.801683816651076e-06, + "loss": 1.4773, + "step": 1193 + }, + { + "epoch": 1.5075757575757576, + "grad_norm": 3.090339422225952, + "learning_rate": 2.799345182413471e-06, + "loss": 1.4012, + "step": 1194 + }, + { + "epoch": 1.5088383838383839, + "grad_norm": 2.977215528488159, + "learning_rate": 2.7970065481758657e-06, + "loss": 1.4651, + "step": 1195 + }, + { + "epoch": 1.51010101010101, + "grad_norm": 3.1349570751190186, + "learning_rate": 2.7946679139382606e-06, + "loss": 1.5572, + "step": 1196 + }, + { + "epoch": 1.5113636363636362, + "grad_norm": 2.8715298175811768, + "learning_rate": 2.792329279700655e-06, + "loss": 1.4558, + "step": 1197 + }, + { + "epoch": 1.5126262626262625, + "grad_norm": 2.9848744869232178, + "learning_rate": 2.78999064546305e-06, + "loss": 1.4383, + "step": 1198 + }, + { + "epoch": 1.5138888888888888, + "grad_norm": 3.0774083137512207, + "learning_rate": 2.7876520112254447e-06, + "loss": 1.4607, + "step": 1199 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.9721343517303467, + "learning_rate": 2.785313376987839e-06, + "loss": 1.4341, + "step": 1200 + }, + { + "epoch": 1.5151515151515151, + "eval_loss": 1.6859924793243408, + "eval_runtime": 37.4589, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 1200 + }, + { + "epoch": 1.5164141414141414, + "grad_norm": 3.041095733642578, + "learning_rate": 2.782974742750234e-06, + "loss": 1.4736, + "step": 1201 + }, + { + "epoch": 1.5176767676767677, + "grad_norm": 2.8400142192840576, + "learning_rate": 2.780636108512629e-06, + "loss": 1.4169, + "step": 1202 + }, + { + "epoch": 1.518939393939394, + "grad_norm": 2.785592555999756, + "learning_rate": 2.7782974742750236e-06, + "loss": 1.4553, + "step": 1203 + }, + { + "epoch": 1.5202020202020203, + "grad_norm": 2.75209641456604, + "learning_rate": 2.775958840037418e-06, + "loss": 1.4267, + "step": 1204 + }, + { + "epoch": 1.5214646464646466, + "grad_norm": 2.927558183670044, + "learning_rate": 2.773620205799813e-06, + "loss": 1.4377, + "step": 1205 + }, + { + "epoch": 1.5227272727272727, + "grad_norm": 2.9335508346557617, + "learning_rate": 2.7712815715622077e-06, + "loss": 1.4295, + "step": 1206 + }, + { + "epoch": 1.523989898989899, + "grad_norm": 2.7812745571136475, + "learning_rate": 2.7689429373246026e-06, + "loss": 1.4029, + "step": 1207 + }, + { + "epoch": 1.5252525252525253, + "grad_norm": 3.230062961578369, + "learning_rate": 2.766604303086998e-06, + "loss": 1.539, + "step": 1208 + }, + { + "epoch": 1.5265151515151514, + "grad_norm": 3.1244332790374756, + "learning_rate": 2.7642656688493923e-06, + "loss": 1.4829, + "step": 1209 + }, + { + "epoch": 1.5277777777777777, + "grad_norm": 3.2391650676727295, + "learning_rate": 2.761927034611787e-06, + "loss": 1.4677, + "step": 1210 + }, + { + "epoch": 1.5277777777777777, + "eval_loss": 1.6849907636642456, + "eval_runtime": 36.8638, + "eval_samples_per_second": 27.127, + "eval_steps_per_second": 1.139, + "step": 1210 + }, + { + "epoch": 1.529040404040404, + "grad_norm": 3.1884987354278564, + "learning_rate": 2.759588400374182e-06, + "loss": 1.4185, + "step": 1211 + }, + { + "epoch": 1.5303030303030303, + "grad_norm": 2.959118366241455, + "learning_rate": 2.7572497661365764e-06, + "loss": 1.4536, + "step": 1212 + }, + { + "epoch": 1.5315656565656566, + "grad_norm": 2.882380247116089, + "learning_rate": 2.7549111318989712e-06, + "loss": 1.4872, + "step": 1213 + }, + { + "epoch": 1.5328282828282829, + "grad_norm": 3.0574429035186768, + "learning_rate": 2.752572497661366e-06, + "loss": 1.4343, + "step": 1214 + }, + { + "epoch": 1.5340909090909092, + "grad_norm": 3.1645233631134033, + "learning_rate": 2.750233863423761e-06, + "loss": 1.5101, + "step": 1215 + }, + { + "epoch": 1.5353535353535355, + "grad_norm": 2.83034086227417, + "learning_rate": 2.7478952291861553e-06, + "loss": 1.45, + "step": 1216 + }, + { + "epoch": 1.5366161616161618, + "grad_norm": 2.964780330657959, + "learning_rate": 2.74555659494855e-06, + "loss": 1.453, + "step": 1217 + }, + { + "epoch": 1.5378787878787878, + "grad_norm": 2.9896774291992188, + "learning_rate": 2.743217960710945e-06, + "loss": 1.4648, + "step": 1218 + }, + { + "epoch": 1.5391414141414141, + "grad_norm": 3.013664484024048, + "learning_rate": 2.74087932647334e-06, + "loss": 1.4828, + "step": 1219 + }, + { + "epoch": 1.5404040404040404, + "grad_norm": 3.0011773109436035, + "learning_rate": 2.7385406922357343e-06, + "loss": 1.4705, + "step": 1220 + }, + { + "epoch": 1.5404040404040404, + "eval_loss": 1.6872731447219849, + "eval_runtime": 36.9864, + "eval_samples_per_second": 27.037, + "eval_steps_per_second": 1.136, + "step": 1220 + }, + { + "epoch": 1.5416666666666665, + "grad_norm": 3.239349842071533, + "learning_rate": 2.736202057998129e-06, + "loss": 1.484, + "step": 1221 + }, + { + "epoch": 1.5429292929292928, + "grad_norm": 2.9267003536224365, + "learning_rate": 2.733863423760524e-06, + "loss": 1.4587, + "step": 1222 + }, + { + "epoch": 1.5441919191919191, + "grad_norm": 2.890094518661499, + "learning_rate": 2.7315247895229192e-06, + "loss": 1.4456, + "step": 1223 + }, + { + "epoch": 1.5454545454545454, + "grad_norm": 3.0287511348724365, + "learning_rate": 2.7291861552853136e-06, + "loss": 1.5059, + "step": 1224 + }, + { + "epoch": 1.5467171717171717, + "grad_norm": 3.041609525680542, + "learning_rate": 2.7268475210477085e-06, + "loss": 1.4341, + "step": 1225 + }, + { + "epoch": 1.547979797979798, + "grad_norm": 2.8771862983703613, + "learning_rate": 2.7245088868101033e-06, + "loss": 1.4308, + "step": 1226 + }, + { + "epoch": 1.5492424242424243, + "grad_norm": 3.1173794269561768, + "learning_rate": 2.722170252572498e-06, + "loss": 1.514, + "step": 1227 + }, + { + "epoch": 1.5505050505050506, + "grad_norm": 2.838275671005249, + "learning_rate": 2.7198316183348926e-06, + "loss": 1.4798, + "step": 1228 + }, + { + "epoch": 1.551767676767677, + "grad_norm": 2.9917166233062744, + "learning_rate": 2.7174929840972874e-06, + "loss": 1.5256, + "step": 1229 + }, + { + "epoch": 1.553030303030303, + "grad_norm": 3.071840763092041, + "learning_rate": 2.7151543498596823e-06, + "loss": 1.5187, + "step": 1230 + }, + { + "epoch": 1.553030303030303, + "eval_loss": 1.6850661039352417, + "eval_runtime": 37.0094, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 1230 + }, + { + "epoch": 1.5542929292929293, + "grad_norm": 3.253169536590576, + "learning_rate": 2.7128157156220767e-06, + "loss": 1.5522, + "step": 1231 + }, + { + "epoch": 1.5555555555555556, + "grad_norm": 3.037151336669922, + "learning_rate": 2.7104770813844715e-06, + "loss": 1.5125, + "step": 1232 + }, + { + "epoch": 1.5568181818181817, + "grad_norm": 2.8486485481262207, + "learning_rate": 2.7081384471468664e-06, + "loss": 1.4169, + "step": 1233 + }, + { + "epoch": 1.558080808080808, + "grad_norm": 2.953824996948242, + "learning_rate": 2.7057998129092612e-06, + "loss": 1.4505, + "step": 1234 + }, + { + "epoch": 1.5593434343434343, + "grad_norm": 3.012303590774536, + "learning_rate": 2.7034611786716557e-06, + "loss": 1.4816, + "step": 1235 + }, + { + "epoch": 1.5606060606060606, + "grad_norm": 3.1798768043518066, + "learning_rate": 2.7011225444340505e-06, + "loss": 1.5116, + "step": 1236 + }, + { + "epoch": 1.5618686868686869, + "grad_norm": 3.11370587348938, + "learning_rate": 2.6987839101964453e-06, + "loss": 1.5035, + "step": 1237 + }, + { + "epoch": 1.5631313131313131, + "grad_norm": 3.0563342571258545, + "learning_rate": 2.69644527595884e-06, + "loss": 1.4387, + "step": 1238 + }, + { + "epoch": 1.5643939393939394, + "grad_norm": 2.9884026050567627, + "learning_rate": 2.6941066417212354e-06, + "loss": 1.428, + "step": 1239 + }, + { + "epoch": 1.5656565656565657, + "grad_norm": 2.955773115158081, + "learning_rate": 2.69176800748363e-06, + "loss": 1.4208, + "step": 1240 + }, + { + "epoch": 1.5656565656565657, + "eval_loss": 1.685523509979248, + "eval_runtime": 37.0248, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1240 + }, + { + "epoch": 1.566919191919192, + "grad_norm": 2.9858005046844482, + "learning_rate": 2.6894293732460247e-06, + "loss": 1.4706, + "step": 1241 + }, + { + "epoch": 1.5681818181818183, + "grad_norm": 3.122340440750122, + "learning_rate": 2.6870907390084196e-06, + "loss": 1.4892, + "step": 1242 + }, + { + "epoch": 1.5694444444444444, + "grad_norm": 2.9828262329101562, + "learning_rate": 2.684752104770814e-06, + "loss": 1.4255, + "step": 1243 + }, + { + "epoch": 1.5707070707070707, + "grad_norm": 2.916653633117676, + "learning_rate": 2.682413470533209e-06, + "loss": 1.4241, + "step": 1244 + }, + { + "epoch": 1.571969696969697, + "grad_norm": 3.1615540981292725, + "learning_rate": 2.6800748362956037e-06, + "loss": 1.4719, + "step": 1245 + }, + { + "epoch": 1.573232323232323, + "grad_norm": 3.154130697250366, + "learning_rate": 2.6777362020579985e-06, + "loss": 1.4731, + "step": 1246 + }, + { + "epoch": 1.5744949494949494, + "grad_norm": 3.040848731994629, + "learning_rate": 2.675397567820393e-06, + "loss": 1.5149, + "step": 1247 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.9808318614959717, + "learning_rate": 2.6730589335827878e-06, + "loss": 1.431, + "step": 1248 + }, + { + "epoch": 1.577020202020202, + "grad_norm": 2.8538382053375244, + "learning_rate": 2.6707202993451826e-06, + "loss": 1.4458, + "step": 1249 + }, + { + "epoch": 1.5782828282828283, + "grad_norm": 3.113945960998535, + "learning_rate": 2.6683816651075774e-06, + "loss": 1.4873, + "step": 1250 + }, + { + "epoch": 1.5782828282828283, + "eval_loss": 1.6823841333389282, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1250 + }, + { + "epoch": 1.5795454545454546, + "grad_norm": 3.0044517517089844, + "learning_rate": 2.666043030869972e-06, + "loss": 1.4904, + "step": 1251 + }, + { + "epoch": 1.5808080808080809, + "grad_norm": 2.845820665359497, + "learning_rate": 2.6637043966323667e-06, + "loss": 1.4174, + "step": 1252 + }, + { + "epoch": 1.5820707070707072, + "grad_norm": 2.9123542308807373, + "learning_rate": 2.6613657623947616e-06, + "loss": 1.4896, + "step": 1253 + }, + { + "epoch": 1.5833333333333335, + "grad_norm": 3.1593222618103027, + "learning_rate": 2.659027128157157e-06, + "loss": 1.5208, + "step": 1254 + }, + { + "epoch": 1.5845959595959596, + "grad_norm": 3.0782854557037354, + "learning_rate": 2.6566884939195512e-06, + "loss": 1.4911, + "step": 1255 + }, + { + "epoch": 1.5858585858585859, + "grad_norm": 2.77309513092041, + "learning_rate": 2.654349859681946e-06, + "loss": 1.4475, + "step": 1256 + }, + { + "epoch": 1.5871212121212122, + "grad_norm": 2.9154622554779053, + "learning_rate": 2.652011225444341e-06, + "loss": 1.4324, + "step": 1257 + }, + { + "epoch": 1.5883838383838382, + "grad_norm": 3.107395887374878, + "learning_rate": 2.6496725912067358e-06, + "loss": 1.5126, + "step": 1258 + }, + { + "epoch": 1.5896464646464645, + "grad_norm": 2.825025796890259, + "learning_rate": 2.64733395696913e-06, + "loss": 1.4723, + "step": 1259 + }, + { + "epoch": 1.5909090909090908, + "grad_norm": 2.833296060562134, + "learning_rate": 2.644995322731525e-06, + "loss": 1.4516, + "step": 1260 + }, + { + "epoch": 1.5909090909090908, + "eval_loss": 1.6950432062149048, + "eval_runtime": 37.0343, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1260 + }, + { + "epoch": 1.5921717171717171, + "grad_norm": 2.9508421421051025, + "learning_rate": 2.64265668849392e-06, + "loss": 1.4002, + "step": 1261 + }, + { + "epoch": 1.5934343434343434, + "grad_norm": 2.8778491020202637, + "learning_rate": 2.6403180542563143e-06, + "loss": 1.4258, + "step": 1262 + }, + { + "epoch": 1.5946969696969697, + "grad_norm": 2.81261944770813, + "learning_rate": 2.637979420018709e-06, + "loss": 1.435, + "step": 1263 + }, + { + "epoch": 1.595959595959596, + "grad_norm": 3.1811511516571045, + "learning_rate": 2.635640785781104e-06, + "loss": 1.4792, + "step": 1264 + }, + { + "epoch": 1.5972222222222223, + "grad_norm": 3.0168304443359375, + "learning_rate": 2.633302151543499e-06, + "loss": 1.5394, + "step": 1265 + }, + { + "epoch": 1.5984848484848486, + "grad_norm": 2.9531311988830566, + "learning_rate": 2.6309635173058932e-06, + "loss": 1.4775, + "step": 1266 + }, + { + "epoch": 1.5997474747474747, + "grad_norm": 2.835839033126831, + "learning_rate": 2.628624883068288e-06, + "loss": 1.478, + "step": 1267 + }, + { + "epoch": 1.601010101010101, + "grad_norm": 2.9014039039611816, + "learning_rate": 2.626286248830683e-06, + "loss": 1.4926, + "step": 1268 + }, + { + "epoch": 1.6022727272727273, + "grad_norm": 2.89461088180542, + "learning_rate": 2.623947614593078e-06, + "loss": 1.448, + "step": 1269 + }, + { + "epoch": 1.6035353535353534, + "grad_norm": 2.8005120754241943, + "learning_rate": 2.621608980355473e-06, + "loss": 1.4733, + "step": 1270 + }, + { + "epoch": 1.6035353535353534, + "eval_loss": 1.6889415979385376, + "eval_runtime": 37.0345, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1270 + }, + { + "epoch": 1.6047979797979797, + "grad_norm": 2.9576175212860107, + "learning_rate": 2.6192703461178675e-06, + "loss": 1.5116, + "step": 1271 + }, + { + "epoch": 1.606060606060606, + "grad_norm": 2.926638603210449, + "learning_rate": 2.6169317118802623e-06, + "loss": 1.4744, + "step": 1272 + }, + { + "epoch": 1.6073232323232323, + "grad_norm": 2.911587715148926, + "learning_rate": 2.614593077642657e-06, + "loss": 1.4695, + "step": 1273 + }, + { + "epoch": 1.6085858585858586, + "grad_norm": 2.8878750801086426, + "learning_rate": 2.6122544434050516e-06, + "loss": 1.4574, + "step": 1274 + }, + { + "epoch": 1.6098484848484849, + "grad_norm": 2.8489067554473877, + "learning_rate": 2.6099158091674464e-06, + "loss": 1.4599, + "step": 1275 + }, + { + "epoch": 1.6111111111111112, + "grad_norm": 2.8223719596862793, + "learning_rate": 2.6075771749298412e-06, + "loss": 1.4431, + "step": 1276 + }, + { + "epoch": 1.6123737373737375, + "grad_norm": 2.869910955429077, + "learning_rate": 2.605238540692236e-06, + "loss": 1.48, + "step": 1277 + }, + { + "epoch": 1.6136363636363638, + "grad_norm": 2.8435168266296387, + "learning_rate": 2.6028999064546305e-06, + "loss": 1.4471, + "step": 1278 + }, + { + "epoch": 1.61489898989899, + "grad_norm": 2.9384665489196777, + "learning_rate": 2.6005612722170253e-06, + "loss": 1.5176, + "step": 1279 + }, + { + "epoch": 1.6161616161616161, + "grad_norm": 3.021785020828247, + "learning_rate": 2.59822263797942e-06, + "loss": 1.4789, + "step": 1280 + }, + { + "epoch": 1.6161616161616161, + "eval_loss": 1.6864516735076904, + "eval_runtime": 37.0249, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1280 + }, + { + "epoch": 1.6174242424242424, + "grad_norm": 3.007183313369751, + "learning_rate": 2.5958840037418146e-06, + "loss": 1.4686, + "step": 1281 + }, + { + "epoch": 1.6186868686868687, + "grad_norm": 3.115910053253174, + "learning_rate": 2.5935453695042095e-06, + "loss": 1.5382, + "step": 1282 + }, + { + "epoch": 1.6199494949494948, + "grad_norm": 2.98108172416687, + "learning_rate": 2.5912067352666043e-06, + "loss": 1.4569, + "step": 1283 + }, + { + "epoch": 1.621212121212121, + "grad_norm": 2.926832437515259, + "learning_rate": 2.5888681010289996e-06, + "loss": 1.4515, + "step": 1284 + }, + { + "epoch": 1.6224747474747474, + "grad_norm": 2.9045534133911133, + "learning_rate": 2.5865294667913944e-06, + "loss": 1.4259, + "step": 1285 + }, + { + "epoch": 1.6237373737373737, + "grad_norm": 3.0757768154144287, + "learning_rate": 2.584190832553789e-06, + "loss": 1.5436, + "step": 1286 + }, + { + "epoch": 1.625, + "grad_norm": 3.045257806777954, + "learning_rate": 2.5818521983161837e-06, + "loss": 1.4378, + "step": 1287 + }, + { + "epoch": 1.6262626262626263, + "grad_norm": 2.8131601810455322, + "learning_rate": 2.5795135640785785e-06, + "loss": 1.4224, + "step": 1288 + }, + { + "epoch": 1.6275252525252526, + "grad_norm": 2.953326940536499, + "learning_rate": 2.5771749298409734e-06, + "loss": 1.4137, + "step": 1289 + }, + { + "epoch": 1.628787878787879, + "grad_norm": 2.946232318878174, + "learning_rate": 2.5748362956033678e-06, + "loss": 1.4575, + "step": 1290 + }, + { + "epoch": 1.628787878787879, + "eval_loss": 1.6896334886550903, + "eval_runtime": 37.317, + "eval_samples_per_second": 26.797, + "eval_steps_per_second": 1.125, + "step": 1290 + }, + { + "epoch": 1.6300505050505052, + "grad_norm": 2.860867738723755, + "learning_rate": 2.5724976613657626e-06, + "loss": 1.4412, + "step": 1291 + }, + { + "epoch": 1.6313131313131313, + "grad_norm": 3.0417261123657227, + "learning_rate": 2.5701590271281575e-06, + "loss": 1.4919, + "step": 1292 + }, + { + "epoch": 1.6325757575757576, + "grad_norm": 3.21425199508667, + "learning_rate": 2.567820392890552e-06, + "loss": 1.5243, + "step": 1293 + }, + { + "epoch": 1.6338383838383839, + "grad_norm": 3.000302314758301, + "learning_rate": 2.5654817586529467e-06, + "loss": 1.4571, + "step": 1294 + }, + { + "epoch": 1.63510101010101, + "grad_norm": 2.88415789604187, + "learning_rate": 2.5631431244153416e-06, + "loss": 1.4527, + "step": 1295 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.7747156620025635, + "learning_rate": 2.5608044901777364e-06, + "loss": 1.3984, + "step": 1296 + }, + { + "epoch": 1.6376262626262625, + "grad_norm": 2.9298319816589355, + "learning_rate": 2.558465855940131e-06, + "loss": 1.4587, + "step": 1297 + }, + { + "epoch": 1.6388888888888888, + "grad_norm": 3.0199968814849854, + "learning_rate": 2.5561272217025257e-06, + "loss": 1.4539, + "step": 1298 + }, + { + "epoch": 1.6401515151515151, + "grad_norm": 3.09503436088562, + "learning_rate": 2.5537885874649205e-06, + "loss": 1.4551, + "step": 1299 + }, + { + "epoch": 1.6414141414141414, + "grad_norm": 2.972161293029785, + "learning_rate": 2.5514499532273158e-06, + "loss": 1.4123, + "step": 1300 + }, + { + "epoch": 1.6414141414141414, + "eval_loss": 1.6877377033233643, + "eval_runtime": 37.4088, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1300 + }, + { + "epoch": 1.6426767676767677, + "grad_norm": 3.081058979034424, + "learning_rate": 2.5491113189897106e-06, + "loss": 1.457, + "step": 1301 + }, + { + "epoch": 1.643939393939394, + "grad_norm": 3.040106773376465, + "learning_rate": 2.546772684752105e-06, + "loss": 1.4465, + "step": 1302 + }, + { + "epoch": 1.6452020202020203, + "grad_norm": 3.133812427520752, + "learning_rate": 2.5444340505145e-06, + "loss": 1.4382, + "step": 1303 + }, + { + "epoch": 1.6464646464646466, + "grad_norm": 2.9701125621795654, + "learning_rate": 2.5420954162768947e-06, + "loss": 1.457, + "step": 1304 + }, + { + "epoch": 1.6477272727272727, + "grad_norm": 2.9272050857543945, + "learning_rate": 2.539756782039289e-06, + "loss": 1.464, + "step": 1305 + }, + { + "epoch": 1.648989898989899, + "grad_norm": 3.048401117324829, + "learning_rate": 2.537418147801684e-06, + "loss": 1.4688, + "step": 1306 + }, + { + "epoch": 1.6502525252525253, + "grad_norm": 2.9103102684020996, + "learning_rate": 2.535079513564079e-06, + "loss": 1.5025, + "step": 1307 + }, + { + "epoch": 1.6515151515151514, + "grad_norm": 3.0841548442840576, + "learning_rate": 2.5327408793264737e-06, + "loss": 1.4938, + "step": 1308 + }, + { + "epoch": 1.6527777777777777, + "grad_norm": 3.022184371948242, + "learning_rate": 2.530402245088868e-06, + "loss": 1.4332, + "step": 1309 + }, + { + "epoch": 1.654040404040404, + "grad_norm": 2.881479024887085, + "learning_rate": 2.528063610851263e-06, + "loss": 1.4541, + "step": 1310 + }, + { + "epoch": 1.654040404040404, + "eval_loss": 1.6814675331115723, + "eval_runtime": 36.9446, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.6553030303030303, + "grad_norm": 2.906719446182251, + "learning_rate": 2.5257249766136578e-06, + "loss": 1.4968, + "step": 1311 + }, + { + "epoch": 1.6565656565656566, + "grad_norm": 3.0566461086273193, + "learning_rate": 2.523386342376052e-06, + "loss": 1.4513, + "step": 1312 + }, + { + "epoch": 1.6578282828282829, + "grad_norm": 2.895026206970215, + "learning_rate": 2.521047708138447e-06, + "loss": 1.4217, + "step": 1313 + }, + { + "epoch": 1.6590909090909092, + "grad_norm": 3.0006465911865234, + "learning_rate": 2.518709073900842e-06, + "loss": 1.4793, + "step": 1314 + }, + { + "epoch": 1.6603535353535355, + "grad_norm": 2.9853575229644775, + "learning_rate": 2.516370439663237e-06, + "loss": 1.4642, + "step": 1315 + }, + { + "epoch": 1.6616161616161618, + "grad_norm": 2.9729392528533936, + "learning_rate": 2.514031805425632e-06, + "loss": 1.4644, + "step": 1316 + }, + { + "epoch": 1.6628787878787878, + "grad_norm": 3.0997586250305176, + "learning_rate": 2.5116931711880264e-06, + "loss": 1.4392, + "step": 1317 + }, + { + "epoch": 1.6641414141414141, + "grad_norm": 2.9621379375457764, + "learning_rate": 2.5093545369504213e-06, + "loss": 1.4387, + "step": 1318 + }, + { + "epoch": 1.6654040404040404, + "grad_norm": 3.0571911334991455, + "learning_rate": 2.507015902712816e-06, + "loss": 1.4638, + "step": 1319 + }, + { + "epoch": 1.6666666666666665, + "grad_norm": 2.9580321311950684, + "learning_rate": 2.504677268475211e-06, + "loss": 1.4985, + "step": 1320 + }, + { + "epoch": 1.6666666666666665, + "eval_loss": 1.6790977716445923, + "eval_runtime": 37.0242, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1320 + }, + { + "epoch": 1.6679292929292928, + "grad_norm": 2.87432599067688, + "learning_rate": 2.5023386342376054e-06, + "loss": 1.4058, + "step": 1321 + }, + { + "epoch": 1.6691919191919191, + "grad_norm": 2.925546407699585, + "learning_rate": 2.5e-06, + "loss": 1.4986, + "step": 1322 + }, + { + "epoch": 1.6704545454545454, + "grad_norm": 2.837743043899536, + "learning_rate": 2.497661365762395e-06, + "loss": 1.455, + "step": 1323 + }, + { + "epoch": 1.6717171717171717, + "grad_norm": 3.111581563949585, + "learning_rate": 2.4953227315247895e-06, + "loss": 1.5005, + "step": 1324 + }, + { + "epoch": 1.672979797979798, + "grad_norm": 2.851076126098633, + "learning_rate": 2.4929840972871843e-06, + "loss": 1.4365, + "step": 1325 + }, + { + "epoch": 1.6742424242424243, + "grad_norm": 2.966965675354004, + "learning_rate": 2.4906454630495796e-06, + "loss": 1.4697, + "step": 1326 + }, + { + "epoch": 1.6755050505050506, + "grad_norm": 3.015193223953247, + "learning_rate": 2.488306828811974e-06, + "loss": 1.4486, + "step": 1327 + }, + { + "epoch": 1.676767676767677, + "grad_norm": 3.017681121826172, + "learning_rate": 2.485968194574369e-06, + "loss": 1.4017, + "step": 1328 + }, + { + "epoch": 1.678030303030303, + "grad_norm": 3.071981906890869, + "learning_rate": 2.4836295603367637e-06, + "loss": 1.4572, + "step": 1329 + }, + { + "epoch": 1.6792929292929293, + "grad_norm": 3.051741123199463, + "learning_rate": 2.481290926099158e-06, + "loss": 1.4498, + "step": 1330 + }, + { + "epoch": 1.6792929292929293, + "eval_loss": 1.6771777868270874, + "eval_runtime": 37.0747, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1330 + }, + { + "epoch": 1.6805555555555556, + "grad_norm": 2.9561071395874023, + "learning_rate": 2.478952291861553e-06, + "loss": 1.4428, + "step": 1331 + }, + { + "epoch": 1.6818181818181817, + "grad_norm": 2.992584466934204, + "learning_rate": 2.4766136576239478e-06, + "loss": 1.461, + "step": 1332 + }, + { + "epoch": 1.683080808080808, + "grad_norm": 3.0473921298980713, + "learning_rate": 2.4742750233863426e-06, + "loss": 1.489, + "step": 1333 + }, + { + "epoch": 1.6843434343434343, + "grad_norm": 2.8721725940704346, + "learning_rate": 2.4719363891487375e-06, + "loss": 1.487, + "step": 1334 + }, + { + "epoch": 1.6856060606060606, + "grad_norm": 2.8650765419006348, + "learning_rate": 2.4695977549111323e-06, + "loss": 1.4504, + "step": 1335 + }, + { + "epoch": 1.6868686868686869, + "grad_norm": 2.998887538909912, + "learning_rate": 2.4672591206735267e-06, + "loss": 1.4586, + "step": 1336 + }, + { + "epoch": 1.6881313131313131, + "grad_norm": 2.915188789367676, + "learning_rate": 2.4649204864359216e-06, + "loss": 1.4333, + "step": 1337 + }, + { + "epoch": 1.6893939393939394, + "grad_norm": 2.9660091400146484, + "learning_rate": 2.4625818521983164e-06, + "loss": 1.4711, + "step": 1338 + }, + { + "epoch": 1.6906565656565657, + "grad_norm": 2.916274070739746, + "learning_rate": 2.4602432179607113e-06, + "loss": 1.4642, + "step": 1339 + }, + { + "epoch": 1.691919191919192, + "grad_norm": 3.04154896736145, + "learning_rate": 2.4579045837231057e-06, + "loss": 1.4233, + "step": 1340 + }, + { + "epoch": 1.691919191919192, + "eval_loss": 1.6764302253723145, + "eval_runtime": 37.0686, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1340 + }, + { + "epoch": 1.6931818181818183, + "grad_norm": 2.9976465702056885, + "learning_rate": 2.4555659494855005e-06, + "loss": 1.4775, + "step": 1341 + }, + { + "epoch": 1.6944444444444444, + "grad_norm": 2.9140090942382812, + "learning_rate": 2.4532273152478954e-06, + "loss": 1.4532, + "step": 1342 + }, + { + "epoch": 1.6957070707070707, + "grad_norm": 3.0224454402923584, + "learning_rate": 2.45088868101029e-06, + "loss": 1.5091, + "step": 1343 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.8979439735412598, + "learning_rate": 2.448550046772685e-06, + "loss": 1.4634, + "step": 1344 + }, + { + "epoch": 1.698232323232323, + "grad_norm": 2.913433313369751, + "learning_rate": 2.44621141253508e-06, + "loss": 1.4483, + "step": 1345 + }, + { + "epoch": 1.6994949494949494, + "grad_norm": 2.813455820083618, + "learning_rate": 2.4438727782974743e-06, + "loss": 1.446, + "step": 1346 + }, + { + "epoch": 1.7007575757575757, + "grad_norm": 2.988722324371338, + "learning_rate": 2.441534144059869e-06, + "loss": 1.4908, + "step": 1347 + }, + { + "epoch": 1.702020202020202, + "grad_norm": 2.850446939468384, + "learning_rate": 2.439195509822264e-06, + "loss": 1.4221, + "step": 1348 + }, + { + "epoch": 1.7032828282828283, + "grad_norm": 2.8405508995056152, + "learning_rate": 2.436856875584659e-06, + "loss": 1.4496, + "step": 1349 + }, + { + "epoch": 1.7045454545454546, + "grad_norm": 3.0188651084899902, + "learning_rate": 2.4345182413470537e-06, + "loss": 1.489, + "step": 1350 + }, + { + "epoch": 1.7045454545454546, + "eval_loss": 1.6780083179473877, + "eval_runtime": 37.0724, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 1350 + }, + { + "epoch": 1.7058080808080809, + "grad_norm": 2.905088424682617, + "learning_rate": 2.4321796071094485e-06, + "loss": 1.4283, + "step": 1351 + }, + { + "epoch": 1.7070707070707072, + "grad_norm": 2.878998279571533, + "learning_rate": 2.429840972871843e-06, + "loss": 1.4357, + "step": 1352 + }, + { + "epoch": 1.7083333333333335, + "grad_norm": 3.1707355976104736, + "learning_rate": 2.427502338634238e-06, + "loss": 1.496, + "step": 1353 + }, + { + "epoch": 1.7095959595959596, + "grad_norm": 2.9610493183135986, + "learning_rate": 2.4251637043966326e-06, + "loss": 1.4905, + "step": 1354 + }, + { + "epoch": 1.7108585858585859, + "grad_norm": 2.8372201919555664, + "learning_rate": 2.422825070159027e-06, + "loss": 1.4563, + "step": 1355 + }, + { + "epoch": 1.7121212121212122, + "grad_norm": 2.8741753101348877, + "learning_rate": 2.420486435921422e-06, + "loss": 1.4525, + "step": 1356 + }, + { + "epoch": 1.7133838383838382, + "grad_norm": 2.973726749420166, + "learning_rate": 2.418147801683817e-06, + "loss": 1.4878, + "step": 1357 + }, + { + "epoch": 1.7146464646464645, + "grad_norm": 3.05717396736145, + "learning_rate": 2.4158091674462116e-06, + "loss": 1.5402, + "step": 1358 + }, + { + "epoch": 1.7159090909090908, + "grad_norm": 2.883772373199463, + "learning_rate": 2.4134705332086064e-06, + "loss": 1.4329, + "step": 1359 + }, + { + "epoch": 1.7171717171717171, + "grad_norm": 2.8246653079986572, + "learning_rate": 2.4111318989710013e-06, + "loss": 1.4821, + "step": 1360 + }, + { + "epoch": 1.7171717171717171, + "eval_loss": 1.6737746000289917, + "eval_runtime": 37.0666, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 1360 + }, + { + "epoch": 1.7184343434343434, + "grad_norm": 2.905977964401245, + "learning_rate": 2.4087932647333957e-06, + "loss": 1.4378, + "step": 1361 + }, + { + "epoch": 1.7196969696969697, + "grad_norm": 2.9268195629119873, + "learning_rate": 2.4064546304957905e-06, + "loss": 1.4765, + "step": 1362 + }, + { + "epoch": 1.720959595959596, + "grad_norm": 3.1369476318359375, + "learning_rate": 2.4041159962581854e-06, + "loss": 1.4532, + "step": 1363 + }, + { + "epoch": 1.7222222222222223, + "grad_norm": 3.04546856880188, + "learning_rate": 2.4017773620205802e-06, + "loss": 1.4921, + "step": 1364 + }, + { + "epoch": 1.7234848484848486, + "grad_norm": 2.872852325439453, + "learning_rate": 2.399438727782975e-06, + "loss": 1.4323, + "step": 1365 + }, + { + "epoch": 1.7247474747474747, + "grad_norm": 2.961681604385376, + "learning_rate": 2.39710009354537e-06, + "loss": 1.5181, + "step": 1366 + }, + { + "epoch": 1.726010101010101, + "grad_norm": 2.997729778289795, + "learning_rate": 2.3947614593077643e-06, + "loss": 1.4817, + "step": 1367 + }, + { + "epoch": 1.7272727272727273, + "grad_norm": 2.9097466468811035, + "learning_rate": 2.392422825070159e-06, + "loss": 1.4478, + "step": 1368 + }, + { + "epoch": 1.7285353535353534, + "grad_norm": 2.82433819770813, + "learning_rate": 2.390084190832554e-06, + "loss": 1.4065, + "step": 1369 + }, + { + "epoch": 1.7297979797979797, + "grad_norm": 2.940910577774048, + "learning_rate": 2.387745556594949e-06, + "loss": 1.4735, + "step": 1370 + }, + { + "epoch": 1.7297979797979797, + "eval_loss": 1.6717814207077026, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1370 + }, + { + "epoch": 1.731060606060606, + "grad_norm": 2.79101300239563, + "learning_rate": 2.3854069223573433e-06, + "loss": 1.4545, + "step": 1371 + }, + { + "epoch": 1.7323232323232323, + "grad_norm": 2.916954517364502, + "learning_rate": 2.3830682881197385e-06, + "loss": 1.4293, + "step": 1372 + }, + { + "epoch": 1.7335858585858586, + "grad_norm": 2.9609909057617188, + "learning_rate": 2.380729653882133e-06, + "loss": 1.4578, + "step": 1373 + }, + { + "epoch": 1.7348484848484849, + "grad_norm": 2.935925245285034, + "learning_rate": 2.378391019644528e-06, + "loss": 1.4577, + "step": 1374 + }, + { + "epoch": 1.7361111111111112, + "grad_norm": 2.9803526401519775, + "learning_rate": 2.3760523854069226e-06, + "loss": 1.4449, + "step": 1375 + }, + { + "epoch": 1.7373737373737375, + "grad_norm": 3.0292913913726807, + "learning_rate": 2.3737137511693175e-06, + "loss": 1.4308, + "step": 1376 + }, + { + "epoch": 1.7386363636363638, + "grad_norm": 3.0247879028320312, + "learning_rate": 2.371375116931712e-06, + "loss": 1.5406, + "step": 1377 + }, + { + "epoch": 1.73989898989899, + "grad_norm": 2.86775541305542, + "learning_rate": 2.3690364826941067e-06, + "loss": 1.4324, + "step": 1378 + }, + { + "epoch": 1.7411616161616161, + "grad_norm": 2.8593552112579346, + "learning_rate": 2.3666978484565016e-06, + "loss": 1.4526, + "step": 1379 + }, + { + "epoch": 1.7424242424242424, + "grad_norm": 2.8735320568084717, + "learning_rate": 2.3643592142188964e-06, + "loss": 1.4097, + "step": 1380 + }, + { + "epoch": 1.7424242424242424, + "eval_loss": 1.675827145576477, + "eval_runtime": 37.1206, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1380 + }, + { + "epoch": 1.7436868686868687, + "grad_norm": 2.948153495788574, + "learning_rate": 2.3620205799812913e-06, + "loss": 1.4234, + "step": 1381 + }, + { + "epoch": 1.7449494949494948, + "grad_norm": 2.9344282150268555, + "learning_rate": 2.359681945743686e-06, + "loss": 1.4653, + "step": 1382 + }, + { + "epoch": 1.746212121212121, + "grad_norm": 2.9309439659118652, + "learning_rate": 2.3573433115060805e-06, + "loss": 1.4676, + "step": 1383 + }, + { + "epoch": 1.7474747474747474, + "grad_norm": 3.0762970447540283, + "learning_rate": 2.3550046772684754e-06, + "loss": 1.4407, + "step": 1384 + }, + { + "epoch": 1.7487373737373737, + "grad_norm": 3.149824857711792, + "learning_rate": 2.3526660430308702e-06, + "loss": 1.4629, + "step": 1385 + }, + { + "epoch": 1.75, + "grad_norm": 3.109571695327759, + "learning_rate": 2.3503274087932646e-06, + "loss": 1.494, + "step": 1386 + }, + { + "epoch": 1.7512626262626263, + "grad_norm": 2.8359758853912354, + "learning_rate": 2.34798877455566e-06, + "loss": 1.4056, + "step": 1387 + }, + { + "epoch": 1.7525252525252526, + "grad_norm": 2.8523781299591064, + "learning_rate": 2.3456501403180548e-06, + "loss": 1.4389, + "step": 1388 + }, + { + "epoch": 1.753787878787879, + "grad_norm": 3.0218093395233154, + "learning_rate": 2.343311506080449e-06, + "loss": 1.4587, + "step": 1389 + }, + { + "epoch": 1.7550505050505052, + "grad_norm": 3.0975496768951416, + "learning_rate": 2.340972871842844e-06, + "loss": 1.5133, + "step": 1390 + }, + { + "epoch": 1.7550505050505052, + "eval_loss": 1.671931266784668, + "eval_runtime": 37.2749, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 1390 + }, + { + "epoch": 1.7563131313131313, + "grad_norm": 3.0177836418151855, + "learning_rate": 2.338634237605239e-06, + "loss": 1.4716, + "step": 1391 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 3.049105167388916, + "learning_rate": 2.3362956033676333e-06, + "loss": 1.4858, + "step": 1392 + }, + { + "epoch": 1.7588383838383839, + "grad_norm": 3.0062026977539062, + "learning_rate": 2.333956969130028e-06, + "loss": 1.4239, + "step": 1393 + }, + { + "epoch": 1.76010101010101, + "grad_norm": 3.071772813796997, + "learning_rate": 2.331618334892423e-06, + "loss": 1.4654, + "step": 1394 + }, + { + "epoch": 1.7613636363636362, + "grad_norm": 3.140510320663452, + "learning_rate": 2.329279700654818e-06, + "loss": 1.4854, + "step": 1395 + }, + { + "epoch": 1.7626262626262625, + "grad_norm": 2.9903907775878906, + "learning_rate": 2.3269410664172126e-06, + "loss": 1.4406, + "step": 1396 + }, + { + "epoch": 1.7638888888888888, + "grad_norm": 2.839891195297241, + "learning_rate": 2.3246024321796075e-06, + "loss": 1.4498, + "step": 1397 + }, + { + "epoch": 1.7651515151515151, + "grad_norm": 3.0252437591552734, + "learning_rate": 2.322263797942002e-06, + "loss": 1.472, + "step": 1398 + }, + { + "epoch": 1.7664141414141414, + "grad_norm": 3.0214762687683105, + "learning_rate": 2.3199251637043968e-06, + "loss": 1.4031, + "step": 1399 + }, + { + "epoch": 1.7676767676767677, + "grad_norm": 3.190088987350464, + "learning_rate": 2.3175865294667916e-06, + "loss": 1.4497, + "step": 1400 + }, + { + "epoch": 1.7676767676767677, + "eval_loss": 1.673120379447937, + "eval_runtime": 37.4414, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1400 + }, + { + "epoch": 1.768939393939394, + "grad_norm": 2.9620201587677, + "learning_rate": 2.3152478952291864e-06, + "loss": 1.4149, + "step": 1401 + }, + { + "epoch": 1.7702020202020203, + "grad_norm": 3.112372875213623, + "learning_rate": 2.312909260991581e-06, + "loss": 1.4569, + "step": 1402 + }, + { + "epoch": 1.7714646464646466, + "grad_norm": 3.138857364654541, + "learning_rate": 2.310570626753976e-06, + "loss": 1.4819, + "step": 1403 + }, + { + "epoch": 1.7727272727272727, + "grad_norm": 3.0957937240600586, + "learning_rate": 2.3082319925163705e-06, + "loss": 1.4374, + "step": 1404 + }, + { + "epoch": 1.773989898989899, + "grad_norm": 3.014039993286133, + "learning_rate": 2.3058933582787654e-06, + "loss": 1.4262, + "step": 1405 + }, + { + "epoch": 1.7752525252525253, + "grad_norm": 2.8127005100250244, + "learning_rate": 2.3035547240411602e-06, + "loss": 1.4468, + "step": 1406 + }, + { + "epoch": 1.7765151515151514, + "grad_norm": 2.9180614948272705, + "learning_rate": 2.301216089803555e-06, + "loss": 1.4259, + "step": 1407 + }, + { + "epoch": 1.7777777777777777, + "grad_norm": 3.0376124382019043, + "learning_rate": 2.2988774555659495e-06, + "loss": 1.4628, + "step": 1408 + }, + { + "epoch": 1.779040404040404, + "grad_norm": 3.031740665435791, + "learning_rate": 2.2965388213283443e-06, + "loss": 1.5058, + "step": 1409 + }, + { + "epoch": 1.7803030303030303, + "grad_norm": 2.9876646995544434, + "learning_rate": 2.294200187090739e-06, + "loss": 1.4605, + "step": 1410 + }, + { + "epoch": 1.7803030303030303, + "eval_loss": 1.668077826499939, + "eval_runtime": 36.9734, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 1410 + }, + { + "epoch": 1.7815656565656566, + "grad_norm": 2.953744888305664, + "learning_rate": 2.291861552853134e-06, + "loss": 1.4257, + "step": 1411 + }, + { + "epoch": 1.7828282828282829, + "grad_norm": 3.049036741256714, + "learning_rate": 2.289522918615529e-06, + "loss": 1.4476, + "step": 1412 + }, + { + "epoch": 1.7840909090909092, + "grad_norm": 3.102116584777832, + "learning_rate": 2.2871842843779237e-06, + "loss": 1.4701, + "step": 1413 + }, + { + "epoch": 1.7853535353535355, + "grad_norm": 2.842970132827759, + "learning_rate": 2.284845650140318e-06, + "loss": 1.415, + "step": 1414 + }, + { + "epoch": 1.7866161616161618, + "grad_norm": 2.9386725425720215, + "learning_rate": 2.282507015902713e-06, + "loss": 1.4561, + "step": 1415 + }, + { + "epoch": 1.7878787878787878, + "grad_norm": 2.942713499069214, + "learning_rate": 2.280168381665108e-06, + "loss": 1.435, + "step": 1416 + }, + { + "epoch": 1.7891414141414141, + "grad_norm": 2.943105936050415, + "learning_rate": 2.2778297474275022e-06, + "loss": 1.4923, + "step": 1417 + }, + { + "epoch": 1.7904040404040404, + "grad_norm": 2.7094717025756836, + "learning_rate": 2.2754911131898975e-06, + "loss": 1.4457, + "step": 1418 + }, + { + "epoch": 1.7916666666666665, + "grad_norm": 2.7859625816345215, + "learning_rate": 2.2731524789522923e-06, + "loss": 1.3993, + "step": 1419 + }, + { + "epoch": 1.7929292929292928, + "grad_norm": 3.0923988819122314, + "learning_rate": 2.2708138447146868e-06, + "loss": 1.5054, + "step": 1420 + }, + { + "epoch": 1.7929292929292928, + "eval_loss": 1.6687692403793335, + "eval_runtime": 37.0567, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.7941919191919191, + "grad_norm": 2.849726676940918, + "learning_rate": 2.2684752104770816e-06, + "loss": 1.3978, + "step": 1421 + }, + { + "epoch": 1.7954545454545454, + "grad_norm": 3.028932809829712, + "learning_rate": 2.2661365762394764e-06, + "loss": 1.4786, + "step": 1422 + }, + { + "epoch": 1.7967171717171717, + "grad_norm": 2.973188638687134, + "learning_rate": 2.263797942001871e-06, + "loss": 1.4393, + "step": 1423 + }, + { + "epoch": 1.797979797979798, + "grad_norm": 2.884032964706421, + "learning_rate": 2.2614593077642657e-06, + "loss": 1.4197, + "step": 1424 + }, + { + "epoch": 1.7992424242424243, + "grad_norm": 2.9058549404144287, + "learning_rate": 2.2591206735266606e-06, + "loss": 1.4313, + "step": 1425 + }, + { + "epoch": 1.8005050505050506, + "grad_norm": 2.905841827392578, + "learning_rate": 2.2567820392890554e-06, + "loss": 1.439, + "step": 1426 + }, + { + "epoch": 1.801767676767677, + "grad_norm": 2.927028179168701, + "learning_rate": 2.2544434050514502e-06, + "loss": 1.4296, + "step": 1427 + }, + { + "epoch": 1.803030303030303, + "grad_norm": 2.965529203414917, + "learning_rate": 2.252104770813845e-06, + "loss": 1.4185, + "step": 1428 + }, + { + "epoch": 1.8042929292929293, + "grad_norm": 2.9494433403015137, + "learning_rate": 2.2497661365762395e-06, + "loss": 1.5087, + "step": 1429 + }, + { + "epoch": 1.8055555555555556, + "grad_norm": 3.0137765407562256, + "learning_rate": 2.2474275023386343e-06, + "loss": 1.4383, + "step": 1430 + }, + { + "epoch": 1.8055555555555556, + "eval_loss": 1.6728461980819702, + "eval_runtime": 37.1033, + "eval_samples_per_second": 26.952, + "eval_steps_per_second": 1.132, + "step": 1430 + }, + { + "epoch": 1.8068181818181817, + "grad_norm": 2.980062484741211, + "learning_rate": 2.245088868101029e-06, + "loss": 1.4333, + "step": 1431 + }, + { + "epoch": 1.808080808080808, + "grad_norm": 3.143759250640869, + "learning_rate": 2.242750233863424e-06, + "loss": 1.4676, + "step": 1432 + }, + { + "epoch": 1.8093434343434343, + "grad_norm": 3.0134518146514893, + "learning_rate": 2.240411599625819e-06, + "loss": 1.4223, + "step": 1433 + }, + { + "epoch": 1.8106060606060606, + "grad_norm": 3.000018358230591, + "learning_rate": 2.2380729653882137e-06, + "loss": 1.4537, + "step": 1434 + }, + { + "epoch": 1.8118686868686869, + "grad_norm": 2.9828288555145264, + "learning_rate": 2.235734331150608e-06, + "loss": 1.4379, + "step": 1435 + }, + { + "epoch": 1.8131313131313131, + "grad_norm": 2.965486764907837, + "learning_rate": 2.233395696913003e-06, + "loss": 1.4244, + "step": 1436 + }, + { + "epoch": 1.8143939393939394, + "grad_norm": 2.9787070751190186, + "learning_rate": 2.231057062675398e-06, + "loss": 1.4117, + "step": 1437 + }, + { + "epoch": 1.8156565656565657, + "grad_norm": 3.0893537998199463, + "learning_rate": 2.2287184284377927e-06, + "loss": 1.4809, + "step": 1438 + }, + { + "epoch": 1.816919191919192, + "grad_norm": 2.8568801879882812, + "learning_rate": 2.226379794200187e-06, + "loss": 1.4232, + "step": 1439 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 3.1005046367645264, + "learning_rate": 2.224041159962582e-06, + "loss": 1.482, + "step": 1440 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.6738661527633667, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1440 + }, + { + "epoch": 1.8194444444444444, + "grad_norm": 2.9064700603485107, + "learning_rate": 2.2217025257249768e-06, + "loss": 1.4179, + "step": 1441 + }, + { + "epoch": 1.8207070707070707, + "grad_norm": 3.0111782550811768, + "learning_rate": 2.2193638914873716e-06, + "loss": 1.4084, + "step": 1442 + }, + { + "epoch": 1.821969696969697, + "grad_norm": 2.9269816875457764, + "learning_rate": 2.2170252572497665e-06, + "loss": 1.4361, + "step": 1443 + }, + { + "epoch": 1.823232323232323, + "grad_norm": 2.9985902309417725, + "learning_rate": 2.2146866230121613e-06, + "loss": 1.4345, + "step": 1444 + }, + { + "epoch": 1.8244949494949494, + "grad_norm": 3.1306066513061523, + "learning_rate": 2.2123479887745557e-06, + "loss": 1.4728, + "step": 1445 + }, + { + "epoch": 1.8257575757575757, + "grad_norm": 2.9876086711883545, + "learning_rate": 2.2100093545369506e-06, + "loss": 1.378, + "step": 1446 + }, + { + "epoch": 1.827020202020202, + "grad_norm": 3.018763780593872, + "learning_rate": 2.2076707202993454e-06, + "loss": 1.4506, + "step": 1447 + }, + { + "epoch": 1.8282828282828283, + "grad_norm": 2.960573673248291, + "learning_rate": 2.2053320860617402e-06, + "loss": 1.4674, + "step": 1448 + }, + { + "epoch": 1.8295454545454546, + "grad_norm": 2.849881887435913, + "learning_rate": 2.202993451824135e-06, + "loss": 1.4687, + "step": 1449 + }, + { + "epoch": 1.8308080808080809, + "grad_norm": 2.8798882961273193, + "learning_rate": 2.2006548175865295e-06, + "loss": 1.3868, + "step": 1450 + }, + { + "epoch": 1.8308080808080809, + "eval_loss": 1.6657519340515137, + "eval_runtime": 37.0993, + "eval_samples_per_second": 26.955, + "eval_steps_per_second": 1.132, + "step": 1450 + }, + { + "epoch": 1.8320707070707072, + "grad_norm": 2.937471628189087, + "learning_rate": 2.1983161833489243e-06, + "loss": 1.4085, + "step": 1451 + }, + { + "epoch": 1.8333333333333335, + "grad_norm": 3.0815699100494385, + "learning_rate": 2.195977549111319e-06, + "loss": 1.4055, + "step": 1452 + }, + { + "epoch": 1.8345959595959596, + "grad_norm": 2.927917718887329, + "learning_rate": 2.193638914873714e-06, + "loss": 1.4014, + "step": 1453 + }, + { + "epoch": 1.8358585858585859, + "grad_norm": 2.9576570987701416, + "learning_rate": 2.1913002806361085e-06, + "loss": 1.4394, + "step": 1454 + }, + { + "epoch": 1.8371212121212122, + "grad_norm": 2.9505863189697266, + "learning_rate": 2.1889616463985033e-06, + "loss": 1.4018, + "step": 1455 + }, + { + "epoch": 1.8383838383838382, + "grad_norm": 3.014965772628784, + "learning_rate": 2.186623012160898e-06, + "loss": 1.3735, + "step": 1456 + }, + { + "epoch": 1.8396464646464645, + "grad_norm": 3.121122121810913, + "learning_rate": 2.184284377923293e-06, + "loss": 1.4289, + "step": 1457 + }, + { + "epoch": 1.8409090909090908, + "grad_norm": 2.9710142612457275, + "learning_rate": 2.181945743685688e-06, + "loss": 1.4494, + "step": 1458 + }, + { + "epoch": 1.8421717171717171, + "grad_norm": 3.028554916381836, + "learning_rate": 2.1796071094480827e-06, + "loss": 1.5151, + "step": 1459 + }, + { + "epoch": 1.8434343434343434, + "grad_norm": 2.8973751068115234, + "learning_rate": 2.177268475210477e-06, + "loss": 1.4171, + "step": 1460 + }, + { + "epoch": 1.8434343434343434, + "eval_loss": 1.6670986413955688, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1460 + }, + { + "epoch": 1.8446969696969697, + "grad_norm": 3.046865701675415, + "learning_rate": 2.174929840972872e-06, + "loss": 1.4016, + "step": 1461 + }, + { + "epoch": 1.845959595959596, + "grad_norm": 3.0509603023529053, + "learning_rate": 2.1725912067352668e-06, + "loss": 1.4196, + "step": 1462 + }, + { + "epoch": 1.8472222222222223, + "grad_norm": 2.951892137527466, + "learning_rate": 2.1702525724976616e-06, + "loss": 1.4063, + "step": 1463 + }, + { + "epoch": 1.8484848484848486, + "grad_norm": 3.2126667499542236, + "learning_rate": 2.1679139382600565e-06, + "loss": 1.5106, + "step": 1464 + }, + { + "epoch": 1.8497474747474747, + "grad_norm": 2.965576648712158, + "learning_rate": 2.1655753040224513e-06, + "loss": 1.4096, + "step": 1465 + }, + { + "epoch": 1.851010101010101, + "grad_norm": 2.8640592098236084, + "learning_rate": 2.1632366697848457e-06, + "loss": 1.389, + "step": 1466 + }, + { + "epoch": 1.8522727272727273, + "grad_norm": 3.1476943492889404, + "learning_rate": 2.1608980355472406e-06, + "loss": 1.4793, + "step": 1467 + }, + { + "epoch": 1.8535353535353534, + "grad_norm": 2.814695119857788, + "learning_rate": 2.1585594013096354e-06, + "loss": 1.4265, + "step": 1468 + }, + { + "epoch": 1.8547979797979797, + "grad_norm": 2.9728431701660156, + "learning_rate": 2.1562207670720302e-06, + "loss": 1.386, + "step": 1469 + }, + { + "epoch": 1.856060606060606, + "grad_norm": 2.9062652587890625, + "learning_rate": 2.1538821328344247e-06, + "loss": 1.3986, + "step": 1470 + }, + { + "epoch": 1.856060606060606, + "eval_loss": 1.6698763370513916, + "eval_runtime": 37.1129, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1470 + }, + { + "epoch": 1.8573232323232323, + "grad_norm": 2.950857162475586, + "learning_rate": 2.15154349859682e-06, + "loss": 1.4193, + "step": 1471 + }, + { + "epoch": 1.8585858585858586, + "grad_norm": 3.017594337463379, + "learning_rate": 2.1492048643592144e-06, + "loss": 1.4318, + "step": 1472 + }, + { + "epoch": 1.8598484848484849, + "grad_norm": 2.991514205932617, + "learning_rate": 2.146866230121609e-06, + "loss": 1.4462, + "step": 1473 + }, + { + "epoch": 1.8611111111111112, + "grad_norm": 3.149475336074829, + "learning_rate": 2.144527595884004e-06, + "loss": 1.4486, + "step": 1474 + }, + { + "epoch": 1.8623737373737375, + "grad_norm": 2.948828935623169, + "learning_rate": 2.1421889616463985e-06, + "loss": 1.4522, + "step": 1475 + }, + { + "epoch": 1.8636363636363638, + "grad_norm": 3.0383694171905518, + "learning_rate": 2.1398503274087933e-06, + "loss": 1.45, + "step": 1476 + }, + { + "epoch": 1.86489898989899, + "grad_norm": 3.093449831008911, + "learning_rate": 2.137511693171188e-06, + "loss": 1.4266, + "step": 1477 + }, + { + "epoch": 1.8661616161616161, + "grad_norm": 3.017549753189087, + "learning_rate": 2.135173058933583e-06, + "loss": 1.4067, + "step": 1478 + }, + { + "epoch": 1.8674242424242424, + "grad_norm": 2.985044002532959, + "learning_rate": 2.132834424695978e-06, + "loss": 1.4904, + "step": 1479 + }, + { + "epoch": 1.8686868686868687, + "grad_norm": 2.9600346088409424, + "learning_rate": 2.1304957904583727e-06, + "loss": 1.4006, + "step": 1480 + }, + { + "epoch": 1.8686868686868687, + "eval_loss": 1.6648480892181396, + "eval_runtime": 37.1225, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1480 + }, + { + "epoch": 1.8699494949494948, + "grad_norm": 2.8905625343322754, + "learning_rate": 2.128157156220767e-06, + "loss": 1.4096, + "step": 1481 + }, + { + "epoch": 1.871212121212121, + "grad_norm": 2.95342755317688, + "learning_rate": 2.125818521983162e-06, + "loss": 1.3888, + "step": 1482 + }, + { + "epoch": 1.8724747474747474, + "grad_norm": 2.9438536167144775, + "learning_rate": 2.1234798877455568e-06, + "loss": 1.413, + "step": 1483 + }, + { + "epoch": 1.8737373737373737, + "grad_norm": 3.1257364749908447, + "learning_rate": 2.1211412535079516e-06, + "loss": 1.4902, + "step": 1484 + }, + { + "epoch": 1.875, + "grad_norm": 3.1377382278442383, + "learning_rate": 2.118802619270346e-06, + "loss": 1.4815, + "step": 1485 + }, + { + "epoch": 1.8762626262626263, + "grad_norm": 3.122253656387329, + "learning_rate": 2.116463985032741e-06, + "loss": 1.4695, + "step": 1486 + }, + { + "epoch": 1.8775252525252526, + "grad_norm": 3.0128729343414307, + "learning_rate": 2.1141253507951357e-06, + "loss": 1.4673, + "step": 1487 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.921835422515869, + "learning_rate": 2.1117867165575306e-06, + "loss": 1.4325, + "step": 1488 + }, + { + "epoch": 1.8800505050505052, + "grad_norm": 3.19289231300354, + "learning_rate": 2.1094480823199254e-06, + "loss": 1.4504, + "step": 1489 + }, + { + "epoch": 1.8813131313131313, + "grad_norm": 2.8579299449920654, + "learning_rate": 2.1071094480823203e-06, + "loss": 1.4042, + "step": 1490 + }, + { + "epoch": 1.8813131313131313, + "eval_loss": 1.6603271961212158, + "eval_runtime": 37.2997, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 1490 + }, + { + "epoch": 1.8825757575757576, + "grad_norm": 2.9113969802856445, + "learning_rate": 2.1047708138447147e-06, + "loss": 1.4128, + "step": 1491 + }, + { + "epoch": 1.8838383838383839, + "grad_norm": 3.058133363723755, + "learning_rate": 2.1024321796071095e-06, + "loss": 1.4574, + "step": 1492 + }, + { + "epoch": 1.88510101010101, + "grad_norm": 3.0066583156585693, + "learning_rate": 2.1000935453695044e-06, + "loss": 1.4521, + "step": 1493 + }, + { + "epoch": 1.8863636363636362, + "grad_norm": 2.9154908657073975, + "learning_rate": 2.097754911131899e-06, + "loss": 1.4103, + "step": 1494 + }, + { + "epoch": 1.8876262626262625, + "grad_norm": 2.977062702178955, + "learning_rate": 2.095416276894294e-06, + "loss": 1.4003, + "step": 1495 + }, + { + "epoch": 1.8888888888888888, + "grad_norm": 2.9048168659210205, + "learning_rate": 2.093077642656689e-06, + "loss": 1.4216, + "step": 1496 + }, + { + "epoch": 1.8901515151515151, + "grad_norm": 3.008733034133911, + "learning_rate": 2.0907390084190833e-06, + "loss": 1.4516, + "step": 1497 + }, + { + "epoch": 1.8914141414141414, + "grad_norm": 3.0832529067993164, + "learning_rate": 2.088400374181478e-06, + "loss": 1.4248, + "step": 1498 + }, + { + "epoch": 1.8926767676767677, + "grad_norm": 2.848066568374634, + "learning_rate": 2.086061739943873e-06, + "loss": 1.4042, + "step": 1499 + }, + { + "epoch": 1.893939393939394, + "grad_norm": 2.846782684326172, + "learning_rate": 2.0837231057062674e-06, + "loss": 1.409, + "step": 1500 + }, + { + "epoch": 1.893939393939394, + "eval_loss": 1.6672260761260986, + "eval_runtime": 37.4695, + "eval_samples_per_second": 26.688, + "eval_steps_per_second": 1.121, + "step": 1500 + }, + { + "epoch": 1.8952020202020203, + "grad_norm": 3.026818037033081, + "learning_rate": 2.0813844714686623e-06, + "loss": 1.4842, + "step": 1501 + }, + { + "epoch": 1.8964646464646466, + "grad_norm": 2.977665662765503, + "learning_rate": 2.0790458372310575e-06, + "loss": 1.4635, + "step": 1502 + }, + { + "epoch": 1.8977272727272727, + "grad_norm": 2.9093048572540283, + "learning_rate": 2.076707202993452e-06, + "loss": 1.4303, + "step": 1503 + }, + { + "epoch": 1.898989898989899, + "grad_norm": 3.0777814388275146, + "learning_rate": 2.0743685687558468e-06, + "loss": 1.4581, + "step": 1504 + }, + { + "epoch": 1.9002525252525253, + "grad_norm": 3.0283970832824707, + "learning_rate": 2.0720299345182416e-06, + "loss": 1.4665, + "step": 1505 + }, + { + "epoch": 1.9015151515151514, + "grad_norm": 2.852412223815918, + "learning_rate": 2.069691300280636e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.9027777777777777, + "grad_norm": 3.1163125038146973, + "learning_rate": 2.067352666043031e-06, + "loss": 1.4907, + "step": 1507 + }, + { + "epoch": 1.904040404040404, + "grad_norm": 2.9857897758483887, + "learning_rate": 2.0650140318054257e-06, + "loss": 1.4262, + "step": 1508 + }, + { + "epoch": 1.9053030303030303, + "grad_norm": 2.957388401031494, + "learning_rate": 2.0626753975678206e-06, + "loss": 1.4525, + "step": 1509 + }, + { + "epoch": 1.9065656565656566, + "grad_norm": 3.0107505321502686, + "learning_rate": 2.0603367633302154e-06, + "loss": 1.4461, + "step": 1510 + }, + { + "epoch": 1.9065656565656566, + "eval_loss": 1.6621215343475342, + "eval_runtime": 36.9336, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1510 + }, + { + "epoch": 1.9078282828282829, + "grad_norm": 3.112478494644165, + "learning_rate": 2.0579981290926103e-06, + "loss": 1.479, + "step": 1511 + }, + { + "epoch": 1.9090909090909092, + "grad_norm": 3.083494186401367, + "learning_rate": 2.0556594948550047e-06, + "loss": 1.4777, + "step": 1512 + }, + { + "epoch": 1.9103535353535355, + "grad_norm": 2.932652711868286, + "learning_rate": 2.0533208606173995e-06, + "loss": 1.3937, + "step": 1513 + }, + { + "epoch": 1.9116161616161618, + "grad_norm": 2.8349449634552, + "learning_rate": 2.0509822263797944e-06, + "loss": 1.4096, + "step": 1514 + }, + { + "epoch": 1.9128787878787878, + "grad_norm": 2.9760472774505615, + "learning_rate": 2.048643592142189e-06, + "loss": 1.4312, + "step": 1515 + }, + { + "epoch": 1.9141414141414141, + "grad_norm": 3.1012730598449707, + "learning_rate": 2.0463049579045836e-06, + "loss": 1.5189, + "step": 1516 + }, + { + "epoch": 1.9154040404040404, + "grad_norm": 2.9072659015655518, + "learning_rate": 2.043966323666979e-06, + "loss": 1.4051, + "step": 1517 + }, + { + "epoch": 1.9166666666666665, + "grad_norm": 2.840203285217285, + "learning_rate": 2.0416276894293733e-06, + "loss": 1.3899, + "step": 1518 + }, + { + "epoch": 1.9179292929292928, + "grad_norm": 3.130812406539917, + "learning_rate": 2.039289055191768e-06, + "loss": 1.4834, + "step": 1519 + }, + { + "epoch": 1.9191919191919191, + "grad_norm": 3.1507933139801025, + "learning_rate": 2.036950420954163e-06, + "loss": 1.402, + "step": 1520 + }, + { + "epoch": 1.9191919191919191, + "eval_loss": 1.6648352146148682, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1520 + }, + { + "epoch": 1.9204545454545454, + "grad_norm": 3.099269151687622, + "learning_rate": 2.034611786716558e-06, + "loss": 1.4189, + "step": 1521 + }, + { + "epoch": 1.9217171717171717, + "grad_norm": 3.0046563148498535, + "learning_rate": 2.0322731524789523e-06, + "loss": 1.405, + "step": 1522 + }, + { + "epoch": 1.922979797979798, + "grad_norm": 3.229635715484619, + "learning_rate": 2.029934518241347e-06, + "loss": 1.5031, + "step": 1523 + }, + { + "epoch": 1.9242424242424243, + "grad_norm": 3.0336225032806396, + "learning_rate": 2.027595884003742e-06, + "loss": 1.4685, + "step": 1524 + }, + { + "epoch": 1.9255050505050506, + "grad_norm": 2.9998443126678467, + "learning_rate": 2.025257249766137e-06, + "loss": 1.4558, + "step": 1525 + }, + { + "epoch": 1.926767676767677, + "grad_norm": 3.1587722301483154, + "learning_rate": 2.0229186155285316e-06, + "loss": 1.4916, + "step": 1526 + }, + { + "epoch": 1.928030303030303, + "grad_norm": 3.2228925228118896, + "learning_rate": 2.0205799812909265e-06, + "loss": 1.4373, + "step": 1527 + }, + { + "epoch": 1.9292929292929293, + "grad_norm": 3.06015944480896, + "learning_rate": 2.018241347053321e-06, + "loss": 1.4147, + "step": 1528 + }, + { + "epoch": 1.9305555555555556, + "grad_norm": 3.02284574508667, + "learning_rate": 2.0159027128157157e-06, + "loss": 1.4574, + "step": 1529 + }, + { + "epoch": 1.9318181818181817, + "grad_norm": 2.9933414459228516, + "learning_rate": 2.0135640785781106e-06, + "loss": 1.4283, + "step": 1530 + }, + { + "epoch": 1.9318181818181817, + "eval_loss": 1.6711069345474243, + "eval_runtime": 37.1182, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1530 + }, + { + "epoch": 1.933080808080808, + "grad_norm": 3.325230121612549, + "learning_rate": 2.011225444340505e-06, + "loss": 1.4532, + "step": 1531 + }, + { + "epoch": 1.9343434343434343, + "grad_norm": 3.0051827430725098, + "learning_rate": 2.0088868101029003e-06, + "loss": 1.3965, + "step": 1532 + }, + { + "epoch": 1.9356060606060606, + "grad_norm": 3.1392757892608643, + "learning_rate": 2.006548175865295e-06, + "loss": 1.4678, + "step": 1533 + }, + { + "epoch": 1.9368686868686869, + "grad_norm": 3.0719001293182373, + "learning_rate": 2.0042095416276895e-06, + "loss": 1.4374, + "step": 1534 + }, + { + "epoch": 1.9381313131313131, + "grad_norm": 3.0406386852264404, + "learning_rate": 2.0018709073900844e-06, + "loss": 1.4199, + "step": 1535 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9481236934661865, + "learning_rate": 1.9995322731524792e-06, + "loss": 1.4599, + "step": 1536 + }, + { + "epoch": 1.9406565656565657, + "grad_norm": 2.901155948638916, + "learning_rate": 1.9971936389148736e-06, + "loss": 1.381, + "step": 1537 + }, + { + "epoch": 1.941919191919192, + "grad_norm": 2.9623661041259766, + "learning_rate": 1.9948550046772685e-06, + "loss": 1.4769, + "step": 1538 + }, + { + "epoch": 1.9431818181818183, + "grad_norm": 2.8457608222961426, + "learning_rate": 1.9925163704396633e-06, + "loss": 1.4117, + "step": 1539 + }, + { + "epoch": 1.9444444444444444, + "grad_norm": 2.9782702922821045, + "learning_rate": 1.990177736202058e-06, + "loss": 1.4824, + "step": 1540 + }, + { + "epoch": 1.9444444444444444, + "eval_loss": 1.6663525104522705, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1540 + }, + { + "epoch": 1.9457070707070707, + "grad_norm": 2.8265817165374756, + "learning_rate": 1.987839101964453e-06, + "loss": 1.4375, + "step": 1541 + }, + { + "epoch": 1.946969696969697, + "grad_norm": 2.9103686809539795, + "learning_rate": 1.985500467726848e-06, + "loss": 1.474, + "step": 1542 + }, + { + "epoch": 1.948232323232323, + "grad_norm": 2.837301254272461, + "learning_rate": 1.9831618334892423e-06, + "loss": 1.4171, + "step": 1543 + }, + { + "epoch": 1.9494949494949494, + "grad_norm": 3.0468413829803467, + "learning_rate": 1.980823199251637e-06, + "loss": 1.5095, + "step": 1544 + }, + { + "epoch": 1.9507575757575757, + "grad_norm": 2.968480348587036, + "learning_rate": 1.978484565014032e-06, + "loss": 1.4203, + "step": 1545 + }, + { + "epoch": 1.952020202020202, + "grad_norm": 2.9865822792053223, + "learning_rate": 1.976145930776427e-06, + "loss": 1.3916, + "step": 1546 + }, + { + "epoch": 1.9532828282828283, + "grad_norm": 2.9733808040618896, + "learning_rate": 1.9738072965388212e-06, + "loss": 1.4216, + "step": 1547 + }, + { + "epoch": 1.9545454545454546, + "grad_norm": 2.941718101501465, + "learning_rate": 1.9714686623012165e-06, + "loss": 1.426, + "step": 1548 + }, + { + "epoch": 1.9558080808080809, + "grad_norm": 3.0980522632598877, + "learning_rate": 1.969130028063611e-06, + "loss": 1.471, + "step": 1549 + }, + { + "epoch": 1.9570707070707072, + "grad_norm": 2.797102689743042, + "learning_rate": 1.9667913938260057e-06, + "loss": 1.3897, + "step": 1550 + }, + { + "epoch": 1.9570707070707072, + "eval_loss": 1.661893367767334, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 1550 + }, + { + "epoch": 1.9583333333333335, + "grad_norm": 2.9562623500823975, + "learning_rate": 1.9644527595884006e-06, + "loss": 1.4481, + "step": 1551 + }, + { + "epoch": 1.9595959595959596, + "grad_norm": 2.9586915969848633, + "learning_rate": 1.9621141253507954e-06, + "loss": 1.4068, + "step": 1552 + }, + { + "epoch": 1.9608585858585859, + "grad_norm": 3.061110496520996, + "learning_rate": 1.95977549111319e-06, + "loss": 1.4524, + "step": 1553 + }, + { + "epoch": 1.9621212121212122, + "grad_norm": 3.080491542816162, + "learning_rate": 1.9574368568755847e-06, + "loss": 1.4457, + "step": 1554 + }, + { + "epoch": 1.9633838383838382, + "grad_norm": 3.083254814147949, + "learning_rate": 1.9550982226379795e-06, + "loss": 1.4775, + "step": 1555 + }, + { + "epoch": 1.9646464646464645, + "grad_norm": 3.008596420288086, + "learning_rate": 1.9527595884003744e-06, + "loss": 1.425, + "step": 1556 + }, + { + "epoch": 1.9659090909090908, + "grad_norm": 2.999195098876953, + "learning_rate": 1.9504209541627692e-06, + "loss": 1.4114, + "step": 1557 + }, + { + "epoch": 1.9671717171717171, + "grad_norm": 3.055265188217163, + "learning_rate": 1.948082319925164e-06, + "loss": 1.4767, + "step": 1558 + }, + { + "epoch": 1.9684343434343434, + "grad_norm": 3.0922064781188965, + "learning_rate": 1.9457436856875585e-06, + "loss": 1.4548, + "step": 1559 + }, + { + "epoch": 1.9696969696969697, + "grad_norm": 2.8740789890289307, + "learning_rate": 1.9434050514499533e-06, + "loss": 1.4582, + "step": 1560 + }, + { + "epoch": 1.9696969696969697, + "eval_loss": 1.6608474254608154, + "eval_runtime": 37.1201, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1560 + }, + { + "epoch": 1.970959595959596, + "grad_norm": 2.983570098876953, + "learning_rate": 1.941066417212348e-06, + "loss": 1.4602, + "step": 1561 + }, + { + "epoch": 1.9722222222222223, + "grad_norm": 3.132474184036255, + "learning_rate": 1.9387277829747426e-06, + "loss": 1.4689, + "step": 1562 + }, + { + "epoch": 1.9734848484848486, + "grad_norm": 2.9685440063476562, + "learning_rate": 1.936389148737138e-06, + "loss": 1.39, + "step": 1563 + }, + { + "epoch": 1.9747474747474747, + "grad_norm": 3.123037815093994, + "learning_rate": 1.9340505144995327e-06, + "loss": 1.4151, + "step": 1564 + }, + { + "epoch": 1.976010101010101, + "grad_norm": 3.135232925415039, + "learning_rate": 1.931711880261927e-06, + "loss": 1.4166, + "step": 1565 + }, + { + "epoch": 1.9772727272727273, + "grad_norm": 2.9131808280944824, + "learning_rate": 1.929373246024322e-06, + "loss": 1.3952, + "step": 1566 + }, + { + "epoch": 1.9785353535353534, + "grad_norm": 3.1187453269958496, + "learning_rate": 1.927034611786717e-06, + "loss": 1.4132, + "step": 1567 + }, + { + "epoch": 1.9797979797979797, + "grad_norm": 2.983447551727295, + "learning_rate": 1.9246959775491112e-06, + "loss": 1.4517, + "step": 1568 + }, + { + "epoch": 1.981060606060606, + "grad_norm": 2.9731497764587402, + "learning_rate": 1.922357343311506e-06, + "loss": 1.432, + "step": 1569 + }, + { + "epoch": 1.9823232323232323, + "grad_norm": 2.8526554107666016, + "learning_rate": 1.920018709073901e-06, + "loss": 1.4141, + "step": 1570 + }, + { + "epoch": 1.9823232323232323, + "eval_loss": 1.6579400300979614, + "eval_runtime": 37.1216, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1570 + }, + { + "epoch": 1.9835858585858586, + "grad_norm": 2.900019407272339, + "learning_rate": 1.9176800748362958e-06, + "loss": 1.4008, + "step": 1571 + }, + { + "epoch": 1.9848484848484849, + "grad_norm": 2.96079683303833, + "learning_rate": 1.9153414405986906e-06, + "loss": 1.4586, + "step": 1572 + }, + { + "epoch": 1.9861111111111112, + "grad_norm": 3.1275715827941895, + "learning_rate": 1.9130028063610854e-06, + "loss": 1.4768, + "step": 1573 + }, + { + "epoch": 1.9873737373737375, + "grad_norm": 3.044041395187378, + "learning_rate": 1.91066417212348e-06, + "loss": 1.4428, + "step": 1574 + }, + { + "epoch": 1.9886363636363638, + "grad_norm": 3.0206832885742188, + "learning_rate": 1.9083255378858747e-06, + "loss": 1.4492, + "step": 1575 + }, + { + "epoch": 1.98989898989899, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.398, + "step": 1576 + }, + { + "epoch": 1.9911616161616161, + "grad_norm": 2.935185194015503, + "learning_rate": 1.9059869036482695e-06, + "loss": 1.4373, + "step": 1577 + }, + { + "epoch": 1.9924242424242424, + "grad_norm": 3.129822254180908, + "learning_rate": 1.9036482694106642e-06, + "loss": 1.4599, + "step": 1578 + }, + { + "epoch": 1.9936868686868687, + "grad_norm": 2.978015661239624, + "learning_rate": 1.9013096351730592e-06, + "loss": 1.409, + "step": 1579 + }, + { + "epoch": 1.9949494949494948, + "grad_norm": 2.980295181274414, + "learning_rate": 1.8989710009354539e-06, + "loss": 1.4251, + "step": 1580 + }, + { + "epoch": 1.9949494949494948, + "eval_loss": 1.661036491394043, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.996212121212121, + "grad_norm": 2.878361225128174, + "learning_rate": 1.8966323666978487e-06, + "loss": 1.4289, + "step": 1581 + }, + { + "epoch": 1.9974747474747474, + "grad_norm": 2.967414379119873, + "learning_rate": 1.8942937324602433e-06, + "loss": 1.5151, + "step": 1582 + }, + { + "epoch": 1.9987373737373737, + "grad_norm": 2.9448516368865967, + "learning_rate": 1.8919550982226382e-06, + "loss": 1.4327, + "step": 1583 + }, + { + "epoch": 2.0, + "grad_norm": 2.9000067710876465, + "learning_rate": 1.8896164639850328e-06, + "loss": 1.4113, + "step": 1584 + }, + { + "epoch": 2.0012626262626263, + "grad_norm": 3.1630377769470215, + "learning_rate": 1.8872778297474277e-06, + "loss": 1.4179, + "step": 1585 + }, + { + "epoch": 2.0025252525252526, + "grad_norm": 3.2642927169799805, + "learning_rate": 1.8849391955098223e-06, + "loss": 1.3873, + "step": 1586 + }, + { + "epoch": 2.003787878787879, + "grad_norm": 3.1669039726257324, + "learning_rate": 1.8826005612722173e-06, + "loss": 1.3834, + "step": 1587 + }, + { + "epoch": 2.005050505050505, + "grad_norm": 3.145040988922119, + "learning_rate": 1.880261927034612e-06, + "loss": 1.3334, + "step": 1588 + }, + { + "epoch": 2.0063131313131315, + "grad_norm": 3.707200050354004, + "learning_rate": 1.8779232927970068e-06, + "loss": 1.3979, + "step": 1589 + }, + { + "epoch": 2.007575757575758, + "grad_norm": 3.936523199081421, + "learning_rate": 1.8755846585594014e-06, + "loss": 1.4221, + "step": 1590 + }, + { + "epoch": 2.007575757575758, + "eval_loss": 1.6722478866577148, + "eval_runtime": 37.2129, + "eval_samples_per_second": 26.872, + "eval_steps_per_second": 1.129, + "step": 1590 + }, + { + "epoch": 2.0088383838383836, + "grad_norm": 3.4221086502075195, + "learning_rate": 1.8732460243217963e-06, + "loss": 1.3894, + "step": 1591 + }, + { + "epoch": 2.01010101010101, + "grad_norm": 3.3115553855895996, + "learning_rate": 1.870907390084191e-06, + "loss": 1.4016, + "step": 1592 + }, + { + "epoch": 2.0113636363636362, + "grad_norm": 3.055779457092285, + "learning_rate": 1.8685687558465855e-06, + "loss": 1.3676, + "step": 1593 + }, + { + "epoch": 2.0126262626262625, + "grad_norm": 3.4615232944488525, + "learning_rate": 1.8662301216089804e-06, + "loss": 1.3624, + "step": 1594 + }, + { + "epoch": 2.013888888888889, + "grad_norm": 3.3298869132995605, + "learning_rate": 1.8638914873713754e-06, + "loss": 1.3939, + "step": 1595 + }, + { + "epoch": 2.015151515151515, + "grad_norm": 3.271921396255493, + "learning_rate": 1.86155285313377e-06, + "loss": 1.3994, + "step": 1596 + }, + { + "epoch": 2.0164141414141414, + "grad_norm": 3.272677421569824, + "learning_rate": 1.859214218896165e-06, + "loss": 1.3934, + "step": 1597 + }, + { + "epoch": 2.0176767676767677, + "grad_norm": 3.3485944271087646, + "learning_rate": 1.8568755846585596e-06, + "loss": 1.3626, + "step": 1598 + }, + { + "epoch": 2.018939393939394, + "grad_norm": 3.426981210708618, + "learning_rate": 1.8545369504209542e-06, + "loss": 1.4231, + "step": 1599 + }, + { + "epoch": 2.0202020202020203, + "grad_norm": 3.3859283924102783, + "learning_rate": 1.852198316183349e-06, + "loss": 1.3587, + "step": 1600 + }, + { + "epoch": 2.0202020202020203, + "eval_loss": 1.6732399463653564, + "eval_runtime": 37.3484, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 1600 + }, + { + "epoch": 2.0214646464646466, + "grad_norm": 3.537806987762451, + "learning_rate": 1.8498596819457437e-06, + "loss": 1.3729, + "step": 1601 + }, + { + "epoch": 2.022727272727273, + "grad_norm": 3.5417566299438477, + "learning_rate": 1.8475210477081387e-06, + "loss": 1.4272, + "step": 1602 + }, + { + "epoch": 2.023989898989899, + "grad_norm": 3.287195920944214, + "learning_rate": 1.8451824134705336e-06, + "loss": 1.3647, + "step": 1603 + }, + { + "epoch": 2.025252525252525, + "grad_norm": 3.233365297317505, + "learning_rate": 1.8428437792329282e-06, + "loss": 1.4062, + "step": 1604 + }, + { + "epoch": 2.0265151515151514, + "grad_norm": 3.296578884124756, + "learning_rate": 1.8405051449953228e-06, + "loss": 1.3841, + "step": 1605 + }, + { + "epoch": 2.0277777777777777, + "grad_norm": 3.2697315216064453, + "learning_rate": 1.8381665107577177e-06, + "loss": 1.3957, + "step": 1606 + }, + { + "epoch": 2.029040404040404, + "grad_norm": 3.1117050647735596, + "learning_rate": 1.8358278765201123e-06, + "loss": 1.3643, + "step": 1607 + }, + { + "epoch": 2.0303030303030303, + "grad_norm": 3.14170241355896, + "learning_rate": 1.8334892422825071e-06, + "loss": 1.3483, + "step": 1608 + }, + { + "epoch": 2.0315656565656566, + "grad_norm": 3.2749483585357666, + "learning_rate": 1.8311506080449018e-06, + "loss": 1.4007, + "step": 1609 + }, + { + "epoch": 2.032828282828283, + "grad_norm": 3.266631603240967, + "learning_rate": 1.8288119738072968e-06, + "loss": 1.3943, + "step": 1610 + }, + { + "epoch": 2.032828282828283, + "eval_loss": 1.6679413318634033, + "eval_runtime": 36.8768, + "eval_samples_per_second": 27.117, + "eval_steps_per_second": 1.139, + "step": 1610 + }, + { + "epoch": 2.034090909090909, + "grad_norm": 3.535625696182251, + "learning_rate": 1.8264733395696914e-06, + "loss": 1.4027, + "step": 1611 + }, + { + "epoch": 2.0353535353535355, + "grad_norm": 3.2741401195526123, + "learning_rate": 1.8241347053320863e-06, + "loss": 1.4075, + "step": 1612 + }, + { + "epoch": 2.0366161616161618, + "grad_norm": 3.3183717727661133, + "learning_rate": 1.821796071094481e-06, + "loss": 1.4223, + "step": 1613 + }, + { + "epoch": 2.037878787878788, + "grad_norm": 3.323057174682617, + "learning_rate": 1.8194574368568758e-06, + "loss": 1.3595, + "step": 1614 + }, + { + "epoch": 2.0391414141414144, + "grad_norm": 3.202054738998413, + "learning_rate": 1.8171188026192704e-06, + "loss": 1.3399, + "step": 1615 + }, + { + "epoch": 2.04040404040404, + "grad_norm": 3.3273448944091797, + "learning_rate": 1.8147801683816652e-06, + "loss": 1.3848, + "step": 1616 + }, + { + "epoch": 2.0416666666666665, + "grad_norm": 3.396843671798706, + "learning_rate": 1.81244153414406e-06, + "loss": 1.3968, + "step": 1617 + }, + { + "epoch": 2.042929292929293, + "grad_norm": 3.2145836353302, + "learning_rate": 1.810102899906455e-06, + "loss": 1.3571, + "step": 1618 + }, + { + "epoch": 2.044191919191919, + "grad_norm": 3.156754732131958, + "learning_rate": 1.8077642656688496e-06, + "loss": 1.3439, + "step": 1619 + }, + { + "epoch": 2.0454545454545454, + "grad_norm": 3.4259471893310547, + "learning_rate": 1.8054256314312444e-06, + "loss": 1.4123, + "step": 1620 + }, + { + "epoch": 2.0454545454545454, + "eval_loss": 1.6703134775161743, + "eval_runtime": 37.002, + "eval_samples_per_second": 27.026, + "eval_steps_per_second": 1.135, + "step": 1620 + }, + { + "epoch": 2.0467171717171717, + "grad_norm": 3.3556408882141113, + "learning_rate": 1.803086997193639e-06, + "loss": 1.4252, + "step": 1621 + }, + { + "epoch": 2.047979797979798, + "grad_norm": 3.6319949626922607, + "learning_rate": 1.8007483629560339e-06, + "loss": 1.4634, + "step": 1622 + }, + { + "epoch": 2.0492424242424243, + "grad_norm": 3.4495089054107666, + "learning_rate": 1.7984097287184285e-06, + "loss": 1.4066, + "step": 1623 + }, + { + "epoch": 2.0505050505050506, + "grad_norm": 3.182297945022583, + "learning_rate": 1.7960710944808231e-06, + "loss": 1.3899, + "step": 1624 + }, + { + "epoch": 2.051767676767677, + "grad_norm": 3.097158193588257, + "learning_rate": 1.7937324602432182e-06, + "loss": 1.293, + "step": 1625 + }, + { + "epoch": 2.053030303030303, + "grad_norm": 3.429065704345703, + "learning_rate": 1.791393826005613e-06, + "loss": 1.3905, + "step": 1626 + }, + { + "epoch": 2.0542929292929295, + "grad_norm": 3.183366298675537, + "learning_rate": 1.7890551917680077e-06, + "loss": 1.3444, + "step": 1627 + }, + { + "epoch": 2.0555555555555554, + "grad_norm": 3.175612449645996, + "learning_rate": 1.7867165575304025e-06, + "loss": 1.3985, + "step": 1628 + }, + { + "epoch": 2.0568181818181817, + "grad_norm": 3.26491641998291, + "learning_rate": 1.7843779232927971e-06, + "loss": 1.3607, + "step": 1629 + }, + { + "epoch": 2.058080808080808, + "grad_norm": 3.2279412746429443, + "learning_rate": 1.7820392890551918e-06, + "loss": 1.3545, + "step": 1630 + }, + { + "epoch": 2.058080808080808, + "eval_loss": 1.6729198694229126, + "eval_runtime": 37.0369, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 1630 + }, + { + "epoch": 2.0593434343434343, + "grad_norm": 3.3255503177642822, + "learning_rate": 1.7797006548175866e-06, + "loss": 1.3938, + "step": 1631 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.2912404537200928, + "learning_rate": 1.7773620205799812e-06, + "loss": 1.366, + "step": 1632 + }, + { + "epoch": 2.061868686868687, + "grad_norm": 3.2314395904541016, + "learning_rate": 1.7750233863423763e-06, + "loss": 1.3298, + "step": 1633 + }, + { + "epoch": 2.063131313131313, + "grad_norm": 3.3304178714752197, + "learning_rate": 1.772684752104771e-06, + "loss": 1.3592, + "step": 1634 + }, + { + "epoch": 2.0643939393939394, + "grad_norm": 3.3113889694213867, + "learning_rate": 1.7703461178671658e-06, + "loss": 1.3865, + "step": 1635 + }, + { + "epoch": 2.0656565656565657, + "grad_norm": 3.260434150695801, + "learning_rate": 1.7680074836295604e-06, + "loss": 1.3306, + "step": 1636 + }, + { + "epoch": 2.066919191919192, + "grad_norm": 3.2890963554382324, + "learning_rate": 1.7656688493919552e-06, + "loss": 1.3533, + "step": 1637 + }, + { + "epoch": 2.0681818181818183, + "grad_norm": 3.3027193546295166, + "learning_rate": 1.7633302151543499e-06, + "loss": 1.3746, + "step": 1638 + }, + { + "epoch": 2.0694444444444446, + "grad_norm": 3.3163208961486816, + "learning_rate": 1.7609915809167447e-06, + "loss": 1.4113, + "step": 1639 + }, + { + "epoch": 2.0707070707070705, + "grad_norm": 3.300557851791382, + "learning_rate": 1.7586529466791396e-06, + "loss": 1.3669, + "step": 1640 + }, + { + "epoch": 2.0707070707070705, + "eval_loss": 1.6676570177078247, + "eval_runtime": 37.0589, + "eval_samples_per_second": 26.984, + "eval_steps_per_second": 1.133, + "step": 1640 + }, + { + "epoch": 2.071969696969697, + "grad_norm": 3.305338144302368, + "learning_rate": 1.7563143124415344e-06, + "loss": 1.3969, + "step": 1641 + }, + { + "epoch": 2.073232323232323, + "grad_norm": 3.2047572135925293, + "learning_rate": 1.753975678203929e-06, + "loss": 1.3856, + "step": 1642 + }, + { + "epoch": 2.0744949494949494, + "grad_norm": 3.575038433074951, + "learning_rate": 1.7516370439663239e-06, + "loss": 1.3947, + "step": 1643 + }, + { + "epoch": 2.0757575757575757, + "grad_norm": 3.44581937789917, + "learning_rate": 1.7492984097287185e-06, + "loss": 1.4037, + "step": 1644 + }, + { + "epoch": 2.077020202020202, + "grad_norm": 3.574326753616333, + "learning_rate": 1.7469597754911134e-06, + "loss": 1.4216, + "step": 1645 + }, + { + "epoch": 2.0782828282828283, + "grad_norm": 3.2762558460235596, + "learning_rate": 1.744621141253508e-06, + "loss": 1.3621, + "step": 1646 + }, + { + "epoch": 2.0795454545454546, + "grad_norm": 3.3749659061431885, + "learning_rate": 1.7422825070159028e-06, + "loss": 1.3762, + "step": 1647 + }, + { + "epoch": 2.080808080808081, + "grad_norm": 3.287447214126587, + "learning_rate": 1.7399438727782977e-06, + "loss": 1.3565, + "step": 1648 + }, + { + "epoch": 2.082070707070707, + "grad_norm": 3.5107264518737793, + "learning_rate": 1.7376052385406925e-06, + "loss": 1.392, + "step": 1649 + }, + { + "epoch": 2.0833333333333335, + "grad_norm": 3.319650173187256, + "learning_rate": 1.7352666043030871e-06, + "loss": 1.3835, + "step": 1650 + }, + { + "epoch": 2.0833333333333335, + "eval_loss": 1.6727185249328613, + "eval_runtime": 37.0658, + "eval_samples_per_second": 26.979, + "eval_steps_per_second": 1.133, + "step": 1650 + }, + { + "epoch": 2.08459595959596, + "grad_norm": 3.4447925090789795, + "learning_rate": 1.732927970065482e-06, + "loss": 1.4186, + "step": 1651 + }, + { + "epoch": 2.0858585858585856, + "grad_norm": 3.5944249629974365, + "learning_rate": 1.7305893358278766e-06, + "loss": 1.4461, + "step": 1652 + }, + { + "epoch": 2.087121212121212, + "grad_norm": 3.5437049865722656, + "learning_rate": 1.7282507015902715e-06, + "loss": 1.3343, + "step": 1653 + }, + { + "epoch": 2.0883838383838382, + "grad_norm": 3.361407518386841, + "learning_rate": 1.725912067352666e-06, + "loss": 1.3653, + "step": 1654 + }, + { + "epoch": 2.0896464646464645, + "grad_norm": 3.4728963375091553, + "learning_rate": 1.7235734331150607e-06, + "loss": 1.4297, + "step": 1655 + }, + { + "epoch": 2.090909090909091, + "grad_norm": 3.3794610500335693, + "learning_rate": 1.7212347988774558e-06, + "loss": 1.3941, + "step": 1656 + }, + { + "epoch": 2.092171717171717, + "grad_norm": 3.280857801437378, + "learning_rate": 1.7188961646398506e-06, + "loss": 1.3445, + "step": 1657 + }, + { + "epoch": 2.0934343434343434, + "grad_norm": 3.29036283493042, + "learning_rate": 1.7165575304022453e-06, + "loss": 1.3581, + "step": 1658 + }, + { + "epoch": 2.0946969696969697, + "grad_norm": 3.1731486320495605, + "learning_rate": 1.7142188961646399e-06, + "loss": 1.3873, + "step": 1659 + }, + { + "epoch": 2.095959595959596, + "grad_norm": 3.2283008098602295, + "learning_rate": 1.7118802619270347e-06, + "loss": 1.3462, + "step": 1660 + }, + { + "epoch": 2.095959595959596, + "eval_loss": 1.6726486682891846, + "eval_runtime": 37.0617, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1660 + }, + { + "epoch": 2.0972222222222223, + "grad_norm": 3.329171657562256, + "learning_rate": 1.7095416276894294e-06, + "loss": 1.3974, + "step": 1661 + }, + { + "epoch": 2.0984848484848486, + "grad_norm": 3.4113657474517822, + "learning_rate": 1.7072029934518242e-06, + "loss": 1.4237, + "step": 1662 + }, + { + "epoch": 2.099747474747475, + "grad_norm": 3.431184768676758, + "learning_rate": 1.7048643592142193e-06, + "loss": 1.38, + "step": 1663 + }, + { + "epoch": 2.101010101010101, + "grad_norm": 3.5471761226654053, + "learning_rate": 1.7025257249766139e-06, + "loss": 1.3762, + "step": 1664 + }, + { + "epoch": 2.102272727272727, + "grad_norm": 3.43166446685791, + "learning_rate": 1.7001870907390085e-06, + "loss": 1.4371, + "step": 1665 + }, + { + "epoch": 2.1035353535353534, + "grad_norm": 3.28086256980896, + "learning_rate": 1.6978484565014034e-06, + "loss": 1.4062, + "step": 1666 + }, + { + "epoch": 2.1047979797979797, + "grad_norm": 3.166083335876465, + "learning_rate": 1.695509822263798e-06, + "loss": 1.3535, + "step": 1667 + }, + { + "epoch": 2.106060606060606, + "grad_norm": 3.4707589149475098, + "learning_rate": 1.6931711880261928e-06, + "loss": 1.4237, + "step": 1668 + }, + { + "epoch": 2.1073232323232323, + "grad_norm": 3.5033648014068604, + "learning_rate": 1.6908325537885875e-06, + "loss": 1.4331, + "step": 1669 + }, + { + "epoch": 2.1085858585858586, + "grad_norm": 3.3915514945983887, + "learning_rate": 1.6884939195509823e-06, + "loss": 1.3783, + "step": 1670 + }, + { + "epoch": 2.1085858585858586, + "eval_loss": 1.6657648086547852, + "eval_runtime": 37.0742, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 1670 + }, + { + "epoch": 2.109848484848485, + "grad_norm": 3.3420865535736084, + "learning_rate": 1.6861552853133772e-06, + "loss": 1.4016, + "step": 1671 + }, + { + "epoch": 2.111111111111111, + "grad_norm": 3.382840394973755, + "learning_rate": 1.683816651075772e-06, + "loss": 1.4209, + "step": 1672 + }, + { + "epoch": 2.1123737373737375, + "grad_norm": 3.3109424114227295, + "learning_rate": 1.6814780168381666e-06, + "loss": 1.3716, + "step": 1673 + }, + { + "epoch": 2.1136363636363638, + "grad_norm": 3.3781373500823975, + "learning_rate": 1.6791393826005615e-06, + "loss": 1.4106, + "step": 1674 + }, + { + "epoch": 2.11489898989899, + "grad_norm": 3.29980731010437, + "learning_rate": 1.676800748362956e-06, + "loss": 1.3628, + "step": 1675 + }, + { + "epoch": 2.1161616161616164, + "grad_norm": 3.38468599319458, + "learning_rate": 1.674462114125351e-06, + "loss": 1.3478, + "step": 1676 + }, + { + "epoch": 2.117424242424242, + "grad_norm": 3.4106788635253906, + "learning_rate": 1.6721234798877456e-06, + "loss": 1.391, + "step": 1677 + }, + { + "epoch": 2.1186868686868685, + "grad_norm": 3.412184476852417, + "learning_rate": 1.6697848456501404e-06, + "loss": 1.4245, + "step": 1678 + }, + { + "epoch": 2.119949494949495, + "grad_norm": 3.3659188747406006, + "learning_rate": 1.6674462114125353e-06, + "loss": 1.3358, + "step": 1679 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.241384744644165, + "learning_rate": 1.66510757717493e-06, + "loss": 1.3855, + "step": 1680 + }, + { + "epoch": 2.121212121212121, + "eval_loss": 1.6706457138061523, + "eval_runtime": 37.0651, + "eval_samples_per_second": 26.98, + "eval_steps_per_second": 1.133, + "step": 1680 + }, + { + "epoch": 2.1224747474747474, + "grad_norm": 3.3625874519348145, + "learning_rate": 1.6627689429373247e-06, + "loss": 1.3955, + "step": 1681 + }, + { + "epoch": 2.1237373737373737, + "grad_norm": 3.378143787384033, + "learning_rate": 1.6604303086997196e-06, + "loss": 1.3635, + "step": 1682 + }, + { + "epoch": 2.125, + "grad_norm": 3.4872324466705322, + "learning_rate": 1.6580916744621142e-06, + "loss": 1.4037, + "step": 1683 + }, + { + "epoch": 2.1262626262626263, + "grad_norm": 3.4733831882476807, + "learning_rate": 1.655753040224509e-06, + "loss": 1.3846, + "step": 1684 + }, + { + "epoch": 2.1275252525252526, + "grad_norm": 3.4518625736236572, + "learning_rate": 1.6534144059869037e-06, + "loss": 1.3984, + "step": 1685 + }, + { + "epoch": 2.128787878787879, + "grad_norm": 3.3702666759490967, + "learning_rate": 1.6510757717492987e-06, + "loss": 1.3712, + "step": 1686 + }, + { + "epoch": 2.130050505050505, + "grad_norm": 3.357708215713501, + "learning_rate": 1.6487371375116934e-06, + "loss": 1.4204, + "step": 1687 + }, + { + "epoch": 2.1313131313131315, + "grad_norm": 3.3902523517608643, + "learning_rate": 1.6463985032740882e-06, + "loss": 1.439, + "step": 1688 + }, + { + "epoch": 2.132575757575758, + "grad_norm": 3.246450424194336, + "learning_rate": 1.6440598690364828e-06, + "loss": 1.4061, + "step": 1689 + }, + { + "epoch": 2.1338383838383836, + "grad_norm": 3.5247316360473633, + "learning_rate": 1.6417212347988775e-06, + "loss": 1.4148, + "step": 1690 + }, + { + "epoch": 2.1338383838383836, + "eval_loss": 1.6684828996658325, + "eval_runtime": 37.3778, + "eval_samples_per_second": 26.754, + "eval_steps_per_second": 1.124, + "step": 1690 + }, + { + "epoch": 2.13510101010101, + "grad_norm": 3.3932440280914307, + "learning_rate": 1.6393826005612723e-06, + "loss": 1.3756, + "step": 1691 + }, + { + "epoch": 2.1363636363636362, + "grad_norm": 3.5242092609405518, + "learning_rate": 1.637043966323667e-06, + "loss": 1.4009, + "step": 1692 + }, + { + "epoch": 2.1376262626262625, + "grad_norm": 3.464235544204712, + "learning_rate": 1.6347053320860618e-06, + "loss": 1.4129, + "step": 1693 + }, + { + "epoch": 2.138888888888889, + "grad_norm": 3.4409964084625244, + "learning_rate": 1.6323666978484568e-06, + "loss": 1.3969, + "step": 1694 + }, + { + "epoch": 2.140151515151515, + "grad_norm": 3.329281806945801, + "learning_rate": 1.6300280636108515e-06, + "loss": 1.367, + "step": 1695 + }, + { + "epoch": 2.1414141414141414, + "grad_norm": 3.28047251701355, + "learning_rate": 1.6276894293732461e-06, + "loss": 1.3994, + "step": 1696 + }, + { + "epoch": 2.1426767676767677, + "grad_norm": 3.1937954425811768, + "learning_rate": 1.625350795135641e-06, + "loss": 1.3626, + "step": 1697 + }, + { + "epoch": 2.143939393939394, + "grad_norm": 3.5084168910980225, + "learning_rate": 1.6230121608980356e-06, + "loss": 1.4067, + "step": 1698 + }, + { + "epoch": 2.1452020202020203, + "grad_norm": 3.3637681007385254, + "learning_rate": 1.6206735266604304e-06, + "loss": 1.3619, + "step": 1699 + }, + { + "epoch": 2.1464646464646466, + "grad_norm": 3.2855823040008545, + "learning_rate": 1.618334892422825e-06, + "loss": 1.4227, + "step": 1700 + }, + { + "epoch": 2.1464646464646466, + "eval_loss": 1.6681337356567383, + "eval_runtime": 37.4375, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1700 + }, + { + "epoch": 2.147727272727273, + "grad_norm": 3.4610795974731445, + "learning_rate": 1.6159962581852201e-06, + "loss": 1.3842, + "step": 1701 + }, + { + "epoch": 2.148989898989899, + "grad_norm": 3.420252561569214, + "learning_rate": 1.6136576239476147e-06, + "loss": 1.3951, + "step": 1702 + }, + { + "epoch": 2.150252525252525, + "grad_norm": 3.4725613594055176, + "learning_rate": 1.6113189897100096e-06, + "loss": 1.3697, + "step": 1703 + }, + { + "epoch": 2.1515151515151514, + "grad_norm": 3.3183791637420654, + "learning_rate": 1.6089803554724042e-06, + "loss": 1.3813, + "step": 1704 + }, + { + "epoch": 2.1527777777777777, + "grad_norm": 3.3727011680603027, + "learning_rate": 1.606641721234799e-06, + "loss": 1.3654, + "step": 1705 + }, + { + "epoch": 2.154040404040404, + "grad_norm": 3.420855760574341, + "learning_rate": 1.6043030869971937e-06, + "loss": 1.3399, + "step": 1706 + }, + { + "epoch": 2.1553030303030303, + "grad_norm": 3.3293979167938232, + "learning_rate": 1.6019644527595885e-06, + "loss": 1.3694, + "step": 1707 + }, + { + "epoch": 2.1565656565656566, + "grad_norm": 3.414578437805176, + "learning_rate": 1.5996258185219832e-06, + "loss": 1.4002, + "step": 1708 + }, + { + "epoch": 2.157828282828283, + "grad_norm": 3.3196682929992676, + "learning_rate": 1.5972871842843782e-06, + "loss": 1.379, + "step": 1709 + }, + { + "epoch": 2.159090909090909, + "grad_norm": 3.2492880821228027, + "learning_rate": 1.5949485500467728e-06, + "loss": 1.3694, + "step": 1710 + }, + { + "epoch": 2.159090909090909, + "eval_loss": 1.6655367612838745, + "eval_runtime": 36.9249, + "eval_samples_per_second": 27.082, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 2.1603535353535355, + "grad_norm": 3.3484954833984375, + "learning_rate": 1.5926099158091677e-06, + "loss": 1.374, + "step": 1711 + }, + { + "epoch": 2.1616161616161618, + "grad_norm": 3.2919235229492188, + "learning_rate": 1.5902712815715623e-06, + "loss": 1.3524, + "step": 1712 + }, + { + "epoch": 2.162878787878788, + "grad_norm": 3.395267963409424, + "learning_rate": 1.5879326473339572e-06, + "loss": 1.3848, + "step": 1713 + }, + { + "epoch": 2.1641414141414144, + "grad_norm": 3.5136897563934326, + "learning_rate": 1.5855940130963518e-06, + "loss": 1.3465, + "step": 1714 + }, + { + "epoch": 2.16540404040404, + "grad_norm": 3.5640273094177246, + "learning_rate": 1.5832553788587464e-06, + "loss": 1.3924, + "step": 1715 + }, + { + "epoch": 2.1666666666666665, + "grad_norm": 3.761011838912964, + "learning_rate": 1.5809167446211413e-06, + "loss": 1.4401, + "step": 1716 + }, + { + "epoch": 2.167929292929293, + "grad_norm": 3.259582042694092, + "learning_rate": 1.5785781103835363e-06, + "loss": 1.3808, + "step": 1717 + }, + { + "epoch": 2.169191919191919, + "grad_norm": 3.365035057067871, + "learning_rate": 1.576239476145931e-06, + "loss": 1.3928, + "step": 1718 + }, + { + "epoch": 2.1704545454545454, + "grad_norm": 3.466689109802246, + "learning_rate": 1.5739008419083258e-06, + "loss": 1.3797, + "step": 1719 + }, + { + "epoch": 2.1717171717171717, + "grad_norm": 3.5278398990631104, + "learning_rate": 1.5715622076707204e-06, + "loss": 1.3752, + "step": 1720 + }, + { + "epoch": 2.1717171717171717, + "eval_loss": 1.6711663007736206, + "eval_runtime": 37.0568, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1720 + }, + { + "epoch": 2.172979797979798, + "grad_norm": 3.43217134475708, + "learning_rate": 1.569223573433115e-06, + "loss": 1.4222, + "step": 1721 + }, + { + "epoch": 2.1742424242424243, + "grad_norm": 3.295238733291626, + "learning_rate": 1.56688493919551e-06, + "loss": 1.3537, + "step": 1722 + }, + { + "epoch": 2.1755050505050506, + "grad_norm": 3.4373223781585693, + "learning_rate": 1.5645463049579045e-06, + "loss": 1.3813, + "step": 1723 + }, + { + "epoch": 2.176767676767677, + "grad_norm": 3.4872143268585205, + "learning_rate": 1.5622076707202996e-06, + "loss": 1.4018, + "step": 1724 + }, + { + "epoch": 2.178030303030303, + "grad_norm": 3.3276984691619873, + "learning_rate": 1.5598690364826944e-06, + "loss": 1.3644, + "step": 1725 + }, + { + "epoch": 2.179292929292929, + "grad_norm": 3.441479444503784, + "learning_rate": 1.557530402245089e-06, + "loss": 1.4105, + "step": 1726 + }, + { + "epoch": 2.1805555555555554, + "grad_norm": 3.3938825130462646, + "learning_rate": 1.5551917680074837e-06, + "loss": 1.3478, + "step": 1727 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.428677797317505, + "learning_rate": 1.5528531337698785e-06, + "loss": 1.3977, + "step": 1728 + }, + { + "epoch": 2.183080808080808, + "grad_norm": 3.3802578449249268, + "learning_rate": 1.5505144995322732e-06, + "loss": 1.3441, + "step": 1729 + }, + { + "epoch": 2.1843434343434343, + "grad_norm": 3.419437885284424, + "learning_rate": 1.548175865294668e-06, + "loss": 1.3811, + "step": 1730 + }, + { + "epoch": 2.1843434343434343, + "eval_loss": 1.669846773147583, + "eval_runtime": 37.0868, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 1730 + }, + { + "epoch": 2.1856060606060606, + "grad_norm": 3.31225848197937, + "learning_rate": 1.5458372310570626e-06, + "loss": 1.3923, + "step": 1731 + }, + { + "epoch": 2.186868686868687, + "grad_norm": 3.5382823944091797, + "learning_rate": 1.5434985968194577e-06, + "loss": 1.4213, + "step": 1732 + }, + { + "epoch": 2.188131313131313, + "grad_norm": 3.207977533340454, + "learning_rate": 1.5411599625818523e-06, + "loss": 1.3175, + "step": 1733 + }, + { + "epoch": 2.1893939393939394, + "grad_norm": 3.3290693759918213, + "learning_rate": 1.5388213283442472e-06, + "loss": 1.4157, + "step": 1734 + }, + { + "epoch": 2.1906565656565657, + "grad_norm": 3.3432180881500244, + "learning_rate": 1.5364826941066418e-06, + "loss": 1.3844, + "step": 1735 + }, + { + "epoch": 2.191919191919192, + "grad_norm": 3.6149415969848633, + "learning_rate": 1.5341440598690366e-06, + "loss": 1.4243, + "step": 1736 + }, + { + "epoch": 2.1931818181818183, + "grad_norm": 3.3316938877105713, + "learning_rate": 1.5318054256314313e-06, + "loss": 1.372, + "step": 1737 + }, + { + "epoch": 2.1944444444444446, + "grad_norm": 3.4198312759399414, + "learning_rate": 1.5294667913938261e-06, + "loss": 1.3177, + "step": 1738 + }, + { + "epoch": 2.195707070707071, + "grad_norm": 3.450404167175293, + "learning_rate": 1.5271281571562207e-06, + "loss": 1.3596, + "step": 1739 + }, + { + "epoch": 2.196969696969697, + "grad_norm": 3.261406898498535, + "learning_rate": 1.5247895229186158e-06, + "loss": 1.3639, + "step": 1740 + }, + { + "epoch": 2.196969696969697, + "eval_loss": 1.6702104806900024, + "eval_runtime": 37.1009, + "eval_samples_per_second": 26.953, + "eval_steps_per_second": 1.132, + "step": 1740 + }, + { + "epoch": 2.198232323232323, + "grad_norm": 3.5527029037475586, + "learning_rate": 1.5224508886810104e-06, + "loss": 1.4129, + "step": 1741 + }, + { + "epoch": 2.1994949494949494, + "grad_norm": 3.498812437057495, + "learning_rate": 1.5201122544434053e-06, + "loss": 1.396, + "step": 1742 + }, + { + "epoch": 2.2007575757575757, + "grad_norm": 3.509518623352051, + "learning_rate": 1.5177736202058e-06, + "loss": 1.4506, + "step": 1743 + }, + { + "epoch": 2.202020202020202, + "grad_norm": 3.2482805252075195, + "learning_rate": 1.5154349859681948e-06, + "loss": 1.3793, + "step": 1744 + }, + { + "epoch": 2.2032828282828283, + "grad_norm": 3.299140214920044, + "learning_rate": 1.5130963517305894e-06, + "loss": 1.4011, + "step": 1745 + }, + { + "epoch": 2.2045454545454546, + "grad_norm": 3.4633231163024902, + "learning_rate": 1.510757717492984e-06, + "loss": 1.3904, + "step": 1746 + }, + { + "epoch": 2.205808080808081, + "grad_norm": 3.2724618911743164, + "learning_rate": 1.508419083255379e-06, + "loss": 1.3699, + "step": 1747 + }, + { + "epoch": 2.207070707070707, + "grad_norm": 3.361274242401123, + "learning_rate": 1.506080449017774e-06, + "loss": 1.3759, + "step": 1748 + }, + { + "epoch": 2.2083333333333335, + "grad_norm": 3.2770802974700928, + "learning_rate": 1.5037418147801685e-06, + "loss": 1.3497, + "step": 1749 + }, + { + "epoch": 2.20959595959596, + "grad_norm": 3.3818726539611816, + "learning_rate": 1.5014031805425634e-06, + "loss": 1.3277, + "step": 1750 + }, + { + "epoch": 2.20959595959596, + "eval_loss": 1.660167932510376, + "eval_runtime": 37.1049, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1750 + }, + { + "epoch": 2.2108585858585856, + "grad_norm": 3.296377420425415, + "learning_rate": 1.499064546304958e-06, + "loss": 1.3584, + "step": 1751 + }, + { + "epoch": 2.212121212121212, + "grad_norm": 3.4132192134857178, + "learning_rate": 1.4967259120673526e-06, + "loss": 1.367, + "step": 1752 + }, + { + "epoch": 2.2133838383838382, + "grad_norm": 3.3144986629486084, + "learning_rate": 1.4943872778297475e-06, + "loss": 1.3406, + "step": 1753 + }, + { + "epoch": 2.2146464646464645, + "grad_norm": 3.4283008575439453, + "learning_rate": 1.4920486435921421e-06, + "loss": 1.355, + "step": 1754 + }, + { + "epoch": 2.215909090909091, + "grad_norm": 3.319946050643921, + "learning_rate": 1.4897100093545372e-06, + "loss": 1.3665, + "step": 1755 + }, + { + "epoch": 2.217171717171717, + "grad_norm": 3.5585882663726807, + "learning_rate": 1.487371375116932e-06, + "loss": 1.4407, + "step": 1756 + }, + { + "epoch": 2.2184343434343434, + "grad_norm": 3.4216599464416504, + "learning_rate": 1.4850327408793267e-06, + "loss": 1.4105, + "step": 1757 + }, + { + "epoch": 2.2196969696969697, + "grad_norm": 3.3320260047912598, + "learning_rate": 1.4826941066417213e-06, + "loss": 1.3822, + "step": 1758 + }, + { + "epoch": 2.220959595959596, + "grad_norm": 3.2759056091308594, + "learning_rate": 1.4803554724041161e-06, + "loss": 1.4091, + "step": 1759 + }, + { + "epoch": 2.2222222222222223, + "grad_norm": 3.2865848541259766, + "learning_rate": 1.4780168381665108e-06, + "loss": 1.3643, + "step": 1760 + }, + { + "epoch": 2.2222222222222223, + "eval_loss": 1.662533164024353, + "eval_runtime": 37.1194, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1760 + }, + { + "epoch": 2.2234848484848486, + "grad_norm": 3.2485909461975098, + "learning_rate": 1.4756782039289056e-06, + "loss": 1.3986, + "step": 1761 + }, + { + "epoch": 2.224747474747475, + "grad_norm": 3.367898941040039, + "learning_rate": 1.4733395696913002e-06, + "loss": 1.37, + "step": 1762 + }, + { + "epoch": 2.226010101010101, + "grad_norm": 3.3919382095336914, + "learning_rate": 1.4710009354536953e-06, + "loss": 1.3517, + "step": 1763 + }, + { + "epoch": 2.227272727272727, + "grad_norm": 3.388627052307129, + "learning_rate": 1.46866230121609e-06, + "loss": 1.394, + "step": 1764 + }, + { + "epoch": 2.2285353535353534, + "grad_norm": 3.470121383666992, + "learning_rate": 1.4663236669784848e-06, + "loss": 1.3705, + "step": 1765 + }, + { + "epoch": 2.2297979797979797, + "grad_norm": 3.3515899181365967, + "learning_rate": 1.4639850327408794e-06, + "loss": 1.3736, + "step": 1766 + }, + { + "epoch": 2.231060606060606, + "grad_norm": 3.427877426147461, + "learning_rate": 1.4616463985032742e-06, + "loss": 1.3819, + "step": 1767 + }, + { + "epoch": 2.2323232323232323, + "grad_norm": 3.4387621879577637, + "learning_rate": 1.4593077642656689e-06, + "loss": 1.4043, + "step": 1768 + }, + { + "epoch": 2.2335858585858586, + "grad_norm": 3.4503068923950195, + "learning_rate": 1.4569691300280637e-06, + "loss": 1.3784, + "step": 1769 + }, + { + "epoch": 2.234848484848485, + "grad_norm": 3.5118091106414795, + "learning_rate": 1.4546304957904585e-06, + "loss": 1.319, + "step": 1770 + }, + { + "epoch": 2.234848484848485, + "eval_loss": 1.6638855934143066, + "eval_runtime": 37.1208, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 2.236111111111111, + "grad_norm": 3.3450958728790283, + "learning_rate": 1.4522918615528534e-06, + "loss": 1.3969, + "step": 1771 + }, + { + "epoch": 2.2373737373737375, + "grad_norm": 3.249265193939209, + "learning_rate": 1.449953227315248e-06, + "loss": 1.3543, + "step": 1772 + }, + { + "epoch": 2.2386363636363638, + "grad_norm": 3.33396315574646, + "learning_rate": 1.4476145930776429e-06, + "loss": 1.3595, + "step": 1773 + }, + { + "epoch": 2.23989898989899, + "grad_norm": 3.2275466918945312, + "learning_rate": 1.4452759588400375e-06, + "loss": 1.387, + "step": 1774 + }, + { + "epoch": 2.2411616161616164, + "grad_norm": 3.1885244846343994, + "learning_rate": 1.4429373246024323e-06, + "loss": 1.3815, + "step": 1775 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.41682767868042, + "learning_rate": 1.440598690364827e-06, + "loss": 1.398, + "step": 1776 + }, + { + "epoch": 2.2436868686868685, + "grad_norm": 3.3739681243896484, + "learning_rate": 1.4382600561272216e-06, + "loss": 1.4076, + "step": 1777 + }, + { + "epoch": 2.244949494949495, + "grad_norm": 3.3971354961395264, + "learning_rate": 1.4359214218896167e-06, + "loss": 1.3665, + "step": 1778 + }, + { + "epoch": 2.246212121212121, + "grad_norm": 3.375438690185547, + "learning_rate": 1.4335827876520115e-06, + "loss": 1.3841, + "step": 1779 + }, + { + "epoch": 2.2474747474747474, + "grad_norm": 3.4381961822509766, + "learning_rate": 1.4312441534144061e-06, + "loss": 1.3725, + "step": 1780 + }, + { + "epoch": 2.2474747474747474, + "eval_loss": 1.664459228515625, + "eval_runtime": 37.1364, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 1780 + }, + { + "epoch": 2.2487373737373737, + "grad_norm": 3.3988287448883057, + "learning_rate": 1.428905519176801e-06, + "loss": 1.3677, + "step": 1781 + }, + { + "epoch": 2.25, + "grad_norm": 3.3283984661102295, + "learning_rate": 1.4265668849391956e-06, + "loss": 1.4028, + "step": 1782 + }, + { + "epoch": 2.2512626262626263, + "grad_norm": 3.377519369125366, + "learning_rate": 1.4242282507015902e-06, + "loss": 1.4629, + "step": 1783 + }, + { + "epoch": 2.2525252525252526, + "grad_norm": 3.3736071586608887, + "learning_rate": 1.421889616463985e-06, + "loss": 1.3599, + "step": 1784 + }, + { + "epoch": 2.253787878787879, + "grad_norm": 3.4322469234466553, + "learning_rate": 1.4195509822263801e-06, + "loss": 1.3742, + "step": 1785 + }, + { + "epoch": 2.255050505050505, + "grad_norm": 3.2117607593536377, + "learning_rate": 1.4172123479887748e-06, + "loss": 1.3642, + "step": 1786 + }, + { + "epoch": 2.2563131313131315, + "grad_norm": 3.3926846981048584, + "learning_rate": 1.4148737137511694e-06, + "loss": 1.3988, + "step": 1787 + }, + { + "epoch": 2.257575757575758, + "grad_norm": 3.371910810470581, + "learning_rate": 1.4125350795135642e-06, + "loss": 1.3552, + "step": 1788 + }, + { + "epoch": 2.2588383838383836, + "grad_norm": 3.447720766067505, + "learning_rate": 1.4101964452759589e-06, + "loss": 1.3906, + "step": 1789 + }, + { + "epoch": 2.26010101010101, + "grad_norm": 3.396575450897217, + "learning_rate": 1.4078578110383537e-06, + "loss": 1.3641, + "step": 1790 + }, + { + "epoch": 2.26010101010101, + "eval_loss": 1.6667912006378174, + "eval_runtime": 37.3275, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 1790 + }, + { + "epoch": 2.2613636363636362, + "grad_norm": 3.532827138900757, + "learning_rate": 1.4055191768007483e-06, + "loss": 1.4078, + "step": 1791 + }, + { + "epoch": 2.2626262626262625, + "grad_norm": 3.438328981399536, + "learning_rate": 1.4031805425631432e-06, + "loss": 1.3589, + "step": 1792 + }, + { + "epoch": 2.263888888888889, + "grad_norm": 3.3978145122528076, + "learning_rate": 1.400841908325538e-06, + "loss": 1.4299, + "step": 1793 + }, + { + "epoch": 2.265151515151515, + "grad_norm": 3.457463502883911, + "learning_rate": 1.3985032740879329e-06, + "loss": 1.3814, + "step": 1794 + }, + { + "epoch": 2.2664141414141414, + "grad_norm": 3.2421817779541016, + "learning_rate": 1.3961646398503275e-06, + "loss": 1.3732, + "step": 1795 + }, + { + "epoch": 2.2676767676767677, + "grad_norm": 3.420693874359131, + "learning_rate": 1.3938260056127223e-06, + "loss": 1.431, + "step": 1796 + }, + { + "epoch": 2.268939393939394, + "grad_norm": 3.3793931007385254, + "learning_rate": 1.391487371375117e-06, + "loss": 1.3338, + "step": 1797 + }, + { + "epoch": 2.2702020202020203, + "grad_norm": 3.4106879234313965, + "learning_rate": 1.3891487371375118e-06, + "loss": 1.366, + "step": 1798 + }, + { + "epoch": 2.2714646464646466, + "grad_norm": 3.391782760620117, + "learning_rate": 1.3868101028999065e-06, + "loss": 1.3898, + "step": 1799 + }, + { + "epoch": 2.2727272727272725, + "grad_norm": 3.329619884490967, + "learning_rate": 1.3844714686623013e-06, + "loss": 1.3494, + "step": 1800 + }, + { + "epoch": 2.2727272727272725, + "eval_loss": 1.6635981798171997, + "eval_runtime": 37.4748, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 2.273989898989899, + "grad_norm": 3.383571147918701, + "learning_rate": 1.3821328344246961e-06, + "loss": 1.3603, + "step": 1801 + }, + { + "epoch": 2.275252525252525, + "grad_norm": 3.4224114418029785, + "learning_rate": 1.379794200187091e-06, + "loss": 1.4003, + "step": 1802 + }, + { + "epoch": 2.2765151515151514, + "grad_norm": 3.4758248329162598, + "learning_rate": 1.3774555659494856e-06, + "loss": 1.357, + "step": 1803 + }, + { + "epoch": 2.2777777777777777, + "grad_norm": 3.410489320755005, + "learning_rate": 1.3751169317118805e-06, + "loss": 1.3507, + "step": 1804 + }, + { + "epoch": 2.279040404040404, + "grad_norm": 3.4158382415771484, + "learning_rate": 1.372778297474275e-06, + "loss": 1.3822, + "step": 1805 + }, + { + "epoch": 2.2803030303030303, + "grad_norm": 3.562300443649292, + "learning_rate": 1.37043966323667e-06, + "loss": 1.3898, + "step": 1806 + }, + { + "epoch": 2.2815656565656566, + "grad_norm": 3.5485477447509766, + "learning_rate": 1.3681010289990646e-06, + "loss": 1.3341, + "step": 1807 + }, + { + "epoch": 2.282828282828283, + "grad_norm": 3.624049186706543, + "learning_rate": 1.3657623947614596e-06, + "loss": 1.4034, + "step": 1808 + }, + { + "epoch": 2.284090909090909, + "grad_norm": 3.37754225730896, + "learning_rate": 1.3634237605238542e-06, + "loss": 1.4328, + "step": 1809 + }, + { + "epoch": 2.2853535353535355, + "grad_norm": 3.5212652683258057, + "learning_rate": 1.361085126286249e-06, + "loss": 1.4038, + "step": 1810 + }, + { + "epoch": 2.2853535353535355, + "eval_loss": 1.6660585403442383, + "eval_runtime": 36.944, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 2.2866161616161618, + "grad_norm": 3.5564818382263184, + "learning_rate": 1.3587464920486437e-06, + "loss": 1.4132, + "step": 1811 + }, + { + "epoch": 2.287878787878788, + "grad_norm": 3.613368272781372, + "learning_rate": 1.3564078578110384e-06, + "loss": 1.4147, + "step": 1812 + }, + { + "epoch": 2.2891414141414144, + "grad_norm": 3.3956761360168457, + "learning_rate": 1.3540692235734332e-06, + "loss": 1.3691, + "step": 1813 + }, + { + "epoch": 2.29040404040404, + "grad_norm": 3.4510204792022705, + "learning_rate": 1.3517305893358278e-06, + "loss": 1.4152, + "step": 1814 + }, + { + "epoch": 2.2916666666666665, + "grad_norm": 3.475883960723877, + "learning_rate": 1.3493919550982227e-06, + "loss": 1.3786, + "step": 1815 + }, + { + "epoch": 2.292929292929293, + "grad_norm": 3.4566831588745117, + "learning_rate": 1.3470533208606177e-06, + "loss": 1.3737, + "step": 1816 + }, + { + "epoch": 2.294191919191919, + "grad_norm": 3.2443597316741943, + "learning_rate": 1.3447146866230124e-06, + "loss": 1.3435, + "step": 1817 + }, + { + "epoch": 2.2954545454545454, + "grad_norm": 3.4844541549682617, + "learning_rate": 1.342376052385407e-06, + "loss": 1.3759, + "step": 1818 + }, + { + "epoch": 2.2967171717171717, + "grad_norm": 3.3630642890930176, + "learning_rate": 1.3400374181478018e-06, + "loss": 1.363, + "step": 1819 + }, + { + "epoch": 2.297979797979798, + "grad_norm": 3.382788896560669, + "learning_rate": 1.3376987839101965e-06, + "loss": 1.4096, + "step": 1820 + }, + { + "epoch": 2.297979797979798, + "eval_loss": 1.664247751235962, + "eval_runtime": 37.0799, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1820 + }, + { + "epoch": 2.2992424242424243, + "grad_norm": 3.287827491760254, + "learning_rate": 1.3353601496725913e-06, + "loss": 1.348, + "step": 1821 + }, + { + "epoch": 2.3005050505050506, + "grad_norm": 3.388911008834839, + "learning_rate": 1.333021515434986e-06, + "loss": 1.3901, + "step": 1822 + }, + { + "epoch": 2.301767676767677, + "grad_norm": 3.3604469299316406, + "learning_rate": 1.3306828811973808e-06, + "loss": 1.3533, + "step": 1823 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.4020297527313232, + "learning_rate": 1.3283442469597756e-06, + "loss": 1.3469, + "step": 1824 + }, + { + "epoch": 2.304292929292929, + "grad_norm": 3.449728488922119, + "learning_rate": 1.3260056127221705e-06, + "loss": 1.3622, + "step": 1825 + }, + { + "epoch": 2.3055555555555554, + "grad_norm": 3.531832218170166, + "learning_rate": 1.323666978484565e-06, + "loss": 1.354, + "step": 1826 + }, + { + "epoch": 2.3068181818181817, + "grad_norm": 3.3406412601470947, + "learning_rate": 1.32132834424696e-06, + "loss": 1.3699, + "step": 1827 + }, + { + "epoch": 2.308080808080808, + "grad_norm": 3.5115885734558105, + "learning_rate": 1.3189897100093546e-06, + "loss": 1.4186, + "step": 1828 + }, + { + "epoch": 2.3093434343434343, + "grad_norm": 3.5668814182281494, + "learning_rate": 1.3166510757717494e-06, + "loss": 1.3894, + "step": 1829 + }, + { + "epoch": 2.3106060606060606, + "grad_norm": 3.5255587100982666, + "learning_rate": 1.314312441534144e-06, + "loss": 1.3567, + "step": 1830 + }, + { + "epoch": 2.3106060606060606, + "eval_loss": 1.6662371158599854, + "eval_runtime": 37.1254, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 1830 + }, + { + "epoch": 2.311868686868687, + "grad_norm": 3.4452052116394043, + "learning_rate": 1.311973807296539e-06, + "loss": 1.331, + "step": 1831 + }, + { + "epoch": 2.313131313131313, + "grad_norm": 3.466440439224243, + "learning_rate": 1.3096351730589337e-06, + "loss": 1.4204, + "step": 1832 + }, + { + "epoch": 2.3143939393939394, + "grad_norm": 3.2339625358581543, + "learning_rate": 1.3072965388213286e-06, + "loss": 1.3663, + "step": 1833 + }, + { + "epoch": 2.3156565656565657, + "grad_norm": 3.4634203910827637, + "learning_rate": 1.3049579045837232e-06, + "loss": 1.3805, + "step": 1834 + }, + { + "epoch": 2.316919191919192, + "grad_norm": 3.4900529384613037, + "learning_rate": 1.302619270346118e-06, + "loss": 1.3851, + "step": 1835 + }, + { + "epoch": 2.3181818181818183, + "grad_norm": 3.314610242843628, + "learning_rate": 1.3002806361085127e-06, + "loss": 1.3971, + "step": 1836 + }, + { + "epoch": 2.3194444444444446, + "grad_norm": 3.3920958042144775, + "learning_rate": 1.2979420018709073e-06, + "loss": 1.404, + "step": 1837 + }, + { + "epoch": 2.320707070707071, + "grad_norm": 3.3682339191436768, + "learning_rate": 1.2956033676333021e-06, + "loss": 1.3803, + "step": 1838 + }, + { + "epoch": 2.321969696969697, + "grad_norm": 3.332442283630371, + "learning_rate": 1.2932647333956972e-06, + "loss": 1.4214, + "step": 1839 + }, + { + "epoch": 2.323232323232323, + "grad_norm": 3.3797237873077393, + "learning_rate": 1.2909260991580918e-06, + "loss": 1.3323, + "step": 1840 + }, + { + "epoch": 2.323232323232323, + "eval_loss": 1.664433479309082, + "eval_runtime": 37.1313, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 2.3244949494949494, + "grad_norm": 3.454676866531372, + "learning_rate": 1.2885874649204867e-06, + "loss": 1.4014, + "step": 1841 + }, + { + "epoch": 2.3257575757575757, + "grad_norm": 3.4918594360351562, + "learning_rate": 1.2862488306828813e-06, + "loss": 1.3926, + "step": 1842 + }, + { + "epoch": 2.327020202020202, + "grad_norm": 3.291851043701172, + "learning_rate": 1.283910196445276e-06, + "loss": 1.3957, + "step": 1843 + }, + { + "epoch": 2.3282828282828283, + "grad_norm": 3.308403491973877, + "learning_rate": 1.2815715622076708e-06, + "loss": 1.3866, + "step": 1844 + }, + { + "epoch": 2.3295454545454546, + "grad_norm": 3.4691977500915527, + "learning_rate": 1.2792329279700654e-06, + "loss": 1.3944, + "step": 1845 + }, + { + "epoch": 2.330808080808081, + "grad_norm": 3.3912553787231445, + "learning_rate": 1.2768942937324603e-06, + "loss": 1.3521, + "step": 1846 + }, + { + "epoch": 2.332070707070707, + "grad_norm": 3.399017810821533, + "learning_rate": 1.2745556594948553e-06, + "loss": 1.3798, + "step": 1847 + }, + { + "epoch": 2.3333333333333335, + "grad_norm": 3.4829556941986084, + "learning_rate": 1.27221702525725e-06, + "loss": 1.3854, + "step": 1848 + }, + { + "epoch": 2.33459595959596, + "grad_norm": 3.363145351409912, + "learning_rate": 1.2698783910196446e-06, + "loss": 1.3581, + "step": 1849 + }, + { + "epoch": 2.3358585858585856, + "grad_norm": 3.3383405208587646, + "learning_rate": 1.2675397567820394e-06, + "loss": 1.3314, + "step": 1850 + }, + { + "epoch": 2.3358585858585856, + "eval_loss": 1.6598613262176514, + "eval_runtime": 37.1286, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 1850 + }, + { + "epoch": 2.337121212121212, + "grad_norm": 3.54443359375, + "learning_rate": 1.265201122544434e-06, + "loss": 1.3831, + "step": 1851 + }, + { + "epoch": 2.3383838383838382, + "grad_norm": 3.373283624649048, + "learning_rate": 1.2628624883068289e-06, + "loss": 1.4073, + "step": 1852 + }, + { + "epoch": 2.3396464646464645, + "grad_norm": 3.5655102729797363, + "learning_rate": 1.2605238540692235e-06, + "loss": 1.4081, + "step": 1853 + }, + { + "epoch": 2.340909090909091, + "grad_norm": 3.5852856636047363, + "learning_rate": 1.2581852198316186e-06, + "loss": 1.4566, + "step": 1854 + }, + { + "epoch": 2.342171717171717, + "grad_norm": 3.370293378829956, + "learning_rate": 1.2558465855940132e-06, + "loss": 1.3959, + "step": 1855 + }, + { + "epoch": 2.3434343434343434, + "grad_norm": 3.556837558746338, + "learning_rate": 1.253507951356408e-06, + "loss": 1.3989, + "step": 1856 + }, + { + "epoch": 2.3446969696969697, + "grad_norm": 3.3351283073425293, + "learning_rate": 1.2511693171188027e-06, + "loss": 1.3425, + "step": 1857 + }, + { + "epoch": 2.345959595959596, + "grad_norm": 3.432695150375366, + "learning_rate": 1.2488306828811975e-06, + "loss": 1.3849, + "step": 1858 + }, + { + "epoch": 2.3472222222222223, + "grad_norm": 3.6309726238250732, + "learning_rate": 1.2464920486435922e-06, + "loss": 1.4181, + "step": 1859 + }, + { + "epoch": 2.3484848484848486, + "grad_norm": 3.465672016143799, + "learning_rate": 1.244153414405987e-06, + "loss": 1.3992, + "step": 1860 + }, + { + "epoch": 2.3484848484848486, + "eval_loss": 1.6598951816558838, + "eval_runtime": 37.1338, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 1860 + }, + { + "epoch": 2.349747474747475, + "grad_norm": 3.333996534347534, + "learning_rate": 1.2418147801683818e-06, + "loss": 1.3765, + "step": 1861 + }, + { + "epoch": 2.351010101010101, + "grad_norm": 3.3039495944976807, + "learning_rate": 1.2394761459307765e-06, + "loss": 1.4055, + "step": 1862 + }, + { + "epoch": 2.3522727272727275, + "grad_norm": 3.42413067817688, + "learning_rate": 1.2371375116931713e-06, + "loss": 1.4089, + "step": 1863 + }, + { + "epoch": 2.3535353535353534, + "grad_norm": 3.626856565475464, + "learning_rate": 1.2347988774555662e-06, + "loss": 1.3382, + "step": 1864 + }, + { + "epoch": 2.3547979797979797, + "grad_norm": 3.677388906478882, + "learning_rate": 1.2324602432179608e-06, + "loss": 1.4251, + "step": 1865 + }, + { + "epoch": 2.356060606060606, + "grad_norm": 3.7962775230407715, + "learning_rate": 1.2301216089803556e-06, + "loss": 1.3808, + "step": 1866 + }, + { + "epoch": 2.3573232323232323, + "grad_norm": 3.4381752014160156, + "learning_rate": 1.2277829747427503e-06, + "loss": 1.3561, + "step": 1867 + }, + { + "epoch": 2.3585858585858586, + "grad_norm": 3.4678566455841064, + "learning_rate": 1.225444340505145e-06, + "loss": 1.3542, + "step": 1868 + }, + { + "epoch": 2.359848484848485, + "grad_norm": 3.4447884559631348, + "learning_rate": 1.22310570626754e-06, + "loss": 1.3809, + "step": 1869 + }, + { + "epoch": 2.361111111111111, + "grad_norm": 3.3679399490356445, + "learning_rate": 1.2207670720299346e-06, + "loss": 1.3825, + "step": 1870 + }, + { + "epoch": 2.361111111111111, + "eval_loss": 1.6562567949295044, + "eval_runtime": 37.1451, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 1870 + }, + { + "epoch": 2.3623737373737375, + "grad_norm": 3.3724935054779053, + "learning_rate": 1.2184284377923294e-06, + "loss": 1.3394, + "step": 1871 + }, + { + "epoch": 2.3636363636363638, + "grad_norm": 3.426579475402832, + "learning_rate": 1.2160898035547243e-06, + "loss": 1.4326, + "step": 1872 + }, + { + "epoch": 2.36489898989899, + "grad_norm": 3.4336345195770264, + "learning_rate": 1.213751169317119e-06, + "loss": 1.3575, + "step": 1873 + }, + { + "epoch": 2.3661616161616164, + "grad_norm": 3.4542808532714844, + "learning_rate": 1.2114125350795135e-06, + "loss": 1.3449, + "step": 1874 + }, + { + "epoch": 2.367424242424242, + "grad_norm": 3.362525701522827, + "learning_rate": 1.2090739008419086e-06, + "loss": 1.3964, + "step": 1875 + }, + { + "epoch": 2.3686868686868685, + "grad_norm": 3.392415761947632, + "learning_rate": 1.2067352666043032e-06, + "loss": 1.4272, + "step": 1876 + }, + { + "epoch": 2.369949494949495, + "grad_norm": 3.4062328338623047, + "learning_rate": 1.2043966323666978e-06, + "loss": 1.3407, + "step": 1877 + }, + { + "epoch": 2.371212121212121, + "grad_norm": 3.5435354709625244, + "learning_rate": 1.2020579981290927e-06, + "loss": 1.3671, + "step": 1878 + }, + { + "epoch": 2.3724747474747474, + "grad_norm": 3.743882656097412, + "learning_rate": 1.1997193638914875e-06, + "loss": 1.4046, + "step": 1879 + }, + { + "epoch": 2.3737373737373737, + "grad_norm": 3.752319574356079, + "learning_rate": 1.1973807296538822e-06, + "loss": 1.3631, + "step": 1880 + }, + { + "epoch": 2.3737373737373737, + "eval_loss": 1.6618274450302124, + "eval_runtime": 37.1614, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1880 + }, + { + "epoch": 2.375, + "grad_norm": 3.57468318939209, + "learning_rate": 1.195042095416277e-06, + "loss": 1.3584, + "step": 1881 + }, + { + "epoch": 2.3762626262626263, + "grad_norm": 3.544557809829712, + "learning_rate": 1.1927034611786716e-06, + "loss": 1.3759, + "step": 1882 + }, + { + "epoch": 2.3775252525252526, + "grad_norm": 3.313106060028076, + "learning_rate": 1.1903648269410665e-06, + "loss": 1.36, + "step": 1883 + }, + { + "epoch": 2.378787878787879, + "grad_norm": 3.347764730453491, + "learning_rate": 1.1880261927034613e-06, + "loss": 1.3789, + "step": 1884 + }, + { + "epoch": 2.380050505050505, + "grad_norm": 3.260697364807129, + "learning_rate": 1.185687558465856e-06, + "loss": 1.3456, + "step": 1885 + }, + { + "epoch": 2.3813131313131315, + "grad_norm": 3.287977695465088, + "learning_rate": 1.1833489242282508e-06, + "loss": 1.3739, + "step": 1886 + }, + { + "epoch": 2.382575757575758, + "grad_norm": 3.27716064453125, + "learning_rate": 1.1810102899906456e-06, + "loss": 1.3792, + "step": 1887 + }, + { + "epoch": 2.3838383838383836, + "grad_norm": 3.4265832901000977, + "learning_rate": 1.1786716557530403e-06, + "loss": 1.3604, + "step": 1888 + }, + { + "epoch": 2.38510101010101, + "grad_norm": 3.5619351863861084, + "learning_rate": 1.1763330215154351e-06, + "loss": 1.3786, + "step": 1889 + }, + { + "epoch": 2.3863636363636362, + "grad_norm": 3.482348680496216, + "learning_rate": 1.17399438727783e-06, + "loss": 1.415, + "step": 1890 + }, + { + "epoch": 2.3863636363636362, + "eval_loss": 1.6625829935073853, + "eval_runtime": 37.2791, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 2.3876262626262625, + "grad_norm": 3.375915288925171, + "learning_rate": 1.1716557530402246e-06, + "loss": 1.335, + "step": 1891 + }, + { + "epoch": 2.388888888888889, + "grad_norm": 3.5826265811920166, + "learning_rate": 1.1693171188026194e-06, + "loss": 1.3612, + "step": 1892 + }, + { + "epoch": 2.390151515151515, + "grad_norm": 3.787109851837158, + "learning_rate": 1.166978484565014e-06, + "loss": 1.386, + "step": 1893 + }, + { + "epoch": 2.3914141414141414, + "grad_norm": 3.5518991947174072, + "learning_rate": 1.164639850327409e-06, + "loss": 1.3372, + "step": 1894 + }, + { + "epoch": 2.3926767676767677, + "grad_norm": 3.4609735012054443, + "learning_rate": 1.1623012160898037e-06, + "loss": 1.362, + "step": 1895 + }, + { + "epoch": 2.393939393939394, + "grad_norm": 3.578777551651001, + "learning_rate": 1.1599625818521984e-06, + "loss": 1.3754, + "step": 1896 + }, + { + "epoch": 2.3952020202020203, + "grad_norm": 3.5557262897491455, + "learning_rate": 1.1576239476145932e-06, + "loss": 1.4048, + "step": 1897 + }, + { + "epoch": 2.3964646464646466, + "grad_norm": 3.3413872718811035, + "learning_rate": 1.155285313376988e-06, + "loss": 1.3531, + "step": 1898 + }, + { + "epoch": 2.3977272727272725, + "grad_norm": 3.207866668701172, + "learning_rate": 1.1529466791393827e-06, + "loss": 1.3479, + "step": 1899 + }, + { + "epoch": 2.398989898989899, + "grad_norm": 3.3909525871276855, + "learning_rate": 1.1506080449017775e-06, + "loss": 1.3663, + "step": 1900 + }, + { + "epoch": 2.398989898989899, + "eval_loss": 1.661224126815796, + "eval_runtime": 37.4772, + "eval_samples_per_second": 26.683, + "eval_steps_per_second": 1.121, + "step": 1900 + }, + { + "epoch": 2.400252525252525, + "grad_norm": 3.4364285469055176, + "learning_rate": 1.1482694106641722e-06, + "loss": 1.3541, + "step": 1901 + }, + { + "epoch": 2.4015151515151514, + "grad_norm": 3.5648012161254883, + "learning_rate": 1.145930776426567e-06, + "loss": 1.3976, + "step": 1902 + }, + { + "epoch": 2.4027777777777777, + "grad_norm": 3.4142239093780518, + "learning_rate": 1.1435921421889619e-06, + "loss": 1.4217, + "step": 1903 + }, + { + "epoch": 2.404040404040404, + "grad_norm": 3.415621757507324, + "learning_rate": 1.1412535079513565e-06, + "loss": 1.4152, + "step": 1904 + }, + { + "epoch": 2.4053030303030303, + "grad_norm": 3.401431083679199, + "learning_rate": 1.1389148737137511e-06, + "loss": 1.3588, + "step": 1905 + }, + { + "epoch": 2.4065656565656566, + "grad_norm": 3.528501272201538, + "learning_rate": 1.1365762394761462e-06, + "loss": 1.4113, + "step": 1906 + }, + { + "epoch": 2.407828282828283, + "grad_norm": 3.5070385932922363, + "learning_rate": 1.1342376052385408e-06, + "loss": 1.4051, + "step": 1907 + }, + { + "epoch": 2.409090909090909, + "grad_norm": 3.2571558952331543, + "learning_rate": 1.1318989710009354e-06, + "loss": 1.3471, + "step": 1908 + }, + { + "epoch": 2.4103535353535355, + "grad_norm": 3.4780328273773193, + "learning_rate": 1.1295603367633303e-06, + "loss": 1.3895, + "step": 1909 + }, + { + "epoch": 2.4116161616161618, + "grad_norm": 3.431661367416382, + "learning_rate": 1.1272217025257251e-06, + "loss": 1.366, + "step": 1910 + }, + { + "epoch": 2.4116161616161618, + "eval_loss": 1.663731575012207, + "eval_runtime": 36.9118, + "eval_samples_per_second": 27.092, + "eval_steps_per_second": 1.138, + "step": 1910 + }, + { + "epoch": 2.412878787878788, + "grad_norm": 3.4981627464294434, + "learning_rate": 1.1248830682881197e-06, + "loss": 1.3476, + "step": 1911 + }, + { + "epoch": 2.4141414141414144, + "grad_norm": 3.1849045753479004, + "learning_rate": 1.1225444340505146e-06, + "loss": 1.32, + "step": 1912 + }, + { + "epoch": 2.41540404040404, + "grad_norm": 3.4398815631866455, + "learning_rate": 1.1202057998129094e-06, + "loss": 1.4065, + "step": 1913 + }, + { + "epoch": 2.4166666666666665, + "grad_norm": 3.5676138401031494, + "learning_rate": 1.117867165575304e-06, + "loss": 1.3973, + "step": 1914 + }, + { + "epoch": 2.417929292929293, + "grad_norm": 3.5278069972991943, + "learning_rate": 1.115528531337699e-06, + "loss": 1.3802, + "step": 1915 + }, + { + "epoch": 2.419191919191919, + "grad_norm": 3.4156792163848877, + "learning_rate": 1.1131898971000935e-06, + "loss": 1.3605, + "step": 1916 + }, + { + "epoch": 2.4204545454545454, + "grad_norm": 3.3457775115966797, + "learning_rate": 1.1108512628624884e-06, + "loss": 1.3688, + "step": 1917 + }, + { + "epoch": 2.4217171717171717, + "grad_norm": 3.3902130126953125, + "learning_rate": 1.1085126286248832e-06, + "loss": 1.3829, + "step": 1918 + }, + { + "epoch": 2.422979797979798, + "grad_norm": 3.361523389816284, + "learning_rate": 1.1061739943872779e-06, + "loss": 1.3495, + "step": 1919 + }, + { + "epoch": 2.4242424242424243, + "grad_norm": 3.2795472145080566, + "learning_rate": 1.1038353601496727e-06, + "loss": 1.3165, + "step": 1920 + }, + { + "epoch": 2.4242424242424243, + "eval_loss": 1.6594749689102173, + "eval_runtime": 37.0393, + "eval_samples_per_second": 26.998, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 2.4255050505050506, + "grad_norm": 3.5426580905914307, + "learning_rate": 1.1014967259120675e-06, + "loss": 1.403, + "step": 1921 + }, + { + "epoch": 2.426767676767677, + "grad_norm": 3.329042434692383, + "learning_rate": 1.0991580916744622e-06, + "loss": 1.3731, + "step": 1922 + }, + { + "epoch": 2.428030303030303, + "grad_norm": 3.419233798980713, + "learning_rate": 1.096819457436857e-06, + "loss": 1.3784, + "step": 1923 + }, + { + "epoch": 2.429292929292929, + "grad_norm": 3.6780271530151367, + "learning_rate": 1.0944808231992516e-06, + "loss": 1.4061, + "step": 1924 + }, + { + "epoch": 2.4305555555555554, + "grad_norm": 3.790707588195801, + "learning_rate": 1.0921421889616465e-06, + "loss": 1.4125, + "step": 1925 + }, + { + "epoch": 2.4318181818181817, + "grad_norm": 3.4600441455841064, + "learning_rate": 1.0898035547240413e-06, + "loss": 1.4084, + "step": 1926 + }, + { + "epoch": 2.433080808080808, + "grad_norm": 3.502103805541992, + "learning_rate": 1.087464920486436e-06, + "loss": 1.3882, + "step": 1927 + }, + { + "epoch": 2.4343434343434343, + "grad_norm": 3.3180925846099854, + "learning_rate": 1.0851262862488308e-06, + "loss": 1.3669, + "step": 1928 + }, + { + "epoch": 2.4356060606060606, + "grad_norm": 3.447608709335327, + "learning_rate": 1.0827876520112257e-06, + "loss": 1.2922, + "step": 1929 + }, + { + "epoch": 2.436868686868687, + "grad_norm": 3.4833571910858154, + "learning_rate": 1.0804490177736203e-06, + "loss": 1.3754, + "step": 1930 + }, + { + "epoch": 2.436868686868687, + "eval_loss": 1.665055751800537, + "eval_runtime": 37.0614, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1930 + }, + { + "epoch": 2.438131313131313, + "grad_norm": 3.426940679550171, + "learning_rate": 1.0781103835360151e-06, + "loss": 1.3659, + "step": 1931 + }, + { + "epoch": 2.4393939393939394, + "grad_norm": 3.4161434173583984, + "learning_rate": 1.07577174929841e-06, + "loss": 1.4161, + "step": 1932 + }, + { + "epoch": 2.4406565656565657, + "grad_norm": 3.5886597633361816, + "learning_rate": 1.0734331150608046e-06, + "loss": 1.448, + "step": 1933 + }, + { + "epoch": 2.441919191919192, + "grad_norm": 3.4271745681762695, + "learning_rate": 1.0710944808231992e-06, + "loss": 1.3608, + "step": 1934 + }, + { + "epoch": 2.4431818181818183, + "grad_norm": 3.357823133468628, + "learning_rate": 1.068755846585594e-06, + "loss": 1.3477, + "step": 1935 + }, + { + "epoch": 2.4444444444444446, + "grad_norm": 3.604701519012451, + "learning_rate": 1.066417212347989e-06, + "loss": 1.4402, + "step": 1936 + }, + { + "epoch": 2.445707070707071, + "grad_norm": 3.3494462966918945, + "learning_rate": 1.0640785781103835e-06, + "loss": 1.3878, + "step": 1937 + }, + { + "epoch": 2.446969696969697, + "grad_norm": 3.3025968074798584, + "learning_rate": 1.0617399438727784e-06, + "loss": 1.3671, + "step": 1938 + }, + { + "epoch": 2.448232323232323, + "grad_norm": 3.430521249771118, + "learning_rate": 1.059401309635173e-06, + "loss": 1.3795, + "step": 1939 + }, + { + "epoch": 2.4494949494949494, + "grad_norm": 3.5415585041046143, + "learning_rate": 1.0570626753975679e-06, + "loss": 1.3867, + "step": 1940 + }, + { + "epoch": 2.4494949494949494, + "eval_loss": 1.6607574224472046, + "eval_runtime": 37.0458, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1940 + }, + { + "epoch": 2.4507575757575757, + "grad_norm": 3.527456521987915, + "learning_rate": 1.0547240411599627e-06, + "loss": 1.3641, + "step": 1941 + }, + { + "epoch": 2.452020202020202, + "grad_norm": 3.5055441856384277, + "learning_rate": 1.0523854069223573e-06, + "loss": 1.3905, + "step": 1942 + }, + { + "epoch": 2.4532828282828283, + "grad_norm": 3.512960910797119, + "learning_rate": 1.0500467726847522e-06, + "loss": 1.3824, + "step": 1943 + }, + { + "epoch": 2.4545454545454546, + "grad_norm": 3.4122636318206787, + "learning_rate": 1.047708138447147e-06, + "loss": 1.3857, + "step": 1944 + }, + { + "epoch": 2.455808080808081, + "grad_norm": 3.665520668029785, + "learning_rate": 1.0453695042095417e-06, + "loss": 1.3866, + "step": 1945 + }, + { + "epoch": 2.457070707070707, + "grad_norm": 3.7062342166900635, + "learning_rate": 1.0430308699719365e-06, + "loss": 1.3968, + "step": 1946 + }, + { + "epoch": 2.4583333333333335, + "grad_norm": 3.558770179748535, + "learning_rate": 1.0406922357343311e-06, + "loss": 1.4005, + "step": 1947 + }, + { + "epoch": 2.45959595959596, + "grad_norm": 3.3678979873657227, + "learning_rate": 1.038353601496726e-06, + "loss": 1.3258, + "step": 1948 + }, + { + "epoch": 2.4608585858585856, + "grad_norm": 3.4766786098480225, + "learning_rate": 1.0360149672591208e-06, + "loss": 1.3634, + "step": 1949 + }, + { + "epoch": 2.462121212121212, + "grad_norm": 3.3427937030792236, + "learning_rate": 1.0336763330215154e-06, + "loss": 1.3419, + "step": 1950 + }, + { + "epoch": 2.462121212121212, + "eval_loss": 1.660905361175537, + "eval_runtime": 37.0483, + "eval_samples_per_second": 26.992, + "eval_steps_per_second": 1.134, + "step": 1950 + }, + { + "epoch": 2.4633838383838382, + "grad_norm": 3.436858654022217, + "learning_rate": 1.0313376987839103e-06, + "loss": 1.377, + "step": 1951 + }, + { + "epoch": 2.4646464646464645, + "grad_norm": 3.4107327461242676, + "learning_rate": 1.0289990645463051e-06, + "loss": 1.3484, + "step": 1952 + }, + { + "epoch": 2.465909090909091, + "grad_norm": 3.3843321800231934, + "learning_rate": 1.0266604303086998e-06, + "loss": 1.3373, + "step": 1953 + }, + { + "epoch": 2.467171717171717, + "grad_norm": 3.368032455444336, + "learning_rate": 1.0243217960710946e-06, + "loss": 1.3245, + "step": 1954 + }, + { + "epoch": 2.4684343434343434, + "grad_norm": 3.3483028411865234, + "learning_rate": 1.0219831618334894e-06, + "loss": 1.3498, + "step": 1955 + }, + { + "epoch": 2.4696969696969697, + "grad_norm": 3.5274465084075928, + "learning_rate": 1.019644527595884e-06, + "loss": 1.4074, + "step": 1956 + }, + { + "epoch": 2.470959595959596, + "grad_norm": 3.4556262493133545, + "learning_rate": 1.017305893358279e-06, + "loss": 1.3778, + "step": 1957 + }, + { + "epoch": 2.4722222222222223, + "grad_norm": 3.4345815181732178, + "learning_rate": 1.0149672591206736e-06, + "loss": 1.3847, + "step": 1958 + }, + { + "epoch": 2.4734848484848486, + "grad_norm": 3.437321662902832, + "learning_rate": 1.0126286248830684e-06, + "loss": 1.3784, + "step": 1959 + }, + { + "epoch": 2.474747474747475, + "grad_norm": 3.4794695377349854, + "learning_rate": 1.0102899906454632e-06, + "loss": 1.3646, + "step": 1960 + }, + { + "epoch": 2.474747474747475, + "eval_loss": 1.6639539003372192, + "eval_runtime": 37.0499, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1960 + }, + { + "epoch": 2.476010101010101, + "grad_norm": 3.457455635070801, + "learning_rate": 1.0079513564078579e-06, + "loss": 1.3654, + "step": 1961 + }, + { + "epoch": 2.4772727272727275, + "grad_norm": 3.432436227798462, + "learning_rate": 1.0056127221702525e-06, + "loss": 1.3224, + "step": 1962 + }, + { + "epoch": 2.4785353535353534, + "grad_norm": 3.454395055770874, + "learning_rate": 1.0032740879326476e-06, + "loss": 1.3513, + "step": 1963 + }, + { + "epoch": 2.4797979797979797, + "grad_norm": 3.5861761569976807, + "learning_rate": 1.0009354536950422e-06, + "loss": 1.3855, + "step": 1964 + }, + { + "epoch": 2.481060606060606, + "grad_norm": 3.2933616638183594, + "learning_rate": 9.985968194574368e-07, + "loss": 1.3303, + "step": 1965 + }, + { + "epoch": 2.4823232323232323, + "grad_norm": 3.368325710296631, + "learning_rate": 9.962581852198317e-07, + "loss": 1.3584, + "step": 1966 + }, + { + "epoch": 2.4835858585858586, + "grad_norm": 3.539789915084839, + "learning_rate": 9.939195509822265e-07, + "loss": 1.4043, + "step": 1967 + }, + { + "epoch": 2.484848484848485, + "grad_norm": 3.4011685848236084, + "learning_rate": 9.915809167446211e-07, + "loss": 1.4291, + "step": 1968 + }, + { + "epoch": 2.486111111111111, + "grad_norm": 3.239996910095215, + "learning_rate": 9.89242282507016e-07, + "loss": 1.3769, + "step": 1969 + }, + { + "epoch": 2.4873737373737375, + "grad_norm": 3.311413288116455, + "learning_rate": 9.869036482694106e-07, + "loss": 1.3636, + "step": 1970 + }, + { + "epoch": 2.4873737373737375, + "eval_loss": 1.6609822511672974, + "eval_runtime": 37.0636, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1970 + }, + { + "epoch": 2.4886363636363638, + "grad_norm": 3.514861583709717, + "learning_rate": 9.845650140318055e-07, + "loss": 1.3879, + "step": 1971 + }, + { + "epoch": 2.48989898989899, + "grad_norm": 3.385402202606201, + "learning_rate": 9.822263797942003e-07, + "loss": 1.336, + "step": 1972 + }, + { + "epoch": 2.4911616161616164, + "grad_norm": 3.3489670753479004, + "learning_rate": 9.79887745556595e-07, + "loss": 1.3734, + "step": 1973 + }, + { + "epoch": 2.492424242424242, + "grad_norm": 3.3082852363586426, + "learning_rate": 9.775491113189898e-07, + "loss": 1.3461, + "step": 1974 + }, + { + "epoch": 2.4936868686868685, + "grad_norm": 3.348860740661621, + "learning_rate": 9.752104770813846e-07, + "loss": 1.3502, + "step": 1975 + }, + { + "epoch": 2.494949494949495, + "grad_norm": 3.4895060062408447, + "learning_rate": 9.728718428437792e-07, + "loss": 1.3889, + "step": 1976 + }, + { + "epoch": 2.496212121212121, + "grad_norm": 3.5898680686950684, + "learning_rate": 9.70533208606174e-07, + "loss": 1.4076, + "step": 1977 + }, + { + "epoch": 2.4974747474747474, + "grad_norm": 3.306730270385742, + "learning_rate": 9.68194574368569e-07, + "loss": 1.3627, + "step": 1978 + }, + { + "epoch": 2.4987373737373737, + "grad_norm": 3.375053882598877, + "learning_rate": 9.658559401309636e-07, + "loss": 1.3422, + "step": 1979 + }, + { + "epoch": 2.5, + "grad_norm": 3.5166397094726562, + "learning_rate": 9.635173058933584e-07, + "loss": 1.3922, + "step": 1980 + }, + { + "epoch": 2.5, + "eval_loss": 1.6624678373336792, + "eval_runtime": 37.0699, + "eval_samples_per_second": 26.976, + "eval_steps_per_second": 1.133, + "step": 1980 + }, + { + "epoch": 2.5012626262626263, + "grad_norm": 3.405553102493286, + "learning_rate": 9.61178671655753e-07, + "loss": 1.3411, + "step": 1981 + }, + { + "epoch": 2.5025252525252526, + "grad_norm": 3.56683349609375, + "learning_rate": 9.588400374181479e-07, + "loss": 1.3832, + "step": 1982 + }, + { + "epoch": 2.503787878787879, + "grad_norm": 3.387821912765503, + "learning_rate": 9.565014031805427e-07, + "loss": 1.3407, + "step": 1983 + }, + { + "epoch": 2.505050505050505, + "grad_norm": 3.5401318073272705, + "learning_rate": 9.541627689429374e-07, + "loss": 1.3903, + "step": 1984 + }, + { + "epoch": 2.5063131313131315, + "grad_norm": 3.5890989303588867, + "learning_rate": 9.518241347053321e-07, + "loss": 1.3839, + "step": 1985 + }, + { + "epoch": 2.507575757575758, + "grad_norm": 3.3846449851989746, + "learning_rate": 9.494855004677269e-07, + "loss": 1.3808, + "step": 1986 + }, + { + "epoch": 2.508838383838384, + "grad_norm": 3.7580108642578125, + "learning_rate": 9.471468662301217e-07, + "loss": 1.403, + "step": 1987 + }, + { + "epoch": 2.51010101010101, + "grad_norm": 3.317741870880127, + "learning_rate": 9.448082319925164e-07, + "loss": 1.315, + "step": 1988 + }, + { + "epoch": 2.5113636363636362, + "grad_norm": 3.453925848007202, + "learning_rate": 9.424695977549111e-07, + "loss": 1.3763, + "step": 1989 + }, + { + "epoch": 2.5126262626262625, + "grad_norm": 3.5050547122955322, + "learning_rate": 9.40130963517306e-07, + "loss": 1.3237, + "step": 1990 + }, + { + "epoch": 2.5126262626262625, + "eval_loss": 1.6650363206863403, + "eval_runtime": 37.2939, + "eval_samples_per_second": 26.814, + "eval_steps_per_second": 1.126, + "step": 1990 + }, + { + "epoch": 2.513888888888889, + "grad_norm": 3.484497308731079, + "learning_rate": 9.377923292797007e-07, + "loss": 1.371, + "step": 1991 + }, + { + "epoch": 2.515151515151515, + "grad_norm": 3.4720635414123535, + "learning_rate": 9.354536950420955e-07, + "loss": 1.3496, + "step": 1992 + }, + { + "epoch": 2.5164141414141414, + "grad_norm": 3.511245012283325, + "learning_rate": 9.331150608044902e-07, + "loss": 1.3208, + "step": 1993 + }, + { + "epoch": 2.5176767676767677, + "grad_norm": 3.4779229164123535, + "learning_rate": 9.30776426566885e-07, + "loss": 1.3743, + "step": 1994 + }, + { + "epoch": 2.518939393939394, + "grad_norm": 3.5714519023895264, + "learning_rate": 9.284377923292798e-07, + "loss": 1.3659, + "step": 1995 + }, + { + "epoch": 2.5202020202020203, + "grad_norm": 3.501225709915161, + "learning_rate": 9.260991580916745e-07, + "loss": 1.4598, + "step": 1996 + }, + { + "epoch": 2.5214646464646466, + "grad_norm": 3.281121253967285, + "learning_rate": 9.237605238540694e-07, + "loss": 1.3342, + "step": 1997 + }, + { + "epoch": 2.5227272727272725, + "grad_norm": 3.4490602016448975, + "learning_rate": 9.214218896164641e-07, + "loss": 1.3958, + "step": 1998 + }, + { + "epoch": 2.523989898989899, + "grad_norm": 3.4208924770355225, + "learning_rate": 9.190832553788588e-07, + "loss": 1.3943, + "step": 1999 + }, + { + "epoch": 2.525252525252525, + "grad_norm": 3.2830605506896973, + "learning_rate": 9.167446211412536e-07, + "loss": 1.3649, + "step": 2000 + }, + { + "epoch": 2.525252525252525, + "eval_loss": 1.6609365940093994, + "eval_runtime": 37.39, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 2000 + }, + { + "epoch": 2.5265151515151514, + "grad_norm": 3.5467946529388428, + "learning_rate": 9.144059869036484e-07, + "loss": 1.4129, + "step": 2001 + }, + { + "epoch": 2.5277777777777777, + "grad_norm": 3.435167074203491, + "learning_rate": 9.120673526660431e-07, + "loss": 1.3497, + "step": 2002 + }, + { + "epoch": 2.529040404040404, + "grad_norm": 3.4507105350494385, + "learning_rate": 9.097287184284379e-07, + "loss": 1.4018, + "step": 2003 + }, + { + "epoch": 2.5303030303030303, + "grad_norm": 3.4729814529418945, + "learning_rate": 9.073900841908326e-07, + "loss": 1.4069, + "step": 2004 + }, + { + "epoch": 2.5315656565656566, + "grad_norm": 3.3762307167053223, + "learning_rate": 9.050514499532275e-07, + "loss": 1.3379, + "step": 2005 + }, + { + "epoch": 2.532828282828283, + "grad_norm": 3.4308269023895264, + "learning_rate": 9.027128157156222e-07, + "loss": 1.385, + "step": 2006 + }, + { + "epoch": 2.534090909090909, + "grad_norm": 3.455213785171509, + "learning_rate": 9.003741814780169e-07, + "loss": 1.3543, + "step": 2007 + }, + { + "epoch": 2.5353535353535355, + "grad_norm": 3.360630989074707, + "learning_rate": 8.980355472404116e-07, + "loss": 1.3399, + "step": 2008 + }, + { + "epoch": 2.5366161616161618, + "grad_norm": 3.4755003452301025, + "learning_rate": 8.956969130028065e-07, + "loss": 1.4, + "step": 2009 + }, + { + "epoch": 2.537878787878788, + "grad_norm": 3.473783016204834, + "learning_rate": 8.933582787652013e-07, + "loss": 1.3265, + "step": 2010 + }, + { + "epoch": 2.537878787878788, + "eval_loss": 1.6632015705108643, + "eval_runtime": 36.9441, + "eval_samples_per_second": 27.068, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.5391414141414144, + "grad_norm": 3.3548569679260254, + "learning_rate": 8.910196445275959e-07, + "loss": 1.3642, + "step": 2011 + }, + { + "epoch": 2.5404040404040407, + "grad_norm": 3.4288694858551025, + "learning_rate": 8.886810102899906e-07, + "loss": 1.3571, + "step": 2012 + }, + { + "epoch": 2.5416666666666665, + "grad_norm": 3.5140700340270996, + "learning_rate": 8.863423760523855e-07, + "loss": 1.3791, + "step": 2013 + }, + { + "epoch": 2.542929292929293, + "grad_norm": 3.4733200073242188, + "learning_rate": 8.840037418147802e-07, + "loss": 1.4035, + "step": 2014 + }, + { + "epoch": 2.544191919191919, + "grad_norm": 3.2244248390197754, + "learning_rate": 8.816651075771749e-07, + "loss": 1.3167, + "step": 2015 + }, + { + "epoch": 2.5454545454545454, + "grad_norm": 3.3097362518310547, + "learning_rate": 8.793264733395698e-07, + "loss": 1.3032, + "step": 2016 + }, + { + "epoch": 2.5467171717171717, + "grad_norm": 3.2589709758758545, + "learning_rate": 8.769878391019645e-07, + "loss": 1.343, + "step": 2017 + }, + { + "epoch": 2.547979797979798, + "grad_norm": 3.3932478427886963, + "learning_rate": 8.746492048643593e-07, + "loss": 1.3539, + "step": 2018 + }, + { + "epoch": 2.5492424242424243, + "grad_norm": 3.305260181427002, + "learning_rate": 8.72310570626754e-07, + "loss": 1.3596, + "step": 2019 + }, + { + "epoch": 2.5505050505050506, + "grad_norm": 3.381182909011841, + "learning_rate": 8.699719363891488e-07, + "loss": 1.3562, + "step": 2020 + }, + { + "epoch": 2.5505050505050506, + "eval_loss": 1.6605889797210693, + "eval_runtime": 37.0728, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.551767676767677, + "grad_norm": 3.406937599182129, + "learning_rate": 8.676333021515436e-07, + "loss": 1.3388, + "step": 2021 + }, + { + "epoch": 2.5530303030303028, + "grad_norm": 3.587719678878784, + "learning_rate": 8.652946679139383e-07, + "loss": 1.4079, + "step": 2022 + }, + { + "epoch": 2.554292929292929, + "grad_norm": 3.5528013706207275, + "learning_rate": 8.62956033676333e-07, + "loss": 1.3852, + "step": 2023 + }, + { + "epoch": 2.5555555555555554, + "grad_norm": 3.4368793964385986, + "learning_rate": 8.606173994387279e-07, + "loss": 1.3738, + "step": 2024 + }, + { + "epoch": 2.5568181818181817, + "grad_norm": 3.517357587814331, + "learning_rate": 8.582787652011226e-07, + "loss": 1.4152, + "step": 2025 + }, + { + "epoch": 2.558080808080808, + "grad_norm": 3.5744385719299316, + "learning_rate": 8.559401309635174e-07, + "loss": 1.381, + "step": 2026 + }, + { + "epoch": 2.5593434343434343, + "grad_norm": 3.3060715198516846, + "learning_rate": 8.536014967259121e-07, + "loss": 1.3174, + "step": 2027 + }, + { + "epoch": 2.5606060606060606, + "grad_norm": 3.3646938800811768, + "learning_rate": 8.512628624883069e-07, + "loss": 1.3647, + "step": 2028 + }, + { + "epoch": 2.561868686868687, + "grad_norm": 3.4408724308013916, + "learning_rate": 8.489242282507017e-07, + "loss": 1.3381, + "step": 2029 + }, + { + "epoch": 2.563131313131313, + "grad_norm": 3.381373405456543, + "learning_rate": 8.465855940130964e-07, + "loss": 1.3228, + "step": 2030 + }, + { + "epoch": 2.563131313131313, + "eval_loss": 1.6585171222686768, + "eval_runtime": 37.1174, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.5643939393939394, + "grad_norm": 3.440295696258545, + "learning_rate": 8.442469597754912e-07, + "loss": 1.4118, + "step": 2031 + }, + { + "epoch": 2.5656565656565657, + "grad_norm": 3.497279644012451, + "learning_rate": 8.41908325537886e-07, + "loss": 1.3721, + "step": 2032 + }, + { + "epoch": 2.566919191919192, + "grad_norm": 3.302016496658325, + "learning_rate": 8.395696913002807e-07, + "loss": 1.3179, + "step": 2033 + }, + { + "epoch": 2.5681818181818183, + "grad_norm": 3.311852216720581, + "learning_rate": 8.372310570626755e-07, + "loss": 1.3641, + "step": 2034 + }, + { + "epoch": 2.5694444444444446, + "grad_norm": 3.566032648086548, + "learning_rate": 8.348924228250702e-07, + "loss": 1.3996, + "step": 2035 + }, + { + "epoch": 2.570707070707071, + "grad_norm": 3.6114914417266846, + "learning_rate": 8.32553788587465e-07, + "loss": 1.3694, + "step": 2036 + }, + { + "epoch": 2.5719696969696972, + "grad_norm": 3.52956223487854, + "learning_rate": 8.302151543498598e-07, + "loss": 1.3811, + "step": 2037 + }, + { + "epoch": 2.573232323232323, + "grad_norm": 3.372699022293091, + "learning_rate": 8.278765201122545e-07, + "loss": 1.3762, + "step": 2038 + }, + { + "epoch": 2.5744949494949494, + "grad_norm": 3.3569037914276123, + "learning_rate": 8.255378858746494e-07, + "loss": 1.3126, + "step": 2039 + }, + { + "epoch": 2.5757575757575757, + "grad_norm": 3.764512062072754, + "learning_rate": 8.231992516370441e-07, + "loss": 1.397, + "step": 2040 + }, + { + "epoch": 2.5757575757575757, + "eval_loss": 1.6595747470855713, + "eval_runtime": 37.1249, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2040 + }, + { + "epoch": 2.577020202020202, + "grad_norm": 3.3517415523529053, + "learning_rate": 8.208606173994387e-07, + "loss": 1.2947, + "step": 2041 + }, + { + "epoch": 2.5782828282828283, + "grad_norm": 3.386699676513672, + "learning_rate": 8.185219831618335e-07, + "loss": 1.3313, + "step": 2042 + }, + { + "epoch": 2.5795454545454546, + "grad_norm": 3.480076313018799, + "learning_rate": 8.161833489242284e-07, + "loss": 1.3852, + "step": 2043 + }, + { + "epoch": 2.580808080808081, + "grad_norm": 3.2437856197357178, + "learning_rate": 8.138447146866231e-07, + "loss": 1.3265, + "step": 2044 + }, + { + "epoch": 2.582070707070707, + "grad_norm": 3.411377191543579, + "learning_rate": 8.115060804490178e-07, + "loss": 1.3815, + "step": 2045 + }, + { + "epoch": 2.5833333333333335, + "grad_norm": 3.3562614917755127, + "learning_rate": 8.091674462114125e-07, + "loss": 1.3599, + "step": 2046 + }, + { + "epoch": 2.5845959595959593, + "grad_norm": 3.4753265380859375, + "learning_rate": 8.068288119738074e-07, + "loss": 1.3585, + "step": 2047 + }, + { + "epoch": 2.5858585858585856, + "grad_norm": 3.36663556098938, + "learning_rate": 8.044901777362021e-07, + "loss": 1.3942, + "step": 2048 + }, + { + "epoch": 2.587121212121212, + "grad_norm": 3.3582539558410645, + "learning_rate": 8.021515434985968e-07, + "loss": 1.3131, + "step": 2049 + }, + { + "epoch": 2.5883838383838382, + "grad_norm": 3.4529905319213867, + "learning_rate": 7.998129092609916e-07, + "loss": 1.3659, + "step": 2050 + }, + { + "epoch": 2.5883838383838382, + "eval_loss": 1.6579172611236572, + "eval_runtime": 37.1157, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.5896464646464645, + "grad_norm": 3.482149124145508, + "learning_rate": 7.974742750233864e-07, + "loss": 1.3699, + "step": 2051 + }, + { + "epoch": 2.590909090909091, + "grad_norm": 3.3723981380462646, + "learning_rate": 7.951356407857812e-07, + "loss": 1.3659, + "step": 2052 + }, + { + "epoch": 2.592171717171717, + "grad_norm": 3.3570878505706787, + "learning_rate": 7.927970065481759e-07, + "loss": 1.3449, + "step": 2053 + }, + { + "epoch": 2.5934343434343434, + "grad_norm": 3.4706034660339355, + "learning_rate": 7.904583723105706e-07, + "loss": 1.3752, + "step": 2054 + }, + { + "epoch": 2.5946969696969697, + "grad_norm": 3.4540836811065674, + "learning_rate": 7.881197380729655e-07, + "loss": 1.3561, + "step": 2055 + }, + { + "epoch": 2.595959595959596, + "grad_norm": 3.3945505619049072, + "learning_rate": 7.857811038353602e-07, + "loss": 1.3815, + "step": 2056 + }, + { + "epoch": 2.5972222222222223, + "grad_norm": 3.5257701873779297, + "learning_rate": 7.83442469597755e-07, + "loss": 1.3682, + "step": 2057 + }, + { + "epoch": 2.5984848484848486, + "grad_norm": 3.3582730293273926, + "learning_rate": 7.811038353601498e-07, + "loss": 1.3481, + "step": 2058 + }, + { + "epoch": 2.599747474747475, + "grad_norm": 3.277299404144287, + "learning_rate": 7.787652011225445e-07, + "loss": 1.2955, + "step": 2059 + }, + { + "epoch": 2.601010101010101, + "grad_norm": 3.414294719696045, + "learning_rate": 7.764265668849393e-07, + "loss": 1.3627, + "step": 2060 + }, + { + "epoch": 2.601010101010101, + "eval_loss": 1.6582155227661133, + "eval_runtime": 37.1197, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 2060 + }, + { + "epoch": 2.6022727272727275, + "grad_norm": 3.399158000946045, + "learning_rate": 7.74087932647334e-07, + "loss": 1.2828, + "step": 2061 + }, + { + "epoch": 2.6035353535353534, + "grad_norm": 3.4865469932556152, + "learning_rate": 7.717492984097288e-07, + "loss": 1.3871, + "step": 2062 + }, + { + "epoch": 2.6047979797979797, + "grad_norm": 3.5197713375091553, + "learning_rate": 7.694106641721236e-07, + "loss": 1.3994, + "step": 2063 + }, + { + "epoch": 2.606060606060606, + "grad_norm": 3.4030301570892334, + "learning_rate": 7.670720299345183e-07, + "loss": 1.3309, + "step": 2064 + }, + { + "epoch": 2.6073232323232323, + "grad_norm": 3.514847755432129, + "learning_rate": 7.647333956969131e-07, + "loss": 1.3955, + "step": 2065 + }, + { + "epoch": 2.6085858585858586, + "grad_norm": 3.4094550609588623, + "learning_rate": 7.623947614593079e-07, + "loss": 1.3486, + "step": 2066 + }, + { + "epoch": 2.609848484848485, + "grad_norm": 3.335812568664551, + "learning_rate": 7.600561272217026e-07, + "loss": 1.3556, + "step": 2067 + }, + { + "epoch": 2.611111111111111, + "grad_norm": 3.3734757900238037, + "learning_rate": 7.577174929840974e-07, + "loss": 1.3443, + "step": 2068 + }, + { + "epoch": 2.6123737373737375, + "grad_norm": 3.4860036373138428, + "learning_rate": 7.55378858746492e-07, + "loss": 1.3652, + "step": 2069 + }, + { + "epoch": 2.6136363636363638, + "grad_norm": 3.323676347732544, + "learning_rate": 7.53040224508887e-07, + "loss": 1.3547, + "step": 2070 + }, + { + "epoch": 2.6136363636363638, + "eval_loss": 1.6536892652511597, + "eval_runtime": 37.1179, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.61489898989899, + "grad_norm": 3.372002363204956, + "learning_rate": 7.507015902712817e-07, + "loss": 1.3369, + "step": 2071 + }, + { + "epoch": 2.616161616161616, + "grad_norm": 3.329336643218994, + "learning_rate": 7.483629560336763e-07, + "loss": 1.3463, + "step": 2072 + }, + { + "epoch": 2.617424242424242, + "grad_norm": 3.4763405323028564, + "learning_rate": 7.460243217960711e-07, + "loss": 1.3634, + "step": 2073 + }, + { + "epoch": 2.6186868686868685, + "grad_norm": 3.662146806716919, + "learning_rate": 7.43685687558466e-07, + "loss": 1.3864, + "step": 2074 + }, + { + "epoch": 2.619949494949495, + "grad_norm": 3.4778242111206055, + "learning_rate": 7.413470533208606e-07, + "loss": 1.4011, + "step": 2075 + }, + { + "epoch": 2.621212121212121, + "grad_norm": 3.494809627532959, + "learning_rate": 7.390084190832554e-07, + "loss": 1.3807, + "step": 2076 + }, + { + "epoch": 2.6224747474747474, + "grad_norm": 3.383918523788452, + "learning_rate": 7.366697848456501e-07, + "loss": 1.3477, + "step": 2077 + }, + { + "epoch": 2.6237373737373737, + "grad_norm": 3.415273666381836, + "learning_rate": 7.34331150608045e-07, + "loss": 1.3611, + "step": 2078 + }, + { + "epoch": 2.625, + "grad_norm": 3.4107258319854736, + "learning_rate": 7.319925163704397e-07, + "loss": 1.3545, + "step": 2079 + }, + { + "epoch": 2.6262626262626263, + "grad_norm": 3.4525718688964844, + "learning_rate": 7.296538821328344e-07, + "loss": 1.3402, + "step": 2080 + }, + { + "epoch": 2.6262626262626263, + "eval_loss": 1.651517629623413, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.6275252525252526, + "grad_norm": 3.373176336288452, + "learning_rate": 7.273152478952293e-07, + "loss": 1.3789, + "step": 2081 + }, + { + "epoch": 2.628787878787879, + "grad_norm": 3.379733085632324, + "learning_rate": 7.24976613657624e-07, + "loss": 1.3595, + "step": 2082 + }, + { + "epoch": 2.630050505050505, + "grad_norm": 3.2968251705169678, + "learning_rate": 7.226379794200187e-07, + "loss": 1.354, + "step": 2083 + }, + { + "epoch": 2.6313131313131315, + "grad_norm": 3.4465534687042236, + "learning_rate": 7.202993451824135e-07, + "loss": 1.3776, + "step": 2084 + }, + { + "epoch": 2.632575757575758, + "grad_norm": 3.4734654426574707, + "learning_rate": 7.179607109448083e-07, + "loss": 1.3893, + "step": 2085 + }, + { + "epoch": 2.633838383838384, + "grad_norm": 3.4402577877044678, + "learning_rate": 7.156220767072031e-07, + "loss": 1.3412, + "step": 2086 + }, + { + "epoch": 2.63510101010101, + "grad_norm": 3.5570950508117676, + "learning_rate": 7.132834424695978e-07, + "loss": 1.38, + "step": 2087 + }, + { + "epoch": 2.6363636363636362, + "grad_norm": 3.4646284580230713, + "learning_rate": 7.109448082319925e-07, + "loss": 1.3816, + "step": 2088 + }, + { + "epoch": 2.6376262626262625, + "grad_norm": 3.5913166999816895, + "learning_rate": 7.086061739943874e-07, + "loss": 1.3754, + "step": 2089 + }, + { + "epoch": 2.638888888888889, + "grad_norm": 3.4469869136810303, + "learning_rate": 7.062675397567821e-07, + "loss": 1.3582, + "step": 2090 + }, + { + "epoch": 2.638888888888889, + "eval_loss": 1.6559001207351685, + "eval_runtime": 37.3478, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.640151515151515, + "grad_norm": 3.5393295288085938, + "learning_rate": 7.039289055191769e-07, + "loss": 1.3602, + "step": 2091 + }, + { + "epoch": 2.6414141414141414, + "grad_norm": 3.431546211242676, + "learning_rate": 7.015902712815716e-07, + "loss": 1.3074, + "step": 2092 + }, + { + "epoch": 2.6426767676767677, + "grad_norm": 3.3025810718536377, + "learning_rate": 6.992516370439664e-07, + "loss": 1.2884, + "step": 2093 + }, + { + "epoch": 2.643939393939394, + "grad_norm": 3.4054274559020996, + "learning_rate": 6.969130028063612e-07, + "loss": 1.3283, + "step": 2094 + }, + { + "epoch": 2.6452020202020203, + "grad_norm": 3.5449748039245605, + "learning_rate": 6.945743685687559e-07, + "loss": 1.3896, + "step": 2095 + }, + { + "epoch": 2.6464646464646466, + "grad_norm": 3.528944253921509, + "learning_rate": 6.922357343311506e-07, + "loss": 1.3487, + "step": 2096 + }, + { + "epoch": 2.6477272727272725, + "grad_norm": 3.5601792335510254, + "learning_rate": 6.898971000935455e-07, + "loss": 1.3995, + "step": 2097 + }, + { + "epoch": 2.648989898989899, + "grad_norm": 3.358328104019165, + "learning_rate": 6.875584658559402e-07, + "loss": 1.3395, + "step": 2098 + }, + { + "epoch": 2.650252525252525, + "grad_norm": 3.392995595932007, + "learning_rate": 6.85219831618335e-07, + "loss": 1.3806, + "step": 2099 + }, + { + "epoch": 2.6515151515151514, + "grad_norm": 3.4869582653045654, + "learning_rate": 6.828811973807298e-07, + "loss": 1.3696, + "step": 2100 + }, + { + "epoch": 2.6515151515151514, + "eval_loss": 1.6575806140899658, + "eval_runtime": 37.4666, + "eval_samples_per_second": 26.69, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.6527777777777777, + "grad_norm": 3.29205060005188, + "learning_rate": 6.805425631431245e-07, + "loss": 1.3119, + "step": 2101 + }, + { + "epoch": 2.654040404040404, + "grad_norm": 3.309109687805176, + "learning_rate": 6.782039289055192e-07, + "loss": 1.326, + "step": 2102 + }, + { + "epoch": 2.6553030303030303, + "grad_norm": 3.3156912326812744, + "learning_rate": 6.758652946679139e-07, + "loss": 1.3296, + "step": 2103 + }, + { + "epoch": 2.6565656565656566, + "grad_norm": 3.45403790473938, + "learning_rate": 6.735266604303089e-07, + "loss": 1.4157, + "step": 2104 + }, + { + "epoch": 2.657828282828283, + "grad_norm": 3.540456533432007, + "learning_rate": 6.711880261927035e-07, + "loss": 1.3798, + "step": 2105 + }, + { + "epoch": 2.659090909090909, + "grad_norm": 3.4654109477996826, + "learning_rate": 6.688493919550982e-07, + "loss": 1.3992, + "step": 2106 + }, + { + "epoch": 2.6603535353535355, + "grad_norm": 3.514021873474121, + "learning_rate": 6.66510757717493e-07, + "loss": 1.3868, + "step": 2107 + }, + { + "epoch": 2.6616161616161618, + "grad_norm": 3.4709298610687256, + "learning_rate": 6.641721234798878e-07, + "loss": 1.3414, + "step": 2108 + }, + { + "epoch": 2.662878787878788, + "grad_norm": 3.4472742080688477, + "learning_rate": 6.618334892422825e-07, + "loss": 1.2879, + "step": 2109 + }, + { + "epoch": 2.6641414141414144, + "grad_norm": 3.7136893272399902, + "learning_rate": 6.594948550046773e-07, + "loss": 1.4012, + "step": 2110 + }, + { + "epoch": 2.6641414141414144, + "eval_loss": 1.658721923828125, + "eval_runtime": 36.9489, + "eval_samples_per_second": 27.064, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.6654040404040407, + "grad_norm": 3.3402535915374756, + "learning_rate": 6.57156220767072e-07, + "loss": 1.3412, + "step": 2111 + }, + { + "epoch": 2.6666666666666665, + "grad_norm": 3.3103139400482178, + "learning_rate": 6.548175865294669e-07, + "loss": 1.3207, + "step": 2112 + }, + { + "epoch": 2.667929292929293, + "grad_norm": 3.3308048248291016, + "learning_rate": 6.524789522918616e-07, + "loss": 1.3047, + "step": 2113 + }, + { + "epoch": 2.669191919191919, + "grad_norm": 3.4672188758850098, + "learning_rate": 6.501403180542563e-07, + "loss": 1.4158, + "step": 2114 + }, + { + "epoch": 2.6704545454545454, + "grad_norm": 3.522867202758789, + "learning_rate": 6.478016838166511e-07, + "loss": 1.3464, + "step": 2115 + }, + { + "epoch": 2.6717171717171717, + "grad_norm": 3.4110591411590576, + "learning_rate": 6.454630495790459e-07, + "loss": 1.3651, + "step": 2116 + }, + { + "epoch": 2.672979797979798, + "grad_norm": 3.4079062938690186, + "learning_rate": 6.431244153414407e-07, + "loss": 1.3554, + "step": 2117 + }, + { + "epoch": 2.6742424242424243, + "grad_norm": 3.4102203845977783, + "learning_rate": 6.407857811038354e-07, + "loss": 1.3953, + "step": 2118 + }, + { + "epoch": 2.6755050505050506, + "grad_norm": 3.381063222885132, + "learning_rate": 6.384471468662301e-07, + "loss": 1.3461, + "step": 2119 + }, + { + "epoch": 2.676767676767677, + "grad_norm": 3.3743748664855957, + "learning_rate": 6.36108512628625e-07, + "loss": 1.3559, + "step": 2120 + }, + { + "epoch": 2.676767676767677, + "eval_loss": 1.6545666456222534, + "eval_runtime": 37.0937, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.6780303030303028, + "grad_norm": 3.450329542160034, + "learning_rate": 6.337698783910197e-07, + "loss": 1.4058, + "step": 2121 + }, + { + "epoch": 2.679292929292929, + "grad_norm": 3.490386486053467, + "learning_rate": 6.314312441534144e-07, + "loss": 1.3704, + "step": 2122 + }, + { + "epoch": 2.6805555555555554, + "grad_norm": 3.689189910888672, + "learning_rate": 6.290926099158093e-07, + "loss": 1.3778, + "step": 2123 + }, + { + "epoch": 2.6818181818181817, + "grad_norm": 3.437407970428467, + "learning_rate": 6.26753975678204e-07, + "loss": 1.3673, + "step": 2124 + }, + { + "epoch": 2.683080808080808, + "grad_norm": 3.6720778942108154, + "learning_rate": 6.244153414405988e-07, + "loss": 1.4601, + "step": 2125 + }, + { + "epoch": 2.6843434343434343, + "grad_norm": 3.5332326889038086, + "learning_rate": 6.220767072029935e-07, + "loss": 1.3204, + "step": 2126 + }, + { + "epoch": 2.6856060606060606, + "grad_norm": 3.474308967590332, + "learning_rate": 6.197380729653882e-07, + "loss": 1.357, + "step": 2127 + }, + { + "epoch": 2.686868686868687, + "grad_norm": 3.4824960231781006, + "learning_rate": 6.173994387277831e-07, + "loss": 1.3412, + "step": 2128 + }, + { + "epoch": 2.688131313131313, + "grad_norm": 3.622313976287842, + "learning_rate": 6.150608044901778e-07, + "loss": 1.4076, + "step": 2129 + }, + { + "epoch": 2.6893939393939394, + "grad_norm": 3.520381212234497, + "learning_rate": 6.127221702525726e-07, + "loss": 1.3341, + "step": 2130 + }, + { + "epoch": 2.6893939393939394, + "eval_loss": 1.6562730073928833, + "eval_runtime": 37.1249, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.6906565656565657, + "grad_norm": 3.5275325775146484, + "learning_rate": 6.103835360149673e-07, + "loss": 1.3876, + "step": 2131 + }, + { + "epoch": 2.691919191919192, + "grad_norm": 3.5301501750946045, + "learning_rate": 6.080449017773621e-07, + "loss": 1.3745, + "step": 2132 + }, + { + "epoch": 2.6931818181818183, + "grad_norm": 3.4208099842071533, + "learning_rate": 6.057062675397568e-07, + "loss": 1.3435, + "step": 2133 + }, + { + "epoch": 2.6944444444444446, + "grad_norm": 3.3709633350372314, + "learning_rate": 6.033676333021516e-07, + "loss": 1.3356, + "step": 2134 + }, + { + "epoch": 2.695707070707071, + "grad_norm": 3.427328109741211, + "learning_rate": 6.010289990645463e-07, + "loss": 1.3395, + "step": 2135 + }, + { + "epoch": 2.6969696969696972, + "grad_norm": 3.543050527572632, + "learning_rate": 5.986903648269411e-07, + "loss": 1.364, + "step": 2136 + }, + { + "epoch": 2.698232323232323, + "grad_norm": 3.422373056411743, + "learning_rate": 5.963517305893358e-07, + "loss": 1.3119, + "step": 2137 + }, + { + "epoch": 2.6994949494949494, + "grad_norm": 3.4447410106658936, + "learning_rate": 5.940130963517307e-07, + "loss": 1.2964, + "step": 2138 + }, + { + "epoch": 2.7007575757575757, + "grad_norm": 3.498711585998535, + "learning_rate": 5.916744621141254e-07, + "loss": 1.3655, + "step": 2139 + }, + { + "epoch": 2.702020202020202, + "grad_norm": 3.433220148086548, + "learning_rate": 5.893358278765201e-07, + "loss": 1.3559, + "step": 2140 + }, + { + "epoch": 2.702020202020202, + "eval_loss": 1.6559937000274658, + "eval_runtime": 37.1318, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.7032828282828283, + "grad_norm": 3.3882076740264893, + "learning_rate": 5.86997193638915e-07, + "loss": 1.3324, + "step": 2141 + }, + { + "epoch": 2.7045454545454546, + "grad_norm": 3.4268569946289062, + "learning_rate": 5.846585594013097e-07, + "loss": 1.3666, + "step": 2142 + }, + { + "epoch": 2.705808080808081, + "grad_norm": 3.507413625717163, + "learning_rate": 5.823199251637045e-07, + "loss": 1.3435, + "step": 2143 + }, + { + "epoch": 2.707070707070707, + "grad_norm": 3.5061936378479004, + "learning_rate": 5.799812909260992e-07, + "loss": 1.3656, + "step": 2144 + }, + { + "epoch": 2.7083333333333335, + "grad_norm": 3.4619619846343994, + "learning_rate": 5.77642656688494e-07, + "loss": 1.3643, + "step": 2145 + }, + { + "epoch": 2.7095959595959593, + "grad_norm": 3.463290214538574, + "learning_rate": 5.753040224508888e-07, + "loss": 1.337, + "step": 2146 + }, + { + "epoch": 2.7108585858585856, + "grad_norm": 3.4522857666015625, + "learning_rate": 5.729653882132835e-07, + "loss": 1.4072, + "step": 2147 + }, + { + "epoch": 2.712121212121212, + "grad_norm": 3.468033790588379, + "learning_rate": 5.706267539756782e-07, + "loss": 1.3417, + "step": 2148 + }, + { + "epoch": 2.7133838383838382, + "grad_norm": 3.556464910507202, + "learning_rate": 5.682881197380731e-07, + "loss": 1.3936, + "step": 2149 + }, + { + "epoch": 2.7146464646464645, + "grad_norm": 3.4759442806243896, + "learning_rate": 5.659494855004677e-07, + "loss": 1.3114, + "step": 2150 + }, + { + "epoch": 2.7146464646464645, + "eval_loss": 1.6544948816299438, + "eval_runtime": 37.1258, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 2150 + }, + { + "epoch": 2.715909090909091, + "grad_norm": 3.5131609439849854, + "learning_rate": 5.636108512628626e-07, + "loss": 1.3116, + "step": 2151 + }, + { + "epoch": 2.717171717171717, + "grad_norm": 3.492794990539551, + "learning_rate": 5.612722170252573e-07, + "loss": 1.3297, + "step": 2152 + }, + { + "epoch": 2.7184343434343434, + "grad_norm": 3.3546905517578125, + "learning_rate": 5.58933582787652e-07, + "loss": 1.3348, + "step": 2153 + }, + { + "epoch": 2.7196969696969697, + "grad_norm": 3.3591794967651367, + "learning_rate": 5.565949485500468e-07, + "loss": 1.3355, + "step": 2154 + }, + { + "epoch": 2.720959595959596, + "grad_norm": 3.4083364009857178, + "learning_rate": 5.542563143124416e-07, + "loss": 1.329, + "step": 2155 + }, + { + "epoch": 2.7222222222222223, + "grad_norm": 3.625196933746338, + "learning_rate": 5.519176800748364e-07, + "loss": 1.4063, + "step": 2156 + }, + { + "epoch": 2.7234848484848486, + "grad_norm": 3.534701347351074, + "learning_rate": 5.495790458372311e-07, + "loss": 1.4223, + "step": 2157 + }, + { + "epoch": 2.724747474747475, + "grad_norm": 3.4186694622039795, + "learning_rate": 5.472404115996258e-07, + "loss": 1.373, + "step": 2158 + }, + { + "epoch": 2.726010101010101, + "grad_norm": 3.31740140914917, + "learning_rate": 5.449017773620207e-07, + "loss": 1.3691, + "step": 2159 + }, + { + "epoch": 2.7272727272727275, + "grad_norm": 3.4586780071258545, + "learning_rate": 5.425631431244154e-07, + "loss": 1.3948, + "step": 2160 + }, + { + "epoch": 2.7272727272727275, + "eval_loss": 1.65037202835083, + "eval_runtime": 37.1256, + "eval_samples_per_second": 26.936, + "eval_steps_per_second": 1.131, + "step": 2160 + }, + { + "epoch": 2.7285353535353534, + "grad_norm": 3.497927188873291, + "learning_rate": 5.402245088868101e-07, + "loss": 1.3483, + "step": 2161 + }, + { + "epoch": 2.7297979797979797, + "grad_norm": 3.4493894577026367, + "learning_rate": 5.37885874649205e-07, + "loss": 1.365, + "step": 2162 + }, + { + "epoch": 2.731060606060606, + "grad_norm": 3.4639272689819336, + "learning_rate": 5.355472404115996e-07, + "loss": 1.3822, + "step": 2163 + }, + { + "epoch": 2.7323232323232323, + "grad_norm": 3.5272274017333984, + "learning_rate": 5.332086061739945e-07, + "loss": 1.3722, + "step": 2164 + }, + { + "epoch": 2.7335858585858586, + "grad_norm": 3.3754897117614746, + "learning_rate": 5.308699719363892e-07, + "loss": 1.3444, + "step": 2165 + }, + { + "epoch": 2.734848484848485, + "grad_norm": 3.374189615249634, + "learning_rate": 5.285313376987839e-07, + "loss": 1.3459, + "step": 2166 + }, + { + "epoch": 2.736111111111111, + "grad_norm": 3.2838594913482666, + "learning_rate": 5.261927034611787e-07, + "loss": 1.3476, + "step": 2167 + }, + { + "epoch": 2.7373737373737375, + "grad_norm": 3.6404688358306885, + "learning_rate": 5.238540692235735e-07, + "loss": 1.4402, + "step": 2168 + }, + { + "epoch": 2.7386363636363638, + "grad_norm": 3.395650625228882, + "learning_rate": 5.215154349859682e-07, + "loss": 1.3657, + "step": 2169 + }, + { + "epoch": 2.73989898989899, + "grad_norm": 3.5969934463500977, + "learning_rate": 5.19176800748363e-07, + "loss": 1.3823, + "step": 2170 + }, + { + "epoch": 2.73989898989899, + "eval_loss": 1.6518349647521973, + "eval_runtime": 37.1587, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.741161616161616, + "grad_norm": 3.495293617248535, + "learning_rate": 5.168381665107577e-07, + "loss": 1.3524, + "step": 2171 + }, + { + "epoch": 2.742424242424242, + "grad_norm": 3.486459255218506, + "learning_rate": 5.144995322731526e-07, + "loss": 1.3554, + "step": 2172 + }, + { + "epoch": 2.7436868686868685, + "grad_norm": 3.5103650093078613, + "learning_rate": 5.121608980355473e-07, + "loss": 1.3699, + "step": 2173 + }, + { + "epoch": 2.744949494949495, + "grad_norm": 3.401812791824341, + "learning_rate": 5.09822263797942e-07, + "loss": 1.4116, + "step": 2174 + }, + { + "epoch": 2.746212121212121, + "grad_norm": 3.5474627017974854, + "learning_rate": 5.074836295603368e-07, + "loss": 1.3821, + "step": 2175 + }, + { + "epoch": 2.7474747474747474, + "grad_norm": 3.347783327102661, + "learning_rate": 5.051449953227316e-07, + "loss": 1.3235, + "step": 2176 + }, + { + "epoch": 2.7487373737373737, + "grad_norm": 3.385366916656494, + "learning_rate": 5.028063610851263e-07, + "loss": 1.379, + "step": 2177 + }, + { + "epoch": 2.75, + "grad_norm": 3.563000202178955, + "learning_rate": 5.004677268475211e-07, + "loss": 1.411, + "step": 2178 + }, + { + "epoch": 2.7512626262626263, + "grad_norm": 3.539781093597412, + "learning_rate": 4.981290926099158e-07, + "loss": 1.3351, + "step": 2179 + }, + { + "epoch": 2.7525252525252526, + "grad_norm": 3.477810859680176, + "learning_rate": 4.957904583723106e-07, + "loss": 1.3385, + "step": 2180 + }, + { + "epoch": 2.7525252525252526, + "eval_loss": 1.6517527103424072, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.753787878787879, + "grad_norm": 3.6181442737579346, + "learning_rate": 4.934518241347053e-07, + "loss": 1.4082, + "step": 2181 + }, + { + "epoch": 2.755050505050505, + "grad_norm": 3.5040857791900635, + "learning_rate": 4.911131898971001e-07, + "loss": 1.3886, + "step": 2182 + }, + { + "epoch": 2.7563131313131315, + "grad_norm": 3.616530656814575, + "learning_rate": 4.887745556594949e-07, + "loss": 1.4308, + "step": 2183 + }, + { + "epoch": 2.757575757575758, + "grad_norm": 3.689122200012207, + "learning_rate": 4.864359214218896e-07, + "loss": 1.3394, + "step": 2184 + }, + { + "epoch": 2.758838383838384, + "grad_norm": 3.432558298110962, + "learning_rate": 4.840972871842845e-07, + "loss": 1.3169, + "step": 2185 + }, + { + "epoch": 2.76010101010101, + "grad_norm": 3.4511947631835938, + "learning_rate": 4.817586529466792e-07, + "loss": 1.3488, + "step": 2186 + }, + { + "epoch": 2.7613636363636362, + "grad_norm": 3.4622750282287598, + "learning_rate": 4.794200187090739e-07, + "loss": 1.3744, + "step": 2187 + }, + { + "epoch": 2.7626262626262625, + "grad_norm": 3.3733813762664795, + "learning_rate": 4.770813844714687e-07, + "loss": 1.3744, + "step": 2188 + }, + { + "epoch": 2.763888888888889, + "grad_norm": 3.4657459259033203, + "learning_rate": 4.7474275023386347e-07, + "loss": 1.3686, + "step": 2189 + }, + { + "epoch": 2.765151515151515, + "grad_norm": 3.5236029624938965, + "learning_rate": 4.724041159962582e-07, + "loss": 1.3809, + "step": 2190 + }, + { + "epoch": 2.765151515151515, + "eval_loss": 1.653585433959961, + "eval_runtime": 37.2341, + "eval_samples_per_second": 26.857, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.7664141414141414, + "grad_norm": 3.5313808917999268, + "learning_rate": 4.70065481758653e-07, + "loss": 1.3772, + "step": 2191 + }, + { + "epoch": 2.7676767676767677, + "grad_norm": 3.4364354610443115, + "learning_rate": 4.6772684752104773e-07, + "loss": 1.3641, + "step": 2192 + }, + { + "epoch": 2.768939393939394, + "grad_norm": 3.5184450149536133, + "learning_rate": 4.653882132834425e-07, + "loss": 1.3629, + "step": 2193 + }, + { + "epoch": 2.7702020202020203, + "grad_norm": 3.4177098274230957, + "learning_rate": 4.6304957904583726e-07, + "loss": 1.3624, + "step": 2194 + }, + { + "epoch": 2.7714646464646466, + "grad_norm": 3.365241527557373, + "learning_rate": 4.6071094480823205e-07, + "loss": 1.3077, + "step": 2195 + }, + { + "epoch": 2.7727272727272725, + "grad_norm": 3.479496955871582, + "learning_rate": 4.583723105706268e-07, + "loss": 1.3509, + "step": 2196 + }, + { + "epoch": 2.773989898989899, + "grad_norm": 3.5142934322357178, + "learning_rate": 4.5603367633302157e-07, + "loss": 1.3876, + "step": 2197 + }, + { + "epoch": 2.775252525252525, + "grad_norm": 3.4927608966827393, + "learning_rate": 4.536950420954163e-07, + "loss": 1.3403, + "step": 2198 + }, + { + "epoch": 2.7765151515151514, + "grad_norm": 3.4133102893829346, + "learning_rate": 4.513564078578111e-07, + "loss": 1.3748, + "step": 2199 + }, + { + "epoch": 2.7777777777777777, + "grad_norm": 3.4805235862731934, + "learning_rate": 4.490177736202058e-07, + "loss": 1.3569, + "step": 2200 + }, + { + "epoch": 2.7777777777777777, + "eval_loss": 1.6538234949111938, + "eval_runtime": 37.4443, + "eval_samples_per_second": 26.706, + "eval_steps_per_second": 1.122, + "step": 2200 + }, + { + "epoch": 2.779040404040404, + "grad_norm": 3.498913526535034, + "learning_rate": 4.4667913938260063e-07, + "loss": 1.2893, + "step": 2201 + }, + { + "epoch": 2.7803030303030303, + "grad_norm": 3.525386333465576, + "learning_rate": 4.443405051449953e-07, + "loss": 1.3991, + "step": 2202 + }, + { + "epoch": 2.7815656565656566, + "grad_norm": 3.5398542881011963, + "learning_rate": 4.420018709073901e-07, + "loss": 1.4024, + "step": 2203 + }, + { + "epoch": 2.782828282828283, + "grad_norm": 3.407313823699951, + "learning_rate": 4.396632366697849e-07, + "loss": 1.315, + "step": 2204 + }, + { + "epoch": 2.784090909090909, + "grad_norm": 3.4025394916534424, + "learning_rate": 4.3732460243217963e-07, + "loss": 1.3267, + "step": 2205 + }, + { + "epoch": 2.7853535353535355, + "grad_norm": 3.4221434593200684, + "learning_rate": 4.349859681945744e-07, + "loss": 1.3376, + "step": 2206 + }, + { + "epoch": 2.7866161616161618, + "grad_norm": 3.5077474117279053, + "learning_rate": 4.3264733395696915e-07, + "loss": 1.3399, + "step": 2207 + }, + { + "epoch": 2.787878787878788, + "grad_norm": 3.442110538482666, + "learning_rate": 4.3030869971936394e-07, + "loss": 1.3272, + "step": 2208 + }, + { + "epoch": 2.7891414141414144, + "grad_norm": 3.4591195583343506, + "learning_rate": 4.279700654817587e-07, + "loss": 1.3445, + "step": 2209 + }, + { + "epoch": 2.7904040404040407, + "grad_norm": 3.455976963043213, + "learning_rate": 4.2563143124415347e-07, + "loss": 1.3594, + "step": 2210 + }, + { + "epoch": 2.7904040404040407, + "eval_loss": 1.653259515762329, + "eval_runtime": 36.9822, + "eval_samples_per_second": 27.04, + "eval_steps_per_second": 1.136, + "step": 2210 + }, + { + "epoch": 2.7916666666666665, + "grad_norm": 3.427316427230835, + "learning_rate": 4.232927970065482e-07, + "loss": 1.3921, + "step": 2211 + }, + { + "epoch": 2.792929292929293, + "grad_norm": 3.426870107650757, + "learning_rate": 4.20954162768943e-07, + "loss": 1.3771, + "step": 2212 + }, + { + "epoch": 2.794191919191919, + "grad_norm": 3.461211919784546, + "learning_rate": 4.1861552853133774e-07, + "loss": 1.3869, + "step": 2213 + }, + { + "epoch": 2.7954545454545454, + "grad_norm": 3.4966578483581543, + "learning_rate": 4.162768942937325e-07, + "loss": 1.369, + "step": 2214 + }, + { + "epoch": 2.7967171717171717, + "grad_norm": 3.6234564781188965, + "learning_rate": 4.1393826005612726e-07, + "loss": 1.3954, + "step": 2215 + }, + { + "epoch": 2.797979797979798, + "grad_norm": 3.300652265548706, + "learning_rate": 4.1159962581852205e-07, + "loss": 1.341, + "step": 2216 + }, + { + "epoch": 2.7992424242424243, + "grad_norm": 3.4092752933502197, + "learning_rate": 4.0926099158091674e-07, + "loss": 1.3775, + "step": 2217 + }, + { + "epoch": 2.8005050505050506, + "grad_norm": 3.3984715938568115, + "learning_rate": 4.0692235734331153e-07, + "loss": 1.3748, + "step": 2218 + }, + { + "epoch": 2.801767676767677, + "grad_norm": 3.4459221363067627, + "learning_rate": 4.0458372310570626e-07, + "loss": 1.327, + "step": 2219 + }, + { + "epoch": 2.8030303030303028, + "grad_norm": 3.4371068477630615, + "learning_rate": 4.0224508886810105e-07, + "loss": 1.3855, + "step": 2220 + }, + { + "epoch": 2.8030303030303028, + "eval_loss": 1.6508564949035645, + "eval_runtime": 37.1213, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 2220 + }, + { + "epoch": 2.804292929292929, + "grad_norm": 3.4269979000091553, + "learning_rate": 3.999064546304958e-07, + "loss": 1.3736, + "step": 2221 + }, + { + "epoch": 2.8055555555555554, + "grad_norm": 3.452821969985962, + "learning_rate": 3.975678203928906e-07, + "loss": 1.344, + "step": 2222 + }, + { + "epoch": 2.8068181818181817, + "grad_norm": 3.5009210109710693, + "learning_rate": 3.952291861552853e-07, + "loss": 1.3673, + "step": 2223 + }, + { + "epoch": 2.808080808080808, + "grad_norm": 3.425577402114868, + "learning_rate": 3.928905519176801e-07, + "loss": 1.3496, + "step": 2224 + }, + { + "epoch": 2.8093434343434343, + "grad_norm": 3.3825881481170654, + "learning_rate": 3.905519176800749e-07, + "loss": 1.367, + "step": 2225 + }, + { + "epoch": 2.8106060606060606, + "grad_norm": 3.5621533393859863, + "learning_rate": 3.8821328344246963e-07, + "loss": 1.3823, + "step": 2226 + }, + { + "epoch": 2.811868686868687, + "grad_norm": 3.5179245471954346, + "learning_rate": 3.858746492048644e-07, + "loss": 1.3498, + "step": 2227 + }, + { + "epoch": 2.813131313131313, + "grad_norm": 3.3433382511138916, + "learning_rate": 3.8353601496725916e-07, + "loss": 1.3658, + "step": 2228 + }, + { + "epoch": 2.8143939393939394, + "grad_norm": 3.5230610370635986, + "learning_rate": 3.8119738072965395e-07, + "loss": 1.399, + "step": 2229 + }, + { + "epoch": 2.8156565656565657, + "grad_norm": 3.5895588397979736, + "learning_rate": 3.788587464920487e-07, + "loss": 1.4021, + "step": 2230 + }, + { + "epoch": 2.8156565656565657, + "eval_loss": 1.6516975164413452, + "eval_runtime": 37.1659, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 2230 + }, + { + "epoch": 2.816919191919192, + "grad_norm": 3.5839803218841553, + "learning_rate": 3.765201122544435e-07, + "loss": 1.4384, + "step": 2231 + }, + { + "epoch": 2.8181818181818183, + "grad_norm": 3.3988068103790283, + "learning_rate": 3.7418147801683816e-07, + "loss": 1.345, + "step": 2232 + }, + { + "epoch": 2.8194444444444446, + "grad_norm": 3.474256992340088, + "learning_rate": 3.71842843779233e-07, + "loss": 1.3632, + "step": 2233 + }, + { + "epoch": 2.820707070707071, + "grad_norm": 3.442566156387329, + "learning_rate": 3.695042095416277e-07, + "loss": 1.4078, + "step": 2234 + }, + { + "epoch": 2.8219696969696972, + "grad_norm": 3.4802756309509277, + "learning_rate": 3.671655753040225e-07, + "loss": 1.3926, + "step": 2235 + }, + { + "epoch": 2.823232323232323, + "grad_norm": 3.440426826477051, + "learning_rate": 3.648269410664172e-07, + "loss": 1.3679, + "step": 2236 + }, + { + "epoch": 2.8244949494949494, + "grad_norm": 3.3994140625, + "learning_rate": 3.62488306828812e-07, + "loss": 1.3665, + "step": 2237 + }, + { + "epoch": 2.8257575757575757, + "grad_norm": 3.4944987297058105, + "learning_rate": 3.6014967259120674e-07, + "loss": 1.3803, + "step": 2238 + }, + { + "epoch": 2.827020202020202, + "grad_norm": 3.3832037448883057, + "learning_rate": 3.5781103835360153e-07, + "loss": 1.368, + "step": 2239 + }, + { + "epoch": 2.8282828282828283, + "grad_norm": 3.3579232692718506, + "learning_rate": 3.5547240411599627e-07, + "loss": 1.365, + "step": 2240 + }, + { + "epoch": 2.8282828282828283, + "eval_loss": 1.6520887613296509, + "eval_runtime": 37.1559, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 2240 + }, + { + "epoch": 2.8295454545454546, + "grad_norm": 3.546870231628418, + "learning_rate": 3.5313376987839106e-07, + "loss": 1.3597, + "step": 2241 + }, + { + "epoch": 2.830808080808081, + "grad_norm": 3.452018976211548, + "learning_rate": 3.507951356407858e-07, + "loss": 1.3504, + "step": 2242 + }, + { + "epoch": 2.832070707070707, + "grad_norm": 3.39516019821167, + "learning_rate": 3.484565014031806e-07, + "loss": 1.3248, + "step": 2243 + }, + { + "epoch": 2.8333333333333335, + "grad_norm": 3.3939480781555176, + "learning_rate": 3.461178671655753e-07, + "loss": 1.3132, + "step": 2244 + }, + { + "epoch": 2.8345959595959593, + "grad_norm": 3.385967493057251, + "learning_rate": 3.437792329279701e-07, + "loss": 1.3351, + "step": 2245 + }, + { + "epoch": 2.8358585858585856, + "grad_norm": 3.4794607162475586, + "learning_rate": 3.414405986903649e-07, + "loss": 1.3758, + "step": 2246 + }, + { + "epoch": 2.837121212121212, + "grad_norm": 3.4628307819366455, + "learning_rate": 3.391019644527596e-07, + "loss": 1.3738, + "step": 2247 + }, + { + "epoch": 2.8383838383838382, + "grad_norm": 3.338982105255127, + "learning_rate": 3.3676333021515443e-07, + "loss": 1.3227, + "step": 2248 + }, + { + "epoch": 2.8396464646464645, + "grad_norm": 3.456300735473633, + "learning_rate": 3.344246959775491e-07, + "loss": 1.3439, + "step": 2249 + }, + { + "epoch": 2.840909090909091, + "grad_norm": 3.3870697021484375, + "learning_rate": 3.320860617399439e-07, + "loss": 1.3215, + "step": 2250 + }, + { + "epoch": 2.840909090909091, + "eval_loss": 1.654848337173462, + "eval_runtime": 37.1651, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 2250 + }, + { + "epoch": 2.842171717171717, + "grad_norm": 3.285362720489502, + "learning_rate": 3.2974742750233864e-07, + "loss": 1.3267, + "step": 2251 + }, + { + "epoch": 2.8434343434343434, + "grad_norm": 3.5300493240356445, + "learning_rate": 3.2740879326473343e-07, + "loss": 1.3731, + "step": 2252 + }, + { + "epoch": 2.8446969696969697, + "grad_norm": 3.3968286514282227, + "learning_rate": 3.2507015902712817e-07, + "loss": 1.3523, + "step": 2253 + }, + { + "epoch": 2.845959595959596, + "grad_norm": 3.546522855758667, + "learning_rate": 3.2273152478952296e-07, + "loss": 1.3853, + "step": 2254 + }, + { + "epoch": 2.8472222222222223, + "grad_norm": 3.357612371444702, + "learning_rate": 3.203928905519177e-07, + "loss": 1.3502, + "step": 2255 + }, + { + "epoch": 2.8484848484848486, + "grad_norm": 3.43454909324646, + "learning_rate": 3.180542563143125e-07, + "loss": 1.3358, + "step": 2256 + }, + { + "epoch": 2.849747474747475, + "grad_norm": 3.264030694961548, + "learning_rate": 3.157156220767072e-07, + "loss": 1.2898, + "step": 2257 + }, + { + "epoch": 2.851010101010101, + "grad_norm": 3.5119142532348633, + "learning_rate": 3.13376987839102e-07, + "loss": 1.4031, + "step": 2258 + }, + { + "epoch": 2.8522727272727275, + "grad_norm": 3.3765857219696045, + "learning_rate": 3.1103835360149675e-07, + "loss": 1.3499, + "step": 2259 + }, + { + "epoch": 2.8535353535353534, + "grad_norm": 3.5452826023101807, + "learning_rate": 3.0869971936389154e-07, + "loss": 1.3963, + "step": 2260 + }, + { + "epoch": 2.8535353535353534, + "eval_loss": 1.652150273323059, + "eval_runtime": 37.1616, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 2260 + }, + { + "epoch": 2.8547979797979797, + "grad_norm": 3.403693199157715, + "learning_rate": 3.063610851262863e-07, + "loss": 1.3509, + "step": 2261 + }, + { + "epoch": 2.856060606060606, + "grad_norm": 3.3340814113616943, + "learning_rate": 3.0402245088868107e-07, + "loss": 1.3238, + "step": 2262 + }, + { + "epoch": 2.8573232323232323, + "grad_norm": 3.4628171920776367, + "learning_rate": 3.016838166510758e-07, + "loss": 1.4144, + "step": 2263 + }, + { + "epoch": 2.8585858585858586, + "grad_norm": 3.4264028072357178, + "learning_rate": 2.9934518241347054e-07, + "loss": 1.3699, + "step": 2264 + }, + { + "epoch": 2.859848484848485, + "grad_norm": 3.281966209411621, + "learning_rate": 2.9700654817586533e-07, + "loss": 1.3189, + "step": 2265 + }, + { + "epoch": 2.861111111111111, + "grad_norm": 3.3605847358703613, + "learning_rate": 2.9466791393826007e-07, + "loss": 1.3155, + "step": 2266 + }, + { + "epoch": 2.8623737373737375, + "grad_norm": 3.4888250827789307, + "learning_rate": 2.9232927970065486e-07, + "loss": 1.3941, + "step": 2267 + }, + { + "epoch": 2.8636363636363638, + "grad_norm": 3.3471014499664307, + "learning_rate": 2.899906454630496e-07, + "loss": 1.3344, + "step": 2268 + }, + { + "epoch": 2.86489898989899, + "grad_norm": 3.410423517227173, + "learning_rate": 2.876520112254444e-07, + "loss": 1.3501, + "step": 2269 + }, + { + "epoch": 2.866161616161616, + "grad_norm": 3.5021302700042725, + "learning_rate": 2.853133769878391e-07, + "loss": 1.3478, + "step": 2270 + }, + { + "epoch": 2.866161616161616, + "eval_loss": 1.6510955095291138, + "eval_runtime": 37.1536, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 2270 + }, + { + "epoch": 2.867424242424242, + "grad_norm": 3.4819681644439697, + "learning_rate": 2.8297474275023386e-07, + "loss": 1.3521, + "step": 2271 + }, + { + "epoch": 2.8686868686868685, + "grad_norm": 3.5270345211029053, + "learning_rate": 2.8063610851262865e-07, + "loss": 1.4129, + "step": 2272 + }, + { + "epoch": 2.869949494949495, + "grad_norm": 3.4656264781951904, + "learning_rate": 2.782974742750234e-07, + "loss": 1.3363, + "step": 2273 + }, + { + "epoch": 2.871212121212121, + "grad_norm": 3.5093917846679688, + "learning_rate": 2.759588400374182e-07, + "loss": 1.3369, + "step": 2274 + }, + { + "epoch": 2.8724747474747474, + "grad_norm": 3.590909719467163, + "learning_rate": 2.736202057998129e-07, + "loss": 1.3865, + "step": 2275 + }, + { + "epoch": 2.8737373737373737, + "grad_norm": 3.407057523727417, + "learning_rate": 2.712815715622077e-07, + "loss": 1.3598, + "step": 2276 + }, + { + "epoch": 2.875, + "grad_norm": 3.543407440185547, + "learning_rate": 2.689429373246025e-07, + "loss": 1.3909, + "step": 2277 + }, + { + "epoch": 2.8762626262626263, + "grad_norm": 3.589087963104248, + "learning_rate": 2.6660430308699723e-07, + "loss": 1.3743, + "step": 2278 + }, + { + "epoch": 2.8775252525252526, + "grad_norm": 3.6114728450775146, + "learning_rate": 2.6426566884939197e-07, + "loss": 1.3511, + "step": 2279 + }, + { + "epoch": 2.878787878787879, + "grad_norm": 3.3339433670043945, + "learning_rate": 2.6192703461178676e-07, + "loss": 1.2917, + "step": 2280 + }, + { + "epoch": 2.878787878787879, + "eval_loss": 1.6515085697174072, + "eval_runtime": 37.1483, + "eval_samples_per_second": 26.919, + "eval_steps_per_second": 1.131, + "step": 2280 + }, + { + "epoch": 2.880050505050505, + "grad_norm": 3.443605661392212, + "learning_rate": 2.595884003741815e-07, + "loss": 1.3677, + "step": 2281 + }, + { + "epoch": 2.8813131313131315, + "grad_norm": 3.5950000286102295, + "learning_rate": 2.572497661365763e-07, + "loss": 1.4666, + "step": 2282 + }, + { + "epoch": 2.882575757575758, + "grad_norm": 3.3595333099365234, + "learning_rate": 2.54911131898971e-07, + "loss": 1.3724, + "step": 2283 + }, + { + "epoch": 2.883838383838384, + "grad_norm": 3.4813501834869385, + "learning_rate": 2.525724976613658e-07, + "loss": 1.3422, + "step": 2284 + }, + { + "epoch": 2.88510101010101, + "grad_norm": 3.372537136077881, + "learning_rate": 2.5023386342376055e-07, + "loss": 1.3379, + "step": 2285 + }, + { + "epoch": 2.8863636363636362, + "grad_norm": 3.496241569519043, + "learning_rate": 2.478952291861553e-07, + "loss": 1.3845, + "step": 2286 + }, + { + "epoch": 2.8876262626262625, + "grad_norm": 3.51509165763855, + "learning_rate": 2.455565949485501e-07, + "loss": 1.3647, + "step": 2287 + }, + { + "epoch": 2.888888888888889, + "grad_norm": 3.3701627254486084, + "learning_rate": 2.432179607109448e-07, + "loss": 1.3547, + "step": 2288 + }, + { + "epoch": 2.890151515151515, + "grad_norm": 3.4733920097351074, + "learning_rate": 2.408793264733396e-07, + "loss": 1.3603, + "step": 2289 + }, + { + "epoch": 2.8914141414141414, + "grad_norm": 3.482700824737549, + "learning_rate": 2.3854069223573434e-07, + "loss": 1.3853, + "step": 2290 + }, + { + "epoch": 2.8914141414141414, + "eval_loss": 1.6494017839431763, + "eval_runtime": 37.2117, + "eval_samples_per_second": 26.873, + "eval_steps_per_second": 1.129, + "step": 2290 + }, + { + "epoch": 2.8926767676767677, + "grad_norm": 3.351447582244873, + "learning_rate": 2.362020579981291e-07, + "loss": 1.3422, + "step": 2291 + }, + { + "epoch": 2.893939393939394, + "grad_norm": 3.4265522956848145, + "learning_rate": 2.3386342376052386e-07, + "loss": 1.352, + "step": 2292 + }, + { + "epoch": 2.8952020202020203, + "grad_norm": 3.434129238128662, + "learning_rate": 2.3152478952291863e-07, + "loss": 1.376, + "step": 2293 + }, + { + "epoch": 2.8964646464646466, + "grad_norm": 3.347670555114746, + "learning_rate": 2.291861552853134e-07, + "loss": 1.3418, + "step": 2294 + }, + { + "epoch": 2.8977272727272725, + "grad_norm": 3.5123982429504395, + "learning_rate": 2.2684752104770815e-07, + "loss": 1.3943, + "step": 2295 + }, + { + "epoch": 2.898989898989899, + "grad_norm": 3.48795747756958, + "learning_rate": 2.245088868101029e-07, + "loss": 1.3747, + "step": 2296 + }, + { + "epoch": 2.900252525252525, + "grad_norm": 3.3663408756256104, + "learning_rate": 2.2217025257249766e-07, + "loss": 1.3863, + "step": 2297 + }, + { + "epoch": 2.9015151515151514, + "grad_norm": 3.4627926349639893, + "learning_rate": 2.1983161833489245e-07, + "loss": 1.4198, + "step": 2298 + }, + { + "epoch": 2.9027777777777777, + "grad_norm": 3.3175923824310303, + "learning_rate": 2.174929840972872e-07, + "loss": 1.3743, + "step": 2299 + }, + { + "epoch": 2.904040404040404, + "grad_norm": 3.362423896789551, + "learning_rate": 2.1515434985968197e-07, + "loss": 1.3724, + "step": 2300 + }, + { + "epoch": 2.904040404040404, + "eval_loss": 1.6507534980773926, + "eval_runtime": 37.3708, + "eval_samples_per_second": 26.759, + "eval_steps_per_second": 1.124, + "step": 2300 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 7.169811641532416e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-2300/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/latest new file mode 100644 index 0000000000000000000000000000000000000000..e5bdf58d4f29d34e909da25905fad376f73e7c29 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/latest @@ -0,0 +1 @@ +global_step400 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..06024c7992821d2ac1dd8a9c0eec199a90910543 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/trainer_state.json @@ -0,0 +1,3153 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.5050505050505051, + "eval_steps": 10, + "global_step": 400, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 1.246923763744768e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-400/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-400/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/latest new file mode 100644 index 0000000000000000000000000000000000000000..12cae1adf3af8546b4141c6f62261c8e99839a54 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/latest @@ -0,0 +1 @@ +global_step600 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..d8bdf7f41c48b265cd81a2dc5851296a4e24f3b9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/trainer_state.json @@ -0,0 +1,4713 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.7575757575757576, + "eval_steps": 10, + "global_step": 600, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 1.870385645617152e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-600/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-600/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/config.json @@ -0,0 +1,36 @@ +{ + "_name_or_path": "meta-llama/Llama-3.2-3B", + "architectures": [ + "LlamaForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "bos_token_id": 128000, + "eos_token_id": 128001, + "head_dim": 128, + "hidden_act": "silu", + "hidden_size": 3072, + "initializer_range": 0.02, + "intermediate_size": 8192, + "max_position_embeddings": 131072, + "mlp_bias": false, + "model_type": "llama", + "num_attention_heads": 24, + "num_hidden_layers": 28, + "num_key_value_heads": 8, + "pretraining_tp": 1, + "rms_norm_eps": 1e-05, + "rope_scaling": { + "factor": 32.0, + "high_freq_factor": 4.0, + "low_freq_factor": 1.0, + "original_max_position_embeddings": 8192, + "rope_type": "llama3" + }, + "rope_theta": 500000.0, + "tie_word_embeddings": true, + "torch_dtype": "float16", + "transformers_version": "4.45.1", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/generation_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "do_sample": true, + "eos_token_id": 128001, + "temperature": 0.6, + "top_p": 0.9, + "transformers_version": "4.45.1" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/latest b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/latest new file mode 100644 index 0000000000000000000000000000000000000000..a0f3e526aa9af2ac647b278f006bb9616843c5d6 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/latest @@ -0,0 +1 @@ +global_step700 \ No newline at end of file diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/model.safetensors.index.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/model.safetensors.index.json @@ -0,0 +1,262 @@ +{ + "metadata": { + "total_size": 7213504512 + }, + "weight_map": { + "lm_head.weight": "model-00002-of-00002.safetensors", + "model.embed_tokens.weight": "model-00001-of-00002.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors", + "model.norm.weight": "model-00002-of-00002.safetensors" + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/special_tokens_map.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/special_tokens_map.json @@ -0,0 +1,23 @@ +{ + "bos_token": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/tokenizer_config.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..8991b49e9c2a43fc527dab9e09ad8171f0cc5943 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/tokenizer_config.json @@ -0,0 +1,2086 @@ +{ + "added_tokens_decoder": { + "128000": { + "content": "<|begin_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128001": { + "content": "<|end_of_text|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128002": { + "content": "<|reserved_special_token_0|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128003": { + "content": "<|reserved_special_token_1|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128004": { + "content": "<|finetune_right_pad_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128005": { + "content": "<|reserved_special_token_2|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128006": { + "content": "<|start_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128007": { + "content": "<|end_header_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128008": { + "content": "<|eom_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128009": { + "content": "<|eot_id|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128010": { + "content": "<|python_tag|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128011": { + "content": "<|reserved_special_token_3|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128012": { + "content": "<|reserved_special_token_4|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128013": { + "content": "<|reserved_special_token_5|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128014": { + "content": "<|reserved_special_token_6|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128015": { + "content": "<|reserved_special_token_7|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128016": { + "content": "<|reserved_special_token_8|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128017": { + "content": "<|reserved_special_token_9|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128018": { + "content": "<|reserved_special_token_10|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128019": { + "content": "<|reserved_special_token_11|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128020": { + "content": "<|reserved_special_token_12|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128021": { + "content": "<|reserved_special_token_13|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128022": { + "content": "<|reserved_special_token_14|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128023": { + "content": "<|reserved_special_token_15|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128024": { + "content": "<|reserved_special_token_16|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128025": { + "content": "<|reserved_special_token_17|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128026": { + "content": "<|reserved_special_token_18|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128027": { + "content": "<|reserved_special_token_19|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128028": { + "content": "<|reserved_special_token_20|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128029": { + "content": "<|reserved_special_token_21|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128030": { + "content": "<|reserved_special_token_22|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128031": { + "content": "<|reserved_special_token_23|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128032": { + "content": "<|reserved_special_token_24|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128033": { + "content": "<|reserved_special_token_25|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128034": { + "content": "<|reserved_special_token_26|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128035": { + "content": "<|reserved_special_token_27|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128036": { + "content": "<|reserved_special_token_28|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128037": { + "content": "<|reserved_special_token_29|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128038": { + "content": "<|reserved_special_token_30|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128039": { + "content": "<|reserved_special_token_31|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128040": { + "content": "<|reserved_special_token_32|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128041": { + "content": "<|reserved_special_token_33|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128042": { + "content": "<|reserved_special_token_34|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128043": { + "content": "<|reserved_special_token_35|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128044": { + "content": "<|reserved_special_token_36|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128045": { + "content": "<|reserved_special_token_37|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128046": { + "content": "<|reserved_special_token_38|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128047": { + "content": "<|reserved_special_token_39|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128048": { + "content": "<|reserved_special_token_40|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128049": { + "content": "<|reserved_special_token_41|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128050": { + "content": "<|reserved_special_token_42|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128051": { + "content": "<|reserved_special_token_43|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128052": { + "content": "<|reserved_special_token_44|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128053": { + "content": "<|reserved_special_token_45|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128054": { + "content": "<|reserved_special_token_46|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128055": { + "content": "<|reserved_special_token_47|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128056": { + "content": "<|reserved_special_token_48|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128057": { + "content": "<|reserved_special_token_49|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128058": { + "content": "<|reserved_special_token_50|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128059": { + "content": "<|reserved_special_token_51|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128060": { + "content": "<|reserved_special_token_52|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128061": { + "content": "<|reserved_special_token_53|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128062": { + "content": "<|reserved_special_token_54|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128063": { + "content": "<|reserved_special_token_55|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128064": { + "content": "<|reserved_special_token_56|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128065": { + "content": "<|reserved_special_token_57|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128066": { + "content": "<|reserved_special_token_58|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128067": { + "content": "<|reserved_special_token_59|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128068": { + "content": "<|reserved_special_token_60|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128069": { + "content": "<|reserved_special_token_61|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128070": { + "content": "<|reserved_special_token_62|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128071": { + "content": "<|reserved_special_token_63|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128072": { + "content": "<|reserved_special_token_64|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128073": { + "content": "<|reserved_special_token_65|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128074": { + "content": "<|reserved_special_token_66|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128075": { + "content": "<|reserved_special_token_67|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128076": { + "content": "<|reserved_special_token_68|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128077": { + "content": "<|reserved_special_token_69|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128078": { + "content": "<|reserved_special_token_70|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128079": { + "content": "<|reserved_special_token_71|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128080": { + "content": "<|reserved_special_token_72|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128081": { + "content": "<|reserved_special_token_73|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128082": { + "content": "<|reserved_special_token_74|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128083": { + "content": "<|reserved_special_token_75|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128084": { + "content": "<|reserved_special_token_76|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128085": { + "content": "<|reserved_special_token_77|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128086": { + "content": "<|reserved_special_token_78|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128087": { + "content": "<|reserved_special_token_79|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128088": { + "content": "<|reserved_special_token_80|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128089": { + "content": "<|reserved_special_token_81|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128090": { + "content": "<|reserved_special_token_82|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128091": { + "content": "<|reserved_special_token_83|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128092": { + "content": "<|reserved_special_token_84|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128093": { + "content": "<|reserved_special_token_85|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128094": { + "content": "<|reserved_special_token_86|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128095": { + "content": "<|reserved_special_token_87|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128096": { + "content": "<|reserved_special_token_88|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128097": { + "content": "<|reserved_special_token_89|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128098": { + "content": "<|reserved_special_token_90|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128099": { + "content": "<|reserved_special_token_91|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128100": { + "content": "<|reserved_special_token_92|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128101": { + "content": "<|reserved_special_token_93|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128102": { + "content": "<|reserved_special_token_94|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128103": { + "content": "<|reserved_special_token_95|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128104": { + "content": "<|reserved_special_token_96|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128105": { + "content": "<|reserved_special_token_97|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128106": { + "content": "<|reserved_special_token_98|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128107": { + "content": "<|reserved_special_token_99|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128108": { + "content": "<|reserved_special_token_100|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128109": { + "content": "<|reserved_special_token_101|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128110": { + "content": "<|reserved_special_token_102|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128111": { + "content": "<|reserved_special_token_103|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128112": { + "content": "<|reserved_special_token_104|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128113": { + "content": "<|reserved_special_token_105|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128114": { + "content": "<|reserved_special_token_106|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128115": { + "content": "<|reserved_special_token_107|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128116": { + "content": "<|reserved_special_token_108|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128117": { + "content": "<|reserved_special_token_109|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128118": { + "content": "<|reserved_special_token_110|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128119": { + "content": "<|reserved_special_token_111|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128120": { + "content": "<|reserved_special_token_112|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128121": { + "content": "<|reserved_special_token_113|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128122": { + "content": "<|reserved_special_token_114|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128123": { + "content": "<|reserved_special_token_115|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128124": { + "content": "<|reserved_special_token_116|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128125": { + "content": "<|reserved_special_token_117|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128126": { + "content": "<|reserved_special_token_118|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128127": { + "content": "<|reserved_special_token_119|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128128": { + "content": "<|reserved_special_token_120|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128129": { + "content": "<|reserved_special_token_121|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128130": { + "content": "<|reserved_special_token_122|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128131": { + "content": "<|reserved_special_token_123|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128132": { + "content": "<|reserved_special_token_124|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128133": { + "content": "<|reserved_special_token_125|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128134": { + "content": "<|reserved_special_token_126|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128135": { + "content": "<|reserved_special_token_127|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128136": { + "content": "<|reserved_special_token_128|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128137": { + "content": "<|reserved_special_token_129|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128138": { + "content": "<|reserved_special_token_130|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128139": { + "content": "<|reserved_special_token_131|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128140": { + "content": "<|reserved_special_token_132|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128141": { + "content": "<|reserved_special_token_133|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128142": { + "content": "<|reserved_special_token_134|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128143": { + "content": "<|reserved_special_token_135|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128144": { + "content": "<|reserved_special_token_136|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128145": { + "content": "<|reserved_special_token_137|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128146": { + "content": "<|reserved_special_token_138|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128147": { + "content": "<|reserved_special_token_139|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128148": { + "content": "<|reserved_special_token_140|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128149": { + "content": "<|reserved_special_token_141|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128150": { + "content": "<|reserved_special_token_142|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128151": { + "content": "<|reserved_special_token_143|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128152": { + "content": "<|reserved_special_token_144|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128153": { + "content": "<|reserved_special_token_145|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128154": { + "content": "<|reserved_special_token_146|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128155": { + "content": "<|reserved_special_token_147|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128156": { + "content": "<|reserved_special_token_148|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128157": { + "content": "<|reserved_special_token_149|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128158": { + "content": "<|reserved_special_token_150|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128159": { + "content": "<|reserved_special_token_151|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128160": { + "content": "<|reserved_special_token_152|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128161": { + "content": "<|reserved_special_token_153|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128162": { + "content": "<|reserved_special_token_154|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128163": { + "content": "<|reserved_special_token_155|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128164": { + "content": "<|reserved_special_token_156|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128165": { + "content": "<|reserved_special_token_157|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128166": { + "content": "<|reserved_special_token_158|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128167": { + "content": "<|reserved_special_token_159|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128168": { + "content": "<|reserved_special_token_160|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128169": { + "content": "<|reserved_special_token_161|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128170": { + "content": "<|reserved_special_token_162|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128171": { + "content": "<|reserved_special_token_163|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128172": { + "content": "<|reserved_special_token_164|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128173": { + "content": "<|reserved_special_token_165|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128174": { + "content": "<|reserved_special_token_166|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128175": { + "content": "<|reserved_special_token_167|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128176": { + "content": "<|reserved_special_token_168|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128177": { + "content": "<|reserved_special_token_169|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128178": { + "content": "<|reserved_special_token_170|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128179": { + "content": "<|reserved_special_token_171|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128180": { + "content": "<|reserved_special_token_172|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128181": { + "content": "<|reserved_special_token_173|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128182": { + "content": "<|reserved_special_token_174|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128183": { + "content": "<|reserved_special_token_175|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128184": { + "content": "<|reserved_special_token_176|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128185": { + "content": "<|reserved_special_token_177|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128186": { + "content": "<|reserved_special_token_178|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128187": { + "content": "<|reserved_special_token_179|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128188": { + "content": "<|reserved_special_token_180|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128189": { + "content": "<|reserved_special_token_181|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128190": { + "content": "<|reserved_special_token_182|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128191": { + "content": "<|reserved_special_token_183|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128192": { + "content": "<|reserved_special_token_184|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128193": { + "content": "<|reserved_special_token_185|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128194": { + "content": "<|reserved_special_token_186|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128195": { + "content": "<|reserved_special_token_187|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128196": { + "content": "<|reserved_special_token_188|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128197": { + "content": "<|reserved_special_token_189|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128198": { + "content": "<|reserved_special_token_190|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128199": { + "content": "<|reserved_special_token_191|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128200": { + "content": "<|reserved_special_token_192|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128201": { + "content": "<|reserved_special_token_193|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128202": { + "content": "<|reserved_special_token_194|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128203": { + "content": "<|reserved_special_token_195|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128204": { + "content": "<|reserved_special_token_196|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128205": { + "content": "<|reserved_special_token_197|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128206": { + "content": "<|reserved_special_token_198|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128207": { + "content": "<|reserved_special_token_199|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128208": { + "content": "<|reserved_special_token_200|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128209": { + "content": "<|reserved_special_token_201|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128210": { + "content": "<|reserved_special_token_202|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128211": { + "content": "<|reserved_special_token_203|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128212": { + "content": "<|reserved_special_token_204|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128213": { + "content": "<|reserved_special_token_205|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128214": { + "content": "<|reserved_special_token_206|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128215": { + "content": "<|reserved_special_token_207|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128216": { + "content": "<|reserved_special_token_208|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128217": { + "content": "<|reserved_special_token_209|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128218": { + "content": "<|reserved_special_token_210|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128219": { + "content": "<|reserved_special_token_211|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128220": { + "content": "<|reserved_special_token_212|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128221": { + "content": "<|reserved_special_token_213|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128222": { + "content": "<|reserved_special_token_214|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128223": { + "content": "<|reserved_special_token_215|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128224": { + "content": "<|reserved_special_token_216|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128225": { + "content": "<|reserved_special_token_217|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128226": { + "content": "<|reserved_special_token_218|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128227": { + "content": "<|reserved_special_token_219|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128228": { + "content": "<|reserved_special_token_220|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128229": { + "content": "<|reserved_special_token_221|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128230": { + "content": "<|reserved_special_token_222|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128231": { + "content": "<|reserved_special_token_223|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128232": { + "content": "<|reserved_special_token_224|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128233": { + "content": "<|reserved_special_token_225|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128234": { + "content": "<|reserved_special_token_226|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128235": { + "content": "<|reserved_special_token_227|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128236": { + "content": "<|reserved_special_token_228|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128237": { + "content": "<|reserved_special_token_229|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128238": { + "content": "<|reserved_special_token_230|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128239": { + "content": "<|reserved_special_token_231|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128240": { + "content": "<|reserved_special_token_232|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128241": { + "content": "<|reserved_special_token_233|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128242": { + "content": "<|reserved_special_token_234|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128243": { + "content": "<|reserved_special_token_235|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128244": { + "content": "<|reserved_special_token_236|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128245": { + "content": "<|reserved_special_token_237|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128246": { + "content": "<|reserved_special_token_238|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128247": { + "content": "<|reserved_special_token_239|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128248": { + "content": "<|reserved_special_token_240|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128249": { + "content": "<|reserved_special_token_241|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128250": { + "content": "<|reserved_special_token_242|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128251": { + "content": "<|reserved_special_token_243|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128252": { + "content": "<|reserved_special_token_244|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128253": { + "content": "<|reserved_special_token_245|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128254": { + "content": "<|reserved_special_token_246|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128255": { + "content": "<|reserved_special_token_247|>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128256": { + "content": "[PAD]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "128257": { + "content": "🅂", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + }, + "128258": { + "content": "🄿", + "lstrip": true, + "normalized": true, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "bos_token": "<|begin_of_text|>", + "clean_up_tokenization_spaces": true, + "eos_token": "<|end_of_text|>", + "model_input_names": [ + "input_ids", + "attention_mask" + ], + "model_max_length": 131072, + "pad_token": "[PAD]", + "tokenizer_class": "PreTrainedTokenizerFast" +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/trainer_state.json b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..225fae439c26d14b6d70075e00004fc7f6267504 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/trainer_state.json @@ -0,0 +1,5493 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.8838383838383839, + "eval_steps": 10, + "global_step": 700, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0012626262626262627, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8571, + "step": 1 + }, + { + "epoch": 0.0025252525252525255, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7758, + "step": 2 + }, + { + "epoch": 0.003787878787878788, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8383, + "step": 3 + }, + { + "epoch": 0.005050505050505051, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9088, + "step": 4 + }, + { + "epoch": 0.006313131313131313, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8748, + "step": 5 + }, + { + "epoch": 0.007575757575757576, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8575, + "step": 6 + }, + { + "epoch": 0.008838383838383838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8271, + "step": 7 + }, + { + "epoch": 0.010101010101010102, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8336, + "step": 8 + }, + { + "epoch": 0.011363636363636364, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8037, + "step": 9 + }, + { + "epoch": 0.012626262626262626, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8323, + "step": 10 + }, + { + "epoch": 0.012626262626262626, + "eval_loss": 2.8736937046051025, + "eval_runtime": 37.7802, + "eval_samples_per_second": 26.469, + "eval_steps_per_second": 1.112, + "step": 10 + }, + { + "epoch": 0.013888888888888888, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8653, + "step": 11 + }, + { + "epoch": 0.015151515151515152, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8104, + "step": 12 + }, + { + "epoch": 0.016414141414141416, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8489, + "step": 13 + }, + { + "epoch": 0.017676767676767676, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8442, + "step": 14 + }, + { + "epoch": 0.01893939393939394, + "grad_norm": 5.356770992279053, + "learning_rate": 2.1008403361344538e-08, + "loss": 2.8207, + "step": 15 + }, + { + "epoch": 0.020202020202020204, + "grad_norm": 4.954462051391602, + "learning_rate": 4.2016806722689076e-08, + "loss": 2.8104, + "step": 16 + }, + { + "epoch": 0.021464646464646464, + "grad_norm": 5.248059272766113, + "learning_rate": 6.302521008403361e-08, + "loss": 2.8178, + "step": 17 + }, + { + "epoch": 0.022727272727272728, + "grad_norm": 5.277204513549805, + "learning_rate": 8.403361344537815e-08, + "loss": 2.8255, + "step": 18 + }, + { + "epoch": 0.023989898989898988, + "grad_norm": 5.344228267669678, + "learning_rate": 1.050420168067227e-07, + "loss": 2.8018, + "step": 19 + }, + { + "epoch": 0.025252525252525252, + "grad_norm": 5.004310131072998, + "learning_rate": 1.2605042016806723e-07, + "loss": 2.8337, + "step": 20 + }, + { + "epoch": 0.025252525252525252, + "eval_loss": 2.872413158416748, + "eval_runtime": 37.7426, + "eval_samples_per_second": 26.495, + "eval_steps_per_second": 1.113, + "step": 20 + }, + { + "epoch": 0.026515151515151516, + "grad_norm": 5.209287166595459, + "learning_rate": 1.4705882352941178e-07, + "loss": 2.8246, + "step": 21 + }, + { + "epoch": 0.027777777777777776, + "grad_norm": 5.369327545166016, + "learning_rate": 1.680672268907563e-07, + "loss": 2.8791, + "step": 22 + }, + { + "epoch": 0.02904040404040404, + "grad_norm": 4.646446228027344, + "learning_rate": 1.8907563025210085e-07, + "loss": 2.8425, + "step": 23 + }, + { + "epoch": 0.030303030303030304, + "grad_norm": 4.949256420135498, + "learning_rate": 2.100840336134454e-07, + "loss": 2.8629, + "step": 24 + }, + { + "epoch": 0.03156565656565657, + "grad_norm": 4.709571838378906, + "learning_rate": 2.3109243697478996e-07, + "loss": 2.8264, + "step": 25 + }, + { + "epoch": 0.03282828282828283, + "grad_norm": 4.519557952880859, + "learning_rate": 2.5210084033613445e-07, + "loss": 2.8143, + "step": 26 + }, + { + "epoch": 0.03409090909090909, + "grad_norm": 4.445594310760498, + "learning_rate": 2.73109243697479e-07, + "loss": 2.8031, + "step": 27 + }, + { + "epoch": 0.03535353535353535, + "grad_norm": 4.26489782333374, + "learning_rate": 2.9411764705882356e-07, + "loss": 2.7719, + "step": 28 + }, + { + "epoch": 0.036616161616161616, + "grad_norm": 4.153639793395996, + "learning_rate": 3.151260504201681e-07, + "loss": 2.8251, + "step": 29 + }, + { + "epoch": 0.03787878787878788, + "grad_norm": 3.40531325340271, + "learning_rate": 3.361344537815126e-07, + "loss": 2.849, + "step": 30 + }, + { + "epoch": 0.03787878787878788, + "eval_loss": 2.825382709503174, + "eval_runtime": 37.0715, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 30 + }, + { + "epoch": 0.039141414141414144, + "grad_norm": 3.4759507179260254, + "learning_rate": 3.5714285714285716e-07, + "loss": 2.7653, + "step": 31 + }, + { + "epoch": 0.04040404040404041, + "grad_norm": 3.4147374629974365, + "learning_rate": 3.781512605042017e-07, + "loss": 2.7916, + "step": 32 + }, + { + "epoch": 0.041666666666666664, + "grad_norm": 3.2775769233703613, + "learning_rate": 3.991596638655462e-07, + "loss": 2.8072, + "step": 33 + }, + { + "epoch": 0.04292929292929293, + "grad_norm": 3.3541715145111084, + "learning_rate": 4.201680672268908e-07, + "loss": 2.7944, + "step": 34 + }, + { + "epoch": 0.04419191919191919, + "grad_norm": 2.991971969604492, + "learning_rate": 4.4117647058823536e-07, + "loss": 2.7827, + "step": 35 + }, + { + "epoch": 0.045454545454545456, + "grad_norm": 2.9127862453460693, + "learning_rate": 4.621848739495799e-07, + "loss": 2.7332, + "step": 36 + }, + { + "epoch": 0.04671717171717172, + "grad_norm": 2.839707136154175, + "learning_rate": 4.831932773109245e-07, + "loss": 2.7568, + "step": 37 + }, + { + "epoch": 0.047979797979797977, + "grad_norm": 2.9395432472229004, + "learning_rate": 5.042016806722689e-07, + "loss": 2.7445, + "step": 38 + }, + { + "epoch": 0.04924242424242424, + "grad_norm": 2.869896650314331, + "learning_rate": 5.252100840336135e-07, + "loss": 2.7157, + "step": 39 + }, + { + "epoch": 0.050505050505050504, + "grad_norm": 2.9064571857452393, + "learning_rate": 5.46218487394958e-07, + "loss": 2.7049, + "step": 40 + }, + { + "epoch": 0.050505050505050504, + "eval_loss": 2.7477543354034424, + "eval_runtime": 37.0669, + "eval_samples_per_second": 26.978, + "eval_steps_per_second": 1.133, + "step": 40 + }, + { + "epoch": 0.05176767676767677, + "grad_norm": 2.6264421939849854, + "learning_rate": 5.672268907563026e-07, + "loss": 2.7651, + "step": 41 + }, + { + "epoch": 0.05303030303030303, + "grad_norm": 2.518627405166626, + "learning_rate": 5.882352941176471e-07, + "loss": 2.7241, + "step": 42 + }, + { + "epoch": 0.054292929292929296, + "grad_norm": 2.521034002304077, + "learning_rate": 6.092436974789917e-07, + "loss": 2.6691, + "step": 43 + }, + { + "epoch": 0.05555555555555555, + "grad_norm": 2.4679362773895264, + "learning_rate": 6.302521008403362e-07, + "loss": 2.743, + "step": 44 + }, + { + "epoch": 0.056818181818181816, + "grad_norm": 2.5817418098449707, + "learning_rate": 6.512605042016808e-07, + "loss": 2.7067, + "step": 45 + }, + { + "epoch": 0.05808080808080808, + "grad_norm": 2.3704075813293457, + "learning_rate": 6.722689075630252e-07, + "loss": 2.6495, + "step": 46 + }, + { + "epoch": 0.059343434343434344, + "grad_norm": 2.2321906089782715, + "learning_rate": 6.932773109243699e-07, + "loss": 2.6606, + "step": 47 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.985407829284668, + "learning_rate": 7.142857142857143e-07, + "loss": 2.662, + "step": 48 + }, + { + "epoch": 0.06186868686868687, + "grad_norm": 1.945688247680664, + "learning_rate": 7.352941176470589e-07, + "loss": 2.6319, + "step": 49 + }, + { + "epoch": 0.06313131313131314, + "grad_norm": 1.9392870664596558, + "learning_rate": 7.563025210084034e-07, + "loss": 2.6094, + "step": 50 + }, + { + "epoch": 0.06313131313131314, + "eval_loss": 2.6554172039031982, + "eval_runtime": 37.0569, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 50 + }, + { + "epoch": 0.06439393939393939, + "grad_norm": 1.9975577592849731, + "learning_rate": 7.77310924369748e-07, + "loss": 2.6045, + "step": 51 + }, + { + "epoch": 0.06565656565656566, + "grad_norm": 1.9733693599700928, + "learning_rate": 7.983193277310924e-07, + "loss": 2.649, + "step": 52 + }, + { + "epoch": 0.06691919191919192, + "grad_norm": 1.7324917316436768, + "learning_rate": 8.193277310924371e-07, + "loss": 2.5689, + "step": 53 + }, + { + "epoch": 0.06818181818181818, + "grad_norm": 1.7553645372390747, + "learning_rate": 8.403361344537816e-07, + "loss": 2.644, + "step": 54 + }, + { + "epoch": 0.06944444444444445, + "grad_norm": 1.8163801431655884, + "learning_rate": 8.613445378151261e-07, + "loss": 2.4943, + "step": 55 + }, + { + "epoch": 0.0707070707070707, + "grad_norm": 1.7717015743255615, + "learning_rate": 8.823529411764707e-07, + "loss": 2.6093, + "step": 56 + }, + { + "epoch": 0.07196969696969698, + "grad_norm": 1.7013663053512573, + "learning_rate": 9.033613445378152e-07, + "loss": 2.6376, + "step": 57 + }, + { + "epoch": 0.07323232323232323, + "grad_norm": 1.752943992614746, + "learning_rate": 9.243697478991598e-07, + "loss": 2.5172, + "step": 58 + }, + { + "epoch": 0.07449494949494949, + "grad_norm": 1.5362415313720703, + "learning_rate": 9.453781512605043e-07, + "loss": 2.6129, + "step": 59 + }, + { + "epoch": 0.07575757575757576, + "grad_norm": 1.514885425567627, + "learning_rate": 9.66386554621849e-07, + "loss": 2.6064, + "step": 60 + }, + { + "epoch": 0.07575757575757576, + "eval_loss": 2.5934340953826904, + "eval_runtime": 37.0782, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 60 + }, + { + "epoch": 0.07702020202020202, + "grad_norm": 1.5915579795837402, + "learning_rate": 9.873949579831934e-07, + "loss": 2.5491, + "step": 61 + }, + { + "epoch": 0.07828282828282829, + "grad_norm": 1.601773977279663, + "learning_rate": 1.0084033613445378e-06, + "loss": 2.5685, + "step": 62 + }, + { + "epoch": 0.07954545454545454, + "grad_norm": 1.8327804803848267, + "learning_rate": 1.0294117647058825e-06, + "loss": 2.5763, + "step": 63 + }, + { + "epoch": 0.08080808080808081, + "grad_norm": 1.5276741981506348, + "learning_rate": 1.050420168067227e-06, + "loss": 2.5494, + "step": 64 + }, + { + "epoch": 0.08207070707070707, + "grad_norm": 1.7065422534942627, + "learning_rate": 1.0714285714285714e-06, + "loss": 2.449, + "step": 65 + }, + { + "epoch": 0.08333333333333333, + "grad_norm": 2.0551910400390625, + "learning_rate": 1.092436974789916e-06, + "loss": 2.4371, + "step": 66 + }, + { + "epoch": 0.0845959595959596, + "grad_norm": 2.154855728149414, + "learning_rate": 1.1134453781512607e-06, + "loss": 2.5417, + "step": 67 + }, + { + "epoch": 0.08585858585858586, + "grad_norm": 1.7922390699386597, + "learning_rate": 1.1344537815126051e-06, + "loss": 2.4754, + "step": 68 + }, + { + "epoch": 0.08712121212121213, + "grad_norm": 1.745632290840149, + "learning_rate": 1.1554621848739498e-06, + "loss": 2.4669, + "step": 69 + }, + { + "epoch": 0.08838383838383838, + "grad_norm": 1.8907713890075684, + "learning_rate": 1.1764705882352942e-06, + "loss": 2.4943, + "step": 70 + }, + { + "epoch": 0.08838383838383838, + "eval_loss": 2.518237829208374, + "eval_runtime": 37.074, + "eval_samples_per_second": 26.973, + "eval_steps_per_second": 1.133, + "step": 70 + }, + { + "epoch": 0.08964646464646464, + "grad_norm": 1.8178164958953857, + "learning_rate": 1.1974789915966389e-06, + "loss": 2.4891, + "step": 71 + }, + { + "epoch": 0.09090909090909091, + "grad_norm": 1.970969796180725, + "learning_rate": 1.2184873949579833e-06, + "loss": 2.4797, + "step": 72 + }, + { + "epoch": 0.09217171717171717, + "grad_norm": 2.0211431980133057, + "learning_rate": 1.2394957983193278e-06, + "loss": 2.4716, + "step": 73 + }, + { + "epoch": 0.09343434343434344, + "grad_norm": 2.650787591934204, + "learning_rate": 1.2605042016806724e-06, + "loss": 2.4915, + "step": 74 + }, + { + "epoch": 0.0946969696969697, + "grad_norm": 2.235499143600464, + "learning_rate": 1.2815126050420169e-06, + "loss": 2.3917, + "step": 75 + }, + { + "epoch": 0.09595959595959595, + "grad_norm": 2.245999813079834, + "learning_rate": 1.3025210084033615e-06, + "loss": 2.4942, + "step": 76 + }, + { + "epoch": 0.09722222222222222, + "grad_norm": 2.4414420127868652, + "learning_rate": 1.323529411764706e-06, + "loss": 2.4004, + "step": 77 + }, + { + "epoch": 0.09848484848484848, + "grad_norm": 2.279937982559204, + "learning_rate": 1.3445378151260504e-06, + "loss": 2.4043, + "step": 78 + }, + { + "epoch": 0.09974747474747475, + "grad_norm": 2.3830068111419678, + "learning_rate": 1.3655462184873949e-06, + "loss": 2.4096, + "step": 79 + }, + { + "epoch": 0.10101010101010101, + "grad_norm": 2.0119733810424805, + "learning_rate": 1.3865546218487397e-06, + "loss": 2.4227, + "step": 80 + }, + { + "epoch": 0.10101010101010101, + "eval_loss": 2.4316186904907227, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 80 + }, + { + "epoch": 0.10227272727272728, + "grad_norm": 2.1273374557495117, + "learning_rate": 1.4075630252100842e-06, + "loss": 2.3541, + "step": 81 + }, + { + "epoch": 0.10353535353535354, + "grad_norm": 2.6852898597717285, + "learning_rate": 1.4285714285714286e-06, + "loss": 2.3321, + "step": 82 + }, + { + "epoch": 0.10479797979797979, + "grad_norm": 2.2622721195220947, + "learning_rate": 1.4495798319327733e-06, + "loss": 2.3166, + "step": 83 + }, + { + "epoch": 0.10606060606060606, + "grad_norm": 2.98258900642395, + "learning_rate": 1.4705882352941177e-06, + "loss": 2.3733, + "step": 84 + }, + { + "epoch": 0.10732323232323232, + "grad_norm": 2.5140459537506104, + "learning_rate": 1.4915966386554624e-06, + "loss": 2.314, + "step": 85 + }, + { + "epoch": 0.10858585858585859, + "grad_norm": 2.637608051300049, + "learning_rate": 1.5126050420168068e-06, + "loss": 2.3479, + "step": 86 + }, + { + "epoch": 0.10984848484848485, + "grad_norm": 2.399508476257324, + "learning_rate": 1.5336134453781515e-06, + "loss": 2.2932, + "step": 87 + }, + { + "epoch": 0.1111111111111111, + "grad_norm": 3.0835793018341064, + "learning_rate": 1.554621848739496e-06, + "loss": 2.2886, + "step": 88 + }, + { + "epoch": 0.11237373737373738, + "grad_norm": 3.288769483566284, + "learning_rate": 1.5756302521008404e-06, + "loss": 2.2914, + "step": 89 + }, + { + "epoch": 0.11363636363636363, + "grad_norm": 2.538361072540283, + "learning_rate": 1.5966386554621848e-06, + "loss": 2.3247, + "step": 90 + }, + { + "epoch": 0.11363636363636363, + "eval_loss": 2.3510072231292725, + "eval_runtime": 37.0847, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 90 + }, + { + "epoch": 0.1148989898989899, + "grad_norm": 2.8016724586486816, + "learning_rate": 1.6176470588235297e-06, + "loss": 2.281, + "step": 91 + }, + { + "epoch": 0.11616161616161616, + "grad_norm": 2.9251859188079834, + "learning_rate": 1.6386554621848741e-06, + "loss": 2.2715, + "step": 92 + }, + { + "epoch": 0.11742424242424243, + "grad_norm": 2.595750570297241, + "learning_rate": 1.6596638655462186e-06, + "loss": 2.2127, + "step": 93 + }, + { + "epoch": 0.11868686868686869, + "grad_norm": 2.6989378929138184, + "learning_rate": 1.6806722689075632e-06, + "loss": 2.2623, + "step": 94 + }, + { + "epoch": 0.11994949494949494, + "grad_norm": 2.554863929748535, + "learning_rate": 1.7016806722689077e-06, + "loss": 2.2649, + "step": 95 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.7013020515441895, + "learning_rate": 1.7226890756302521e-06, + "loss": 2.2428, + "step": 96 + }, + { + "epoch": 0.12247474747474747, + "grad_norm": 2.7415151596069336, + "learning_rate": 1.7436974789915968e-06, + "loss": 2.1882, + "step": 97 + }, + { + "epoch": 0.12373737373737374, + "grad_norm": 2.672806739807129, + "learning_rate": 1.7647058823529414e-06, + "loss": 2.2302, + "step": 98 + }, + { + "epoch": 0.125, + "grad_norm": 2.9508326053619385, + "learning_rate": 1.7857142857142859e-06, + "loss": 2.1894, + "step": 99 + }, + { + "epoch": 0.12626262626262627, + "grad_norm": 2.4668166637420654, + "learning_rate": 1.8067226890756303e-06, + "loss": 2.2043, + "step": 100 + }, + { + "epoch": 0.12626262626262627, + "eval_loss": 2.2788307666778564, + "eval_runtime": 37.1156, + "eval_samples_per_second": 26.943, + "eval_steps_per_second": 1.132, + "step": 100 + }, + { + "epoch": 0.1275252525252525, + "grad_norm": 2.8241467475891113, + "learning_rate": 1.8277310924369748e-06, + "loss": 2.2109, + "step": 101 + }, + { + "epoch": 0.12878787878787878, + "grad_norm": 2.4179229736328125, + "learning_rate": 1.8487394957983196e-06, + "loss": 2.2193, + "step": 102 + }, + { + "epoch": 0.13005050505050506, + "grad_norm": 2.714315891265869, + "learning_rate": 1.869747899159664e-06, + "loss": 2.2474, + "step": 103 + }, + { + "epoch": 0.13131313131313133, + "grad_norm": 2.8560359477996826, + "learning_rate": 1.8907563025210085e-06, + "loss": 2.2205, + "step": 104 + }, + { + "epoch": 0.13257575757575757, + "grad_norm": 2.7780051231384277, + "learning_rate": 1.9117647058823528e-06, + "loss": 2.1834, + "step": 105 + }, + { + "epoch": 0.13383838383838384, + "grad_norm": 3.3773467540740967, + "learning_rate": 1.932773109243698e-06, + "loss": 2.2063, + "step": 106 + }, + { + "epoch": 0.1351010101010101, + "grad_norm": 2.927551746368408, + "learning_rate": 1.953781512605042e-06, + "loss": 2.1592, + "step": 107 + }, + { + "epoch": 0.13636363636363635, + "grad_norm": 2.8094165325164795, + "learning_rate": 1.9747899159663867e-06, + "loss": 2.2619, + "step": 108 + }, + { + "epoch": 0.13762626262626262, + "grad_norm": 2.5312564373016357, + "learning_rate": 1.9957983193277314e-06, + "loss": 2.2323, + "step": 109 + }, + { + "epoch": 0.1388888888888889, + "grad_norm": 3.175746202468872, + "learning_rate": 2.0168067226890756e-06, + "loss": 2.1776, + "step": 110 + }, + { + "epoch": 0.1388888888888889, + "eval_loss": 2.2208564281463623, + "eval_runtime": 37.3885, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 110 + }, + { + "epoch": 0.14015151515151514, + "grad_norm": 2.660886287689209, + "learning_rate": 2.0378151260504203e-06, + "loss": 2.1626, + "step": 111 + }, + { + "epoch": 0.1414141414141414, + "grad_norm": 2.6948511600494385, + "learning_rate": 2.058823529411765e-06, + "loss": 2.1731, + "step": 112 + }, + { + "epoch": 0.14267676767676768, + "grad_norm": 2.826976776123047, + "learning_rate": 2.0798319327731096e-06, + "loss": 2.1865, + "step": 113 + }, + { + "epoch": 0.14393939393939395, + "grad_norm": 3.210033655166626, + "learning_rate": 2.100840336134454e-06, + "loss": 2.1325, + "step": 114 + }, + { + "epoch": 0.1452020202020202, + "grad_norm": 2.509060859680176, + "learning_rate": 2.1218487394957985e-06, + "loss": 2.0926, + "step": 115 + }, + { + "epoch": 0.14646464646464646, + "grad_norm": 3.51330304145813, + "learning_rate": 2.1428571428571427e-06, + "loss": 2.1577, + "step": 116 + }, + { + "epoch": 0.14772727272727273, + "grad_norm": 2.468395471572876, + "learning_rate": 2.163865546218488e-06, + "loss": 2.1, + "step": 117 + }, + { + "epoch": 0.14898989898989898, + "grad_norm": 3.081529378890991, + "learning_rate": 2.184873949579832e-06, + "loss": 2.1082, + "step": 118 + }, + { + "epoch": 0.15025252525252525, + "grad_norm": 2.8856358528137207, + "learning_rate": 2.2058823529411767e-06, + "loss": 2.1288, + "step": 119 + }, + { + "epoch": 0.15151515151515152, + "grad_norm": 3.323089838027954, + "learning_rate": 2.2268907563025214e-06, + "loss": 2.0933, + "step": 120 + }, + { + "epoch": 0.15151515151515152, + "eval_loss": 2.1834487915039062, + "eval_runtime": 37.4344, + "eval_samples_per_second": 26.713, + "eval_steps_per_second": 1.122, + "step": 120 + }, + { + "epoch": 0.1527777777777778, + "grad_norm": 2.8871560096740723, + "learning_rate": 2.2478991596638656e-06, + "loss": 2.1519, + "step": 121 + }, + { + "epoch": 0.15404040404040403, + "grad_norm": 3.0718588829040527, + "learning_rate": 2.2689075630252102e-06, + "loss": 2.0321, + "step": 122 + }, + { + "epoch": 0.1553030303030303, + "grad_norm": 2.820739507675171, + "learning_rate": 2.289915966386555e-06, + "loss": 2.0944, + "step": 123 + }, + { + "epoch": 0.15656565656565657, + "grad_norm": 3.1259310245513916, + "learning_rate": 2.3109243697478996e-06, + "loss": 2.0141, + "step": 124 + }, + { + "epoch": 0.15782828282828282, + "grad_norm": 3.303004741668701, + "learning_rate": 2.331932773109244e-06, + "loss": 2.1095, + "step": 125 + }, + { + "epoch": 0.1590909090909091, + "grad_norm": 3.1838674545288086, + "learning_rate": 2.3529411764705885e-06, + "loss": 2.0727, + "step": 126 + }, + { + "epoch": 0.16035353535353536, + "grad_norm": 3.1013786792755127, + "learning_rate": 2.3739495798319327e-06, + "loss": 2.0963, + "step": 127 + }, + { + "epoch": 0.16161616161616163, + "grad_norm": 2.803283214569092, + "learning_rate": 2.3949579831932778e-06, + "loss": 2.0551, + "step": 128 + }, + { + "epoch": 0.16287878787878787, + "grad_norm": 3.1616814136505127, + "learning_rate": 2.415966386554622e-06, + "loss": 2.0485, + "step": 129 + }, + { + "epoch": 0.16414141414141414, + "grad_norm": 3.333782196044922, + "learning_rate": 2.4369747899159667e-06, + "loss": 2.041, + "step": 130 + }, + { + "epoch": 0.16414141414141414, + "eval_loss": 2.150329351425171, + "eval_runtime": 37.494, + "eval_samples_per_second": 26.671, + "eval_steps_per_second": 1.12, + "step": 130 + }, + { + "epoch": 0.16540404040404041, + "grad_norm": 2.9714694023132324, + "learning_rate": 2.457983193277311e-06, + "loss": 2.0686, + "step": 131 + }, + { + "epoch": 0.16666666666666666, + "grad_norm": 3.216895818710327, + "learning_rate": 2.4789915966386555e-06, + "loss": 2.003, + "step": 132 + }, + { + "epoch": 0.16792929292929293, + "grad_norm": 2.9636809825897217, + "learning_rate": 2.5e-06, + "loss": 2.0534, + "step": 133 + }, + { + "epoch": 0.1691919191919192, + "grad_norm": 3.6477177143096924, + "learning_rate": 2.521008403361345e-06, + "loss": 2.077, + "step": 134 + }, + { + "epoch": 0.17045454545454544, + "grad_norm": 2.7437169551849365, + "learning_rate": 2.542016806722689e-06, + "loss": 2.0501, + "step": 135 + }, + { + "epoch": 0.1717171717171717, + "grad_norm": 3.2992663383483887, + "learning_rate": 2.5630252100840338e-06, + "loss": 2.0284, + "step": 136 + }, + { + "epoch": 0.17297979797979798, + "grad_norm": 3.0735950469970703, + "learning_rate": 2.5840336134453784e-06, + "loss": 2.025, + "step": 137 + }, + { + "epoch": 0.17424242424242425, + "grad_norm": 3.588418960571289, + "learning_rate": 2.605042016806723e-06, + "loss": 2.002, + "step": 138 + }, + { + "epoch": 0.1755050505050505, + "grad_norm": 2.8772335052490234, + "learning_rate": 2.6260504201680677e-06, + "loss": 1.9873, + "step": 139 + }, + { + "epoch": 0.17676767676767677, + "grad_norm": 2.9078969955444336, + "learning_rate": 2.647058823529412e-06, + "loss": 2.011, + "step": 140 + }, + { + "epoch": 0.17676767676767677, + "eval_loss": 2.1225531101226807, + "eval_runtime": 37.4531, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 140 + }, + { + "epoch": 0.17803030303030304, + "grad_norm": 3.556654214859009, + "learning_rate": 2.6680672268907566e-06, + "loss": 2.0401, + "step": 141 + }, + { + "epoch": 0.17929292929292928, + "grad_norm": 2.9375200271606445, + "learning_rate": 2.689075630252101e-06, + "loss": 2.0277, + "step": 142 + }, + { + "epoch": 0.18055555555555555, + "grad_norm": 3.4019014835357666, + "learning_rate": 2.7100840336134455e-06, + "loss": 1.975, + "step": 143 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 2.7653748989105225, + "learning_rate": 2.7310924369747897e-06, + "loss": 2.0102, + "step": 144 + }, + { + "epoch": 0.1830808080808081, + "grad_norm": 3.4416797161102295, + "learning_rate": 2.752100840336135e-06, + "loss": 2.0093, + "step": 145 + }, + { + "epoch": 0.18434343434343434, + "grad_norm": 3.2531726360321045, + "learning_rate": 2.7731092436974795e-06, + "loss": 1.9852, + "step": 146 + }, + { + "epoch": 0.1856060606060606, + "grad_norm": 3.171502113342285, + "learning_rate": 2.7941176470588237e-06, + "loss": 1.9165, + "step": 147 + }, + { + "epoch": 0.18686868686868688, + "grad_norm": 3.114550828933716, + "learning_rate": 2.8151260504201684e-06, + "loss": 1.9516, + "step": 148 + }, + { + "epoch": 0.18813131313131312, + "grad_norm": 3.3601152896881104, + "learning_rate": 2.8361344537815126e-06, + "loss": 2.0183, + "step": 149 + }, + { + "epoch": 0.1893939393939394, + "grad_norm": 3.059622049331665, + "learning_rate": 2.8571428571428573e-06, + "loss": 1.9682, + "step": 150 + }, + { + "epoch": 0.1893939393939394, + "eval_loss": 2.0962483882904053, + "eval_runtime": 37.2364, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 150 + }, + { + "epoch": 0.19065656565656566, + "grad_norm": 2.8749427795410156, + "learning_rate": 2.8781512605042015e-06, + "loss": 2.0089, + "step": 151 + }, + { + "epoch": 0.1919191919191919, + "grad_norm": 3.067267894744873, + "learning_rate": 2.8991596638655466e-06, + "loss": 1.9661, + "step": 152 + }, + { + "epoch": 0.19318181818181818, + "grad_norm": 2.7126266956329346, + "learning_rate": 2.9201680672268912e-06, + "loss": 1.982, + "step": 153 + }, + { + "epoch": 0.19444444444444445, + "grad_norm": 3.1330652236938477, + "learning_rate": 2.9411764705882355e-06, + "loss": 1.9753, + "step": 154 + }, + { + "epoch": 0.19570707070707072, + "grad_norm": 3.158313274383545, + "learning_rate": 2.96218487394958e-06, + "loss": 1.9431, + "step": 155 + }, + { + "epoch": 0.19696969696969696, + "grad_norm": 3.30806303024292, + "learning_rate": 2.9831932773109248e-06, + "loss": 1.9812, + "step": 156 + }, + { + "epoch": 0.19823232323232323, + "grad_norm": 3.0344884395599365, + "learning_rate": 3.004201680672269e-06, + "loss": 1.9677, + "step": 157 + }, + { + "epoch": 0.1994949494949495, + "grad_norm": 2.9032909870147705, + "learning_rate": 3.0252100840336137e-06, + "loss": 1.9132, + "step": 158 + }, + { + "epoch": 0.20075757575757575, + "grad_norm": 3.459934711456299, + "learning_rate": 3.046218487394958e-06, + "loss": 1.9581, + "step": 159 + }, + { + "epoch": 0.20202020202020202, + "grad_norm": 2.9455478191375732, + "learning_rate": 3.067226890756303e-06, + "loss": 1.9509, + "step": 160 + }, + { + "epoch": 0.20202020202020202, + "eval_loss": 2.074032783508301, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 160 + }, + { + "epoch": 0.2032828282828283, + "grad_norm": 3.2347211837768555, + "learning_rate": 3.0882352941176476e-06, + "loss": 1.9743, + "step": 161 + }, + { + "epoch": 0.20454545454545456, + "grad_norm": 3.3152947425842285, + "learning_rate": 3.109243697478992e-06, + "loss": 1.8971, + "step": 162 + }, + { + "epoch": 0.2058080808080808, + "grad_norm": 3.035548210144043, + "learning_rate": 3.1302521008403365e-06, + "loss": 1.8762, + "step": 163 + }, + { + "epoch": 0.20707070707070707, + "grad_norm": 3.1004066467285156, + "learning_rate": 3.1512605042016808e-06, + "loss": 1.9512, + "step": 164 + }, + { + "epoch": 0.20833333333333334, + "grad_norm": 3.2280282974243164, + "learning_rate": 3.1722689075630254e-06, + "loss": 1.8856, + "step": 165 + }, + { + "epoch": 0.20959595959595959, + "grad_norm": 3.021969795227051, + "learning_rate": 3.1932773109243696e-06, + "loss": 1.9127, + "step": 166 + }, + { + "epoch": 0.21085858585858586, + "grad_norm": 3.1647000312805176, + "learning_rate": 3.2142857142857147e-06, + "loss": 2.038, + "step": 167 + }, + { + "epoch": 0.21212121212121213, + "grad_norm": 3.0084872245788574, + "learning_rate": 3.2352941176470594e-06, + "loss": 1.9536, + "step": 168 + }, + { + "epoch": 0.21338383838383837, + "grad_norm": 2.951509952545166, + "learning_rate": 3.2563025210084036e-06, + "loss": 1.933, + "step": 169 + }, + { + "epoch": 0.21464646464646464, + "grad_norm": 3.472856283187866, + "learning_rate": 3.2773109243697483e-06, + "loss": 1.934, + "step": 170 + }, + { + "epoch": 0.21464646464646464, + "eval_loss": 2.0450048446655273, + "eval_runtime": 37.1669, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 170 + }, + { + "epoch": 0.2159090909090909, + "grad_norm": 3.1104896068573, + "learning_rate": 3.2983193277310925e-06, + "loss": 1.9008, + "step": 171 + }, + { + "epoch": 0.21717171717171718, + "grad_norm": 2.847781181335449, + "learning_rate": 3.319327731092437e-06, + "loss": 1.9592, + "step": 172 + }, + { + "epoch": 0.21843434343434343, + "grad_norm": 3.384554624557495, + "learning_rate": 3.3403361344537814e-06, + "loss": 1.9087, + "step": 173 + }, + { + "epoch": 0.2196969696969697, + "grad_norm": 3.221564292907715, + "learning_rate": 3.3613445378151265e-06, + "loss": 1.9107, + "step": 174 + }, + { + "epoch": 0.22095959595959597, + "grad_norm": 3.30204439163208, + "learning_rate": 3.382352941176471e-06, + "loss": 1.8879, + "step": 175 + }, + { + "epoch": 0.2222222222222222, + "grad_norm": 3.549259662628174, + "learning_rate": 3.4033613445378154e-06, + "loss": 1.9592, + "step": 176 + }, + { + "epoch": 0.22348484848484848, + "grad_norm": 3.498199462890625, + "learning_rate": 3.42436974789916e-06, + "loss": 1.984, + "step": 177 + }, + { + "epoch": 0.22474747474747475, + "grad_norm": 3.576317548751831, + "learning_rate": 3.4453781512605043e-06, + "loss": 1.8897, + "step": 178 + }, + { + "epoch": 0.22601010101010102, + "grad_norm": 3.5337026119232178, + "learning_rate": 3.466386554621849e-06, + "loss": 1.9607, + "step": 179 + }, + { + "epoch": 0.22727272727272727, + "grad_norm": 3.342886209487915, + "learning_rate": 3.4873949579831936e-06, + "loss": 1.8797, + "step": 180 + }, + { + "epoch": 0.22727272727272727, + "eval_loss": 2.02158522605896, + "eval_runtime": 37.1382, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 180 + }, + { + "epoch": 0.22853535353535354, + "grad_norm": 2.941906213760376, + "learning_rate": 3.508403361344538e-06, + "loss": 1.9, + "step": 181 + }, + { + "epoch": 0.2297979797979798, + "grad_norm": 3.1913063526153564, + "learning_rate": 3.529411764705883e-06, + "loss": 1.8704, + "step": 182 + }, + { + "epoch": 0.23106060606060605, + "grad_norm": 3.0619900226593018, + "learning_rate": 3.550420168067227e-06, + "loss": 1.8851, + "step": 183 + }, + { + "epoch": 0.23232323232323232, + "grad_norm": 3.051288604736328, + "learning_rate": 3.5714285714285718e-06, + "loss": 1.9422, + "step": 184 + }, + { + "epoch": 0.2335858585858586, + "grad_norm": 2.999845266342163, + "learning_rate": 3.5924369747899164e-06, + "loss": 1.8757, + "step": 185 + }, + { + "epoch": 0.23484848484848486, + "grad_norm": 3.262355089187622, + "learning_rate": 3.6134453781512607e-06, + "loss": 1.9114, + "step": 186 + }, + { + "epoch": 0.2361111111111111, + "grad_norm": 3.0354340076446533, + "learning_rate": 3.6344537815126053e-06, + "loss": 1.8547, + "step": 187 + }, + { + "epoch": 0.23737373737373738, + "grad_norm": 2.9401369094848633, + "learning_rate": 3.6554621848739496e-06, + "loss": 1.8974, + "step": 188 + }, + { + "epoch": 0.23863636363636365, + "grad_norm": 3.0544443130493164, + "learning_rate": 3.6764705882352946e-06, + "loss": 1.8662, + "step": 189 + }, + { + "epoch": 0.2398989898989899, + "grad_norm": 2.7257750034332275, + "learning_rate": 3.6974789915966393e-06, + "loss": 1.8737, + "step": 190 + }, + { + "epoch": 0.2398989898989899, + "eval_loss": 2.0065128803253174, + "eval_runtime": 37.1296, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 190 + }, + { + "epoch": 0.24116161616161616, + "grad_norm": 2.8592522144317627, + "learning_rate": 3.7184873949579835e-06, + "loss": 1.8984, + "step": 191 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.833958864212036, + "learning_rate": 3.739495798319328e-06, + "loss": 1.9084, + "step": 192 + }, + { + "epoch": 0.24368686868686867, + "grad_norm": 2.7842230796813965, + "learning_rate": 3.7605042016806724e-06, + "loss": 1.8843, + "step": 193 + }, + { + "epoch": 0.24494949494949494, + "grad_norm": 3.1309409141540527, + "learning_rate": 3.781512605042017e-06, + "loss": 1.8259, + "step": 194 + }, + { + "epoch": 0.24621212121212122, + "grad_norm": 3.120788335800171, + "learning_rate": 3.8025210084033613e-06, + "loss": 1.8951, + "step": 195 + }, + { + "epoch": 0.2474747474747475, + "grad_norm": 3.2228143215179443, + "learning_rate": 3.8235294117647055e-06, + "loss": 1.8608, + "step": 196 + }, + { + "epoch": 0.24873737373737373, + "grad_norm": 3.3362114429473877, + "learning_rate": 3.844537815126051e-06, + "loss": 1.9163, + "step": 197 + }, + { + "epoch": 0.25, + "grad_norm": 3.0973799228668213, + "learning_rate": 3.865546218487396e-06, + "loss": 1.911, + "step": 198 + }, + { + "epoch": 0.25126262626262624, + "grad_norm": 3.5567545890808105, + "learning_rate": 3.8865546218487395e-06, + "loss": 1.8795, + "step": 199 + }, + { + "epoch": 0.25252525252525254, + "grad_norm": 3.4758830070495605, + "learning_rate": 3.907563025210084e-06, + "loss": 1.8813, + "step": 200 + }, + { + "epoch": 0.25252525252525254, + "eval_loss": 1.9892102479934692, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 200 + }, + { + "epoch": 0.2537878787878788, + "grad_norm": 3.5564475059509277, + "learning_rate": 3.928571428571429e-06, + "loss": 1.8469, + "step": 201 + }, + { + "epoch": 0.255050505050505, + "grad_norm": 3.7004899978637695, + "learning_rate": 3.9495798319327735e-06, + "loss": 1.873, + "step": 202 + }, + { + "epoch": 0.2563131313131313, + "grad_norm": 3.9285225868225098, + "learning_rate": 3.970588235294118e-06, + "loss": 1.8596, + "step": 203 + }, + { + "epoch": 0.25757575757575757, + "grad_norm": 3.1388907432556152, + "learning_rate": 3.991596638655463e-06, + "loss": 1.7991, + "step": 204 + }, + { + "epoch": 0.2588383838383838, + "grad_norm": 3.3797011375427246, + "learning_rate": 4.0126050420168075e-06, + "loss": 1.8682, + "step": 205 + }, + { + "epoch": 0.2601010101010101, + "grad_norm": 3.32183575630188, + "learning_rate": 4.033613445378151e-06, + "loss": 1.8396, + "step": 206 + }, + { + "epoch": 0.26136363636363635, + "grad_norm": 3.457291603088379, + "learning_rate": 4.054621848739496e-06, + "loss": 1.8314, + "step": 207 + }, + { + "epoch": 0.26262626262626265, + "grad_norm": 3.2906265258789062, + "learning_rate": 4.075630252100841e-06, + "loss": 1.844, + "step": 208 + }, + { + "epoch": 0.2638888888888889, + "grad_norm": 3.081913948059082, + "learning_rate": 4.096638655462185e-06, + "loss": 1.8198, + "step": 209 + }, + { + "epoch": 0.26515151515151514, + "grad_norm": 2.9697105884552, + "learning_rate": 4.11764705882353e-06, + "loss": 1.7885, + "step": 210 + }, + { + "epoch": 0.26515151515151514, + "eval_loss": 1.9749963283538818, + "eval_runtime": 36.962, + "eval_samples_per_second": 27.055, + "eval_steps_per_second": 1.136, + "step": 210 + }, + { + "epoch": 0.26641414141414144, + "grad_norm": 3.0732104778289795, + "learning_rate": 4.138655462184874e-06, + "loss": 1.8463, + "step": 211 + }, + { + "epoch": 0.2676767676767677, + "grad_norm": 3.4714958667755127, + "learning_rate": 4.159663865546219e-06, + "loss": 1.8537, + "step": 212 + }, + { + "epoch": 0.2689393939393939, + "grad_norm": 2.955864429473877, + "learning_rate": 4.180672268907564e-06, + "loss": 1.8566, + "step": 213 + }, + { + "epoch": 0.2702020202020202, + "grad_norm": 3.5314948558807373, + "learning_rate": 4.201680672268908e-06, + "loss": 1.7625, + "step": 214 + }, + { + "epoch": 0.27146464646464646, + "grad_norm": 2.8855810165405273, + "learning_rate": 4.222689075630252e-06, + "loss": 1.7968, + "step": 215 + }, + { + "epoch": 0.2727272727272727, + "grad_norm": 3.7866363525390625, + "learning_rate": 4.243697478991597e-06, + "loss": 1.8341, + "step": 216 + }, + { + "epoch": 0.273989898989899, + "grad_norm": 3.0352272987365723, + "learning_rate": 4.264705882352942e-06, + "loss": 1.8276, + "step": 217 + }, + { + "epoch": 0.27525252525252525, + "grad_norm": 3.4880247116088867, + "learning_rate": 4.2857142857142855e-06, + "loss": 1.8194, + "step": 218 + }, + { + "epoch": 0.2765151515151515, + "grad_norm": 3.6807212829589844, + "learning_rate": 4.306722689075631e-06, + "loss": 1.8495, + "step": 219 + }, + { + "epoch": 0.2777777777777778, + "grad_norm": 3.029017448425293, + "learning_rate": 4.327731092436976e-06, + "loss": 1.8125, + "step": 220 + }, + { + "epoch": 0.2777777777777778, + "eval_loss": 1.9633830785751343, + "eval_runtime": 37.1259, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 220 + }, + { + "epoch": 0.27904040404040403, + "grad_norm": 3.4820573329925537, + "learning_rate": 4.3487394957983194e-06, + "loss": 1.8159, + "step": 221 + }, + { + "epoch": 0.2803030303030303, + "grad_norm": 3.2057833671569824, + "learning_rate": 4.369747899159664e-06, + "loss": 1.8215, + "step": 222 + }, + { + "epoch": 0.2815656565656566, + "grad_norm": 3.477302312850952, + "learning_rate": 4.390756302521009e-06, + "loss": 1.7837, + "step": 223 + }, + { + "epoch": 0.2828282828282828, + "grad_norm": 2.8349595069885254, + "learning_rate": 4.411764705882353e-06, + "loss": 1.8555, + "step": 224 + }, + { + "epoch": 0.2840909090909091, + "grad_norm": 3.616478443145752, + "learning_rate": 4.432773109243698e-06, + "loss": 1.8067, + "step": 225 + }, + { + "epoch": 0.28535353535353536, + "grad_norm": 3.111969470977783, + "learning_rate": 4.453781512605043e-06, + "loss": 1.8322, + "step": 226 + }, + { + "epoch": 0.2866161616161616, + "grad_norm": 3.248695135116577, + "learning_rate": 4.474789915966387e-06, + "loss": 1.8004, + "step": 227 + }, + { + "epoch": 0.2878787878787879, + "grad_norm": 3.017082691192627, + "learning_rate": 4.495798319327731e-06, + "loss": 1.776, + "step": 228 + }, + { + "epoch": 0.28914141414141414, + "grad_norm": 3.6790099143981934, + "learning_rate": 4.516806722689076e-06, + "loss": 1.8176, + "step": 229 + }, + { + "epoch": 0.2904040404040404, + "grad_norm": 3.1513001918792725, + "learning_rate": 4.5378151260504205e-06, + "loss": 1.8016, + "step": 230 + }, + { + "epoch": 0.2904040404040404, + "eval_loss": 1.949769377708435, + "eval_runtime": 37.2573, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 230 + }, + { + "epoch": 0.2916666666666667, + "grad_norm": 3.0559096336364746, + "learning_rate": 4.558823529411765e-06, + "loss": 1.7915, + "step": 231 + }, + { + "epoch": 0.29292929292929293, + "grad_norm": 3.0974984169006348, + "learning_rate": 4.57983193277311e-06, + "loss": 1.8277, + "step": 232 + }, + { + "epoch": 0.29419191919191917, + "grad_norm": 2.8514139652252197, + "learning_rate": 4.600840336134454e-06, + "loss": 1.7616, + "step": 233 + }, + { + "epoch": 0.29545454545454547, + "grad_norm": 3.355198383331299, + "learning_rate": 4.621848739495799e-06, + "loss": 1.841, + "step": 234 + }, + { + "epoch": 0.2967171717171717, + "grad_norm": 3.1467435359954834, + "learning_rate": 4.642857142857144e-06, + "loss": 1.7393, + "step": 235 + }, + { + "epoch": 0.29797979797979796, + "grad_norm": 4.065819263458252, + "learning_rate": 4.663865546218488e-06, + "loss": 1.7508, + "step": 236 + }, + { + "epoch": 0.29924242424242425, + "grad_norm": 3.2602460384368896, + "learning_rate": 4.684873949579832e-06, + "loss": 1.7415, + "step": 237 + }, + { + "epoch": 0.3005050505050505, + "grad_norm": 3.6884894371032715, + "learning_rate": 4.705882352941177e-06, + "loss": 1.7608, + "step": 238 + }, + { + "epoch": 0.30176767676767674, + "grad_norm": 3.9843380451202393, + "learning_rate": 4.7268907563025216e-06, + "loss": 1.8519, + "step": 239 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.2823405265808105, + "learning_rate": 4.747899159663865e-06, + "loss": 1.7689, + "step": 240 + }, + { + "epoch": 0.30303030303030304, + "eval_loss": 1.939724326133728, + "eval_runtime": 37.465, + "eval_samples_per_second": 26.692, + "eval_steps_per_second": 1.121, + "step": 240 + }, + { + "epoch": 0.3042929292929293, + "grad_norm": 3.4200189113616943, + "learning_rate": 4.768907563025211e-06, + "loss": 1.7944, + "step": 241 + }, + { + "epoch": 0.3055555555555556, + "grad_norm": 3.7541189193725586, + "learning_rate": 4.7899159663865555e-06, + "loss": 1.8151, + "step": 242 + }, + { + "epoch": 0.3068181818181818, + "grad_norm": 3.9103832244873047, + "learning_rate": 4.810924369747899e-06, + "loss": 1.7921, + "step": 243 + }, + { + "epoch": 0.30808080808080807, + "grad_norm": 3.2769975662231445, + "learning_rate": 4.831932773109244e-06, + "loss": 1.8408, + "step": 244 + }, + { + "epoch": 0.30934343434343436, + "grad_norm": 3.414168357849121, + "learning_rate": 4.852941176470589e-06, + "loss": 1.7937, + "step": 245 + }, + { + "epoch": 0.3106060606060606, + "grad_norm": 3.219625473022461, + "learning_rate": 4.873949579831933e-06, + "loss": 1.796, + "step": 246 + }, + { + "epoch": 0.31186868686868685, + "grad_norm": 3.2573912143707275, + "learning_rate": 4.894957983193277e-06, + "loss": 1.8052, + "step": 247 + }, + { + "epoch": 0.31313131313131315, + "grad_norm": 3.4201040267944336, + "learning_rate": 4.915966386554622e-06, + "loss": 1.8529, + "step": 248 + }, + { + "epoch": 0.3143939393939394, + "grad_norm": 3.7448058128356934, + "learning_rate": 4.936974789915967e-06, + "loss": 1.8183, + "step": 249 + }, + { + "epoch": 0.31565656565656564, + "grad_norm": 3.4108386039733887, + "learning_rate": 4.957983193277311e-06, + "loss": 1.7926, + "step": 250 + }, + { + "epoch": 0.31565656565656564, + "eval_loss": 1.9319047927856445, + "eval_runtime": 37.4745, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 250 + }, + { + "epoch": 0.31691919191919193, + "grad_norm": 3.50185227394104, + "learning_rate": 4.978991596638656e-06, + "loss": 1.792, + "step": 251 + }, + { + "epoch": 0.3181818181818182, + "grad_norm": 3.225581407546997, + "learning_rate": 5e-06, + "loss": 1.7286, + "step": 252 + }, + { + "epoch": 0.3194444444444444, + "grad_norm": 3.3070669174194336, + "learning_rate": 4.997661365762395e-06, + "loss": 1.7957, + "step": 253 + }, + { + "epoch": 0.3207070707070707, + "grad_norm": 3.498079776763916, + "learning_rate": 4.99532273152479e-06, + "loss": 1.823, + "step": 254 + }, + { + "epoch": 0.32196969696969696, + "grad_norm": 3.316534996032715, + "learning_rate": 4.9929840972871845e-06, + "loss": 1.8249, + "step": 255 + }, + { + "epoch": 0.32323232323232326, + "grad_norm": 3.1641833782196045, + "learning_rate": 4.990645463049579e-06, + "loss": 1.7658, + "step": 256 + }, + { + "epoch": 0.3244949494949495, + "grad_norm": 3.0794270038604736, + "learning_rate": 4.988306828811974e-06, + "loss": 1.8146, + "step": 257 + }, + { + "epoch": 0.32575757575757575, + "grad_norm": 3.2410383224487305, + "learning_rate": 4.985968194574369e-06, + "loss": 1.7894, + "step": 258 + }, + { + "epoch": 0.32702020202020204, + "grad_norm": 3.0899274349212646, + "learning_rate": 4.983629560336764e-06, + "loss": 1.7381, + "step": 259 + }, + { + "epoch": 0.3282828282828283, + "grad_norm": 3.0132803916931152, + "learning_rate": 4.981290926099159e-06, + "loss": 1.7713, + "step": 260 + }, + { + "epoch": 0.3282828282828283, + "eval_loss": 1.9123589992523193, + "eval_runtime": 37.3201, + "eval_samples_per_second": 26.795, + "eval_steps_per_second": 1.125, + "step": 260 + }, + { + "epoch": 0.32954545454545453, + "grad_norm": 2.92564058303833, + "learning_rate": 4.9789522918615536e-06, + "loss": 1.7659, + "step": 261 + }, + { + "epoch": 0.33080808080808083, + "grad_norm": 3.1402878761291504, + "learning_rate": 4.976613657623948e-06, + "loss": 1.8439, + "step": 262 + }, + { + "epoch": 0.33207070707070707, + "grad_norm": 3.170881509780884, + "learning_rate": 4.974275023386343e-06, + "loss": 1.7605, + "step": 263 + }, + { + "epoch": 0.3333333333333333, + "grad_norm": 3.014850378036499, + "learning_rate": 4.971936389148738e-06, + "loss": 1.7899, + "step": 264 + }, + { + "epoch": 0.3345959595959596, + "grad_norm": 3.169818878173828, + "learning_rate": 4.969597754911132e-06, + "loss": 1.8263, + "step": 265 + }, + { + "epoch": 0.33585858585858586, + "grad_norm": 2.8709678649902344, + "learning_rate": 4.967259120673527e-06, + "loss": 1.7834, + "step": 266 + }, + { + "epoch": 0.3371212121212121, + "grad_norm": 2.7527096271514893, + "learning_rate": 4.964920486435922e-06, + "loss": 1.8106, + "step": 267 + }, + { + "epoch": 0.3383838383838384, + "grad_norm": 2.9960227012634277, + "learning_rate": 4.962581852198316e-06, + "loss": 1.7452, + "step": 268 + }, + { + "epoch": 0.33964646464646464, + "grad_norm": 2.979163885116577, + "learning_rate": 4.9602432179607115e-06, + "loss": 1.6943, + "step": 269 + }, + { + "epoch": 0.3409090909090909, + "grad_norm": 3.2396063804626465, + "learning_rate": 4.957904583723106e-06, + "loss": 1.7798, + "step": 270 + }, + { + "epoch": 0.3409090909090909, + "eval_loss": 1.9126806259155273, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 270 + }, + { + "epoch": 0.3421717171717172, + "grad_norm": 2.7868010997772217, + "learning_rate": 4.955565949485501e-06, + "loss": 1.8053, + "step": 271 + }, + { + "epoch": 0.3434343434343434, + "grad_norm": 2.8895349502563477, + "learning_rate": 4.9532273152478956e-06, + "loss": 1.7672, + "step": 272 + }, + { + "epoch": 0.3446969696969697, + "grad_norm": 3.0839710235595703, + "learning_rate": 4.95088868101029e-06, + "loss": 1.7414, + "step": 273 + }, + { + "epoch": 0.34595959595959597, + "grad_norm": 2.81974458694458, + "learning_rate": 4.948550046772685e-06, + "loss": 1.7527, + "step": 274 + }, + { + "epoch": 0.3472222222222222, + "grad_norm": 3.7469775676727295, + "learning_rate": 4.9462114125350805e-06, + "loss": 1.7689, + "step": 275 + }, + { + "epoch": 0.3484848484848485, + "grad_norm": 3.434573173522949, + "learning_rate": 4.943872778297475e-06, + "loss": 1.739, + "step": 276 + }, + { + "epoch": 0.34974747474747475, + "grad_norm": 2.985041379928589, + "learning_rate": 4.941534144059869e-06, + "loss": 1.7894, + "step": 277 + }, + { + "epoch": 0.351010101010101, + "grad_norm": 3.441204309463501, + "learning_rate": 4.939195509822265e-06, + "loss": 1.7804, + "step": 278 + }, + { + "epoch": 0.3522727272727273, + "grad_norm": 2.8371102809906006, + "learning_rate": 4.936856875584659e-06, + "loss": 1.7415, + "step": 279 + }, + { + "epoch": 0.35353535353535354, + "grad_norm": 3.662773847579956, + "learning_rate": 4.9345182413470535e-06, + "loss": 1.7894, + "step": 280 + }, + { + "epoch": 0.35353535353535354, + "eval_loss": 1.915844440460205, + "eval_runtime": 37.0941, + "eval_samples_per_second": 26.958, + "eval_steps_per_second": 1.132, + "step": 280 + }, + { + "epoch": 0.3547979797979798, + "grad_norm": 3.3254036903381348, + "learning_rate": 4.932179607109449e-06, + "loss": 1.7238, + "step": 281 + }, + { + "epoch": 0.3560606060606061, + "grad_norm": 2.9872665405273438, + "learning_rate": 4.929840972871843e-06, + "loss": 1.769, + "step": 282 + }, + { + "epoch": 0.3573232323232323, + "grad_norm": 3.0163521766662598, + "learning_rate": 4.9275023386342376e-06, + "loss": 1.7702, + "step": 283 + }, + { + "epoch": 0.35858585858585856, + "grad_norm": 2.9687387943267822, + "learning_rate": 4.925163704396633e-06, + "loss": 1.7927, + "step": 284 + }, + { + "epoch": 0.35984848484848486, + "grad_norm": 2.7064011096954346, + "learning_rate": 4.922825070159027e-06, + "loss": 1.7638, + "step": 285 + }, + { + "epoch": 0.3611111111111111, + "grad_norm": 3.0043134689331055, + "learning_rate": 4.9204864359214225e-06, + "loss": 1.8041, + "step": 286 + }, + { + "epoch": 0.36237373737373735, + "grad_norm": 3.037100315093994, + "learning_rate": 4.918147801683817e-06, + "loss": 1.7579, + "step": 287 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7528676986694336, + "learning_rate": 4.915809167446211e-06, + "loss": 1.7629, + "step": 288 + }, + { + "epoch": 0.3648989898989899, + "grad_norm": 3.075815439224243, + "learning_rate": 4.913470533208607e-06, + "loss": 1.7966, + "step": 289 + }, + { + "epoch": 0.3661616161616162, + "grad_norm": 2.964785575866699, + "learning_rate": 4.911131898971001e-06, + "loss": 1.801, + "step": 290 + }, + { + "epoch": 0.3661616161616162, + "eval_loss": 1.8839410543441772, + "eval_runtime": 37.0844, + "eval_samples_per_second": 26.965, + "eval_steps_per_second": 1.133, + "step": 290 + }, + { + "epoch": 0.36742424242424243, + "grad_norm": 2.668747663497925, + "learning_rate": 4.908793264733396e-06, + "loss": 1.7703, + "step": 291 + }, + { + "epoch": 0.3686868686868687, + "grad_norm": 2.814751148223877, + "learning_rate": 4.906454630495791e-06, + "loss": 1.7075, + "step": 292 + }, + { + "epoch": 0.369949494949495, + "grad_norm": 3.005922794342041, + "learning_rate": 4.904115996258186e-06, + "loss": 1.7418, + "step": 293 + }, + { + "epoch": 0.3712121212121212, + "grad_norm": 3.050936460494995, + "learning_rate": 4.90177736202058e-06, + "loss": 1.793, + "step": 294 + }, + { + "epoch": 0.37247474747474746, + "grad_norm": 2.822004556655884, + "learning_rate": 4.899438727782975e-06, + "loss": 1.8303, + "step": 295 + }, + { + "epoch": 0.37373737373737376, + "grad_norm": 2.950254201889038, + "learning_rate": 4.89710009354537e-06, + "loss": 1.808, + "step": 296 + }, + { + "epoch": 0.375, + "grad_norm": 2.8140666484832764, + "learning_rate": 4.8947614593077645e-06, + "loss": 1.7262, + "step": 297 + }, + { + "epoch": 0.37626262626262624, + "grad_norm": 2.899620532989502, + "learning_rate": 4.89242282507016e-06, + "loss": 1.7157, + "step": 298 + }, + { + "epoch": 0.37752525252525254, + "grad_norm": 2.805753469467163, + "learning_rate": 4.890084190832554e-06, + "loss": 1.7294, + "step": 299 + }, + { + "epoch": 0.3787878787878788, + "grad_norm": 3.110246181488037, + "learning_rate": 4.887745556594949e-06, + "loss": 1.7415, + "step": 300 + }, + { + "epoch": 0.3787878787878788, + "eval_loss": 1.892285943031311, + "eval_runtime": 37.0732, + "eval_samples_per_second": 26.974, + "eval_steps_per_second": 1.133, + "step": 300 + }, + { + "epoch": 0.380050505050505, + "grad_norm": 2.9197490215301514, + "learning_rate": 4.885406922357344e-06, + "loss": 1.7132, + "step": 301 + }, + { + "epoch": 0.3813131313131313, + "grad_norm": 3.3165123462677, + "learning_rate": 4.883068288119738e-06, + "loss": 1.777, + "step": 302 + }, + { + "epoch": 0.38257575757575757, + "grad_norm": 2.8409054279327393, + "learning_rate": 4.880729653882133e-06, + "loss": 1.7092, + "step": 303 + }, + { + "epoch": 0.3838383838383838, + "grad_norm": 2.6800901889801025, + "learning_rate": 4.878391019644528e-06, + "loss": 1.7622, + "step": 304 + }, + { + "epoch": 0.3851010101010101, + "grad_norm": 3.2847490310668945, + "learning_rate": 4.876052385406922e-06, + "loss": 1.729, + "step": 305 + }, + { + "epoch": 0.38636363636363635, + "grad_norm": 2.786057949066162, + "learning_rate": 4.873713751169318e-06, + "loss": 1.7168, + "step": 306 + }, + { + "epoch": 0.38762626262626265, + "grad_norm": 3.236050605773926, + "learning_rate": 4.871375116931712e-06, + "loss": 1.6809, + "step": 307 + }, + { + "epoch": 0.3888888888888889, + "grad_norm": 2.851910352706909, + "learning_rate": 4.869036482694107e-06, + "loss": 1.7336, + "step": 308 + }, + { + "epoch": 0.39015151515151514, + "grad_norm": 2.892421007156372, + "learning_rate": 4.866697848456502e-06, + "loss": 1.7451, + "step": 309 + }, + { + "epoch": 0.39141414141414144, + "grad_norm": 2.8175048828125, + "learning_rate": 4.864359214218897e-06, + "loss": 1.7734, + "step": 310 + }, + { + "epoch": 0.39141414141414144, + "eval_loss": 1.8777843713760376, + "eval_runtime": 36.9321, + "eval_samples_per_second": 27.077, + "eval_steps_per_second": 1.137, + "step": 310 + }, + { + "epoch": 0.3926767676767677, + "grad_norm": 3.005329132080078, + "learning_rate": 4.8620205799812915e-06, + "loss": 1.7476, + "step": 311 + }, + { + "epoch": 0.3939393939393939, + "grad_norm": 2.849017858505249, + "learning_rate": 4.859681945743686e-06, + "loss": 1.7854, + "step": 312 + }, + { + "epoch": 0.3952020202020202, + "grad_norm": 2.801048517227173, + "learning_rate": 4.857343311506081e-06, + "loss": 1.7186, + "step": 313 + }, + { + "epoch": 0.39646464646464646, + "grad_norm": 2.5772831439971924, + "learning_rate": 4.855004677268476e-06, + "loss": 1.7279, + "step": 314 + }, + { + "epoch": 0.3977272727272727, + "grad_norm": 2.7969934940338135, + "learning_rate": 4.85266604303087e-06, + "loss": 1.7475, + "step": 315 + }, + { + "epoch": 0.398989898989899, + "grad_norm": 2.7067770957946777, + "learning_rate": 4.850327408793265e-06, + "loss": 1.7055, + "step": 316 + }, + { + "epoch": 0.40025252525252525, + "grad_norm": 2.70928955078125, + "learning_rate": 4.84798877455566e-06, + "loss": 1.7203, + "step": 317 + }, + { + "epoch": 0.4015151515151515, + "grad_norm": 2.7704474925994873, + "learning_rate": 4.845650140318054e-06, + "loss": 1.8058, + "step": 318 + }, + { + "epoch": 0.4027777777777778, + "grad_norm": 2.7680182456970215, + "learning_rate": 4.843311506080449e-06, + "loss": 1.7162, + "step": 319 + }, + { + "epoch": 0.40404040404040403, + "grad_norm": 3.0953330993652344, + "learning_rate": 4.840972871842844e-06, + "loss": 1.7617, + "step": 320 + }, + { + "epoch": 0.40404040404040403, + "eval_loss": 1.8680965900421143, + "eval_runtime": 37.0382, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 320 + }, + { + "epoch": 0.4053030303030303, + "grad_norm": 2.8340392112731934, + "learning_rate": 4.838634237605239e-06, + "loss": 1.7767, + "step": 321 + }, + { + "epoch": 0.4065656565656566, + "grad_norm": 3.057861804962158, + "learning_rate": 4.836295603367634e-06, + "loss": 1.8043, + "step": 322 + }, + { + "epoch": 0.4078282828282828, + "grad_norm": 3.0123419761657715, + "learning_rate": 4.833956969130029e-06, + "loss": 1.7561, + "step": 323 + }, + { + "epoch": 0.4090909090909091, + "grad_norm": 3.0873124599456787, + "learning_rate": 4.831618334892423e-06, + "loss": 1.7581, + "step": 324 + }, + { + "epoch": 0.41035353535353536, + "grad_norm": 2.6600539684295654, + "learning_rate": 4.8292797006548184e-06, + "loss": 1.7005, + "step": 325 + }, + { + "epoch": 0.4116161616161616, + "grad_norm": 2.905280590057373, + "learning_rate": 4.826941066417213e-06, + "loss": 1.7592, + "step": 326 + }, + { + "epoch": 0.4128787878787879, + "grad_norm": 3.0642189979553223, + "learning_rate": 4.824602432179607e-06, + "loss": 1.6456, + "step": 327 + }, + { + "epoch": 0.41414141414141414, + "grad_norm": 2.606531858444214, + "learning_rate": 4.8222637979420025e-06, + "loss": 1.6731, + "step": 328 + }, + { + "epoch": 0.4154040404040404, + "grad_norm": 2.9918158054351807, + "learning_rate": 4.819925163704397e-06, + "loss": 1.6854, + "step": 329 + }, + { + "epoch": 0.4166666666666667, + "grad_norm": 2.721940040588379, + "learning_rate": 4.817586529466791e-06, + "loss": 1.7346, + "step": 330 + }, + { + "epoch": 0.4166666666666667, + "eval_loss": 1.863736629486084, + "eval_runtime": 37.0887, + "eval_samples_per_second": 26.962, + "eval_steps_per_second": 1.132, + "step": 330 + }, + { + "epoch": 0.41792929292929293, + "grad_norm": 3.1076390743255615, + "learning_rate": 4.815247895229187e-06, + "loss": 1.6286, + "step": 331 + }, + { + "epoch": 0.41919191919191917, + "grad_norm": 3.1235783100128174, + "learning_rate": 4.812909260991581e-06, + "loss": 1.7404, + "step": 332 + }, + { + "epoch": 0.42045454545454547, + "grad_norm": 3.1211330890655518, + "learning_rate": 4.8105706267539755e-06, + "loss": 1.7143, + "step": 333 + }, + { + "epoch": 0.4217171717171717, + "grad_norm": 3.072317600250244, + "learning_rate": 4.808231992516371e-06, + "loss": 1.7514, + "step": 334 + }, + { + "epoch": 0.42297979797979796, + "grad_norm": 2.880671739578247, + "learning_rate": 4.805893358278765e-06, + "loss": 1.7018, + "step": 335 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.700063943862915, + "learning_rate": 4.8035547240411604e-06, + "loss": 1.7089, + "step": 336 + }, + { + "epoch": 0.4255050505050505, + "grad_norm": 2.9038445949554443, + "learning_rate": 4.801216089803556e-06, + "loss": 1.7483, + "step": 337 + }, + { + "epoch": 0.42676767676767674, + "grad_norm": 2.7822766304016113, + "learning_rate": 4.79887745556595e-06, + "loss": 1.6927, + "step": 338 + }, + { + "epoch": 0.42803030303030304, + "grad_norm": 2.957554340362549, + "learning_rate": 4.7965388213283445e-06, + "loss": 1.7162, + "step": 339 + }, + { + "epoch": 0.4292929292929293, + "grad_norm": 3.1333892345428467, + "learning_rate": 4.79420018709074e-06, + "loss": 1.731, + "step": 340 + }, + { + "epoch": 0.4292929292929293, + "eval_loss": 1.852413535118103, + "eval_runtime": 37.1809, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 340 + }, + { + "epoch": 0.4305555555555556, + "grad_norm": 2.9801084995269775, + "learning_rate": 4.791861552853134e-06, + "loss": 1.6892, + "step": 341 + }, + { + "epoch": 0.4318181818181818, + "grad_norm": 2.5077271461486816, + "learning_rate": 4.789522918615529e-06, + "loss": 1.637, + "step": 342 + }, + { + "epoch": 0.43308080808080807, + "grad_norm": 2.8654391765594482, + "learning_rate": 4.787184284377924e-06, + "loss": 1.6915, + "step": 343 + }, + { + "epoch": 0.43434343434343436, + "grad_norm": 2.6204264163970947, + "learning_rate": 4.784845650140318e-06, + "loss": 1.6722, + "step": 344 + }, + { + "epoch": 0.4356060606060606, + "grad_norm": 2.633991003036499, + "learning_rate": 4.782507015902713e-06, + "loss": 1.6797, + "step": 345 + }, + { + "epoch": 0.43686868686868685, + "grad_norm": 2.7203924655914307, + "learning_rate": 4.780168381665108e-06, + "loss": 1.7018, + "step": 346 + }, + { + "epoch": 0.43813131313131315, + "grad_norm": 2.6789209842681885, + "learning_rate": 4.7778297474275024e-06, + "loss": 1.7153, + "step": 347 + }, + { + "epoch": 0.4393939393939394, + "grad_norm": 2.6285433769226074, + "learning_rate": 4.775491113189898e-06, + "loss": 1.7082, + "step": 348 + }, + { + "epoch": 0.44065656565656564, + "grad_norm": 2.623793840408325, + "learning_rate": 4.773152478952292e-06, + "loss": 1.7094, + "step": 349 + }, + { + "epoch": 0.44191919191919193, + "grad_norm": 2.647160291671753, + "learning_rate": 4.7708138447146865e-06, + "loss": 1.6528, + "step": 350 + }, + { + "epoch": 0.44191919191919193, + "eval_loss": 1.8506718873977661, + "eval_runtime": 37.4232, + "eval_samples_per_second": 26.721, + "eval_steps_per_second": 1.122, + "step": 350 + }, + { + "epoch": 0.4431818181818182, + "grad_norm": 2.664553642272949, + "learning_rate": 4.768475210477082e-06, + "loss": 1.7008, + "step": 351 + }, + { + "epoch": 0.4444444444444444, + "grad_norm": 2.8486251831054688, + "learning_rate": 4.766136576239477e-06, + "loss": 1.6811, + "step": 352 + }, + { + "epoch": 0.4457070707070707, + "grad_norm": 2.7514266967773438, + "learning_rate": 4.7637979420018715e-06, + "loss": 1.6969, + "step": 353 + }, + { + "epoch": 0.44696969696969696, + "grad_norm": 2.848083257675171, + "learning_rate": 4.761459307764266e-06, + "loss": 1.7058, + "step": 354 + }, + { + "epoch": 0.44823232323232326, + "grad_norm": 2.759218454360962, + "learning_rate": 4.759120673526661e-06, + "loss": 1.6109, + "step": 355 + }, + { + "epoch": 0.4494949494949495, + "grad_norm": 2.905318021774292, + "learning_rate": 4.756782039289056e-06, + "loss": 1.7689, + "step": 356 + }, + { + "epoch": 0.45075757575757575, + "grad_norm": 2.845646381378174, + "learning_rate": 4.75444340505145e-06, + "loss": 1.6687, + "step": 357 + }, + { + "epoch": 0.45202020202020204, + "grad_norm": 2.8609306812286377, + "learning_rate": 4.752104770813845e-06, + "loss": 1.6332, + "step": 358 + }, + { + "epoch": 0.4532828282828283, + "grad_norm": 3.016479969024658, + "learning_rate": 4.74976613657624e-06, + "loss": 1.6692, + "step": 359 + }, + { + "epoch": 0.45454545454545453, + "grad_norm": 3.2790887355804443, + "learning_rate": 4.747427502338635e-06, + "loss": 1.704, + "step": 360 + }, + { + "epoch": 0.45454545454545453, + "eval_loss": 1.848663330078125, + "eval_runtime": 37.4817, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 360 + }, + { + "epoch": 0.45580808080808083, + "grad_norm": 3.1137545108795166, + "learning_rate": 4.745088868101029e-06, + "loss": 1.6797, + "step": 361 + }, + { + "epoch": 0.45707070707070707, + "grad_norm": 2.7598798274993896, + "learning_rate": 4.742750233863424e-06, + "loss": 1.6477, + "step": 362 + }, + { + "epoch": 0.4583333333333333, + "grad_norm": 3.4851579666137695, + "learning_rate": 4.740411599625819e-06, + "loss": 1.6481, + "step": 363 + }, + { + "epoch": 0.4595959595959596, + "grad_norm": 2.9285919666290283, + "learning_rate": 4.7380729653882135e-06, + "loss": 1.7094, + "step": 364 + }, + { + "epoch": 0.46085858585858586, + "grad_norm": 2.901458501815796, + "learning_rate": 4.735734331150608e-06, + "loss": 1.6906, + "step": 365 + }, + { + "epoch": 0.4621212121212121, + "grad_norm": 2.8737902641296387, + "learning_rate": 4.733395696913003e-06, + "loss": 1.6819, + "step": 366 + }, + { + "epoch": 0.4633838383838384, + "grad_norm": 3.2480952739715576, + "learning_rate": 4.7310570626753984e-06, + "loss": 1.6827, + "step": 367 + }, + { + "epoch": 0.46464646464646464, + "grad_norm": 2.8714661598205566, + "learning_rate": 4.728718428437793e-06, + "loss": 1.7349, + "step": 368 + }, + { + "epoch": 0.4659090909090909, + "grad_norm": 2.8410098552703857, + "learning_rate": 4.726379794200187e-06, + "loss": 1.7214, + "step": 369 + }, + { + "epoch": 0.4671717171717172, + "grad_norm": 2.744807720184326, + "learning_rate": 4.7240411599625826e-06, + "loss": 1.6742, + "step": 370 + }, + { + "epoch": 0.4671717171717172, + "eval_loss": 1.842191457748413, + "eval_runtime": 37.454, + "eval_samples_per_second": 26.699, + "eval_steps_per_second": 1.121, + "step": 370 + }, + { + "epoch": 0.4684343434343434, + "grad_norm": 2.8135552406311035, + "learning_rate": 4.721702525724977e-06, + "loss": 1.6957, + "step": 371 + }, + { + "epoch": 0.4696969696969697, + "grad_norm": 2.856764078140259, + "learning_rate": 4.719363891487372e-06, + "loss": 1.6802, + "step": 372 + }, + { + "epoch": 0.47095959595959597, + "grad_norm": 3.0707688331604004, + "learning_rate": 4.717025257249767e-06, + "loss": 1.7383, + "step": 373 + }, + { + "epoch": 0.4722222222222222, + "grad_norm": 2.8832876682281494, + "learning_rate": 4.714686623012161e-06, + "loss": 1.6309, + "step": 374 + }, + { + "epoch": 0.4734848484848485, + "grad_norm": 2.6840622425079346, + "learning_rate": 4.712347988774556e-06, + "loss": 1.6624, + "step": 375 + }, + { + "epoch": 0.47474747474747475, + "grad_norm": 3.2920548915863037, + "learning_rate": 4.710009354536951e-06, + "loss": 1.6887, + "step": 376 + }, + { + "epoch": 0.476010101010101, + "grad_norm": 3.0317189693450928, + "learning_rate": 4.707670720299345e-06, + "loss": 1.6514, + "step": 377 + }, + { + "epoch": 0.4772727272727273, + "grad_norm": 2.7284793853759766, + "learning_rate": 4.7053320860617404e-06, + "loss": 1.6595, + "step": 378 + }, + { + "epoch": 0.47853535353535354, + "grad_norm": 2.86665415763855, + "learning_rate": 4.702993451824135e-06, + "loss": 1.7219, + "step": 379 + }, + { + "epoch": 0.4797979797979798, + "grad_norm": 3.041191577911377, + "learning_rate": 4.700654817586529e-06, + "loss": 1.6959, + "step": 380 + }, + { + "epoch": 0.4797979797979798, + "eval_loss": 1.8414148092269897, + "eval_runtime": 37.2537, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.4810606060606061, + "grad_norm": 3.1729421615600586, + "learning_rate": 4.6983161833489246e-06, + "loss": 1.6538, + "step": 381 + }, + { + "epoch": 0.4823232323232323, + "grad_norm": 2.8562991619110107, + "learning_rate": 4.69597754911132e-06, + "loss": 1.6667, + "step": 382 + }, + { + "epoch": 0.48358585858585856, + "grad_norm": 3.0348711013793945, + "learning_rate": 4.693638914873714e-06, + "loss": 1.6888, + "step": 383 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.9156477451324463, + "learning_rate": 4.6913002806361095e-06, + "loss": 1.628, + "step": 384 + }, + { + "epoch": 0.4861111111111111, + "grad_norm": 2.832689046859741, + "learning_rate": 4.688961646398504e-06, + "loss": 1.664, + "step": 385 + }, + { + "epoch": 0.48737373737373735, + "grad_norm": 2.9858946800231934, + "learning_rate": 4.686623012160898e-06, + "loss": 1.6696, + "step": 386 + }, + { + "epoch": 0.48863636363636365, + "grad_norm": 2.732468843460083, + "learning_rate": 4.684284377923294e-06, + "loss": 1.6444, + "step": 387 + }, + { + "epoch": 0.4898989898989899, + "grad_norm": 2.7801132202148438, + "learning_rate": 4.681945743685688e-06, + "loss": 1.7084, + "step": 388 + }, + { + "epoch": 0.4911616161616162, + "grad_norm": 2.646303415298462, + "learning_rate": 4.6796071094480824e-06, + "loss": 1.6963, + "step": 389 + }, + { + "epoch": 0.49242424242424243, + "grad_norm": 2.8750250339508057, + "learning_rate": 4.677268475210478e-06, + "loss": 1.6814, + "step": 390 + }, + { + "epoch": 0.49242424242424243, + "eval_loss": 1.8283894062042236, + "eval_runtime": 37.2058, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 390 + }, + { + "epoch": 0.4936868686868687, + "grad_norm": 2.5373270511627197, + "learning_rate": 4.674929840972872e-06, + "loss": 1.5956, + "step": 391 + }, + { + "epoch": 0.494949494949495, + "grad_norm": 2.837170124053955, + "learning_rate": 4.6725912067352666e-06, + "loss": 1.6237, + "step": 392 + }, + { + "epoch": 0.4962121212121212, + "grad_norm": 3.199463367462158, + "learning_rate": 4.670252572497662e-06, + "loss": 1.7087, + "step": 393 + }, + { + "epoch": 0.49747474747474746, + "grad_norm": 2.9850947856903076, + "learning_rate": 4.667913938260056e-06, + "loss": 1.5913, + "step": 394 + }, + { + "epoch": 0.49873737373737376, + "grad_norm": 2.954148292541504, + "learning_rate": 4.665575304022451e-06, + "loss": 1.6272, + "step": 395 + }, + { + "epoch": 0.5, + "grad_norm": 2.8625757694244385, + "learning_rate": 4.663236669784846e-06, + "loss": 1.6623, + "step": 396 + }, + { + "epoch": 0.5012626262626263, + "grad_norm": 3.196354627609253, + "learning_rate": 4.66089803554724e-06, + "loss": 1.7249, + "step": 397 + }, + { + "epoch": 0.5025252525252525, + "grad_norm": 2.7257323265075684, + "learning_rate": 4.658559401309636e-06, + "loss": 1.665, + "step": 398 + }, + { + "epoch": 0.5037878787878788, + "grad_norm": 3.0976016521453857, + "learning_rate": 4.656220767072031e-06, + "loss": 1.6763, + "step": 399 + }, + { + "epoch": 0.5050505050505051, + "grad_norm": 3.0398128032684326, + "learning_rate": 4.653882132834425e-06, + "loss": 1.6375, + "step": 400 + }, + { + "epoch": 0.5050505050505051, + "eval_loss": 1.8324880599975586, + "eval_runtime": 37.1754, + "eval_samples_per_second": 26.9, + "eval_steps_per_second": 1.13, + "step": 400 + }, + { + "epoch": 0.5063131313131313, + "grad_norm": 3.314958095550537, + "learning_rate": 4.65154349859682e-06, + "loss": 1.6699, + "step": 401 + }, + { + "epoch": 0.5075757575757576, + "grad_norm": 2.8478870391845703, + "learning_rate": 4.649204864359215e-06, + "loss": 1.6495, + "step": 402 + }, + { + "epoch": 0.5088383838383839, + "grad_norm": 2.8476510047912598, + "learning_rate": 4.646866230121609e-06, + "loss": 1.633, + "step": 403 + }, + { + "epoch": 0.51010101010101, + "grad_norm": 2.916667938232422, + "learning_rate": 4.644527595884004e-06, + "loss": 1.6092, + "step": 404 + }, + { + "epoch": 0.5113636363636364, + "grad_norm": 2.977226495742798, + "learning_rate": 4.642188961646399e-06, + "loss": 1.6588, + "step": 405 + }, + { + "epoch": 0.5126262626262627, + "grad_norm": 3.0540225505828857, + "learning_rate": 4.6398503274087935e-06, + "loss": 1.6529, + "step": 406 + }, + { + "epoch": 0.5138888888888888, + "grad_norm": 2.902035713195801, + "learning_rate": 4.637511693171188e-06, + "loss": 1.5983, + "step": 407 + }, + { + "epoch": 0.5151515151515151, + "grad_norm": 2.9009487628936768, + "learning_rate": 4.635173058933583e-06, + "loss": 1.6389, + "step": 408 + }, + { + "epoch": 0.5164141414141414, + "grad_norm": 2.904982805252075, + "learning_rate": 4.632834424695978e-06, + "loss": 1.6971, + "step": 409 + }, + { + "epoch": 0.5176767676767676, + "grad_norm": 2.8340721130371094, + "learning_rate": 4.630495790458373e-06, + "loss": 1.6336, + "step": 410 + }, + { + "epoch": 0.5176767676767676, + "eval_loss": 1.8218064308166504, + "eval_runtime": 36.9963, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 410 + }, + { + "epoch": 0.5189393939393939, + "grad_norm": 2.9507815837860107, + "learning_rate": 4.628157156220767e-06, + "loss": 1.6795, + "step": 411 + }, + { + "epoch": 0.5202020202020202, + "grad_norm": 2.868793249130249, + "learning_rate": 4.625818521983162e-06, + "loss": 1.6494, + "step": 412 + }, + { + "epoch": 0.5214646464646465, + "grad_norm": 2.786395788192749, + "learning_rate": 4.623479887745557e-06, + "loss": 1.7156, + "step": 413 + }, + { + "epoch": 0.5227272727272727, + "grad_norm": 2.7456934452056885, + "learning_rate": 4.621141253507952e-06, + "loss": 1.6528, + "step": 414 + }, + { + "epoch": 0.523989898989899, + "grad_norm": 3.1211695671081543, + "learning_rate": 4.618802619270347e-06, + "loss": 1.6891, + "step": 415 + }, + { + "epoch": 0.5252525252525253, + "grad_norm": 3.2319254875183105, + "learning_rate": 4.616463985032741e-06, + "loss": 1.6452, + "step": 416 + }, + { + "epoch": 0.5265151515151515, + "grad_norm": 2.6718053817749023, + "learning_rate": 4.614125350795136e-06, + "loss": 1.6442, + "step": 417 + }, + { + "epoch": 0.5277777777777778, + "grad_norm": 2.985020637512207, + "learning_rate": 4.611786716557531e-06, + "loss": 1.6935, + "step": 418 + }, + { + "epoch": 0.5290404040404041, + "grad_norm": 2.798720598220825, + "learning_rate": 4.609448082319925e-06, + "loss": 1.6303, + "step": 419 + }, + { + "epoch": 0.5303030303030303, + "grad_norm": 2.860403299331665, + "learning_rate": 4.6071094480823205e-06, + "loss": 1.6445, + "step": 420 + }, + { + "epoch": 0.5303030303030303, + "eval_loss": 1.8173401355743408, + "eval_runtime": 37.1334, + "eval_samples_per_second": 26.93, + "eval_steps_per_second": 1.131, + "step": 420 + }, + { + "epoch": 0.5315656565656566, + "grad_norm": 2.883209705352783, + "learning_rate": 4.604770813844715e-06, + "loss": 1.6182, + "step": 421 + }, + { + "epoch": 0.5328282828282829, + "grad_norm": 3.0330183506011963, + "learning_rate": 4.60243217960711e-06, + "loss": 1.6208, + "step": 422 + }, + { + "epoch": 0.5340909090909091, + "grad_norm": 3.224184274673462, + "learning_rate": 4.6000935453695046e-06, + "loss": 1.6687, + "step": 423 + }, + { + "epoch": 0.5353535353535354, + "grad_norm": 3.0345349311828613, + "learning_rate": 4.597754911131899e-06, + "loss": 1.6654, + "step": 424 + }, + { + "epoch": 0.5366161616161617, + "grad_norm": 3.1425750255584717, + "learning_rate": 4.595416276894294e-06, + "loss": 1.6569, + "step": 425 + }, + { + "epoch": 0.5378787878787878, + "grad_norm": 2.6387979984283447, + "learning_rate": 4.593077642656689e-06, + "loss": 1.6868, + "step": 426 + }, + { + "epoch": 0.5391414141414141, + "grad_norm": 2.959446430206299, + "learning_rate": 4.590739008419083e-06, + "loss": 1.621, + "step": 427 + }, + { + "epoch": 0.5404040404040404, + "grad_norm": 3.009673833847046, + "learning_rate": 4.588400374181478e-06, + "loss": 1.6414, + "step": 428 + }, + { + "epoch": 0.5416666666666666, + "grad_norm": 2.827531576156616, + "learning_rate": 4.586061739943874e-06, + "loss": 1.6561, + "step": 429 + }, + { + "epoch": 0.5429292929292929, + "grad_norm": 2.8770735263824463, + "learning_rate": 4.583723105706268e-06, + "loss": 1.6685, + "step": 430 + }, + { + "epoch": 0.5429292929292929, + "eval_loss": 1.8234716653823853, + "eval_runtime": 37.1647, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 430 + }, + { + "epoch": 0.5441919191919192, + "grad_norm": 3.1215782165527344, + "learning_rate": 4.5813844714686625e-06, + "loss": 1.6548, + "step": 431 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.6587517261505127, + "learning_rate": 4.579045837231058e-06, + "loss": 1.649, + "step": 432 + }, + { + "epoch": 0.5467171717171717, + "grad_norm": 2.7059030532836914, + "learning_rate": 4.576707202993452e-06, + "loss": 1.6293, + "step": 433 + }, + { + "epoch": 0.547979797979798, + "grad_norm": 2.7390925884246826, + "learning_rate": 4.574368568755847e-06, + "loss": 1.6957, + "step": 434 + }, + { + "epoch": 0.5492424242424242, + "grad_norm": 2.9457757472991943, + "learning_rate": 4.572029934518242e-06, + "loss": 1.6714, + "step": 435 + }, + { + "epoch": 0.5505050505050505, + "grad_norm": 2.6871347427368164, + "learning_rate": 4.569691300280636e-06, + "loss": 1.6648, + "step": 436 + }, + { + "epoch": 0.5517676767676768, + "grad_norm": 3.1708056926727295, + "learning_rate": 4.5673526660430315e-06, + "loss": 1.6415, + "step": 437 + }, + { + "epoch": 0.553030303030303, + "grad_norm": 2.8767378330230713, + "learning_rate": 4.565014031805426e-06, + "loss": 1.6475, + "step": 438 + }, + { + "epoch": 0.5542929292929293, + "grad_norm": 3.083566188812256, + "learning_rate": 4.56267539756782e-06, + "loss": 1.632, + "step": 439 + }, + { + "epoch": 0.5555555555555556, + "grad_norm": 2.584460735321045, + "learning_rate": 4.560336763330216e-06, + "loss": 1.6469, + "step": 440 + }, + { + "epoch": 0.5555555555555556, + "eval_loss": 1.8140095472335815, + "eval_runtime": 37.1986, + "eval_samples_per_second": 26.883, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.5568181818181818, + "grad_norm": 2.5885088443756104, + "learning_rate": 4.55799812909261e-06, + "loss": 1.6058, + "step": 441 + }, + { + "epoch": 0.5580808080808081, + "grad_norm": 2.8467836380004883, + "learning_rate": 4.5556594948550045e-06, + "loss": 1.6959, + "step": 442 + }, + { + "epoch": 0.5593434343434344, + "grad_norm": 2.764326333999634, + "learning_rate": 4.5533208606174e-06, + "loss": 1.6565, + "step": 443 + }, + { + "epoch": 0.5606060606060606, + "grad_norm": 2.88089919090271, + "learning_rate": 4.550982226379795e-06, + "loss": 1.6651, + "step": 444 + }, + { + "epoch": 0.5618686868686869, + "grad_norm": 2.6239469051361084, + "learning_rate": 4.548643592142189e-06, + "loss": 1.6255, + "step": 445 + }, + { + "epoch": 0.5631313131313131, + "grad_norm": 2.7748732566833496, + "learning_rate": 4.546304957904585e-06, + "loss": 1.6587, + "step": 446 + }, + { + "epoch": 0.5643939393939394, + "grad_norm": 2.7075202465057373, + "learning_rate": 4.543966323666979e-06, + "loss": 1.6448, + "step": 447 + }, + { + "epoch": 0.5656565656565656, + "grad_norm": 2.9291634559631348, + "learning_rate": 4.5416276894293735e-06, + "loss": 1.6564, + "step": 448 + }, + { + "epoch": 0.5669191919191919, + "grad_norm": 3.0039470195770264, + "learning_rate": 4.539289055191769e-06, + "loss": 1.623, + "step": 449 + }, + { + "epoch": 0.5681818181818182, + "grad_norm": 3.085277557373047, + "learning_rate": 4.536950420954163e-06, + "loss": 1.6176, + "step": 450 + }, + { + "epoch": 0.5681818181818182, + "eval_loss": 1.8022958040237427, + "eval_runtime": 37.2204, + "eval_samples_per_second": 26.867, + "eval_steps_per_second": 1.128, + "step": 450 + }, + { + "epoch": 0.5694444444444444, + "grad_norm": 2.93575119972229, + "learning_rate": 4.534611786716558e-06, + "loss": 1.615, + "step": 451 + }, + { + "epoch": 0.5707070707070707, + "grad_norm": 2.847141742706299, + "learning_rate": 4.532273152478953e-06, + "loss": 1.6474, + "step": 452 + }, + { + "epoch": 0.571969696969697, + "grad_norm": 3.0299603939056396, + "learning_rate": 4.529934518241347e-06, + "loss": 1.6424, + "step": 453 + }, + { + "epoch": 0.5732323232323232, + "grad_norm": 3.0307669639587402, + "learning_rate": 4.527595884003742e-06, + "loss": 1.6473, + "step": 454 + }, + { + "epoch": 0.5744949494949495, + "grad_norm": 2.864269971847534, + "learning_rate": 4.525257249766137e-06, + "loss": 1.6631, + "step": 455 + }, + { + "epoch": 0.5757575757575758, + "grad_norm": 2.606637716293335, + "learning_rate": 4.522918615528531e-06, + "loss": 1.6118, + "step": 456 + }, + { + "epoch": 0.577020202020202, + "grad_norm": 3.127891778945923, + "learning_rate": 4.520579981290926e-06, + "loss": 1.6286, + "step": 457 + }, + { + "epoch": 0.5782828282828283, + "grad_norm": 2.8809893131256104, + "learning_rate": 4.518241347053321e-06, + "loss": 1.6158, + "step": 458 + }, + { + "epoch": 0.5795454545454546, + "grad_norm": 3.0662436485290527, + "learning_rate": 4.515902712815716e-06, + "loss": 1.7336, + "step": 459 + }, + { + "epoch": 0.5808080808080808, + "grad_norm": 3.0795772075653076, + "learning_rate": 4.513564078578111e-06, + "loss": 1.6974, + "step": 460 + }, + { + "epoch": 0.5808080808080808, + "eval_loss": 1.7981404066085815, + "eval_runtime": 37.299, + "eval_samples_per_second": 26.81, + "eval_steps_per_second": 1.126, + "step": 460 + }, + { + "epoch": 0.5820707070707071, + "grad_norm": 3.091688871383667, + "learning_rate": 4.511225444340506e-06, + "loss": 1.6803, + "step": 461 + }, + { + "epoch": 0.5833333333333334, + "grad_norm": 2.5597071647644043, + "learning_rate": 4.5088868101029005e-06, + "loss": 1.6104, + "step": 462 + }, + { + "epoch": 0.5845959595959596, + "grad_norm": 2.849806070327759, + "learning_rate": 4.506548175865295e-06, + "loss": 1.6147, + "step": 463 + }, + { + "epoch": 0.5858585858585859, + "grad_norm": 2.9791204929351807, + "learning_rate": 4.50420954162769e-06, + "loss": 1.6376, + "step": 464 + }, + { + "epoch": 0.5871212121212122, + "grad_norm": 2.8582305908203125, + "learning_rate": 4.501870907390085e-06, + "loss": 1.6831, + "step": 465 + }, + { + "epoch": 0.5883838383838383, + "grad_norm": 2.8445184230804443, + "learning_rate": 4.499532273152479e-06, + "loss": 1.6463, + "step": 466 + }, + { + "epoch": 0.5896464646464646, + "grad_norm": 2.858168125152588, + "learning_rate": 4.497193638914874e-06, + "loss": 1.6662, + "step": 467 + }, + { + "epoch": 0.5909090909090909, + "grad_norm": 2.829702854156494, + "learning_rate": 4.494855004677269e-06, + "loss": 1.6058, + "step": 468 + }, + { + "epoch": 0.5921717171717171, + "grad_norm": 2.9292726516723633, + "learning_rate": 4.492516370439663e-06, + "loss": 1.6341, + "step": 469 + }, + { + "epoch": 0.5934343434343434, + "grad_norm": 2.9006292819976807, + "learning_rate": 4.490177736202058e-06, + "loss": 1.5726, + "step": 470 + }, + { + "epoch": 0.5934343434343434, + "eval_loss": 1.7953916788101196, + "eval_runtime": 37.5058, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 470 + }, + { + "epoch": 0.5946969696969697, + "grad_norm": 2.8116230964660645, + "learning_rate": 4.487839101964453e-06, + "loss": 1.6083, + "step": 471 + }, + { + "epoch": 0.5959595959595959, + "grad_norm": 2.540607452392578, + "learning_rate": 4.485500467726848e-06, + "loss": 1.6208, + "step": 472 + }, + { + "epoch": 0.5972222222222222, + "grad_norm": 2.7339096069335938, + "learning_rate": 4.4831618334892425e-06, + "loss": 1.6839, + "step": 473 + }, + { + "epoch": 0.5984848484848485, + "grad_norm": 2.9094762802124023, + "learning_rate": 4.480823199251638e-06, + "loss": 1.6962, + "step": 474 + }, + { + "epoch": 0.5997474747474747, + "grad_norm": 2.7733027935028076, + "learning_rate": 4.478484565014032e-06, + "loss": 1.6252, + "step": 475 + }, + { + "epoch": 0.601010101010101, + "grad_norm": 2.5709428787231445, + "learning_rate": 4.4761459307764274e-06, + "loss": 1.5896, + "step": 476 + }, + { + "epoch": 0.6022727272727273, + "grad_norm": 2.8358278274536133, + "learning_rate": 4.473807296538822e-06, + "loss": 1.6247, + "step": 477 + }, + { + "epoch": 0.6035353535353535, + "grad_norm": 2.7504897117614746, + "learning_rate": 4.471468662301216e-06, + "loss": 1.7138, + "step": 478 + }, + { + "epoch": 0.6047979797979798, + "grad_norm": 2.6965620517730713, + "learning_rate": 4.4691300280636115e-06, + "loss": 1.6054, + "step": 479 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.9401891231536865, + "learning_rate": 4.466791393826006e-06, + "loss": 1.6172, + "step": 480 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.7946635484695435, + "eval_runtime": 37.5366, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 480 + }, + { + "epoch": 0.6073232323232324, + "grad_norm": 2.71535587310791, + "learning_rate": 4.4644527595884e-06, + "loss": 1.61, + "step": 481 + }, + { + "epoch": 0.6085858585858586, + "grad_norm": 2.7910654544830322, + "learning_rate": 4.462114125350796e-06, + "loss": 1.6193, + "step": 482 + }, + { + "epoch": 0.6098484848484849, + "grad_norm": 2.809241533279419, + "learning_rate": 4.45977549111319e-06, + "loss": 1.602, + "step": 483 + }, + { + "epoch": 0.6111111111111112, + "grad_norm": 2.7152137756347656, + "learning_rate": 4.457436856875585e-06, + "loss": 1.6516, + "step": 484 + }, + { + "epoch": 0.6123737373737373, + "grad_norm": 2.7653465270996094, + "learning_rate": 4.45509822263798e-06, + "loss": 1.5904, + "step": 485 + }, + { + "epoch": 0.6136363636363636, + "grad_norm": 2.729876756668091, + "learning_rate": 4.452759588400374e-06, + "loss": 1.6259, + "step": 486 + }, + { + "epoch": 0.61489898989899, + "grad_norm": 2.6290180683135986, + "learning_rate": 4.4504209541627694e-06, + "loss": 1.5796, + "step": 487 + }, + { + "epoch": 0.6161616161616161, + "grad_norm": 2.939884662628174, + "learning_rate": 4.448082319925164e-06, + "loss": 1.6774, + "step": 488 + }, + { + "epoch": 0.6174242424242424, + "grad_norm": 2.7554702758789062, + "learning_rate": 4.445743685687559e-06, + "loss": 1.6226, + "step": 489 + }, + { + "epoch": 0.6186868686868687, + "grad_norm": 2.890458106994629, + "learning_rate": 4.4434050514499535e-06, + "loss": 1.592, + "step": 490 + }, + { + "epoch": 0.6186868686868687, + "eval_loss": 1.7868437767028809, + "eval_runtime": 37.5267, + "eval_samples_per_second": 26.648, + "eval_steps_per_second": 1.119, + "step": 490 + }, + { + "epoch": 0.6199494949494949, + "grad_norm": 2.757178783416748, + "learning_rate": 4.441066417212349e-06, + "loss": 1.6297, + "step": 491 + }, + { + "epoch": 0.6212121212121212, + "grad_norm": 2.7179954051971436, + "learning_rate": 4.438727782974743e-06, + "loss": 1.5685, + "step": 492 + }, + { + "epoch": 0.6224747474747475, + "grad_norm": 2.824514389038086, + "learning_rate": 4.436389148737138e-06, + "loss": 1.6625, + "step": 493 + }, + { + "epoch": 0.6237373737373737, + "grad_norm": 3.0062615871429443, + "learning_rate": 4.434050514499533e-06, + "loss": 1.5889, + "step": 494 + }, + { + "epoch": 0.625, + "grad_norm": 2.962176561355591, + "learning_rate": 4.431711880261927e-06, + "loss": 1.6813, + "step": 495 + }, + { + "epoch": 0.6262626262626263, + "grad_norm": 2.7531204223632812, + "learning_rate": 4.429373246024323e-06, + "loss": 1.6152, + "step": 496 + }, + { + "epoch": 0.6275252525252525, + "grad_norm": 2.764021873474121, + "learning_rate": 4.427034611786717e-06, + "loss": 1.6164, + "step": 497 + }, + { + "epoch": 0.6287878787878788, + "grad_norm": 2.836826801300049, + "learning_rate": 4.4246959775491114e-06, + "loss": 1.6192, + "step": 498 + }, + { + "epoch": 0.6300505050505051, + "grad_norm": 2.871220827102661, + "learning_rate": 4.422357343311507e-06, + "loss": 1.6096, + "step": 499 + }, + { + "epoch": 0.6313131313131313, + "grad_norm": 2.8129167556762695, + "learning_rate": 4.420018709073901e-06, + "loss": 1.6674, + "step": 500 + }, + { + "epoch": 0.6313131313131313, + "eval_loss": 1.785156011581421, + "eval_runtime": 37.2971, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 500 + }, + { + "epoch": 0.6325757575757576, + "grad_norm": 2.775108814239502, + "learning_rate": 4.4176800748362955e-06, + "loss": 1.5851, + "step": 501 + }, + { + "epoch": 0.6338383838383839, + "grad_norm": 2.7775518894195557, + "learning_rate": 4.415341440598691e-06, + "loss": 1.6839, + "step": 502 + }, + { + "epoch": 0.63510101010101, + "grad_norm": 2.7997918128967285, + "learning_rate": 4.413002806361085e-06, + "loss": 1.5849, + "step": 503 + }, + { + "epoch": 0.6363636363636364, + "grad_norm": 2.91103196144104, + "learning_rate": 4.4106641721234805e-06, + "loss": 1.6369, + "step": 504 + }, + { + "epoch": 0.6376262626262627, + "grad_norm": 2.6800882816314697, + "learning_rate": 4.408325537885875e-06, + "loss": 1.6049, + "step": 505 + }, + { + "epoch": 0.6388888888888888, + "grad_norm": 2.7451021671295166, + "learning_rate": 4.40598690364827e-06, + "loss": 1.658, + "step": 506 + }, + { + "epoch": 0.6401515151515151, + "grad_norm": 2.794813632965088, + "learning_rate": 4.403648269410665e-06, + "loss": 1.6015, + "step": 507 + }, + { + "epoch": 0.6414141414141414, + "grad_norm": 2.769373655319214, + "learning_rate": 4.401309635173059e-06, + "loss": 1.6217, + "step": 508 + }, + { + "epoch": 0.6426767676767676, + "grad_norm": 2.9081523418426514, + "learning_rate": 4.398971000935454e-06, + "loss": 1.6235, + "step": 509 + }, + { + "epoch": 0.6439393939393939, + "grad_norm": 2.707618236541748, + "learning_rate": 4.396632366697849e-06, + "loss": 1.6003, + "step": 510 + }, + { + "epoch": 0.6439393939393939, + "eval_loss": 1.7876425981521606, + "eval_runtime": 36.9832, + "eval_samples_per_second": 27.039, + "eval_steps_per_second": 1.136, + "step": 510 + }, + { + "epoch": 0.6452020202020202, + "grad_norm": 2.7923099994659424, + "learning_rate": 4.394293732460244e-06, + "loss": 1.5647, + "step": 511 + }, + { + "epoch": 0.6464646464646465, + "grad_norm": 2.8449175357818604, + "learning_rate": 4.391955098222638e-06, + "loss": 1.6456, + "step": 512 + }, + { + "epoch": 0.6477272727272727, + "grad_norm": 2.755856990814209, + "learning_rate": 4.389616463985033e-06, + "loss": 1.6962, + "step": 513 + }, + { + "epoch": 0.648989898989899, + "grad_norm": 2.5578861236572266, + "learning_rate": 4.387277829747428e-06, + "loss": 1.6097, + "step": 514 + }, + { + "epoch": 0.6502525252525253, + "grad_norm": 2.8804502487182617, + "learning_rate": 4.3849391955098225e-06, + "loss": 1.6212, + "step": 515 + }, + { + "epoch": 0.6515151515151515, + "grad_norm": 2.684215545654297, + "learning_rate": 4.382600561272217e-06, + "loss": 1.5529, + "step": 516 + }, + { + "epoch": 0.6527777777777778, + "grad_norm": 2.9128026962280273, + "learning_rate": 4.380261927034612e-06, + "loss": 1.6424, + "step": 517 + }, + { + "epoch": 0.6540404040404041, + "grad_norm": 2.733415126800537, + "learning_rate": 4.377923292797007e-06, + "loss": 1.6201, + "step": 518 + }, + { + "epoch": 0.6553030303030303, + "grad_norm": 2.7211062908172607, + "learning_rate": 4.375584658559401e-06, + "loss": 1.5833, + "step": 519 + }, + { + "epoch": 0.6565656565656566, + "grad_norm": 2.7308316230773926, + "learning_rate": 4.373246024321796e-06, + "loss": 1.6288, + "step": 520 + }, + { + "epoch": 0.6565656565656566, + "eval_loss": 1.7787725925445557, + "eval_runtime": 37.1089, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 520 + }, + { + "epoch": 0.6578282828282829, + "grad_norm": 2.760362148284912, + "learning_rate": 4.3709073900841915e-06, + "loss": 1.5814, + "step": 521 + }, + { + "epoch": 0.6590909090909091, + "grad_norm": 2.7625153064727783, + "learning_rate": 4.368568755846586e-06, + "loss": 1.6239, + "step": 522 + }, + { + "epoch": 0.6603535353535354, + "grad_norm": 2.715710401535034, + "learning_rate": 4.366230121608981e-06, + "loss": 1.6069, + "step": 523 + }, + { + "epoch": 0.6616161616161617, + "grad_norm": 2.7753846645355225, + "learning_rate": 4.363891487371376e-06, + "loss": 1.6178, + "step": 524 + }, + { + "epoch": 0.6628787878787878, + "grad_norm": 2.650716543197632, + "learning_rate": 4.36155285313377e-06, + "loss": 1.6534, + "step": 525 + }, + { + "epoch": 0.6641414141414141, + "grad_norm": 2.573969602584839, + "learning_rate": 4.359214218896165e-06, + "loss": 1.6721, + "step": 526 + }, + { + "epoch": 0.6654040404040404, + "grad_norm": 2.9017715454101562, + "learning_rate": 4.35687558465856e-06, + "loss": 1.6273, + "step": 527 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.691643714904785, + "learning_rate": 4.354536950420954e-06, + "loss": 1.6199, + "step": 528 + }, + { + "epoch": 0.6679292929292929, + "grad_norm": 2.818662166595459, + "learning_rate": 4.3521983161833494e-06, + "loss": 1.5758, + "step": 529 + }, + { + "epoch": 0.6691919191919192, + "grad_norm": 3.017738103866577, + "learning_rate": 4.349859681945744e-06, + "loss": 1.6437, + "step": 530 + }, + { + "epoch": 0.6691919191919192, + "eval_loss": 1.7835458517074585, + "eval_runtime": 37.1359, + "eval_samples_per_second": 26.928, + "eval_steps_per_second": 1.131, + "step": 530 + }, + { + "epoch": 0.6704545454545454, + "grad_norm": 2.8027284145355225, + "learning_rate": 4.347521047708138e-06, + "loss": 1.6688, + "step": 531 + }, + { + "epoch": 0.6717171717171717, + "grad_norm": 2.6642649173736572, + "learning_rate": 4.3451824134705335e-06, + "loss": 1.6327, + "step": 532 + }, + { + "epoch": 0.672979797979798, + "grad_norm": 2.7406396865844727, + "learning_rate": 4.342843779232928e-06, + "loss": 1.6297, + "step": 533 + }, + { + "epoch": 0.6742424242424242, + "grad_norm": 2.758538007736206, + "learning_rate": 4.340505144995323e-06, + "loss": 1.5276, + "step": 534 + }, + { + "epoch": 0.6755050505050505, + "grad_norm": 2.815817356109619, + "learning_rate": 4.3381665107577185e-06, + "loss": 1.6192, + "step": 535 + }, + { + "epoch": 0.6767676767676768, + "grad_norm": 2.8416755199432373, + "learning_rate": 4.335827876520113e-06, + "loss": 1.5966, + "step": 536 + }, + { + "epoch": 0.678030303030303, + "grad_norm": 3.014401435852051, + "learning_rate": 4.333489242282507e-06, + "loss": 1.6104, + "step": 537 + }, + { + "epoch": 0.6792929292929293, + "grad_norm": 2.731356143951416, + "learning_rate": 4.331150608044903e-06, + "loss": 1.6177, + "step": 538 + }, + { + "epoch": 0.6805555555555556, + "grad_norm": 2.633310317993164, + "learning_rate": 4.328811973807297e-06, + "loss": 1.5004, + "step": 539 + }, + { + "epoch": 0.6818181818181818, + "grad_norm": 2.977476119995117, + "learning_rate": 4.3264733395696914e-06, + "loss": 1.6171, + "step": 540 + }, + { + "epoch": 0.6818181818181818, + "eval_loss": 1.7790141105651855, + "eval_runtime": 37.12, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 540 + }, + { + "epoch": 0.6830808080808081, + "grad_norm": 2.769422769546509, + "learning_rate": 4.324134705332087e-06, + "loss": 1.5591, + "step": 541 + }, + { + "epoch": 0.6843434343434344, + "grad_norm": 3.088426351547241, + "learning_rate": 4.321796071094481e-06, + "loss": 1.5603, + "step": 542 + }, + { + "epoch": 0.6856060606060606, + "grad_norm": 2.6784212589263916, + "learning_rate": 4.3194574368568755e-06, + "loss": 1.6061, + "step": 543 + }, + { + "epoch": 0.6868686868686869, + "grad_norm": 3.209821939468384, + "learning_rate": 4.317118802619271e-06, + "loss": 1.585, + "step": 544 + }, + { + "epoch": 0.6881313131313131, + "grad_norm": 2.940791368484497, + "learning_rate": 4.314780168381665e-06, + "loss": 1.6092, + "step": 545 + }, + { + "epoch": 0.6893939393939394, + "grad_norm": 2.679884672164917, + "learning_rate": 4.3124415341440605e-06, + "loss": 1.6207, + "step": 546 + }, + { + "epoch": 0.6906565656565656, + "grad_norm": 2.853809118270874, + "learning_rate": 4.310102899906455e-06, + "loss": 1.5674, + "step": 547 + }, + { + "epoch": 0.6919191919191919, + "grad_norm": 2.693838119506836, + "learning_rate": 4.307764265668849e-06, + "loss": 1.5886, + "step": 548 + }, + { + "epoch": 0.6931818181818182, + "grad_norm": 2.871680736541748, + "learning_rate": 4.305425631431245e-06, + "loss": 1.6309, + "step": 549 + }, + { + "epoch": 0.6944444444444444, + "grad_norm": 2.9908292293548584, + "learning_rate": 4.30308699719364e-06, + "loss": 1.5577, + "step": 550 + }, + { + "epoch": 0.6944444444444444, + "eval_loss": 1.7790372371673584, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 550 + }, + { + "epoch": 0.6957070707070707, + "grad_norm": 2.883481740951538, + "learning_rate": 4.300748362956034e-06, + "loss": 1.5749, + "step": 551 + }, + { + "epoch": 0.696969696969697, + "grad_norm": 2.7523458003997803, + "learning_rate": 4.298409728718429e-06, + "loss": 1.624, + "step": 552 + }, + { + "epoch": 0.6982323232323232, + "grad_norm": 2.69509220123291, + "learning_rate": 4.296071094480824e-06, + "loss": 1.5856, + "step": 553 + }, + { + "epoch": 0.6994949494949495, + "grad_norm": 2.6340222358703613, + "learning_rate": 4.293732460243218e-06, + "loss": 1.5857, + "step": 554 + }, + { + "epoch": 0.7007575757575758, + "grad_norm": 2.4998607635498047, + "learning_rate": 4.291393826005613e-06, + "loss": 1.6038, + "step": 555 + }, + { + "epoch": 0.702020202020202, + "grad_norm": 2.7913858890533447, + "learning_rate": 4.289055191768008e-06, + "loss": 1.6411, + "step": 556 + }, + { + "epoch": 0.7032828282828283, + "grad_norm": 2.7149479389190674, + "learning_rate": 4.2867165575304025e-06, + "loss": 1.592, + "step": 557 + }, + { + "epoch": 0.7045454545454546, + "grad_norm": 2.7187132835388184, + "learning_rate": 4.284377923292797e-06, + "loss": 1.5413, + "step": 558 + }, + { + "epoch": 0.7058080808080808, + "grad_norm": 2.8913674354553223, + "learning_rate": 4.282039289055192e-06, + "loss": 1.6105, + "step": 559 + }, + { + "epoch": 0.7070707070707071, + "grad_norm": 2.8470897674560547, + "learning_rate": 4.279700654817587e-06, + "loss": 1.6481, + "step": 560 + }, + { + "epoch": 0.7070707070707071, + "eval_loss": 1.7711386680603027, + "eval_runtime": 37.1243, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 560 + }, + { + "epoch": 0.7083333333333334, + "grad_norm": 2.7525687217712402, + "learning_rate": 4.277362020579982e-06, + "loss": 1.5749, + "step": 561 + }, + { + "epoch": 0.7095959595959596, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.637, + "step": 562 + }, + { + "epoch": 0.7108585858585859, + "grad_norm": 2.6935207843780518, + "learning_rate": 4.275023386342376e-06, + "loss": 1.5903, + "step": 563 + }, + { + "epoch": 0.7121212121212122, + "grad_norm": 2.992992401123047, + "learning_rate": 4.272684752104771e-06, + "loss": 1.6654, + "step": 564 + }, + { + "epoch": 0.7133838383838383, + "grad_norm": 2.7314348220825195, + "learning_rate": 4.270346117867166e-06, + "loss": 1.6775, + "step": 565 + }, + { + "epoch": 0.7146464646464646, + "grad_norm": 2.6952013969421387, + "learning_rate": 4.26800748362956e-06, + "loss": 1.5936, + "step": 566 + }, + { + "epoch": 0.7159090909090909, + "grad_norm": 2.8244829177856445, + "learning_rate": 4.265668849391956e-06, + "loss": 1.6228, + "step": 567 + }, + { + "epoch": 0.7171717171717171, + "grad_norm": 2.9110143184661865, + "learning_rate": 4.26333021515435e-06, + "loss": 1.6016, + "step": 568 + }, + { + "epoch": 0.7184343434343434, + "grad_norm": 2.7953436374664307, + "learning_rate": 4.260991580916745e-06, + "loss": 1.611, + "step": 569 + }, + { + "epoch": 0.7196969696969697, + "grad_norm": 2.854447364807129, + "learning_rate": 4.25865294667914e-06, + "loss": 1.6365, + "step": 570 + }, + { + "epoch": 0.7196969696969697, + "eval_loss": 1.7711069583892822, + "eval_runtime": 37.1374, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 570 + }, + { + "epoch": 0.7209595959595959, + "grad_norm": 2.7885239124298096, + "learning_rate": 4.256314312441534e-06, + "loss": 1.6106, + "step": 571 + }, + { + "epoch": 0.7222222222222222, + "grad_norm": 2.8012993335723877, + "learning_rate": 4.2539756782039295e-06, + "loss": 1.6081, + "step": 572 + }, + { + "epoch": 0.7234848484848485, + "grad_norm": 2.9223906993865967, + "learning_rate": 4.251637043966324e-06, + "loss": 1.5836, + "step": 573 + }, + { + "epoch": 0.7247474747474747, + "grad_norm": 2.885830879211426, + "learning_rate": 4.249298409728719e-06, + "loss": 1.5616, + "step": 574 + }, + { + "epoch": 0.726010101010101, + "grad_norm": 2.687372922897339, + "learning_rate": 4.2469597754911136e-06, + "loss": 1.5873, + "step": 575 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 3.0156168937683105, + "learning_rate": 4.244621141253508e-06, + "loss": 1.5763, + "step": 576 + }, + { + "epoch": 0.7285353535353535, + "grad_norm": 3.061842918395996, + "learning_rate": 4.242282507015903e-06, + "loss": 1.6148, + "step": 577 + }, + { + "epoch": 0.7297979797979798, + "grad_norm": 2.7467715740203857, + "learning_rate": 4.239943872778298e-06, + "loss": 1.577, + "step": 578 + }, + { + "epoch": 0.7310606060606061, + "grad_norm": 2.669501543045044, + "learning_rate": 4.237605238540692e-06, + "loss": 1.5653, + "step": 579 + }, + { + "epoch": 0.7323232323232324, + "grad_norm": 2.874349355697632, + "learning_rate": 4.235266604303087e-06, + "loss": 1.6523, + "step": 580 + }, + { + "epoch": 0.7323232323232324, + "eval_loss": 1.7680890560150146, + "eval_runtime": 37.2931, + "eval_samples_per_second": 26.815, + "eval_steps_per_second": 1.126, + "step": 580 + }, + { + "epoch": 0.7335858585858586, + "grad_norm": 2.7468504905700684, + "learning_rate": 4.232927970065482e-06, + "loss": 1.5935, + "step": 581 + }, + { + "epoch": 0.7348484848484849, + "grad_norm": 2.936697244644165, + "learning_rate": 4.230589335827877e-06, + "loss": 1.6335, + "step": 582 + }, + { + "epoch": 0.7361111111111112, + "grad_norm": 2.6878974437713623, + "learning_rate": 4.2282507015902715e-06, + "loss": 1.5699, + "step": 583 + }, + { + "epoch": 0.7373737373737373, + "grad_norm": 3.3578853607177734, + "learning_rate": 4.225912067352667e-06, + "loss": 1.5979, + "step": 584 + }, + { + "epoch": 0.7386363636363636, + "grad_norm": 2.8441176414489746, + "learning_rate": 4.223573433115061e-06, + "loss": 1.5736, + "step": 585 + }, + { + "epoch": 0.73989898989899, + "grad_norm": 2.7186129093170166, + "learning_rate": 4.221234798877456e-06, + "loss": 1.636, + "step": 586 + }, + { + "epoch": 0.7411616161616161, + "grad_norm": 2.940002918243408, + "learning_rate": 4.218896164639851e-06, + "loss": 1.5721, + "step": 587 + }, + { + "epoch": 0.7424242424242424, + "grad_norm": 2.676284074783325, + "learning_rate": 4.216557530402245e-06, + "loss": 1.5972, + "step": 588 + }, + { + "epoch": 0.7436868686868687, + "grad_norm": 2.9176571369171143, + "learning_rate": 4.2142188961646405e-06, + "loss": 1.6045, + "step": 589 + }, + { + "epoch": 0.7449494949494949, + "grad_norm": 2.7716288566589355, + "learning_rate": 4.211880261927035e-06, + "loss": 1.5988, + "step": 590 + }, + { + "epoch": 0.7449494949494949, + "eval_loss": 1.775631070137024, + "eval_runtime": 37.4113, + "eval_samples_per_second": 26.73, + "eval_steps_per_second": 1.123, + "step": 590 + }, + { + "epoch": 0.7462121212121212, + "grad_norm": 2.7201945781707764, + "learning_rate": 4.209541627689429e-06, + "loss": 1.5767, + "step": 591 + }, + { + "epoch": 0.7474747474747475, + "grad_norm": 2.838362455368042, + "learning_rate": 4.207202993451825e-06, + "loss": 1.5793, + "step": 592 + }, + { + "epoch": 0.7487373737373737, + "grad_norm": 2.6511716842651367, + "learning_rate": 4.204864359214219e-06, + "loss": 1.5776, + "step": 593 + }, + { + "epoch": 0.75, + "grad_norm": 2.7062957286834717, + "learning_rate": 4.2025257249766135e-06, + "loss": 1.5593, + "step": 594 + }, + { + "epoch": 0.7512626262626263, + "grad_norm": 2.808570384979248, + "learning_rate": 4.200187090739009e-06, + "loss": 1.6249, + "step": 595 + }, + { + "epoch": 0.7525252525252525, + "grad_norm": 2.600278377532959, + "learning_rate": 4.197848456501403e-06, + "loss": 1.6171, + "step": 596 + }, + { + "epoch": 0.7537878787878788, + "grad_norm": 2.9669382572174072, + "learning_rate": 4.195509822263798e-06, + "loss": 1.608, + "step": 597 + }, + { + "epoch": 0.7550505050505051, + "grad_norm": 2.6576340198516846, + "learning_rate": 4.193171188026194e-06, + "loss": 1.5986, + "step": 598 + }, + { + "epoch": 0.7563131313131313, + "grad_norm": 2.787060022354126, + "learning_rate": 4.190832553788588e-06, + "loss": 1.548, + "step": 599 + }, + { + "epoch": 0.7575757575757576, + "grad_norm": 2.854971408843994, + "learning_rate": 4.1884939195509825e-06, + "loss": 1.6421, + "step": 600 + }, + { + "epoch": 0.7575757575757576, + "eval_loss": 1.7721046209335327, + "eval_runtime": 37.4471, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 600 + }, + { + "epoch": 0.7588383838383839, + "grad_norm": 2.7417967319488525, + "learning_rate": 4.186155285313378e-06, + "loss": 1.6399, + "step": 601 + }, + { + "epoch": 0.76010101010101, + "grad_norm": 3.0036182403564453, + "learning_rate": 4.183816651075772e-06, + "loss": 1.5953, + "step": 602 + }, + { + "epoch": 0.7613636363636364, + "grad_norm": 2.502861261367798, + "learning_rate": 4.181478016838167e-06, + "loss": 1.5284, + "step": 603 + }, + { + "epoch": 0.7626262626262627, + "grad_norm": 2.7262094020843506, + "learning_rate": 4.179139382600562e-06, + "loss": 1.619, + "step": 604 + }, + { + "epoch": 0.7638888888888888, + "grad_norm": 2.8748724460601807, + "learning_rate": 4.176800748362956e-06, + "loss": 1.5625, + "step": 605 + }, + { + "epoch": 0.7651515151515151, + "grad_norm": 2.7272109985351562, + "learning_rate": 4.174462114125351e-06, + "loss": 1.5935, + "step": 606 + }, + { + "epoch": 0.7664141414141414, + "grad_norm": 2.7930479049682617, + "learning_rate": 4.172123479887746e-06, + "loss": 1.61, + "step": 607 + }, + { + "epoch": 0.7676767676767676, + "grad_norm": 2.952056407928467, + "learning_rate": 4.16978484565014e-06, + "loss": 1.6369, + "step": 608 + }, + { + "epoch": 0.7689393939393939, + "grad_norm": 2.979564905166626, + "learning_rate": 4.167446211412535e-06, + "loss": 1.5748, + "step": 609 + }, + { + "epoch": 0.7702020202020202, + "grad_norm": 2.7945773601531982, + "learning_rate": 4.16510757717493e-06, + "loss": 1.6096, + "step": 610 + }, + { + "epoch": 0.7702020202020202, + "eval_loss": 1.7568180561065674, + "eval_runtime": 36.969, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 610 + }, + { + "epoch": 0.7714646464646465, + "grad_norm": 2.9217135906219482, + "learning_rate": 4.1627689429373245e-06, + "loss": 1.5902, + "step": 611 + }, + { + "epoch": 0.7727272727272727, + "grad_norm": 2.8184452056884766, + "learning_rate": 4.16043030869972e-06, + "loss": 1.5775, + "step": 612 + }, + { + "epoch": 0.773989898989899, + "grad_norm": 2.7121212482452393, + "learning_rate": 4.158091674462115e-06, + "loss": 1.5778, + "step": 613 + }, + { + "epoch": 0.7752525252525253, + "grad_norm": 2.7600955963134766, + "learning_rate": 4.1557530402245095e-06, + "loss": 1.6145, + "step": 614 + }, + { + "epoch": 0.7765151515151515, + "grad_norm": 2.6388845443725586, + "learning_rate": 4.153414405986904e-06, + "loss": 1.6475, + "step": 615 + }, + { + "epoch": 0.7777777777777778, + "grad_norm": 2.684379816055298, + "learning_rate": 4.151075771749299e-06, + "loss": 1.6187, + "step": 616 + }, + { + "epoch": 0.7790404040404041, + "grad_norm": 2.6907055377960205, + "learning_rate": 4.1487371375116936e-06, + "loss": 1.6108, + "step": 617 + }, + { + "epoch": 0.7803030303030303, + "grad_norm": 2.8458995819091797, + "learning_rate": 4.146398503274088e-06, + "loss": 1.5596, + "step": 618 + }, + { + "epoch": 0.7815656565656566, + "grad_norm": 2.68843674659729, + "learning_rate": 4.144059869036483e-06, + "loss": 1.5912, + "step": 619 + }, + { + "epoch": 0.7828282828282829, + "grad_norm": 2.7620413303375244, + "learning_rate": 4.141721234798878e-06, + "loss": 1.627, + "step": 620 + }, + { + "epoch": 0.7828282828282829, + "eval_loss": 1.757076621055603, + "eval_runtime": 37.0489, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 620 + }, + { + "epoch": 0.7840909090909091, + "grad_norm": 2.7084949016571045, + "learning_rate": 4.139382600561272e-06, + "loss": 1.5249, + "step": 621 + }, + { + "epoch": 0.7853535353535354, + "grad_norm": 3.0410406589508057, + "learning_rate": 4.137043966323667e-06, + "loss": 1.6458, + "step": 622 + }, + { + "epoch": 0.7866161616161617, + "grad_norm": 2.8445630073547363, + "learning_rate": 4.134705332086062e-06, + "loss": 1.6404, + "step": 623 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 3.0764353275299072, + "learning_rate": 4.132366697848457e-06, + "loss": 1.5041, + "step": 624 + }, + { + "epoch": 0.7891414141414141, + "grad_norm": 2.626544237136841, + "learning_rate": 4.1300280636108515e-06, + "loss": 1.6145, + "step": 625 + }, + { + "epoch": 0.7904040404040404, + "grad_norm": 2.9735796451568604, + "learning_rate": 4.127689429373246e-06, + "loss": 1.5672, + "step": 626 + }, + { + "epoch": 0.7916666666666666, + "grad_norm": 2.7798562049865723, + "learning_rate": 4.125350795135641e-06, + "loss": 1.6272, + "step": 627 + }, + { + "epoch": 0.7929292929292929, + "grad_norm": 2.7438931465148926, + "learning_rate": 4.123012160898036e-06, + "loss": 1.5318, + "step": 628 + }, + { + "epoch": 0.7941919191919192, + "grad_norm": 3.103142499923706, + "learning_rate": 4.120673526660431e-06, + "loss": 1.6, + "step": 629 + }, + { + "epoch": 0.7954545454545454, + "grad_norm": 2.9056239128112793, + "learning_rate": 4.118334892422825e-06, + "loss": 1.6581, + "step": 630 + }, + { + "epoch": 0.7954545454545454, + "eval_loss": 1.7552870512008667, + "eval_runtime": 37.1223, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 630 + }, + { + "epoch": 0.7967171717171717, + "grad_norm": 2.7320070266723633, + "learning_rate": 4.1159962581852205e-06, + "loss": 1.5808, + "step": 631 + }, + { + "epoch": 0.797979797979798, + "grad_norm": 2.9518637657165527, + "learning_rate": 4.113657623947615e-06, + "loss": 1.6034, + "step": 632 + }, + { + "epoch": 0.7992424242424242, + "grad_norm": 2.7135164737701416, + "learning_rate": 4.111318989710009e-06, + "loss": 1.5684, + "step": 633 + }, + { + "epoch": 0.8005050505050505, + "grad_norm": 2.948230266571045, + "learning_rate": 4.108980355472405e-06, + "loss": 1.6378, + "step": 634 + }, + { + "epoch": 0.8017676767676768, + "grad_norm": 2.816206216812134, + "learning_rate": 4.106641721234799e-06, + "loss": 1.6118, + "step": 635 + }, + { + "epoch": 0.803030303030303, + "grad_norm": 2.8032639026641846, + "learning_rate": 4.104303086997194e-06, + "loss": 1.5627, + "step": 636 + }, + { + "epoch": 0.8042929292929293, + "grad_norm": 3.032348871231079, + "learning_rate": 4.101964452759589e-06, + "loss": 1.6192, + "step": 637 + }, + { + "epoch": 0.8055555555555556, + "grad_norm": 2.9512710571289062, + "learning_rate": 4.099625818521983e-06, + "loss": 1.5655, + "step": 638 + }, + { + "epoch": 0.8068181818181818, + "grad_norm": 2.738973379135132, + "learning_rate": 4.097287184284378e-06, + "loss": 1.5686, + "step": 639 + }, + { + "epoch": 0.8080808080808081, + "grad_norm": 2.6510915756225586, + "learning_rate": 4.094948550046773e-06, + "loss": 1.5889, + "step": 640 + }, + { + "epoch": 0.8080808080808081, + "eval_loss": 1.762879490852356, + "eval_runtime": 37.1457, + "eval_samples_per_second": 26.921, + "eval_steps_per_second": 1.131, + "step": 640 + }, + { + "epoch": 0.8093434343434344, + "grad_norm": 2.6933884620666504, + "learning_rate": 4.092609915809167e-06, + "loss": 1.5436, + "step": 641 + }, + { + "epoch": 0.8106060606060606, + "grad_norm": 2.703784227371216, + "learning_rate": 4.0902712815715625e-06, + "loss": 1.5869, + "step": 642 + }, + { + "epoch": 0.8118686868686869, + "grad_norm": 2.7742321491241455, + "learning_rate": 4.087932647333958e-06, + "loss": 1.5779, + "step": 643 + }, + { + "epoch": 0.8131313131313131, + "grad_norm": 2.836832284927368, + "learning_rate": 4.085594013096352e-06, + "loss": 1.5511, + "step": 644 + }, + { + "epoch": 0.8143939393939394, + "grad_norm": 2.9212028980255127, + "learning_rate": 4.083255378858747e-06, + "loss": 1.5604, + "step": 645 + }, + { + "epoch": 0.8156565656565656, + "grad_norm": 2.8608527183532715, + "learning_rate": 4.080916744621142e-06, + "loss": 1.5975, + "step": 646 + }, + { + "epoch": 0.8169191919191919, + "grad_norm": 2.8246703147888184, + "learning_rate": 4.078578110383536e-06, + "loss": 1.5962, + "step": 647 + }, + { + "epoch": 0.8181818181818182, + "grad_norm": 3.0648651123046875, + "learning_rate": 4.076239476145932e-06, + "loss": 1.6267, + "step": 648 + }, + { + "epoch": 0.8194444444444444, + "grad_norm": 2.849240303039551, + "learning_rate": 4.073900841908326e-06, + "loss": 1.6271, + "step": 649 + }, + { + "epoch": 0.8207070707070707, + "grad_norm": 2.8817555904388428, + "learning_rate": 4.07156220767072e-06, + "loss": 1.5916, + "step": 650 + }, + { + "epoch": 0.8207070707070707, + "eval_loss": 1.7502237558364868, + "eval_runtime": 37.1497, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 650 + }, + { + "epoch": 0.821969696969697, + "grad_norm": 2.5747952461242676, + "learning_rate": 4.069223573433116e-06, + "loss": 1.5661, + "step": 651 + }, + { + "epoch": 0.8232323232323232, + "grad_norm": 2.7721974849700928, + "learning_rate": 4.06688493919551e-06, + "loss": 1.5903, + "step": 652 + }, + { + "epoch": 0.8244949494949495, + "grad_norm": 2.9606690406799316, + "learning_rate": 4.0645463049579045e-06, + "loss": 1.5431, + "step": 653 + }, + { + "epoch": 0.8257575757575758, + "grad_norm": 2.941169261932373, + "learning_rate": 4.0622076707203e-06, + "loss": 1.5711, + "step": 654 + }, + { + "epoch": 0.827020202020202, + "grad_norm": 2.8408572673797607, + "learning_rate": 4.059869036482694e-06, + "loss": 1.5975, + "step": 655 + }, + { + "epoch": 0.8282828282828283, + "grad_norm": 2.619875431060791, + "learning_rate": 4.057530402245089e-06, + "loss": 1.5194, + "step": 656 + }, + { + "epoch": 0.8295454545454546, + "grad_norm": 2.910982847213745, + "learning_rate": 4.055191768007484e-06, + "loss": 1.6318, + "step": 657 + }, + { + "epoch": 0.8308080808080808, + "grad_norm": 2.747422456741333, + "learning_rate": 4.052853133769879e-06, + "loss": 1.6205, + "step": 658 + }, + { + "epoch": 0.8320707070707071, + "grad_norm": 2.6889796257019043, + "learning_rate": 4.050514499532274e-06, + "loss": 1.5537, + "step": 659 + }, + { + "epoch": 0.8333333333333334, + "grad_norm": 2.9849202632904053, + "learning_rate": 4.048175865294669e-06, + "loss": 1.5438, + "step": 660 + }, + { + "epoch": 0.8333333333333334, + "eval_loss": 1.7484922409057617, + "eval_runtime": 37.161, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.8345959595959596, + "grad_norm": 2.816930055618286, + "learning_rate": 4.045837231057063e-06, + "loss": 1.555, + "step": 661 + }, + { + "epoch": 0.8358585858585859, + "grad_norm": 2.990126371383667, + "learning_rate": 4.043498596819458e-06, + "loss": 1.6263, + "step": 662 + }, + { + "epoch": 0.8371212121212122, + "grad_norm": 2.898263692855835, + "learning_rate": 4.041159962581853e-06, + "loss": 1.5522, + "step": 663 + }, + { + "epoch": 0.8383838383838383, + "grad_norm": 2.7824151515960693, + "learning_rate": 4.038821328344247e-06, + "loss": 1.5366, + "step": 664 + }, + { + "epoch": 0.8396464646464646, + "grad_norm": 2.9943954944610596, + "learning_rate": 4.036482694106642e-06, + "loss": 1.6214, + "step": 665 + }, + { + "epoch": 0.8409090909090909, + "grad_norm": 2.628308057785034, + "learning_rate": 4.034144059869037e-06, + "loss": 1.5797, + "step": 666 + }, + { + "epoch": 0.8421717171717171, + "grad_norm": 2.7565176486968994, + "learning_rate": 4.0318054256314315e-06, + "loss": 1.5567, + "step": 667 + }, + { + "epoch": 0.8434343434343434, + "grad_norm": 2.688354969024658, + "learning_rate": 4.029466791393826e-06, + "loss": 1.5516, + "step": 668 + }, + { + "epoch": 0.8446969696969697, + "grad_norm": 2.899620771408081, + "learning_rate": 4.027128157156221e-06, + "loss": 1.6014, + "step": 669 + }, + { + "epoch": 0.8459595959595959, + "grad_norm": 2.7894718647003174, + "learning_rate": 4.024789522918616e-06, + "loss": 1.5688, + "step": 670 + }, + { + "epoch": 0.8459595959595959, + "eval_loss": 1.7486555576324463, + "eval_runtime": 37.1824, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 670 + }, + { + "epoch": 0.8472222222222222, + "grad_norm": 2.7441444396972656, + "learning_rate": 4.02245088868101e-06, + "loss": 1.5949, + "step": 671 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.7088451385498047, + "learning_rate": 4.020112254443405e-06, + "loss": 1.5878, + "step": 672 + }, + { + "epoch": 0.8497474747474747, + "grad_norm": 2.756920576095581, + "learning_rate": 4.0177736202058005e-06, + "loss": 1.6356, + "step": 673 + }, + { + "epoch": 0.851010101010101, + "grad_norm": 2.5605697631835938, + "learning_rate": 4.015434985968195e-06, + "loss": 1.5586, + "step": 674 + }, + { + "epoch": 0.8522727272727273, + "grad_norm": 2.946573257446289, + "learning_rate": 4.01309635173059e-06, + "loss": 1.5537, + "step": 675 + }, + { + "epoch": 0.8535353535353535, + "grad_norm": 2.65500545501709, + "learning_rate": 4.010757717492985e-06, + "loss": 1.5875, + "step": 676 + }, + { + "epoch": 0.8547979797979798, + "grad_norm": 2.6632118225097656, + "learning_rate": 4.008419083255379e-06, + "loss": 1.5329, + "step": 677 + }, + { + "epoch": 0.8560606060606061, + "grad_norm": 2.8909032344818115, + "learning_rate": 4.006080449017774e-06, + "loss": 1.6396, + "step": 678 + }, + { + "epoch": 0.8573232323232324, + "grad_norm": 2.8780951499938965, + "learning_rate": 4.003741814780169e-06, + "loss": 1.6189, + "step": 679 + }, + { + "epoch": 0.8585858585858586, + "grad_norm": 2.7757010459899902, + "learning_rate": 4.001403180542563e-06, + "loss": 1.5968, + "step": 680 + }, + { + "epoch": 0.8585858585858586, + "eval_loss": 1.7423925399780273, + "eval_runtime": 37.1537, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 680 + }, + { + "epoch": 0.8598484848484849, + "grad_norm": 2.6610794067382812, + "learning_rate": 3.9990645463049584e-06, + "loss": 1.5258, + "step": 681 + }, + { + "epoch": 0.8611111111111112, + "grad_norm": 2.9292140007019043, + "learning_rate": 3.996725912067353e-06, + "loss": 1.5669, + "step": 682 + }, + { + "epoch": 0.8623737373737373, + "grad_norm": 2.7885360717773438, + "learning_rate": 3.994387277829747e-06, + "loss": 1.5737, + "step": 683 + }, + { + "epoch": 0.8636363636363636, + "grad_norm": 2.7511544227600098, + "learning_rate": 3.9920486435921425e-06, + "loss": 1.5715, + "step": 684 + }, + { + "epoch": 0.86489898989899, + "grad_norm": 2.8561315536499023, + "learning_rate": 3.989710009354537e-06, + "loss": 1.5965, + "step": 685 + }, + { + "epoch": 0.8661616161616161, + "grad_norm": 2.9127237796783447, + "learning_rate": 3.987371375116932e-06, + "loss": 1.5407, + "step": 686 + }, + { + "epoch": 0.8674242424242424, + "grad_norm": 2.735577344894409, + "learning_rate": 3.985032740879327e-06, + "loss": 1.5449, + "step": 687 + }, + { + "epoch": 0.8686868686868687, + "grad_norm": 2.6626689434051514, + "learning_rate": 3.982694106641721e-06, + "loss": 1.5173, + "step": 688 + }, + { + "epoch": 0.8699494949494949, + "grad_norm": 2.901592969894409, + "learning_rate": 3.980355472404116e-06, + "loss": 1.5757, + "step": 689 + }, + { + "epoch": 0.8712121212121212, + "grad_norm": 2.8801231384277344, + "learning_rate": 3.978016838166512e-06, + "loss": 1.5633, + "step": 690 + }, + { + "epoch": 0.8712121212121212, + "eval_loss": 1.7453432083129883, + "eval_runtime": 37.4368, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 690 + }, + { + "epoch": 0.8724747474747475, + "grad_norm": 2.7463626861572266, + "learning_rate": 3.975678203928906e-06, + "loss": 1.5786, + "step": 691 + }, + { + "epoch": 0.8737373737373737, + "grad_norm": 2.659820318222046, + "learning_rate": 3.9733395696913004e-06, + "loss": 1.5923, + "step": 692 + }, + { + "epoch": 0.875, + "grad_norm": 2.759030342102051, + "learning_rate": 3.971000935453696e-06, + "loss": 1.6217, + "step": 693 + }, + { + "epoch": 0.8762626262626263, + "grad_norm": 2.689967632293701, + "learning_rate": 3.96866230121609e-06, + "loss": 1.5533, + "step": 694 + }, + { + "epoch": 0.8775252525252525, + "grad_norm": 3.1230740547180176, + "learning_rate": 3.9663236669784845e-06, + "loss": 1.6538, + "step": 695 + }, + { + "epoch": 0.8787878787878788, + "grad_norm": 2.657022476196289, + "learning_rate": 3.96398503274088e-06, + "loss": 1.5432, + "step": 696 + }, + { + "epoch": 0.8800505050505051, + "grad_norm": 2.919379711151123, + "learning_rate": 3.961646398503274e-06, + "loss": 1.5867, + "step": 697 + }, + { + "epoch": 0.8813131313131313, + "grad_norm": 2.607135057449341, + "learning_rate": 3.9593077642656695e-06, + "loss": 1.5079, + "step": 698 + }, + { + "epoch": 0.8825757575757576, + "grad_norm": 2.9978976249694824, + "learning_rate": 3.956969130028064e-06, + "loss": 1.5905, + "step": 699 + }, + { + "epoch": 0.8838383838383839, + "grad_norm": 2.8117148876190186, + "learning_rate": 3.954630495790458e-06, + "loss": 1.5588, + "step": 700 + }, + { + "epoch": 0.8838383838383839, + "eval_loss": 1.7376645803451538, + "eval_runtime": 37.5045, + "eval_samples_per_second": 26.663, + "eval_steps_per_second": 1.12, + "step": 700 + } + ], + "logging_steps": 1, + "max_steps": 2376, + "num_input_tokens_seen": 0, + "num_train_epochs": 3, + "save_steps": 100, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": false + }, + "attributes": {} + } + }, + "total_flos": 2.182116586553344e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/babylm_hop_words4_10M_seed0/runs/checkpoint-700/zero_to_fp32.py b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/babylm_hop_words4_10M_seed0/runs/checkpoint-700/zero_to_fp32.py @@ -0,0 +1,604 @@ +#!/usr/bin/env python + +# Copyright (c) Microsoft Corporation. +# SPDX-License-Identifier: Apache-2.0 + +# DeepSpeed Team + +# This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets +# copied into the top level checkpoint dir, so the user can easily do the conversion at any point in +# the future. Once extracted, the weights don't require DeepSpeed and can be used in any +# application. +# +# example: python zero_to_fp32.py . pytorch_model.bin + +import argparse +import torch +import glob +import math +import os +import re +from collections import OrderedDict +from dataclasses import dataclass + +# while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with +# DeepSpeed data structures it has to be available in the current python environment. +from deepspeed.utils import logger +from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, + FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, + FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS) + + +@dataclass +class zero_model_state: + buffers: dict() + param_shapes: dict() + shared_params: list + ds_version: int + frozen_param_shapes: dict() + frozen_param_fragments: dict() + + +debug = 0 + +# load to cpu +device = torch.device('cpu') + + +def atoi(text): + return int(text) if text.isdigit() else text + + +def natural_keys(text): + ''' + alist.sort(key=natural_keys) sorts in human order + http://nedbatchelder.com/blog/200712/human_sorting.html + (See Toothy's implementation in the comments) + ''' + return [atoi(c) for c in re.split(r'(\d+)', text)] + + +def get_model_state_file(checkpoint_dir, zero_stage): + if not os.path.isdir(checkpoint_dir): + raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist") + + # there should be only one file + if zero_stage <= 2: + file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt") + elif zero_stage == 3: + file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt") + + if not os.path.exists(file): + raise FileNotFoundError(f"can't find model states file at '{file}'") + + return file + + +def get_checkpoint_files(checkpoint_dir, glob_pattern): + # XXX: need to test that this simple glob rule works for multi-node setup too + ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys) + + if len(ckpt_files) == 0: + raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'") + + return ckpt_files + + +def get_optim_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt") + + +def get_model_state_files(checkpoint_dir): + return get_checkpoint_files(checkpoint_dir, "*_model_states.pt") + + +def parse_model_states(files): + zero_model_states = [] + for file in files: + state_dict = torch.load(file, map_location=device) + + if BUFFER_NAMES not in state_dict: + raise ValueError(f"{file} is not a model state checkpoint") + buffer_names = state_dict[BUFFER_NAMES] + if debug: + print("Found buffers:", buffer_names) + + # recover just the buffers while restoring them to fp32 if they were saved in fp16 + buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names} + param_shapes = state_dict[PARAM_SHAPES] + + # collect parameters that are included in param_shapes + param_names = [] + for s in param_shapes: + for name in s.keys(): + param_names.append(name) + + # update with frozen parameters + frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None) + if frozen_param_shapes is not None: + if debug: + print(f"Found frozen_param_shapes: {frozen_param_shapes}") + param_names += list(frozen_param_shapes.keys()) + + # handle shared params + shared_params = [[k, v] for k, v in state_dict["shared_params"].items()] + + ds_version = state_dict.get(DS_VERSION, None) + + frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None) + + z_model_state = zero_model_state(buffers=buffers, + param_shapes=param_shapes, + shared_params=shared_params, + ds_version=ds_version, + frozen_param_shapes=frozen_param_shapes, + frozen_param_fragments=frozen_param_fragments) + zero_model_states.append(z_model_state) + + return zero_model_states + + +def parse_optim_states(files, ds_checkpoint_dir): + + total_files = len(files) + state_dicts = [] + for f in files: + state_dict = torch.load(f, map_location=device) + # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights + # and also handle the case where it was already removed by another helper script + state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None) + state_dicts.append(state_dict) + + if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]: + raise ValueError(f"{files[0]} is not a zero checkpoint") + zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE] + world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT] + + # For ZeRO-2 each param group can have different partition_count as data parallelism for expert + # parameters can be different from data parallelism for non-expert parameters. So we can just + # use the max of the partition_count to get the dp world_size. + + if type(world_size) is list: + world_size = max(world_size) + + if world_size != total_files: + raise ValueError( + f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. " + "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes." + ) + + # the groups are named differently in each stage + if zero_stage <= 2: + fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS + elif zero_stage == 3: + fp32_groups_key = FP32_FLAT_GROUPS + else: + raise ValueError(f"unknown zero stage {zero_stage}") + + if zero_stage <= 2: + fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))] + elif zero_stage == 3: + # if there is more than one param group, there will be multiple flattened tensors - one + # flattened tensor per group - for simplicity merge them into a single tensor + # + # XXX: could make the script more memory efficient for when there are multiple groups - it + # will require matching the sub-lists of param_shapes for each param group flattened tensor + + fp32_flat_groups = [ + torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts)) + ] + + return zero_stage, world_size, fp32_flat_groups + + +def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters): + """ + Returns fp32 state_dict reconstructed from ds checkpoint + + Args: + - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are) + + """ + print(f"Processing zero checkpoint '{ds_checkpoint_dir}'") + + optim_files = get_optim_files(ds_checkpoint_dir) + zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir) + print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}") + + model_files = get_model_state_files(ds_checkpoint_dir) + + zero_model_states = parse_model_states(model_files) + print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}') + + if zero_stage <= 2: + return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + elif zero_stage == 3: + return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters) + + +def _zero2_merge_frozen_params(state_dict, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + frozen_param_fragments = zero_model_states[0].frozen_param_fragments + + if debug: + num_elem = sum(s.numel() for s in frozen_param_shapes.values()) + print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in frozen_param_fragments.values()]) + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + state_dict[name] = frozen_param_fragments[name] + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _has_callable(obj, fn): + attr = getattr(obj, fn, None) + return callable(attr) + + +def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + + # Reconstruction protocol: + # + # XXX: document this + + if debug: + for i in range(world_size): + for j in range(len(fp32_flat_groups[0])): + print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}") + + # XXX: memory usage doubles here (zero2) + num_param_groups = len(fp32_flat_groups[0]) + merged_single_partition_of_fp32_groups = [] + for i in range(num_param_groups): + merged_partitions = [sd[i] for sd in fp32_flat_groups] + full_single_fp32_vector = torch.cat(merged_partitions, 0) + merged_single_partition_of_fp32_groups.append(full_single_fp32_vector) + avail_numel = sum( + [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups]) + + if debug: + wanted_params = sum([len(shapes) for shapes in param_shapes]) + wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes]) + # not asserting if there is a mismatch due to possible padding + print(f"Have {avail_numel} numels to process.") + print(f"Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + total_numel = 0 + total_params = 0 + for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups): + offset = 0 + avail_numel = full_single_fp32_vector.numel() + for name, shape in shapes.items(): + + unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape) + total_numel += unpartitioned_numel + total_params += 1 + + if debug: + print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ") + state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape) + offset += unpartitioned_numel + + # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and + # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex + # paddings performed in the code it's almost impossible to predict the exact numbers w/o the + # live optimizer object, so we are checking that the numbers are within the right range + align_to = 2 * world_size + + def zero2_align(x): + return align_to * math.ceil(x / align_to) + + if debug: + print(f"original offset={offset}, avail_numel={avail_numel}") + + offset = zero2_align(offset) + avail_numel = zero2_align(avail_numel) + + if debug: + print(f"aligned offset={offset}, avail_numel={avail_numel}") + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero2_merge_frozen_params(state_dict, zero_model_states) + + _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def zero3_partitioned_param_info(unpartitioned_numel, world_size): + remainder = unpartitioned_numel % world_size + padding_numel = (world_size - remainder) if remainder else 0 + partitioned_numel = math.ceil(unpartitioned_numel / world_size) + return partitioned_numel, padding_numel + + +def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states): + if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0: + return + + if debug: + for i in range(world_size): + num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values()) + print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}') + + frozen_param_shapes = zero_model_states[0].frozen_param_shapes + wanted_params = len(frozen_param_shapes) + wanted_numel = sum(s.numel() for s in frozen_param_shapes.values()) + avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size + print(f'Frozen params: Have {avail_numel} numels to process.') + print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params') + + total_params = 0 + total_numel = 0 + for name, shape in zero_model_states[0].frozen_param_shapes.items(): + total_params += 1 + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + + param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states) + state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape) + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements") + + +def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states): + param_shapes = zero_model_states[0].param_shapes + avail_numel = fp32_flat_groups[0].numel() * world_size + # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each + # param, re-consolidating each param, while dealing with padding if any + + # merge list of dicts, preserving order + param_shapes = {k: v for d in param_shapes for k, v in d.items()} + + if debug: + for i in range(world_size): + print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}") + + wanted_params = len(param_shapes) + wanted_numel = sum(shape.numel() for shape in param_shapes.values()) + # not asserting if there is a mismatch due to possible padding + avail_numel = fp32_flat_groups[0].numel() * world_size + print(f"Trainable params: Have {avail_numel} numels to process.") + print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.") + + # params + # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support + # out-of-core computing solution + offset = 0 + total_numel = 0 + total_params = 0 + for name, shape in param_shapes.items(): + + unpartitioned_numel = shape.numel() + total_numel += unpartitioned_numel + total_params += 1 + + partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size) + + if debug: + print( + f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}" + ) + + # XXX: memory usage doubles here + state_dict[name] = torch.cat( + tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)), + 0).narrow(0, 0, unpartitioned_numel).view(shape) + offset += partitioned_numel + + offset *= world_size + + # Sanity check + if offset != avail_numel: + raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong") + + print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements") + + +def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states, + exclude_frozen_parameters): + state_dict = OrderedDict() + + # buffers + buffers = zero_model_states[0].buffers + state_dict.update(buffers) + if debug: + print(f"added {len(buffers)} buffers") + + if not exclude_frozen_parameters: + _zero3_merge_frozen_params(state_dict, world_size, zero_model_states) + + _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states) + + # recover shared parameters + for pair in zero_model_states[0].shared_params: + if pair[1] in state_dict: + state_dict[pair[0]] = state_dict[pair[1]] + + return state_dict + + +def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with + ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example + via a model hub. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + + Returns: + - pytorch ``state_dict`` + + Note: this approach may not work if your application doesn't have sufficient free CPU memory and + you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with + the checkpoint. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint + # do the training and checkpoint saving + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu + model = model.cpu() # move to cpu + model.load_state_dict(state_dict) + # submit to model hub or save the model to share with others + + In this example the ``model`` will no longer be usable in the deepspeed context of the same + application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead. + + """ + if tag is None: + latest_path = os.path.join(checkpoint_dir, 'latest') + if os.path.isfile(latest_path): + with open(latest_path, 'r') as fd: + tag = fd.read().strip() + else: + raise ValueError(f"Unable to find 'latest' file at {latest_path}") + + ds_checkpoint_dir = os.path.join(checkpoint_dir, tag) + + if not os.path.isdir(ds_checkpoint_dir): + raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist") + + return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters) + + +def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False): + """ + Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be + loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed. + + Args: + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + - ``exclude_frozen_parameters``: exclude frozen parameters + """ + + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters) + print(f"Saving fp32 state dict to {output_file}") + torch.save(state_dict, output_file) + + +def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None): + """ + 1. Put the provided model to cpu + 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` + 3. Load it into the provided model + + Args: + - ``model``: the model object to update + - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``) + - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14`` + + Returns: + - ``model`: modified model + + Make sure you have plenty of CPU memory available before you call this function. If you don't + have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it + conveniently placed for you in the checkpoint folder. + + A typical usage might be :: + + from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint + model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir) + # submit to model hub or save the model to share with others + + Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context + of the same application. i.e. you will need to re-initialize the deepspeed engine, since + ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it. + + """ + logger.info(f"Extracting fp32 weights") + state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag) + + logger.info(f"Overwriting model with fp32 weights") + model = model.cpu() + model.load_state_dict(state_dict, strict=False) + + return model + + +if __name__ == "__main__": + + parser = argparse.ArgumentParser() + parser.add_argument("checkpoint_dir", + type=str, + help="path to the desired checkpoint folder, e.g., path/checkpoint-12") + parser.add_argument( + "output_file", + type=str, + help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)") + parser.add_argument("-t", + "--tag", + type=str, + default=None, + help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1") + parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters") + parser.add_argument("-d", "--debug", action='store_true', help="enable debug") + args = parser.parse_args() + + debug = args.debug + + convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir, + args.output_file, + tag=args.tag, + exclude_frozen_parameters=args.exclude_frozen_parameters) diff --git a/babylm_reverse_full_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors b/babylm_reverse_full_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e362b81bf93cef8c6d96b08171bab63416808c70 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:584d8d3e3f82f7964955174dfe5e3b1cf117a9d859f022cfdf7fcb884856e002 +size 4965799096 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-100/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-100/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..78658c48ae9d6dfb406ed6edf6b0bb49e37625ba --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-100/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:902a2d4c0ba883647d4fd0115d71854c4a73100d5fb8352c83f61fbfaaaa79a2 +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-1100/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-1100/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..96e34ed0400316c5cb73eaa4a353b038fe447eb0 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-1100/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:663420395cec4d9f4835800b83650b40af5e83661ad0a7ac31fcd805a4ab074a +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-1300/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-1300/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7d8486f3b1fba58dfbc6a8a5e39f239e9117917d --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-1300/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:940c54ac51cf109c0e25a0f555c57e874c0810297d50e3c53b699afbe3d2ce85 +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-1400/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-1400/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..62912ae51cbc89e00ad7272dbf8f422241171a46 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-1400/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:95ad3e989c141fac52daba6d9bd25b429e3a63a04134736f80de84c98236ce8b +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-1600/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-1600/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..daeeeb7b668820ddbbf552fe87d2d14e8092bf64 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-1600/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d026cac1168a27a1f8811f5737b40999ab965a8212cac0968b36eef8969ae3ac +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-2100/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-2100/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..aa5346db04016d08e9ef42dc4bfb79df869c1db6 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-2100/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af17d3d596fd5aa9fb343f3afe3902904467adde6c7a54aec2412935a13c103c +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-2600/model-00001-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-2600/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ba7124a9759b22c315867954a7b490c4e7f5b03f --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-2600/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a3b7e0b99294386826724c4271ce3315f708e599cac59d341606b963594a463 +size 4965798912 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-2700/model-00001-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-2700/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7fcc433863027fff0ce7a2da1647e3bd60cd90f7 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-2700/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e25029a0883b9c1a5ccfb4dffd791e59ff1fbcd14ae643231abf8e9097021b4a +size 4965798912 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-400/model-00002-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-400/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..596e16ab490c2f658cebb8b8ee69ea4ca1285b26 --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-400/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eaae5bc1ddff3c3a5fd3beabe654466d1dc37136539aeabc90f6a866d585969f +size 2247734920 diff --git a/babylm_reverse_full_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors b/babylm_reverse_full_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f2b788a156be61b9ed5256203642f469b7c8e9ea --- /dev/null +++ b/babylm_reverse_full_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91a1ef7ad883f52ef3d4cd1ba695e50b7663e26165d6ef37265bfc91925fba9b +size 4965798912