diff --git a/.gitattributes b/.gitattributes index 6547bb60107513670e969210626cee8facadd12f..12daf091a6f8341c13efd4e033b7b52f3b82c1d0 100644 --- a/.gitattributes +++ b/.gitattributes @@ -626,3 +626,9 @@ checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/artifacts/models--met checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/4719a04514ec2f060240711b7c33ab21187cac730ecaba3040b7a0fd95a9cefb filter=lfs diff=lfs merge=lfs -text checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-900/tokenizer.json filter=lfs diff=lfs merge=lfs -text checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-100/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/tokenizer.json filter=lfs diff=lfs merge=lfs -text +checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/tokenizer.json filter=lfs diff=lfs merge=lfs -text diff --git a/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f86b34bc143cc5c1223cc7cd6683ff5a099b7ff3 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2200/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de41c97f1b85191e92c9748ae33c27b2d434c86fe7e93eff7b504a9f4adc65c1 +size 4965798912 diff --git a/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2300/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2300/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2121ad99f07a1f10987a2379e1e626e31e622d12 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_hop_control_10M_seed0/runs/checkpoint-2300/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2cf9a4443150167813cb08d2019804bff1012d4de0b26e029d3d9ecaa172c144 +size 4965798912 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/2d73a6863086ff9d491c28e49df9fb697cd92c2b.lock b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/2d73a6863086ff9d491c28e49df9fb697cd92c2b.lock new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/4719a04514ec2f060240711b7c33ab21187cac730ecaba3040b7a0fd95a9cefb.lock b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/4719a04514ec2f060240711b7c33ab21187cac730ecaba3040b7a0fd95a9cefb.lock new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/47d4a5aa69cdef91a53b77f5c5583647a578ca0e.lock b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/47d4a5aa69cdef91a53b77f5c5583647a578ca0e.lock new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/584d8d3e3f82f7964955174dfe5e3b1cf117a9d859f022cfdf7fcb884856e002.lock b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/584d8d3e3f82f7964955174dfe5e3b1cf117a9d859f022cfdf7fcb884856e002.lock new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58.lock b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/.locks/models--meta-llama--Llama-3.2-3B/d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58.lock new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/.no_exist/13afe5124825b4f3751f836b40dafda64c1ed062/adapter_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/.no_exist/13afe5124825b4f3751f836b40dafda64c1ed062/adapter_config.json new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/.no_exist/13afe5124825b4f3751f836b40dafda64c1ed062/model.safetensors b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/.no_exist/13afe5124825b4f3751f836b40dafda64c1ed062/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/2d73a6863086ff9d491c28e49df9fb697cd92c2b b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/2d73a6863086ff9d491c28e49df9fb697cd92c2b new file mode 100644 index 0000000000000000000000000000000000000000..2d73a6863086ff9d491c28e49df9fb697cd92c2b --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/2d73a6863086ff9d491c28e49df9fb697cd92c2b @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "eos_token_id": 128001, + "transformers_version": "4.45.0.dev0", + "do_sample": true, + "temperature": 0.6, + "top_p": 0.9 +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/47d4a5aa69cdef91a53b77f5c5583647a578ca0e b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/47d4a5aa69cdef91a53b77f5c5583647a578ca0e new file mode 100644 index 0000000000000000000000000000000000000000..47d4a5aa69cdef91a53b77f5c5583647a578ca0e --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/47d4a5aa69cdef91a53b77f5c5583647a578ca0e @@ -0,0 +1,35 @@ +{ + "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": "bfloat16", + "transformers_version": "4.45.0.dev0", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58 b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58 new file mode 100644 index 0000000000000000000000000000000000000000..d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/blobs/d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58 @@ -0,0 +1,261 @@ +{ + "metadata": { + "total_size": 6425499648 + }, + "weight_map": { + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/refs/main b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/refs/main new file mode 100644 index 0000000000000000000000000000000000000000..940c46e96db631910d2cc5186ab76cd83e32b124 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/refs/main @@ -0,0 +1 @@ +13afe5124825b4f3751f836b40dafda64c1ed062 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/config.json new file mode 100644 index 0000000000000000000000000000000000000000..47d4a5aa69cdef91a53b77f5c5583647a578ca0e --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/config.json @@ -0,0 +1,35 @@ +{ + "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": "bfloat16", + "transformers_version": "4.45.0.dev0", + "use_cache": true, + "vocab_size": 128256 +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..2d73a6863086ff9d491c28e49df9fb697cd92c2b --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/generation_config.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 128000, + "eos_token_id": 128001, + "transformers_version": "4.45.0.dev0", + "do_sample": true, + "temperature": 0.6, + "top_p": 0.9 +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..d3a1f0f5f401eeadca0c7a6786bd9e877fd42e58 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/artifacts/models--meta-llama--Llama-3.2-3B/snapshots/13afe5124825b4f3751f836b40dafda64c1ed062/model.safetensors.index.json @@ -0,0 +1,261 @@ +{ + "metadata": { + "total_size": 6425499648 + }, + "weight_map": { + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-100/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-100/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-100/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..fad444f8143e1badd2a8d15a75e619abd485ceab --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:186b472c4073bfc086d5b4584aeaefc23f9c40f1a8199963bda08775617e1e12 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_1.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..39b6620f3f8d750109fe44053f80f2e221c6e290 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d8957d748b592f49a56723f1936535313c244cee56956dbc893c99d40f6ef84b +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..04d30c4b88bb88b6860f56816d3db2c6c7742594 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:be2cd13083af0074c65198c01c90c0a3c599fbe6e0ea705a8835ca51f393d0c0 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7248f7ac479a56a769e8cdf6ce637746c6f7ca09 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21b780d4d6c03815e446fb8e448f580da14aa8b4fd07b55a3d6fb50aae0bd49c +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1100/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..3259c6605b4fec3d2abdf15c7c3ccbdef85f5d96 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_1.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..9f4723485ff30930b59224e8afe0cb0b2b3c51bc --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..6f245457df6ae96d9dae4e4269752c2f036e6ee3 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..4cb50922fa57de26ef77cb3a816c7db227a89a34 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:256d946551f12d380f3f12957da773c4f4a180fdb3c9be7365660e302f4193cf +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1200/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..491d7aa6873ac345ad7d2d8238b3751df06748aa --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..4cf5bfa682e62dbe5c2e496536a84208dc8433d0 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..1a96cc64fd6aecc18f077ec9e3f3f5b8df0525cb --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:71a91896c602899bb3d0f2a27971e2497301e79a066f465c04907a578ddfae3a +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1500/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..5509f0e1ece25901639e38b63337f0f28e33b98d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0123f1b841cbb90ad2cc1a10c38027bc63872dc02a831b18d254a6186f0ffff0 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_1.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..bdaddc4981c690246dfa665b10da40b38a679bc3 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:86e82dbca3ee98b4aff4d87464271202c99a718b90530a7719851b8b6a191547 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..fd1ca9202bb0004e7be63108d043e05db918daf3 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fdc0a29e8caac820aa227d61191352d70074e7bc59c8656d34721d88d568ccb2 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..ec5f7bd05f13db1133b65abf3adad94122fb1c8c --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:535e85e29b289ae27e42bbc01d5f9f60bb0732f7e838f447fec6e6bcbb245d0f +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1600/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..173f12bde4ca76ddf4b03b0964b7524eec1d0cf8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_0.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:103551ff1c3e292b45450c5a23cdaf1958c78f242d8cf8b564df9603b5e96d5c +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_1.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..b0aed44f2fa8fd35795e9b8bfde272b7a98d0ec9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_1.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d69b787213a6aea61f516b084db235c8c80de8bef9f33a5157ae51d85fa9135 +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..487463491926bb630fa016cfe776df854285e817 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/rng_state_2.pth @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ffe85d84d1459bc77a8988c2ddaa541ec71b9112c567063ec91af255e984793b +size 16567 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b2974868394a2ff53fd6eb99c226932a56541b8e --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2a9347f45e367120c490407b2ae3675b1a0757fdaa34cfb976ff3b35e0b9eb7e +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-1700/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/latest b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/latest new file mode 100644 index 0000000000000000000000000000000000000000..adbdf3f70e1e0db88a47bff3edcb0592ed0a2e61 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/latest @@ -0,0 +1 @@ +global_step2200 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..0b782d4602b361f7577abb52b710a8c0e088588d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/tokenizer_config.json @@ -0,0 +1,2078 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..bdb6622e18de1175bdd998376744e2cab7682cce --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/trainer_state.json @@ -0,0 +1,17193 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.2598870056497176, + "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.001027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7522, + "step": 1 + }, + { + "epoch": 0.002054442732408834, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7541, + "step": 2 + }, + { + "epoch": 0.0030816640986132513, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8031, + "step": 3 + }, + { + "epoch": 0.004108885464817668, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8062, + "step": 4 + }, + { + "epoch": 0.005136106831022085, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7733, + "step": 5 + }, + { + "epoch": 0.0061633281972265025, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7949, + "step": 6 + }, + { + "epoch": 0.007190549563430919, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7657, + "step": 7 + }, + { + "epoch": 0.008217770929635337, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.803, + "step": 8 + }, + { + "epoch": 0.009244992295839754, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7793, + "step": 9 + }, + { + "epoch": 0.01027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7935, + "step": 10 + }, + { + "epoch": 0.01027221366204417, + "eval_loss": 2.8114004135131836, + "eval_runtime": 37.8463, + "eval_samples_per_second": 26.423, + "eval_steps_per_second": 1.11, + "step": 10 + }, + { + "epoch": 0.011299435028248588, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7546, + "step": 11 + }, + { + "epoch": 0.012326656394453005, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7423, + "step": 12 + }, + { + "epoch": 0.01335387776065742, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8057, + "step": 13 + }, + { + "epoch": 0.014381099126861838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7663, + "step": 14 + }, + { + "epoch": 0.015408320493066256, + "grad_norm": 4.694424152374268, + "learning_rate": 1.7123287671232876e-08, + "loss": 2.7715, + "step": 15 + }, + { + "epoch": 0.016435541859270673, + "grad_norm": 4.905329704284668, + "learning_rate": 3.424657534246575e-08, + "loss": 2.751, + "step": 16 + }, + { + "epoch": 0.01746276322547509, + "grad_norm": 4.705279350280762, + "learning_rate": 5.136986301369863e-08, + "loss": 2.7186, + "step": 17 + }, + { + "epoch": 0.01848998459167951, + "grad_norm": 5.154810905456543, + "learning_rate": 6.84931506849315e-08, + "loss": 2.8073, + "step": 18 + }, + { + "epoch": 0.019517205957883924, + "grad_norm": 5.070403099060059, + "learning_rate": 8.561643835616439e-08, + "loss": 2.7237, + "step": 19 + }, + { + "epoch": 0.02054442732408834, + "grad_norm": 4.901423454284668, + "learning_rate": 1.0273972602739726e-07, + "loss": 2.7882, + "step": 20 + }, + { + "epoch": 0.02054442732408834, + "eval_loss": 2.810307502746582, + "eval_runtime": 37.65, + "eval_samples_per_second": 26.56, + "eval_steps_per_second": 1.116, + "step": 20 + }, + { + "epoch": 0.02157164869029276, + "grad_norm": 4.914156436920166, + "learning_rate": 1.1986301369863014e-07, + "loss": 2.8005, + "step": 21 + }, + { + "epoch": 0.022598870056497175, + "grad_norm": 4.937129974365234, + "learning_rate": 1.36986301369863e-07, + "loss": 2.7902, + "step": 22 + }, + { + "epoch": 0.02362609142270159, + "grad_norm": 5.0690202713012695, + "learning_rate": 1.541095890410959e-07, + "loss": 2.7474, + "step": 23 + }, + { + "epoch": 0.02465331278890601, + "grad_norm": 4.980844974517822, + "learning_rate": 1.7123287671232878e-07, + "loss": 2.7695, + "step": 24 + }, + { + "epoch": 0.025680534155110426, + "grad_norm": 4.8092451095581055, + "learning_rate": 1.8835616438356165e-07, + "loss": 2.7611, + "step": 25 + }, + { + "epoch": 0.02670775552131484, + "grad_norm": 4.765528202056885, + "learning_rate": 2.0547945205479452e-07, + "loss": 2.7823, + "step": 26 + }, + { + "epoch": 0.02773497688751926, + "grad_norm": 3.9759938716888428, + "learning_rate": 2.226027397260274e-07, + "loss": 2.7445, + "step": 27 + }, + { + "epoch": 0.028762198253723677, + "grad_norm": 4.023419380187988, + "learning_rate": 2.397260273972603e-07, + "loss": 2.7679, + "step": 28 + }, + { + "epoch": 0.029789419619928096, + "grad_norm": 4.161378383636475, + "learning_rate": 2.568493150684932e-07, + "loss": 2.7704, + "step": 29 + }, + { + "epoch": 0.030816640986132512, + "grad_norm": 3.8363842964172363, + "learning_rate": 2.73972602739726e-07, + "loss": 2.7598, + "step": 30 + }, + { + "epoch": 0.030816640986132512, + "eval_loss": 2.772540330886841, + "eval_runtime": 37.2624, + "eval_samples_per_second": 26.837, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03184386235233693, + "grad_norm": 3.718747615814209, + "learning_rate": 2.910958904109589e-07, + "loss": 2.7241, + "step": 31 + }, + { + "epoch": 0.03287108371854135, + "grad_norm": 3.555985927581787, + "learning_rate": 3.082191780821918e-07, + "loss": 2.7751, + "step": 32 + }, + { + "epoch": 0.03389830508474576, + "grad_norm": 3.6041972637176514, + "learning_rate": 3.2534246575342466e-07, + "loss": 2.7322, + "step": 33 + }, + { + "epoch": 0.03492552645095018, + "grad_norm": 3.4408438205718994, + "learning_rate": 3.4246575342465755e-07, + "loss": 2.727, + "step": 34 + }, + { + "epoch": 0.035952747817154594, + "grad_norm": 3.4169280529022217, + "learning_rate": 3.595890410958904e-07, + "loss": 2.7062, + "step": 35 + }, + { + "epoch": 0.03697996918335902, + "grad_norm": 3.3882925510406494, + "learning_rate": 3.767123287671233e-07, + "loss": 2.6271, + "step": 36 + }, + { + "epoch": 0.03800719054956343, + "grad_norm": 3.0491855144500732, + "learning_rate": 3.938356164383562e-07, + "loss": 2.7309, + "step": 37 + }, + { + "epoch": 0.03903441191576785, + "grad_norm": 2.8444902896881104, + "learning_rate": 4.1095890410958903e-07, + "loss": 2.647, + "step": 38 + }, + { + "epoch": 0.040061633281972264, + "grad_norm": 2.7438602447509766, + "learning_rate": 4.2808219178082193e-07, + "loss": 2.6478, + "step": 39 + }, + { + "epoch": 0.04108885464817668, + "grad_norm": 2.8886020183563232, + "learning_rate": 4.452054794520548e-07, + "loss": 2.7065, + "step": 40 + }, + { + "epoch": 0.04108885464817668, + "eval_loss": 2.7021424770355225, + "eval_runtime": 37.3227, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.042116076014381096, + "grad_norm": 2.8825268745422363, + "learning_rate": 4.6232876712328767e-07, + "loss": 2.6666, + "step": 41 + }, + { + "epoch": 0.04314329738058552, + "grad_norm": 2.8848748207092285, + "learning_rate": 4.794520547945206e-07, + "loss": 2.597, + "step": 42 + }, + { + "epoch": 0.044170518746789934, + "grad_norm": 2.9176979064941406, + "learning_rate": 4.965753424657534e-07, + "loss": 2.6638, + "step": 43 + }, + { + "epoch": 0.04519774011299435, + "grad_norm": 2.6903059482574463, + "learning_rate": 5.136986301369864e-07, + "loss": 2.6202, + "step": 44 + }, + { + "epoch": 0.046224961479198766, + "grad_norm": 2.5349855422973633, + "learning_rate": 5.308219178082192e-07, + "loss": 2.6466, + "step": 45 + }, + { + "epoch": 0.04725218284540318, + "grad_norm": 2.405383825302124, + "learning_rate": 5.47945205479452e-07, + "loss": 2.6291, + "step": 46 + }, + { + "epoch": 0.048279404211607604, + "grad_norm": 2.526106595993042, + "learning_rate": 5.65068493150685e-07, + "loss": 2.6718, + "step": 47 + }, + { + "epoch": 0.04930662557781202, + "grad_norm": 2.3981597423553467, + "learning_rate": 5.821917808219178e-07, + "loss": 2.592, + "step": 48 + }, + { + "epoch": 0.050333846944016436, + "grad_norm": 2.61842679977417, + "learning_rate": 5.993150684931507e-07, + "loss": 2.6039, + "step": 49 + }, + { + "epoch": 0.05136106831022085, + "grad_norm": 2.3001420497894287, + "learning_rate": 6.164383561643836e-07, + "loss": 2.5481, + "step": 50 + }, + { + "epoch": 0.05136106831022085, + "eval_loss": 2.6244444847106934, + "eval_runtime": 37.3248, + "eval_samples_per_second": 26.792, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05238828967642527, + "grad_norm": 2.12744140625, + "learning_rate": 6.335616438356165e-07, + "loss": 2.5336, + "step": 51 + }, + { + "epoch": 0.05341551104262968, + "grad_norm": 1.974413275718689, + "learning_rate": 6.506849315068493e-07, + "loss": 2.5882, + "step": 52 + }, + { + "epoch": 0.054442732408834106, + "grad_norm": 2.098527669906616, + "learning_rate": 6.678082191780823e-07, + "loss": 2.5073, + "step": 53 + }, + { + "epoch": 0.05546995377503852, + "grad_norm": 2.056288719177246, + "learning_rate": 6.849315068493151e-07, + "loss": 2.5612, + "step": 54 + }, + { + "epoch": 0.05649717514124294, + "grad_norm": 2.2291629314422607, + "learning_rate": 7.020547945205481e-07, + "loss": 2.527, + "step": 55 + }, + { + "epoch": 0.05752439650744735, + "grad_norm": 2.140164375305176, + "learning_rate": 7.191780821917808e-07, + "loss": 2.4972, + "step": 56 + }, + { + "epoch": 0.05855161787365177, + "grad_norm": 1.8682485818862915, + "learning_rate": 7.363013698630137e-07, + "loss": 2.5231, + "step": 57 + }, + { + "epoch": 0.05957883923985619, + "grad_norm": 1.9373308420181274, + "learning_rate": 7.534246575342466e-07, + "loss": 2.4509, + "step": 58 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.8535712957382202, + "learning_rate": 7.705479452054795e-07, + "loss": 2.5164, + "step": 59 + }, + { + "epoch": 0.061633281972265024, + "grad_norm": 1.7497329711914062, + "learning_rate": 7.876712328767124e-07, + "loss": 2.5177, + "step": 60 + }, + { + "epoch": 0.061633281972265024, + "eval_loss": 2.5608646869659424, + "eval_runtime": 37.3426, + "eval_samples_per_second": 26.779, + "eval_steps_per_second": 1.125, + "step": 60 + }, + { + "epoch": 0.06266050333846944, + "grad_norm": 1.8476629257202148, + "learning_rate": 8.047945205479453e-07, + "loss": 2.4838, + "step": 61 + }, + { + "epoch": 0.06368772470467386, + "grad_norm": 1.7130696773529053, + "learning_rate": 8.219178082191781e-07, + "loss": 2.473, + "step": 62 + }, + { + "epoch": 0.06471494607087827, + "grad_norm": 1.751081109046936, + "learning_rate": 8.39041095890411e-07, + "loss": 2.5152, + "step": 63 + }, + { + "epoch": 0.0657421674370827, + "grad_norm": 1.7758578062057495, + "learning_rate": 8.561643835616439e-07, + "loss": 2.5031, + "step": 64 + }, + { + "epoch": 0.0667693888032871, + "grad_norm": 1.9316363334655762, + "learning_rate": 8.732876712328768e-07, + "loss": 2.3695, + "step": 65 + }, + { + "epoch": 0.06779661016949153, + "grad_norm": 1.7765878438949585, + "learning_rate": 8.904109589041097e-07, + "loss": 2.4669, + "step": 66 + }, + { + "epoch": 0.06882383153569595, + "grad_norm": 1.7339810132980347, + "learning_rate": 9.075342465753426e-07, + "loss": 2.4112, + "step": 67 + }, + { + "epoch": 0.06985105290190036, + "grad_norm": 1.717781662940979, + "learning_rate": 9.246575342465753e-07, + "loss": 2.4677, + "step": 68 + }, + { + "epoch": 0.07087827426810478, + "grad_norm": 1.7711046934127808, + "learning_rate": 9.417808219178083e-07, + "loss": 2.4131, + "step": 69 + }, + { + "epoch": 0.07190549563430919, + "grad_norm": 1.708680510520935, + "learning_rate": 9.589041095890411e-07, + "loss": 2.5191, + "step": 70 + }, + { + "epoch": 0.07190549563430919, + "eval_loss": 2.4920735359191895, + "eval_runtime": 37.3501, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 70 + }, + { + "epoch": 0.07293271700051361, + "grad_norm": 1.9429680109024048, + "learning_rate": 9.76027397260274e-07, + "loss": 2.4099, + "step": 71 + }, + { + "epoch": 0.07395993836671803, + "grad_norm": 1.7411247491836548, + "learning_rate": 9.931506849315068e-07, + "loss": 2.3632, + "step": 72 + }, + { + "epoch": 0.07498715973292244, + "grad_norm": 1.7577983140945435, + "learning_rate": 1.0102739726027399e-06, + "loss": 2.4214, + "step": 73 + }, + { + "epoch": 0.07601438109912687, + "grad_norm": 2.025926351547241, + "learning_rate": 1.0273972602739727e-06, + "loss": 2.394, + "step": 74 + }, + { + "epoch": 0.07704160246533127, + "grad_norm": 2.4366860389709473, + "learning_rate": 1.0445205479452056e-06, + "loss": 2.3943, + "step": 75 + }, + { + "epoch": 0.0780688238315357, + "grad_norm": 2.437108278274536, + "learning_rate": 1.0616438356164384e-06, + "loss": 2.3844, + "step": 76 + }, + { + "epoch": 0.07909604519774012, + "grad_norm": 2.1352434158325195, + "learning_rate": 1.0787671232876712e-06, + "loss": 2.3871, + "step": 77 + }, + { + "epoch": 0.08012326656394453, + "grad_norm": 1.8895397186279297, + "learning_rate": 1.095890410958904e-06, + "loss": 2.3927, + "step": 78 + }, + { + "epoch": 0.08115048793014895, + "grad_norm": 2.2960093021392822, + "learning_rate": 1.1130136986301371e-06, + "loss": 2.3343, + "step": 79 + }, + { + "epoch": 0.08217770929635336, + "grad_norm": 2.0891849994659424, + "learning_rate": 1.13013698630137e-06, + "loss": 2.3732, + "step": 80 + }, + { + "epoch": 0.08217770929635336, + "eval_loss": 2.4189229011535645, + "eval_runtime": 37.3645, + "eval_samples_per_second": 26.763, + "eval_steps_per_second": 1.124, + "step": 80 + }, + { + "epoch": 0.08320493066255778, + "grad_norm": 2.450587034225464, + "learning_rate": 1.1472602739726028e-06, + "loss": 2.3163, + "step": 81 + }, + { + "epoch": 0.08423215202876219, + "grad_norm": 2.079949378967285, + "learning_rate": 1.1643835616438357e-06, + "loss": 2.3381, + "step": 82 + }, + { + "epoch": 0.08525937339496661, + "grad_norm": 2.160737991333008, + "learning_rate": 1.1815068493150685e-06, + "loss": 2.3376, + "step": 83 + }, + { + "epoch": 0.08628659476117104, + "grad_norm": 2.463890552520752, + "learning_rate": 1.1986301369863014e-06, + "loss": 2.308, + "step": 84 + }, + { + "epoch": 0.08731381612737545, + "grad_norm": 2.3923065662384033, + "learning_rate": 1.2157534246575344e-06, + "loss": 2.2682, + "step": 85 + }, + { + "epoch": 0.08834103749357987, + "grad_norm": 2.2096004486083984, + "learning_rate": 1.2328767123287673e-06, + "loss": 2.3256, + "step": 86 + }, + { + "epoch": 0.08936825885978428, + "grad_norm": 2.1402556896209717, + "learning_rate": 1.25e-06, + "loss": 2.2767, + "step": 87 + }, + { + "epoch": 0.0903954802259887, + "grad_norm": 2.500629186630249, + "learning_rate": 1.267123287671233e-06, + "loss": 2.2576, + "step": 88 + }, + { + "epoch": 0.09142270159219312, + "grad_norm": 2.0257675647735596, + "learning_rate": 1.284246575342466e-06, + "loss": 2.2722, + "step": 89 + }, + { + "epoch": 0.09244992295839753, + "grad_norm": 2.148428440093994, + "learning_rate": 1.3013698630136986e-06, + "loss": 2.2309, + "step": 90 + }, + { + "epoch": 0.09244992295839753, + "eval_loss": 2.3439993858337402, + "eval_runtime": 37.3803, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 90 + }, + { + "epoch": 0.09347714432460195, + "grad_norm": 2.3541946411132812, + "learning_rate": 1.3184931506849317e-06, + "loss": 2.2347, + "step": 91 + }, + { + "epoch": 0.09450436569080636, + "grad_norm": 2.6239383220672607, + "learning_rate": 1.3356164383561645e-06, + "loss": 2.2589, + "step": 92 + }, + { + "epoch": 0.09553158705701079, + "grad_norm": 2.1981804370880127, + "learning_rate": 1.3527397260273976e-06, + "loss": 2.2372, + "step": 93 + }, + { + "epoch": 0.09655880842321521, + "grad_norm": 2.3652358055114746, + "learning_rate": 1.3698630136986302e-06, + "loss": 2.2169, + "step": 94 + }, + { + "epoch": 0.09758602978941962, + "grad_norm": 2.2078158855438232, + "learning_rate": 1.386986301369863e-06, + "loss": 2.2761, + "step": 95 + }, + { + "epoch": 0.09861325115562404, + "grad_norm": 2.494685649871826, + "learning_rate": 1.4041095890410961e-06, + "loss": 2.1691, + "step": 96 + }, + { + "epoch": 0.09964047252182845, + "grad_norm": 2.3293051719665527, + "learning_rate": 1.421232876712329e-06, + "loss": 2.2246, + "step": 97 + }, + { + "epoch": 0.10066769388803287, + "grad_norm": 2.410687208175659, + "learning_rate": 1.4383561643835616e-06, + "loss": 2.1166, + "step": 98 + }, + { + "epoch": 0.1016949152542373, + "grad_norm": 2.5168960094451904, + "learning_rate": 1.4554794520547946e-06, + "loss": 2.1417, + "step": 99 + }, + { + "epoch": 0.1027221366204417, + "grad_norm": 2.178412914276123, + "learning_rate": 1.4726027397260275e-06, + "loss": 2.1638, + "step": 100 + }, + { + "epoch": 0.1027221366204417, + "eval_loss": 2.273622751235962, + "eval_runtime": 37.3892, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 100 + }, + { + "epoch": 0.10374935798664613, + "grad_norm": 2.8951706886291504, + "learning_rate": 1.4897260273972605e-06, + "loss": 2.1631, + "step": 101 + }, + { + "epoch": 0.10477657935285054, + "grad_norm": 2.103206157684326, + "learning_rate": 1.5068493150684932e-06, + "loss": 2.1419, + "step": 102 + }, + { + "epoch": 0.10580380071905496, + "grad_norm": 2.495969772338867, + "learning_rate": 1.5239726027397262e-06, + "loss": 2.1029, + "step": 103 + }, + { + "epoch": 0.10683102208525937, + "grad_norm": 2.5686826705932617, + "learning_rate": 1.541095890410959e-06, + "loss": 2.1235, + "step": 104 + }, + { + "epoch": 0.10785824345146379, + "grad_norm": 3.4710471630096436, + "learning_rate": 1.5582191780821921e-06, + "loss": 2.1832, + "step": 105 + }, + { + "epoch": 0.10888546481766821, + "grad_norm": 2.2563316822052, + "learning_rate": 1.5753424657534248e-06, + "loss": 2.1441, + "step": 106 + }, + { + "epoch": 0.10991268618387262, + "grad_norm": 3.860506534576416, + "learning_rate": 1.5924657534246576e-06, + "loss": 2.1764, + "step": 107 + }, + { + "epoch": 0.11093990755007704, + "grad_norm": 2.550135850906372, + "learning_rate": 1.6095890410958907e-06, + "loss": 2.1455, + "step": 108 + }, + { + "epoch": 0.11196712891628145, + "grad_norm": 3.168283700942993, + "learning_rate": 1.6267123287671235e-06, + "loss": 2.0246, + "step": 109 + }, + { + "epoch": 0.11299435028248588, + "grad_norm": 2.577069044113159, + "learning_rate": 1.6438356164383561e-06, + "loss": 2.0921, + "step": 110 + }, + { + "epoch": 0.11299435028248588, + "eval_loss": 2.2218997478485107, + "eval_runtime": 37.246, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 110 + }, + { + "epoch": 0.1140215716486903, + "grad_norm": 2.3515374660491943, + "learning_rate": 1.6609589041095892e-06, + "loss": 2.0209, + "step": 111 + }, + { + "epoch": 0.1150487930148947, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.678082191780822e-06, + "loss": 2.1165, + "step": 112 + }, + { + "epoch": 0.11607601438109913, + "grad_norm": 2.567336082458496, + "learning_rate": 1.695205479452055e-06, + "loss": 2.0529, + "step": 113 + }, + { + "epoch": 0.11710323574730354, + "grad_norm": 2.4126152992248535, + "learning_rate": 1.7123287671232877e-06, + "loss": 2.1028, + "step": 114 + }, + { + "epoch": 0.11813045711350796, + "grad_norm": 3.577528476715088, + "learning_rate": 1.7294520547945206e-06, + "loss": 2.0317, + "step": 115 + }, + { + "epoch": 0.11915767847971238, + "grad_norm": 2.9624195098876953, + "learning_rate": 1.7465753424657536e-06, + "loss": 2.1318, + "step": 116 + }, + { + "epoch": 0.12018489984591679, + "grad_norm": 2.789854049682617, + "learning_rate": 1.7636986301369865e-06, + "loss": 2.0822, + "step": 117 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.8168365955352783, + "learning_rate": 1.7808219178082193e-06, + "loss": 2.08, + "step": 118 + }, + { + "epoch": 0.12223934257832562, + "grad_norm": 2.7616257667541504, + "learning_rate": 1.7979452054794521e-06, + "loss": 2.0668, + "step": 119 + }, + { + "epoch": 0.12326656394453005, + "grad_norm": 2.5610315799713135, + "learning_rate": 1.8150684931506852e-06, + "loss": 2.1518, + "step": 120 + }, + { + "epoch": 0.12326656394453005, + "eval_loss": 2.184033155441284, + "eval_runtime": 37.3784, + "eval_samples_per_second": 26.753, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.12429378531073447, + "grad_norm": 2.817063093185425, + "learning_rate": 1.832191780821918e-06, + "loss": 2.0813, + "step": 121 + }, + { + "epoch": 0.12532100667693888, + "grad_norm": 2.987205982208252, + "learning_rate": 1.8493150684931507e-06, + "loss": 2.0421, + "step": 122 + }, + { + "epoch": 0.1263482280431433, + "grad_norm": 2.8480496406555176, + "learning_rate": 1.8664383561643837e-06, + "loss": 2.0018, + "step": 123 + }, + { + "epoch": 0.12737544940934772, + "grad_norm": 2.994274616241455, + "learning_rate": 1.8835616438356166e-06, + "loss": 2.0266, + "step": 124 + }, + { + "epoch": 0.12840267077555212, + "grad_norm": 2.64136004447937, + "learning_rate": 1.9006849315068496e-06, + "loss": 1.9873, + "step": 125 + }, + { + "epoch": 0.12942989214175654, + "grad_norm": 2.9426915645599365, + "learning_rate": 1.9178082191780823e-06, + "loss": 2.0566, + "step": 126 + }, + { + "epoch": 0.13045711350796096, + "grad_norm": 2.9018619060516357, + "learning_rate": 1.9349315068493153e-06, + "loss": 1.9674, + "step": 127 + }, + { + "epoch": 0.1314843348741654, + "grad_norm": 3.6921045780181885, + "learning_rate": 1.952054794520548e-06, + "loss": 2.0295, + "step": 128 + }, + { + "epoch": 0.1325115562403698, + "grad_norm": 2.4194657802581787, + "learning_rate": 1.969178082191781e-06, + "loss": 2.0639, + "step": 129 + }, + { + "epoch": 0.1335387776065742, + "grad_norm": 3.252349615097046, + "learning_rate": 1.9863013698630136e-06, + "loss": 2.0613, + "step": 130 + }, + { + "epoch": 0.1335387776065742, + "eval_loss": 2.153157949447632, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 130 + }, + { + "epoch": 0.13456599897277863, + "grad_norm": 2.9361023902893066, + "learning_rate": 2.0034246575342467e-06, + "loss": 1.9817, + "step": 131 + }, + { + "epoch": 0.13559322033898305, + "grad_norm": 2.834803581237793, + "learning_rate": 2.0205479452054797e-06, + "loss": 2.0539, + "step": 132 + }, + { + "epoch": 0.13662044170518747, + "grad_norm": 2.8814785480499268, + "learning_rate": 2.037671232876713e-06, + "loss": 2.0296, + "step": 133 + }, + { + "epoch": 0.1376476630713919, + "grad_norm": 2.647125482559204, + "learning_rate": 2.0547945205479454e-06, + "loss": 1.9739, + "step": 134 + }, + { + "epoch": 0.1386748844375963, + "grad_norm": 2.9608750343322754, + "learning_rate": 2.071917808219178e-06, + "loss": 2.0083, + "step": 135 + }, + { + "epoch": 0.1397021058038007, + "grad_norm": 2.8570032119750977, + "learning_rate": 2.089041095890411e-06, + "loss": 2.0216, + "step": 136 + }, + { + "epoch": 0.14072932717000514, + "grad_norm": 2.806358575820923, + "learning_rate": 2.106164383561644e-06, + "loss": 1.9457, + "step": 137 + }, + { + "epoch": 0.14175654853620956, + "grad_norm": 3.1003353595733643, + "learning_rate": 2.123287671232877e-06, + "loss": 2.0324, + "step": 138 + }, + { + "epoch": 0.14278376990241398, + "grad_norm": 2.679588556289673, + "learning_rate": 2.14041095890411e-06, + "loss": 1.9699, + "step": 139 + }, + { + "epoch": 0.14381099126861838, + "grad_norm": 2.9436194896698, + "learning_rate": 2.1575342465753425e-06, + "loss": 1.9824, + "step": 140 + }, + { + "epoch": 0.14381099126861838, + "eval_loss": 2.123953104019165, + "eval_runtime": 37.4214, + "eval_samples_per_second": 26.723, + "eval_steps_per_second": 1.122, + "step": 140 + }, + { + "epoch": 0.1448382126348228, + "grad_norm": 3.088301420211792, + "learning_rate": 2.1746575342465755e-06, + "loss": 1.9861, + "step": 141 + }, + { + "epoch": 0.14586543400102722, + "grad_norm": 3.011859178543091, + "learning_rate": 2.191780821917808e-06, + "loss": 1.9383, + "step": 142 + }, + { + "epoch": 0.14689265536723164, + "grad_norm": 2.930264472961426, + "learning_rate": 2.2089041095890412e-06, + "loss": 2.0674, + "step": 143 + }, + { + "epoch": 0.14791987673343607, + "grad_norm": 3.361328601837158, + "learning_rate": 2.2260273972602743e-06, + "loss": 1.9495, + "step": 144 + }, + { + "epoch": 0.14894709809964046, + "grad_norm": 2.7825112342834473, + "learning_rate": 2.243150684931507e-06, + "loss": 1.9335, + "step": 145 + }, + { + "epoch": 0.14997431946584489, + "grad_norm": 3.7215847969055176, + "learning_rate": 2.26027397260274e-06, + "loss": 1.9403, + "step": 146 + }, + { + "epoch": 0.1510015408320493, + "grad_norm": 2.5125460624694824, + "learning_rate": 2.2773972602739726e-06, + "loss": 1.9437, + "step": 147 + }, + { + "epoch": 0.15202876219825373, + "grad_norm": 3.0963492393493652, + "learning_rate": 2.2945205479452057e-06, + "loss": 1.9651, + "step": 148 + }, + { + "epoch": 0.15305598356445815, + "grad_norm": 2.9147121906280518, + "learning_rate": 2.3116438356164387e-06, + "loss": 2.0567, + "step": 149 + }, + { + "epoch": 0.15408320493066255, + "grad_norm": 3.5275802612304688, + "learning_rate": 2.3287671232876713e-06, + "loss": 1.9773, + "step": 150 + }, + { + "epoch": 0.15408320493066255, + "eval_loss": 2.112344264984131, + "eval_runtime": 37.4142, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 150 + }, + { + "epoch": 0.15511042629686697, + "grad_norm": 3.013786554336548, + "learning_rate": 2.3458904109589044e-06, + "loss": 1.8753, + "step": 151 + }, + { + "epoch": 0.1561376476630714, + "grad_norm": 3.433615207672119, + "learning_rate": 2.363013698630137e-06, + "loss": 1.9256, + "step": 152 + }, + { + "epoch": 0.15716486902927582, + "grad_norm": 3.1526033878326416, + "learning_rate": 2.38013698630137e-06, + "loss": 1.9815, + "step": 153 + }, + { + "epoch": 0.15819209039548024, + "grad_norm": 3.2099974155426025, + "learning_rate": 2.3972602739726027e-06, + "loss": 1.9628, + "step": 154 + }, + { + "epoch": 0.15921931176168463, + "grad_norm": 3.2713584899902344, + "learning_rate": 2.4143835616438358e-06, + "loss": 1.9128, + "step": 155 + }, + { + "epoch": 0.16024653312788906, + "grad_norm": 3.287649393081665, + "learning_rate": 2.431506849315069e-06, + "loss": 1.8764, + "step": 156 + }, + { + "epoch": 0.16127375449409348, + "grad_norm": 3.1439101696014404, + "learning_rate": 2.4486301369863015e-06, + "loss": 1.9428, + "step": 157 + }, + { + "epoch": 0.1623009758602979, + "grad_norm": 3.006457805633545, + "learning_rate": 2.4657534246575345e-06, + "loss": 1.9516, + "step": 158 + }, + { + "epoch": 0.1633281972265023, + "grad_norm": 2.9567787647247314, + "learning_rate": 2.482876712328767e-06, + "loss": 1.915, + "step": 159 + }, + { + "epoch": 0.16435541859270672, + "grad_norm": 3.2475216388702393, + "learning_rate": 2.5e-06, + "loss": 1.9451, + "step": 160 + }, + { + "epoch": 0.16435541859270672, + "eval_loss": 2.086299419403076, + "eval_runtime": 37.3939, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 160 + }, + { + "epoch": 0.16538263995891114, + "grad_norm": 2.9586498737335205, + "learning_rate": 2.5171232876712333e-06, + "loss": 1.9298, + "step": 161 + }, + { + "epoch": 0.16640986132511557, + "grad_norm": 3.066910982131958, + "learning_rate": 2.534246575342466e-06, + "loss": 1.9103, + "step": 162 + }, + { + "epoch": 0.16743708269132, + "grad_norm": 2.814741611480713, + "learning_rate": 2.551369863013699e-06, + "loss": 1.8678, + "step": 163 + }, + { + "epoch": 0.16846430405752438, + "grad_norm": 3.7239339351654053, + "learning_rate": 2.568493150684932e-06, + "loss": 1.9321, + "step": 164 + }, + { + "epoch": 0.1694915254237288, + "grad_norm": 2.829899787902832, + "learning_rate": 2.585616438356164e-06, + "loss": 1.9587, + "step": 165 + }, + { + "epoch": 0.17051874678993323, + "grad_norm": 3.826289176940918, + "learning_rate": 2.6027397260273973e-06, + "loss": 1.9288, + "step": 166 + }, + { + "epoch": 0.17154596815613765, + "grad_norm": 3.1028499603271484, + "learning_rate": 2.6198630136986303e-06, + "loss": 1.8685, + "step": 167 + }, + { + "epoch": 0.17257318952234207, + "grad_norm": 3.6923508644104004, + "learning_rate": 2.6369863013698634e-06, + "loss": 1.9276, + "step": 168 + }, + { + "epoch": 0.17360041088854647, + "grad_norm": 3.0577642917633057, + "learning_rate": 2.654109589041096e-06, + "loss": 1.9653, + "step": 169 + }, + { + "epoch": 0.1746276322547509, + "grad_norm": 3.363762617111206, + "learning_rate": 2.671232876712329e-06, + "loss": 1.9465, + "step": 170 + }, + { + "epoch": 0.1746276322547509, + "eval_loss": 2.0753531455993652, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 170 + }, + { + "epoch": 0.17565485362095531, + "grad_norm": 2.9063165187835693, + "learning_rate": 2.688356164383562e-06, + "loss": 1.8827, + "step": 171 + }, + { + "epoch": 0.17668207498715974, + "grad_norm": 3.4709181785583496, + "learning_rate": 2.705479452054795e-06, + "loss": 1.882, + "step": 172 + }, + { + "epoch": 0.17770929635336416, + "grad_norm": 3.395397663116455, + "learning_rate": 2.7226027397260274e-06, + "loss": 1.8711, + "step": 173 + }, + { + "epoch": 0.17873651771956856, + "grad_norm": 3.9973363876342773, + "learning_rate": 2.7397260273972604e-06, + "loss": 1.8223, + "step": 174 + }, + { + "epoch": 0.17976373908577298, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.947, + "step": 175 + }, + { + "epoch": 0.1807909604519774, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.9192, + "step": 176 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 3.84712290763855, + "learning_rate": 2.773972602739726e-06, + "loss": 1.8344, + "step": 177 + }, + { + "epoch": 0.18284540318438625, + "grad_norm": 3.487525463104248, + "learning_rate": 2.791095890410959e-06, + "loss": 1.9242, + "step": 178 + }, + { + "epoch": 0.18387262455059064, + "grad_norm": 3.1980319023132324, + "learning_rate": 2.8082191780821922e-06, + "loss": 1.7959, + "step": 179 + }, + { + "epoch": 0.18489984591679506, + "grad_norm": 3.56462025642395, + "learning_rate": 2.8253424657534253e-06, + "loss": 1.9668, + "step": 180 + }, + { + "epoch": 0.18489984591679506, + "eval_loss": 2.0606908798217773, + "eval_runtime": 37.3994, + "eval_samples_per_second": 26.738, + "eval_steps_per_second": 1.123, + "step": 180 + }, + { + "epoch": 0.1859270672829995, + "grad_norm": 3.613842248916626, + "learning_rate": 2.842465753424658e-06, + "loss": 1.925, + "step": 181 + }, + { + "epoch": 0.1869542886492039, + "grad_norm": 2.9755642414093018, + "learning_rate": 2.8595890410958905e-06, + "loss": 1.9251, + "step": 182 + }, + { + "epoch": 0.18798151001540833, + "grad_norm": 2.9678146839141846, + "learning_rate": 2.876712328767123e-06, + "loss": 1.9406, + "step": 183 + }, + { + "epoch": 0.18900873138161273, + "grad_norm": 3.3536393642425537, + "learning_rate": 2.8938356164383562e-06, + "loss": 1.8041, + "step": 184 + }, + { + "epoch": 0.19003595274781715, + "grad_norm": 2.714902877807617, + "learning_rate": 2.9109589041095893e-06, + "loss": 1.8213, + "step": 185 + }, + { + "epoch": 0.19106317411402157, + "grad_norm": 3.0686237812042236, + "learning_rate": 2.9280821917808223e-06, + "loss": 1.817, + "step": 186 + }, + { + "epoch": 0.192090395480226, + "grad_norm": 3.510448932647705, + "learning_rate": 2.945205479452055e-06, + "loss": 1.8317, + "step": 187 + }, + { + "epoch": 0.19311761684643042, + "grad_norm": 3.2060787677764893, + "learning_rate": 2.962328767123288e-06, + "loss": 1.894, + "step": 188 + }, + { + "epoch": 0.1941448382126348, + "grad_norm": 3.1756906509399414, + "learning_rate": 2.979452054794521e-06, + "loss": 1.8788, + "step": 189 + }, + { + "epoch": 0.19517205957883924, + "grad_norm": 3.059863805770874, + "learning_rate": 2.9965753424657533e-06, + "loss": 1.8198, + "step": 190 + }, + { + "epoch": 0.19517205957883924, + "eval_loss": 2.0506513118743896, + "eval_runtime": 37.4014, + "eval_samples_per_second": 26.737, + "eval_steps_per_second": 1.123, + "step": 190 + }, + { + "epoch": 0.19619928094504366, + "grad_norm": 3.193685531616211, + "learning_rate": 3.0136986301369864e-06, + "loss": 1.8818, + "step": 191 + }, + { + "epoch": 0.19722650231124808, + "grad_norm": 2.905402183532715, + "learning_rate": 3.0308219178082194e-06, + "loss": 1.8546, + "step": 192 + }, + { + "epoch": 0.1982537236774525, + "grad_norm": 3.423109531402588, + "learning_rate": 3.0479452054794525e-06, + "loss": 1.8142, + "step": 193 + }, + { + "epoch": 0.1992809450436569, + "grad_norm": 3.136094331741333, + "learning_rate": 3.065068493150685e-06, + "loss": 1.8553, + "step": 194 + }, + { + "epoch": 0.20030816640986132, + "grad_norm": 3.0746538639068604, + "learning_rate": 3.082191780821918e-06, + "loss": 1.8153, + "step": 195 + }, + { + "epoch": 0.20133538777606574, + "grad_norm": 2.9354681968688965, + "learning_rate": 3.099315068493151e-06, + "loss": 1.8301, + "step": 196 + }, + { + "epoch": 0.20236260914227017, + "grad_norm": 3.00426983833313, + "learning_rate": 3.1164383561643843e-06, + "loss": 1.8233, + "step": 197 + }, + { + "epoch": 0.2033898305084746, + "grad_norm": 3.2754054069519043, + "learning_rate": 3.1335616438356165e-06, + "loss": 1.787, + "step": 198 + }, + { + "epoch": 0.20441705187467898, + "grad_norm": 2.9854509830474854, + "learning_rate": 3.1506849315068495e-06, + "loss": 1.9341, + "step": 199 + }, + { + "epoch": 0.2054442732408834, + "grad_norm": 3.5655088424682617, + "learning_rate": 3.167808219178082e-06, + "loss": 1.7964, + "step": 200 + }, + { + "epoch": 0.2054442732408834, + "eval_loss": 2.0440263748168945, + "eval_runtime": 37.4054, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 200 + }, + { + "epoch": 0.20647149460708783, + "grad_norm": 3.132373571395874, + "learning_rate": 3.184931506849315e-06, + "loss": 1.8019, + "step": 201 + }, + { + "epoch": 0.20749871597329225, + "grad_norm": 3.3017780780792236, + "learning_rate": 3.2020547945205483e-06, + "loss": 1.7786, + "step": 202 + }, + { + "epoch": 0.20852593733949668, + "grad_norm": 3.009382486343384, + "learning_rate": 3.2191780821917813e-06, + "loss": 1.7639, + "step": 203 + }, + { + "epoch": 0.20955315870570107, + "grad_norm": 3.5784854888916016, + "learning_rate": 3.236301369863014e-06, + "loss": 1.76, + "step": 204 + }, + { + "epoch": 0.2105803800719055, + "grad_norm": 3.236858367919922, + "learning_rate": 3.253424657534247e-06, + "loss": 1.7295, + "step": 205 + }, + { + "epoch": 0.21160760143810992, + "grad_norm": 3.6157479286193848, + "learning_rate": 3.2705479452054796e-06, + "loss": 1.8287, + "step": 206 + }, + { + "epoch": 0.21263482280431434, + "grad_norm": 3.51790452003479, + "learning_rate": 3.2876712328767123e-06, + "loss": 1.8004, + "step": 207 + }, + { + "epoch": 0.21366204417051873, + "grad_norm": 3.0700764656066895, + "learning_rate": 3.3047945205479453e-06, + "loss": 1.7581, + "step": 208 + }, + { + "epoch": 0.21468926553672316, + "grad_norm": 3.2651591300964355, + "learning_rate": 3.3219178082191784e-06, + "loss": 1.7724, + "step": 209 + }, + { + "epoch": 0.21571648690292758, + "grad_norm": 2.9267404079437256, + "learning_rate": 3.3390410958904114e-06, + "loss": 1.8154, + "step": 210 + }, + { + "epoch": 0.21571648690292758, + "eval_loss": 2.0323731899261475, + "eval_runtime": 37.2156, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 210 + }, + { + "epoch": 0.216743708269132, + "grad_norm": 3.463411808013916, + "learning_rate": 3.356164383561644e-06, + "loss": 1.8309, + "step": 211 + }, + { + "epoch": 0.21777092963533642, + "grad_norm": 3.123573064804077, + "learning_rate": 3.373287671232877e-06, + "loss": 1.8793, + "step": 212 + }, + { + "epoch": 0.21879815100154082, + "grad_norm": 3.427051305770874, + "learning_rate": 3.39041095890411e-06, + "loss": 1.8259, + "step": 213 + }, + { + "epoch": 0.21982537236774524, + "grad_norm": 3.306366443634033, + "learning_rate": 3.4075342465753424e-06, + "loss": 1.8565, + "step": 214 + }, + { + "epoch": 0.22085259373394966, + "grad_norm": 3.557340383529663, + "learning_rate": 3.4246575342465754e-06, + "loss": 1.7531, + "step": 215 + }, + { + "epoch": 0.2218798151001541, + "grad_norm": 3.20525860786438, + "learning_rate": 3.4417808219178085e-06, + "loss": 1.8032, + "step": 216 + }, + { + "epoch": 0.2229070364663585, + "grad_norm": 3.2657511234283447, + "learning_rate": 3.458904109589041e-06, + "loss": 1.8026, + "step": 217 + }, + { + "epoch": 0.2239342578325629, + "grad_norm": 3.1869699954986572, + "learning_rate": 3.476027397260274e-06, + "loss": 1.814, + "step": 218 + }, + { + "epoch": 0.22496147919876733, + "grad_norm": 2.661536931991577, + "learning_rate": 3.4931506849315072e-06, + "loss": 1.8005, + "step": 219 + }, + { + "epoch": 0.22598870056497175, + "grad_norm": 3.1108739376068115, + "learning_rate": 3.5102739726027403e-06, + "loss": 1.8095, + "step": 220 + }, + { + "epoch": 0.22598870056497175, + "eval_loss": 2.015268325805664, + "eval_runtime": 37.2867, + "eval_samples_per_second": 26.819, + "eval_steps_per_second": 1.126, + "step": 220 + }, + { + "epoch": 0.22701592193117617, + "grad_norm": 2.8078837394714355, + "learning_rate": 3.527397260273973e-06, + "loss": 1.8016, + "step": 221 + }, + { + "epoch": 0.2280431432973806, + "grad_norm": 3.131399631500244, + "learning_rate": 3.5445205479452056e-06, + "loss": 1.7787, + "step": 222 + }, + { + "epoch": 0.229070364663585, + "grad_norm": 3.114422559738159, + "learning_rate": 3.5616438356164386e-06, + "loss": 1.7899, + "step": 223 + }, + { + "epoch": 0.2300975860297894, + "grad_norm": 3.040816068649292, + "learning_rate": 3.5787671232876712e-06, + "loss": 1.8601, + "step": 224 + }, + { + "epoch": 0.23112480739599384, + "grad_norm": 3.043100357055664, + "learning_rate": 3.5958904109589043e-06, + "loss": 1.8244, + "step": 225 + }, + { + "epoch": 0.23215202876219826, + "grad_norm": 3.4232981204986572, + "learning_rate": 3.6130136986301373e-06, + "loss": 1.7634, + "step": 226 + }, + { + "epoch": 0.23317925012840268, + "grad_norm": 2.8170571327209473, + "learning_rate": 3.6301369863013704e-06, + "loss": 1.7793, + "step": 227 + }, + { + "epoch": 0.23420647149460708, + "grad_norm": 3.9133572578430176, + "learning_rate": 3.647260273972603e-06, + "loss": 1.8239, + "step": 228 + }, + { + "epoch": 0.2352336928608115, + "grad_norm": 2.7053382396698, + "learning_rate": 3.664383561643836e-06, + "loss": 1.8209, + "step": 229 + }, + { + "epoch": 0.23626091422701592, + "grad_norm": 3.4366891384124756, + "learning_rate": 3.6815068493150687e-06, + "loss": 1.7539, + "step": 230 + }, + { + "epoch": 0.23626091422701592, + "eval_loss": 2.0176751613616943, + "eval_runtime": 37.3064, + "eval_samples_per_second": 26.805, + "eval_steps_per_second": 1.126, + "step": 230 + }, + { + "epoch": 0.23728813559322035, + "grad_norm": 2.592280149459839, + "learning_rate": 3.6986301369863014e-06, + "loss": 1.7858, + "step": 231 + }, + { + "epoch": 0.23831535695942477, + "grad_norm": 2.8370869159698486, + "learning_rate": 3.7157534246575344e-06, + "loss": 1.7414, + "step": 232 + }, + { + "epoch": 0.23934257832562916, + "grad_norm": 3.2120535373687744, + "learning_rate": 3.7328767123287675e-06, + "loss": 1.8467, + "step": 233 + }, + { + "epoch": 0.24036979969183359, + "grad_norm": 2.9314255714416504, + "learning_rate": 3.7500000000000005e-06, + "loss": 1.7602, + "step": 234 + }, + { + "epoch": 0.241397021058038, + "grad_norm": 3.056293249130249, + "learning_rate": 3.767123287671233e-06, + "loss": 1.7972, + "step": 235 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.5918917655944824, + "learning_rate": 3.784246575342466e-06, + "loss": 1.8274, + "step": 236 + }, + { + "epoch": 0.24345146379044685, + "grad_norm": 2.7911906242370605, + "learning_rate": 3.8013698630136993e-06, + "loss": 1.753, + "step": 237 + }, + { + "epoch": 0.24447868515665125, + "grad_norm": 2.810612916946411, + "learning_rate": 3.818493150684932e-06, + "loss": 1.7822, + "step": 238 + }, + { + "epoch": 0.24550590652285567, + "grad_norm": 2.8863744735717773, + "learning_rate": 3.8356164383561645e-06, + "loss": 1.7258, + "step": 239 + }, + { + "epoch": 0.2465331278890601, + "grad_norm": 3.002087116241455, + "learning_rate": 3.852739726027397e-06, + "loss": 1.8423, + "step": 240 + }, + { + "epoch": 0.2465331278890601, + "eval_loss": 2.020223617553711, + "eval_runtime": 37.3197, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 240 + }, + { + "epoch": 0.24756034925526452, + "grad_norm": 2.892317056655884, + "learning_rate": 3.869863013698631e-06, + "loss": 1.8181, + "step": 241 + }, + { + "epoch": 0.24858757062146894, + "grad_norm": 2.895620584487915, + "learning_rate": 3.886986301369863e-06, + "loss": 1.7538, + "step": 242 + }, + { + "epoch": 0.24961479198767333, + "grad_norm": 3.015660524368286, + "learning_rate": 3.904109589041096e-06, + "loss": 1.766, + "step": 243 + }, + { + "epoch": 0.25064201335387776, + "grad_norm": 2.7516112327575684, + "learning_rate": 3.921232876712329e-06, + "loss": 1.7324, + "step": 244 + }, + { + "epoch": 0.2516692347200822, + "grad_norm": 2.937777519226074, + "learning_rate": 3.938356164383562e-06, + "loss": 1.7468, + "step": 245 + }, + { + "epoch": 0.2526964560862866, + "grad_norm": 3.1706550121307373, + "learning_rate": 3.9554794520547955e-06, + "loss": 1.7035, + "step": 246 + }, + { + "epoch": 0.253723677452491, + "grad_norm": 2.7795372009277344, + "learning_rate": 3.972602739726027e-06, + "loss": 1.7912, + "step": 247 + }, + { + "epoch": 0.25475089881869545, + "grad_norm": 3.0375940799713135, + "learning_rate": 3.989726027397261e-06, + "loss": 1.7208, + "step": 248 + }, + { + "epoch": 0.25577812018489987, + "grad_norm": 2.863913059234619, + "learning_rate": 4.006849315068493e-06, + "loss": 1.7891, + "step": 249 + }, + { + "epoch": 0.25680534155110424, + "grad_norm": 3.02720046043396, + "learning_rate": 4.023972602739726e-06, + "loss": 1.7864, + "step": 250 + }, + { + "epoch": 0.25680534155110424, + "eval_loss": 1.9981943368911743, + "eval_runtime": 37.3019, + "eval_samples_per_second": 26.808, + "eval_steps_per_second": 1.126, + "step": 250 + }, + { + "epoch": 0.25783256291730866, + "grad_norm": 2.90144681930542, + "learning_rate": 4.0410958904109595e-06, + "loss": 1.7508, + "step": 251 + }, + { + "epoch": 0.2588597842835131, + "grad_norm": 2.7670342922210693, + "learning_rate": 4.058219178082192e-06, + "loss": 1.7668, + "step": 252 + }, + { + "epoch": 0.2598870056497175, + "grad_norm": 3.0309600830078125, + "learning_rate": 4.075342465753426e-06, + "loss": 1.7853, + "step": 253 + }, + { + "epoch": 0.26091422701592193, + "grad_norm": 2.995190143585205, + "learning_rate": 4.092465753424658e-06, + "loss": 1.7409, + "step": 254 + }, + { + "epoch": 0.26194144838212635, + "grad_norm": 3.127767324447632, + "learning_rate": 4.109589041095891e-06, + "loss": 1.743, + "step": 255 + }, + { + "epoch": 0.2629686697483308, + "grad_norm": 3.7104549407958984, + "learning_rate": 4.1267123287671235e-06, + "loss": 1.8149, + "step": 256 + }, + { + "epoch": 0.2639958911145352, + "grad_norm": 3.042569160461426, + "learning_rate": 4.143835616438356e-06, + "loss": 1.7544, + "step": 257 + }, + { + "epoch": 0.2650231124807396, + "grad_norm": 2.9337198734283447, + "learning_rate": 4.16095890410959e-06, + "loss": 1.823, + "step": 258 + }, + { + "epoch": 0.26605033384694404, + "grad_norm": 3.1853089332580566, + "learning_rate": 4.178082191780822e-06, + "loss": 1.8002, + "step": 259 + }, + { + "epoch": 0.2670775552131484, + "grad_norm": 2.9620680809020996, + "learning_rate": 4.195205479452055e-06, + "loss": 1.7401, + "step": 260 + }, + { + "epoch": 0.2670775552131484, + "eval_loss": 1.9844669103622437, + "eval_runtime": 37.3162, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 260 + }, + { + "epoch": 0.26810477657935283, + "grad_norm": 3.477985382080078, + "learning_rate": 4.212328767123288e-06, + "loss": 1.7456, + "step": 261 + }, + { + "epoch": 0.26913199794555726, + "grad_norm": 2.881277084350586, + "learning_rate": 4.229452054794521e-06, + "loss": 1.7427, + "step": 262 + }, + { + "epoch": 0.2701592193117617, + "grad_norm": 3.3313355445861816, + "learning_rate": 4.246575342465754e-06, + "loss": 1.7537, + "step": 263 + }, + { + "epoch": 0.2711864406779661, + "grad_norm": 2.947119951248169, + "learning_rate": 4.263698630136986e-06, + "loss": 1.7349, + "step": 264 + }, + { + "epoch": 0.2722136620441705, + "grad_norm": 3.0780889987945557, + "learning_rate": 4.28082191780822e-06, + "loss": 1.694, + "step": 265 + }, + { + "epoch": 0.27324088341037495, + "grad_norm": 3.3371143341064453, + "learning_rate": 4.297945205479452e-06, + "loss": 1.8489, + "step": 266 + }, + { + "epoch": 0.27426810477657937, + "grad_norm": 3.01694917678833, + "learning_rate": 4.315068493150685e-06, + "loss": 1.7478, + "step": 267 + }, + { + "epoch": 0.2752953261427838, + "grad_norm": 3.001850128173828, + "learning_rate": 4.3321917808219185e-06, + "loss": 1.7298, + "step": 268 + }, + { + "epoch": 0.2763225475089882, + "grad_norm": 2.9096696376800537, + "learning_rate": 4.349315068493151e-06, + "loss": 1.7542, + "step": 269 + }, + { + "epoch": 0.2773497688751926, + "grad_norm": 3.209601402282715, + "learning_rate": 4.3664383561643846e-06, + "loss": 1.7248, + "step": 270 + }, + { + "epoch": 0.2773497688751926, + "eval_loss": 1.9732507467269897, + "eval_runtime": 37.3151, + "eval_samples_per_second": 26.799, + "eval_steps_per_second": 1.126, + "step": 270 + }, + { + "epoch": 0.278376990241397, + "grad_norm": 2.9635558128356934, + "learning_rate": 4.383561643835616e-06, + "loss": 1.727, + "step": 271 + }, + { + "epoch": 0.2794042116076014, + "grad_norm": 3.143821954727173, + "learning_rate": 4.40068493150685e-06, + "loss": 1.8045, + "step": 272 + }, + { + "epoch": 0.28043143297380585, + "grad_norm": 3.4248714447021484, + "learning_rate": 4.4178082191780825e-06, + "loss": 1.7649, + "step": 273 + }, + { + "epoch": 0.2814586543400103, + "grad_norm": 2.902162790298462, + "learning_rate": 4.434931506849315e-06, + "loss": 1.7324, + "step": 274 + }, + { + "epoch": 0.2824858757062147, + "grad_norm": 2.8899712562561035, + "learning_rate": 4.4520547945205486e-06, + "loss": 1.6886, + "step": 275 + }, + { + "epoch": 0.2835130970724191, + "grad_norm": 3.2267258167266846, + "learning_rate": 4.469178082191781e-06, + "loss": 1.7442, + "step": 276 + }, + { + "epoch": 0.28454031843862354, + "grad_norm": 3.275735378265381, + "learning_rate": 4.486301369863014e-06, + "loss": 1.7334, + "step": 277 + }, + { + "epoch": 0.28556753980482796, + "grad_norm": 2.991509199142456, + "learning_rate": 4.503424657534247e-06, + "loss": 1.7853, + "step": 278 + }, + { + "epoch": 0.28659476117103233, + "grad_norm": 2.8380391597747803, + "learning_rate": 4.52054794520548e-06, + "loss": 1.6885, + "step": 279 + }, + { + "epoch": 0.28762198253723675, + "grad_norm": 2.909980297088623, + "learning_rate": 4.537671232876713e-06, + "loss": 1.7146, + "step": 280 + }, + { + "epoch": 0.28762198253723675, + "eval_loss": 1.9755980968475342, + "eval_runtime": 37.3524, + "eval_samples_per_second": 26.772, + "eval_steps_per_second": 1.124, + "step": 280 + }, + { + "epoch": 0.2886492039034412, + "grad_norm": 3.2689175605773926, + "learning_rate": 4.554794520547945e-06, + "loss": 1.7747, + "step": 281 + }, + { + "epoch": 0.2896764252696456, + "grad_norm": 2.8242290019989014, + "learning_rate": 4.571917808219179e-06, + "loss": 1.7657, + "step": 282 + }, + { + "epoch": 0.29070364663585, + "grad_norm": 3.03828501701355, + "learning_rate": 4.589041095890411e-06, + "loss": 1.7555, + "step": 283 + }, + { + "epoch": 0.29173086800205444, + "grad_norm": 2.714351177215576, + "learning_rate": 4.606164383561644e-06, + "loss": 1.7055, + "step": 284 + }, + { + "epoch": 0.29275808936825887, + "grad_norm": 2.9540348052978516, + "learning_rate": 4.6232876712328774e-06, + "loss": 1.7258, + "step": 285 + }, + { + "epoch": 0.2937853107344633, + "grad_norm": 3.107060670852661, + "learning_rate": 4.64041095890411e-06, + "loss": 1.7376, + "step": 286 + }, + { + "epoch": 0.2948125321006677, + "grad_norm": 3.3266639709472656, + "learning_rate": 4.657534246575343e-06, + "loss": 1.7063, + "step": 287 + }, + { + "epoch": 0.29583975346687214, + "grad_norm": 3.3113174438476562, + "learning_rate": 4.674657534246575e-06, + "loss": 1.7514, + "step": 288 + }, + { + "epoch": 0.2968669748330765, + "grad_norm": 3.313321352005005, + "learning_rate": 4.691780821917809e-06, + "loss": 1.73, + "step": 289 + }, + { + "epoch": 0.2978941961992809, + "grad_norm": 2.901522159576416, + "learning_rate": 4.7089041095890414e-06, + "loss": 1.7709, + "step": 290 + }, + { + "epoch": 0.2978941961992809, + "eval_loss": 1.9608837366104126, + "eval_runtime": 37.3687, + "eval_samples_per_second": 26.76, + "eval_steps_per_second": 1.124, + "step": 290 + }, + { + "epoch": 0.29892141756548535, + "grad_norm": 3.4208264350891113, + "learning_rate": 4.726027397260274e-06, + "loss": 1.73, + "step": 291 + }, + { + "epoch": 0.29994863893168977, + "grad_norm": 2.8982441425323486, + "learning_rate": 4.7431506849315075e-06, + "loss": 1.7416, + "step": 292 + }, + { + "epoch": 0.3009758602978942, + "grad_norm": 3.2743732929229736, + "learning_rate": 4.76027397260274e-06, + "loss": 1.7443, + "step": 293 + }, + { + "epoch": 0.3020030816640986, + "grad_norm": 3.028764486312866, + "learning_rate": 4.777397260273973e-06, + "loss": 1.6879, + "step": 294 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.3531785011291504, + "learning_rate": 4.7945205479452054e-06, + "loss": 1.7096, + "step": 295 + }, + { + "epoch": 0.30405752439650746, + "grad_norm": 2.5602540969848633, + "learning_rate": 4.811643835616439e-06, + "loss": 1.7125, + "step": 296 + }, + { + "epoch": 0.3050847457627119, + "grad_norm": 3.1823129653930664, + "learning_rate": 4.8287671232876716e-06, + "loss": 1.6531, + "step": 297 + }, + { + "epoch": 0.3061119671289163, + "grad_norm": 2.6700282096862793, + "learning_rate": 4.845890410958904e-06, + "loss": 1.6983, + "step": 298 + }, + { + "epoch": 0.3071391884951207, + "grad_norm": 2.7118732929229736, + "learning_rate": 4.863013698630138e-06, + "loss": 1.7056, + "step": 299 + }, + { + "epoch": 0.3081664098613251, + "grad_norm": 3.3493311405181885, + "learning_rate": 4.88013698630137e-06, + "loss": 1.6944, + "step": 300 + }, + { + "epoch": 0.3081664098613251, + "eval_loss": 1.9489911794662476, + "eval_runtime": 37.35, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 300 + }, + { + "epoch": 0.3091936312275295, + "grad_norm": 2.9436538219451904, + "learning_rate": 4.897260273972603e-06, + "loss": 1.6717, + "step": 301 + }, + { + "epoch": 0.31022085259373394, + "grad_norm": 2.894737482070923, + "learning_rate": 4.914383561643836e-06, + "loss": 1.7895, + "step": 302 + }, + { + "epoch": 0.31124807395993837, + "grad_norm": 2.8645665645599365, + "learning_rate": 4.931506849315069e-06, + "loss": 1.721, + "step": 303 + }, + { + "epoch": 0.3122752953261428, + "grad_norm": 3.050203323364258, + "learning_rate": 4.948630136986302e-06, + "loss": 1.7564, + "step": 304 + }, + { + "epoch": 0.3133025166923472, + "grad_norm": 3.146395206451416, + "learning_rate": 4.965753424657534e-06, + "loss": 1.7938, + "step": 305 + }, + { + "epoch": 0.31432973805855163, + "grad_norm": 3.1021251678466797, + "learning_rate": 4.982876712328768e-06, + "loss": 1.6682, + "step": 306 + }, + { + "epoch": 0.31535695942475606, + "grad_norm": 2.7301125526428223, + "learning_rate": 5e-06, + "loss": 1.6974, + "step": 307 + }, + { + "epoch": 0.3163841807909605, + "grad_norm": 2.7602250576019287, + "learning_rate": 4.998096688237534e-06, + "loss": 1.6523, + "step": 308 + }, + { + "epoch": 0.31741140215716485, + "grad_norm": 2.8149876594543457, + "learning_rate": 4.996193376475067e-06, + "loss": 1.7535, + "step": 309 + }, + { + "epoch": 0.31843862352336927, + "grad_norm": 2.54823637008667, + "learning_rate": 4.9942900647126005e-06, + "loss": 1.63, + "step": 310 + }, + { + "epoch": 0.31843862352336927, + "eval_loss": 1.928893804550171, + "eval_runtime": 37.2236, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3194658448895737, + "grad_norm": 2.7819766998291016, + "learning_rate": 4.992386752950134e-06, + "loss": 1.6461, + "step": 311 + }, + { + "epoch": 0.3204930662557781, + "grad_norm": 2.999209403991699, + "learning_rate": 4.990483441187667e-06, + "loss": 1.6861, + "step": 312 + }, + { + "epoch": 0.32152028762198254, + "grad_norm": 2.7288851737976074, + "learning_rate": 4.9885801294252e-06, + "loss": 1.7076, + "step": 313 + }, + { + "epoch": 0.32254750898818696, + "grad_norm": 2.8722457885742188, + "learning_rate": 4.986676817662733e-06, + "loss": 1.7079, + "step": 314 + }, + { + "epoch": 0.3235747303543914, + "grad_norm": 2.853344678878784, + "learning_rate": 4.9847735059002665e-06, + "loss": 1.6699, + "step": 315 + }, + { + "epoch": 0.3246019517205958, + "grad_norm": 2.7404720783233643, + "learning_rate": 4.9828701941378e-06, + "loss": 1.7423, + "step": 316 + }, + { + "epoch": 0.3256291730868002, + "grad_norm": 2.6062633991241455, + "learning_rate": 4.980966882375333e-06, + "loss": 1.6811, + "step": 317 + }, + { + "epoch": 0.3266563944530046, + "grad_norm": 2.834162712097168, + "learning_rate": 4.979063570612867e-06, + "loss": 1.6575, + "step": 318 + }, + { + "epoch": 0.327683615819209, + "grad_norm": 2.7741048336029053, + "learning_rate": 4.9771602588504e-06, + "loss": 1.6158, + "step": 319 + }, + { + "epoch": 0.32871083718541344, + "grad_norm": 2.6216888427734375, + "learning_rate": 4.975256947087933e-06, + "loss": 1.7295, + "step": 320 + }, + { + "epoch": 0.32871083718541344, + "eval_loss": 1.9309543371200562, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 320 + }, + { + "epoch": 0.32973805855161786, + "grad_norm": 2.6282548904418945, + "learning_rate": 4.973353635325467e-06, + "loss": 1.6266, + "step": 321 + }, + { + "epoch": 0.3307652799178223, + "grad_norm": 2.632124662399292, + "learning_rate": 4.971450323563e-06, + "loss": 1.6887, + "step": 322 + }, + { + "epoch": 0.3317925012840267, + "grad_norm": 2.5844857692718506, + "learning_rate": 4.969547011800533e-06, + "loss": 1.6287, + "step": 323 + }, + { + "epoch": 0.33281972265023113, + "grad_norm": 2.597620964050293, + "learning_rate": 4.967643700038067e-06, + "loss": 1.6876, + "step": 324 + }, + { + "epoch": 0.33384694401643555, + "grad_norm": 2.6529343128204346, + "learning_rate": 4.9657403882756e-06, + "loss": 1.6883, + "step": 325 + }, + { + "epoch": 0.33487416538264, + "grad_norm": 2.592874526977539, + "learning_rate": 4.9638370765131335e-06, + "loss": 1.653, + "step": 326 + }, + { + "epoch": 0.3359013867488444, + "grad_norm": 2.523489236831665, + "learning_rate": 4.961933764750667e-06, + "loss": 1.6501, + "step": 327 + }, + { + "epoch": 0.33692860811504877, + "grad_norm": 2.6721699237823486, + "learning_rate": 4.9600304529882e-06, + "loss": 1.7126, + "step": 328 + }, + { + "epoch": 0.3379558294812532, + "grad_norm": 2.513122797012329, + "learning_rate": 4.9581271412257335e-06, + "loss": 1.7268, + "step": 329 + }, + { + "epoch": 0.3389830508474576, + "grad_norm": 2.571118116378784, + "learning_rate": 4.956223829463266e-06, + "loss": 1.6862, + "step": 330 + }, + { + "epoch": 0.3389830508474576, + "eval_loss": 1.9501315355300903, + "eval_runtime": 37.3287, + "eval_samples_per_second": 26.789, + "eval_steps_per_second": 1.125, + "step": 330 + }, + { + "epoch": 0.34001027221366203, + "grad_norm": 2.5783371925354004, + "learning_rate": 4.954320517700799e-06, + "loss": 1.6041, + "step": 331 + }, + { + "epoch": 0.34103749357986646, + "grad_norm": 2.6829233169555664, + "learning_rate": 4.952417205938333e-06, + "loss": 1.7111, + "step": 332 + }, + { + "epoch": 0.3420647149460709, + "grad_norm": 2.6643247604370117, + "learning_rate": 4.950513894175866e-06, + "loss": 1.6797, + "step": 333 + }, + { + "epoch": 0.3430919363122753, + "grad_norm": 2.776308298110962, + "learning_rate": 4.9486105824133995e-06, + "loss": 1.6766, + "step": 334 + }, + { + "epoch": 0.3441191576784797, + "grad_norm": 2.722667932510376, + "learning_rate": 4.946707270650933e-06, + "loss": 1.6865, + "step": 335 + }, + { + "epoch": 0.34514637904468415, + "grad_norm": 2.7050845623016357, + "learning_rate": 4.944803958888466e-06, + "loss": 1.6967, + "step": 336 + }, + { + "epoch": 0.34617360041088857, + "grad_norm": 3.0009098052978516, + "learning_rate": 4.942900647126e-06, + "loss": 1.714, + "step": 337 + }, + { + "epoch": 0.34720082177709294, + "grad_norm": 2.793621301651001, + "learning_rate": 4.940997335363533e-06, + "loss": 1.6591, + "step": 338 + }, + { + "epoch": 0.34822804314329736, + "grad_norm": 2.844383478164673, + "learning_rate": 4.939094023601066e-06, + "loss": 1.611, + "step": 339 + }, + { + "epoch": 0.3492552645095018, + "grad_norm": 2.829378366470337, + "learning_rate": 4.9371907118386e-06, + "loss": 1.666, + "step": 340 + }, + { + "epoch": 0.3492552645095018, + "eval_loss": 1.96309494972229, + "eval_runtime": 37.3346, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 340 + }, + { + "epoch": 0.3502824858757062, + "grad_norm": 2.9702041149139404, + "learning_rate": 4.935287400076133e-06, + "loss": 1.6703, + "step": 341 + }, + { + "epoch": 0.35130970724191063, + "grad_norm": 2.7555935382843018, + "learning_rate": 4.933384088313666e-06, + "loss": 1.6879, + "step": 342 + }, + { + "epoch": 0.35233692860811505, + "grad_norm": 2.8968169689178467, + "learning_rate": 4.9314807765512e-06, + "loss": 1.6342, + "step": 343 + }, + { + "epoch": 0.3533641499743195, + "grad_norm": 3.010115385055542, + "learning_rate": 4.929577464788733e-06, + "loss": 1.5883, + "step": 344 + }, + { + "epoch": 0.3543913713405239, + "grad_norm": 2.6449785232543945, + "learning_rate": 4.9276741530262665e-06, + "loss": 1.654, + "step": 345 + }, + { + "epoch": 0.3554185927067283, + "grad_norm": 2.976109266281128, + "learning_rate": 4.9257708412638e-06, + "loss": 1.6483, + "step": 346 + }, + { + "epoch": 0.35644581407293274, + "grad_norm": 2.856144666671753, + "learning_rate": 4.923867529501332e-06, + "loss": 1.6536, + "step": 347 + }, + { + "epoch": 0.3574730354391371, + "grad_norm": 2.8142166137695312, + "learning_rate": 4.921964217738866e-06, + "loss": 1.6415, + "step": 348 + }, + { + "epoch": 0.35850025680534153, + "grad_norm": 2.70807147026062, + "learning_rate": 4.920060905976399e-06, + "loss": 1.7099, + "step": 349 + }, + { + "epoch": 0.35952747817154596, + "grad_norm": 2.7715506553649902, + "learning_rate": 4.9181575942139325e-06, + "loss": 1.6947, + "step": 350 + }, + { + "epoch": 0.35952747817154596, + "eval_loss": 1.9507063627243042, + "eval_runtime": 37.3094, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 350 + }, + { + "epoch": 0.3605546995377504, + "grad_norm": 2.8816769123077393, + "learning_rate": 4.916254282451466e-06, + "loss": 1.6906, + "step": 351 + }, + { + "epoch": 0.3615819209039548, + "grad_norm": 3.08797025680542, + "learning_rate": 4.914350970688999e-06, + "loss": 1.7387, + "step": 352 + }, + { + "epoch": 0.3626091422701592, + "grad_norm": 2.6018118858337402, + "learning_rate": 4.9124476589265325e-06, + "loss": 1.6319, + "step": 353 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7790255546569824, + "learning_rate": 4.910544347164066e-06, + "loss": 1.6357, + "step": 354 + }, + { + "epoch": 0.36466358500256807, + "grad_norm": 2.7360289096832275, + "learning_rate": 4.908641035401599e-06, + "loss": 1.7127, + "step": 355 + }, + { + "epoch": 0.3656908063687725, + "grad_norm": 2.6693854331970215, + "learning_rate": 4.906737723639133e-06, + "loss": 1.7747, + "step": 356 + }, + { + "epoch": 0.3667180277349769, + "grad_norm": 2.642868995666504, + "learning_rate": 4.904834411876665e-06, + "loss": 1.6401, + "step": 357 + }, + { + "epoch": 0.3677452491011813, + "grad_norm": 2.7230374813079834, + "learning_rate": 4.902931100114199e-06, + "loss": 1.6833, + "step": 358 + }, + { + "epoch": 0.3687724704673857, + "grad_norm": 2.499457359313965, + "learning_rate": 4.901027788351733e-06, + "loss": 1.6007, + "step": 359 + }, + { + "epoch": 0.3697996918335901, + "grad_norm": 2.7695276737213135, + "learning_rate": 4.899124476589266e-06, + "loss": 1.6486, + "step": 360 + }, + { + "epoch": 0.3697996918335901, + "eval_loss": 1.9434194564819336, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.37082691319979455, + "grad_norm": 2.5715794563293457, + "learning_rate": 4.8972211648267994e-06, + "loss": 1.6183, + "step": 361 + }, + { + "epoch": 0.371854134565999, + "grad_norm": 2.637859582901001, + "learning_rate": 4.895317853064333e-06, + "loss": 1.6648, + "step": 362 + }, + { + "epoch": 0.3728813559322034, + "grad_norm": 2.5601117610931396, + "learning_rate": 4.893414541301866e-06, + "loss": 1.6475, + "step": 363 + }, + { + "epoch": 0.3739085772984078, + "grad_norm": 2.429337739944458, + "learning_rate": 4.891511229539399e-06, + "loss": 1.6435, + "step": 364 + }, + { + "epoch": 0.37493579866461224, + "grad_norm": 2.6890885829925537, + "learning_rate": 4.889607917776932e-06, + "loss": 1.6577, + "step": 365 + }, + { + "epoch": 0.37596302003081666, + "grad_norm": 2.55492901802063, + "learning_rate": 4.887704606014465e-06, + "loss": 1.6523, + "step": 366 + }, + { + "epoch": 0.37699024139702103, + "grad_norm": 2.841362476348877, + "learning_rate": 4.885801294251999e-06, + "loss": 1.6742, + "step": 367 + }, + { + "epoch": 0.37801746276322545, + "grad_norm": 2.8041367530822754, + "learning_rate": 4.883897982489532e-06, + "loss": 1.5946, + "step": 368 + }, + { + "epoch": 0.3790446841294299, + "grad_norm": 2.7135822772979736, + "learning_rate": 4.8819946707270655e-06, + "loss": 1.6909, + "step": 369 + }, + { + "epoch": 0.3800719054956343, + "grad_norm": 2.9020211696624756, + "learning_rate": 4.880091358964599e-06, + "loss": 1.6804, + "step": 370 + }, + { + "epoch": 0.3800719054956343, + "eval_loss": 1.9353317022323608, + "eval_runtime": 37.3188, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 370 + }, + { + "epoch": 0.3810991268618387, + "grad_norm": 2.567310333251953, + "learning_rate": 4.878188047202132e-06, + "loss": 1.5885, + "step": 371 + }, + { + "epoch": 0.38212634822804314, + "grad_norm": 2.910834789276123, + "learning_rate": 4.876284735439666e-06, + "loss": 1.7681, + "step": 372 + }, + { + "epoch": 0.38315356959424757, + "grad_norm": 2.7507543563842773, + "learning_rate": 4.874381423677199e-06, + "loss": 1.7228, + "step": 373 + }, + { + "epoch": 0.384180790960452, + "grad_norm": 2.9230854511260986, + "learning_rate": 4.8724781119147315e-06, + "loss": 1.6491, + "step": 374 + }, + { + "epoch": 0.3852080123266564, + "grad_norm": 2.608980178833008, + "learning_rate": 4.870574800152265e-06, + "loss": 1.6757, + "step": 375 + }, + { + "epoch": 0.38623523369286084, + "grad_norm": 2.9557063579559326, + "learning_rate": 4.868671488389798e-06, + "loss": 1.607, + "step": 376 + }, + { + "epoch": 0.3872624550590652, + "grad_norm": 2.982243537902832, + "learning_rate": 4.8667681766273316e-06, + "loss": 1.6169, + "step": 377 + }, + { + "epoch": 0.3882896764252696, + "grad_norm": 2.775012254714966, + "learning_rate": 4.864864864864866e-06, + "loss": 1.5835, + "step": 378 + }, + { + "epoch": 0.38931689779147405, + "grad_norm": 2.9500741958618164, + "learning_rate": 4.862961553102399e-06, + "loss": 1.6504, + "step": 379 + }, + { + "epoch": 0.39034411915767847, + "grad_norm": 2.889702796936035, + "learning_rate": 4.8610582413399325e-06, + "loss": 1.6814, + "step": 380 + }, + { + "epoch": 0.39034411915767847, + "eval_loss": 1.9426418542861938, + "eval_runtime": 37.3276, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 380 + }, + { + "epoch": 0.3913713405238829, + "grad_norm": 2.5075154304504395, + "learning_rate": 4.859154929577465e-06, + "loss": 1.6368, + "step": 381 + }, + { + "epoch": 0.3923985618900873, + "grad_norm": 2.6461148262023926, + "learning_rate": 4.857251617814998e-06, + "loss": 1.6526, + "step": 382 + }, + { + "epoch": 0.39342578325629174, + "grad_norm": 2.681501626968384, + "learning_rate": 4.855348306052532e-06, + "loss": 1.6846, + "step": 383 + }, + { + "epoch": 0.39445300462249616, + "grad_norm": 2.557311773300171, + "learning_rate": 4.853444994290065e-06, + "loss": 1.6645, + "step": 384 + }, + { + "epoch": 0.3954802259887006, + "grad_norm": 2.6819186210632324, + "learning_rate": 4.8515416825275985e-06, + "loss": 1.6323, + "step": 385 + }, + { + "epoch": 0.396507447354905, + "grad_norm": 2.8803794384002686, + "learning_rate": 4.849638370765132e-06, + "loss": 1.6991, + "step": 386 + }, + { + "epoch": 0.3975346687211094, + "grad_norm": 2.7181358337402344, + "learning_rate": 4.847735059002665e-06, + "loss": 1.6256, + "step": 387 + }, + { + "epoch": 0.3985618900873138, + "grad_norm": 2.5668489933013916, + "learning_rate": 4.8458317472401985e-06, + "loss": 1.58, + "step": 388 + }, + { + "epoch": 0.3995891114535182, + "grad_norm": 2.708804130554199, + "learning_rate": 4.843928435477732e-06, + "loss": 1.67, + "step": 389 + }, + { + "epoch": 0.40061633281972264, + "grad_norm": 2.6565728187561035, + "learning_rate": 4.842025123715265e-06, + "loss": 1.622, + "step": 390 + }, + { + "epoch": 0.40061633281972264, + "eval_loss": 1.9529728889465332, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 390 + }, + { + "epoch": 0.40164355418592707, + "grad_norm": 2.8743324279785156, + "learning_rate": 4.840121811952798e-06, + "loss": 1.6618, + "step": 391 + }, + { + "epoch": 0.4026707755521315, + "grad_norm": 2.86134672164917, + "learning_rate": 4.838218500190331e-06, + "loss": 1.6544, + "step": 392 + }, + { + "epoch": 0.4036979969183359, + "grad_norm": 3.3492071628570557, + "learning_rate": 4.8363151884278645e-06, + "loss": 1.6683, + "step": 393 + }, + { + "epoch": 0.40472521828454033, + "grad_norm": 2.6630825996398926, + "learning_rate": 4.834411876665398e-06, + "loss": 1.6447, + "step": 394 + }, + { + "epoch": 0.40575243965074476, + "grad_norm": 2.820096492767334, + "learning_rate": 4.832508564902931e-06, + "loss": 1.6579, + "step": 395 + }, + { + "epoch": 0.4067796610169492, + "grad_norm": 3.076234817504883, + "learning_rate": 4.830605253140465e-06, + "loss": 1.6039, + "step": 396 + }, + { + "epoch": 0.40780688238315355, + "grad_norm": 2.667719841003418, + "learning_rate": 4.828701941377998e-06, + "loss": 1.6364, + "step": 397 + }, + { + "epoch": 0.40883410374935797, + "grad_norm": 3.2132771015167236, + "learning_rate": 4.826798629615531e-06, + "loss": 1.5906, + "step": 398 + }, + { + "epoch": 0.4098613251155624, + "grad_norm": 2.661484479904175, + "learning_rate": 4.824895317853065e-06, + "loss": 1.613, + "step": 399 + }, + { + "epoch": 0.4108885464817668, + "grad_norm": 3.002636671066284, + "learning_rate": 4.822992006090598e-06, + "loss": 1.6305, + "step": 400 + }, + { + "epoch": 0.4108885464817668, + "eval_loss": 1.9345510005950928, + "eval_runtime": 37.3369, + "eval_samples_per_second": 26.783, + "eval_steps_per_second": 1.125, + "step": 400 + }, + { + "epoch": 0.41191576784797124, + "grad_norm": 2.6405227184295654, + "learning_rate": 4.821088694328131e-06, + "loss": 1.5881, + "step": 401 + }, + { + "epoch": 0.41294298921417566, + "grad_norm": 3.006169080734253, + "learning_rate": 4.819185382565665e-06, + "loss": 1.6126, + "step": 402 + }, + { + "epoch": 0.4139702105803801, + "grad_norm": 2.7614858150482178, + "learning_rate": 4.817282070803198e-06, + "loss": 1.6695, + "step": 403 + }, + { + "epoch": 0.4149974319465845, + "grad_norm": 2.797260046005249, + "learning_rate": 4.8153787590407315e-06, + "loss": 1.6539, + "step": 404 + }, + { + "epoch": 0.41602465331278893, + "grad_norm": 2.8156301975250244, + "learning_rate": 4.813475447278265e-06, + "loss": 1.5845, + "step": 405 + }, + { + "epoch": 0.41705187467899335, + "grad_norm": 2.74100399017334, + "learning_rate": 4.811572135515798e-06, + "loss": 1.6699, + "step": 406 + }, + { + "epoch": 0.4180790960451977, + "grad_norm": 2.7785775661468506, + "learning_rate": 4.809668823753332e-06, + "loss": 1.5987, + "step": 407 + }, + { + "epoch": 0.41910631741140214, + "grad_norm": 2.663504123687744, + "learning_rate": 4.807765511990864e-06, + "loss": 1.5987, + "step": 408 + }, + { + "epoch": 0.42013353877760656, + "grad_norm": 2.903374433517456, + "learning_rate": 4.8058622002283975e-06, + "loss": 1.6302, + "step": 409 + }, + { + "epoch": 0.421160760143811, + "grad_norm": 2.696049690246582, + "learning_rate": 4.803958888465931e-06, + "loss": 1.671, + "step": 410 + }, + { + "epoch": 0.421160760143811, + "eval_loss": 1.9333878755569458, + "eval_runtime": 37.2472, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 410 + }, + { + "epoch": 0.4221879815100154, + "grad_norm": 2.486461877822876, + "learning_rate": 4.802055576703464e-06, + "loss": 1.6282, + "step": 411 + }, + { + "epoch": 0.42321520287621983, + "grad_norm": 2.6812057495117188, + "learning_rate": 4.8001522649409976e-06, + "loss": 1.7144, + "step": 412 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.8370249271392822, + "learning_rate": 4.798248953178531e-06, + "loss": 1.5979, + "step": 413 + }, + { + "epoch": 0.4252696456086287, + "grad_norm": 2.6073036193847656, + "learning_rate": 4.796345641416064e-06, + "loss": 1.5703, + "step": 414 + }, + { + "epoch": 0.4262968669748331, + "grad_norm": 2.5891220569610596, + "learning_rate": 4.794442329653598e-06, + "loss": 1.6212, + "step": 415 + }, + { + "epoch": 0.42732408834103747, + "grad_norm": 2.6296865940093994, + "learning_rate": 4.792539017891131e-06, + "loss": 1.6711, + "step": 416 + }, + { + "epoch": 0.4283513097072419, + "grad_norm": 2.573472499847412, + "learning_rate": 4.790635706128664e-06, + "loss": 1.5994, + "step": 417 + }, + { + "epoch": 0.4293785310734463, + "grad_norm": 2.734464168548584, + "learning_rate": 4.788732394366197e-06, + "loss": 1.6383, + "step": 418 + }, + { + "epoch": 0.43040575243965074, + "grad_norm": 2.6412851810455322, + "learning_rate": 4.786829082603731e-06, + "loss": 1.5646, + "step": 419 + }, + { + "epoch": 0.43143297380585516, + "grad_norm": 2.5065555572509766, + "learning_rate": 4.7849257708412645e-06, + "loss": 1.5636, + "step": 420 + }, + { + "epoch": 0.43143297380585516, + "eval_loss": 1.9357348680496216, + "eval_runtime": 37.372, + "eval_samples_per_second": 26.758, + "eval_steps_per_second": 1.124, + "step": 420 + }, + { + "epoch": 0.4324601951720596, + "grad_norm": 2.4889843463897705, + "learning_rate": 4.783022459078798e-06, + "loss": 1.6457, + "step": 421 + }, + { + "epoch": 0.433487416538264, + "grad_norm": 2.544801950454712, + "learning_rate": 4.781119147316331e-06, + "loss": 1.5897, + "step": 422 + }, + { + "epoch": 0.4345146379044684, + "grad_norm": 2.7834064960479736, + "learning_rate": 4.7792158355538645e-06, + "loss": 1.6174, + "step": 423 + }, + { + "epoch": 0.43554185927067285, + "grad_norm": 2.602952480316162, + "learning_rate": 4.777312523791398e-06, + "loss": 1.6152, + "step": 424 + }, + { + "epoch": 0.43656908063687727, + "grad_norm": 2.629746198654175, + "learning_rate": 4.77540921202893e-06, + "loss": 1.5991, + "step": 425 + }, + { + "epoch": 0.43759630200308164, + "grad_norm": 2.4987270832061768, + "learning_rate": 4.773505900266464e-06, + "loss": 1.6342, + "step": 426 + }, + { + "epoch": 0.43862352336928606, + "grad_norm": 2.607611894607544, + "learning_rate": 4.771602588503997e-06, + "loss": 1.6145, + "step": 427 + }, + { + "epoch": 0.4396507447354905, + "grad_norm": 2.5654876232147217, + "learning_rate": 4.7696992767415305e-06, + "loss": 1.6162, + "step": 428 + }, + { + "epoch": 0.4406779661016949, + "grad_norm": 2.5701870918273926, + "learning_rate": 4.767795964979064e-06, + "loss": 1.6516, + "step": 429 + }, + { + "epoch": 0.44170518746789933, + "grad_norm": 2.8381104469299316, + "learning_rate": 4.765892653216597e-06, + "loss": 1.659, + "step": 430 + }, + { + "epoch": 0.44170518746789933, + "eval_loss": 1.9259562492370605, + "eval_runtime": 37.3906, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 430 + }, + { + "epoch": 0.44273240883410375, + "grad_norm": 2.5777862071990967, + "learning_rate": 4.763989341454131e-06, + "loss": 1.6571, + "step": 431 + }, + { + "epoch": 0.4437596302003082, + "grad_norm": 2.6656861305236816, + "learning_rate": 4.762086029691664e-06, + "loss": 1.5786, + "step": 432 + }, + { + "epoch": 0.4447868515665126, + "grad_norm": 2.724851131439209, + "learning_rate": 4.760182717929197e-06, + "loss": 1.5916, + "step": 433 + }, + { + "epoch": 0.445814072932717, + "grad_norm": 2.532191514968872, + "learning_rate": 4.758279406166731e-06, + "loss": 1.608, + "step": 434 + }, + { + "epoch": 0.44684129429892144, + "grad_norm": 2.6252050399780273, + "learning_rate": 4.756376094404263e-06, + "loss": 1.5297, + "step": 435 + }, + { + "epoch": 0.4478685156651258, + "grad_norm": 2.674530029296875, + "learning_rate": 4.7544727826417966e-06, + "loss": 1.615, + "step": 436 + }, + { + "epoch": 0.44889573703133023, + "grad_norm": 2.9209136962890625, + "learning_rate": 4.75256947087933e-06, + "loss": 1.6749, + "step": 437 + }, + { + "epoch": 0.44992295839753466, + "grad_norm": 2.4959394931793213, + "learning_rate": 4.750666159116863e-06, + "loss": 1.6186, + "step": 438 + }, + { + "epoch": 0.4509501797637391, + "grad_norm": 2.6317503452301025, + "learning_rate": 4.7487628473543975e-06, + "loss": 1.5873, + "step": 439 + }, + { + "epoch": 0.4519774011299435, + "grad_norm": 2.6485610008239746, + "learning_rate": 4.746859535591931e-06, + "loss": 1.5451, + "step": 440 + }, + { + "epoch": 0.4519774011299435, + "eval_loss": 1.9235283136367798, + "eval_runtime": 37.367, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 440 + }, + { + "epoch": 0.4530046224961479, + "grad_norm": 2.491852045059204, + "learning_rate": 4.744956223829464e-06, + "loss": 1.6016, + "step": 441 + }, + { + "epoch": 0.45403184386235235, + "grad_norm": 2.5354514122009277, + "learning_rate": 4.743052912066997e-06, + "loss": 1.5795, + "step": 442 + }, + { + "epoch": 0.45505906522855677, + "grad_norm": 2.7756357192993164, + "learning_rate": 4.74114960030453e-06, + "loss": 1.6299, + "step": 443 + }, + { + "epoch": 0.4560862865947612, + "grad_norm": 2.638906955718994, + "learning_rate": 4.7392462885420635e-06, + "loss": 1.6429, + "step": 444 + }, + { + "epoch": 0.4571135079609656, + "grad_norm": 2.5858829021453857, + "learning_rate": 4.737342976779597e-06, + "loss": 1.5316, + "step": 445 + }, + { + "epoch": 0.45814072932717, + "grad_norm": 2.798567295074463, + "learning_rate": 4.73543966501713e-06, + "loss": 1.6166, + "step": 446 + }, + { + "epoch": 0.4591679506933744, + "grad_norm": 2.512143135070801, + "learning_rate": 4.7335363532546636e-06, + "loss": 1.5952, + "step": 447 + }, + { + "epoch": 0.4601951720595788, + "grad_norm": 2.7063655853271484, + "learning_rate": 4.731633041492197e-06, + "loss": 1.6051, + "step": 448 + }, + { + "epoch": 0.46122239342578325, + "grad_norm": 2.6358017921447754, + "learning_rate": 4.72972972972973e-06, + "loss": 1.6008, + "step": 449 + }, + { + "epoch": 0.4622496147919877, + "grad_norm": 2.75992751121521, + "learning_rate": 4.727826417967264e-06, + "loss": 1.644, + "step": 450 + }, + { + "epoch": 0.4622496147919877, + "eval_loss": 1.927345633506775, + "eval_runtime": 37.3707, + "eval_samples_per_second": 26.759, + "eval_steps_per_second": 1.124, + "step": 450 + }, + { + "epoch": 0.4632768361581921, + "grad_norm": 2.9103775024414062, + "learning_rate": 4.725923106204797e-06, + "loss": 1.6176, + "step": 451 + }, + { + "epoch": 0.4643040575243965, + "grad_norm": 2.627481460571289, + "learning_rate": 4.72401979444233e-06, + "loss": 1.586, + "step": 452 + }, + { + "epoch": 0.46533127889060094, + "grad_norm": 2.8432116508483887, + "learning_rate": 4.722116482679863e-06, + "loss": 1.6303, + "step": 453 + }, + { + "epoch": 0.46635850025680536, + "grad_norm": 2.8088958263397217, + "learning_rate": 4.720213170917396e-06, + "loss": 1.6163, + "step": 454 + }, + { + "epoch": 0.46738572162300973, + "grad_norm": 2.6040408611297607, + "learning_rate": 4.71830985915493e-06, + "loss": 1.6346, + "step": 455 + }, + { + "epoch": 0.46841294298921415, + "grad_norm": 2.8951191902160645, + "learning_rate": 4.716406547392463e-06, + "loss": 1.5859, + "step": 456 + }, + { + "epoch": 0.4694401643554186, + "grad_norm": 2.792473793029785, + "learning_rate": 4.714503235629996e-06, + "loss": 1.6441, + "step": 457 + }, + { + "epoch": 0.470467385721623, + "grad_norm": 2.6793878078460693, + "learning_rate": 4.71259992386753e-06, + "loss": 1.5699, + "step": 458 + }, + { + "epoch": 0.4714946070878274, + "grad_norm": 2.7223684787750244, + "learning_rate": 4.710696612105063e-06, + "loss": 1.6038, + "step": 459 + }, + { + "epoch": 0.47252182845403184, + "grad_norm": 2.8043856620788574, + "learning_rate": 4.708793300342596e-06, + "loss": 1.6018, + "step": 460 + }, + { + "epoch": 0.47252182845403184, + "eval_loss": 1.9213919639587402, + "eval_runtime": 37.3451, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 460 + }, + { + "epoch": 0.47354904982023627, + "grad_norm": 2.7657470703125, + "learning_rate": 4.70688998858013e-06, + "loss": 1.5896, + "step": 461 + }, + { + "epoch": 0.4745762711864407, + "grad_norm": 2.8417739868164062, + "learning_rate": 4.704986676817663e-06, + "loss": 1.5892, + "step": 462 + }, + { + "epoch": 0.4756034925526451, + "grad_norm": 2.5660197734832764, + "learning_rate": 4.7030833650551965e-06, + "loss": 1.615, + "step": 463 + }, + { + "epoch": 0.47663071391884954, + "grad_norm": 2.762880325317383, + "learning_rate": 4.70118005329273e-06, + "loss": 1.5747, + "step": 464 + }, + { + "epoch": 0.4776579352850539, + "grad_norm": 2.7191319465637207, + "learning_rate": 4.699276741530263e-06, + "loss": 1.6529, + "step": 465 + }, + { + "epoch": 0.4786851566512583, + "grad_norm": 2.4995453357696533, + "learning_rate": 4.697373429767797e-06, + "loss": 1.6285, + "step": 466 + }, + { + "epoch": 0.47971237801746275, + "grad_norm": 2.8300375938415527, + "learning_rate": 4.69547011800533e-06, + "loss": 1.6171, + "step": 467 + }, + { + "epoch": 0.48073959938366717, + "grad_norm": 2.6028811931610107, + "learning_rate": 4.693566806242863e-06, + "loss": 1.6014, + "step": 468 + }, + { + "epoch": 0.4817668207498716, + "grad_norm": 3.080156087875366, + "learning_rate": 4.691663494480397e-06, + "loss": 1.6055, + "step": 469 + }, + { + "epoch": 0.482794042116076, + "grad_norm": 2.6707916259765625, + "learning_rate": 4.689760182717929e-06, + "loss": 1.627, + "step": 470 + }, + { + "epoch": 0.482794042116076, + "eval_loss": 1.9015462398529053, + "eval_runtime": 37.3476, + "eval_samples_per_second": 26.776, + "eval_steps_per_second": 1.125, + "step": 470 + }, + { + "epoch": 0.48382126348228044, + "grad_norm": 2.6820614337921143, + "learning_rate": 4.6878568709554626e-06, + "loss": 1.5261, + "step": 471 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.841031789779663, + "learning_rate": 4.685953559192996e-06, + "loss": 1.5903, + "step": 472 + }, + { + "epoch": 0.4858757062146893, + "grad_norm": 2.470867156982422, + "learning_rate": 4.684050247430529e-06, + "loss": 1.6091, + "step": 473 + }, + { + "epoch": 0.4869029275808937, + "grad_norm": 2.4955153465270996, + "learning_rate": 4.682146935668063e-06, + "loss": 1.544, + "step": 474 + }, + { + "epoch": 0.4879301489470981, + "grad_norm": 2.9290695190429688, + "learning_rate": 4.680243623905596e-06, + "loss": 1.5893, + "step": 475 + }, + { + "epoch": 0.4889573703133025, + "grad_norm": 2.5902767181396484, + "learning_rate": 4.678340312143129e-06, + "loss": 1.6117, + "step": 476 + }, + { + "epoch": 0.4899845916795069, + "grad_norm": 2.6067540645599365, + "learning_rate": 4.676437000380663e-06, + "loss": 1.5703, + "step": 477 + }, + { + "epoch": 0.49101181304571134, + "grad_norm": 2.490105152130127, + "learning_rate": 4.674533688618196e-06, + "loss": 1.6931, + "step": 478 + }, + { + "epoch": 0.49203903441191577, + "grad_norm": 2.7736282348632812, + "learning_rate": 4.6726303768557295e-06, + "loss": 1.6204, + "step": 479 + }, + { + "epoch": 0.4930662557781202, + "grad_norm": 2.741196870803833, + "learning_rate": 4.670727065093263e-06, + "loss": 1.5331, + "step": 480 + }, + { + "epoch": 0.4930662557781202, + "eval_loss": 1.9095442295074463, + "eval_runtime": 37.3351, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 480 + }, + { + "epoch": 0.4940934771443246, + "grad_norm": 2.7624332904815674, + "learning_rate": 4.668823753330796e-06, + "loss": 1.5591, + "step": 481 + }, + { + "epoch": 0.49512069851052903, + "grad_norm": 2.86407470703125, + "learning_rate": 4.6669204415683295e-06, + "loss": 1.5844, + "step": 482 + }, + { + "epoch": 0.49614791987673346, + "grad_norm": 2.770030975341797, + "learning_rate": 4.665017129805863e-06, + "loss": 1.6515, + "step": 483 + }, + { + "epoch": 0.4971751412429379, + "grad_norm": 2.569862127304077, + "learning_rate": 4.663113818043396e-06, + "loss": 1.6228, + "step": 484 + }, + { + "epoch": 0.49820236260914225, + "grad_norm": 2.7176733016967773, + "learning_rate": 4.66121050628093e-06, + "loss": 1.5938, + "step": 485 + }, + { + "epoch": 0.49922958397534667, + "grad_norm": 2.4514758586883545, + "learning_rate": 4.659307194518463e-06, + "loss": 1.6307, + "step": 486 + }, + { + "epoch": 0.5002568053415511, + "grad_norm": 2.5439529418945312, + "learning_rate": 4.6574038827559955e-06, + "loss": 1.5855, + "step": 487 + }, + { + "epoch": 0.5012840267077555, + "grad_norm": 2.6395790576934814, + "learning_rate": 4.655500570993529e-06, + "loss": 1.6124, + "step": 488 + }, + { + "epoch": 0.50231124807396, + "grad_norm": 2.69413685798645, + "learning_rate": 4.653597259231062e-06, + "loss": 1.6358, + "step": 489 + }, + { + "epoch": 0.5033384694401644, + "grad_norm": 2.693129777908325, + "learning_rate": 4.651693947468596e-06, + "loss": 1.5218, + "step": 490 + }, + { + "epoch": 0.5033384694401644, + "eval_loss": 1.9201244115829468, + "eval_runtime": 37.3324, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 490 + }, + { + "epoch": 0.5043656908063687, + "grad_norm": 2.630596876144409, + "learning_rate": 4.649790635706129e-06, + "loss": 1.6719, + "step": 491 + }, + { + "epoch": 0.5053929121725732, + "grad_norm": 2.7230262756347656, + "learning_rate": 4.647887323943662e-06, + "loss": 1.5705, + "step": 492 + }, + { + "epoch": 0.5064201335387776, + "grad_norm": 2.529944896697998, + "learning_rate": 4.645984012181196e-06, + "loss": 1.5767, + "step": 493 + }, + { + "epoch": 0.507447354904982, + "grad_norm": 2.546849489212036, + "learning_rate": 4.644080700418729e-06, + "loss": 1.5039, + "step": 494 + }, + { + "epoch": 0.5084745762711864, + "grad_norm": 2.6698246002197266, + "learning_rate": 4.642177388656262e-06, + "loss": 1.6229, + "step": 495 + }, + { + "epoch": 0.5095017976373909, + "grad_norm": 2.7093148231506348, + "learning_rate": 4.640274076893796e-06, + "loss": 1.5797, + "step": 496 + }, + { + "epoch": 0.5105290190035953, + "grad_norm": 2.48629093170166, + "learning_rate": 4.638370765131328e-06, + "loss": 1.6366, + "step": 497 + }, + { + "epoch": 0.5115562403697997, + "grad_norm": 2.5818612575531006, + "learning_rate": 4.636467453368862e-06, + "loss": 1.5608, + "step": 498 + }, + { + "epoch": 0.5125834617360041, + "grad_norm": 2.6263651847839355, + "learning_rate": 4.634564141606396e-06, + "loss": 1.5518, + "step": 499 + }, + { + "epoch": 0.5136106831022085, + "grad_norm": 2.5724668502807617, + "learning_rate": 4.632660829843929e-06, + "loss": 1.6195, + "step": 500 + }, + { + "epoch": 0.5136106831022085, + "eval_loss": 1.9094111919403076, + "eval_runtime": 37.3483, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 500 + }, + { + "epoch": 0.514637904468413, + "grad_norm": 2.5028088092803955, + "learning_rate": 4.630757518081463e-06, + "loss": 1.5454, + "step": 501 + }, + { + "epoch": 0.5156651258346173, + "grad_norm": 2.588115692138672, + "learning_rate": 4.628854206318996e-06, + "loss": 1.6676, + "step": 502 + }, + { + "epoch": 0.5166923472008218, + "grad_norm": 2.9701285362243652, + "learning_rate": 4.626950894556529e-06, + "loss": 1.5822, + "step": 503 + }, + { + "epoch": 0.5177195685670262, + "grad_norm": 2.593867301940918, + "learning_rate": 4.625047582794062e-06, + "loss": 1.6091, + "step": 504 + }, + { + "epoch": 0.5187467899332306, + "grad_norm": 2.559642791748047, + "learning_rate": 4.623144271031595e-06, + "loss": 1.5359, + "step": 505 + }, + { + "epoch": 0.519774011299435, + "grad_norm": 2.4158153533935547, + "learning_rate": 4.6212409592691286e-06, + "loss": 1.5476, + "step": 506 + }, + { + "epoch": 0.5208012326656395, + "grad_norm": 2.666099786758423, + "learning_rate": 4.619337647506662e-06, + "loss": 1.6297, + "step": 507 + }, + { + "epoch": 0.5218284540318439, + "grad_norm": 2.7026472091674805, + "learning_rate": 4.617434335744195e-06, + "loss": 1.5891, + "step": 508 + }, + { + "epoch": 0.5228556753980482, + "grad_norm": 2.4591612815856934, + "learning_rate": 4.615531023981729e-06, + "loss": 1.5546, + "step": 509 + }, + { + "epoch": 0.5238828967642527, + "grad_norm": 2.870255708694458, + "learning_rate": 4.613627712219262e-06, + "loss": 1.5817, + "step": 510 + }, + { + "epoch": 0.5238828967642527, + "eval_loss": 1.910629153251648, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 510 + }, + { + "epoch": 0.5249101181304571, + "grad_norm": 2.653801679611206, + "learning_rate": 4.611724400456795e-06, + "loss": 1.586, + "step": 511 + }, + { + "epoch": 0.5259373394966615, + "grad_norm": 2.4235377311706543, + "learning_rate": 4.609821088694329e-06, + "loss": 1.5886, + "step": 512 + }, + { + "epoch": 0.5269645608628659, + "grad_norm": 2.8782477378845215, + "learning_rate": 4.607917776931862e-06, + "loss": 1.5634, + "step": 513 + }, + { + "epoch": 0.5279917822290704, + "grad_norm": 2.792876958847046, + "learning_rate": 4.606014465169395e-06, + "loss": 1.6073, + "step": 514 + }, + { + "epoch": 0.5290190035952748, + "grad_norm": 2.5579030513763428, + "learning_rate": 4.604111153406928e-06, + "loss": 1.6161, + "step": 515 + }, + { + "epoch": 0.5300462249614792, + "grad_norm": 2.6486189365386963, + "learning_rate": 4.602207841644461e-06, + "loss": 1.5297, + "step": 516 + }, + { + "epoch": 0.5310734463276836, + "grad_norm": 2.9047019481658936, + "learning_rate": 4.600304529881995e-06, + "loss": 1.626, + "step": 517 + }, + { + "epoch": 0.5321006676938881, + "grad_norm": 2.9164223670959473, + "learning_rate": 4.598401218119528e-06, + "loss": 1.6051, + "step": 518 + }, + { + "epoch": 0.5331278890600925, + "grad_norm": 2.544311046600342, + "learning_rate": 4.5964979063570614e-06, + "loss": 1.6247, + "step": 519 + }, + { + "epoch": 0.5341551104262968, + "grad_norm": 2.8632311820983887, + "learning_rate": 4.594594594594596e-06, + "loss": 1.5615, + "step": 520 + }, + { + "epoch": 0.5341551104262968, + "eval_loss": 1.9171669483184814, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 520 + }, + { + "epoch": 0.5351823317925013, + "grad_norm": 2.507697105407715, + "learning_rate": 4.592691282832128e-06, + "loss": 1.5929, + "step": 521 + }, + { + "epoch": 0.5362095531587057, + "grad_norm": 2.657752752304077, + "learning_rate": 4.5907879710696615e-06, + "loss": 1.6159, + "step": 522 + }, + { + "epoch": 0.5372367745249101, + "grad_norm": 2.32254695892334, + "learning_rate": 4.588884659307195e-06, + "loss": 1.5318, + "step": 523 + }, + { + "epoch": 0.5382639958911145, + "grad_norm": 2.5979015827178955, + "learning_rate": 4.586981347544728e-06, + "loss": 1.5999, + "step": 524 + }, + { + "epoch": 0.539291217257319, + "grad_norm": 2.839956521987915, + "learning_rate": 4.585078035782262e-06, + "loss": 1.547, + "step": 525 + }, + { + "epoch": 0.5403184386235234, + "grad_norm": 2.700131893157959, + "learning_rate": 4.583174724019795e-06, + "loss": 1.5831, + "step": 526 + }, + { + "epoch": 0.5413456599897278, + "grad_norm": 2.365183115005493, + "learning_rate": 4.581271412257328e-06, + "loss": 1.5357, + "step": 527 + }, + { + "epoch": 0.5423728813559322, + "grad_norm": 2.675567865371704, + "learning_rate": 4.579368100494862e-06, + "loss": 1.5212, + "step": 528 + }, + { + "epoch": 0.5434001027221366, + "grad_norm": 2.9096713066101074, + "learning_rate": 4.577464788732395e-06, + "loss": 1.6196, + "step": 529 + }, + { + "epoch": 0.544427324088341, + "grad_norm": 2.6186270713806152, + "learning_rate": 4.575561476969928e-06, + "loss": 1.5895, + "step": 530 + }, + { + "epoch": 0.544427324088341, + "eval_loss": 1.926998496055603, + "eval_runtime": 37.2248, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 530 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.7291345596313477, + "learning_rate": 4.573658165207461e-06, + "loss": 1.6335, + "step": 531 + }, + { + "epoch": 0.5464817668207499, + "grad_norm": 2.7084522247314453, + "learning_rate": 4.571754853444994e-06, + "loss": 1.5968, + "step": 532 + }, + { + "epoch": 0.5475089881869543, + "grad_norm": 2.699721336364746, + "learning_rate": 4.569851541682528e-06, + "loss": 1.5775, + "step": 533 + }, + { + "epoch": 0.5485362095531587, + "grad_norm": 2.602526903152466, + "learning_rate": 4.567948229920061e-06, + "loss": 1.5497, + "step": 534 + }, + { + "epoch": 0.5495634309193631, + "grad_norm": 2.597362518310547, + "learning_rate": 4.566044918157594e-06, + "loss": 1.5218, + "step": 535 + }, + { + "epoch": 0.5505906522855676, + "grad_norm": 2.6624414920806885, + "learning_rate": 4.564141606395128e-06, + "loss": 1.5553, + "step": 536 + }, + { + "epoch": 0.551617873651772, + "grad_norm": 2.750807285308838, + "learning_rate": 4.562238294632661e-06, + "loss": 1.6078, + "step": 537 + }, + { + "epoch": 0.5526450950179764, + "grad_norm": 2.7480952739715576, + "learning_rate": 4.5603349828701945e-06, + "loss": 1.5707, + "step": 538 + }, + { + "epoch": 0.5536723163841808, + "grad_norm": 2.6287965774536133, + "learning_rate": 4.558431671107728e-06, + "loss": 1.5961, + "step": 539 + }, + { + "epoch": 0.5546995377503852, + "grad_norm": 2.721311569213867, + "learning_rate": 4.556528359345261e-06, + "loss": 1.6335, + "step": 540 + }, + { + "epoch": 0.5546995377503852, + "eval_loss": 1.9290324449539185, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5557267591165896, + "grad_norm": 2.470325469970703, + "learning_rate": 4.5546250475827946e-06, + "loss": 1.5645, + "step": 541 + }, + { + "epoch": 0.556753980482794, + "grad_norm": 2.6089508533477783, + "learning_rate": 4.552721735820328e-06, + "loss": 1.5256, + "step": 542 + }, + { + "epoch": 0.5577812018489985, + "grad_norm": 2.541785478591919, + "learning_rate": 4.550818424057861e-06, + "loss": 1.5641, + "step": 543 + }, + { + "epoch": 0.5588084232152029, + "grad_norm": 2.6722488403320312, + "learning_rate": 4.548915112295395e-06, + "loss": 1.5708, + "step": 544 + }, + { + "epoch": 0.5598356445814073, + "grad_norm": 2.583695650100708, + "learning_rate": 4.547011800532928e-06, + "loss": 1.5799, + "step": 545 + }, + { + "epoch": 0.5608628659476117, + "grad_norm": 2.956885576248169, + "learning_rate": 4.545108488770461e-06, + "loss": 1.5345, + "step": 546 + }, + { + "epoch": 0.5618900873138162, + "grad_norm": 2.5534746646881104, + "learning_rate": 4.543205177007995e-06, + "loss": 1.636, + "step": 547 + }, + { + "epoch": 0.5629173086800205, + "grad_norm": 2.8133325576782227, + "learning_rate": 4.541301865245527e-06, + "loss": 1.5678, + "step": 548 + }, + { + "epoch": 0.5639445300462249, + "grad_norm": 2.5969228744506836, + "learning_rate": 4.539398553483061e-06, + "loss": 1.5975, + "step": 549 + }, + { + "epoch": 0.5649717514124294, + "grad_norm": 2.7594687938690186, + "learning_rate": 4.537495241720594e-06, + "loss": 1.5622, + "step": 550 + }, + { + "epoch": 0.5649717514124294, + "eval_loss": 1.8870681524276733, + "eval_runtime": 37.2094, + "eval_samples_per_second": 26.875, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.5659989727786338, + "grad_norm": 2.7413411140441895, + "learning_rate": 4.535591929958127e-06, + "loss": 1.6031, + "step": 551 + }, + { + "epoch": 0.5670261941448382, + "grad_norm": 2.6182661056518555, + "learning_rate": 4.533688618195661e-06, + "loss": 1.5828, + "step": 552 + }, + { + "epoch": 0.5680534155110426, + "grad_norm": 2.5016422271728516, + "learning_rate": 4.531785306433194e-06, + "loss": 1.5712, + "step": 553 + }, + { + "epoch": 0.5690806368772471, + "grad_norm": 2.7788732051849365, + "learning_rate": 4.529881994670727e-06, + "loss": 1.5472, + "step": 554 + }, + { + "epoch": 0.5701078582434514, + "grad_norm": 2.7602436542510986, + "learning_rate": 4.527978682908261e-06, + "loss": 1.5526, + "step": 555 + }, + { + "epoch": 0.5711350796096559, + "grad_norm": 2.5898544788360596, + "learning_rate": 4.526075371145794e-06, + "loss": 1.5644, + "step": 556 + }, + { + "epoch": 0.5721623009758603, + "grad_norm": 2.439344882965088, + "learning_rate": 4.5241720593833275e-06, + "loss": 1.5892, + "step": 557 + }, + { + "epoch": 0.5731895223420647, + "grad_norm": 2.6324563026428223, + "learning_rate": 4.52226874762086e-06, + "loss": 1.5729, + "step": 558 + }, + { + "epoch": 0.5742167437082691, + "grad_norm": 2.674042224884033, + "learning_rate": 4.520365435858393e-06, + "loss": 1.6016, + "step": 559 + }, + { + "epoch": 0.5752439650744735, + "grad_norm": 2.692619800567627, + "learning_rate": 4.518462124095928e-06, + "loss": 1.6229, + "step": 560 + }, + { + "epoch": 0.5752439650744735, + "eval_loss": 1.8689097166061401, + "eval_runtime": 37.2177, + "eval_samples_per_second": 26.869, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.576271186440678, + "grad_norm": 2.636924982070923, + "learning_rate": 4.516558812333461e-06, + "loss": 1.5748, + "step": 561 + }, + { + "epoch": 0.5772984078068824, + "grad_norm": 2.549285888671875, + "learning_rate": 4.514655500570994e-06, + "loss": 1.5656, + "step": 562 + }, + { + "epoch": 0.5783256291730868, + "grad_norm": 2.5873067378997803, + "learning_rate": 4.512752188808528e-06, + "loss": 1.631, + "step": 563 + }, + { + "epoch": 0.5793528505392912, + "grad_norm": 2.552121162414551, + "learning_rate": 4.510848877046061e-06, + "loss": 1.6334, + "step": 564 + }, + { + "epoch": 0.5803800719054957, + "grad_norm": 2.697615146636963, + "learning_rate": 4.5089455652835936e-06, + "loss": 1.5951, + "step": 565 + }, + { + "epoch": 0.5814072932717, + "grad_norm": 2.64585280418396, + "learning_rate": 4.507042253521127e-06, + "loss": 1.556, + "step": 566 + }, + { + "epoch": 0.5824345146379045, + "grad_norm": 2.6383707523345947, + "learning_rate": 4.50513894175866e-06, + "loss": 1.5184, + "step": 567 + }, + { + "epoch": 0.5834617360041089, + "grad_norm": 2.7571511268615723, + "learning_rate": 4.503235629996194e-06, + "loss": 1.5984, + "step": 568 + }, + { + "epoch": 0.5844889573703133, + "grad_norm": 2.6637237071990967, + "learning_rate": 4.501332318233727e-06, + "loss": 1.5989, + "step": 569 + }, + { + "epoch": 0.5855161787365177, + "grad_norm": 2.725754976272583, + "learning_rate": 4.49942900647126e-06, + "loss": 1.5594, + "step": 570 + }, + { + "epoch": 0.5855161787365177, + "eval_loss": 1.8722314834594727, + "eval_runtime": 37.1859, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 570 + }, + { + "epoch": 0.5865434001027221, + "grad_norm": 2.6657378673553467, + "learning_rate": 4.497525694708794e-06, + "loss": 1.5954, + "step": 571 + }, + { + "epoch": 0.5875706214689266, + "grad_norm": 2.78940486907959, + "learning_rate": 4.495622382946327e-06, + "loss": 1.592, + "step": 572 + }, + { + "epoch": 0.588597842835131, + "grad_norm": 2.580956220626831, + "learning_rate": 4.4937190711838605e-06, + "loss": 1.5444, + "step": 573 + }, + { + "epoch": 0.5896250642013354, + "grad_norm": 2.7783710956573486, + "learning_rate": 4.491815759421394e-06, + "loss": 1.6257, + "step": 574 + }, + { + "epoch": 0.5906522855675398, + "grad_norm": 2.9381537437438965, + "learning_rate": 4.489912447658926e-06, + "loss": 1.5489, + "step": 575 + }, + { + "epoch": 0.5916795069337443, + "grad_norm": 2.7915165424346924, + "learning_rate": 4.48800913589646e-06, + "loss": 1.5471, + "step": 576 + }, + { + "epoch": 0.5927067282999486, + "grad_norm": 2.5928280353546143, + "learning_rate": 4.486105824133993e-06, + "loss": 1.5864, + "step": 577 + }, + { + "epoch": 0.593733949666153, + "grad_norm": 2.844512462615967, + "learning_rate": 4.4842025123715264e-06, + "loss": 1.5841, + "step": 578 + }, + { + "epoch": 0.5947611710323575, + "grad_norm": 2.562718391418457, + "learning_rate": 4.48229920060906e-06, + "loss": 1.5476, + "step": 579 + }, + { + "epoch": 0.5957883923985619, + "grad_norm": 2.535759925842285, + "learning_rate": 4.480395888846594e-06, + "loss": 1.5274, + "step": 580 + }, + { + "epoch": 0.5957883923985619, + "eval_loss": 1.8881579637527466, + "eval_runtime": 37.2533, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.5968156137647663, + "grad_norm": 2.6940653324127197, + "learning_rate": 4.478492577084127e-06, + "loss": 1.6377, + "step": 581 + }, + { + "epoch": 0.5978428351309707, + "grad_norm": 2.4839298725128174, + "learning_rate": 4.47658926532166e-06, + "loss": 1.534, + "step": 582 + }, + { + "epoch": 0.5988700564971752, + "grad_norm": 2.5035688877105713, + "learning_rate": 4.474685953559193e-06, + "loss": 1.5764, + "step": 583 + }, + { + "epoch": 0.5998972778633795, + "grad_norm": 2.791907548904419, + "learning_rate": 4.472782641796727e-06, + "loss": 1.704, + "step": 584 + }, + { + "epoch": 0.600924499229584, + "grad_norm": 2.5384254455566406, + "learning_rate": 4.47087933003426e-06, + "loss": 1.5184, + "step": 585 + }, + { + "epoch": 0.6019517205957884, + "grad_norm": 2.4990930557250977, + "learning_rate": 4.468976018271793e-06, + "loss": 1.5335, + "step": 586 + }, + { + "epoch": 0.6029789419619929, + "grad_norm": 2.470872163772583, + "learning_rate": 4.467072706509327e-06, + "loss": 1.5464, + "step": 587 + }, + { + "epoch": 0.6040061633281972, + "grad_norm": 2.539036989212036, + "learning_rate": 4.46516939474686e-06, + "loss": 1.5793, + "step": 588 + }, + { + "epoch": 0.6050333846944016, + "grad_norm": 2.495309829711914, + "learning_rate": 4.463266082984393e-06, + "loss": 1.5312, + "step": 589 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.5937912464141846, + "learning_rate": 4.461362771221927e-06, + "loss": 1.6508, + "step": 590 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.8893816471099854, + "eval_runtime": 37.2539, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 590 + }, + { + "epoch": 0.6070878274268104, + "grad_norm": 2.53300404548645, + "learning_rate": 4.45945945945946e-06, + "loss": 1.5408, + "step": 591 + }, + { + "epoch": 0.6081150487930149, + "grad_norm": 2.5760276317596436, + "learning_rate": 4.457556147696993e-06, + "loss": 1.5575, + "step": 592 + }, + { + "epoch": 0.6091422701592193, + "grad_norm": 2.638353109359741, + "learning_rate": 4.455652835934526e-06, + "loss": 1.6286, + "step": 593 + }, + { + "epoch": 0.6101694915254238, + "grad_norm": 2.916550397872925, + "learning_rate": 4.453749524172059e-06, + "loss": 1.6334, + "step": 594 + }, + { + "epoch": 0.6111967128916281, + "grad_norm": 2.560878038406372, + "learning_rate": 4.451846212409593e-06, + "loss": 1.5806, + "step": 595 + }, + { + "epoch": 0.6122239342578326, + "grad_norm": 2.637882947921753, + "learning_rate": 4.449942900647126e-06, + "loss": 1.5791, + "step": 596 + }, + { + "epoch": 0.613251155624037, + "grad_norm": 2.752870798110962, + "learning_rate": 4.4480395888846595e-06, + "loss": 1.5117, + "step": 597 + }, + { + "epoch": 0.6142783769902413, + "grad_norm": 2.702087640762329, + "learning_rate": 4.446136277122193e-06, + "loss": 1.5783, + "step": 598 + }, + { + "epoch": 0.6153055983564458, + "grad_norm": 2.504819631576538, + "learning_rate": 4.444232965359726e-06, + "loss": 1.5136, + "step": 599 + }, + { + "epoch": 0.6163328197226502, + "grad_norm": 2.521670341491699, + "learning_rate": 4.4423296535972596e-06, + "loss": 1.5741, + "step": 600 + }, + { + "epoch": 0.6163328197226502, + "eval_loss": 1.8973714113235474, + "eval_runtime": 37.3167, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 600 + }, + { + "epoch": 0.6173600410888547, + "grad_norm": 2.927755355834961, + "learning_rate": 4.440426341834793e-06, + "loss": 1.5262, + "step": 601 + }, + { + "epoch": 0.618387262455059, + "grad_norm": 2.477053642272949, + "learning_rate": 4.438523030072326e-06, + "loss": 1.572, + "step": 602 + }, + { + "epoch": 0.6194144838212635, + "grad_norm": 2.6500253677368164, + "learning_rate": 4.43661971830986e-06, + "loss": 1.596, + "step": 603 + }, + { + "epoch": 0.6204417051874679, + "grad_norm": 2.5134825706481934, + "learning_rate": 4.434716406547393e-06, + "loss": 1.5432, + "step": 604 + }, + { + "epoch": 0.6214689265536724, + "grad_norm": 2.4142072200775146, + "learning_rate": 4.432813094784926e-06, + "loss": 1.5391, + "step": 605 + }, + { + "epoch": 0.6224961479198767, + "grad_norm": 2.752556800842285, + "learning_rate": 4.43090978302246e-06, + "loss": 1.6055, + "step": 606 + }, + { + "epoch": 0.6235233692860811, + "grad_norm": 2.5731008052825928, + "learning_rate": 4.429006471259993e-06, + "loss": 1.5648, + "step": 607 + }, + { + "epoch": 0.6245505906522856, + "grad_norm": 2.531656503677368, + "learning_rate": 4.4271031594975265e-06, + "loss": 1.5492, + "step": 608 + }, + { + "epoch": 0.6255778120184899, + "grad_norm": 2.6665475368499756, + "learning_rate": 4.425199847735059e-06, + "loss": 1.5194, + "step": 609 + }, + { + "epoch": 0.6266050333846944, + "grad_norm": 2.437051773071289, + "learning_rate": 4.423296535972592e-06, + "loss": 1.5271, + "step": 610 + }, + { + "epoch": 0.6266050333846944, + "eval_loss": 1.88953697681427, + "eval_runtime": 36.8703, + "eval_samples_per_second": 27.122, + "eval_steps_per_second": 1.139, + "step": 610 + }, + { + "epoch": 0.6276322547508988, + "grad_norm": 2.5932440757751465, + "learning_rate": 4.421393224210126e-06, + "loss": 1.5563, + "step": 611 + }, + { + "epoch": 0.6286594761171033, + "grad_norm": 2.589447021484375, + "learning_rate": 4.419489912447659e-06, + "loss": 1.5207, + "step": 612 + }, + { + "epoch": 0.6296866974833076, + "grad_norm": 2.570096731185913, + "learning_rate": 4.4175866006851924e-06, + "loss": 1.5995, + "step": 613 + }, + { + "epoch": 0.6307139188495121, + "grad_norm": 2.5356791019439697, + "learning_rate": 4.415683288922726e-06, + "loss": 1.5703, + "step": 614 + }, + { + "epoch": 0.6317411402157165, + "grad_norm": 2.705193281173706, + "learning_rate": 4.413779977160259e-06, + "loss": 1.5405, + "step": 615 + }, + { + "epoch": 0.632768361581921, + "grad_norm": 2.555168390274048, + "learning_rate": 4.4118766653977925e-06, + "loss": 1.5487, + "step": 616 + }, + { + "epoch": 0.6337955829481253, + "grad_norm": 2.804696559906006, + "learning_rate": 4.409973353635326e-06, + "loss": 1.6094, + "step": 617 + }, + { + "epoch": 0.6348228043143297, + "grad_norm": 2.6889376640319824, + "learning_rate": 4.408070041872859e-06, + "loss": 1.5656, + "step": 618 + }, + { + "epoch": 0.6358500256805342, + "grad_norm": 2.576746702194214, + "learning_rate": 4.406166730110393e-06, + "loss": 1.5244, + "step": 619 + }, + { + "epoch": 0.6368772470467385, + "grad_norm": 2.811156749725342, + "learning_rate": 4.404263418347925e-06, + "loss": 1.5153, + "step": 620 + }, + { + "epoch": 0.6368772470467385, + "eval_loss": 1.8949775695800781, + "eval_runtime": 36.978, + "eval_samples_per_second": 27.043, + "eval_steps_per_second": 1.136, + "step": 620 + }, + { + "epoch": 0.637904468412943, + "grad_norm": 2.6560428142547607, + "learning_rate": 4.402360106585459e-06, + "loss": 1.5347, + "step": 621 + }, + { + "epoch": 0.6389316897791474, + "grad_norm": 2.668100118637085, + "learning_rate": 4.400456794822993e-06, + "loss": 1.5236, + "step": 622 + }, + { + "epoch": 0.6399589111453519, + "grad_norm": 2.925978660583496, + "learning_rate": 4.398553483060526e-06, + "loss": 1.5494, + "step": 623 + }, + { + "epoch": 0.6409861325115562, + "grad_norm": 2.5964415073394775, + "learning_rate": 4.396650171298059e-06, + "loss": 1.522, + "step": 624 + }, + { + "epoch": 0.6420133538777607, + "grad_norm": 2.643888235092163, + "learning_rate": 4.394746859535593e-06, + "loss": 1.5332, + "step": 625 + }, + { + "epoch": 0.6430405752439651, + "grad_norm": 2.805192708969116, + "learning_rate": 4.392843547773125e-06, + "loss": 1.5031, + "step": 626 + }, + { + "epoch": 0.6440677966101694, + "grad_norm": 2.6862149238586426, + "learning_rate": 4.390940236010659e-06, + "loss": 1.5335, + "step": 627 + }, + { + "epoch": 0.6450950179763739, + "grad_norm": 2.5579185485839844, + "learning_rate": 4.389036924248192e-06, + "loss": 1.5466, + "step": 628 + }, + { + "epoch": 0.6461222393425783, + "grad_norm": 2.4550797939300537, + "learning_rate": 4.387133612485725e-06, + "loss": 1.5776, + "step": 629 + }, + { + "epoch": 0.6471494607087828, + "grad_norm": 2.6051297187805176, + "learning_rate": 4.385230300723259e-06, + "loss": 1.5435, + "step": 630 + }, + { + "epoch": 0.6471494607087828, + "eval_loss": 1.8860505819320679, + "eval_runtime": 36.989, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 630 + }, + { + "epoch": 0.6481766820749871, + "grad_norm": 2.3723294734954834, + "learning_rate": 4.383326988960792e-06, + "loss": 1.5413, + "step": 631 + }, + { + "epoch": 0.6492039034411916, + "grad_norm": 2.44370174407959, + "learning_rate": 4.3814236771983255e-06, + "loss": 1.5152, + "step": 632 + }, + { + "epoch": 0.650231124807396, + "grad_norm": 2.474081039428711, + "learning_rate": 4.379520365435859e-06, + "loss": 1.579, + "step": 633 + }, + { + "epoch": 0.6512583461736005, + "grad_norm": 2.837843179702759, + "learning_rate": 4.377617053673392e-06, + "loss": 1.559, + "step": 634 + }, + { + "epoch": 0.6522855675398048, + "grad_norm": 2.593322515487671, + "learning_rate": 4.3757137419109256e-06, + "loss": 1.552, + "step": 635 + }, + { + "epoch": 0.6533127889060092, + "grad_norm": 3.018051862716675, + "learning_rate": 4.373810430148459e-06, + "loss": 1.6231, + "step": 636 + }, + { + "epoch": 0.6543400102722137, + "grad_norm": 2.6338863372802734, + "learning_rate": 4.3719071183859914e-06, + "loss": 1.517, + "step": 637 + }, + { + "epoch": 0.655367231638418, + "grad_norm": 2.544140577316284, + "learning_rate": 4.370003806623525e-06, + "loss": 1.5763, + "step": 638 + }, + { + "epoch": 0.6563944530046225, + "grad_norm": 2.6375792026519775, + "learning_rate": 4.368100494861058e-06, + "loss": 1.5098, + "step": 639 + }, + { + "epoch": 0.6574216743708269, + "grad_norm": 2.5452051162719727, + "learning_rate": 4.3661971830985915e-06, + "loss": 1.5671, + "step": 640 + }, + { + "epoch": 0.6574216743708269, + "eval_loss": 1.881069540977478, + "eval_runtime": 36.9961, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 640 + }, + { + "epoch": 0.6584488957370314, + "grad_norm": 2.623411178588867, + "learning_rate": 4.364293871336126e-06, + "loss": 1.5215, + "step": 641 + }, + { + "epoch": 0.6594761171032357, + "grad_norm": 2.6149892807006836, + "learning_rate": 4.362390559573659e-06, + "loss": 1.5507, + "step": 642 + }, + { + "epoch": 0.6605033384694402, + "grad_norm": 2.5570266246795654, + "learning_rate": 4.3604872478111925e-06, + "loss": 1.5143, + "step": 643 + }, + { + "epoch": 0.6615305598356446, + "grad_norm": 2.4813005924224854, + "learning_rate": 4.358583936048725e-06, + "loss": 1.4518, + "step": 644 + }, + { + "epoch": 0.662557781201849, + "grad_norm": 3.0318472385406494, + "learning_rate": 4.356680624286258e-06, + "loss": 1.5812, + "step": 645 + }, + { + "epoch": 0.6635850025680534, + "grad_norm": 2.46936297416687, + "learning_rate": 4.354777312523792e-06, + "loss": 1.5396, + "step": 646 + }, + { + "epoch": 0.6646122239342578, + "grad_norm": 2.5295984745025635, + "learning_rate": 4.352874000761325e-06, + "loss": 1.586, + "step": 647 + }, + { + "epoch": 0.6656394453004623, + "grad_norm": 2.7826883792877197, + "learning_rate": 4.3509706889988584e-06, + "loss": 1.5339, + "step": 648 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.4273526668548584, + "learning_rate": 4.349067377236392e-06, + "loss": 1.598, + "step": 649 + }, + { + "epoch": 0.6676938880328711, + "grad_norm": 2.644308090209961, + "learning_rate": 4.347164065473925e-06, + "loss": 1.5889, + "step": 650 + }, + { + "epoch": 0.6676938880328711, + "eval_loss": 1.8861087560653687, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 650 + }, + { + "epoch": 0.6687211093990755, + "grad_norm": 2.827376127243042, + "learning_rate": 4.3452607537114585e-06, + "loss": 1.606, + "step": 651 + }, + { + "epoch": 0.66974833076528, + "grad_norm": 2.7463080883026123, + "learning_rate": 4.343357441948992e-06, + "loss": 1.5208, + "step": 652 + }, + { + "epoch": 0.6707755521314843, + "grad_norm": 2.6082730293273926, + "learning_rate": 4.341454130186525e-06, + "loss": 1.5597, + "step": 653 + }, + { + "epoch": 0.6718027734976888, + "grad_norm": 2.7131078243255615, + "learning_rate": 4.339550818424058e-06, + "loss": 1.526, + "step": 654 + }, + { + "epoch": 0.6728299948638932, + "grad_norm": 2.6635708808898926, + "learning_rate": 4.337647506661591e-06, + "loss": 1.5052, + "step": 655 + }, + { + "epoch": 0.6738572162300975, + "grad_norm": 2.5686581134796143, + "learning_rate": 4.3357441948991245e-06, + "loss": 1.5374, + "step": 656 + }, + { + "epoch": 0.674884437596302, + "grad_norm": 3.0532894134521484, + "learning_rate": 4.333840883136658e-06, + "loss": 1.5787, + "step": 657 + }, + { + "epoch": 0.6759116589625064, + "grad_norm": 2.6494507789611816, + "learning_rate": 4.331937571374191e-06, + "loss": 1.5121, + "step": 658 + }, + { + "epoch": 0.6769388803287109, + "grad_norm": 2.748142719268799, + "learning_rate": 4.3300342596117246e-06, + "loss": 1.5294, + "step": 659 + }, + { + "epoch": 0.6779661016949152, + "grad_norm": 2.837799310684204, + "learning_rate": 4.328130947849258e-06, + "loss": 1.5424, + "step": 660 + }, + { + "epoch": 0.6779661016949152, + "eval_loss": 1.8644919395446777, + "eval_runtime": 37.0096, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 660 + }, + { + "epoch": 0.6789933230611197, + "grad_norm": 2.7405757904052734, + "learning_rate": 4.326227636086791e-06, + "loss": 1.4949, + "step": 661 + }, + { + "epoch": 0.6800205444273241, + "grad_norm": 2.545466661453247, + "learning_rate": 4.324324324324325e-06, + "loss": 1.5445, + "step": 662 + }, + { + "epoch": 0.6810477657935285, + "grad_norm": 2.5715537071228027, + "learning_rate": 4.322421012561858e-06, + "loss": 1.5468, + "step": 663 + }, + { + "epoch": 0.6820749871597329, + "grad_norm": 2.7073662281036377, + "learning_rate": 4.320517700799391e-06, + "loss": 1.5293, + "step": 664 + }, + { + "epoch": 0.6831022085259374, + "grad_norm": 2.62264347076416, + "learning_rate": 4.318614389036925e-06, + "loss": 1.6258, + "step": 665 + }, + { + "epoch": 0.6841294298921418, + "grad_norm": 2.4506373405456543, + "learning_rate": 4.316711077274458e-06, + "loss": 1.5845, + "step": 666 + }, + { + "epoch": 0.6851566512583461, + "grad_norm": 2.849473237991333, + "learning_rate": 4.3148077655119915e-06, + "loss": 1.4959, + "step": 667 + }, + { + "epoch": 0.6861838726245506, + "grad_norm": 2.5865421295166016, + "learning_rate": 4.312904453749525e-06, + "loss": 1.599, + "step": 668 + }, + { + "epoch": 0.687211093990755, + "grad_norm": 2.4636387825012207, + "learning_rate": 4.311001141987058e-06, + "loss": 1.5495, + "step": 669 + }, + { + "epoch": 0.6882383153569595, + "grad_norm": 2.638047218322754, + "learning_rate": 4.3090978302245916e-06, + "loss": 1.5507, + "step": 670 + }, + { + "epoch": 0.6882383153569595, + "eval_loss": 1.8834826946258545, + "eval_runtime": 37.0335, + "eval_samples_per_second": 27.003, + "eval_steps_per_second": 1.134, + "step": 670 + }, + { + "epoch": 0.6892655367231638, + "grad_norm": 2.669201612472534, + "learning_rate": 4.307194518462124e-06, + "loss": 1.5494, + "step": 671 + }, + { + "epoch": 0.6902927580893683, + "grad_norm": 2.634634494781494, + "learning_rate": 4.3052912066996574e-06, + "loss": 1.6177, + "step": 672 + }, + { + "epoch": 0.6913199794555727, + "grad_norm": 2.5489139556884766, + "learning_rate": 4.303387894937191e-06, + "loss": 1.5876, + "step": 673 + }, + { + "epoch": 0.6923472008217771, + "grad_norm": 2.580073833465576, + "learning_rate": 4.301484583174724e-06, + "loss": 1.5106, + "step": 674 + }, + { + "epoch": 0.6933744221879815, + "grad_norm": 2.5126636028289795, + "learning_rate": 4.2995812714122575e-06, + "loss": 1.4988, + "step": 675 + }, + { + "epoch": 0.6944016435541859, + "grad_norm": 2.4630191326141357, + "learning_rate": 4.297677959649791e-06, + "loss": 1.5218, + "step": 676 + }, + { + "epoch": 0.6954288649203904, + "grad_norm": 2.8822433948516846, + "learning_rate": 4.295774647887324e-06, + "loss": 1.5592, + "step": 677 + }, + { + "epoch": 0.6964560862865947, + "grad_norm": 2.5498058795928955, + "learning_rate": 4.293871336124858e-06, + "loss": 1.5713, + "step": 678 + }, + { + "epoch": 0.6974833076527992, + "grad_norm": 2.6661157608032227, + "learning_rate": 4.291968024362391e-06, + "loss": 1.5431, + "step": 679 + }, + { + "epoch": 0.6985105290190036, + "grad_norm": 2.617971897125244, + "learning_rate": 4.290064712599924e-06, + "loss": 1.5792, + "step": 680 + }, + { + "epoch": 0.6985105290190036, + "eval_loss": 1.8702285289764404, + "eval_runtime": 37.0376, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 680 + }, + { + "epoch": 0.699537750385208, + "grad_norm": 2.5892348289489746, + "learning_rate": 4.288161400837458e-06, + "loss": 1.5626, + "step": 681 + }, + { + "epoch": 0.7005649717514124, + "grad_norm": 2.5460269451141357, + "learning_rate": 4.286258089074991e-06, + "loss": 1.4876, + "step": 682 + }, + { + "epoch": 0.7015921931176169, + "grad_norm": 2.601632595062256, + "learning_rate": 4.284354777312524e-06, + "loss": 1.5547, + "step": 683 + }, + { + "epoch": 0.7026194144838213, + "grad_norm": 2.6032373905181885, + "learning_rate": 4.282451465550058e-06, + "loss": 1.5133, + "step": 684 + }, + { + "epoch": 0.7036466358500256, + "grad_norm": 2.6283886432647705, + "learning_rate": 4.280548153787591e-06, + "loss": 1.5375, + "step": 685 + }, + { + "epoch": 0.7046738572162301, + "grad_norm": 2.5892159938812256, + "learning_rate": 4.2786448420251245e-06, + "loss": 1.5338, + "step": 686 + }, + { + "epoch": 0.7057010785824345, + "grad_norm": 2.518629789352417, + "learning_rate": 4.276741530262658e-06, + "loss": 1.5003, + "step": 687 + }, + { + "epoch": 0.706728299948639, + "grad_norm": 2.8705344200134277, + "learning_rate": 4.27483821850019e-06, + "loss": 1.5178, + "step": 688 + }, + { + "epoch": 0.7077555213148433, + "grad_norm": 2.7153639793395996, + "learning_rate": 4.272934906737724e-06, + "loss": 1.5914, + "step": 689 + }, + { + "epoch": 0.7087827426810478, + "grad_norm": 2.4585697650909424, + "learning_rate": 4.271031594975257e-06, + "loss": 1.5726, + "step": 690 + }, + { + "epoch": 0.7087827426810478, + "eval_loss": 1.8637797832489014, + "eval_runtime": 37.0465, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 690 + }, + { + "epoch": 0.7098099640472522, + "grad_norm": 2.6370503902435303, + "learning_rate": 4.2691282832127905e-06, + "loss": 1.5706, + "step": 691 + }, + { + "epoch": 0.7108371854134566, + "grad_norm": 2.70202898979187, + "learning_rate": 4.267224971450324e-06, + "loss": 1.5298, + "step": 692 + }, + { + "epoch": 0.711864406779661, + "grad_norm": 2.6149585247039795, + "learning_rate": 4.265321659687857e-06, + "loss": 1.477, + "step": 693 + }, + { + "epoch": 0.7128916281458655, + "grad_norm": 2.9561493396759033, + "learning_rate": 4.2634183479253906e-06, + "loss": 1.5189, + "step": 694 + }, + { + "epoch": 0.7139188495120699, + "grad_norm": 2.697680711746216, + "learning_rate": 4.261515036162924e-06, + "loss": 1.5001, + "step": 695 + }, + { + "epoch": 0.7149460708782742, + "grad_norm": 2.55727219581604, + "learning_rate": 4.259611724400457e-06, + "loss": 1.533, + "step": 696 + }, + { + "epoch": 0.7159732922444787, + "grad_norm": 2.854290723800659, + "learning_rate": 4.257708412637991e-06, + "loss": 1.5057, + "step": 697 + }, + { + "epoch": 0.7170005136106831, + "grad_norm": 2.73824405670166, + "learning_rate": 4.255805100875523e-06, + "loss": 1.4567, + "step": 698 + }, + { + "epoch": 0.7180277349768875, + "grad_norm": 2.730050802230835, + "learning_rate": 4.2539017891130565e-06, + "loss": 1.4873, + "step": 699 + }, + { + "epoch": 0.7190549563430919, + "grad_norm": 2.966660737991333, + "learning_rate": 4.25199847735059e-06, + "loss": 1.574, + "step": 700 + }, + { + "epoch": 0.7190549563430919, + "eval_loss": 1.8623589277267456, + "eval_runtime": 37.0761, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 700 + }, + { + "epoch": 0.7200821777092964, + "grad_norm": 2.7526276111602783, + "learning_rate": 4.250095165588123e-06, + "loss": 1.4831, + "step": 701 + }, + { + "epoch": 0.7211093990755008, + "grad_norm": 2.9722564220428467, + "learning_rate": 4.2481918538256575e-06, + "loss": 1.548, + "step": 702 + }, + { + "epoch": 0.7221366204417052, + "grad_norm": 2.679373025894165, + "learning_rate": 4.246288542063191e-06, + "loss": 1.5145, + "step": 703 + }, + { + "epoch": 0.7231638418079096, + "grad_norm": 2.733139753341675, + "learning_rate": 4.244385230300724e-06, + "loss": 1.5542, + "step": 704 + }, + { + "epoch": 0.724191063174114, + "grad_norm": 2.6549174785614014, + "learning_rate": 4.242481918538257e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.7252182845403184, + "grad_norm": 2.562958002090454, + "learning_rate": 4.24057860677579e-06, + "loss": 1.524, + "step": 706 + }, + { + "epoch": 0.7262455059065228, + "grad_norm": 2.641123056411743, + "learning_rate": 4.2386752950133234e-06, + "loss": 1.5723, + "step": 707 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 2.8128373622894287, + "learning_rate": 4.236771983250857e-06, + "loss": 1.5151, + "step": 708 + }, + { + "epoch": 0.7282999486389317, + "grad_norm": 2.676765203475952, + "learning_rate": 4.23486867148839e-06, + "loss": 1.5024, + "step": 709 + }, + { + "epoch": 0.7293271700051361, + "grad_norm": 2.917178153991699, + "learning_rate": 4.2329653597259235e-06, + "loss": 1.5326, + "step": 710 + }, + { + "epoch": 0.7293271700051361, + "eval_loss": 1.8676133155822754, + "eval_runtime": 36.8304, + "eval_samples_per_second": 27.152, + "eval_steps_per_second": 1.14, + "step": 710 + }, + { + "epoch": 0.7303543913713405, + "grad_norm": 2.8313796520233154, + "learning_rate": 4.231062047963457e-06, + "loss": 1.5623, + "step": 711 + }, + { + "epoch": 0.731381612737545, + "grad_norm": 2.61542010307312, + "learning_rate": 4.22915873620099e-06, + "loss": 1.5725, + "step": 712 + }, + { + "epoch": 0.7324088341037494, + "grad_norm": 2.5788609981536865, + "learning_rate": 4.227255424438524e-06, + "loss": 1.5381, + "step": 713 + }, + { + "epoch": 0.7334360554699538, + "grad_norm": 2.720306158065796, + "learning_rate": 4.225352112676057e-06, + "loss": 1.572, + "step": 714 + }, + { + "epoch": 0.7344632768361582, + "grad_norm": 2.780654191970825, + "learning_rate": 4.2234488009135895e-06, + "loss": 1.5084, + "step": 715 + }, + { + "epoch": 0.7354904982023626, + "grad_norm": 2.4448273181915283, + "learning_rate": 4.221545489151123e-06, + "loss": 1.4614, + "step": 716 + }, + { + "epoch": 0.736517719568567, + "grad_norm": 2.6579184532165527, + "learning_rate": 4.219642177388656e-06, + "loss": 1.5022, + "step": 717 + }, + { + "epoch": 0.7375449409347714, + "grad_norm": 2.7092721462249756, + "learning_rate": 4.2177388656261896e-06, + "loss": 1.5307, + "step": 718 + }, + { + "epoch": 0.7385721623009759, + "grad_norm": 2.65014910697937, + "learning_rate": 4.215835553863723e-06, + "loss": 1.4936, + "step": 719 + }, + { + "epoch": 0.7395993836671803, + "grad_norm": 2.7403628826141357, + "learning_rate": 4.213932242101256e-06, + "loss": 1.4458, + "step": 720 + }, + { + "epoch": 0.7395993836671803, + "eval_loss": 1.8778676986694336, + "eval_runtime": 36.9378, + "eval_samples_per_second": 27.073, + "eval_steps_per_second": 1.137, + "step": 720 + }, + { + "epoch": 0.7406266050333847, + "grad_norm": 2.3937759399414062, + "learning_rate": 4.21202893033879e-06, + "loss": 1.5086, + "step": 721 + }, + { + "epoch": 0.7416538263995891, + "grad_norm": 2.5786867141723633, + "learning_rate": 4.210125618576323e-06, + "loss": 1.5321, + "step": 722 + }, + { + "epoch": 0.7426810477657936, + "grad_norm": 2.607236862182617, + "learning_rate": 4.208222306813856e-06, + "loss": 1.5244, + "step": 723 + }, + { + "epoch": 0.743708269131998, + "grad_norm": 2.5461602210998535, + "learning_rate": 4.20631899505139e-06, + "loss": 1.5841, + "step": 724 + }, + { + "epoch": 0.7447354904982023, + "grad_norm": 2.5051612854003906, + "learning_rate": 4.204415683288923e-06, + "loss": 1.43, + "step": 725 + }, + { + "epoch": 0.7457627118644068, + "grad_norm": 2.692816734313965, + "learning_rate": 4.2025123715264565e-06, + "loss": 1.5439, + "step": 726 + }, + { + "epoch": 0.7467899332306112, + "grad_norm": 2.4827022552490234, + "learning_rate": 4.20060905976399e-06, + "loss": 1.4987, + "step": 727 + }, + { + "epoch": 0.7478171545968156, + "grad_norm": 2.7261135578155518, + "learning_rate": 4.198705748001523e-06, + "loss": 1.4573, + "step": 728 + }, + { + "epoch": 0.74884437596302, + "grad_norm": 2.5710837841033936, + "learning_rate": 4.1968024362390566e-06, + "loss": 1.564, + "step": 729 + }, + { + "epoch": 0.7498715973292245, + "grad_norm": 2.61181640625, + "learning_rate": 4.19489912447659e-06, + "loss": 1.5211, + "step": 730 + }, + { + "epoch": 0.7498715973292245, + "eval_loss": 1.8794434070587158, + "eval_runtime": 36.9676, + "eval_samples_per_second": 27.051, + "eval_steps_per_second": 1.136, + "step": 730 + }, + { + "epoch": 0.7508988186954288, + "grad_norm": 2.6946470737457275, + "learning_rate": 4.192995812714123e-06, + "loss": 1.5837, + "step": 731 + }, + { + "epoch": 0.7519260400616333, + "grad_norm": 2.5337390899658203, + "learning_rate": 4.191092500951656e-06, + "loss": 1.5251, + "step": 732 + }, + { + "epoch": 0.7529532614278377, + "grad_norm": 2.541532278060913, + "learning_rate": 4.189189189189189e-06, + "loss": 1.503, + "step": 733 + }, + { + "epoch": 0.7539804827940421, + "grad_norm": 2.639124631881714, + "learning_rate": 4.1872858774267225e-06, + "loss": 1.4963, + "step": 734 + }, + { + "epoch": 0.7550077041602465, + "grad_norm": 2.872861862182617, + "learning_rate": 4.185382565664256e-06, + "loss": 1.5632, + "step": 735 + }, + { + "epoch": 0.7560349255264509, + "grad_norm": 2.588980197906494, + "learning_rate": 4.183479253901789e-06, + "loss": 1.4998, + "step": 736 + }, + { + "epoch": 0.7570621468926554, + "grad_norm": 2.9199588298797607, + "learning_rate": 4.181575942139323e-06, + "loss": 1.466, + "step": 737 + }, + { + "epoch": 0.7580893682588598, + "grad_norm": 2.5677788257598877, + "learning_rate": 4.179672630376856e-06, + "loss": 1.5247, + "step": 738 + }, + { + "epoch": 0.7591165896250642, + "grad_norm": 2.481142520904541, + "learning_rate": 4.177769318614389e-06, + "loss": 1.4915, + "step": 739 + }, + { + "epoch": 0.7601438109912686, + "grad_norm": 2.7072012424468994, + "learning_rate": 4.175866006851923e-06, + "loss": 1.5783, + "step": 740 + }, + { + "epoch": 0.7601438109912686, + "eval_loss": 1.8817057609558105, + "eval_runtime": 36.9635, + "eval_samples_per_second": 27.054, + "eval_steps_per_second": 1.136, + "step": 740 + }, + { + "epoch": 0.7611710323574731, + "grad_norm": 2.5433032512664795, + "learning_rate": 4.173962695089456e-06, + "loss": 1.5457, + "step": 741 + }, + { + "epoch": 0.7621982537236774, + "grad_norm": 2.702077627182007, + "learning_rate": 4.1720593833269894e-06, + "loss": 1.5285, + "step": 742 + }, + { + "epoch": 0.7632254750898819, + "grad_norm": 2.6678309440612793, + "learning_rate": 4.170156071564523e-06, + "loss": 1.5381, + "step": 743 + }, + { + "epoch": 0.7642526964560863, + "grad_norm": 2.6097185611724854, + "learning_rate": 4.168252759802056e-06, + "loss": 1.6076, + "step": 744 + }, + { + "epoch": 0.7652799178222907, + "grad_norm": 2.5975704193115234, + "learning_rate": 4.1663494480395895e-06, + "loss": 1.515, + "step": 745 + }, + { + "epoch": 0.7663071391884951, + "grad_norm": 2.6858558654785156, + "learning_rate": 4.164446136277123e-06, + "loss": 1.5499, + "step": 746 + }, + { + "epoch": 0.7673343605546995, + "grad_norm": 2.565023899078369, + "learning_rate": 4.162542824514656e-06, + "loss": 1.5502, + "step": 747 + }, + { + "epoch": 0.768361581920904, + "grad_norm": 2.61715030670166, + "learning_rate": 4.16063951275219e-06, + "loss": 1.5271, + "step": 748 + }, + { + "epoch": 0.7693888032871083, + "grad_norm": 2.564382314682007, + "learning_rate": 4.158736200989722e-06, + "loss": 1.4756, + "step": 749 + }, + { + "epoch": 0.7704160246533128, + "grad_norm": 2.64457368850708, + "learning_rate": 4.1568328892272555e-06, + "loss": 1.5246, + "step": 750 + }, + { + "epoch": 0.7704160246533128, + "eval_loss": 1.8777766227722168, + "eval_runtime": 36.9737, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 750 + }, + { + "epoch": 0.7714432460195172, + "grad_norm": 2.773602247238159, + "learning_rate": 4.154929577464789e-06, + "loss": 1.5337, + "step": 751 + }, + { + "epoch": 0.7724704673857217, + "grad_norm": 2.5466997623443604, + "learning_rate": 4.153026265702322e-06, + "loss": 1.5269, + "step": 752 + }, + { + "epoch": 0.773497688751926, + "grad_norm": 2.702888250350952, + "learning_rate": 4.1511229539398556e-06, + "loss": 1.4636, + "step": 753 + }, + { + "epoch": 0.7745249101181304, + "grad_norm": 2.629251718521118, + "learning_rate": 4.149219642177389e-06, + "loss": 1.513, + "step": 754 + }, + { + "epoch": 0.7755521314843349, + "grad_norm": 2.52544903755188, + "learning_rate": 4.147316330414922e-06, + "loss": 1.5225, + "step": 755 + }, + { + "epoch": 0.7765793528505393, + "grad_norm": 2.4267778396606445, + "learning_rate": 4.145413018652456e-06, + "loss": 1.4377, + "step": 756 + }, + { + "epoch": 0.7776065742167437, + "grad_norm": 2.6061577796936035, + "learning_rate": 4.143509706889989e-06, + "loss": 1.5402, + "step": 757 + }, + { + "epoch": 0.7786337955829481, + "grad_norm": 2.6571085453033447, + "learning_rate": 4.141606395127522e-06, + "loss": 1.527, + "step": 758 + }, + { + "epoch": 0.7796610169491526, + "grad_norm": 2.48740291595459, + "learning_rate": 4.139703083365056e-06, + "loss": 1.517, + "step": 759 + }, + { + "epoch": 0.7806882383153569, + "grad_norm": 2.5044426918029785, + "learning_rate": 4.137799771602588e-06, + "loss": 1.5579, + "step": 760 + }, + { + "epoch": 0.7806882383153569, + "eval_loss": 1.8778951168060303, + "eval_runtime": 37.0045, + "eval_samples_per_second": 27.024, + "eval_steps_per_second": 1.135, + "step": 760 + }, + { + "epoch": 0.7817154596815614, + "grad_norm": 2.4728074073791504, + "learning_rate": 4.135896459840122e-06, + "loss": 1.5321, + "step": 761 + }, + { + "epoch": 0.7827426810477658, + "grad_norm": 2.6541659832000732, + "learning_rate": 4.133993148077656e-06, + "loss": 1.4866, + "step": 762 + }, + { + "epoch": 0.7837699024139703, + "grad_norm": 2.526716947555542, + "learning_rate": 4.132089836315189e-06, + "loss": 1.5035, + "step": 763 + }, + { + "epoch": 0.7847971237801746, + "grad_norm": 2.683306932449341, + "learning_rate": 4.1301865245527226e-06, + "loss": 1.5221, + "step": 764 + }, + { + "epoch": 0.785824345146379, + "grad_norm": 2.4980056285858154, + "learning_rate": 4.128283212790256e-06, + "loss": 1.4643, + "step": 765 + }, + { + "epoch": 0.7868515665125835, + "grad_norm": 2.3985488414764404, + "learning_rate": 4.1263799010277884e-06, + "loss": 1.4348, + "step": 766 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 2.600689649581909, + "learning_rate": 4.124476589265322e-06, + "loss": 1.5647, + "step": 767 + }, + { + "epoch": 0.7889060092449923, + "grad_norm": 2.5934395790100098, + "learning_rate": 4.122573277502855e-06, + "loss": 1.4887, + "step": 768 + }, + { + "epoch": 0.7899332306111967, + "grad_norm": 2.5209388732910156, + "learning_rate": 4.1206699657403885e-06, + "loss": 1.4893, + "step": 769 + }, + { + "epoch": 0.7909604519774012, + "grad_norm": 2.6656298637390137, + "learning_rate": 4.118766653977922e-06, + "loss": 1.5239, + "step": 770 + }, + { + "epoch": 0.7909604519774012, + "eval_loss": 1.8737632036209106, + "eval_runtime": 37.056, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 770 + }, + { + "epoch": 0.7919876733436055, + "grad_norm": 2.5147693157196045, + "learning_rate": 4.116863342215455e-06, + "loss": 1.5, + "step": 771 + }, + { + "epoch": 0.79301489470981, + "grad_norm": 2.4653773307800293, + "learning_rate": 4.114960030452989e-06, + "loss": 1.4802, + "step": 772 + }, + { + "epoch": 0.7940421160760144, + "grad_norm": 2.643160820007324, + "learning_rate": 4.113056718690522e-06, + "loss": 1.5086, + "step": 773 + }, + { + "epoch": 0.7950693374422187, + "grad_norm": 2.4873125553131104, + "learning_rate": 4.111153406928055e-06, + "loss": 1.4558, + "step": 774 + }, + { + "epoch": 0.7960965588084232, + "grad_norm": 2.541308879852295, + "learning_rate": 4.109250095165589e-06, + "loss": 1.5483, + "step": 775 + }, + { + "epoch": 0.7971237801746276, + "grad_norm": 2.701874256134033, + "learning_rate": 4.107346783403122e-06, + "loss": 1.5551, + "step": 776 + }, + { + "epoch": 0.7981510015408321, + "grad_norm": 2.457371234893799, + "learning_rate": 4.105443471640655e-06, + "loss": 1.4968, + "step": 777 + }, + { + "epoch": 0.7991782229070364, + "grad_norm": 2.5294973850250244, + "learning_rate": 4.103540159878188e-06, + "loss": 1.4756, + "step": 778 + }, + { + "epoch": 0.8002054442732409, + "grad_norm": 2.464932918548584, + "learning_rate": 4.101636848115721e-06, + "loss": 1.5215, + "step": 779 + }, + { + "epoch": 0.8012326656394453, + "grad_norm": 2.5919454097747803, + "learning_rate": 4.099733536353255e-06, + "loss": 1.4807, + "step": 780 + }, + { + "epoch": 0.8012326656394453, + "eval_loss": 1.875250220298767, + "eval_runtime": 37.035, + "eval_samples_per_second": 27.001, + "eval_steps_per_second": 1.134, + "step": 780 + }, + { + "epoch": 0.8022598870056498, + "grad_norm": 2.505749464035034, + "learning_rate": 4.097830224590788e-06, + "loss": 1.5224, + "step": 781 + }, + { + "epoch": 0.8032871083718541, + "grad_norm": 2.747262954711914, + "learning_rate": 4.095926912828322e-06, + "loss": 1.6051, + "step": 782 + }, + { + "epoch": 0.8043143297380585, + "grad_norm": 2.6147146224975586, + "learning_rate": 4.094023601065855e-06, + "loss": 1.565, + "step": 783 + }, + { + "epoch": 0.805341551104263, + "grad_norm": 2.5381031036376953, + "learning_rate": 4.092120289303388e-06, + "loss": 1.511, + "step": 784 + }, + { + "epoch": 0.8063687724704673, + "grad_norm": 2.6476190090179443, + "learning_rate": 4.0902169775409215e-06, + "loss": 1.5147, + "step": 785 + }, + { + "epoch": 0.8073959938366718, + "grad_norm": 2.486919641494751, + "learning_rate": 4.088313665778455e-06, + "loss": 1.4697, + "step": 786 + }, + { + "epoch": 0.8084232152028762, + "grad_norm": 2.7006654739379883, + "learning_rate": 4.086410354015988e-06, + "loss": 1.4965, + "step": 787 + }, + { + "epoch": 0.8094504365690807, + "grad_norm": 2.7587921619415283, + "learning_rate": 4.0845070422535216e-06, + "loss": 1.4781, + "step": 788 + }, + { + "epoch": 0.810477657935285, + "grad_norm": 2.64239501953125, + "learning_rate": 4.082603730491055e-06, + "loss": 1.4945, + "step": 789 + }, + { + "epoch": 0.8115048793014895, + "grad_norm": 2.659919261932373, + "learning_rate": 4.080700418728588e-06, + "loss": 1.5378, + "step": 790 + }, + { + "epoch": 0.8115048793014895, + "eval_loss": 1.881503939628601, + "eval_runtime": 37.2277, + "eval_samples_per_second": 26.862, + "eval_steps_per_second": 1.128, + "step": 790 + }, + { + "epoch": 0.8125321006676939, + "grad_norm": 2.820996046066284, + "learning_rate": 4.078797106966122e-06, + "loss": 1.5134, + "step": 791 + }, + { + "epoch": 0.8135593220338984, + "grad_norm": 2.4667489528656006, + "learning_rate": 4.076893795203655e-06, + "loss": 1.4165, + "step": 792 + }, + { + "epoch": 0.8145865434001027, + "grad_norm": 2.5528056621551514, + "learning_rate": 4.074990483441188e-06, + "loss": 1.481, + "step": 793 + }, + { + "epoch": 0.8156137647663071, + "grad_norm": 2.632922887802124, + "learning_rate": 4.073087171678721e-06, + "loss": 1.5502, + "step": 794 + }, + { + "epoch": 0.8166409861325116, + "grad_norm": 2.662536144256592, + "learning_rate": 4.071183859916254e-06, + "loss": 1.5302, + "step": 795 + }, + { + "epoch": 0.8176682074987159, + "grad_norm": 2.662454605102539, + "learning_rate": 4.069280548153788e-06, + "loss": 1.5131, + "step": 796 + }, + { + "epoch": 0.8186954288649204, + "grad_norm": 2.6387386322021484, + "learning_rate": 4.067377236391321e-06, + "loss": 1.4677, + "step": 797 + }, + { + "epoch": 0.8197226502311248, + "grad_norm": 2.8879177570343018, + "learning_rate": 4.065473924628854e-06, + "loss": 1.5106, + "step": 798 + }, + { + "epoch": 0.8207498715973293, + "grad_norm": 2.680610418319702, + "learning_rate": 4.063570612866388e-06, + "loss": 1.5389, + "step": 799 + }, + { + "epoch": 0.8217770929635336, + "grad_norm": 2.659590244293213, + "learning_rate": 4.061667301103921e-06, + "loss": 1.5254, + "step": 800 + }, + { + "epoch": 0.8217770929635336, + "eval_loss": 1.881851077079773, + "eval_runtime": 37.3385, + "eval_samples_per_second": 26.782, + "eval_steps_per_second": 1.125, + "step": 800 + }, + { + "epoch": 0.8228043143297381, + "grad_norm": 2.711195945739746, + "learning_rate": 4.0597639893414544e-06, + "loss": 1.5026, + "step": 801 + }, + { + "epoch": 0.8238315356959425, + "grad_norm": 2.6387939453125, + "learning_rate": 4.057860677578988e-06, + "loss": 1.5101, + "step": 802 + }, + { + "epoch": 0.8248587570621468, + "grad_norm": 2.507387638092041, + "learning_rate": 4.055957365816521e-06, + "loss": 1.5277, + "step": 803 + }, + { + "epoch": 0.8258859784283513, + "grad_norm": 2.610370397567749, + "learning_rate": 4.0540540540540545e-06, + "loss": 1.4695, + "step": 804 + }, + { + "epoch": 0.8269131997945557, + "grad_norm": 2.8345787525177, + "learning_rate": 4.052150742291588e-06, + "loss": 1.467, + "step": 805 + }, + { + "epoch": 0.8279404211607602, + "grad_norm": 2.5142805576324463, + "learning_rate": 4.050247430529121e-06, + "loss": 1.4637, + "step": 806 + }, + { + "epoch": 0.8289676425269645, + "grad_norm": 2.7722535133361816, + "learning_rate": 4.048344118766655e-06, + "loss": 1.5427, + "step": 807 + }, + { + "epoch": 0.829994863893169, + "grad_norm": 2.816789150238037, + "learning_rate": 4.046440807004188e-06, + "loss": 1.5422, + "step": 808 + }, + { + "epoch": 0.8310220852593734, + "grad_norm": 2.6485376358032227, + "learning_rate": 4.044537495241721e-06, + "loss": 1.5078, + "step": 809 + }, + { + "epoch": 0.8320493066255779, + "grad_norm": 2.4760921001434326, + "learning_rate": 4.042634183479255e-06, + "loss": 1.4827, + "step": 810 + }, + { + "epoch": 0.8320493066255779, + "eval_loss": 1.8709155321121216, + "eval_runtime": 36.9015, + "eval_samples_per_second": 27.099, + "eval_steps_per_second": 1.138, + "step": 810 + }, + { + "epoch": 0.8330765279917822, + "grad_norm": 2.658062219619751, + "learning_rate": 4.040730871716787e-06, + "loss": 1.5434, + "step": 811 + }, + { + "epoch": 0.8341037493579867, + "grad_norm": 2.5443618297576904, + "learning_rate": 4.038827559954321e-06, + "loss": 1.5165, + "step": 812 + }, + { + "epoch": 0.8351309707241911, + "grad_norm": 2.698432207107544, + "learning_rate": 4.036924248191854e-06, + "loss": 1.5553, + "step": 813 + }, + { + "epoch": 0.8361581920903954, + "grad_norm": 2.5598180294036865, + "learning_rate": 4.035020936429387e-06, + "loss": 1.5289, + "step": 814 + }, + { + "epoch": 0.8371854134565999, + "grad_norm": 2.7002663612365723, + "learning_rate": 4.033117624666921e-06, + "loss": 1.509, + "step": 815 + }, + { + "epoch": 0.8382126348228043, + "grad_norm": 2.7882776260375977, + "learning_rate": 4.031214312904454e-06, + "loss": 1.4787, + "step": 816 + }, + { + "epoch": 0.8392398561890088, + "grad_norm": 2.5583086013793945, + "learning_rate": 4.029311001141987e-06, + "loss": 1.4642, + "step": 817 + }, + { + "epoch": 0.8402670775552131, + "grad_norm": 2.584573268890381, + "learning_rate": 4.027407689379521e-06, + "loss": 1.4343, + "step": 818 + }, + { + "epoch": 0.8412942989214176, + "grad_norm": 2.5062575340270996, + "learning_rate": 4.025504377617054e-06, + "loss": 1.4975, + "step": 819 + }, + { + "epoch": 0.842321520287622, + "grad_norm": 2.3546457290649414, + "learning_rate": 4.0236010658545875e-06, + "loss": 1.4795, + "step": 820 + }, + { + "epoch": 0.842321520287622, + "eval_loss": 1.8716342449188232, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 0.8433487416538265, + "grad_norm": 2.513228178024292, + "learning_rate": 4.02169775409212e-06, + "loss": 1.5191, + "step": 821 + }, + { + "epoch": 0.8443759630200308, + "grad_norm": 2.6360220909118652, + "learning_rate": 4.019794442329653e-06, + "loss": 1.5739, + "step": 822 + }, + { + "epoch": 0.8454031843862352, + "grad_norm": 2.5351850986480713, + "learning_rate": 4.0178911305671876e-06, + "loss": 1.4487, + "step": 823 + }, + { + "epoch": 0.8464304057524397, + "grad_norm": 2.6693851947784424, + "learning_rate": 4.015987818804721e-06, + "loss": 1.5319, + "step": 824 + }, + { + "epoch": 0.847457627118644, + "grad_norm": 2.5210046768188477, + "learning_rate": 4.014084507042254e-06, + "loss": 1.4723, + "step": 825 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.5799994468688965, + "learning_rate": 4.012181195279788e-06, + "loss": 1.4409, + "step": 826 + }, + { + "epoch": 0.8495120698510529, + "grad_norm": 2.5206313133239746, + "learning_rate": 4.010277883517321e-06, + "loss": 1.5449, + "step": 827 + }, + { + "epoch": 0.8505392912172574, + "grad_norm": 2.480584144592285, + "learning_rate": 4.0083745717548535e-06, + "loss": 1.5306, + "step": 828 + }, + { + "epoch": 0.8515665125834617, + "grad_norm": 2.5229053497314453, + "learning_rate": 4.006471259992387e-06, + "loss": 1.5001, + "step": 829 + }, + { + "epoch": 0.8525937339496662, + "grad_norm": 2.53357195854187, + "learning_rate": 4.00456794822992e-06, + "loss": 1.5188, + "step": 830 + }, + { + "epoch": 0.8525937339496662, + "eval_loss": 1.8724256753921509, + "eval_runtime": 37.1114, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 830 + }, + { + "epoch": 0.8536209553158706, + "grad_norm": 2.6796159744262695, + "learning_rate": 4.002664636467454e-06, + "loss": 1.5577, + "step": 831 + }, + { + "epoch": 0.8546481766820749, + "grad_norm": 2.4717817306518555, + "learning_rate": 4.000761324704987e-06, + "loss": 1.5124, + "step": 832 + }, + { + "epoch": 0.8556753980482794, + "grad_norm": 2.649057626724243, + "learning_rate": 3.99885801294252e-06, + "loss": 1.4516, + "step": 833 + }, + { + "epoch": 0.8567026194144838, + "grad_norm": 2.592406749725342, + "learning_rate": 3.996954701180054e-06, + "loss": 1.5215, + "step": 834 + }, + { + "epoch": 0.8577298407806883, + "grad_norm": 2.46956205368042, + "learning_rate": 3.995051389417587e-06, + "loss": 1.4324, + "step": 835 + }, + { + "epoch": 0.8587570621468926, + "grad_norm": 2.8484537601470947, + "learning_rate": 3.9931480776551204e-06, + "loss": 1.4574, + "step": 836 + }, + { + "epoch": 0.8597842835130971, + "grad_norm": 2.6475489139556885, + "learning_rate": 3.991244765892654e-06, + "loss": 1.5581, + "step": 837 + }, + { + "epoch": 0.8608115048793015, + "grad_norm": 2.527153491973877, + "learning_rate": 3.989341454130186e-06, + "loss": 1.4713, + "step": 838 + }, + { + "epoch": 0.861838726245506, + "grad_norm": 2.6767020225524902, + "learning_rate": 3.98743814236772e-06, + "loss": 1.4459, + "step": 839 + }, + { + "epoch": 0.8628659476117103, + "grad_norm": 2.6406850814819336, + "learning_rate": 3.985534830605253e-06, + "loss": 1.4964, + "step": 840 + }, + { + "epoch": 0.8628659476117103, + "eval_loss": 1.886762261390686, + "eval_runtime": 37.1378, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 840 + }, + { + "epoch": 0.8638931689779148, + "grad_norm": 2.466372489929199, + "learning_rate": 3.983631518842786e-06, + "loss": 1.5136, + "step": 841 + }, + { + "epoch": 0.8649203903441192, + "grad_norm": 2.830249071121216, + "learning_rate": 3.98172820708032e-06, + "loss": 1.5372, + "step": 842 + }, + { + "epoch": 0.8659476117103235, + "grad_norm": 2.675889015197754, + "learning_rate": 3.979824895317854e-06, + "loss": 1.4881, + "step": 843 + }, + { + "epoch": 0.866974833076528, + "grad_norm": 2.7418212890625, + "learning_rate": 3.977921583555387e-06, + "loss": 1.4541, + "step": 844 + }, + { + "epoch": 0.8680020544427324, + "grad_norm": 2.572862386703491, + "learning_rate": 3.97601827179292e-06, + "loss": 1.5191, + "step": 845 + }, + { + "epoch": 0.8690292758089369, + "grad_norm": 2.6524274349212646, + "learning_rate": 3.974114960030453e-06, + "loss": 1.4647, + "step": 846 + }, + { + "epoch": 0.8700564971751412, + "grad_norm": 2.794415235519409, + "learning_rate": 3.9722116482679866e-06, + "loss": 1.504, + "step": 847 + }, + { + "epoch": 0.8710837185413457, + "grad_norm": 2.5947723388671875, + "learning_rate": 3.97030833650552e-06, + "loss": 1.492, + "step": 848 + }, + { + "epoch": 0.8721109399075501, + "grad_norm": 2.8539984226226807, + "learning_rate": 3.968405024743053e-06, + "loss": 1.5049, + "step": 849 + }, + { + "epoch": 0.8731381612737545, + "grad_norm": 2.522786855697632, + "learning_rate": 3.966501712980587e-06, + "loss": 1.4223, + "step": 850 + }, + { + "epoch": 0.8731381612737545, + "eval_loss": 1.8691684007644653, + "eval_runtime": 37.1287, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 850 + }, + { + "epoch": 0.8741653826399589, + "grad_norm": 2.6483359336853027, + "learning_rate": 3.96459840121812e-06, + "loss": 1.4815, + "step": 851 + }, + { + "epoch": 0.8751926040061633, + "grad_norm": 2.581429958343506, + "learning_rate": 3.962695089455653e-06, + "loss": 1.4837, + "step": 852 + }, + { + "epoch": 0.8762198253723678, + "grad_norm": 2.5967583656311035, + "learning_rate": 3.960791777693187e-06, + "loss": 1.5289, + "step": 853 + }, + { + "epoch": 0.8772470467385721, + "grad_norm": 2.7169225215911865, + "learning_rate": 3.95888846593072e-06, + "loss": 1.5173, + "step": 854 + }, + { + "epoch": 0.8782742681047766, + "grad_norm": 2.644656181335449, + "learning_rate": 3.956985154168253e-06, + "loss": 1.5095, + "step": 855 + }, + { + "epoch": 0.879301489470981, + "grad_norm": 2.6397511959075928, + "learning_rate": 3.955081842405786e-06, + "loss": 1.5422, + "step": 856 + }, + { + "epoch": 0.8803287108371854, + "grad_norm": 2.691540479660034, + "learning_rate": 3.953178530643319e-06, + "loss": 1.5703, + "step": 857 + }, + { + "epoch": 0.8813559322033898, + "grad_norm": 2.50526762008667, + "learning_rate": 3.951275218880853e-06, + "loss": 1.5885, + "step": 858 + }, + { + "epoch": 0.8823831535695943, + "grad_norm": 2.3846609592437744, + "learning_rate": 3.949371907118386e-06, + "loss": 1.4248, + "step": 859 + }, + { + "epoch": 0.8834103749357987, + "grad_norm": 2.7107760906219482, + "learning_rate": 3.9474685953559194e-06, + "loss": 1.5358, + "step": 860 + }, + { + "epoch": 0.8834103749357987, + "eval_loss": 1.876391887664795, + "eval_runtime": 37.1284, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.884437596302003, + "grad_norm": 2.6206252574920654, + "learning_rate": 3.945565283593453e-06, + "loss": 1.4907, + "step": 861 + }, + { + "epoch": 0.8854648176682075, + "grad_norm": 2.6641664505004883, + "learning_rate": 3.943661971830986e-06, + "loss": 1.4888, + "step": 862 + }, + { + "epoch": 0.8864920390344119, + "grad_norm": 2.7342002391815186, + "learning_rate": 3.9417586600685195e-06, + "loss": 1.5421, + "step": 863 + }, + { + "epoch": 0.8875192604006163, + "grad_norm": 2.7120890617370605, + "learning_rate": 3.939855348306053e-06, + "loss": 1.5154, + "step": 864 + }, + { + "epoch": 0.8885464817668207, + "grad_norm": 2.6350717544555664, + "learning_rate": 3.937952036543586e-06, + "loss": 1.5135, + "step": 865 + }, + { + "epoch": 0.8895737031330252, + "grad_norm": 2.570857524871826, + "learning_rate": 3.93604872478112e-06, + "loss": 1.5072, + "step": 866 + }, + { + "epoch": 0.8906009244992296, + "grad_norm": 2.855071544647217, + "learning_rate": 3.934145413018653e-06, + "loss": 1.5171, + "step": 867 + }, + { + "epoch": 0.891628145865434, + "grad_norm": 2.5592195987701416, + "learning_rate": 3.932242101256186e-06, + "loss": 1.4695, + "step": 868 + }, + { + "epoch": 0.8926553672316384, + "grad_norm": 2.6110830307006836, + "learning_rate": 3.93033878949372e-06, + "loss": 1.4802, + "step": 869 + }, + { + "epoch": 0.8936825885978429, + "grad_norm": 2.7121548652648926, + "learning_rate": 3.928435477731253e-06, + "loss": 1.4422, + "step": 870 + }, + { + "epoch": 0.8936825885978429, + "eval_loss": 1.8629941940307617, + "eval_runtime": 37.1091, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.8947098099640473, + "grad_norm": 2.7763099670410156, + "learning_rate": 3.9265321659687864e-06, + "loss": 1.5275, + "step": 871 + }, + { + "epoch": 0.8957370313302516, + "grad_norm": 2.596220016479492, + "learning_rate": 3.924628854206319e-06, + "loss": 1.5374, + "step": 872 + }, + { + "epoch": 0.8967642526964561, + "grad_norm": 2.575990915298462, + "learning_rate": 3.922725542443852e-06, + "loss": 1.5071, + "step": 873 + }, + { + "epoch": 0.8977914740626605, + "grad_norm": 2.7052223682403564, + "learning_rate": 3.920822230681386e-06, + "loss": 1.4656, + "step": 874 + }, + { + "epoch": 0.8988186954288649, + "grad_norm": 2.550267219543457, + "learning_rate": 3.918918918918919e-06, + "loss": 1.4837, + "step": 875 + }, + { + "epoch": 0.8998459167950693, + "grad_norm": 2.6948060989379883, + "learning_rate": 3.917015607156452e-06, + "loss": 1.5156, + "step": 876 + }, + { + "epoch": 0.9008731381612738, + "grad_norm": 2.575596809387207, + "learning_rate": 3.915112295393986e-06, + "loss": 1.4881, + "step": 877 + }, + { + "epoch": 0.9019003595274782, + "grad_norm": 2.9835500717163086, + "learning_rate": 3.913208983631519e-06, + "loss": 1.5729, + "step": 878 + }, + { + "epoch": 0.9029275808936826, + "grad_norm": 2.5961520671844482, + "learning_rate": 3.9113056718690525e-06, + "loss": 1.5469, + "step": 879 + }, + { + "epoch": 0.903954802259887, + "grad_norm": 2.518260955810547, + "learning_rate": 3.909402360106586e-06, + "loss": 1.4704, + "step": 880 + }, + { + "epoch": 0.903954802259887, + "eval_loss": 1.8570330142974854, + "eval_runtime": 37.1293, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 880 + }, + { + "epoch": 0.9049820236260914, + "grad_norm": 2.740879535675049, + "learning_rate": 3.907499048344119e-06, + "loss": 1.5092, + "step": 881 + }, + { + "epoch": 0.9060092449922958, + "grad_norm": 2.5227973461151123, + "learning_rate": 3.905595736581652e-06, + "loss": 1.5252, + "step": 882 + }, + { + "epoch": 0.9070364663585002, + "grad_norm": 2.483125925064087, + "learning_rate": 3.903692424819186e-06, + "loss": 1.5216, + "step": 883 + }, + { + "epoch": 0.9080636877247047, + "grad_norm": 2.4979538917541504, + "learning_rate": 3.901789113056719e-06, + "loss": 1.4734, + "step": 884 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.707669496536255, + "learning_rate": 3.899885801294253e-06, + "loss": 1.5347, + "step": 885 + }, + { + "epoch": 0.9101181304571135, + "grad_norm": 2.6548409461975098, + "learning_rate": 3.897982489531786e-06, + "loss": 1.5079, + "step": 886 + }, + { + "epoch": 0.9111453518233179, + "grad_norm": 2.6678848266601562, + "learning_rate": 3.896079177769319e-06, + "loss": 1.5659, + "step": 887 + }, + { + "epoch": 0.9121725731895224, + "grad_norm": 2.570976734161377, + "learning_rate": 3.894175866006853e-06, + "loss": 1.5281, + "step": 888 + }, + { + "epoch": 0.9131997945557268, + "grad_norm": 2.5939226150512695, + "learning_rate": 3.892272554244385e-06, + "loss": 1.56, + "step": 889 + }, + { + "epoch": 0.9142270159219312, + "grad_norm": 2.4577109813690186, + "learning_rate": 3.890369242481919e-06, + "loss": 1.4702, + "step": 890 + }, + { + "epoch": 0.9142270159219312, + "eval_loss": 1.8534926176071167, + "eval_runtime": 37.3353, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 890 + }, + { + "epoch": 0.9152542372881356, + "grad_norm": 2.5012643337249756, + "learning_rate": 3.888465930719452e-06, + "loss": 1.5411, + "step": 891 + }, + { + "epoch": 0.91628145865434, + "grad_norm": 2.470905065536499, + "learning_rate": 3.886562618956985e-06, + "loss": 1.4766, + "step": 892 + }, + { + "epoch": 0.9173086800205444, + "grad_norm": 2.4610393047332764, + "learning_rate": 3.884659307194519e-06, + "loss": 1.4925, + "step": 893 + }, + { + "epoch": 0.9183359013867488, + "grad_norm": 2.573151111602783, + "learning_rate": 3.882755995432052e-06, + "loss": 1.4389, + "step": 894 + }, + { + "epoch": 0.9193631227529533, + "grad_norm": 2.628934144973755, + "learning_rate": 3.8808526836695854e-06, + "loss": 1.5099, + "step": 895 + }, + { + "epoch": 0.9203903441191577, + "grad_norm": 2.639920949935913, + "learning_rate": 3.878949371907119e-06, + "loss": 1.4724, + "step": 896 + }, + { + "epoch": 0.9214175654853621, + "grad_norm": 2.4775230884552, + "learning_rate": 3.877046060144652e-06, + "loss": 1.4657, + "step": 897 + }, + { + "epoch": 0.9224447868515665, + "grad_norm": 2.5242490768432617, + "learning_rate": 3.8751427483821855e-06, + "loss": 1.459, + "step": 898 + }, + { + "epoch": 0.923472008217771, + "grad_norm": 2.4444870948791504, + "learning_rate": 3.873239436619718e-06, + "loss": 1.494, + "step": 899 + }, + { + "epoch": 0.9244992295839753, + "grad_norm": 2.4326930046081543, + "learning_rate": 3.871336124857251e-06, + "loss": 1.4449, + "step": 900 + }, + { + "epoch": 0.9244992295839753, + "eval_loss": 1.8844366073608398, + "eval_runtime": 37.432, + "eval_samples_per_second": 26.715, + "eval_steps_per_second": 1.122, + "step": 900 + }, + { + "epoch": 0.9255264509501797, + "grad_norm": 2.621652603149414, + "learning_rate": 3.869432813094785e-06, + "loss": 1.5055, + "step": 901 + }, + { + "epoch": 0.9265536723163842, + "grad_norm": 2.5909814834594727, + "learning_rate": 3.867529501332318e-06, + "loss": 1.5176, + "step": 902 + }, + { + "epoch": 0.9275808936825886, + "grad_norm": 2.512760877609253, + "learning_rate": 3.8656261895698515e-06, + "loss": 1.5027, + "step": 903 + }, + { + "epoch": 0.928608115048793, + "grad_norm": 2.6954784393310547, + "learning_rate": 3.863722877807386e-06, + "loss": 1.4912, + "step": 904 + }, + { + "epoch": 0.9296353364149974, + "grad_norm": 2.481905221939087, + "learning_rate": 3.861819566044919e-06, + "loss": 1.529, + "step": 905 + }, + { + "epoch": 0.9306625577812019, + "grad_norm": 2.612758159637451, + "learning_rate": 3.859916254282452e-06, + "loss": 1.5658, + "step": 906 + }, + { + "epoch": 0.9316897791474062, + "grad_norm": 2.523171901702881, + "learning_rate": 3.858012942519985e-06, + "loss": 1.4804, + "step": 907 + }, + { + "epoch": 0.9327170005136107, + "grad_norm": 2.5053422451019287, + "learning_rate": 3.856109630757518e-06, + "loss": 1.5007, + "step": 908 + }, + { + "epoch": 0.9337442218798151, + "grad_norm": 2.6110904216766357, + "learning_rate": 3.854206318995052e-06, + "loss": 1.4758, + "step": 909 + }, + { + "epoch": 0.9347714432460195, + "grad_norm": 2.7004992961883545, + "learning_rate": 3.852303007232585e-06, + "loss": 1.5255, + "step": 910 + }, + { + "epoch": 0.9347714432460195, + "eval_loss": 1.892087697982788, + "eval_runtime": 36.9356, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 910 + }, + { + "epoch": 0.9357986646122239, + "grad_norm": 2.617180109024048, + "learning_rate": 3.850399695470118e-06, + "loss": 1.4563, + "step": 911 + }, + { + "epoch": 0.9368258859784283, + "grad_norm": 2.5294559001922607, + "learning_rate": 3.848496383707652e-06, + "loss": 1.4767, + "step": 912 + }, + { + "epoch": 0.9378531073446328, + "grad_norm": 2.594238042831421, + "learning_rate": 3.846593071945185e-06, + "loss": 1.4617, + "step": 913 + }, + { + "epoch": 0.9388803287108372, + "grad_norm": 2.6076996326446533, + "learning_rate": 3.8446897601827185e-06, + "loss": 1.446, + "step": 914 + }, + { + "epoch": 0.9399075500770416, + "grad_norm": 2.6047134399414062, + "learning_rate": 3.842786448420252e-06, + "loss": 1.5698, + "step": 915 + }, + { + "epoch": 0.940934771443246, + "grad_norm": 2.5670995712280273, + "learning_rate": 3.840883136657785e-06, + "loss": 1.5803, + "step": 916 + }, + { + "epoch": 0.9419619928094505, + "grad_norm": 2.6153712272644043, + "learning_rate": 3.838979824895318e-06, + "loss": 1.5565, + "step": 917 + }, + { + "epoch": 0.9429892141756548, + "grad_norm": 2.6707098484039307, + "learning_rate": 3.837076513132851e-06, + "loss": 1.4417, + "step": 918 + }, + { + "epoch": 0.9440164355418593, + "grad_norm": 2.591336488723755, + "learning_rate": 3.8351732013703844e-06, + "loss": 1.5159, + "step": 919 + }, + { + "epoch": 0.9450436569080637, + "grad_norm": 2.4689993858337402, + "learning_rate": 3.833269889607918e-06, + "loss": 1.4735, + "step": 920 + }, + { + "epoch": 0.9450436569080637, + "eval_loss": 1.875109076499939, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 920 + }, + { + "epoch": 0.9460708782742681, + "grad_norm": 2.739758253097534, + "learning_rate": 3.831366577845451e-06, + "loss": 1.4566, + "step": 921 + }, + { + "epoch": 0.9470980996404725, + "grad_norm": 2.65278697013855, + "learning_rate": 3.8294632660829845e-06, + "loss": 1.4545, + "step": 922 + }, + { + "epoch": 0.9481253210066769, + "grad_norm": 2.678321123123169, + "learning_rate": 3.827559954320518e-06, + "loss": 1.4864, + "step": 923 + }, + { + "epoch": 0.9491525423728814, + "grad_norm": 2.7516839504241943, + "learning_rate": 3.825656642558051e-06, + "loss": 1.5031, + "step": 924 + }, + { + "epoch": 0.9501797637390857, + "grad_norm": 2.6125247478485107, + "learning_rate": 3.823753330795585e-06, + "loss": 1.521, + "step": 925 + }, + { + "epoch": 0.9512069851052902, + "grad_norm": 2.5134871006011963, + "learning_rate": 3.821850019033118e-06, + "loss": 1.5019, + "step": 926 + }, + { + "epoch": 0.9522342064714946, + "grad_norm": 2.5274763107299805, + "learning_rate": 3.819946707270651e-06, + "loss": 1.4358, + "step": 927 + }, + { + "epoch": 0.9532614278376991, + "grad_norm": 2.480628728866577, + "learning_rate": 3.818043395508185e-06, + "loss": 1.483, + "step": 928 + }, + { + "epoch": 0.9542886492039034, + "grad_norm": 2.6450936794281006, + "learning_rate": 3.816140083745718e-06, + "loss": 1.4892, + "step": 929 + }, + { + "epoch": 0.9553158705701078, + "grad_norm": 2.6758296489715576, + "learning_rate": 3.8142367719832514e-06, + "loss": 1.5701, + "step": 930 + }, + { + "epoch": 0.9553158705701078, + "eval_loss": 1.8651176691055298, + "eval_runtime": 37.0553, + "eval_samples_per_second": 26.987, + "eval_steps_per_second": 1.133, + "step": 930 + }, + { + "epoch": 0.9563430919363123, + "grad_norm": 2.6887295246124268, + "learning_rate": 3.8123334602207844e-06, + "loss": 1.4596, + "step": 931 + }, + { + "epoch": 0.9573703133025167, + "grad_norm": 2.577507734298706, + "learning_rate": 3.8104301484583177e-06, + "loss": 1.4664, + "step": 932 + }, + { + "epoch": 0.9583975346687211, + "grad_norm": 2.6431517601013184, + "learning_rate": 3.808526836695851e-06, + "loss": 1.4715, + "step": 933 + }, + { + "epoch": 0.9594247560349255, + "grad_norm": 2.8451743125915527, + "learning_rate": 3.8066235249333845e-06, + "loss": 1.5096, + "step": 934 + }, + { + "epoch": 0.96045197740113, + "grad_norm": 2.5525193214416504, + "learning_rate": 3.804720213170918e-06, + "loss": 1.4975, + "step": 935 + }, + { + "epoch": 0.9614791987673343, + "grad_norm": 2.6353704929351807, + "learning_rate": 3.8028169014084508e-06, + "loss": 1.4979, + "step": 936 + }, + { + "epoch": 0.9625064201335388, + "grad_norm": 2.923781156539917, + "learning_rate": 3.800913589645984e-06, + "loss": 1.5118, + "step": 937 + }, + { + "epoch": 0.9635336414997432, + "grad_norm": 2.7363650798797607, + "learning_rate": 3.7990102778835175e-06, + "loss": 1.5016, + "step": 938 + }, + { + "epoch": 0.9645608628659477, + "grad_norm": 2.7126598358154297, + "learning_rate": 3.797106966121051e-06, + "loss": 1.5022, + "step": 939 + }, + { + "epoch": 0.965588084232152, + "grad_norm": 3.0102460384368896, + "learning_rate": 3.7952036543585842e-06, + "loss": 1.5638, + "step": 940 + }, + { + "epoch": 0.965588084232152, + "eval_loss": 1.8475635051727295, + "eval_runtime": 37.0871, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 940 + }, + { + "epoch": 0.9666153055983564, + "grad_norm": 2.5455925464630127, + "learning_rate": 3.793300342596117e-06, + "loss": 1.5218, + "step": 941 + }, + { + "epoch": 0.9676425269645609, + "grad_norm": 2.5489683151245117, + "learning_rate": 3.7913970308336505e-06, + "loss": 1.5375, + "step": 942 + }, + { + "epoch": 0.9686697483307652, + "grad_norm": 2.672626256942749, + "learning_rate": 3.789493719071184e-06, + "loss": 1.5217, + "step": 943 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.498378038406372, + "learning_rate": 3.7875904073087177e-06, + "loss": 1.475, + "step": 944 + }, + { + "epoch": 0.9707241910631741, + "grad_norm": 2.515190362930298, + "learning_rate": 3.785687095546251e-06, + "loss": 1.4341, + "step": 945 + }, + { + "epoch": 0.9717514124293786, + "grad_norm": 2.7561733722686768, + "learning_rate": 3.7837837837837844e-06, + "loss": 1.4838, + "step": 946 + }, + { + "epoch": 0.9727786337955829, + "grad_norm": 2.544776201248169, + "learning_rate": 3.7818804720213178e-06, + "loss": 1.4539, + "step": 947 + }, + { + "epoch": 0.9738058551617874, + "grad_norm": 2.744417428970337, + "learning_rate": 3.7799771602588507e-06, + "loss": 1.4296, + "step": 948 + }, + { + "epoch": 0.9748330765279918, + "grad_norm": 2.7759811878204346, + "learning_rate": 3.778073848496384e-06, + "loss": 1.4314, + "step": 949 + }, + { + "epoch": 0.9758602978941961, + "grad_norm": 2.5204875469207764, + "learning_rate": 3.7761705367339174e-06, + "loss": 1.495, + "step": 950 + }, + { + "epoch": 0.9758602978941961, + "eval_loss": 1.8491986989974976, + "eval_runtime": 37.1266, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 0.9768875192604006, + "grad_norm": 2.5156631469726562, + "learning_rate": 3.7742672249714508e-06, + "loss": 1.5228, + "step": 951 + }, + { + "epoch": 0.977914740626605, + "grad_norm": 2.469834089279175, + "learning_rate": 3.772363913208984e-06, + "loss": 1.5188, + "step": 952 + }, + { + "epoch": 0.9789419619928095, + "grad_norm": 2.4933393001556396, + "learning_rate": 3.770460601446517e-06, + "loss": 1.4818, + "step": 953 + }, + { + "epoch": 0.9799691833590138, + "grad_norm": 2.5382790565490723, + "learning_rate": 3.7685572896840504e-06, + "loss": 1.5107, + "step": 954 + }, + { + "epoch": 0.9809964047252183, + "grad_norm": 2.6962392330169678, + "learning_rate": 3.766653977921584e-06, + "loss": 1.4534, + "step": 955 + }, + { + "epoch": 0.9820236260914227, + "grad_norm": 2.5411603450775146, + "learning_rate": 3.764750666159117e-06, + "loss": 1.3942, + "step": 956 + }, + { + "epoch": 0.9830508474576272, + "grad_norm": 2.7101845741271973, + "learning_rate": 3.7628473543966505e-06, + "loss": 1.5003, + "step": 957 + }, + { + "epoch": 0.9840780688238315, + "grad_norm": 2.5780858993530273, + "learning_rate": 3.7609440426341835e-06, + "loss": 1.4508, + "step": 958 + }, + { + "epoch": 0.9851052901900359, + "grad_norm": 2.814004421234131, + "learning_rate": 3.759040730871717e-06, + "loss": 1.5435, + "step": 959 + }, + { + "epoch": 0.9861325115562404, + "grad_norm": 2.5701088905334473, + "learning_rate": 3.75713741910925e-06, + "loss": 1.4386, + "step": 960 + }, + { + "epoch": 0.9861325115562404, + "eval_loss": 1.8556313514709473, + "eval_runtime": 37.1375, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 0.9871597329224447, + "grad_norm": 2.407271146774292, + "learning_rate": 3.7552341073467836e-06, + "loss": 1.4338, + "step": 961 + }, + { + "epoch": 0.9881869542886492, + "grad_norm": 2.433256149291992, + "learning_rate": 3.753330795584317e-06, + "loss": 1.47, + "step": 962 + }, + { + "epoch": 0.9892141756548536, + "grad_norm": 2.679339647293091, + "learning_rate": 3.75142748382185e-06, + "loss": 1.5459, + "step": 963 + }, + { + "epoch": 0.9902413970210581, + "grad_norm": 2.325840473175049, + "learning_rate": 3.749524172059384e-06, + "loss": 1.4385, + "step": 964 + }, + { + "epoch": 0.9912686183872624, + "grad_norm": 2.7060739994049072, + "learning_rate": 3.747620860296917e-06, + "loss": 1.4938, + "step": 965 + }, + { + "epoch": 0.9922958397534669, + "grad_norm": 2.589808464050293, + "learning_rate": 3.7457175485344504e-06, + "loss": 1.5258, + "step": 966 + }, + { + "epoch": 0.9933230611196713, + "grad_norm": 2.4876766204833984, + "learning_rate": 3.7438142367719837e-06, + "loss": 1.5361, + "step": 967 + }, + { + "epoch": 0.9943502824858758, + "grad_norm": 2.7370550632476807, + "learning_rate": 3.741910925009517e-06, + "loss": 1.4999, + "step": 968 + }, + { + "epoch": 0.9953775038520801, + "grad_norm": 2.6552398204803467, + "learning_rate": 3.7400076132470505e-06, + "loss": 1.4612, + "step": 969 + }, + { + "epoch": 0.9964047252182845, + "grad_norm": 2.5229146480560303, + "learning_rate": 3.7381043014845834e-06, + "loss": 1.4794, + "step": 970 + }, + { + "epoch": 0.9964047252182845, + "eval_loss": 1.8614000082015991, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 0.997431946584489, + "grad_norm": 2.5604071617126465, + "learning_rate": 3.7362009897221168e-06, + "loss": 1.5078, + "step": 971 + }, + { + "epoch": 0.9984591679506933, + "grad_norm": 2.614192485809326, + "learning_rate": 3.73429767795965e-06, + "loss": 1.4849, + "step": 972 + }, + { + "epoch": 0.9994863893168978, + "grad_norm": 2.5245072841644287, + "learning_rate": 3.7323943661971835e-06, + "loss": 1.4993, + "step": 973 + }, + { + "epoch": 1.0005136106831023, + "grad_norm": 2.5631163120269775, + "learning_rate": 3.730491054434717e-06, + "loss": 1.4136, + "step": 974 + }, + { + "epoch": 1.0015408320493067, + "grad_norm": 2.625396728515625, + "learning_rate": 3.72858774267225e-06, + "loss": 1.3623, + "step": 975 + }, + { + "epoch": 1.002568053415511, + "grad_norm": 2.7067251205444336, + "learning_rate": 3.726684430909783e-06, + "loss": 1.4527, + "step": 976 + }, + { + "epoch": 1.0035952747817154, + "grad_norm": 2.5352942943573, + "learning_rate": 3.7247811191473165e-06, + "loss": 1.474, + "step": 977 + }, + { + "epoch": 1.00462249614792, + "grad_norm": 2.818143367767334, + "learning_rate": 3.72287780738485e-06, + "loss": 1.4852, + "step": 978 + }, + { + "epoch": 1.0056497175141244, + "grad_norm": 2.8269577026367188, + "learning_rate": 3.7209744956223832e-06, + "loss": 1.4667, + "step": 979 + }, + { + "epoch": 1.0066769388803287, + "grad_norm": 2.837454319000244, + "learning_rate": 3.719071183859916e-06, + "loss": 1.4728, + "step": 980 + }, + { + "epoch": 1.0066769388803287, + "eval_loss": 1.8701857328414917, + "eval_runtime": 37.2859, + "eval_samples_per_second": 26.82, + "eval_steps_per_second": 1.126, + "step": 980 + }, + { + "epoch": 1.007704160246533, + "grad_norm": 2.6601309776306152, + "learning_rate": 3.7171678720974495e-06, + "loss": 1.3917, + "step": 981 + }, + { + "epoch": 1.0087313816127375, + "grad_norm": 2.761413097381592, + "learning_rate": 3.715264560334983e-06, + "loss": 1.5049, + "step": 982 + }, + { + "epoch": 1.009758602978942, + "grad_norm": 2.681429147720337, + "learning_rate": 3.7133612485725163e-06, + "loss": 1.4579, + "step": 983 + }, + { + "epoch": 1.0107858243451464, + "grad_norm": 2.693094253540039, + "learning_rate": 3.71145793681005e-06, + "loss": 1.4766, + "step": 984 + }, + { + "epoch": 1.0118130457113508, + "grad_norm": 2.589165210723877, + "learning_rate": 3.7095546250475834e-06, + "loss": 1.4011, + "step": 985 + }, + { + "epoch": 1.0128402670775551, + "grad_norm": 2.738140821456909, + "learning_rate": 3.7076513132851168e-06, + "loss": 1.5109, + "step": 986 + }, + { + "epoch": 1.0138674884437597, + "grad_norm": 2.7863996028900146, + "learning_rate": 3.7057480015226497e-06, + "loss": 1.4344, + "step": 987 + }, + { + "epoch": 1.014894709809964, + "grad_norm": 2.6011335849761963, + "learning_rate": 3.703844689760183e-06, + "loss": 1.3981, + "step": 988 + }, + { + "epoch": 1.0159219311761685, + "grad_norm": 2.5715689659118652, + "learning_rate": 3.7019413779977164e-06, + "loss": 1.4876, + "step": 989 + }, + { + "epoch": 1.0169491525423728, + "grad_norm": 2.6804113388061523, + "learning_rate": 3.70003806623525e-06, + "loss": 1.4576, + "step": 990 + }, + { + "epoch": 1.0169491525423728, + "eval_loss": 1.8802876472473145, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 990 + }, + { + "epoch": 1.0179763739085772, + "grad_norm": 2.737604856491089, + "learning_rate": 3.698134754472783e-06, + "loss": 1.4399, + "step": 991 + }, + { + "epoch": 1.0190035952747818, + "grad_norm": 2.658388137817383, + "learning_rate": 3.696231442710316e-06, + "loss": 1.5035, + "step": 992 + }, + { + "epoch": 1.0200308166409862, + "grad_norm": 2.7263925075531006, + "learning_rate": 3.6943281309478495e-06, + "loss": 1.4469, + "step": 993 + }, + { + "epoch": 1.0210580380071905, + "grad_norm": 2.7885773181915283, + "learning_rate": 3.692424819185383e-06, + "loss": 1.4641, + "step": 994 + }, + { + "epoch": 1.022085259373395, + "grad_norm": 2.8929779529571533, + "learning_rate": 3.690521507422916e-06, + "loss": 1.4727, + "step": 995 + }, + { + "epoch": 1.0231124807395995, + "grad_norm": 2.6605303287506104, + "learning_rate": 3.6886181956604496e-06, + "loss": 1.4153, + "step": 996 + }, + { + "epoch": 1.0241397021058039, + "grad_norm": 2.8678219318389893, + "learning_rate": 3.686714883897983e-06, + "loss": 1.4199, + "step": 997 + }, + { + "epoch": 1.0251669234720082, + "grad_norm": 2.884453058242798, + "learning_rate": 3.684811572135516e-06, + "loss": 1.4909, + "step": 998 + }, + { + "epoch": 1.0261941448382126, + "grad_norm": 2.6681578159332275, + "learning_rate": 3.6829082603730492e-06, + "loss": 1.4903, + "step": 999 + }, + { + "epoch": 1.027221366204417, + "grad_norm": 2.8013756275177, + "learning_rate": 3.6810049486105826e-06, + "loss": 1.449, + "step": 1000 + }, + { + "epoch": 1.027221366204417, + "eval_loss": 1.8735578060150146, + "eval_runtime": 37.4416, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1000 + }, + { + "epoch": 1.0282485875706215, + "grad_norm": 2.787750244140625, + "learning_rate": 3.679101636848116e-06, + "loss": 1.4808, + "step": 1001 + }, + { + "epoch": 1.029275808936826, + "grad_norm": 2.666874647140503, + "learning_rate": 3.6771983250856493e-06, + "loss": 1.4739, + "step": 1002 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.720586061477661, + "learning_rate": 3.6752950133231822e-06, + "loss": 1.4439, + "step": 1003 + }, + { + "epoch": 1.0313302516692346, + "grad_norm": 2.707771062850952, + "learning_rate": 3.6733917015607156e-06, + "loss": 1.4765, + "step": 1004 + }, + { + "epoch": 1.0323574730354392, + "grad_norm": 2.658092737197876, + "learning_rate": 3.6714883897982494e-06, + "loss": 1.4822, + "step": 1005 + }, + { + "epoch": 1.0333846944016436, + "grad_norm": 2.66788911819458, + "learning_rate": 3.6695850780357828e-06, + "loss": 1.4237, + "step": 1006 + }, + { + "epoch": 1.034411915767848, + "grad_norm": 2.638956308364868, + "learning_rate": 3.667681766273316e-06, + "loss": 1.4168, + "step": 1007 + }, + { + "epoch": 1.0354391371340523, + "grad_norm": 2.739384412765503, + "learning_rate": 3.6657784545108495e-06, + "loss": 1.4338, + "step": 1008 + }, + { + "epoch": 1.0364663585002567, + "grad_norm": 2.599773406982422, + "learning_rate": 3.6638751427483824e-06, + "loss": 1.5188, + "step": 1009 + }, + { + "epoch": 1.0374935798664613, + "grad_norm": 2.705876588821411, + "learning_rate": 3.6619718309859158e-06, + "loss": 1.468, + "step": 1010 + }, + { + "epoch": 1.0374935798664613, + "eval_loss": 1.8751709461212158, + "eval_runtime": 36.9681, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 1010 + }, + { + "epoch": 1.0385208012326657, + "grad_norm": 2.712103843688965, + "learning_rate": 3.660068519223449e-06, + "loss": 1.508, + "step": 1011 + }, + { + "epoch": 1.03954802259887, + "grad_norm": 2.6621742248535156, + "learning_rate": 3.6581652074609825e-06, + "loss": 1.464, + "step": 1012 + }, + { + "epoch": 1.0405752439650744, + "grad_norm": 2.6141257286071777, + "learning_rate": 3.656261895698516e-06, + "loss": 1.4429, + "step": 1013 + }, + { + "epoch": 1.041602465331279, + "grad_norm": 2.5651750564575195, + "learning_rate": 3.6543585839360492e-06, + "loss": 1.4254, + "step": 1014 + }, + { + "epoch": 1.0426296866974833, + "grad_norm": 2.598686933517456, + "learning_rate": 3.652455272173582e-06, + "loss": 1.436, + "step": 1015 + }, + { + "epoch": 1.0436569080636877, + "grad_norm": 2.559375524520874, + "learning_rate": 3.6505519604111155e-06, + "loss": 1.4359, + "step": 1016 + }, + { + "epoch": 1.044684129429892, + "grad_norm": 2.57920503616333, + "learning_rate": 3.648648648648649e-06, + "loss": 1.4524, + "step": 1017 + }, + { + "epoch": 1.0457113507960965, + "grad_norm": 2.6629042625427246, + "learning_rate": 3.6467453368861823e-06, + "loss": 1.4901, + "step": 1018 + }, + { + "epoch": 1.046738572162301, + "grad_norm": 2.6846625804901123, + "learning_rate": 3.6448420251237156e-06, + "loss": 1.4345, + "step": 1019 + }, + { + "epoch": 1.0477657935285054, + "grad_norm": 2.7288694381713867, + "learning_rate": 3.6429387133612486e-06, + "loss": 1.4632, + "step": 1020 + }, + { + "epoch": 1.0477657935285054, + "eval_loss": 1.878115177154541, + "eval_runtime": 37.0627, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.0487930148947098, + "grad_norm": 2.6689510345458984, + "learning_rate": 3.641035401598782e-06, + "loss": 1.4934, + "step": 1021 + }, + { + "epoch": 1.0498202362609141, + "grad_norm": 2.6762588024139404, + "learning_rate": 3.6391320898363153e-06, + "loss": 1.4284, + "step": 1022 + }, + { + "epoch": 1.0508474576271187, + "grad_norm": 2.7875330448150635, + "learning_rate": 3.6372287780738487e-06, + "loss": 1.4717, + "step": 1023 + }, + { + "epoch": 1.051874678993323, + "grad_norm": 2.9239425659179688, + "learning_rate": 3.635325466311382e-06, + "loss": 1.4504, + "step": 1024 + }, + { + "epoch": 1.0529019003595275, + "grad_norm": 2.700913429260254, + "learning_rate": 3.633422154548916e-06, + "loss": 1.4701, + "step": 1025 + }, + { + "epoch": 1.0539291217257318, + "grad_norm": 2.693148612976074, + "learning_rate": 3.6315188427864487e-06, + "loss": 1.4889, + "step": 1026 + }, + { + "epoch": 1.0549563430919364, + "grad_norm": 2.6215498447418213, + "learning_rate": 3.629615531023982e-06, + "loss": 1.4586, + "step": 1027 + }, + { + "epoch": 1.0559835644581408, + "grad_norm": 2.704425573348999, + "learning_rate": 3.6277122192615155e-06, + "loss": 1.4182, + "step": 1028 + }, + { + "epoch": 1.0570107858243452, + "grad_norm": 2.691352128982544, + "learning_rate": 3.625808907499049e-06, + "loss": 1.468, + "step": 1029 + }, + { + "epoch": 1.0580380071905495, + "grad_norm": 2.769744634628296, + "learning_rate": 3.623905595736582e-06, + "loss": 1.4375, + "step": 1030 + }, + { + "epoch": 1.0580380071905495, + "eval_loss": 1.8686742782592773, + "eval_runtime": 37.1278, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.059065228556754, + "grad_norm": 2.6012840270996094, + "learning_rate": 3.6220022839741156e-06, + "loss": 1.4334, + "step": 1031 + }, + { + "epoch": 1.0600924499229585, + "grad_norm": 2.641416072845459, + "learning_rate": 3.6200989722116485e-06, + "loss": 1.4186, + "step": 1032 + }, + { + "epoch": 1.0611196712891628, + "grad_norm": 2.821176052093506, + "learning_rate": 3.618195660449182e-06, + "loss": 1.4438, + "step": 1033 + }, + { + "epoch": 1.0621468926553672, + "grad_norm": 2.719303607940674, + "learning_rate": 3.6162923486867152e-06, + "loss": 1.4519, + "step": 1034 + }, + { + "epoch": 1.0631741140215716, + "grad_norm": 2.76882004737854, + "learning_rate": 3.6143890369242486e-06, + "loss": 1.3941, + "step": 1035 + }, + { + "epoch": 1.064201335387776, + "grad_norm": 2.601137399673462, + "learning_rate": 3.612485725161782e-06, + "loss": 1.4167, + "step": 1036 + }, + { + "epoch": 1.0652285567539805, + "grad_norm": 2.691140651702881, + "learning_rate": 3.610582413399315e-06, + "loss": 1.4361, + "step": 1037 + }, + { + "epoch": 1.066255778120185, + "grad_norm": 2.5779640674591064, + "learning_rate": 3.6086791016368482e-06, + "loss": 1.4237, + "step": 1038 + }, + { + "epoch": 1.0672829994863893, + "grad_norm": 2.6535801887512207, + "learning_rate": 3.6067757898743816e-06, + "loss": 1.436, + "step": 1039 + }, + { + "epoch": 1.0683102208525936, + "grad_norm": 2.8365395069122314, + "learning_rate": 3.604872478111915e-06, + "loss": 1.5067, + "step": 1040 + }, + { + "epoch": 1.0683102208525936, + "eval_loss": 1.8686083555221558, + "eval_runtime": 37.1495, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 1040 + }, + { + "epoch": 1.0693374422187982, + "grad_norm": 2.704638957977295, + "learning_rate": 3.6029691663494483e-06, + "loss": 1.5105, + "step": 1041 + }, + { + "epoch": 1.0703646635850026, + "grad_norm": 2.6670243740081787, + "learning_rate": 3.6010658545869813e-06, + "loss": 1.451, + "step": 1042 + }, + { + "epoch": 1.071391884951207, + "grad_norm": 2.8808884620666504, + "learning_rate": 3.5991625428245146e-06, + "loss": 1.4795, + "step": 1043 + }, + { + "epoch": 1.0724191063174113, + "grad_norm": 2.800075054168701, + "learning_rate": 3.597259231062048e-06, + "loss": 1.4758, + "step": 1044 + }, + { + "epoch": 1.073446327683616, + "grad_norm": 2.4989686012268066, + "learning_rate": 3.5953559192995818e-06, + "loss": 1.42, + "step": 1045 + }, + { + "epoch": 1.0744735490498203, + "grad_norm": 2.7234370708465576, + "learning_rate": 3.593452607537115e-06, + "loss": 1.4549, + "step": 1046 + }, + { + "epoch": 1.0755007704160247, + "grad_norm": 2.7219443321228027, + "learning_rate": 3.5915492957746485e-06, + "loss": 1.5245, + "step": 1047 + }, + { + "epoch": 1.076527991782229, + "grad_norm": 2.6963443756103516, + "learning_rate": 3.589645984012182e-06, + "loss": 1.4472, + "step": 1048 + }, + { + "epoch": 1.0775552131484334, + "grad_norm": 2.7101387977600098, + "learning_rate": 3.587742672249715e-06, + "loss": 1.4659, + "step": 1049 + }, + { + "epoch": 1.078582434514638, + "grad_norm": 2.7673099040985107, + "learning_rate": 3.585839360487248e-06, + "loss": 1.4284, + "step": 1050 + }, + { + "epoch": 1.078582434514638, + "eval_loss": 1.879286289215088, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.13, + "step": 1050 + }, + { + "epoch": 1.0796096558808423, + "grad_norm": 2.81614351272583, + "learning_rate": 3.5839360487247815e-06, + "loss": 1.4251, + "step": 1051 + }, + { + "epoch": 1.0806368772470467, + "grad_norm": 2.770350933074951, + "learning_rate": 3.582032736962315e-06, + "loss": 1.4372, + "step": 1052 + }, + { + "epoch": 1.081664098613251, + "grad_norm": 2.8099257946014404, + "learning_rate": 3.5801294251998483e-06, + "loss": 1.495, + "step": 1053 + }, + { + "epoch": 1.0826913199794557, + "grad_norm": 2.582378387451172, + "learning_rate": 3.578226113437381e-06, + "loss": 1.4005, + "step": 1054 + }, + { + "epoch": 1.08371854134566, + "grad_norm": 2.6215474605560303, + "learning_rate": 3.5763228016749146e-06, + "loss": 1.4428, + "step": 1055 + }, + { + "epoch": 1.0847457627118644, + "grad_norm": 2.7204151153564453, + "learning_rate": 3.574419489912448e-06, + "loss": 1.5197, + "step": 1056 + }, + { + "epoch": 1.0857729840780688, + "grad_norm": 2.6563117504119873, + "learning_rate": 3.5725161781499813e-06, + "loss": 1.457, + "step": 1057 + }, + { + "epoch": 1.0868002054442731, + "grad_norm": 2.559816837310791, + "learning_rate": 3.5706128663875147e-06, + "loss": 1.4376, + "step": 1058 + }, + { + "epoch": 1.0878274268104777, + "grad_norm": 2.6742372512817383, + "learning_rate": 3.5687095546250476e-06, + "loss": 1.4626, + "step": 1059 + }, + { + "epoch": 1.088854648176682, + "grad_norm": 2.8052852153778076, + "learning_rate": 3.566806242862581e-06, + "loss": 1.4785, + "step": 1060 + }, + { + "epoch": 1.088854648176682, + "eval_loss": 1.8775676488876343, + "eval_runtime": 37.1523, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1060 + }, + { + "epoch": 1.0898818695428865, + "grad_norm": 2.687411308288574, + "learning_rate": 3.5649029311001143e-06, + "loss": 1.4491, + "step": 1061 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.662604808807373, + "learning_rate": 3.5629996193376477e-06, + "loss": 1.4274, + "step": 1062 + }, + { + "epoch": 1.0919363122752954, + "grad_norm": 2.619046211242676, + "learning_rate": 3.561096307575181e-06, + "loss": 1.4449, + "step": 1063 + }, + { + "epoch": 1.0929635336414998, + "grad_norm": 2.672553539276123, + "learning_rate": 3.559192995812714e-06, + "loss": 1.4719, + "step": 1064 + }, + { + "epoch": 1.0939907550077042, + "grad_norm": 2.868760824203491, + "learning_rate": 3.557289684050248e-06, + "loss": 1.4955, + "step": 1065 + }, + { + "epoch": 1.0950179763739085, + "grad_norm": 2.7635276317596436, + "learning_rate": 3.555386372287781e-06, + "loss": 1.4765, + "step": 1066 + }, + { + "epoch": 1.0960451977401129, + "grad_norm": 2.883657455444336, + "learning_rate": 3.5534830605253145e-06, + "loss": 1.4923, + "step": 1067 + }, + { + "epoch": 1.0970724191063175, + "grad_norm": 2.7595503330230713, + "learning_rate": 3.551579748762848e-06, + "loss": 1.4527, + "step": 1068 + }, + { + "epoch": 1.0980996404725218, + "grad_norm": 2.8310179710388184, + "learning_rate": 3.5496764370003812e-06, + "loss": 1.4485, + "step": 1069 + }, + { + "epoch": 1.0991268618387262, + "grad_norm": 2.699432134628296, + "learning_rate": 3.5477731252379146e-06, + "loss": 1.4322, + "step": 1070 + }, + { + "epoch": 1.0991268618387262, + "eval_loss": 1.8706557750701904, + "eval_runtime": 37.1509, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1070 + }, + { + "epoch": 1.1001540832049306, + "grad_norm": 2.6249091625213623, + "learning_rate": 3.5458698134754475e-06, + "loss": 1.4031, + "step": 1071 + }, + { + "epoch": 1.1011813045711352, + "grad_norm": 2.6882944107055664, + "learning_rate": 3.543966501712981e-06, + "loss": 1.4075, + "step": 1072 + }, + { + "epoch": 1.1022085259373395, + "grad_norm": 2.6578996181488037, + "learning_rate": 3.5420631899505142e-06, + "loss": 1.4256, + "step": 1073 + }, + { + "epoch": 1.103235747303544, + "grad_norm": 2.8124637603759766, + "learning_rate": 3.5401598781880476e-06, + "loss": 1.5263, + "step": 1074 + }, + { + "epoch": 1.1042629686697483, + "grad_norm": 2.6022191047668457, + "learning_rate": 3.538256566425581e-06, + "loss": 1.3482, + "step": 1075 + }, + { + "epoch": 1.1052901900359529, + "grad_norm": 2.753009080886841, + "learning_rate": 3.536353254663114e-06, + "loss": 1.483, + "step": 1076 + }, + { + "epoch": 1.1063174114021572, + "grad_norm": 2.575666904449463, + "learning_rate": 3.5344499429006473e-06, + "loss": 1.4325, + "step": 1077 + }, + { + "epoch": 1.1073446327683616, + "grad_norm": 2.594136953353882, + "learning_rate": 3.5325466311381806e-06, + "loss": 1.4187, + "step": 1078 + }, + { + "epoch": 1.108371854134566, + "grad_norm": 2.807074785232544, + "learning_rate": 3.530643319375714e-06, + "loss": 1.4763, + "step": 1079 + }, + { + "epoch": 1.1093990755007703, + "grad_norm": 2.5865330696105957, + "learning_rate": 3.5287400076132474e-06, + "loss": 1.3896, + "step": 1080 + }, + { + "epoch": 1.1093990755007703, + "eval_loss": 1.8865792751312256, + "eval_runtime": 37.1514, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1080 + }, + { + "epoch": 1.110426296866975, + "grad_norm": 2.715419054031372, + "learning_rate": 3.5268366958507803e-06, + "loss": 1.4259, + "step": 1081 + }, + { + "epoch": 1.1114535182331793, + "grad_norm": 2.6837334632873535, + "learning_rate": 3.5249333840883137e-06, + "loss": 1.4581, + "step": 1082 + }, + { + "epoch": 1.1124807395993837, + "grad_norm": 2.722128391265869, + "learning_rate": 3.523030072325847e-06, + "loss": 1.4486, + "step": 1083 + }, + { + "epoch": 1.113507960965588, + "grad_norm": 2.7390329837799072, + "learning_rate": 3.5211267605633804e-06, + "loss": 1.4225, + "step": 1084 + }, + { + "epoch": 1.1145351823317924, + "grad_norm": 2.740271806716919, + "learning_rate": 3.519223448800914e-06, + "loss": 1.4274, + "step": 1085 + }, + { + "epoch": 1.115562403697997, + "grad_norm": 2.8321313858032227, + "learning_rate": 3.5173201370384475e-06, + "loss": 1.4337, + "step": 1086 + }, + { + "epoch": 1.1165896250642013, + "grad_norm": 2.740826368331909, + "learning_rate": 3.515416825275981e-06, + "loss": 1.4331, + "step": 1087 + }, + { + "epoch": 1.1176168464304057, + "grad_norm": 2.7382850646972656, + "learning_rate": 3.513513513513514e-06, + "loss": 1.4495, + "step": 1088 + }, + { + "epoch": 1.11864406779661, + "grad_norm": 2.7677972316741943, + "learning_rate": 3.511610201751047e-06, + "loss": 1.4628, + "step": 1089 + }, + { + "epoch": 1.1196712891628147, + "grad_norm": 2.8850181102752686, + "learning_rate": 3.5097068899885806e-06, + "loss": 1.3739, + "step": 1090 + }, + { + "epoch": 1.1196712891628147, + "eval_loss": 1.8794182538986206, + "eval_runtime": 37.4388, + "eval_samples_per_second": 26.71, + "eval_steps_per_second": 1.122, + "step": 1090 + }, + { + "epoch": 1.120698510529019, + "grad_norm": 2.73730731010437, + "learning_rate": 3.507803578226114e-06, + "loss": 1.4444, + "step": 1091 + }, + { + "epoch": 1.1217257318952234, + "grad_norm": 3.027599811553955, + "learning_rate": 3.5059002664636473e-06, + "loss": 1.5123, + "step": 1092 + }, + { + "epoch": 1.1227529532614278, + "grad_norm": 2.613658905029297, + "learning_rate": 3.5039969547011802e-06, + "loss": 1.3798, + "step": 1093 + }, + { + "epoch": 1.1237801746276324, + "grad_norm": 2.6396987438201904, + "learning_rate": 3.5020936429387136e-06, + "loss": 1.4357, + "step": 1094 + }, + { + "epoch": 1.1248073959938367, + "grad_norm": 2.625540018081665, + "learning_rate": 3.500190331176247e-06, + "loss": 1.4354, + "step": 1095 + }, + { + "epoch": 1.125834617360041, + "grad_norm": 2.8413279056549072, + "learning_rate": 3.4982870194137803e-06, + "loss": 1.4846, + "step": 1096 + }, + { + "epoch": 1.1268618387262455, + "grad_norm": 2.7898473739624023, + "learning_rate": 3.4963837076513137e-06, + "loss": 1.421, + "step": 1097 + }, + { + "epoch": 1.1278890600924498, + "grad_norm": 2.6385879516601562, + "learning_rate": 3.4944803958888466e-06, + "loss": 1.4241, + "step": 1098 + }, + { + "epoch": 1.1289162814586544, + "grad_norm": 2.677098512649536, + "learning_rate": 3.49257708412638e-06, + "loss": 1.3671, + "step": 1099 + }, + { + "epoch": 1.1299435028248588, + "grad_norm": 2.8582913875579834, + "learning_rate": 3.4906737723639133e-06, + "loss": 1.4886, + "step": 1100 + }, + { + "epoch": 1.1299435028248588, + "eval_loss": 1.8696506023406982, + "eval_runtime": 37.462, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 1100 + }, + { + "epoch": 1.1309707241910631, + "grad_norm": 2.7157933712005615, + "learning_rate": 3.4887704606014467e-06, + "loss": 1.4964, + "step": 1101 + }, + { + "epoch": 1.1319979455572675, + "grad_norm": 2.6703319549560547, + "learning_rate": 3.48686714883898e-06, + "loss": 1.4509, + "step": 1102 + }, + { + "epoch": 1.133025166923472, + "grad_norm": 2.6307899951934814, + "learning_rate": 3.484963837076513e-06, + "loss": 1.485, + "step": 1103 + }, + { + "epoch": 1.1340523882896765, + "grad_norm": 2.8333141803741455, + "learning_rate": 3.4830605253140464e-06, + "loss": 1.432, + "step": 1104 + }, + { + "epoch": 1.1350796096558808, + "grad_norm": 2.545804023742676, + "learning_rate": 3.4811572135515797e-06, + "loss": 1.3799, + "step": 1105 + }, + { + "epoch": 1.1361068310220852, + "grad_norm": 2.7070441246032715, + "learning_rate": 3.4792539017891135e-06, + "loss": 1.4193, + "step": 1106 + }, + { + "epoch": 1.1371340523882898, + "grad_norm": 2.697140693664551, + "learning_rate": 3.477350590026647e-06, + "loss": 1.398, + "step": 1107 + }, + { + "epoch": 1.1381612737544942, + "grad_norm": 2.784795045852661, + "learning_rate": 3.4754472782641802e-06, + "loss": 1.4708, + "step": 1108 + }, + { + "epoch": 1.1391884951206985, + "grad_norm": 2.8518831729888916, + "learning_rate": 3.4735439665017136e-06, + "loss": 1.4707, + "step": 1109 + }, + { + "epoch": 1.140215716486903, + "grad_norm": 3.0481607913970947, + "learning_rate": 3.4716406547392465e-06, + "loss": 1.4385, + "step": 1110 + }, + { + "epoch": 1.140215716486903, + "eval_loss": 1.8367345333099365, + "eval_runtime": 36.9539, + "eval_samples_per_second": 27.061, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.1412429378531073, + "grad_norm": 2.803053140640259, + "learning_rate": 3.46973734297678e-06, + "loss": 1.4361, + "step": 1111 + }, + { + "epoch": 1.1422701592193119, + "grad_norm": 2.762638568878174, + "learning_rate": 3.4678340312143133e-06, + "loss": 1.4522, + "step": 1112 + }, + { + "epoch": 1.1432973805855162, + "grad_norm": 2.9458558559417725, + "learning_rate": 3.4659307194518466e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.1443246019517206, + "grad_norm": 2.774721622467041, + "learning_rate": 3.46402740768938e-06, + "loss": 1.4723, + "step": 1114 + }, + { + "epoch": 1.145351823317925, + "grad_norm": 2.6556880474090576, + "learning_rate": 3.462124095926913e-06, + "loss": 1.4073, + "step": 1115 + }, + { + "epoch": 1.1463790446841293, + "grad_norm": 2.7482454776763916, + "learning_rate": 3.4602207841644463e-06, + "loss": 1.4022, + "step": 1116 + }, + { + "epoch": 1.147406266050334, + "grad_norm": 2.627567768096924, + "learning_rate": 3.4583174724019797e-06, + "loss": 1.4428, + "step": 1117 + }, + { + "epoch": 1.1484334874165383, + "grad_norm": 2.750666856765747, + "learning_rate": 3.456414160639513e-06, + "loss": 1.4972, + "step": 1118 + }, + { + "epoch": 1.1494607087827426, + "grad_norm": 2.842184543609619, + "learning_rate": 3.4545108488770464e-06, + "loss": 1.4753, + "step": 1119 + }, + { + "epoch": 1.150487930148947, + "grad_norm": 2.829437732696533, + "learning_rate": 3.4526075371145793e-06, + "loss": 1.4525, + "step": 1120 + }, + { + "epoch": 1.150487930148947, + "eval_loss": 1.8498426675796509, + "eval_runtime": 37.0234, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 1120 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.7674920558929443, + "learning_rate": 3.4507042253521127e-06, + "loss": 1.4516, + "step": 1121 + }, + { + "epoch": 1.152542372881356, + "grad_norm": 2.726806879043579, + "learning_rate": 3.448800913589646e-06, + "loss": 1.5113, + "step": 1122 + }, + { + "epoch": 1.1535695942475603, + "grad_norm": 2.7698283195495605, + "learning_rate": 3.4468976018271794e-06, + "loss": 1.4295, + "step": 1123 + }, + { + "epoch": 1.1545968156137647, + "grad_norm": 2.6224706172943115, + "learning_rate": 3.4449942900647128e-06, + "loss": 1.5179, + "step": 1124 + }, + { + "epoch": 1.1556240369799693, + "grad_norm": 2.7266266345977783, + "learning_rate": 3.443090978302246e-06, + "loss": 1.4784, + "step": 1125 + }, + { + "epoch": 1.1566512583461737, + "grad_norm": 2.631751298904419, + "learning_rate": 3.44118766653978e-06, + "loss": 1.4659, + "step": 1126 + }, + { + "epoch": 1.157678479712378, + "grad_norm": 2.7077064514160156, + "learning_rate": 3.439284354777313e-06, + "loss": 1.4654, + "step": 1127 + }, + { + "epoch": 1.1587057010785824, + "grad_norm": 2.735102415084839, + "learning_rate": 3.4373810430148462e-06, + "loss": 1.4604, + "step": 1128 + }, + { + "epoch": 1.1597329224447868, + "grad_norm": 2.9314727783203125, + "learning_rate": 3.4354777312523796e-06, + "loss": 1.3964, + "step": 1129 + }, + { + "epoch": 1.1607601438109914, + "grad_norm": 2.658930778503418, + "learning_rate": 3.433574419489913e-06, + "loss": 1.4331, + "step": 1130 + }, + { + "epoch": 1.1607601438109914, + "eval_loss": 1.8545702695846558, + "eval_runtime": 37.0493, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1130 + }, + { + "epoch": 1.1617873651771957, + "grad_norm": 2.8146262168884277, + "learning_rate": 3.4316711077274463e-06, + "loss": 1.4711, + "step": 1131 + }, + { + "epoch": 1.1628145865434, + "grad_norm": 2.6918272972106934, + "learning_rate": 3.4297677959649792e-06, + "loss": 1.4327, + "step": 1132 + }, + { + "epoch": 1.1638418079096045, + "grad_norm": 2.680551528930664, + "learning_rate": 3.4278644842025126e-06, + "loss": 1.4886, + "step": 1133 + }, + { + "epoch": 1.1648690292758088, + "grad_norm": 2.617286443710327, + "learning_rate": 3.425961172440046e-06, + "loss": 1.4167, + "step": 1134 + }, + { + "epoch": 1.1658962506420134, + "grad_norm": 2.6772518157958984, + "learning_rate": 3.4240578606775793e-06, + "loss": 1.5274, + "step": 1135 + }, + { + "epoch": 1.1669234720082178, + "grad_norm": 2.640723705291748, + "learning_rate": 3.4221545489151127e-06, + "loss": 1.4398, + "step": 1136 + }, + { + "epoch": 1.1679506933744221, + "grad_norm": 2.7104315757751465, + "learning_rate": 3.4202512371526456e-06, + "loss": 1.4367, + "step": 1137 + }, + { + "epoch": 1.1689779147406265, + "grad_norm": 2.831645965576172, + "learning_rate": 3.418347925390179e-06, + "loss": 1.5329, + "step": 1138 + }, + { + "epoch": 1.170005136106831, + "grad_norm": 2.7174274921417236, + "learning_rate": 3.4164446136277124e-06, + "loss": 1.481, + "step": 1139 + }, + { + "epoch": 1.1710323574730355, + "grad_norm": 2.6496286392211914, + "learning_rate": 3.4145413018652457e-06, + "loss": 1.4489, + "step": 1140 + }, + { + "epoch": 1.1710323574730355, + "eval_loss": 1.8524384498596191, + "eval_runtime": 37.1347, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 1140 + }, + { + "epoch": 1.1720595788392398, + "grad_norm": 2.610058069229126, + "learning_rate": 3.412637990102779e-06, + "loss": 1.4755, + "step": 1141 + }, + { + "epoch": 1.1730868002054442, + "grad_norm": 2.642070770263672, + "learning_rate": 3.4107346783403125e-06, + "loss": 1.4981, + "step": 1142 + }, + { + "epoch": 1.1741140215716488, + "grad_norm": 2.57493257522583, + "learning_rate": 3.4088313665778454e-06, + "loss": 1.4039, + "step": 1143 + }, + { + "epoch": 1.1751412429378532, + "grad_norm": 2.5919628143310547, + "learning_rate": 3.4069280548153788e-06, + "loss": 1.4391, + "step": 1144 + }, + { + "epoch": 1.1761684643040575, + "grad_norm": 2.698862314224243, + "learning_rate": 3.405024743052912e-06, + "loss": 1.4198, + "step": 1145 + }, + { + "epoch": 1.177195685670262, + "grad_norm": 2.6387715339660645, + "learning_rate": 3.403121431290446e-06, + "loss": 1.4373, + "step": 1146 + }, + { + "epoch": 1.1782229070364663, + "grad_norm": 2.764643430709839, + "learning_rate": 3.4012181195279793e-06, + "loss": 1.4348, + "step": 1147 + }, + { + "epoch": 1.1792501284026708, + "grad_norm": 2.721038579940796, + "learning_rate": 3.3993148077655126e-06, + "loss": 1.4232, + "step": 1148 + }, + { + "epoch": 1.1802773497688752, + "grad_norm": 2.8942360877990723, + "learning_rate": 3.3974114960030456e-06, + "loss": 1.4405, + "step": 1149 + }, + { + "epoch": 1.1813045711350796, + "grad_norm": 2.687194585800171, + "learning_rate": 3.395508184240579e-06, + "loss": 1.3854, + "step": 1150 + }, + { + "epoch": 1.1813045711350796, + "eval_loss": 1.8701989650726318, + "eval_runtime": 37.132, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.182331792501284, + "grad_norm": 2.7137982845306396, + "learning_rate": 3.3936048724781123e-06, + "loss": 1.4174, + "step": 1151 + }, + { + "epoch": 1.1833590138674885, + "grad_norm": 2.8916432857513428, + "learning_rate": 3.3917015607156457e-06, + "loss": 1.5043, + "step": 1152 + }, + { + "epoch": 1.184386235233693, + "grad_norm": 2.6625723838806152, + "learning_rate": 3.389798248953179e-06, + "loss": 1.3832, + "step": 1153 + }, + { + "epoch": 1.1854134565998973, + "grad_norm": 2.8155555725097656, + "learning_rate": 3.387894937190712e-06, + "loss": 1.4098, + "step": 1154 + }, + { + "epoch": 1.1864406779661016, + "grad_norm": 2.634751796722412, + "learning_rate": 3.3859916254282453e-06, + "loss": 1.4522, + "step": 1155 + }, + { + "epoch": 1.1874678993323062, + "grad_norm": 2.7085063457489014, + "learning_rate": 3.3840883136657787e-06, + "loss": 1.4539, + "step": 1156 + }, + { + "epoch": 1.1884951206985106, + "grad_norm": 2.65588116645813, + "learning_rate": 3.382185001903312e-06, + "loss": 1.5084, + "step": 1157 + }, + { + "epoch": 1.189522342064715, + "grad_norm": 2.582489252090454, + "learning_rate": 3.3802816901408454e-06, + "loss": 1.4218, + "step": 1158 + }, + { + "epoch": 1.1905495634309193, + "grad_norm": 2.6364378929138184, + "learning_rate": 3.3783783783783788e-06, + "loss": 1.4522, + "step": 1159 + }, + { + "epoch": 1.1915767847971237, + "grad_norm": 2.6329915523529053, + "learning_rate": 3.3764750666159117e-06, + "loss": 1.442, + "step": 1160 + }, + { + "epoch": 1.1915767847971237, + "eval_loss": 1.8593298196792603, + "eval_runtime": 37.1143, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1160 + }, + { + "epoch": 1.1926040061633283, + "grad_norm": 2.610050916671753, + "learning_rate": 3.374571754853445e-06, + "loss": 1.4397, + "step": 1161 + }, + { + "epoch": 1.1936312275295327, + "grad_norm": 2.5503945350646973, + "learning_rate": 3.3726684430909784e-06, + "loss": 1.4211, + "step": 1162 + }, + { + "epoch": 1.194658448895737, + "grad_norm": 2.6648080348968506, + "learning_rate": 3.370765131328512e-06, + "loss": 1.3938, + "step": 1163 + }, + { + "epoch": 1.1956856702619414, + "grad_norm": 2.6705355644226074, + "learning_rate": 3.368861819566045e-06, + "loss": 1.4107, + "step": 1164 + }, + { + "epoch": 1.1967128916281458, + "grad_norm": 2.7582640647888184, + "learning_rate": 3.366958507803578e-06, + "loss": 1.4327, + "step": 1165 + }, + { + "epoch": 1.1977401129943503, + "grad_norm": 2.7768008708953857, + "learning_rate": 3.365055196041112e-06, + "loss": 1.454, + "step": 1166 + }, + { + "epoch": 1.1987673343605547, + "grad_norm": 2.628997325897217, + "learning_rate": 3.3631518842786452e-06, + "loss": 1.3988, + "step": 1167 + }, + { + "epoch": 1.199794555726759, + "grad_norm": 2.637505531311035, + "learning_rate": 3.3612485725161786e-06, + "loss": 1.4766, + "step": 1168 + }, + { + "epoch": 1.2008217770929634, + "grad_norm": 2.6935489177703857, + "learning_rate": 3.359345260753712e-06, + "loss": 1.4063, + "step": 1169 + }, + { + "epoch": 1.201848998459168, + "grad_norm": 2.7827870845794678, + "learning_rate": 3.3574419489912453e-06, + "loss": 1.4882, + "step": 1170 + }, + { + "epoch": 1.201848998459168, + "eval_loss": 1.8575458526611328, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.2028762198253724, + "grad_norm": 2.789053201675415, + "learning_rate": 3.3555386372287783e-06, + "loss": 1.4699, + "step": 1171 + }, + { + "epoch": 1.2039034411915768, + "grad_norm": 3.053184747695923, + "learning_rate": 3.3536353254663116e-06, + "loss": 1.4993, + "step": 1172 + }, + { + "epoch": 1.2049306625577811, + "grad_norm": 2.6963112354278564, + "learning_rate": 3.351732013703845e-06, + "loss": 1.3985, + "step": 1173 + }, + { + "epoch": 1.2059578839239857, + "grad_norm": 2.689101219177246, + "learning_rate": 3.3498287019413784e-06, + "loss": 1.4464, + "step": 1174 + }, + { + "epoch": 1.20698510529019, + "grad_norm": 2.7748899459838867, + "learning_rate": 3.3479253901789117e-06, + "loss": 1.4554, + "step": 1175 + }, + { + "epoch": 1.2080123266563945, + "grad_norm": 2.8673830032348633, + "learning_rate": 3.346022078416445e-06, + "loss": 1.4796, + "step": 1176 + }, + { + "epoch": 1.2090395480225988, + "grad_norm": 2.7719533443450928, + "learning_rate": 3.344118766653978e-06, + "loss": 1.3716, + "step": 1177 + }, + { + "epoch": 1.2100667693888032, + "grad_norm": 2.71771240234375, + "learning_rate": 3.3422154548915114e-06, + "loss": 1.4554, + "step": 1178 + }, + { + "epoch": 1.2110939907550078, + "grad_norm": 2.7095048427581787, + "learning_rate": 3.3403121431290448e-06, + "loss": 1.4233, + "step": 1179 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 2.631638526916504, + "learning_rate": 3.338408831366578e-06, + "loss": 1.4581, + "step": 1180 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.855039119720459, + "eval_runtime": 37.1083, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1180 + }, + { + "epoch": 1.2131484334874165, + "grad_norm": 2.657257318496704, + "learning_rate": 3.3365055196041115e-06, + "loss": 1.3864, + "step": 1181 + }, + { + "epoch": 1.214175654853621, + "grad_norm": 2.894169569015503, + "learning_rate": 3.3346022078416444e-06, + "loss": 1.4673, + "step": 1182 + }, + { + "epoch": 1.2152028762198253, + "grad_norm": 2.852644920349121, + "learning_rate": 3.3326988960791778e-06, + "loss": 1.5026, + "step": 1183 + }, + { + "epoch": 1.2162300975860298, + "grad_norm": 2.720878839492798, + "learning_rate": 3.330795584316711e-06, + "loss": 1.4281, + "step": 1184 + }, + { + "epoch": 1.2172573189522342, + "grad_norm": 2.6779472827911377, + "learning_rate": 3.3288922725542445e-06, + "loss": 1.3559, + "step": 1185 + }, + { + "epoch": 1.2182845403184386, + "grad_norm": 2.836782217025757, + "learning_rate": 3.326988960791778e-06, + "loss": 1.4185, + "step": 1186 + }, + { + "epoch": 1.219311761684643, + "grad_norm": 2.9181265830993652, + "learning_rate": 3.3250856490293117e-06, + "loss": 1.4379, + "step": 1187 + }, + { + "epoch": 1.2203389830508475, + "grad_norm": 2.7315778732299805, + "learning_rate": 3.323182337266845e-06, + "loss": 1.4824, + "step": 1188 + }, + { + "epoch": 1.221366204417052, + "grad_norm": 2.8483448028564453, + "learning_rate": 3.321279025504378e-06, + "loss": 1.4766, + "step": 1189 + }, + { + "epoch": 1.2223934257832563, + "grad_norm": 2.736323118209839, + "learning_rate": 3.3193757137419113e-06, + "loss": 1.4647, + "step": 1190 + }, + { + "epoch": 1.2223934257832563, + "eval_loss": 1.8581864833831787, + "eval_runtime": 37.3401, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.2234206471494606, + "grad_norm": 2.8863027095794678, + "learning_rate": 3.3174724019794447e-06, + "loss": 1.4533, + "step": 1191 + }, + { + "epoch": 1.2244478685156652, + "grad_norm": 2.951228380203247, + "learning_rate": 3.315569090216978e-06, + "loss": 1.495, + "step": 1192 + }, + { + "epoch": 1.2254750898818696, + "grad_norm": 2.7486696243286133, + "learning_rate": 3.3136657784545114e-06, + "loss": 1.4082, + "step": 1193 + }, + { + "epoch": 1.226502311248074, + "grad_norm": 2.9105849266052246, + "learning_rate": 3.3117624666920443e-06, + "loss": 1.4988, + "step": 1194 + }, + { + "epoch": 1.2275295326142783, + "grad_norm": 2.8165998458862305, + "learning_rate": 3.3098591549295777e-06, + "loss": 1.4162, + "step": 1195 + }, + { + "epoch": 1.2285567539804827, + "grad_norm": 2.8999876976013184, + "learning_rate": 3.307955843167111e-06, + "loss": 1.3909, + "step": 1196 + }, + { + "epoch": 1.2295839753466873, + "grad_norm": 2.8325014114379883, + "learning_rate": 3.3060525314046444e-06, + "loss": 1.4625, + "step": 1197 + }, + { + "epoch": 1.2306111967128917, + "grad_norm": 2.6553618907928467, + "learning_rate": 3.304149219642178e-06, + "loss": 1.4453, + "step": 1198 + }, + { + "epoch": 1.231638418079096, + "grad_norm": 2.584757089614868, + "learning_rate": 3.3022459078797107e-06, + "loss": 1.4412, + "step": 1199 + }, + { + "epoch": 1.2326656394453004, + "grad_norm": 2.68215274810791, + "learning_rate": 3.300342596117244e-06, + "loss": 1.4511, + "step": 1200 + }, + { + "epoch": 1.2326656394453004, + "eval_loss": 1.849502682685852, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 1200 + }, + { + "epoch": 1.2336928608115048, + "grad_norm": 2.735494375228882, + "learning_rate": 3.2984392843547775e-06, + "loss": 1.4433, + "step": 1201 + }, + { + "epoch": 1.2347200821777093, + "grad_norm": 2.6691815853118896, + "learning_rate": 3.296535972592311e-06, + "loss": 1.4033, + "step": 1202 + }, + { + "epoch": 1.2357473035439137, + "grad_norm": 2.6378355026245117, + "learning_rate": 3.294632660829844e-06, + "loss": 1.378, + "step": 1203 + }, + { + "epoch": 1.236774524910118, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4227, + "step": 1204 + }, + { + "epoch": 1.2378017462763227, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4204, + "step": 1205 + }, + { + "epoch": 1.238828967642527, + "grad_norm": 2.786475896835327, + "learning_rate": 3.2908260373049105e-06, + "loss": 1.4165, + "step": 1206 + }, + { + "epoch": 1.2398561890087314, + "grad_norm": 2.936825752258301, + "learning_rate": 3.288922725542444e-06, + "loss": 1.4712, + "step": 1207 + }, + { + "epoch": 1.2408834103749358, + "grad_norm": 2.7255940437316895, + "learning_rate": 3.2870194137799776e-06, + "loss": 1.4177, + "step": 1208 + }, + { + "epoch": 1.2419106317411401, + "grad_norm": 2.6526482105255127, + "learning_rate": 3.285116102017511e-06, + "loss": 1.3975, + "step": 1209 + }, + { + "epoch": 1.2429378531073447, + "grad_norm": 2.7539143562316895, + "learning_rate": 3.2832127902550444e-06, + "loss": 1.4717, + "step": 1210 + }, + { + "epoch": 1.2429378531073447, + "eval_loss": 1.8445062637329102, + "eval_runtime": 36.9889, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 1210 + }, + { + "epoch": 1.243965074473549, + "grad_norm": 2.907252073287964, + "learning_rate": 3.2813094784925777e-06, + "loss": 1.4484, + "step": 1211 + }, + { + "epoch": 1.2449922958397535, + "grad_norm": 2.7312281131744385, + "learning_rate": 3.2794061667301107e-06, + "loss": 1.3863, + "step": 1212 + }, + { + "epoch": 1.2460195172059578, + "grad_norm": 2.739184617996216, + "learning_rate": 3.277502854967644e-06, + "loss": 1.4302, + "step": 1213 + }, + { + "epoch": 1.2470467385721622, + "grad_norm": 2.798630475997925, + "learning_rate": 3.2755995432051774e-06, + "loss": 1.4134, + "step": 1214 + }, + { + "epoch": 1.2480739599383668, + "grad_norm": 2.835686683654785, + "learning_rate": 3.2736962314427107e-06, + "loss": 1.4258, + "step": 1215 + }, + { + "epoch": 1.2491011813045712, + "grad_norm": 2.7404251098632812, + "learning_rate": 3.271792919680244e-06, + "loss": 1.4347, + "step": 1216 + }, + { + "epoch": 1.2501284026707755, + "grad_norm": 2.8362114429473877, + "learning_rate": 3.269889607917777e-06, + "loss": 1.4909, + "step": 1217 + }, + { + "epoch": 1.25115562403698, + "grad_norm": 2.842283248901367, + "learning_rate": 3.2679862961553104e-06, + "loss": 1.4065, + "step": 1218 + }, + { + "epoch": 1.2521828454031843, + "grad_norm": 2.717536449432373, + "learning_rate": 3.2660829843928438e-06, + "loss": 1.4446, + "step": 1219 + }, + { + "epoch": 1.2532100667693888, + "grad_norm": 2.687009572982788, + "learning_rate": 3.264179672630377e-06, + "loss": 1.4826, + "step": 1220 + }, + { + "epoch": 1.2532100667693888, + "eval_loss": 1.8576964139938354, + "eval_runtime": 37.0759, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1220 + }, + { + "epoch": 1.2542372881355932, + "grad_norm": 2.7234137058258057, + "learning_rate": 3.2622763608679105e-06, + "loss": 1.4503, + "step": 1221 + }, + { + "epoch": 1.2552645095017976, + "grad_norm": 2.733876943588257, + "learning_rate": 3.2603730491054434e-06, + "loss": 1.4076, + "step": 1222 + }, + { + "epoch": 1.2562917308680022, + "grad_norm": 2.6480257511138916, + "learning_rate": 3.258469737342977e-06, + "loss": 1.422, + "step": 1223 + }, + { + "epoch": 1.2573189522342065, + "grad_norm": 2.7006945610046387, + "learning_rate": 3.25656642558051e-06, + "loss": 1.3975, + "step": 1224 + }, + { + "epoch": 1.258346173600411, + "grad_norm": 2.6176297664642334, + "learning_rate": 3.2546631138180435e-06, + "loss": 1.4138, + "step": 1225 + }, + { + "epoch": 1.2593733949666153, + "grad_norm": 2.717742443084717, + "learning_rate": 3.252759802055577e-06, + "loss": 1.4092, + "step": 1226 + }, + { + "epoch": 1.2604006163328196, + "grad_norm": 2.693643093109131, + "learning_rate": 3.25085649029311e-06, + "loss": 1.4221, + "step": 1227 + }, + { + "epoch": 1.2614278376990242, + "grad_norm": 2.709026575088501, + "learning_rate": 3.248953178530644e-06, + "loss": 1.4474, + "step": 1228 + }, + { + "epoch": 1.2624550590652286, + "grad_norm": 2.628399610519409, + "learning_rate": 3.247049866768177e-06, + "loss": 1.4075, + "step": 1229 + }, + { + "epoch": 1.263482280431433, + "grad_norm": 2.681164503097534, + "learning_rate": 3.2451465550057103e-06, + "loss": 1.3683, + "step": 1230 + }, + { + "epoch": 1.263482280431433, + "eval_loss": 1.8543845415115356, + "eval_runtime": 37.1602, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1230 + }, + { + "epoch": 1.2645095017976373, + "grad_norm": 2.6538004875183105, + "learning_rate": 3.2432432432432437e-06, + "loss": 1.4214, + "step": 1231 + }, + { + "epoch": 1.2655367231638417, + "grad_norm": 2.714805841445923, + "learning_rate": 3.241339931480777e-06, + "loss": 1.4617, + "step": 1232 + }, + { + "epoch": 1.2665639445300463, + "grad_norm": 2.743016481399536, + "learning_rate": 3.2394366197183104e-06, + "loss": 1.4556, + "step": 1233 + }, + { + "epoch": 1.2675911658962506, + "grad_norm": 2.618269681930542, + "learning_rate": 3.2375333079558434e-06, + "loss": 1.419, + "step": 1234 + }, + { + "epoch": 1.268618387262455, + "grad_norm": 2.672124147415161, + "learning_rate": 3.2356299961933767e-06, + "loss": 1.4477, + "step": 1235 + }, + { + "epoch": 1.2696456086286596, + "grad_norm": 2.831037998199463, + "learning_rate": 3.23372668443091e-06, + "loss": 1.4552, + "step": 1236 + }, + { + "epoch": 1.270672829994864, + "grad_norm": 2.713193893432617, + "learning_rate": 3.2318233726684435e-06, + "loss": 1.4653, + "step": 1237 + }, + { + "epoch": 1.2717000513610683, + "grad_norm": 2.645097017288208, + "learning_rate": 3.229920060905977e-06, + "loss": 1.4087, + "step": 1238 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.741934061050415, + "learning_rate": 3.2280167491435098e-06, + "loss": 1.4155, + "step": 1239 + }, + { + "epoch": 1.273754494093477, + "grad_norm": 2.5912556648254395, + "learning_rate": 3.226113437381043e-06, + "loss": 1.4214, + "step": 1240 + }, + { + "epoch": 1.273754494093477, + "eval_loss": 1.8524503707885742, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1240 + }, + { + "epoch": 1.2747817154596817, + "grad_norm": 2.5774216651916504, + "learning_rate": 3.2242101256185765e-06, + "loss": 1.4309, + "step": 1241 + }, + { + "epoch": 1.275808936825886, + "grad_norm": 2.7143211364746094, + "learning_rate": 3.22230681385611e-06, + "loss": 1.4928, + "step": 1242 + }, + { + "epoch": 1.2768361581920904, + "grad_norm": 2.8101789951324463, + "learning_rate": 3.220403502093643e-06, + "loss": 1.4325, + "step": 1243 + }, + { + "epoch": 1.2778633795582948, + "grad_norm": 2.704641580581665, + "learning_rate": 3.218500190331176e-06, + "loss": 1.4729, + "step": 1244 + }, + { + "epoch": 1.2788906009244991, + "grad_norm": 2.649195671081543, + "learning_rate": 3.2165968785687095e-06, + "loss": 1.4186, + "step": 1245 + }, + { + "epoch": 1.2799178222907037, + "grad_norm": 2.7863168716430664, + "learning_rate": 3.214693566806243e-06, + "loss": 1.4474, + "step": 1246 + }, + { + "epoch": 1.280945043656908, + "grad_norm": 2.693589687347412, + "learning_rate": 3.2127902550437762e-06, + "loss": 1.4293, + "step": 1247 + }, + { + "epoch": 1.2819722650231125, + "grad_norm": 2.7287509441375732, + "learning_rate": 3.21088694328131e-06, + "loss": 1.4305, + "step": 1248 + }, + { + "epoch": 1.2829994863893168, + "grad_norm": 2.5521676540374756, + "learning_rate": 3.2089836315188434e-06, + "loss": 1.4203, + "step": 1249 + }, + { + "epoch": 1.2840267077555212, + "grad_norm": 2.745425224304199, + "learning_rate": 3.2070803197563767e-06, + "loss": 1.4234, + "step": 1250 + }, + { + "epoch": 1.2840267077555212, + "eval_loss": 1.8586679697036743, + "eval_runtime": 37.1173, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1250 + }, + { + "epoch": 1.2850539291217258, + "grad_norm": 2.687504768371582, + "learning_rate": 3.2051770079939097e-06, + "loss": 1.443, + "step": 1251 + }, + { + "epoch": 1.2860811504879301, + "grad_norm": 2.7841391563415527, + "learning_rate": 3.203273696231443e-06, + "loss": 1.4722, + "step": 1252 + }, + { + "epoch": 1.2871083718541345, + "grad_norm": 2.7715537548065186, + "learning_rate": 3.2013703844689764e-06, + "loss": 1.4063, + "step": 1253 + }, + { + "epoch": 1.288135593220339, + "grad_norm": 2.8183677196502686, + "learning_rate": 3.1994670727065098e-06, + "loss": 1.4534, + "step": 1254 + }, + { + "epoch": 1.2891628145865435, + "grad_norm": 2.795215606689453, + "learning_rate": 3.197563760944043e-06, + "loss": 1.4779, + "step": 1255 + }, + { + "epoch": 1.2901900359527478, + "grad_norm": 2.7606441974639893, + "learning_rate": 3.195660449181576e-06, + "loss": 1.4668, + "step": 1256 + }, + { + "epoch": 1.2912172573189522, + "grad_norm": 2.629718065261841, + "learning_rate": 3.1937571374191094e-06, + "loss": 1.4114, + "step": 1257 + }, + { + "epoch": 1.2922444786851566, + "grad_norm": 2.776845932006836, + "learning_rate": 3.191853825656643e-06, + "loss": 1.4348, + "step": 1258 + }, + { + "epoch": 1.2932717000513612, + "grad_norm": 2.672982931137085, + "learning_rate": 3.189950513894176e-06, + "loss": 1.4274, + "step": 1259 + }, + { + "epoch": 1.2942989214175655, + "grad_norm": 2.5834884643554688, + "learning_rate": 3.1880472021317095e-06, + "loss": 1.4282, + "step": 1260 + }, + { + "epoch": 1.2942989214175655, + "eval_loss": 1.8541877269744873, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1260 + }, + { + "epoch": 1.29532614278377, + "grad_norm": 2.752054452896118, + "learning_rate": 3.1861438903692425e-06, + "loss": 1.3797, + "step": 1261 + }, + { + "epoch": 1.2963533641499743, + "grad_norm": 2.723689556121826, + "learning_rate": 3.184240578606776e-06, + "loss": 1.4137, + "step": 1262 + }, + { + "epoch": 1.2973805855161786, + "grad_norm": 2.806439161300659, + "learning_rate": 3.182337266844309e-06, + "loss": 1.4396, + "step": 1263 + }, + { + "epoch": 1.2984078068823832, + "grad_norm": 2.8140552043914795, + "learning_rate": 3.1804339550818426e-06, + "loss": 1.4677, + "step": 1264 + }, + { + "epoch": 1.2994350282485876, + "grad_norm": 2.78153920173645, + "learning_rate": 3.178530643319376e-06, + "loss": 1.433, + "step": 1265 + }, + { + "epoch": 1.300462249614792, + "grad_norm": 2.6926140785217285, + "learning_rate": 3.176627331556909e-06, + "loss": 1.3901, + "step": 1266 + }, + { + "epoch": 1.3014894709809965, + "grad_norm": 2.636629581451416, + "learning_rate": 3.1747240197944422e-06, + "loss": 1.4331, + "step": 1267 + }, + { + "epoch": 1.3025166923472007, + "grad_norm": 2.625173330307007, + "learning_rate": 3.172820708031976e-06, + "loss": 1.3918, + "step": 1268 + }, + { + "epoch": 1.3035439137134053, + "grad_norm": 2.745666980743408, + "learning_rate": 3.1709173962695094e-06, + "loss": 1.4669, + "step": 1269 + }, + { + "epoch": 1.3045711350796096, + "grad_norm": 2.8781471252441406, + "learning_rate": 3.1690140845070427e-06, + "loss": 1.4586, + "step": 1270 + }, + { + "epoch": 1.3045711350796096, + "eval_loss": 1.8759416341781616, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1270 + }, + { + "epoch": 1.305598356445814, + "grad_norm": 2.832078456878662, + "learning_rate": 3.167110772744576e-06, + "loss": 1.4385, + "step": 1271 + }, + { + "epoch": 1.3066255778120186, + "grad_norm": 2.7153384685516357, + "learning_rate": 3.1652074609821095e-06, + "loss": 1.427, + "step": 1272 + }, + { + "epoch": 1.307652799178223, + "grad_norm": 2.6645848751068115, + "learning_rate": 3.1633041492196424e-06, + "loss": 1.4147, + "step": 1273 + }, + { + "epoch": 1.3086800205444273, + "grad_norm": 2.7231268882751465, + "learning_rate": 3.1614008374571758e-06, + "loss": 1.436, + "step": 1274 + }, + { + "epoch": 1.3097072419106317, + "grad_norm": 2.8131442070007324, + "learning_rate": 3.159497525694709e-06, + "loss": 1.4357, + "step": 1275 + }, + { + "epoch": 1.310734463276836, + "grad_norm": 2.6366660594940186, + "learning_rate": 3.1575942139322425e-06, + "loss": 1.4089, + "step": 1276 + }, + { + "epoch": 1.3117616846430407, + "grad_norm": 2.7849857807159424, + "learning_rate": 3.155690902169776e-06, + "loss": 1.4486, + "step": 1277 + }, + { + "epoch": 1.312788906009245, + "grad_norm": 2.771298408508301, + "learning_rate": 3.1537875904073088e-06, + "loss": 1.4085, + "step": 1278 + }, + { + "epoch": 1.3138161273754494, + "grad_norm": 2.7618181705474854, + "learning_rate": 3.151884278644842e-06, + "loss": 1.4272, + "step": 1279 + }, + { + "epoch": 1.3148433487416538, + "grad_norm": 2.6741080284118652, + "learning_rate": 3.1499809668823755e-06, + "loss": 1.4135, + "step": 1280 + }, + { + "epoch": 1.3148433487416538, + "eval_loss": 1.8567897081375122, + "eval_runtime": 37.1429, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1280 + }, + { + "epoch": 1.3158705701078581, + "grad_norm": 2.6671698093414307, + "learning_rate": 3.148077655119909e-06, + "loss": 1.4598, + "step": 1281 + }, + { + "epoch": 1.3168977914740627, + "grad_norm": 2.6641244888305664, + "learning_rate": 3.1461743433574422e-06, + "loss": 1.3998, + "step": 1282 + }, + { + "epoch": 1.317925012840267, + "grad_norm": 2.747417449951172, + "learning_rate": 3.144271031594975e-06, + "loss": 1.4602, + "step": 1283 + }, + { + "epoch": 1.3189522342064715, + "grad_norm": 2.708984136581421, + "learning_rate": 3.1423677198325085e-06, + "loss": 1.4562, + "step": 1284 + }, + { + "epoch": 1.319979455572676, + "grad_norm": 2.5973010063171387, + "learning_rate": 3.140464408070042e-06, + "loss": 1.4121, + "step": 1285 + }, + { + "epoch": 1.3210066769388804, + "grad_norm": 2.6334621906280518, + "learning_rate": 3.1385610963075753e-06, + "loss": 1.4049, + "step": 1286 + }, + { + "epoch": 1.3220338983050848, + "grad_norm": 2.772655963897705, + "learning_rate": 3.1366577845451086e-06, + "loss": 1.3953, + "step": 1287 + }, + { + "epoch": 1.3230611196712891, + "grad_norm": 2.5006916522979736, + "learning_rate": 3.134754472782642e-06, + "loss": 1.3842, + "step": 1288 + }, + { + "epoch": 1.3240883410374935, + "grad_norm": 2.6052703857421875, + "learning_rate": 3.1328511610201758e-06, + "loss": 1.3801, + "step": 1289 + }, + { + "epoch": 1.325115562403698, + "grad_norm": 2.703693151473999, + "learning_rate": 3.1309478492577087e-06, + "loss": 1.3672, + "step": 1290 + }, + { + "epoch": 1.325115562403698, + "eval_loss": 1.8470649719238281, + "eval_runtime": 37.2714, + "eval_samples_per_second": 26.83, + "eval_steps_per_second": 1.127, + "step": 1290 + }, + { + "epoch": 1.3261427837699025, + "grad_norm": 2.6864330768585205, + "learning_rate": 3.129044537495242e-06, + "loss": 1.4252, + "step": 1291 + }, + { + "epoch": 1.3271700051361068, + "grad_norm": 2.7768399715423584, + "learning_rate": 3.1271412257327754e-06, + "loss": 1.4483, + "step": 1292 + }, + { + "epoch": 1.3281972265023112, + "grad_norm": 2.6658599376678467, + "learning_rate": 3.125237913970309e-06, + "loss": 1.4059, + "step": 1293 + }, + { + "epoch": 1.3292244478685156, + "grad_norm": 2.754718065261841, + "learning_rate": 3.123334602207842e-06, + "loss": 1.3787, + "step": 1294 + }, + { + "epoch": 1.3302516692347202, + "grad_norm": 2.7994842529296875, + "learning_rate": 3.121431290445375e-06, + "loss": 1.4623, + "step": 1295 + }, + { + "epoch": 1.3312788906009245, + "grad_norm": 2.805941343307495, + "learning_rate": 3.1195279786829085e-06, + "loss": 1.5186, + "step": 1296 + }, + { + "epoch": 1.332306111967129, + "grad_norm": 2.6829605102539062, + "learning_rate": 3.117624666920442e-06, + "loss": 1.4052, + "step": 1297 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.7221593856811523, + "learning_rate": 3.115721355157975e-06, + "loss": 1.4031, + "step": 1298 + }, + { + "epoch": 1.3343605546995376, + "grad_norm": 2.579638719558716, + "learning_rate": 3.1138180433955085e-06, + "loss": 1.3627, + "step": 1299 + }, + { + "epoch": 1.3353877760657422, + "grad_norm": 2.861189842224121, + "learning_rate": 3.1119147316330415e-06, + "loss": 1.4405, + "step": 1300 + }, + { + "epoch": 1.3353877760657422, + "eval_loss": 1.8476718664169312, + "eval_runtime": 37.4298, + "eval_samples_per_second": 26.717, + "eval_steps_per_second": 1.122, + "step": 1300 + }, + { + "epoch": 1.3364149974319466, + "grad_norm": 2.8968658447265625, + "learning_rate": 3.110011419870575e-06, + "loss": 1.4774, + "step": 1301 + }, + { + "epoch": 1.337442218798151, + "grad_norm": 2.8708558082580566, + "learning_rate": 3.1081081081081082e-06, + "loss": 1.4085, + "step": 1302 + }, + { + "epoch": 1.3384694401643555, + "grad_norm": 2.802922487258911, + "learning_rate": 3.1062047963456416e-06, + "loss": 1.4486, + "step": 1303 + }, + { + "epoch": 1.33949666153056, + "grad_norm": 2.9320907592773438, + "learning_rate": 3.104301484583175e-06, + "loss": 1.4932, + "step": 1304 + }, + { + "epoch": 1.3405238828967643, + "grad_norm": 2.7708749771118164, + "learning_rate": 3.1023981728207083e-06, + "loss": 1.4601, + "step": 1305 + }, + { + "epoch": 1.3415511042629686, + "grad_norm": 2.7464756965637207, + "learning_rate": 3.1004948610582412e-06, + "loss": 1.4245, + "step": 1306 + }, + { + "epoch": 1.342578325629173, + "grad_norm": 2.767235517501831, + "learning_rate": 3.0985915492957746e-06, + "loss": 1.4199, + "step": 1307 + }, + { + "epoch": 1.3436055469953776, + "grad_norm": 2.6810076236724854, + "learning_rate": 3.096688237533308e-06, + "loss": 1.3851, + "step": 1308 + }, + { + "epoch": 1.344632768361582, + "grad_norm": 2.6390531063079834, + "learning_rate": 3.0947849257708418e-06, + "loss": 1.4688, + "step": 1309 + }, + { + "epoch": 1.3456599897277863, + "grad_norm": 2.729029893875122, + "learning_rate": 3.092881614008375e-06, + "loss": 1.4248, + "step": 1310 + }, + { + "epoch": 1.3456599897277863, + "eval_loss": 1.854402780532837, + "eval_runtime": 36.9357, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.3466872110939907, + "grad_norm": 2.8098297119140625, + "learning_rate": 3.0909783022459085e-06, + "loss": 1.4761, + "step": 1311 + }, + { + "epoch": 1.347714432460195, + "grad_norm": 2.8076772689819336, + "learning_rate": 3.0890749904834414e-06, + "loss": 1.4119, + "step": 1312 + }, + { + "epoch": 1.3487416538263997, + "grad_norm": 2.8214125633239746, + "learning_rate": 3.0871716787209748e-06, + "loss": 1.4693, + "step": 1313 + }, + { + "epoch": 1.349768875192604, + "grad_norm": 2.830083131790161, + "learning_rate": 3.085268366958508e-06, + "loss": 1.438, + "step": 1314 + }, + { + "epoch": 1.3507960965588084, + "grad_norm": 2.7671611309051514, + "learning_rate": 3.0833650551960415e-06, + "loss": 1.4417, + "step": 1315 + }, + { + "epoch": 1.3518233179250128, + "grad_norm": 2.6464383602142334, + "learning_rate": 3.081461743433575e-06, + "loss": 1.4096, + "step": 1316 + }, + { + "epoch": 1.3528505392912171, + "grad_norm": 2.651646137237549, + "learning_rate": 3.079558431671108e-06, + "loss": 1.4264, + "step": 1317 + }, + { + "epoch": 1.3538777606574217, + "grad_norm": 2.6222681999206543, + "learning_rate": 3.077655119908641e-06, + "loss": 1.3836, + "step": 1318 + }, + { + "epoch": 1.354904982023626, + "grad_norm": 2.5674147605895996, + "learning_rate": 3.0757518081461745e-06, + "loss": 1.4206, + "step": 1319 + }, + { + "epoch": 1.3559322033898304, + "grad_norm": 2.7767961025238037, + "learning_rate": 3.073848496383708e-06, + "loss": 1.4348, + "step": 1320 + }, + { + "epoch": 1.3559322033898304, + "eval_loss": 1.8607728481292725, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 1320 + }, + { + "epoch": 1.356959424756035, + "grad_norm": 2.7811837196350098, + "learning_rate": 3.0719451846212413e-06, + "loss": 1.4082, + "step": 1321 + }, + { + "epoch": 1.3579866461222394, + "grad_norm": 2.5633227825164795, + "learning_rate": 3.0700418728587746e-06, + "loss": 1.4352, + "step": 1322 + }, + { + "epoch": 1.3590138674884438, + "grad_norm": 2.7205374240875244, + "learning_rate": 3.0681385610963076e-06, + "loss": 1.405, + "step": 1323 + }, + { + "epoch": 1.3600410888546481, + "grad_norm": 2.82409930229187, + "learning_rate": 3.066235249333841e-06, + "loss": 1.448, + "step": 1324 + }, + { + "epoch": 1.3610683102208525, + "grad_norm": 2.7932820320129395, + "learning_rate": 3.0643319375713743e-06, + "loss": 1.4785, + "step": 1325 + }, + { + "epoch": 1.362095531587057, + "grad_norm": 2.68669056892395, + "learning_rate": 3.0624286258089076e-06, + "loss": 1.4232, + "step": 1326 + }, + { + "epoch": 1.3631227529532615, + "grad_norm": 2.6492018699645996, + "learning_rate": 3.060525314046441e-06, + "loss": 1.4052, + "step": 1327 + }, + { + "epoch": 1.3641499743194658, + "grad_norm": 2.7677114009857178, + "learning_rate": 3.058622002283974e-06, + "loss": 1.4426, + "step": 1328 + }, + { + "epoch": 1.3651771956856702, + "grad_norm": 2.7124569416046143, + "learning_rate": 3.0567186905215077e-06, + "loss": 1.4631, + "step": 1329 + }, + { + "epoch": 1.3662044170518746, + "grad_norm": 2.7084693908691406, + "learning_rate": 3.054815378759041e-06, + "loss": 1.4656, + "step": 1330 + }, + { + "epoch": 1.3662044170518746, + "eval_loss": 1.8457012176513672, + "eval_runtime": 37.0141, + "eval_samples_per_second": 27.017, + "eval_steps_per_second": 1.135, + "step": 1330 + }, + { + "epoch": 1.3672316384180792, + "grad_norm": 2.7612688541412354, + "learning_rate": 3.0529120669965745e-06, + "loss": 1.455, + "step": 1331 + }, + { + "epoch": 1.3682588597842835, + "grad_norm": 2.5172924995422363, + "learning_rate": 3.051008755234108e-06, + "loss": 1.3915, + "step": 1332 + }, + { + "epoch": 1.3692860811504879, + "grad_norm": 2.7225120067596436, + "learning_rate": 3.049105443471641e-06, + "loss": 1.4025, + "step": 1333 + }, + { + "epoch": 1.3703133025166925, + "grad_norm": 2.835470676422119, + "learning_rate": 3.0472021317091745e-06, + "loss": 1.5423, + "step": 1334 + }, + { + "epoch": 1.3713405238828968, + "grad_norm": 2.6247284412384033, + "learning_rate": 3.0452988199467075e-06, + "loss": 1.4562, + "step": 1335 + }, + { + "epoch": 1.3723677452491012, + "grad_norm": 2.6346960067749023, + "learning_rate": 3.043395508184241e-06, + "loss": 1.4566, + "step": 1336 + }, + { + "epoch": 1.3733949666153056, + "grad_norm": 2.6278724670410156, + "learning_rate": 3.041492196421774e-06, + "loss": 1.403, + "step": 1337 + }, + { + "epoch": 1.37442218798151, + "grad_norm": 2.8066110610961914, + "learning_rate": 3.0395888846593076e-06, + "loss": 1.4433, + "step": 1338 + }, + { + "epoch": 1.3754494093477145, + "grad_norm": 2.6565256118774414, + "learning_rate": 3.037685572896841e-06, + "loss": 1.4177, + "step": 1339 + }, + { + "epoch": 1.376476630713919, + "grad_norm": 2.6529884338378906, + "learning_rate": 3.035782261134374e-06, + "loss": 1.4378, + "step": 1340 + }, + { + "epoch": 1.376476630713919, + "eval_loss": 1.8618167638778687, + "eval_runtime": 37.025, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1340 + }, + { + "epoch": 1.3775038520801233, + "grad_norm": 2.820838689804077, + "learning_rate": 3.0338789493719072e-06, + "loss": 1.4471, + "step": 1341 + }, + { + "epoch": 1.3785310734463276, + "grad_norm": 2.6786739826202393, + "learning_rate": 3.0319756376094406e-06, + "loss": 1.4393, + "step": 1342 + }, + { + "epoch": 1.379558294812532, + "grad_norm": 2.7962610721588135, + "learning_rate": 3.030072325846974e-06, + "loss": 1.4407, + "step": 1343 + }, + { + "epoch": 1.3805855161787366, + "grad_norm": 2.739959716796875, + "learning_rate": 3.0281690140845073e-06, + "loss": 1.4059, + "step": 1344 + }, + { + "epoch": 1.381612737544941, + "grad_norm": 2.7812039852142334, + "learning_rate": 3.0262657023220403e-06, + "loss": 1.4449, + "step": 1345 + }, + { + "epoch": 1.3826399589111453, + "grad_norm": 2.749671459197998, + "learning_rate": 3.0243623905595736e-06, + "loss": 1.4459, + "step": 1346 + }, + { + "epoch": 1.3836671802773497, + "grad_norm": 2.7494630813598633, + "learning_rate": 3.022459078797107e-06, + "loss": 1.4204, + "step": 1347 + }, + { + "epoch": 1.384694401643554, + "grad_norm": 2.7188141345977783, + "learning_rate": 3.0205557670346404e-06, + "loss": 1.4249, + "step": 1348 + }, + { + "epoch": 1.3857216230097587, + "grad_norm": 2.712442636489868, + "learning_rate": 3.018652455272174e-06, + "loss": 1.3975, + "step": 1349 + }, + { + "epoch": 1.386748844375963, + "grad_norm": 2.9888060092926025, + "learning_rate": 3.0167491435097075e-06, + "loss": 1.5258, + "step": 1350 + }, + { + "epoch": 1.386748844375963, + "eval_loss": 1.8591758012771606, + "eval_runtime": 37.0078, + "eval_samples_per_second": 27.021, + "eval_steps_per_second": 1.135, + "step": 1350 + }, + { + "epoch": 1.3877760657421674, + "grad_norm": 2.789698600769043, + "learning_rate": 3.014845831747241e-06, + "loss": 1.4305, + "step": 1351 + }, + { + "epoch": 1.388803287108372, + "grad_norm": 2.747840166091919, + "learning_rate": 3.012942519984774e-06, + "loss": 1.4031, + "step": 1352 + }, + { + "epoch": 1.3898305084745763, + "grad_norm": 2.7078518867492676, + "learning_rate": 3.011039208222307e-06, + "loss": 1.4139, + "step": 1353 + }, + { + "epoch": 1.3908577298407807, + "grad_norm": 2.615677833557129, + "learning_rate": 3.0091358964598405e-06, + "loss": 1.3728, + "step": 1354 + }, + { + "epoch": 1.391884951206985, + "grad_norm": 2.7485580444335938, + "learning_rate": 3.007232584697374e-06, + "loss": 1.4272, + "step": 1355 + }, + { + "epoch": 1.3929121725731894, + "grad_norm": 2.7168128490448, + "learning_rate": 3.0053292729349073e-06, + "loss": 1.4353, + "step": 1356 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 2.7295777797698975, + "learning_rate": 3.00342596117244e-06, + "loss": 1.4324, + "step": 1357 + }, + { + "epoch": 1.3949666153055984, + "grad_norm": 2.8791208267211914, + "learning_rate": 3.0015226494099736e-06, + "loss": 1.4303, + "step": 1358 + }, + { + "epoch": 1.3959938366718028, + "grad_norm": 2.647028684616089, + "learning_rate": 2.999619337647507e-06, + "loss": 1.402, + "step": 1359 + }, + { + "epoch": 1.3970210580380071, + "grad_norm": 2.869901180267334, + "learning_rate": 2.9977160258850403e-06, + "loss": 1.404, + "step": 1360 + }, + { + "epoch": 1.3970210580380071, + "eval_loss": 1.8588491678237915, + "eval_runtime": 37.0159, + "eval_samples_per_second": 27.015, + "eval_steps_per_second": 1.135, + "step": 1360 + }, + { + "epoch": 1.3980482794042115, + "grad_norm": 2.7664403915405273, + "learning_rate": 2.9958127141225736e-06, + "loss": 1.3895, + "step": 1361 + }, + { + "epoch": 1.399075500770416, + "grad_norm": 2.8101043701171875, + "learning_rate": 2.9939094023601066e-06, + "loss": 1.4629, + "step": 1362 + }, + { + "epoch": 1.4001027221366205, + "grad_norm": 2.7114098072052, + "learning_rate": 2.99200609059764e-06, + "loss": 1.4365, + "step": 1363 + }, + { + "epoch": 1.4011299435028248, + "grad_norm": 2.6739253997802734, + "learning_rate": 2.9901027788351733e-06, + "loss": 1.382, + "step": 1364 + }, + { + "epoch": 1.4021571648690292, + "grad_norm": 2.8102970123291016, + "learning_rate": 2.9881994670727067e-06, + "loss": 1.392, + "step": 1365 + }, + { + "epoch": 1.4031843862352336, + "grad_norm": 2.645071506500244, + "learning_rate": 2.98629615531024e-06, + "loss": 1.4157, + "step": 1366 + }, + { + "epoch": 1.4042116076014381, + "grad_norm": 2.6110310554504395, + "learning_rate": 2.984392843547773e-06, + "loss": 1.3996, + "step": 1367 + }, + { + "epoch": 1.4052388289676425, + "grad_norm": 2.784090042114258, + "learning_rate": 2.9824895317853063e-06, + "loss": 1.4178, + "step": 1368 + }, + { + "epoch": 1.4062660503338469, + "grad_norm": 2.7293829917907715, + "learning_rate": 2.98058622002284e-06, + "loss": 1.4193, + "step": 1369 + }, + { + "epoch": 1.4072932717000515, + "grad_norm": 2.6924493312835693, + "learning_rate": 2.9786829082603735e-06, + "loss": 1.476, + "step": 1370 + }, + { + "epoch": 1.4072932717000515, + "eval_loss": 1.8642805814743042, + "eval_runtime": 37.0452, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1370 + }, + { + "epoch": 1.4083204930662558, + "grad_norm": 2.7714414596557617, + "learning_rate": 2.976779596497907e-06, + "loss": 1.4574, + "step": 1371 + }, + { + "epoch": 1.4093477144324602, + "grad_norm": 2.7013490200042725, + "learning_rate": 2.97487628473544e-06, + "loss": 1.4718, + "step": 1372 + }, + { + "epoch": 1.4103749357986646, + "grad_norm": 2.9120547771453857, + "learning_rate": 2.9729729729729736e-06, + "loss": 1.4298, + "step": 1373 + }, + { + "epoch": 1.411402157164869, + "grad_norm": 2.9127988815307617, + "learning_rate": 2.9710696612105065e-06, + "loss": 1.4703, + "step": 1374 + }, + { + "epoch": 1.4124293785310735, + "grad_norm": 2.696772575378418, + "learning_rate": 2.96916634944804e-06, + "loss": 1.412, + "step": 1375 + }, + { + "epoch": 1.413456599897278, + "grad_norm": 2.8330435752868652, + "learning_rate": 2.9672630376855732e-06, + "loss": 1.4731, + "step": 1376 + }, + { + "epoch": 1.4144838212634823, + "grad_norm": 2.7354564666748047, + "learning_rate": 2.9653597259231066e-06, + "loss": 1.4342, + "step": 1377 + }, + { + "epoch": 1.4155110426296866, + "grad_norm": 2.7386012077331543, + "learning_rate": 2.96345641416064e-06, + "loss": 1.3935, + "step": 1378 + }, + { + "epoch": 1.416538263995891, + "grad_norm": 2.5881052017211914, + "learning_rate": 2.961553102398173e-06, + "loss": 1.3282, + "step": 1379 + }, + { + "epoch": 1.4175654853620956, + "grad_norm": 2.805558443069458, + "learning_rate": 2.9596497906357063e-06, + "loss": 1.4691, + "step": 1380 + }, + { + "epoch": 1.4175654853620956, + "eval_loss": 1.84714937210083, + "eval_runtime": 37.0349, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1380 + }, + { + "epoch": 1.4185927067283, + "grad_norm": 2.8639137744903564, + "learning_rate": 2.9577464788732396e-06, + "loss": 1.4745, + "step": 1381 + }, + { + "epoch": 1.4196199280945043, + "grad_norm": 2.6688785552978516, + "learning_rate": 2.955843167110773e-06, + "loss": 1.4016, + "step": 1382 + }, + { + "epoch": 1.420647149460709, + "grad_norm": 2.6851978302001953, + "learning_rate": 2.9539398553483064e-06, + "loss": 1.4205, + "step": 1383 + }, + { + "epoch": 1.4216743708269133, + "grad_norm": 2.7381017208099365, + "learning_rate": 2.9520365435858393e-06, + "loss": 1.4315, + "step": 1384 + }, + { + "epoch": 1.4227015921931176, + "grad_norm": 2.9249424934387207, + "learning_rate": 2.9501332318233727e-06, + "loss": 1.49, + "step": 1385 + }, + { + "epoch": 1.423728813559322, + "grad_norm": 2.765490770339966, + "learning_rate": 2.948229920060906e-06, + "loss": 1.4359, + "step": 1386 + }, + { + "epoch": 1.4247560349255264, + "grad_norm": 2.578561544418335, + "learning_rate": 2.9463266082984394e-06, + "loss": 1.4032, + "step": 1387 + }, + { + "epoch": 1.425783256291731, + "grad_norm": 2.6673238277435303, + "learning_rate": 2.9444232965359727e-06, + "loss": 1.4144, + "step": 1388 + }, + { + "epoch": 1.4268104776579353, + "grad_norm": 2.737516164779663, + "learning_rate": 2.9425199847735057e-06, + "loss": 1.4375, + "step": 1389 + }, + { + "epoch": 1.4278376990241397, + "grad_norm": 2.656177282333374, + "learning_rate": 2.94061667301104e-06, + "loss": 1.4007, + "step": 1390 + }, + { + "epoch": 1.4278376990241397, + "eval_loss": 1.8427128791809082, + "eval_runtime": 37.1322, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1390 + }, + { + "epoch": 1.428864920390344, + "grad_norm": 2.6011996269226074, + "learning_rate": 2.938713361248573e-06, + "loss": 1.339, + "step": 1391 + }, + { + "epoch": 1.4298921417565484, + "grad_norm": 2.6374619007110596, + "learning_rate": 2.936810049486106e-06, + "loss": 1.4092, + "step": 1392 + }, + { + "epoch": 1.430919363122753, + "grad_norm": 2.6879520416259766, + "learning_rate": 2.9349067377236396e-06, + "loss": 1.4235, + "step": 1393 + }, + { + "epoch": 1.4319465844889574, + "grad_norm": 2.8400580883026123, + "learning_rate": 2.933003425961173e-06, + "loss": 1.381, + "step": 1394 + }, + { + "epoch": 1.4329738058551618, + "grad_norm": 2.742767810821533, + "learning_rate": 2.9311001141987063e-06, + "loss": 1.4055, + "step": 1395 + }, + { + "epoch": 1.4340010272213661, + "grad_norm": 2.734239101409912, + "learning_rate": 2.9291968024362392e-06, + "loss": 1.3997, + "step": 1396 + }, + { + "epoch": 1.4350282485875705, + "grad_norm": 2.6337532997131348, + "learning_rate": 2.9272934906737726e-06, + "loss": 1.3825, + "step": 1397 + }, + { + "epoch": 1.436055469953775, + "grad_norm": 2.668790578842163, + "learning_rate": 2.925390178911306e-06, + "loss": 1.3918, + "step": 1398 + }, + { + "epoch": 1.4370826913199795, + "grad_norm": 2.8307507038116455, + "learning_rate": 2.9234868671488393e-06, + "loss": 1.4183, + "step": 1399 + }, + { + "epoch": 1.4381099126861838, + "grad_norm": 2.6661787033081055, + "learning_rate": 2.9215835553863727e-06, + "loss": 1.4225, + "step": 1400 + }, + { + "epoch": 1.4381099126861838, + "eval_loss": 1.8518847227096558, + "eval_runtime": 37.3672, + "eval_samples_per_second": 26.761, + "eval_steps_per_second": 1.124, + "step": 1400 + }, + { + "epoch": 1.4391371340523884, + "grad_norm": 2.698025703430176, + "learning_rate": 2.9196802436239056e-06, + "loss": 1.4115, + "step": 1401 + }, + { + "epoch": 1.4401643554185928, + "grad_norm": 2.6792502403259277, + "learning_rate": 2.917776931861439e-06, + "loss": 1.3869, + "step": 1402 + }, + { + "epoch": 1.4411915767847971, + "grad_norm": 2.693514347076416, + "learning_rate": 2.9158736200989723e-06, + "loss": 1.3875, + "step": 1403 + }, + { + "epoch": 1.4422187981510015, + "grad_norm": 2.6383583545684814, + "learning_rate": 2.9139703083365057e-06, + "loss": 1.3942, + "step": 1404 + }, + { + "epoch": 1.4432460195172059, + "grad_norm": 2.6992499828338623, + "learning_rate": 2.912066996574039e-06, + "loss": 1.4243, + "step": 1405 + }, + { + "epoch": 1.4442732408834105, + "grad_norm": 2.7901880741119385, + "learning_rate": 2.910163684811572e-06, + "loss": 1.3559, + "step": 1406 + }, + { + "epoch": 1.4453004622496148, + "grad_norm": 2.9373207092285156, + "learning_rate": 2.9082603730491054e-06, + "loss": 1.4734, + "step": 1407 + }, + { + "epoch": 1.4463276836158192, + "grad_norm": 2.8224740028381348, + "learning_rate": 2.9063570612866387e-06, + "loss": 1.4474, + "step": 1408 + }, + { + "epoch": 1.4473549049820236, + "grad_norm": 2.9534835815429688, + "learning_rate": 2.904453749524172e-06, + "loss": 1.3691, + "step": 1409 + }, + { + "epoch": 1.448382126348228, + "grad_norm": 2.8153076171875, + "learning_rate": 2.902550437761706e-06, + "loss": 1.3954, + "step": 1410 + }, + { + "epoch": 1.448382126348228, + "eval_loss": 1.842891812324524, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1410 + }, + { + "epoch": 1.4494093477144325, + "grad_norm": 2.8490853309631348, + "learning_rate": 2.9006471259992392e-06, + "loss": 1.4114, + "step": 1411 + }, + { + "epoch": 1.450436569080637, + "grad_norm": 2.7475955486297607, + "learning_rate": 2.8987438142367726e-06, + "loss": 1.4068, + "step": 1412 + }, + { + "epoch": 1.4514637904468413, + "grad_norm": 2.7209911346435547, + "learning_rate": 2.8968405024743055e-06, + "loss": 1.4658, + "step": 1413 + }, + { + "epoch": 1.4524910118130456, + "grad_norm": 2.578934907913208, + "learning_rate": 2.894937190711839e-06, + "loss": 1.3686, + "step": 1414 + }, + { + "epoch": 1.45351823317925, + "grad_norm": 2.722381114959717, + "learning_rate": 2.8930338789493723e-06, + "loss": 1.4337, + "step": 1415 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.702178478240967, + "learning_rate": 2.8911305671869056e-06, + "loss": 1.4163, + "step": 1416 + }, + { + "epoch": 1.455572675911659, + "grad_norm": 2.6742405891418457, + "learning_rate": 2.889227255424439e-06, + "loss": 1.3585, + "step": 1417 + }, + { + "epoch": 1.4565998972778633, + "grad_norm": 2.6732680797576904, + "learning_rate": 2.887323943661972e-06, + "loss": 1.406, + "step": 1418 + }, + { + "epoch": 1.457627118644068, + "grad_norm": 2.698864459991455, + "learning_rate": 2.8854206318995053e-06, + "loss": 1.3872, + "step": 1419 + }, + { + "epoch": 1.4586543400102723, + "grad_norm": 2.7534351348876953, + "learning_rate": 2.8835173201370386e-06, + "loss": 1.4055, + "step": 1420 + }, + { + "epoch": 1.4586543400102723, + "eval_loss": 1.8564038276672363, + "eval_runtime": 37.0708, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.4596815613764766, + "grad_norm": 2.9538612365722656, + "learning_rate": 2.881614008374572e-06, + "loss": 1.4596, + "step": 1421 + }, + { + "epoch": 1.460708782742681, + "grad_norm": 2.829641103744507, + "learning_rate": 2.8797106966121054e-06, + "loss": 1.4265, + "step": 1422 + }, + { + "epoch": 1.4617360041088854, + "grad_norm": 2.6814422607421875, + "learning_rate": 2.8778073848496383e-06, + "loss": 1.4112, + "step": 1423 + }, + { + "epoch": 1.46276322547509, + "grad_norm": 2.7880561351776123, + "learning_rate": 2.8759040730871717e-06, + "loss": 1.4435, + "step": 1424 + }, + { + "epoch": 1.4637904468412943, + "grad_norm": 2.676196336746216, + "learning_rate": 2.874000761324705e-06, + "loss": 1.4104, + "step": 1425 + }, + { + "epoch": 1.4648176682074987, + "grad_norm": 2.7417874336242676, + "learning_rate": 2.8720974495622384e-06, + "loss": 1.4335, + "step": 1426 + }, + { + "epoch": 1.465844889573703, + "grad_norm": 2.7870116233825684, + "learning_rate": 2.8701941377997718e-06, + "loss": 1.4226, + "step": 1427 + }, + { + "epoch": 1.4668721109399074, + "grad_norm": 2.765194892883301, + "learning_rate": 2.8682908260373047e-06, + "loss": 1.4582, + "step": 1428 + }, + { + "epoch": 1.467899332306112, + "grad_norm": 2.608175277709961, + "learning_rate": 2.866387514274838e-06, + "loss": 1.4285, + "step": 1429 + }, + { + "epoch": 1.4689265536723164, + "grad_norm": 2.7573461532592773, + "learning_rate": 2.864484202512372e-06, + "loss": 1.4395, + "step": 1430 + }, + { + "epoch": 1.4689265536723164, + "eval_loss": 1.8600319623947144, + "eval_runtime": 37.1192, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1430 + }, + { + "epoch": 1.4699537750385208, + "grad_norm": 2.635175943374634, + "learning_rate": 2.8625808907499052e-06, + "loss": 1.402, + "step": 1431 + }, + { + "epoch": 1.4709809964047253, + "grad_norm": 2.7318811416625977, + "learning_rate": 2.8606775789874386e-06, + "loss": 1.4352, + "step": 1432 + }, + { + "epoch": 1.4720082177709297, + "grad_norm": 2.783669948577881, + "learning_rate": 2.858774267224972e-06, + "loss": 1.4052, + "step": 1433 + }, + { + "epoch": 1.473035439137134, + "grad_norm": 2.7327511310577393, + "learning_rate": 2.8568709554625053e-06, + "loss": 1.4209, + "step": 1434 + }, + { + "epoch": 1.4740626605033385, + "grad_norm": 2.76542329788208, + "learning_rate": 2.8549676437000382e-06, + "loss": 1.4753, + "step": 1435 + }, + { + "epoch": 1.4750898818695428, + "grad_norm": 2.8031232357025146, + "learning_rate": 2.8530643319375716e-06, + "loss": 1.4399, + "step": 1436 + }, + { + "epoch": 1.4761171032357474, + "grad_norm": 2.737053155899048, + "learning_rate": 2.851161020175105e-06, + "loss": 1.4085, + "step": 1437 + }, + { + "epoch": 1.4771443246019518, + "grad_norm": 2.609987735748291, + "learning_rate": 2.8492577084126383e-06, + "loss": 1.3809, + "step": 1438 + }, + { + "epoch": 1.4781715459681561, + "grad_norm": 2.7482123374938965, + "learning_rate": 2.8473543966501717e-06, + "loss": 1.4234, + "step": 1439 + }, + { + "epoch": 1.4791987673343605, + "grad_norm": 2.80330753326416, + "learning_rate": 2.8454510848877046e-06, + "loss": 1.4665, + "step": 1440 + }, + { + "epoch": 1.4791987673343605, + "eval_loss": 1.8637592792510986, + "eval_runtime": 37.1066, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 1440 + }, + { + "epoch": 1.4802259887005649, + "grad_norm": 2.698328733444214, + "learning_rate": 2.843547773125238e-06, + "loss": 1.4224, + "step": 1441 + }, + { + "epoch": 1.4812532100667695, + "grad_norm": 2.663471221923828, + "learning_rate": 2.8416444613627714e-06, + "loss": 1.4205, + "step": 1442 + }, + { + "epoch": 1.4822804314329738, + "grad_norm": 2.7955918312072754, + "learning_rate": 2.8397411496003047e-06, + "loss": 1.3694, + "step": 1443 + }, + { + "epoch": 1.4833076527991782, + "grad_norm": 2.7320404052734375, + "learning_rate": 2.837837837837838e-06, + "loss": 1.4208, + "step": 1444 + }, + { + "epoch": 1.4843348741653826, + "grad_norm": 2.7786989212036133, + "learning_rate": 2.835934526075371e-06, + "loss": 1.4392, + "step": 1445 + }, + { + "epoch": 1.485362095531587, + "grad_norm": 2.6282734870910645, + "learning_rate": 2.8340312143129044e-06, + "loss": 1.3866, + "step": 1446 + }, + { + "epoch": 1.4863893168977915, + "grad_norm": 2.615072011947632, + "learning_rate": 2.8321279025504377e-06, + "loss": 1.3695, + "step": 1447 + }, + { + "epoch": 1.487416538263996, + "grad_norm": 2.884423017501831, + "learning_rate": 2.830224590787971e-06, + "loss": 1.4867, + "step": 1448 + }, + { + "epoch": 1.4884437596302003, + "grad_norm": 2.7745864391326904, + "learning_rate": 2.8283212790255045e-06, + "loss": 1.42, + "step": 1449 + }, + { + "epoch": 1.4894709809964048, + "grad_norm": 2.718130111694336, + "learning_rate": 2.8264179672630383e-06, + "loss": 1.4665, + "step": 1450 + }, + { + "epoch": 1.4894709809964048, + "eval_loss": 1.8556917905807495, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1450 + }, + { + "epoch": 1.4904982023626092, + "grad_norm": 2.7502355575561523, + "learning_rate": 2.8245146555005716e-06, + "loss": 1.3946, + "step": 1451 + }, + { + "epoch": 1.4915254237288136, + "grad_norm": 2.635500192642212, + "learning_rate": 2.8226113437381046e-06, + "loss": 1.3612, + "step": 1452 + }, + { + "epoch": 1.492552645095018, + "grad_norm": 2.617408275604248, + "learning_rate": 2.820708031975638e-06, + "loss": 1.4146, + "step": 1453 + }, + { + "epoch": 1.4935798664612223, + "grad_norm": 2.7706692218780518, + "learning_rate": 2.8188047202131713e-06, + "loss": 1.4474, + "step": 1454 + }, + { + "epoch": 1.494607087827427, + "grad_norm": 2.7609055042266846, + "learning_rate": 2.8169014084507046e-06, + "loss": 1.4618, + "step": 1455 + }, + { + "epoch": 1.4956343091936313, + "grad_norm": 2.757258892059326, + "learning_rate": 2.814998096688238e-06, + "loss": 1.4131, + "step": 1456 + }, + { + "epoch": 1.4966615305598356, + "grad_norm": 2.5547049045562744, + "learning_rate": 2.813094784925771e-06, + "loss": 1.3692, + "step": 1457 + }, + { + "epoch": 1.49768875192604, + "grad_norm": 2.817617177963257, + "learning_rate": 2.8111914731633043e-06, + "loss": 1.4664, + "step": 1458 + }, + { + "epoch": 1.4987159732922444, + "grad_norm": 2.6999337673187256, + "learning_rate": 2.8092881614008377e-06, + "loss": 1.4126, + "step": 1459 + }, + { + "epoch": 1.499743194658449, + "grad_norm": 2.571727991104126, + "learning_rate": 2.807384849638371e-06, + "loss": 1.3828, + "step": 1460 + }, + { + "epoch": 1.499743194658449, + "eval_loss": 1.8413318395614624, + "eval_runtime": 37.1803, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1460 + }, + { + "epoch": 1.5007704160246533, + "grad_norm": 2.615567684173584, + "learning_rate": 2.8054815378759044e-06, + "loss": 1.4336, + "step": 1461 + }, + { + "epoch": 1.5017976373908577, + "grad_norm": 2.696854591369629, + "learning_rate": 2.8035782261134378e-06, + "loss": 1.4015, + "step": 1462 + }, + { + "epoch": 1.5028248587570623, + "grad_norm": 2.7395496368408203, + "learning_rate": 2.8016749143509707e-06, + "loss": 1.4009, + "step": 1463 + }, + { + "epoch": 1.5038520801232664, + "grad_norm": 2.8281755447387695, + "learning_rate": 2.799771602588504e-06, + "loss": 1.4069, + "step": 1464 + }, + { + "epoch": 1.504879301489471, + "grad_norm": 2.641793727874756, + "learning_rate": 2.7978682908260374e-06, + "loss": 1.4341, + "step": 1465 + }, + { + "epoch": 1.5059065228556754, + "grad_norm": 2.8969004154205322, + "learning_rate": 2.795964979063571e-06, + "loss": 1.4493, + "step": 1466 + }, + { + "epoch": 1.5069337442218798, + "grad_norm": 2.758255958557129, + "learning_rate": 2.794061667301104e-06, + "loss": 1.423, + "step": 1467 + }, + { + "epoch": 1.5079609655880843, + "grad_norm": 2.7737441062927246, + "learning_rate": 2.792158355538637e-06, + "loss": 1.43, + "step": 1468 + }, + { + "epoch": 1.5089881869542885, + "grad_norm": 2.798095226287842, + "learning_rate": 2.7902550437761705e-06, + "loss": 1.4276, + "step": 1469 + }, + { + "epoch": 1.510015408320493, + "grad_norm": 2.760437488555908, + "learning_rate": 2.7883517320137042e-06, + "loss": 1.3927, + "step": 1470 + }, + { + "epoch": 1.510015408320493, + "eval_loss": 1.8517959117889404, + "eval_runtime": 37.163, + "eval_samples_per_second": 26.908, + "eval_steps_per_second": 1.13, + "step": 1470 + }, + { + "epoch": 1.5110426296866974, + "grad_norm": 2.689023733139038, + "learning_rate": 2.7864484202512376e-06, + "loss": 1.387, + "step": 1471 + }, + { + "epoch": 1.5120698510529018, + "grad_norm": 2.876424551010132, + "learning_rate": 2.784545108488771e-06, + "loss": 1.4183, + "step": 1472 + }, + { + "epoch": 1.5130970724191064, + "grad_norm": 2.789368152618408, + "learning_rate": 2.7826417967263043e-06, + "loss": 1.4239, + "step": 1473 + }, + { + "epoch": 1.5141242937853108, + "grad_norm": 2.945103883743286, + "learning_rate": 2.7807384849638373e-06, + "loss": 1.458, + "step": 1474 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.694589614868164, + "learning_rate": 2.7788351732013706e-06, + "loss": 1.4145, + "step": 1475 + }, + { + "epoch": 1.5161787365177197, + "grad_norm": 2.7276177406311035, + "learning_rate": 2.776931861438904e-06, + "loss": 1.4063, + "step": 1476 + }, + { + "epoch": 1.5172059578839239, + "grad_norm": 2.694197177886963, + "learning_rate": 2.7750285496764374e-06, + "loss": 1.4125, + "step": 1477 + }, + { + "epoch": 1.5182331792501285, + "grad_norm": 2.832883834838867, + "learning_rate": 2.7731252379139707e-06, + "loss": 1.4176, + "step": 1478 + }, + { + "epoch": 1.5192604006163328, + "grad_norm": 2.67136549949646, + "learning_rate": 2.771221926151504e-06, + "loss": 1.3633, + "step": 1479 + }, + { + "epoch": 1.5202876219825372, + "grad_norm": 2.9618499279022217, + "learning_rate": 2.769318614389037e-06, + "loss": 1.4283, + "step": 1480 + }, + { + "epoch": 1.5202876219825372, + "eval_loss": 1.8486552238464355, + "eval_runtime": 37.1808, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1480 + }, + { + "epoch": 1.5213148433487418, + "grad_norm": 2.91403865814209, + "learning_rate": 2.7674153026265704e-06, + "loss": 1.4254, + "step": 1481 + }, + { + "epoch": 1.522342064714946, + "grad_norm": 2.9121804237365723, + "learning_rate": 2.7655119908641037e-06, + "loss": 1.4251, + "step": 1482 + }, + { + "epoch": 1.5233692860811505, + "grad_norm": 2.7484965324401855, + "learning_rate": 2.763608679101637e-06, + "loss": 1.4141, + "step": 1483 + }, + { + "epoch": 1.5243965074473549, + "grad_norm": 2.913203716278076, + "learning_rate": 2.7617053673391705e-06, + "loss": 1.4516, + "step": 1484 + }, + { + "epoch": 1.5254237288135593, + "grad_norm": 2.7844395637512207, + "learning_rate": 2.7598020555767034e-06, + "loss": 1.45, + "step": 1485 + }, + { + "epoch": 1.5264509501797638, + "grad_norm": 2.8250396251678467, + "learning_rate": 2.7578987438142368e-06, + "loss": 1.3746, + "step": 1486 + }, + { + "epoch": 1.5274781715459682, + "grad_norm": 2.749629497528076, + "learning_rate": 2.75599543205177e-06, + "loss": 1.433, + "step": 1487 + }, + { + "epoch": 1.5285053929121726, + "grad_norm": 2.7559521198272705, + "learning_rate": 2.7540921202893035e-06, + "loss": 1.452, + "step": 1488 + }, + { + "epoch": 1.529532614278377, + "grad_norm": 2.825148344039917, + "learning_rate": 2.752188808526837e-06, + "loss": 1.427, + "step": 1489 + }, + { + "epoch": 1.5305598356445813, + "grad_norm": 2.645327091217041, + "learning_rate": 2.75028549676437e-06, + "loss": 1.4089, + "step": 1490 + }, + { + "epoch": 1.5305598356445813, + "eval_loss": 1.8497828245162964, + "eval_runtime": 37.3756, + "eval_samples_per_second": 26.755, + "eval_steps_per_second": 1.124, + "step": 1490 + }, + { + "epoch": 1.531587057010786, + "grad_norm": 2.753156900405884, + "learning_rate": 2.7483821850019036e-06, + "loss": 1.4163, + "step": 1491 + }, + { + "epoch": 1.5326142783769903, + "grad_norm": 2.8842933177948, + "learning_rate": 2.746478873239437e-06, + "loss": 1.4041, + "step": 1492 + }, + { + "epoch": 1.5336414997431946, + "grad_norm": 2.7529265880584717, + "learning_rate": 2.7445755614769703e-06, + "loss": 1.41, + "step": 1493 + }, + { + "epoch": 1.5346687211093992, + "grad_norm": 2.6907176971435547, + "learning_rate": 2.7426722497145037e-06, + "loss": 1.3841, + "step": 1494 + }, + { + "epoch": 1.5356959424756034, + "grad_norm": 2.793073892593384, + "learning_rate": 2.740768937952037e-06, + "loss": 1.3885, + "step": 1495 + }, + { + "epoch": 1.536723163841808, + "grad_norm": 2.7895116806030273, + "learning_rate": 2.7388656261895704e-06, + "loss": 1.4065, + "step": 1496 + }, + { + "epoch": 1.5377503852080123, + "grad_norm": 2.903615713119507, + "learning_rate": 2.7369623144271033e-06, + "loss": 1.4812, + "step": 1497 + }, + { + "epoch": 1.5387776065742167, + "grad_norm": 2.7081339359283447, + "learning_rate": 2.7350590026646367e-06, + "loss": 1.403, + "step": 1498 + }, + { + "epoch": 1.5398048279404213, + "grad_norm": 2.75414776802063, + "learning_rate": 2.73315569090217e-06, + "loss": 1.3629, + "step": 1499 + }, + { + "epoch": 1.5408320493066254, + "grad_norm": 2.795982599258423, + "learning_rate": 2.7312523791397034e-06, + "loss": 1.4551, + "step": 1500 + }, + { + "epoch": 1.5408320493066254, + "eval_loss": 1.8510199785232544, + "eval_runtime": 37.3806, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 1500 + }, + { + "epoch": 1.54185927067283, + "grad_norm": 2.894488573074341, + "learning_rate": 2.7293490673772368e-06, + "loss": 1.4794, + "step": 1501 + }, + { + "epoch": 1.5428864920390344, + "grad_norm": 2.7982053756713867, + "learning_rate": 2.7274457556147697e-06, + "loss": 1.4365, + "step": 1502 + }, + { + "epoch": 1.5439137134052388, + "grad_norm": 2.913367509841919, + "learning_rate": 2.725542443852303e-06, + "loss": 1.4132, + "step": 1503 + }, + { + "epoch": 1.5449409347714433, + "grad_norm": 2.6658358573913574, + "learning_rate": 2.7236391320898365e-06, + "loss": 1.4288, + "step": 1504 + }, + { + "epoch": 1.5459681561376477, + "grad_norm": 2.710700035095215, + "learning_rate": 2.72173582032737e-06, + "loss": 1.3936, + "step": 1505 + }, + { + "epoch": 1.546995377503852, + "grad_norm": 2.7881481647491455, + "learning_rate": 2.719832508564903e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.5480225988700564, + "grad_norm": 2.7621042728424072, + "learning_rate": 2.717929196802436e-06, + "loss": 1.4218, + "step": 1507 + }, + { + "epoch": 1.5490498202362608, + "grad_norm": 2.7504117488861084, + "learning_rate": 2.7160258850399695e-06, + "loss": 1.3918, + "step": 1508 + }, + { + "epoch": 1.5500770416024654, + "grad_norm": 2.719702959060669, + "learning_rate": 2.714122573277503e-06, + "loss": 1.4154, + "step": 1509 + }, + { + "epoch": 1.5511042629686698, + "grad_norm": 2.6698062419891357, + "learning_rate": 2.712219261515036e-06, + "loss": 1.4084, + "step": 1510 + }, + { + "epoch": 1.5511042629686698, + "eval_loss": 1.8468451499938965, + "eval_runtime": 36.9179, + "eval_samples_per_second": 27.087, + "eval_steps_per_second": 1.138, + "step": 1510 + }, + { + "epoch": 1.5521314843348741, + "grad_norm": 2.816978693008423, + "learning_rate": 2.71031594975257e-06, + "loss": 1.4195, + "step": 1511 + }, + { + "epoch": 1.5531587057010787, + "grad_norm": 2.8047056198120117, + "learning_rate": 2.7084126379901034e-06, + "loss": 1.4395, + "step": 1512 + }, + { + "epoch": 1.5541859270672829, + "grad_norm": 2.8461363315582275, + "learning_rate": 2.7065093262276367e-06, + "loss": 1.4472, + "step": 1513 + }, + { + "epoch": 1.5552131484334875, + "grad_norm": 2.712890863418579, + "learning_rate": 2.7046060144651697e-06, + "loss": 1.4538, + "step": 1514 + }, + { + "epoch": 1.5562403697996918, + "grad_norm": 2.844165802001953, + "learning_rate": 2.702702702702703e-06, + "loss": 1.4719, + "step": 1515 + }, + { + "epoch": 1.5572675911658962, + "grad_norm": 2.629138231277466, + "learning_rate": 2.7007993909402364e-06, + "loss": 1.3441, + "step": 1516 + }, + { + "epoch": 1.5582948125321008, + "grad_norm": 2.7072019577026367, + "learning_rate": 2.6988960791777697e-06, + "loss": 1.4692, + "step": 1517 + }, + { + "epoch": 1.559322033898305, + "grad_norm": 2.739567279815674, + "learning_rate": 2.696992767415303e-06, + "loss": 1.4215, + "step": 1518 + }, + { + "epoch": 1.5603492552645095, + "grad_norm": 2.7846791744232178, + "learning_rate": 2.695089455652836e-06, + "loss": 1.4151, + "step": 1519 + }, + { + "epoch": 1.5613764766307139, + "grad_norm": 2.7417876720428467, + "learning_rate": 2.6931861438903694e-06, + "loss": 1.3797, + "step": 1520 + }, + { + "epoch": 1.5613764766307139, + "eval_loss": 1.847011923789978, + "eval_runtime": 37.0386, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1520 + }, + { + "epoch": 1.5624036979969183, + "grad_norm": 2.743271827697754, + "learning_rate": 2.6912828321279028e-06, + "loss": 1.4372, + "step": 1521 + }, + { + "epoch": 1.5634309193631228, + "grad_norm": 2.8119285106658936, + "learning_rate": 2.689379520365436e-06, + "loss": 1.4075, + "step": 1522 + }, + { + "epoch": 1.5644581407293272, + "grad_norm": 2.7443952560424805, + "learning_rate": 2.6874762086029695e-06, + "loss": 1.4442, + "step": 1523 + }, + { + "epoch": 1.5654853620955316, + "grad_norm": 2.7869346141815186, + "learning_rate": 2.6855728968405024e-06, + "loss": 1.3948, + "step": 1524 + }, + { + "epoch": 1.5665125834617362, + "grad_norm": 2.750185251235962, + "learning_rate": 2.683669585078036e-06, + "loss": 1.4313, + "step": 1525 + }, + { + "epoch": 1.5675398048279403, + "grad_norm": 2.787389039993286, + "learning_rate": 2.681766273315569e-06, + "loss": 1.3748, + "step": 1526 + }, + { + "epoch": 1.568567026194145, + "grad_norm": 2.6675543785095215, + "learning_rate": 2.6798629615531025e-06, + "loss": 1.4089, + "step": 1527 + }, + { + "epoch": 1.5695942475603493, + "grad_norm": 2.6468663215637207, + "learning_rate": 2.677959649790636e-06, + "loss": 1.4148, + "step": 1528 + }, + { + "epoch": 1.5706214689265536, + "grad_norm": 2.7241196632385254, + "learning_rate": 2.676056338028169e-06, + "loss": 1.4498, + "step": 1529 + }, + { + "epoch": 1.5716486902927582, + "grad_norm": 2.705486536026001, + "learning_rate": 2.674153026265702e-06, + "loss": 1.4324, + "step": 1530 + }, + { + "epoch": 1.5716486902927582, + "eval_loss": 1.8467438220977783, + "eval_runtime": 37.0576, + "eval_samples_per_second": 26.985, + "eval_steps_per_second": 1.133, + "step": 1530 + }, + { + "epoch": 1.5726759116589624, + "grad_norm": 2.805429458618164, + "learning_rate": 2.672249714503236e-06, + "loss": 1.4475, + "step": 1531 + }, + { + "epoch": 1.573703133025167, + "grad_norm": 2.7074034214019775, + "learning_rate": 2.6703464027407693e-06, + "loss": 1.3997, + "step": 1532 + }, + { + "epoch": 1.5747303543913713, + "grad_norm": 2.737571954727173, + "learning_rate": 2.6684430909783027e-06, + "loss": 1.4238, + "step": 1533 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.703036069869995, + "learning_rate": 2.666539779215836e-06, + "loss": 1.4107, + "step": 1534 + }, + { + "epoch": 1.5767847971237803, + "grad_norm": 2.7987451553344727, + "learning_rate": 2.6646364674533694e-06, + "loss": 1.3927, + "step": 1535 + }, + { + "epoch": 1.5778120184899846, + "grad_norm": 2.622497081756592, + "learning_rate": 2.6627331556909024e-06, + "loss": 1.3215, + "step": 1536 + }, + { + "epoch": 1.578839239856189, + "grad_norm": 2.5890607833862305, + "learning_rate": 2.6608298439284357e-06, + "loss": 1.3994, + "step": 1537 + }, + { + "epoch": 1.5798664612223934, + "grad_norm": 2.6438405513763428, + "learning_rate": 2.658926532165969e-06, + "loss": 1.4042, + "step": 1538 + }, + { + "epoch": 1.5808936825885977, + "grad_norm": 2.690793752670288, + "learning_rate": 2.6570232204035024e-06, + "loss": 1.3673, + "step": 1539 + }, + { + "epoch": 1.5819209039548023, + "grad_norm": 2.7485456466674805, + "learning_rate": 2.655119908641036e-06, + "loss": 1.3979, + "step": 1540 + }, + { + "epoch": 1.5819209039548023, + "eval_loss": 1.852581262588501, + "eval_runtime": 37.0688, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1540 + }, + { + "epoch": 1.5829481253210067, + "grad_norm": 2.8531415462493896, + "learning_rate": 2.6532165968785687e-06, + "loss": 1.4094, + "step": 1541 + }, + { + "epoch": 1.583975346687211, + "grad_norm": 2.638291358947754, + "learning_rate": 2.651313285116102e-06, + "loss": 1.3708, + "step": 1542 + }, + { + "epoch": 1.5850025680534157, + "grad_norm": 2.717550754547119, + "learning_rate": 2.6494099733536355e-06, + "loss": 1.3821, + "step": 1543 + }, + { + "epoch": 1.5860297894196198, + "grad_norm": 2.91959810256958, + "learning_rate": 2.647506661591169e-06, + "loss": 1.5532, + "step": 1544 + }, + { + "epoch": 1.5870570107858244, + "grad_norm": 2.7281882762908936, + "learning_rate": 2.645603349828702e-06, + "loss": 1.4361, + "step": 1545 + }, + { + "epoch": 1.5880842321520288, + "grad_norm": 2.732283592224121, + "learning_rate": 2.643700038066235e-06, + "loss": 1.3669, + "step": 1546 + }, + { + "epoch": 1.5891114535182331, + "grad_norm": 2.731055498123169, + "learning_rate": 2.6417967263037685e-06, + "loss": 1.4223, + "step": 1547 + }, + { + "epoch": 1.5901386748844377, + "grad_norm": 2.831334352493286, + "learning_rate": 2.639893414541302e-06, + "loss": 1.4454, + "step": 1548 + }, + { + "epoch": 1.5911658962506419, + "grad_norm": 2.920548677444458, + "learning_rate": 2.6379901027788352e-06, + "loss": 1.4686, + "step": 1549 + }, + { + "epoch": 1.5921931176168465, + "grad_norm": 2.6671719551086426, + "learning_rate": 2.6360867910163686e-06, + "loss": 1.398, + "step": 1550 + }, + { + "epoch": 1.5921931176168465, + "eval_loss": 1.857794165611267, + "eval_runtime": 37.0817, + "eval_samples_per_second": 26.967, + "eval_steps_per_second": 1.133, + "step": 1550 + }, + { + "epoch": 1.5932203389830508, + "grad_norm": 2.685175657272339, + "learning_rate": 2.6341834792539024e-06, + "loss": 1.4176, + "step": 1551 + }, + { + "epoch": 1.5942475603492552, + "grad_norm": 2.7412822246551514, + "learning_rate": 2.6322801674914357e-06, + "loss": 1.4195, + "step": 1552 + }, + { + "epoch": 1.5952747817154598, + "grad_norm": 2.7461724281311035, + "learning_rate": 2.6303768557289687e-06, + "loss": 1.4081, + "step": 1553 + }, + { + "epoch": 1.5963020030816641, + "grad_norm": 2.6494853496551514, + "learning_rate": 2.628473543966502e-06, + "loss": 1.3759, + "step": 1554 + }, + { + "epoch": 1.5973292244478685, + "grad_norm": 2.736752986907959, + "learning_rate": 2.6265702322040354e-06, + "loss": 1.4415, + "step": 1555 + }, + { + "epoch": 1.5983564458140729, + "grad_norm": 2.733452796936035, + "learning_rate": 2.6246669204415688e-06, + "loss": 1.3867, + "step": 1556 + }, + { + "epoch": 1.5993836671802772, + "grad_norm": 2.692986249923706, + "learning_rate": 2.622763608679102e-06, + "loss": 1.3811, + "step": 1557 + }, + { + "epoch": 1.6004108885464818, + "grad_norm": 2.580052614212036, + "learning_rate": 2.620860296916635e-06, + "loss": 1.399, + "step": 1558 + }, + { + "epoch": 1.6014381099126862, + "grad_norm": 2.651503086090088, + "learning_rate": 2.6189569851541684e-06, + "loss": 1.34, + "step": 1559 + }, + { + "epoch": 1.6024653312788906, + "grad_norm": 2.8157193660736084, + "learning_rate": 2.617053673391702e-06, + "loss": 1.4437, + "step": 1560 + }, + { + "epoch": 1.6024653312788906, + "eval_loss": 1.8588496446609497, + "eval_runtime": 37.0786, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 1560 + }, + { + "epoch": 1.6034925526450952, + "grad_norm": 2.893589973449707, + "learning_rate": 2.615150361629235e-06, + "loss": 1.4775, + "step": 1561 + }, + { + "epoch": 1.6045197740112993, + "grad_norm": 2.719688653945923, + "learning_rate": 2.6132470498667685e-06, + "loss": 1.432, + "step": 1562 + }, + { + "epoch": 1.605546995377504, + "grad_norm": 2.5626275539398193, + "learning_rate": 2.6113437381043015e-06, + "loss": 1.4118, + "step": 1563 + }, + { + "epoch": 1.6065742167437083, + "grad_norm": 2.6879632472991943, + "learning_rate": 2.609440426341835e-06, + "loss": 1.398, + "step": 1564 + }, + { + "epoch": 1.6076014381099126, + "grad_norm": 2.890155792236328, + "learning_rate": 2.607537114579368e-06, + "loss": 1.4336, + "step": 1565 + }, + { + "epoch": 1.6086286594761172, + "grad_norm": 2.8333349227905273, + "learning_rate": 2.6056338028169015e-06, + "loss": 1.4148, + "step": 1566 + }, + { + "epoch": 1.6096558808423214, + "grad_norm": 2.729215383529663, + "learning_rate": 2.603730491054435e-06, + "loss": 1.3831, + "step": 1567 + }, + { + "epoch": 1.610683102208526, + "grad_norm": 2.800276279449463, + "learning_rate": 2.601827179291968e-06, + "loss": 1.4315, + "step": 1568 + }, + { + "epoch": 1.6117103235747303, + "grad_norm": 2.7900125980377197, + "learning_rate": 2.599923867529501e-06, + "loss": 1.4257, + "step": 1569 + }, + { + "epoch": 1.6127375449409347, + "grad_norm": 2.7753336429595947, + "learning_rate": 2.5980205557670346e-06, + "loss": 1.4089, + "step": 1570 + }, + { + "epoch": 1.6127375449409347, + "eval_loss": 1.8533762693405151, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1570 + }, + { + "epoch": 1.6137647663071393, + "grad_norm": 2.9164156913757324, + "learning_rate": 2.5961172440045684e-06, + "loss": 1.4467, + "step": 1571 + }, + { + "epoch": 1.6147919876733436, + "grad_norm": 2.747676134109497, + "learning_rate": 2.5942139322421017e-06, + "loss": 1.4521, + "step": 1572 + }, + { + "epoch": 1.615819209039548, + "grad_norm": 2.6693432331085205, + "learning_rate": 2.592310620479635e-06, + "loss": 1.3982, + "step": 1573 + }, + { + "epoch": 1.6168464304057526, + "grad_norm": 2.5721216201782227, + "learning_rate": 2.5904073087171684e-06, + "loss": 1.3657, + "step": 1574 + }, + { + "epoch": 1.6178736517719567, + "grad_norm": 2.706143617630005, + "learning_rate": 2.5885039969547014e-06, + "loss": 1.4315, + "step": 1575 + }, + { + "epoch": 1.6189008731381613, + "grad_norm": 2.59006404876709, + "learning_rate": 2.5866006851922347e-06, + "loss": 1.372, + "step": 1576 + }, + { + "epoch": 1.6199280945043657, + "grad_norm": 2.7117512226104736, + "learning_rate": 2.584697373429768e-06, + "loss": 1.3834, + "step": 1577 + }, + { + "epoch": 1.62095531587057, + "grad_norm": 2.718029737472534, + "learning_rate": 2.5827940616673015e-06, + "loss": 1.3978, + "step": 1578 + }, + { + "epoch": 1.6219825372367747, + "grad_norm": 2.7660539150238037, + "learning_rate": 2.580890749904835e-06, + "loss": 1.4505, + "step": 1579 + }, + { + "epoch": 1.6230097586029788, + "grad_norm": 2.7671236991882324, + "learning_rate": 2.5789874381423678e-06, + "loss": 1.3951, + "step": 1580 + }, + { + "epoch": 1.6230097586029788, + "eval_loss": 1.8412485122680664, + "eval_runtime": 37.1162, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.6240369799691834, + "grad_norm": 2.6504807472229004, + "learning_rate": 2.577084126379901e-06, + "loss": 1.3957, + "step": 1581 + }, + { + "epoch": 1.6250642013353878, + "grad_norm": 2.688166618347168, + "learning_rate": 2.5751808146174345e-06, + "loss": 1.3686, + "step": 1582 + }, + { + "epoch": 1.6260914227015921, + "grad_norm": 2.7034637928009033, + "learning_rate": 2.573277502854968e-06, + "loss": 1.4201, + "step": 1583 + }, + { + "epoch": 1.6271186440677967, + "grad_norm": 2.6097683906555176, + "learning_rate": 2.5713741910925012e-06, + "loss": 1.3881, + "step": 1584 + }, + { + "epoch": 1.6281458654340009, + "grad_norm": 2.6197433471679688, + "learning_rate": 2.569470879330034e-06, + "loss": 1.3962, + "step": 1585 + }, + { + "epoch": 1.6291730868002054, + "grad_norm": 3.0436367988586426, + "learning_rate": 2.5675675675675675e-06, + "loss": 1.4714, + "step": 1586 + }, + { + "epoch": 1.6302003081664098, + "grad_norm": 2.708744764328003, + "learning_rate": 2.565664255805101e-06, + "loss": 1.4007, + "step": 1587 + }, + { + "epoch": 1.6312275295326142, + "grad_norm": 2.708279848098755, + "learning_rate": 2.5637609440426343e-06, + "loss": 1.4284, + "step": 1588 + }, + { + "epoch": 1.6322547508988188, + "grad_norm": 2.7955429553985596, + "learning_rate": 2.5618576322801676e-06, + "loss": 1.3736, + "step": 1589 + }, + { + "epoch": 1.6332819722650231, + "grad_norm": 2.890211820602417, + "learning_rate": 2.5599543205177006e-06, + "loss": 1.4671, + "step": 1590 + }, + { + "epoch": 1.6332819722650231, + "eval_loss": 1.8467649221420288, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 1590 + }, + { + "epoch": 1.6343091936312275, + "grad_norm": 2.698066234588623, + "learning_rate": 2.558051008755234e-06, + "loss": 1.4229, + "step": 1591 + }, + { + "epoch": 1.635336414997432, + "grad_norm": 2.8393352031707764, + "learning_rate": 2.5561476969927677e-06, + "loss": 1.469, + "step": 1592 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.589914321899414, + "learning_rate": 2.554244385230301e-06, + "loss": 1.3853, + "step": 1593 + }, + { + "epoch": 1.6373908577298408, + "grad_norm": 2.630443572998047, + "learning_rate": 2.5523410734678344e-06, + "loss": 1.4176, + "step": 1594 + }, + { + "epoch": 1.6384180790960452, + "grad_norm": 2.784212112426758, + "learning_rate": 2.550437761705368e-06, + "loss": 1.344, + "step": 1595 + }, + { + "epoch": 1.6394453004622496, + "grad_norm": 2.736009359359741, + "learning_rate": 2.548534449942901e-06, + "loss": 1.3867, + "step": 1596 + }, + { + "epoch": 1.6404725218284542, + "grad_norm": 2.9134042263031006, + "learning_rate": 2.546631138180434e-06, + "loss": 1.4214, + "step": 1597 + }, + { + "epoch": 1.6414997431946583, + "grad_norm": 2.995575428009033, + "learning_rate": 2.5447278264179675e-06, + "loss": 1.409, + "step": 1598 + }, + { + "epoch": 1.642526964560863, + "grad_norm": 2.744499683380127, + "learning_rate": 2.542824514655501e-06, + "loss": 1.3727, + "step": 1599 + }, + { + "epoch": 1.6435541859270673, + "grad_norm": 2.835494041442871, + "learning_rate": 2.540921202893034e-06, + "loss": 1.4636, + "step": 1600 + }, + { + "epoch": 1.6435541859270673, + "eval_loss": 1.8454105854034424, + "eval_runtime": 37.4381, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1600 + }, + { + "epoch": 1.6445814072932716, + "grad_norm": 2.817537307739258, + "learning_rate": 2.5390178911305675e-06, + "loss": 1.4416, + "step": 1601 + }, + { + "epoch": 1.6456086286594762, + "grad_norm": 2.868093252182007, + "learning_rate": 2.5371145793681005e-06, + "loss": 1.4187, + "step": 1602 + }, + { + "epoch": 1.6466358500256806, + "grad_norm": 2.861865758895874, + "learning_rate": 2.535211267605634e-06, + "loss": 1.4067, + "step": 1603 + }, + { + "epoch": 1.647663071391885, + "grad_norm": 2.7015087604522705, + "learning_rate": 2.533307955843167e-06, + "loss": 1.3528, + "step": 1604 + }, + { + "epoch": 1.6486902927580893, + "grad_norm": 2.7585439682006836, + "learning_rate": 2.5314046440807006e-06, + "loss": 1.4157, + "step": 1605 + }, + { + "epoch": 1.6497175141242937, + "grad_norm": 2.853614330291748, + "learning_rate": 2.529501332318234e-06, + "loss": 1.3932, + "step": 1606 + }, + { + "epoch": 1.6507447354904983, + "grad_norm": 2.9736239910125732, + "learning_rate": 2.5275980205557673e-06, + "loss": 1.4511, + "step": 1607 + }, + { + "epoch": 1.6517719568567026, + "grad_norm": 2.955389976501465, + "learning_rate": 2.5256947087933002e-06, + "loss": 1.4777, + "step": 1608 + }, + { + "epoch": 1.652799178222907, + "grad_norm": 2.9508774280548096, + "learning_rate": 2.5237913970308336e-06, + "loss": 1.4523, + "step": 1609 + }, + { + "epoch": 1.6538263995891116, + "grad_norm": 2.88675856590271, + "learning_rate": 2.521888085268367e-06, + "loss": 1.4064, + "step": 1610 + }, + { + "epoch": 1.6538263995891116, + "eval_loss": 1.8569127321243286, + "eval_runtime": 36.9819, + "eval_samples_per_second": 27.04, + "eval_steps_per_second": 1.136, + "step": 1610 + }, + { + "epoch": 1.6548536209553157, + "grad_norm": 2.8439440727233887, + "learning_rate": 2.5199847735059003e-06, + "loss": 1.4591, + "step": 1611 + }, + { + "epoch": 1.6558808423215203, + "grad_norm": 2.9314420223236084, + "learning_rate": 2.518081461743434e-06, + "loss": 1.39, + "step": 1612 + }, + { + "epoch": 1.6569080636877247, + "grad_norm": 2.6517245769500732, + "learning_rate": 2.5161781499809675e-06, + "loss": 1.4234, + "step": 1613 + }, + { + "epoch": 1.657935285053929, + "grad_norm": 2.5690908432006836, + "learning_rate": 2.5142748382185004e-06, + "loss": 1.3762, + "step": 1614 + }, + { + "epoch": 1.6589625064201337, + "grad_norm": 2.808892011642456, + "learning_rate": 2.5123715264560338e-06, + "loss": 1.3727, + "step": 1615 + }, + { + "epoch": 1.6599897277863378, + "grad_norm": 2.777846336364746, + "learning_rate": 2.510468214693567e-06, + "loss": 1.3941, + "step": 1616 + }, + { + "epoch": 1.6610169491525424, + "grad_norm": 2.8298423290252686, + "learning_rate": 2.5085649029311005e-06, + "loss": 1.3455, + "step": 1617 + }, + { + "epoch": 1.6620441705187468, + "grad_norm": 2.6464784145355225, + "learning_rate": 2.506661591168634e-06, + "loss": 1.3061, + "step": 1618 + }, + { + "epoch": 1.6630713918849511, + "grad_norm": 2.7149951457977295, + "learning_rate": 2.504758279406167e-06, + "loss": 1.3995, + "step": 1619 + }, + { + "epoch": 1.6640986132511557, + "grad_norm": 3.0328569412231445, + "learning_rate": 2.5028549676437e-06, + "loss": 1.4343, + "step": 1620 + }, + { + "epoch": 1.6640986132511557, + "eval_loss": 1.8629075288772583, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1620 + }, + { + "epoch": 1.66512583461736, + "grad_norm": 2.8838441371917725, + "learning_rate": 2.5009516558812335e-06, + "loss": 1.3817, + "step": 1621 + }, + { + "epoch": 1.6661530559835644, + "grad_norm": 2.843242645263672, + "learning_rate": 2.499048344118767e-06, + "loss": 1.4322, + "step": 1622 + }, + { + "epoch": 1.667180277349769, + "grad_norm": 2.642946720123291, + "learning_rate": 2.4971450323563002e-06, + "loss": 1.4702, + "step": 1623 + }, + { + "epoch": 1.6682074987159732, + "grad_norm": 2.666194200515747, + "learning_rate": 2.4952417205938336e-06, + "loss": 1.3808, + "step": 1624 + }, + { + "epoch": 1.6692347200821778, + "grad_norm": 2.7515366077423096, + "learning_rate": 2.4933384088313666e-06, + "loss": 1.3892, + "step": 1625 + }, + { + "epoch": 1.6702619414483821, + "grad_norm": 2.7051661014556885, + "learning_rate": 2.4914350970689e-06, + "loss": 1.3809, + "step": 1626 + }, + { + "epoch": 1.6712891628145865, + "grad_norm": 2.7311975955963135, + "learning_rate": 2.4895317853064337e-06, + "loss": 1.3952, + "step": 1627 + }, + { + "epoch": 1.672316384180791, + "grad_norm": 2.8365893363952637, + "learning_rate": 2.4876284735439666e-06, + "loss": 1.4505, + "step": 1628 + }, + { + "epoch": 1.6733436055469952, + "grad_norm": 2.644507884979248, + "learning_rate": 2.4857251617815e-06, + "loss": 1.4114, + "step": 1629 + }, + { + "epoch": 1.6743708269131998, + "grad_norm": 2.7952826023101807, + "learning_rate": 2.4838218500190334e-06, + "loss": 1.4223, + "step": 1630 + }, + { + "epoch": 1.6743708269131998, + "eval_loss": 1.8577097654342651, + "eval_runtime": 37.1645, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 1630 + }, + { + "epoch": 1.6753980482794042, + "grad_norm": 2.787752866744995, + "learning_rate": 2.4819185382565667e-06, + "loss": 1.4022, + "step": 1631 + }, + { + "epoch": 1.6764252696456086, + "grad_norm": 2.6963443756103516, + "learning_rate": 2.4800152264941e-06, + "loss": 1.3974, + "step": 1632 + }, + { + "epoch": 1.6774524910118132, + "grad_norm": 2.6753835678100586, + "learning_rate": 2.478111914731633e-06, + "loss": 1.3639, + "step": 1633 + }, + { + "epoch": 1.6784797123780173, + "grad_norm": 2.7086808681488037, + "learning_rate": 2.4762086029691664e-06, + "loss": 1.3987, + "step": 1634 + }, + { + "epoch": 1.6795069337442219, + "grad_norm": 3.2228872776031494, + "learning_rate": 2.4743052912066998e-06, + "loss": 1.5202, + "step": 1635 + }, + { + "epoch": 1.6805341551104263, + "grad_norm": 2.8254599571228027, + "learning_rate": 2.472401979444233e-06, + "loss": 1.3781, + "step": 1636 + }, + { + "epoch": 1.6815613764766306, + "grad_norm": 2.9406354427337646, + "learning_rate": 2.4704986676817665e-06, + "loss": 1.4686, + "step": 1637 + }, + { + "epoch": 1.6825885978428352, + "grad_norm": 2.8287105560302734, + "learning_rate": 2.4685953559193e-06, + "loss": 1.4026, + "step": 1638 + }, + { + "epoch": 1.6836158192090396, + "grad_norm": 2.923633337020874, + "learning_rate": 2.466692044156833e-06, + "loss": 1.4493, + "step": 1639 + }, + { + "epoch": 1.684643040575244, + "grad_norm": 2.9239513874053955, + "learning_rate": 2.4647887323943666e-06, + "loss": 1.3878, + "step": 1640 + }, + { + "epoch": 1.684643040575244, + "eval_loss": 1.853607416152954, + "eval_runtime": 37.1661, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1640 + }, + { + "epoch": 1.6856702619414485, + "grad_norm": 2.699998617172241, + "learning_rate": 2.4628854206319e-06, + "loss": 1.3637, + "step": 1641 + }, + { + "epoch": 1.6866974833076527, + "grad_norm": 2.901700258255005, + "learning_rate": 2.460982108869433e-06, + "loss": 1.3554, + "step": 1642 + }, + { + "epoch": 1.6877247046738573, + "grad_norm": 3.011157512664795, + "learning_rate": 2.4590787971069662e-06, + "loss": 1.4487, + "step": 1643 + }, + { + "epoch": 1.6887519260400616, + "grad_norm": 2.672830581665039, + "learning_rate": 2.4571754853444996e-06, + "loss": 1.3906, + "step": 1644 + }, + { + "epoch": 1.689779147406266, + "grad_norm": 2.7201497554779053, + "learning_rate": 2.455272173582033e-06, + "loss": 1.3747, + "step": 1645 + }, + { + "epoch": 1.6908063687724706, + "grad_norm": 2.979395627975464, + "learning_rate": 2.4533688618195663e-06, + "loss": 1.3895, + "step": 1646 + }, + { + "epoch": 1.6918335901386747, + "grad_norm": 2.8048512935638428, + "learning_rate": 2.4514655500570997e-06, + "loss": 1.3271, + "step": 1647 + }, + { + "epoch": 1.6928608115048793, + "grad_norm": 2.6457700729370117, + "learning_rate": 2.449562238294633e-06, + "loss": 1.3679, + "step": 1648 + }, + { + "epoch": 1.6938880328710837, + "grad_norm": 2.732454299926758, + "learning_rate": 2.4476589265321664e-06, + "loss": 1.3332, + "step": 1649 + }, + { + "epoch": 1.694915254237288, + "grad_norm": 3.190871238708496, + "learning_rate": 2.4457556147696993e-06, + "loss": 1.4437, + "step": 1650 + }, + { + "epoch": 1.694915254237288, + "eval_loss": 1.8642219305038452, + "eval_runtime": 37.1865, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 1650 + }, + { + "epoch": 1.6959424756034926, + "grad_norm": 2.85784912109375, + "learning_rate": 2.4438523030072327e-06, + "loss": 1.4121, + "step": 1651 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.766545534133911, + "learning_rate": 2.441948991244766e-06, + "loss": 1.4231, + "step": 1652 + }, + { + "epoch": 1.6979969183359014, + "grad_norm": 2.8962197303771973, + "learning_rate": 2.4400456794822994e-06, + "loss": 1.4146, + "step": 1653 + }, + { + "epoch": 1.6990241397021058, + "grad_norm": 2.907231569290161, + "learning_rate": 2.438142367719833e-06, + "loss": 1.4375, + "step": 1654 + }, + { + "epoch": 1.7000513610683101, + "grad_norm": 2.8746557235717773, + "learning_rate": 2.4362390559573657e-06, + "loss": 1.4002, + "step": 1655 + }, + { + "epoch": 1.7010785824345147, + "grad_norm": 2.741013765335083, + "learning_rate": 2.434335744194899e-06, + "loss": 1.3865, + "step": 1656 + }, + { + "epoch": 1.702105803800719, + "grad_norm": 3.118908643722534, + "learning_rate": 2.432432432432433e-06, + "loss": 1.4412, + "step": 1657 + }, + { + "epoch": 1.7031330251669234, + "grad_norm": 2.789179563522339, + "learning_rate": 2.4305291206699662e-06, + "loss": 1.3446, + "step": 1658 + }, + { + "epoch": 1.704160246533128, + "grad_norm": 2.9241371154785156, + "learning_rate": 2.428625808907499e-06, + "loss": 1.4263, + "step": 1659 + }, + { + "epoch": 1.7051874678993322, + "grad_norm": 2.8632888793945312, + "learning_rate": 2.4267224971450325e-06, + "loss": 1.4413, + "step": 1660 + }, + { + "epoch": 1.7051874678993322, + "eval_loss": 1.85648512840271, + "eval_runtime": 37.199, + "eval_samples_per_second": 26.882, + "eval_steps_per_second": 1.129, + "step": 1660 + }, + { + "epoch": 1.7062146892655368, + "grad_norm": 2.948183059692383, + "learning_rate": 2.424819185382566e-06, + "loss": 1.4058, + "step": 1661 + }, + { + "epoch": 1.7072419106317411, + "grad_norm": 3.04127836227417, + "learning_rate": 2.4229158736200993e-06, + "loss": 1.3582, + "step": 1662 + }, + { + "epoch": 1.7082691319979455, + "grad_norm": 2.6956393718719482, + "learning_rate": 2.4210125618576326e-06, + "loss": 1.3705, + "step": 1663 + }, + { + "epoch": 1.70929635336415, + "grad_norm": 2.7690250873565674, + "learning_rate": 2.4191092500951656e-06, + "loss": 1.3869, + "step": 1664 + }, + { + "epoch": 1.7103235747303542, + "grad_norm": 2.851231813430786, + "learning_rate": 2.417205938332699e-06, + "loss": 1.4198, + "step": 1665 + }, + { + "epoch": 1.7113507960965588, + "grad_norm": 2.9241318702697754, + "learning_rate": 2.4153026265702323e-06, + "loss": 1.3441, + "step": 1666 + }, + { + "epoch": 1.7123780174627632, + "grad_norm": 2.7888636589050293, + "learning_rate": 2.4133993148077657e-06, + "loss": 1.3765, + "step": 1667 + }, + { + "epoch": 1.7134052388289676, + "grad_norm": 2.891366481781006, + "learning_rate": 2.411496003045299e-06, + "loss": 1.4393, + "step": 1668 + }, + { + "epoch": 1.7144324601951721, + "grad_norm": 2.8162693977355957, + "learning_rate": 2.4095926912828324e-06, + "loss": 1.3811, + "step": 1669 + }, + { + "epoch": 1.7154596815613765, + "grad_norm": 2.6741931438446045, + "learning_rate": 2.4076893795203658e-06, + "loss": 1.404, + "step": 1670 + }, + { + "epoch": 1.7154596815613765, + "eval_loss": 1.8440946340560913, + "eval_runtime": 37.1273, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1670 + }, + { + "epoch": 1.7164869029275809, + "grad_norm": 2.834951400756836, + "learning_rate": 2.405786067757899e-06, + "loss": 1.377, + "step": 1671 + }, + { + "epoch": 1.7175141242937855, + "grad_norm": 2.846113681793213, + "learning_rate": 2.403882755995432e-06, + "loss": 1.4689, + "step": 1672 + }, + { + "epoch": 1.7185413456599896, + "grad_norm": 2.6862854957580566, + "learning_rate": 2.4019794442329654e-06, + "loss": 1.3979, + "step": 1673 + }, + { + "epoch": 1.7195685670261942, + "grad_norm": 2.7214972972869873, + "learning_rate": 2.4000761324704988e-06, + "loss": 1.3738, + "step": 1674 + }, + { + "epoch": 1.7205957883923986, + "grad_norm": 2.7037723064422607, + "learning_rate": 2.398172820708032e-06, + "loss": 1.4184, + "step": 1675 + }, + { + "epoch": 1.721623009758603, + "grad_norm": 2.7854912281036377, + "learning_rate": 2.3962695089455655e-06, + "loss": 1.3854, + "step": 1676 + }, + { + "epoch": 1.7226502311248075, + "grad_norm": 2.640059471130371, + "learning_rate": 2.3943661971830984e-06, + "loss": 1.3656, + "step": 1677 + }, + { + "epoch": 1.7236774524910117, + "grad_norm": 2.6955630779266357, + "learning_rate": 2.3924628854206322e-06, + "loss": 1.3764, + "step": 1678 + }, + { + "epoch": 1.7247046738572163, + "grad_norm": 3.018975257873535, + "learning_rate": 2.3905595736581656e-06, + "loss": 1.3922, + "step": 1679 + }, + { + "epoch": 1.7257318952234206, + "grad_norm": 2.7709803581237793, + "learning_rate": 2.388656261895699e-06, + "loss": 1.4288, + "step": 1680 + }, + { + "epoch": 1.7257318952234206, + "eval_loss": 1.8474006652832031, + "eval_runtime": 37.1233, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 1680 + }, + { + "epoch": 1.726759116589625, + "grad_norm": 2.8898143768310547, + "learning_rate": 2.386752950133232e-06, + "loss": 1.4143, + "step": 1681 + }, + { + "epoch": 1.7277863379558296, + "grad_norm": 2.9092485904693604, + "learning_rate": 2.3848496383707653e-06, + "loss": 1.4097, + "step": 1682 + }, + { + "epoch": 1.7288135593220337, + "grad_norm": 2.886662483215332, + "learning_rate": 2.3829463266082986e-06, + "loss": 1.3968, + "step": 1683 + }, + { + "epoch": 1.7298407806882383, + "grad_norm": 2.7398223876953125, + "learning_rate": 2.381043014845832e-06, + "loss": 1.4198, + "step": 1684 + }, + { + "epoch": 1.7308680020544427, + "grad_norm": 2.626732587814331, + "learning_rate": 2.3791397030833653e-06, + "loss": 1.4116, + "step": 1685 + }, + { + "epoch": 1.731895223420647, + "grad_norm": 2.6434433460235596, + "learning_rate": 2.3772363913208983e-06, + "loss": 1.3857, + "step": 1686 + }, + { + "epoch": 1.7329224447868516, + "grad_norm": 2.83876633644104, + "learning_rate": 2.3753330795584316e-06, + "loss": 1.3672, + "step": 1687 + }, + { + "epoch": 1.733949666153056, + "grad_norm": 2.925100564956665, + "learning_rate": 2.3734297677959654e-06, + "loss": 1.3957, + "step": 1688 + }, + { + "epoch": 1.7349768875192604, + "grad_norm": 2.833876371383667, + "learning_rate": 2.3715264560334984e-06, + "loss": 1.442, + "step": 1689 + }, + { + "epoch": 1.736004108885465, + "grad_norm": 2.8355112075805664, + "learning_rate": 2.3696231442710317e-06, + "loss": 1.4153, + "step": 1690 + }, + { + "epoch": 1.736004108885465, + "eval_loss": 1.8604445457458496, + "eval_runtime": 37.2018, + "eval_samples_per_second": 26.88, + "eval_steps_per_second": 1.129, + "step": 1690 + }, + { + "epoch": 1.7370313302516691, + "grad_norm": 2.7500364780426025, + "learning_rate": 2.367719832508565e-06, + "loss": 1.4165, + "step": 1691 + }, + { + "epoch": 1.7380585516178737, + "grad_norm": 2.87259578704834, + "learning_rate": 2.3658165207460985e-06, + "loss": 1.4431, + "step": 1692 + }, + { + "epoch": 1.739085772984078, + "grad_norm": 2.899538993835449, + "learning_rate": 2.363913208983632e-06, + "loss": 1.4134, + "step": 1693 + }, + { + "epoch": 1.7401129943502824, + "grad_norm": 2.799593687057495, + "learning_rate": 2.362009897221165e-06, + "loss": 1.4249, + "step": 1694 + }, + { + "epoch": 1.741140215716487, + "grad_norm": 2.8966968059539795, + "learning_rate": 2.360106585458698e-06, + "loss": 1.4184, + "step": 1695 + }, + { + "epoch": 1.7421674370826912, + "grad_norm": 2.6944682598114014, + "learning_rate": 2.3582032736962315e-06, + "loss": 1.3678, + "step": 1696 + }, + { + "epoch": 1.7431946584488958, + "grad_norm": 2.819845199584961, + "learning_rate": 2.356299961933765e-06, + "loss": 1.407, + "step": 1697 + }, + { + "epoch": 1.7442218798151001, + "grad_norm": 2.6873581409454346, + "learning_rate": 2.354396650171298e-06, + "loss": 1.3771, + "step": 1698 + }, + { + "epoch": 1.7452491011813045, + "grad_norm": 2.6941304206848145, + "learning_rate": 2.3524933384088316e-06, + "loss": 1.3743, + "step": 1699 + }, + { + "epoch": 1.746276322547509, + "grad_norm": 2.925926685333252, + "learning_rate": 2.350590026646365e-06, + "loss": 1.4549, + "step": 1700 + }, + { + "epoch": 1.746276322547509, + "eval_loss": 1.8619909286499023, + "eval_runtime": 37.3963, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 1700 + }, + { + "epoch": 1.7473035439137135, + "grad_norm": 2.7275006771087646, + "learning_rate": 2.3486867148838983e-06, + "loss": 1.385, + "step": 1701 + }, + { + "epoch": 1.7483307652799178, + "grad_norm": 2.7096338272094727, + "learning_rate": 2.3467834031214317e-06, + "loss": 1.3732, + "step": 1702 + }, + { + "epoch": 1.7493579866461222, + "grad_norm": 2.718862295150757, + "learning_rate": 2.3448800913589646e-06, + "loss": 1.3889, + "step": 1703 + }, + { + "epoch": 1.7503852080123266, + "grad_norm": 2.8386173248291016, + "learning_rate": 2.342976779596498e-06, + "loss": 1.4083, + "step": 1704 + }, + { + "epoch": 1.7514124293785311, + "grad_norm": 2.868790864944458, + "learning_rate": 2.3410734678340313e-06, + "loss": 1.3958, + "step": 1705 + }, + { + "epoch": 1.7524396507447355, + "grad_norm": 2.869119882583618, + "learning_rate": 2.3391701560715647e-06, + "loss": 1.4344, + "step": 1706 + }, + { + "epoch": 1.7534668721109399, + "grad_norm": 2.702211380004883, + "learning_rate": 2.337266844309098e-06, + "loss": 1.3928, + "step": 1707 + }, + { + "epoch": 1.7544940934771445, + "grad_norm": 2.6727139949798584, + "learning_rate": 2.3353635325466314e-06, + "loss": 1.3872, + "step": 1708 + }, + { + "epoch": 1.7555213148433486, + "grad_norm": 2.875605583190918, + "learning_rate": 2.3334602207841648e-06, + "loss": 1.4097, + "step": 1709 + }, + { + "epoch": 1.7565485362095532, + "grad_norm": 2.7349722385406494, + "learning_rate": 2.331556909021698e-06, + "loss": 1.4041, + "step": 1710 + }, + { + "epoch": 1.7565485362095532, + "eval_loss": 1.8533939123153687, + "eval_runtime": 36.9293, + "eval_samples_per_second": 27.079, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 2.7475039958953857, + "learning_rate": 2.3296535972592315e-06, + "loss": 1.4339, + "step": 1711 + }, + { + "epoch": 1.758602978941962, + "grad_norm": 2.705641746520996, + "learning_rate": 2.3277502854967644e-06, + "loss": 1.3763, + "step": 1712 + }, + { + "epoch": 1.7596302003081665, + "grad_norm": 2.77583646774292, + "learning_rate": 2.325846973734298e-06, + "loss": 1.4423, + "step": 1713 + }, + { + "epoch": 1.7606574216743707, + "grad_norm": 2.662750482559204, + "learning_rate": 2.323943661971831e-06, + "loss": 1.3864, + "step": 1714 + }, + { + "epoch": 1.7616846430405753, + "grad_norm": 2.769313335418701, + "learning_rate": 2.3220403502093645e-06, + "loss": 1.4272, + "step": 1715 + }, + { + "epoch": 1.7627118644067796, + "grad_norm": 2.7852976322174072, + "learning_rate": 2.320137038446898e-06, + "loss": 1.4398, + "step": 1716 + }, + { + "epoch": 1.763739085772984, + "grad_norm": 2.7452852725982666, + "learning_rate": 2.318233726684431e-06, + "loss": 1.4504, + "step": 1717 + }, + { + "epoch": 1.7647663071391886, + "grad_norm": 2.762258291244507, + "learning_rate": 2.3163304149219646e-06, + "loss": 1.4325, + "step": 1718 + }, + { + "epoch": 1.765793528505393, + "grad_norm": 2.7612709999084473, + "learning_rate": 2.314427103159498e-06, + "loss": 1.4027, + "step": 1719 + }, + { + "epoch": 1.7668207498715973, + "grad_norm": 2.7290871143341064, + "learning_rate": 2.312523791397031e-06, + "loss": 1.3919, + "step": 1720 + }, + { + "epoch": 1.7668207498715973, + "eval_loss": 1.8454608917236328, + "eval_runtime": 37.0509, + "eval_samples_per_second": 26.99, + "eval_steps_per_second": 1.134, + "step": 1720 + }, + { + "epoch": 1.767847971237802, + "grad_norm": 2.5960779190063477, + "learning_rate": 2.3106204796345643e-06, + "loss": 1.3632, + "step": 1721 + }, + { + "epoch": 1.768875192604006, + "grad_norm": 2.7832717895507812, + "learning_rate": 2.3087171678720976e-06, + "loss": 1.3899, + "step": 1722 + }, + { + "epoch": 1.7699024139702106, + "grad_norm": 2.8227648735046387, + "learning_rate": 2.306813856109631e-06, + "loss": 1.4517, + "step": 1723 + }, + { + "epoch": 1.770929635336415, + "grad_norm": 2.880059242248535, + "learning_rate": 2.3049105443471644e-06, + "loss": 1.4104, + "step": 1724 + }, + { + "epoch": 1.7719568567026194, + "grad_norm": 2.6653501987457275, + "learning_rate": 2.3030072325846973e-06, + "loss": 1.3908, + "step": 1725 + }, + { + "epoch": 1.772984078068824, + "grad_norm": 2.844914197921753, + "learning_rate": 2.3011039208222307e-06, + "loss": 1.3781, + "step": 1726 + }, + { + "epoch": 1.774011299435028, + "grad_norm": 2.8133163452148438, + "learning_rate": 2.299200609059764e-06, + "loss": 1.3775, + "step": 1727 + }, + { + "epoch": 1.7750385208012327, + "grad_norm": 2.7466955184936523, + "learning_rate": 2.297297297297298e-06, + "loss": 1.3926, + "step": 1728 + }, + { + "epoch": 1.776065742167437, + "grad_norm": 2.6311862468719482, + "learning_rate": 2.2953939855348308e-06, + "loss": 1.3486, + "step": 1729 + }, + { + "epoch": 1.7770929635336414, + "grad_norm": 2.7905211448669434, + "learning_rate": 2.293490673772364e-06, + "loss": 1.4369, + "step": 1730 + }, + { + "epoch": 1.7770929635336414, + "eval_loss": 1.8418751955032349, + "eval_runtime": 37.0448, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1730 + }, + { + "epoch": 1.778120184899846, + "grad_norm": 2.862295627593994, + "learning_rate": 2.2915873620098975e-06, + "loss": 1.4209, + "step": 1731 + }, + { + "epoch": 1.7791474062660502, + "grad_norm": 2.765862464904785, + "learning_rate": 2.289684050247431e-06, + "loss": 1.4135, + "step": 1732 + }, + { + "epoch": 1.7801746276322548, + "grad_norm": 2.7296290397644043, + "learning_rate": 2.287780738484964e-06, + "loss": 1.3814, + "step": 1733 + }, + { + "epoch": 1.7812018489984591, + "grad_norm": 2.7937533855438232, + "learning_rate": 2.285877426722497e-06, + "loss": 1.4203, + "step": 1734 + }, + { + "epoch": 1.7822290703646635, + "grad_norm": 2.7680230140686035, + "learning_rate": 2.2839741149600305e-06, + "loss": 1.4354, + "step": 1735 + }, + { + "epoch": 1.783256291730868, + "grad_norm": 2.8785109519958496, + "learning_rate": 2.282070803197564e-06, + "loss": 1.4338, + "step": 1736 + }, + { + "epoch": 1.7842835130970724, + "grad_norm": 2.747436046600342, + "learning_rate": 2.2801674914350972e-06, + "loss": 1.4087, + "step": 1737 + }, + { + "epoch": 1.7853107344632768, + "grad_norm": 2.6810362339019775, + "learning_rate": 2.2782641796726306e-06, + "loss": 1.3993, + "step": 1738 + }, + { + "epoch": 1.7863379558294814, + "grad_norm": 2.7365734577178955, + "learning_rate": 2.276360867910164e-06, + "loss": 1.4235, + "step": 1739 + }, + { + "epoch": 1.7873651771956856, + "grad_norm": 2.6946303844451904, + "learning_rate": 2.2744575561476973e-06, + "loss": 1.3626, + "step": 1740 + }, + { + "epoch": 1.7873651771956856, + "eval_loss": 1.8456190824508667, + "eval_runtime": 37.069, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1740 + }, + { + "epoch": 1.7883923985618901, + "grad_norm": 2.756920337677002, + "learning_rate": 2.2725542443852307e-06, + "loss": 1.396, + "step": 1741 + }, + { + "epoch": 1.7894196199280945, + "grad_norm": 2.765428066253662, + "learning_rate": 2.2706509326227636e-06, + "loss": 1.4374, + "step": 1742 + }, + { + "epoch": 1.7904468412942989, + "grad_norm": 2.8224971294403076, + "learning_rate": 2.268747620860297e-06, + "loss": 1.4113, + "step": 1743 + }, + { + "epoch": 1.7914740626605035, + "grad_norm": 2.90537428855896, + "learning_rate": 2.2668443090978303e-06, + "loss": 1.4124, + "step": 1744 + }, + { + "epoch": 1.7925012840267076, + "grad_norm": 2.712862491607666, + "learning_rate": 2.2649409973353637e-06, + "loss": 1.4256, + "step": 1745 + }, + { + "epoch": 1.7935285053929122, + "grad_norm": 2.688558340072632, + "learning_rate": 2.263037685572897e-06, + "loss": 1.3842, + "step": 1746 + }, + { + "epoch": 1.7945557267591166, + "grad_norm": 2.6504769325256348, + "learning_rate": 2.26113437381043e-06, + "loss": 1.3703, + "step": 1747 + }, + { + "epoch": 1.795582948125321, + "grad_norm": 3.0093724727630615, + "learning_rate": 2.259231062047964e-06, + "loss": 1.4106, + "step": 1748 + }, + { + "epoch": 1.7966101694915255, + "grad_norm": 2.727687120437622, + "learning_rate": 2.257327750285497e-06, + "loss": 1.4286, + "step": 1749 + }, + { + "epoch": 1.7976373908577299, + "grad_norm": 2.679896116256714, + "learning_rate": 2.2554244385230305e-06, + "loss": 1.3937, + "step": 1750 + }, + { + "epoch": 1.7976373908577299, + "eval_loss": 1.8520207405090332, + "eval_runtime": 37.062, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1750 + }, + { + "epoch": 1.7986646122239343, + "grad_norm": 2.8119945526123047, + "learning_rate": 2.2535211267605635e-06, + "loss": 1.3935, + "step": 1751 + }, + { + "epoch": 1.7996918335901386, + "grad_norm": 2.8470423221588135, + "learning_rate": 2.251617814998097e-06, + "loss": 1.4138, + "step": 1752 + }, + { + "epoch": 1.800719054956343, + "grad_norm": 2.6278114318847656, + "learning_rate": 2.24971450323563e-06, + "loss": 1.328, + "step": 1753 + }, + { + "epoch": 1.8017462763225476, + "grad_norm": 2.861860990524292, + "learning_rate": 2.2478111914731636e-06, + "loss": 1.416, + "step": 1754 + }, + { + "epoch": 1.802773497688752, + "grad_norm": 2.559736728668213, + "learning_rate": 2.245907879710697e-06, + "loss": 1.2891, + "step": 1755 + }, + { + "epoch": 1.8038007190549563, + "grad_norm": 2.8461058139801025, + "learning_rate": 2.24400456794823e-06, + "loss": 1.4194, + "step": 1756 + }, + { + "epoch": 1.804827940421161, + "grad_norm": 2.7069878578186035, + "learning_rate": 2.2421012561857632e-06, + "loss": 1.382, + "step": 1757 + }, + { + "epoch": 1.805855161787365, + "grad_norm": 2.727435827255249, + "learning_rate": 2.240197944423297e-06, + "loss": 1.3714, + "step": 1758 + }, + { + "epoch": 1.8068823831535696, + "grad_norm": 2.7298061847686768, + "learning_rate": 2.23829463266083e-06, + "loss": 1.3834, + "step": 1759 + }, + { + "epoch": 1.807909604519774, + "grad_norm": 2.7419843673706055, + "learning_rate": 2.2363913208983633e-06, + "loss": 1.3687, + "step": 1760 + }, + { + "epoch": 1.807909604519774, + "eval_loss": 1.8575356006622314, + "eval_runtime": 37.0563, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1760 + }, + { + "epoch": 1.8089368258859784, + "grad_norm": 2.8058817386627197, + "learning_rate": 2.2344880091358967e-06, + "loss": 1.415, + "step": 1761 + }, + { + "epoch": 1.809964047252183, + "grad_norm": 2.7287003993988037, + "learning_rate": 2.23258469737343e-06, + "loss": 1.3778, + "step": 1762 + }, + { + "epoch": 1.810991268618387, + "grad_norm": 2.7943124771118164, + "learning_rate": 2.2306813856109634e-06, + "loss": 1.433, + "step": 1763 + }, + { + "epoch": 1.8120184899845917, + "grad_norm": 2.7254626750946045, + "learning_rate": 2.2287780738484963e-06, + "loss": 1.3822, + "step": 1764 + }, + { + "epoch": 1.813045711350796, + "grad_norm": 2.799793004989624, + "learning_rate": 2.2268747620860297e-06, + "loss": 1.4595, + "step": 1765 + }, + { + "epoch": 1.8140729327170004, + "grad_norm": 2.79420804977417, + "learning_rate": 2.224971450323563e-06, + "loss": 1.3949, + "step": 1766 + }, + { + "epoch": 1.815100154083205, + "grad_norm": 2.861016035079956, + "learning_rate": 2.2230681385610964e-06, + "loss": 1.3301, + "step": 1767 + }, + { + "epoch": 1.8161273754494094, + "grad_norm": 2.9270451068878174, + "learning_rate": 2.2211648267986298e-06, + "loss": 1.3895, + "step": 1768 + }, + { + "epoch": 1.8171545968156138, + "grad_norm": 2.709031105041504, + "learning_rate": 2.219261515036163e-06, + "loss": 1.3757, + "step": 1769 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 2.7763030529022217, + "learning_rate": 2.2173582032736965e-06, + "loss": 1.4135, + "step": 1770 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.8521159887313843, + "eval_runtime": 37.1265, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 1.8192090395480225, + "grad_norm": 2.97048282623291, + "learning_rate": 2.21545489151123e-06, + "loss": 1.4345, + "step": 1771 + }, + { + "epoch": 1.820236260914227, + "grad_norm": 2.8238744735717773, + "learning_rate": 2.2135515797487632e-06, + "loss": 1.3948, + "step": 1772 + }, + { + "epoch": 1.8212634822804314, + "grad_norm": 2.6430611610412598, + "learning_rate": 2.211648267986296e-06, + "loss": 1.3878, + "step": 1773 + }, + { + "epoch": 1.8222907036466358, + "grad_norm": 2.602883815765381, + "learning_rate": 2.2097449562238295e-06, + "loss": 1.3978, + "step": 1774 + }, + { + "epoch": 1.8233179250128404, + "grad_norm": 2.707813262939453, + "learning_rate": 2.207841644461363e-06, + "loss": 1.3937, + "step": 1775 + }, + { + "epoch": 1.8243451463790445, + "grad_norm": 2.662684679031372, + "learning_rate": 2.2059383326988963e-06, + "loss": 1.4381, + "step": 1776 + }, + { + "epoch": 1.8253723677452491, + "grad_norm": 2.9407193660736084, + "learning_rate": 2.2040350209364296e-06, + "loss": 1.4059, + "step": 1777 + }, + { + "epoch": 1.8263995891114535, + "grad_norm": 2.6678171157836914, + "learning_rate": 2.2021317091739626e-06, + "loss": 1.3718, + "step": 1778 + }, + { + "epoch": 1.8274268104776579, + "grad_norm": 2.7536516189575195, + "learning_rate": 2.2002283974114963e-06, + "loss": 1.3998, + "step": 1779 + }, + { + "epoch": 1.8284540318438625, + "grad_norm": 2.7340128421783447, + "learning_rate": 2.1983250856490297e-06, + "loss": 1.3299, + "step": 1780 + }, + { + "epoch": 1.8284540318438625, + "eval_loss": 1.854264736175537, + "eval_runtime": 37.0874, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1780 + }, + { + "epoch": 1.8294812532100666, + "grad_norm": 2.844285726547241, + "learning_rate": 2.1964217738865626e-06, + "loss": 1.3746, + "step": 1781 + }, + { + "epoch": 1.8305084745762712, + "grad_norm": 2.9005022048950195, + "learning_rate": 2.194518462124096e-06, + "loss": 1.4087, + "step": 1782 + }, + { + "epoch": 1.8315356959424756, + "grad_norm": 2.8825247287750244, + "learning_rate": 2.1926151503616294e-06, + "loss": 1.4458, + "step": 1783 + }, + { + "epoch": 1.83256291730868, + "grad_norm": 2.712778091430664, + "learning_rate": 2.1907118385991627e-06, + "loss": 1.414, + "step": 1784 + }, + { + "epoch": 1.8335901386748845, + "grad_norm": 2.7036001682281494, + "learning_rate": 2.188808526836696e-06, + "loss": 1.3632, + "step": 1785 + }, + { + "epoch": 1.8346173600410889, + "grad_norm": 2.704606771469116, + "learning_rate": 2.1869052150742295e-06, + "loss": 1.3992, + "step": 1786 + }, + { + "epoch": 1.8356445814072933, + "grad_norm": 2.7564282417297363, + "learning_rate": 2.1850019033117624e-06, + "loss": 1.3331, + "step": 1787 + }, + { + "epoch": 1.8366718027734978, + "grad_norm": 2.898256540298462, + "learning_rate": 2.1830985915492958e-06, + "loss": 1.3881, + "step": 1788 + }, + { + "epoch": 1.837699024139702, + "grad_norm": 2.7897069454193115, + "learning_rate": 2.1811952797868295e-06, + "loss": 1.4081, + "step": 1789 + }, + { + "epoch": 1.8387262455059066, + "grad_norm": 2.7293193340301514, + "learning_rate": 2.1792919680243625e-06, + "loss": 1.3705, + "step": 1790 + }, + { + "epoch": 1.8387262455059066, + "eval_loss": 1.8483723402023315, + "eval_runtime": 37.4058, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 1790 + }, + { + "epoch": 1.839753466872111, + "grad_norm": 2.9257752895355225, + "learning_rate": 2.177388656261896e-06, + "loss": 1.4071, + "step": 1791 + }, + { + "epoch": 1.8407806882383153, + "grad_norm": 3.016033172607422, + "learning_rate": 2.1754853444994292e-06, + "loss": 1.4774, + "step": 1792 + }, + { + "epoch": 1.84180790960452, + "grad_norm": 2.74198842048645, + "learning_rate": 2.1735820327369626e-06, + "loss": 1.301, + "step": 1793 + }, + { + "epoch": 1.842835130970724, + "grad_norm": 2.809215545654297, + "learning_rate": 2.171678720974496e-06, + "loss": 1.4192, + "step": 1794 + }, + { + "epoch": 1.8438623523369286, + "grad_norm": 2.673997402191162, + "learning_rate": 2.169775409212029e-06, + "loss": 1.4182, + "step": 1795 + }, + { + "epoch": 1.844889573703133, + "grad_norm": 2.9519944190979004, + "learning_rate": 2.1678720974495622e-06, + "loss": 1.4253, + "step": 1796 + }, + { + "epoch": 1.8459167950693374, + "grad_norm": 2.7683424949645996, + "learning_rate": 2.1659687856870956e-06, + "loss": 1.4162, + "step": 1797 + }, + { + "epoch": 1.846944016435542, + "grad_norm": 2.861701726913452, + "learning_rate": 2.164065473924629e-06, + "loss": 1.4167, + "step": 1798 + }, + { + "epoch": 1.8479712378017463, + "grad_norm": 2.62581467628479, + "learning_rate": 2.1621621621621623e-06, + "loss": 1.3568, + "step": 1799 + }, + { + "epoch": 1.8489984591679507, + "grad_norm": 2.782197952270508, + "learning_rate": 2.1602588503996957e-06, + "loss": 1.4184, + "step": 1800 + }, + { + "epoch": 1.8489984591679507, + "eval_loss": 1.8527827262878418, + "eval_runtime": 37.4743, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 1.850025680534155, + "grad_norm": 2.7289304733276367, + "learning_rate": 2.158355538637229e-06, + "loss": 1.3979, + "step": 1801 + }, + { + "epoch": 1.8510529019003594, + "grad_norm": 2.7153542041778564, + "learning_rate": 2.1564522268747624e-06, + "loss": 1.3863, + "step": 1802 + }, + { + "epoch": 1.852080123266564, + "grad_norm": 2.884798765182495, + "learning_rate": 2.1545489151122958e-06, + "loss": 1.4656, + "step": 1803 + }, + { + "epoch": 1.8531073446327684, + "grad_norm": 2.6483845710754395, + "learning_rate": 2.1526456033498287e-06, + "loss": 1.4118, + "step": 1804 + }, + { + "epoch": 1.8541345659989728, + "grad_norm": 2.7074198722839355, + "learning_rate": 2.150742291587362e-06, + "loss": 1.41, + "step": 1805 + }, + { + "epoch": 1.8551617873651773, + "grad_norm": 2.734137535095215, + "learning_rate": 2.1488389798248954e-06, + "loss": 1.3968, + "step": 1806 + }, + { + "epoch": 1.8561890087313815, + "grad_norm": 2.7855992317199707, + "learning_rate": 2.146935668062429e-06, + "loss": 1.427, + "step": 1807 + }, + { + "epoch": 1.857216230097586, + "grad_norm": 2.7769339084625244, + "learning_rate": 2.145032356299962e-06, + "loss": 1.3476, + "step": 1808 + }, + { + "epoch": 1.8582434514637904, + "grad_norm": 2.6978964805603027, + "learning_rate": 2.1431290445374955e-06, + "loss": 1.4047, + "step": 1809 + }, + { + "epoch": 1.8592706728299948, + "grad_norm": 2.806438446044922, + "learning_rate": 2.141225732775029e-06, + "loss": 1.3693, + "step": 1810 + }, + { + "epoch": 1.8592706728299948, + "eval_loss": 1.8606462478637695, + "eval_runtime": 36.9431, + "eval_samples_per_second": 27.069, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 1.8602978941961994, + "grad_norm": 2.886394500732422, + "learning_rate": 2.1393224210125623e-06, + "loss": 1.4022, + "step": 1811 + }, + { + "epoch": 1.8613251155624035, + "grad_norm": 2.7849435806274414, + "learning_rate": 2.137419109250095e-06, + "loss": 1.3907, + "step": 1812 + }, + { + "epoch": 1.8623523369286081, + "grad_norm": 2.85810923576355, + "learning_rate": 2.1355157974876286e-06, + "loss": 1.4451, + "step": 1813 + }, + { + "epoch": 1.8633795582948125, + "grad_norm": 2.8717806339263916, + "learning_rate": 2.133612485725162e-06, + "loss": 1.4303, + "step": 1814 + }, + { + "epoch": 1.8644067796610169, + "grad_norm": 2.6955783367156982, + "learning_rate": 2.1317091739626953e-06, + "loss": 1.3454, + "step": 1815 + }, + { + "epoch": 1.8654340010272215, + "grad_norm": 2.835347890853882, + "learning_rate": 2.1298058622002286e-06, + "loss": 1.3671, + "step": 1816 + }, + { + "epoch": 1.8664612223934258, + "grad_norm": 2.804835796356201, + "learning_rate": 2.1279025504377616e-06, + "loss": 1.388, + "step": 1817 + }, + { + "epoch": 1.8674884437596302, + "grad_norm": 2.69130539894104, + "learning_rate": 2.125999238675295e-06, + "loss": 1.3815, + "step": 1818 + }, + { + "epoch": 1.8685156651258348, + "grad_norm": 2.856457233428955, + "learning_rate": 2.1240959269128287e-06, + "loss": 1.4323, + "step": 1819 + }, + { + "epoch": 1.869542886492039, + "grad_norm": 2.8100039958953857, + "learning_rate": 2.122192615150362e-06, + "loss": 1.3942, + "step": 1820 + }, + { + "epoch": 1.869542886492039, + "eval_loss": 1.860397219657898, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 1820 + }, + { + "epoch": 1.8705701078582435, + "grad_norm": 2.6977529525756836, + "learning_rate": 2.120289303387895e-06, + "loss": 1.3371, + "step": 1821 + }, + { + "epoch": 1.8715973292244479, + "grad_norm": 2.8403844833374023, + "learning_rate": 2.1183859916254284e-06, + "loss": 1.3776, + "step": 1822 + }, + { + "epoch": 1.8726245505906522, + "grad_norm": 2.829756498336792, + "learning_rate": 2.1164826798629618e-06, + "loss": 1.4229, + "step": 1823 + }, + { + "epoch": 1.8736517719568568, + "grad_norm": 2.6810450553894043, + "learning_rate": 2.114579368100495e-06, + "loss": 1.3757, + "step": 1824 + }, + { + "epoch": 1.874678993323061, + "grad_norm": 2.988982677459717, + "learning_rate": 2.1126760563380285e-06, + "loss": 1.3671, + "step": 1825 + }, + { + "epoch": 1.8757062146892656, + "grad_norm": 2.661953926086426, + "learning_rate": 2.1107727445755614e-06, + "loss": 1.3382, + "step": 1826 + }, + { + "epoch": 1.87673343605547, + "grad_norm": 2.7993199825286865, + "learning_rate": 2.1088694328130948e-06, + "loss": 1.4621, + "step": 1827 + }, + { + "epoch": 1.8777606574216743, + "grad_norm": 2.681609869003296, + "learning_rate": 2.106966121050628e-06, + "loss": 1.3596, + "step": 1828 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.988734722137451, + "learning_rate": 2.1050628092881615e-06, + "loss": 1.4269, + "step": 1829 + }, + { + "epoch": 1.879815100154083, + "grad_norm": 2.867772102355957, + "learning_rate": 2.103159497525695e-06, + "loss": 1.4618, + "step": 1830 + }, + { + "epoch": 1.879815100154083, + "eval_loss": 1.8595913648605347, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1830 + }, + { + "epoch": 1.8808423215202876, + "grad_norm": 2.7212493419647217, + "learning_rate": 2.1012561857632282e-06, + "loss": 1.3738, + "step": 1831 + }, + { + "epoch": 1.881869542886492, + "grad_norm": 2.7997097969055176, + "learning_rate": 2.0993528740007616e-06, + "loss": 1.4268, + "step": 1832 + }, + { + "epoch": 1.8828967642526964, + "grad_norm": 2.8695220947265625, + "learning_rate": 2.097449562238295e-06, + "loss": 1.4397, + "step": 1833 + }, + { + "epoch": 1.883923985618901, + "grad_norm": 2.681015729904175, + "learning_rate": 2.095546250475828e-06, + "loss": 1.3873, + "step": 1834 + }, + { + "epoch": 1.8849512069851053, + "grad_norm": 2.787360429763794, + "learning_rate": 2.0936429387133613e-06, + "loss": 1.3944, + "step": 1835 + }, + { + "epoch": 1.8859784283513097, + "grad_norm": 2.830164670944214, + "learning_rate": 2.0917396269508946e-06, + "loss": 1.4485, + "step": 1836 + }, + { + "epoch": 1.8870056497175143, + "grad_norm": 2.9803659915924072, + "learning_rate": 2.089836315188428e-06, + "loss": 1.4259, + "step": 1837 + }, + { + "epoch": 1.8880328710837184, + "grad_norm": 2.846390962600708, + "learning_rate": 2.0879330034259614e-06, + "loss": 1.4371, + "step": 1838 + }, + { + "epoch": 1.889060092449923, + "grad_norm": 2.636085033416748, + "learning_rate": 2.0860296916634947e-06, + "loss": 1.3679, + "step": 1839 + }, + { + "epoch": 1.8900873138161274, + "grad_norm": 2.806547164916992, + "learning_rate": 2.084126379901028e-06, + "loss": 1.4662, + "step": 1840 + }, + { + "epoch": 1.8900873138161274, + "eval_loss": 1.8502073287963867, + "eval_runtime": 37.1321, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 1.8911145351823317, + "grad_norm": 2.8658089637756348, + "learning_rate": 2.0822230681385614e-06, + "loss": 1.3593, + "step": 1841 + }, + { + "epoch": 1.8921417565485363, + "grad_norm": 2.87060284614563, + "learning_rate": 2.080319756376095e-06, + "loss": 1.4189, + "step": 1842 + }, + { + "epoch": 1.8931689779147405, + "grad_norm": 2.7965638637542725, + "learning_rate": 2.0784164446136277e-06, + "loss": 1.4208, + "step": 1843 + }, + { + "epoch": 1.894196199280945, + "grad_norm": 2.5999045372009277, + "learning_rate": 2.076513132851161e-06, + "loss": 1.375, + "step": 1844 + }, + { + "epoch": 1.8952234206471494, + "grad_norm": 2.833935260772705, + "learning_rate": 2.0746098210886945e-06, + "loss": 1.4137, + "step": 1845 + }, + { + "epoch": 1.8962506420133538, + "grad_norm": 2.9809396266937256, + "learning_rate": 2.072706509326228e-06, + "loss": 1.3874, + "step": 1846 + }, + { + "epoch": 1.8972778633795584, + "grad_norm": 2.9341282844543457, + "learning_rate": 2.070803197563761e-06, + "loss": 1.3953, + "step": 1847 + }, + { + "epoch": 1.8983050847457628, + "grad_norm": 2.989488124847412, + "learning_rate": 2.068899885801294e-06, + "loss": 1.3913, + "step": 1848 + }, + { + "epoch": 1.8993323061119671, + "grad_norm": 2.8224217891693115, + "learning_rate": 2.066996574038828e-06, + "loss": 1.4032, + "step": 1849 + }, + { + "epoch": 1.9003595274781715, + "grad_norm": 2.8850255012512207, + "learning_rate": 2.0650932622763613e-06, + "loss": 1.4402, + "step": 1850 + }, + { + "epoch": 1.9003595274781715, + "eval_loss": 1.8440051078796387, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1850 + }, + { + "epoch": 1.9013867488443759, + "grad_norm": 2.927288293838501, + "learning_rate": 2.0631899505138942e-06, + "loss": 1.3215, + "step": 1851 + }, + { + "epoch": 1.9024139702105805, + "grad_norm": 2.7680368423461914, + "learning_rate": 2.0612866387514276e-06, + "loss": 1.3636, + "step": 1852 + }, + { + "epoch": 1.9034411915767848, + "grad_norm": 2.8515689373016357, + "learning_rate": 2.059383326988961e-06, + "loss": 1.4049, + "step": 1853 + }, + { + "epoch": 1.9044684129429892, + "grad_norm": 2.7973034381866455, + "learning_rate": 2.0574800152264943e-06, + "loss": 1.3876, + "step": 1854 + }, + { + "epoch": 1.9054956343091938, + "grad_norm": 2.7860023975372314, + "learning_rate": 2.0555767034640277e-06, + "loss": 1.3687, + "step": 1855 + }, + { + "epoch": 1.906522855675398, + "grad_norm": 2.807511568069458, + "learning_rate": 2.053673391701561e-06, + "loss": 1.3382, + "step": 1856 + }, + { + "epoch": 1.9075500770416025, + "grad_norm": 2.9342591762542725, + "learning_rate": 2.051770079939094e-06, + "loss": 1.4025, + "step": 1857 + }, + { + "epoch": 1.9085772984078069, + "grad_norm": 2.85943865776062, + "learning_rate": 2.0498667681766273e-06, + "loss": 1.3626, + "step": 1858 + }, + { + "epoch": 1.9096045197740112, + "grad_norm": 2.8514609336853027, + "learning_rate": 2.047963456414161e-06, + "loss": 1.3675, + "step": 1859 + }, + { + "epoch": 1.9106317411402158, + "grad_norm": 2.8177332878112793, + "learning_rate": 2.046060144651694e-06, + "loss": 1.3968, + "step": 1860 + }, + { + "epoch": 1.9106317411402158, + "eval_loss": 1.8451731204986572, + "eval_runtime": 37.1047, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1860 + }, + { + "epoch": 1.91165896250642, + "grad_norm": 2.74876070022583, + "learning_rate": 2.0441568328892274e-06, + "loss": 1.4022, + "step": 1861 + }, + { + "epoch": 1.9126861838726246, + "grad_norm": 2.6633286476135254, + "learning_rate": 2.0422535211267608e-06, + "loss": 1.4004, + "step": 1862 + }, + { + "epoch": 1.913713405238829, + "grad_norm": 2.6457369327545166, + "learning_rate": 2.040350209364294e-06, + "loss": 1.3767, + "step": 1863 + }, + { + "epoch": 1.9147406266050333, + "grad_norm": 2.7011449337005615, + "learning_rate": 2.0384468976018275e-06, + "loss": 1.373, + "step": 1864 + }, + { + "epoch": 1.915767847971238, + "grad_norm": 2.676337480545044, + "learning_rate": 2.0365435858393604e-06, + "loss": 1.3713, + "step": 1865 + }, + { + "epoch": 1.9167950693374423, + "grad_norm": 2.7401342391967773, + "learning_rate": 2.034640274076894e-06, + "loss": 1.3644, + "step": 1866 + }, + { + "epoch": 1.9178222907036466, + "grad_norm": 2.7799527645111084, + "learning_rate": 2.032736962314427e-06, + "loss": 1.433, + "step": 1867 + }, + { + "epoch": 1.9188495120698512, + "grad_norm": 2.7372307777404785, + "learning_rate": 2.0308336505519605e-06, + "loss": 1.3781, + "step": 1868 + }, + { + "epoch": 1.9198767334360554, + "grad_norm": 2.986297845840454, + "learning_rate": 2.028930338789494e-06, + "loss": 1.4406, + "step": 1869 + }, + { + "epoch": 1.92090395480226, + "grad_norm": 2.815706491470337, + "learning_rate": 2.0270270270270273e-06, + "loss": 1.3555, + "step": 1870 + }, + { + "epoch": 1.92090395480226, + "eval_loss": 1.8498139381408691, + "eval_runtime": 37.1146, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1870 + }, + { + "epoch": 1.9219311761684643, + "grad_norm": 2.734189987182617, + "learning_rate": 2.0251237152645606e-06, + "loss": 1.373, + "step": 1871 + }, + { + "epoch": 1.9229583975346687, + "grad_norm": 2.7900359630584717, + "learning_rate": 2.023220403502094e-06, + "loss": 1.3623, + "step": 1872 + }, + { + "epoch": 1.9239856189008733, + "grad_norm": 2.8590710163116455, + "learning_rate": 2.0213170917396273e-06, + "loss": 1.3424, + "step": 1873 + }, + { + "epoch": 1.9250128402670774, + "grad_norm": 2.880814790725708, + "learning_rate": 2.0194137799771603e-06, + "loss": 1.3683, + "step": 1874 + }, + { + "epoch": 1.926040061633282, + "grad_norm": 2.7775630950927734, + "learning_rate": 2.0175104682146937e-06, + "loss": 1.3879, + "step": 1875 + }, + { + "epoch": 1.9270672829994864, + "grad_norm": 2.713557481765747, + "learning_rate": 2.015607156452227e-06, + "loss": 1.3216, + "step": 1876 + }, + { + "epoch": 1.9280945043656907, + "grad_norm": 2.8787734508514404, + "learning_rate": 2.0137038446897604e-06, + "loss": 1.3659, + "step": 1877 + }, + { + "epoch": 1.9291217257318953, + "grad_norm": 2.8224005699157715, + "learning_rate": 2.0118005329272937e-06, + "loss": 1.3572, + "step": 1878 + }, + { + "epoch": 1.9301489470980995, + "grad_norm": 2.865316390991211, + "learning_rate": 2.0098972211648267e-06, + "loss": 1.3817, + "step": 1879 + }, + { + "epoch": 1.931176168464304, + "grad_norm": 2.7488853931427, + "learning_rate": 2.0079939094023605e-06, + "loss": 1.3766, + "step": 1880 + }, + { + "epoch": 1.931176168464304, + "eval_loss": 1.8444081544876099, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1880 + }, + { + "epoch": 1.9322033898305084, + "grad_norm": 2.902949333190918, + "learning_rate": 2.006090597639894e-06, + "loss": 1.4183, + "step": 1881 + }, + { + "epoch": 1.9332306111967128, + "grad_norm": 2.8154122829437256, + "learning_rate": 2.0041872858774268e-06, + "loss": 1.4525, + "step": 1882 + }, + { + "epoch": 1.9342578325629174, + "grad_norm": 2.8343429565429688, + "learning_rate": 2.00228397411496e-06, + "loss": 1.4099, + "step": 1883 + }, + { + "epoch": 1.9352850539291218, + "grad_norm": 2.863067388534546, + "learning_rate": 2.0003806623524935e-06, + "loss": 1.3973, + "step": 1884 + }, + { + "epoch": 1.9363122752953261, + "grad_norm": 2.784085988998413, + "learning_rate": 1.998477350590027e-06, + "loss": 1.3117, + "step": 1885 + }, + { + "epoch": 1.9373394966615307, + "grad_norm": 2.7667958736419678, + "learning_rate": 1.9965740388275602e-06, + "loss": 1.4056, + "step": 1886 + }, + { + "epoch": 1.9383667180277349, + "grad_norm": 2.747432231903076, + "learning_rate": 1.994670727065093e-06, + "loss": 1.3927, + "step": 1887 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9162001609802246, + "learning_rate": 1.9927674153026265e-06, + "loss": 1.3453, + "step": 1888 + }, + { + "epoch": 1.9404211607601438, + "grad_norm": 3.0909414291381836, + "learning_rate": 1.99086410354016e-06, + "loss": 1.4148, + "step": 1889 + }, + { + "epoch": 1.9414483821263482, + "grad_norm": 2.774672746658325, + "learning_rate": 1.9889607917776937e-06, + "loss": 1.3762, + "step": 1890 + }, + { + "epoch": 1.9414483821263482, + "eval_loss": 1.848209023475647, + "eval_runtime": 37.2784, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 1.9424756034925528, + "grad_norm": 2.7985880374908447, + "learning_rate": 1.9870574800152266e-06, + "loss": 1.3908, + "step": 1891 + }, + { + "epoch": 1.943502824858757, + "grad_norm": 2.749664783477783, + "learning_rate": 1.98515416825276e-06, + "loss": 1.4088, + "step": 1892 + }, + { + "epoch": 1.9445300462249615, + "grad_norm": 2.770726442337036, + "learning_rate": 1.9832508564902933e-06, + "loss": 1.4054, + "step": 1893 + }, + { + "epoch": 1.9455572675911659, + "grad_norm": 2.634516477584839, + "learning_rate": 1.9813475447278267e-06, + "loss": 1.3247, + "step": 1894 + }, + { + "epoch": 1.9465844889573702, + "grad_norm": 2.886481285095215, + "learning_rate": 1.97944423296536e-06, + "loss": 1.4025, + "step": 1895 + }, + { + "epoch": 1.9476117103235748, + "grad_norm": 2.8149733543395996, + "learning_rate": 1.977540921202893e-06, + "loss": 1.3745, + "step": 1896 + }, + { + "epoch": 1.9486389316897792, + "grad_norm": 2.643970251083374, + "learning_rate": 1.9756376094404264e-06, + "loss": 1.3581, + "step": 1897 + }, + { + "epoch": 1.9496661530559836, + "grad_norm": 2.7485244274139404, + "learning_rate": 1.9737342976779597e-06, + "loss": 1.3426, + "step": 1898 + }, + { + "epoch": 1.950693374422188, + "grad_norm": 2.930513858795166, + "learning_rate": 1.971830985915493e-06, + "loss": 1.4194, + "step": 1899 + }, + { + "epoch": 1.9517205957883923, + "grad_norm": 2.7637267112731934, + "learning_rate": 1.9699276741530264e-06, + "loss": 1.3908, + "step": 1900 + }, + { + "epoch": 1.9517205957883923, + "eval_loss": 1.8510984182357788, + "eval_runtime": 37.4089, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1900 + }, + { + "epoch": 1.9527478171545969, + "grad_norm": 2.8946938514709473, + "learning_rate": 1.96802436239056e-06, + "loss": 1.3892, + "step": 1901 + }, + { + "epoch": 1.9537750385208013, + "grad_norm": 2.7064263820648193, + "learning_rate": 1.966121050628093e-06, + "loss": 1.3628, + "step": 1902 + }, + { + "epoch": 1.9548022598870056, + "grad_norm": 2.8588147163391113, + "learning_rate": 1.9642177388656265e-06, + "loss": 1.4097, + "step": 1903 + }, + { + "epoch": 1.9558294812532102, + "grad_norm": 2.7746825218200684, + "learning_rate": 1.9623144271031595e-06, + "loss": 1.4255, + "step": 1904 + }, + { + "epoch": 1.9568567026194144, + "grad_norm": 2.872587203979492, + "learning_rate": 1.960411115340693e-06, + "loss": 1.4308, + "step": 1905 + }, + { + "epoch": 1.957883923985619, + "grad_norm": 2.714769124984741, + "learning_rate": 1.958507803578226e-06, + "loss": 1.3847, + "step": 1906 + }, + { + "epoch": 1.9589111453518233, + "grad_norm": 2.7117342948913574, + "learning_rate": 1.9566044918157596e-06, + "loss": 1.406, + "step": 1907 + }, + { + "epoch": 1.9599383667180277, + "grad_norm": 2.754995346069336, + "learning_rate": 1.954701180053293e-06, + "loss": 1.4216, + "step": 1908 + }, + { + "epoch": 1.9609655880842323, + "grad_norm": 2.6597256660461426, + "learning_rate": 1.952797868290826e-06, + "loss": 1.3748, + "step": 1909 + }, + { + "epoch": 1.9619928094504364, + "grad_norm": 2.7966320514678955, + "learning_rate": 1.9508945565283596e-06, + "loss": 1.4237, + "step": 1910 + }, + { + "epoch": 1.9619928094504364, + "eval_loss": 1.8453261852264404, + "eval_runtime": 36.9326, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1910 + }, + { + "epoch": 1.963020030816641, + "grad_norm": 2.7504498958587646, + "learning_rate": 1.948991244765893e-06, + "loss": 1.3832, + "step": 1911 + }, + { + "epoch": 1.9640472521828454, + "grad_norm": 2.7729601860046387, + "learning_rate": 1.9470879330034264e-06, + "loss": 1.4176, + "step": 1912 + }, + { + "epoch": 1.9650744735490497, + "grad_norm": 2.6513381004333496, + "learning_rate": 1.9451846212409593e-06, + "loss": 1.3524, + "step": 1913 + }, + { + "epoch": 1.9661016949152543, + "grad_norm": 2.8147811889648438, + "learning_rate": 1.9432813094784927e-06, + "loss": 1.4119, + "step": 1914 + }, + { + "epoch": 1.9671289162814587, + "grad_norm": 2.7253763675689697, + "learning_rate": 1.941377997716026e-06, + "loss": 1.3458, + "step": 1915 + }, + { + "epoch": 1.968156137647663, + "grad_norm": 2.6741955280303955, + "learning_rate": 1.9394746859535594e-06, + "loss": 1.3764, + "step": 1916 + }, + { + "epoch": 1.9691833590138677, + "grad_norm": 2.785856008529663, + "learning_rate": 1.9375713741910928e-06, + "loss": 1.3484, + "step": 1917 + }, + { + "epoch": 1.9702105803800718, + "grad_norm": 2.743168592453003, + "learning_rate": 1.9356680624286257e-06, + "loss": 1.3779, + "step": 1918 + }, + { + "epoch": 1.9712378017462764, + "grad_norm": 2.827883243560791, + "learning_rate": 1.933764750666159e-06, + "loss": 1.3246, + "step": 1919 + }, + { + "epoch": 1.9722650231124808, + "grad_norm": 2.7379672527313232, + "learning_rate": 1.931861438903693e-06, + "loss": 1.383, + "step": 1920 + }, + { + "epoch": 1.9722650231124808, + "eval_loss": 1.8433737754821777, + "eval_runtime": 37.0421, + "eval_samples_per_second": 26.996, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 1.9732922444786851, + "grad_norm": 2.7514965534210205, + "learning_rate": 1.929958127141226e-06, + "loss": 1.3513, + "step": 1921 + }, + { + "epoch": 1.9743194658448897, + "grad_norm": 2.6764447689056396, + "learning_rate": 1.928054815378759e-06, + "loss": 1.3718, + "step": 1922 + }, + { + "epoch": 1.9753466872110939, + "grad_norm": 2.8592004776000977, + "learning_rate": 1.9261515036162925e-06, + "loss": 1.4479, + "step": 1923 + }, + { + "epoch": 1.9763739085772984, + "grad_norm": 2.8035528659820557, + "learning_rate": 1.924248191853826e-06, + "loss": 1.4222, + "step": 1924 + }, + { + "epoch": 1.9774011299435028, + "grad_norm": 2.7685303688049316, + "learning_rate": 1.9223448800913592e-06, + "loss": 1.3939, + "step": 1925 + }, + { + "epoch": 1.9784283513097072, + "grad_norm": 2.6679201126098633, + "learning_rate": 1.9204415683288926e-06, + "loss": 1.385, + "step": 1926 + }, + { + "epoch": 1.9794555726759118, + "grad_norm": 2.7546098232269287, + "learning_rate": 1.9185382565664255e-06, + "loss": 1.4134, + "step": 1927 + }, + { + "epoch": 1.980482794042116, + "grad_norm": 2.794991970062256, + "learning_rate": 1.916634944803959e-06, + "loss": 1.4212, + "step": 1928 + }, + { + "epoch": 1.9815100154083205, + "grad_norm": 2.767075538635254, + "learning_rate": 1.9147316330414923e-06, + "loss": 1.3979, + "step": 1929 + }, + { + "epoch": 1.9825372367745249, + "grad_norm": 2.7712795734405518, + "learning_rate": 1.9128283212790256e-06, + "loss": 1.3827, + "step": 1930 + }, + { + "epoch": 1.9825372367745249, + "eval_loss": 1.8437753915786743, + "eval_runtime": 37.0885, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1930 + }, + { + "epoch": 1.9835644581407292, + "grad_norm": 2.791165351867676, + "learning_rate": 1.910925009516559e-06, + "loss": 1.4328, + "step": 1931 + }, + { + "epoch": 1.9845916795069338, + "grad_norm": 2.7170767784118652, + "learning_rate": 1.9090216977540924e-06, + "loss": 1.3958, + "step": 1932 + }, + { + "epoch": 1.9856189008731382, + "grad_norm": 2.8529574871063232, + "learning_rate": 1.9071183859916257e-06, + "loss": 1.4379, + "step": 1933 + }, + { + "epoch": 1.9866461222393426, + "grad_norm": 2.8424859046936035, + "learning_rate": 1.9052150742291589e-06, + "loss": 1.4066, + "step": 1934 + }, + { + "epoch": 1.9876733436055471, + "grad_norm": 2.8055100440979004, + "learning_rate": 1.9033117624666922e-06, + "loss": 1.4, + "step": 1935 + }, + { + "epoch": 1.9887005649717513, + "grad_norm": 2.73415470123291, + "learning_rate": 1.9014084507042254e-06, + "loss": 1.3652, + "step": 1936 + }, + { + "epoch": 1.9897277863379559, + "grad_norm": 2.7907590866088867, + "learning_rate": 1.8995051389417587e-06, + "loss": 1.3475, + "step": 1937 + }, + { + "epoch": 1.9907550077041603, + "grad_norm": 2.6994218826293945, + "learning_rate": 1.8976018271792921e-06, + "loss": 1.3428, + "step": 1938 + }, + { + "epoch": 1.9917822290703646, + "grad_norm": 2.8237955570220947, + "learning_rate": 1.8956985154168253e-06, + "loss": 1.3918, + "step": 1939 + }, + { + "epoch": 1.9928094504365692, + "grad_norm": 2.779270648956299, + "learning_rate": 1.8937952036543588e-06, + "loss": 1.3726, + "step": 1940 + }, + { + "epoch": 1.9928094504365692, + "eval_loss": 1.8435344696044922, + "eval_runtime": 37.0797, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1940 + }, + { + "epoch": 1.9938366718027734, + "grad_norm": 2.7808098793029785, + "learning_rate": 1.8918918918918922e-06, + "loss": 1.3368, + "step": 1941 + }, + { + "epoch": 1.994863893168978, + "grad_norm": 2.7355751991271973, + "learning_rate": 1.8899885801294253e-06, + "loss": 1.39, + "step": 1942 + }, + { + "epoch": 1.9958911145351823, + "grad_norm": 2.907468318939209, + "learning_rate": 1.8880852683669587e-06, + "loss": 1.399, + "step": 1943 + }, + { + "epoch": 1.9969183359013867, + "grad_norm": 2.8597185611724854, + "learning_rate": 1.886181956604492e-06, + "loss": 1.3731, + "step": 1944 + }, + { + "epoch": 1.9979455572675913, + "grad_norm": 2.8638176918029785, + "learning_rate": 1.8842786448420252e-06, + "loss": 1.3696, + "step": 1945 + }, + { + "epoch": 1.9989727786337956, + "grad_norm": 2.880169153213501, + "learning_rate": 1.8823753330795586e-06, + "loss": 1.3971, + "step": 1946 + }, + { + "epoch": 2.0, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.8804720213170917e-06, + "loss": 1.3026, + "step": 1947 + }, + { + "epoch": 2.0010272213662046, + "grad_norm": 2.9506170749664307, + "learning_rate": 1.878568709554625e-06, + "loss": 1.3298, + "step": 1948 + }, + { + "epoch": 2.0020544427324087, + "grad_norm": 2.8743488788604736, + "learning_rate": 1.8766653977921585e-06, + "loss": 1.3808, + "step": 1949 + }, + { + "epoch": 2.0030816640986133, + "grad_norm": 2.8676979541778564, + "learning_rate": 1.874762086029692e-06, + "loss": 1.3394, + "step": 1950 + }, + { + "epoch": 2.0030816640986133, + "eval_loss": 1.8473349809646606, + "eval_runtime": 37.093, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1950 + }, + { + "epoch": 2.0041088854648175, + "grad_norm": 3.1344428062438965, + "learning_rate": 1.8728587742672252e-06, + "loss": 1.3774, + "step": 1951 + }, + { + "epoch": 2.005136106831022, + "grad_norm": 3.1567065715789795, + "learning_rate": 1.8709554625047585e-06, + "loss": 1.3863, + "step": 1952 + }, + { + "epoch": 2.0061633281972266, + "grad_norm": 3.109293222427368, + "learning_rate": 1.8690521507422917e-06, + "loss": 1.3196, + "step": 1953 + }, + { + "epoch": 2.007190549563431, + "grad_norm": 3.1035029888153076, + "learning_rate": 1.867148838979825e-06, + "loss": 1.3473, + "step": 1954 + }, + { + "epoch": 2.0082177709296354, + "grad_norm": 3.02809476852417, + "learning_rate": 1.8652455272173584e-06, + "loss": 1.3718, + "step": 1955 + }, + { + "epoch": 2.00924499229584, + "grad_norm": 2.918850898742676, + "learning_rate": 1.8633422154548916e-06, + "loss": 1.3794, + "step": 1956 + }, + { + "epoch": 2.010272213662044, + "grad_norm": 2.89725923538208, + "learning_rate": 1.861438903692425e-06, + "loss": 1.2861, + "step": 1957 + }, + { + "epoch": 2.0112994350282487, + "grad_norm": 3.051868200302124, + "learning_rate": 1.859535591929958e-06, + "loss": 1.3122, + "step": 1958 + }, + { + "epoch": 2.012326656394453, + "grad_norm": 3.1091904640197754, + "learning_rate": 1.8576322801674915e-06, + "loss": 1.3941, + "step": 1959 + }, + { + "epoch": 2.0133538777606574, + "grad_norm": 3.046663761138916, + "learning_rate": 1.855728968405025e-06, + "loss": 1.3293, + "step": 1960 + }, + { + "epoch": 2.0133538777606574, + "eval_loss": 1.851180911064148, + "eval_runtime": 37.0842, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1960 + }, + { + "epoch": 2.014381099126862, + "grad_norm": 3.2143752574920654, + "learning_rate": 1.8538256566425584e-06, + "loss": 1.3431, + "step": 1961 + }, + { + "epoch": 2.015408320493066, + "grad_norm": 3.162942409515381, + "learning_rate": 1.8519223448800915e-06, + "loss": 1.3883, + "step": 1962 + }, + { + "epoch": 2.0164355418592708, + "grad_norm": 3.1271183490753174, + "learning_rate": 1.850019033117625e-06, + "loss": 1.3295, + "step": 1963 + }, + { + "epoch": 2.017462763225475, + "grad_norm": 3.07658314704895, + "learning_rate": 1.848115721355158e-06, + "loss": 1.3323, + "step": 1964 + }, + { + "epoch": 2.0184899845916795, + "grad_norm": 3.075946569442749, + "learning_rate": 1.8462124095926914e-06, + "loss": 1.3524, + "step": 1965 + }, + { + "epoch": 2.019517205957884, + "grad_norm": 3.0359086990356445, + "learning_rate": 1.8443090978302248e-06, + "loss": 1.3578, + "step": 1966 + }, + { + "epoch": 2.0205444273240882, + "grad_norm": 3.1957743167877197, + "learning_rate": 1.842405786067758e-06, + "loss": 1.3909, + "step": 1967 + }, + { + "epoch": 2.021571648690293, + "grad_norm": 3.002462863922119, + "learning_rate": 1.8405024743052913e-06, + "loss": 1.3466, + "step": 1968 + }, + { + "epoch": 2.022598870056497, + "grad_norm": 3.0507266521453857, + "learning_rate": 1.8385991625428247e-06, + "loss": 1.3717, + "step": 1969 + }, + { + "epoch": 2.0236260914227016, + "grad_norm": 3.0719404220581055, + "learning_rate": 1.8366958507803578e-06, + "loss": 1.313, + "step": 1970 + }, + { + "epoch": 2.0236260914227016, + "eval_loss": 1.8496923446655273, + "eval_runtime": 37.1112, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 1970 + }, + { + "epoch": 2.024653312788906, + "grad_norm": 3.068819522857666, + "learning_rate": 1.8347925390178914e-06, + "loss": 1.3614, + "step": 1971 + }, + { + "epoch": 2.0256805341551103, + "grad_norm": 3.005855083465576, + "learning_rate": 1.8328892272554247e-06, + "loss": 1.3007, + "step": 1972 + }, + { + "epoch": 2.026707755521315, + "grad_norm": 3.068032741546631, + "learning_rate": 1.8309859154929579e-06, + "loss": 1.3644, + "step": 1973 + }, + { + "epoch": 2.0277349768875195, + "grad_norm": 3.066153049468994, + "learning_rate": 1.8290826037304913e-06, + "loss": 1.3518, + "step": 1974 + }, + { + "epoch": 2.0287621982537236, + "grad_norm": 3.019429922103882, + "learning_rate": 1.8271792919680246e-06, + "loss": 1.3208, + "step": 1975 + }, + { + "epoch": 2.029789419619928, + "grad_norm": 3.2104897499084473, + "learning_rate": 1.8252759802055578e-06, + "loss": 1.329, + "step": 1976 + }, + { + "epoch": 2.0308166409861323, + "grad_norm": 3.272581100463867, + "learning_rate": 1.8233726684430911e-06, + "loss": 1.3665, + "step": 1977 + }, + { + "epoch": 2.031843862352337, + "grad_norm": 3.2155191898345947, + "learning_rate": 1.8214693566806243e-06, + "loss": 1.3178, + "step": 1978 + }, + { + "epoch": 2.0328710837185415, + "grad_norm": 3.130934238433838, + "learning_rate": 1.8195660449181576e-06, + "loss": 1.3492, + "step": 1979 + }, + { + "epoch": 2.0338983050847457, + "grad_norm": 3.211571216583252, + "learning_rate": 1.817662733155691e-06, + "loss": 1.3477, + "step": 1980 + }, + { + "epoch": 2.0338983050847457, + "eval_loss": 1.8516111373901367, + "eval_runtime": 37.1099, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 1980 + }, + { + "epoch": 2.0349255264509503, + "grad_norm": 2.9799704551696777, + "learning_rate": 1.8157594213932244e-06, + "loss": 1.3469, + "step": 1981 + }, + { + "epoch": 2.0359527478171544, + "grad_norm": 3.0847010612487793, + "learning_rate": 1.8138561096307577e-06, + "loss": 1.3587, + "step": 1982 + }, + { + "epoch": 2.036979969183359, + "grad_norm": 3.0739681720733643, + "learning_rate": 1.811952797868291e-06, + "loss": 1.3053, + "step": 1983 + }, + { + "epoch": 2.0380071905495636, + "grad_norm": 3.0315258502960205, + "learning_rate": 1.8100494861058242e-06, + "loss": 1.3381, + "step": 1984 + }, + { + "epoch": 2.0390344119157677, + "grad_norm": 2.9987258911132812, + "learning_rate": 1.8081461743433576e-06, + "loss": 1.3545, + "step": 1985 + }, + { + "epoch": 2.0400616332819723, + "grad_norm": 2.9383034706115723, + "learning_rate": 1.806242862580891e-06, + "loss": 1.3085, + "step": 1986 + }, + { + "epoch": 2.0410888546481765, + "grad_norm": 3.064450740814209, + "learning_rate": 1.8043395508184241e-06, + "loss": 1.3572, + "step": 1987 + }, + { + "epoch": 2.042116076014381, + "grad_norm": 3.102689027786255, + "learning_rate": 1.8024362390559575e-06, + "loss": 1.3453, + "step": 1988 + }, + { + "epoch": 2.0431432973805856, + "grad_norm": 3.0369763374328613, + "learning_rate": 1.8005329272934906e-06, + "loss": 1.3123, + "step": 1989 + }, + { + "epoch": 2.04417051874679, + "grad_norm": 3.2758162021636963, + "learning_rate": 1.798629615531024e-06, + "loss": 1.4068, + "step": 1990 + }, + { + "epoch": 2.04417051874679, + "eval_loss": 1.8500858545303345, + "eval_runtime": 37.334, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 1990 + }, + { + "epoch": 2.0451977401129944, + "grad_norm": 3.1648876667022705, + "learning_rate": 1.7967263037685576e-06, + "loss": 1.3328, + "step": 1991 + }, + { + "epoch": 2.046224961479199, + "grad_norm": 3.104123830795288, + "learning_rate": 1.794822992006091e-06, + "loss": 1.3297, + "step": 1992 + }, + { + "epoch": 2.047252182845403, + "grad_norm": 3.198483943939209, + "learning_rate": 1.792919680243624e-06, + "loss": 1.3851, + "step": 1993 + }, + { + "epoch": 2.0482794042116077, + "grad_norm": 3.212456226348877, + "learning_rate": 1.7910163684811574e-06, + "loss": 1.3062, + "step": 1994 + }, + { + "epoch": 2.049306625577812, + "grad_norm": 3.058353900909424, + "learning_rate": 1.7891130567186906e-06, + "loss": 1.3347, + "step": 1995 + }, + { + "epoch": 2.0503338469440164, + "grad_norm": 2.963183879852295, + "learning_rate": 1.787209744956224e-06, + "loss": 1.3163, + "step": 1996 + }, + { + "epoch": 2.051361068310221, + "grad_norm": 3.1013290882110596, + "learning_rate": 1.7853064331937573e-06, + "loss": 1.3377, + "step": 1997 + }, + { + "epoch": 2.052388289676425, + "grad_norm": 3.068808078765869, + "learning_rate": 1.7834031214312905e-06, + "loss": 1.3772, + "step": 1998 + }, + { + "epoch": 2.0534155110426298, + "grad_norm": 3.0556447505950928, + "learning_rate": 1.7814998096688238e-06, + "loss": 1.3113, + "step": 1999 + }, + { + "epoch": 2.054442732408834, + "grad_norm": 3.115323543548584, + "learning_rate": 1.779596497906357e-06, + "loss": 1.3499, + "step": 2000 + }, + { + "epoch": 2.054442732408834, + "eval_loss": 1.8396267890930176, + "eval_runtime": 37.4474, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 2000 + }, + { + "epoch": 2.0554699537750385, + "grad_norm": 3.133096694946289, + "learning_rate": 1.7776931861438906e-06, + "loss": 1.3938, + "step": 2001 + }, + { + "epoch": 2.056497175141243, + "grad_norm": 3.259174108505249, + "learning_rate": 1.775789874381424e-06, + "loss": 1.382, + "step": 2002 + }, + { + "epoch": 2.0575243965074472, + "grad_norm": 3.17948317527771, + "learning_rate": 1.7738865626189573e-06, + "loss": 1.3643, + "step": 2003 + }, + { + "epoch": 2.058551617873652, + "grad_norm": 3.0918309688568115, + "learning_rate": 1.7719832508564904e-06, + "loss": 1.3067, + "step": 2004 + }, + { + "epoch": 2.0595788392398564, + "grad_norm": 2.9949889183044434, + "learning_rate": 1.7700799390940238e-06, + "loss": 1.3323, + "step": 2005 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.044935464859009, + "learning_rate": 1.768176627331557e-06, + "loss": 1.359, + "step": 2006 + }, + { + "epoch": 2.061633281972265, + "grad_norm": 2.9965357780456543, + "learning_rate": 1.7662733155690903e-06, + "loss": 1.3456, + "step": 2007 + }, + { + "epoch": 2.0626605033384693, + "grad_norm": 3.092984676361084, + "learning_rate": 1.7643700038066237e-06, + "loss": 1.3831, + "step": 2008 + }, + { + "epoch": 2.063687724704674, + "grad_norm": 3.004263162612915, + "learning_rate": 1.7624666920441568e-06, + "loss": 1.3087, + "step": 2009 + }, + { + "epoch": 2.0647149460708785, + "grad_norm": 3.0408692359924316, + "learning_rate": 1.7605633802816902e-06, + "loss": 1.3058, + "step": 2010 + }, + { + "epoch": 2.0647149460708785, + "eval_loss": 1.8413820266723633, + "eval_runtime": 36.9354, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.0657421674370826, + "grad_norm": 3.155033826828003, + "learning_rate": 1.7586600685192238e-06, + "loss": 1.394, + "step": 2011 + }, + { + "epoch": 2.066769388803287, + "grad_norm": 3.3466567993164062, + "learning_rate": 1.756756756756757e-06, + "loss": 1.401, + "step": 2012 + }, + { + "epoch": 2.0677966101694913, + "grad_norm": 3.219165802001953, + "learning_rate": 1.7548534449942903e-06, + "loss": 1.3564, + "step": 2013 + }, + { + "epoch": 2.068823831535696, + "grad_norm": 3.1733009815216064, + "learning_rate": 1.7529501332318236e-06, + "loss": 1.3446, + "step": 2014 + }, + { + "epoch": 2.0698510529019005, + "grad_norm": 3.1392745971679688, + "learning_rate": 1.7510468214693568e-06, + "loss": 1.3212, + "step": 2015 + }, + { + "epoch": 2.0708782742681047, + "grad_norm": 2.9845967292785645, + "learning_rate": 1.7491435097068902e-06, + "loss": 1.3117, + "step": 2016 + }, + { + "epoch": 2.0719054956343093, + "grad_norm": 3.2182602882385254, + "learning_rate": 1.7472401979444233e-06, + "loss": 1.3539, + "step": 2017 + }, + { + "epoch": 2.0729327170005134, + "grad_norm": 3.0809013843536377, + "learning_rate": 1.7453368861819567e-06, + "loss": 1.2891, + "step": 2018 + }, + { + "epoch": 2.073959938366718, + "grad_norm": 3.0690958499908447, + "learning_rate": 1.74343357441949e-06, + "loss": 1.3381, + "step": 2019 + }, + { + "epoch": 2.0749871597329226, + "grad_norm": 3.091454267501831, + "learning_rate": 1.7415302626570232e-06, + "loss": 1.3874, + "step": 2020 + }, + { + "epoch": 2.0749871597329226, + "eval_loss": 1.8516818284988403, + "eval_runtime": 37.0619, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.0760143810991267, + "grad_norm": 3.039327621459961, + "learning_rate": 1.7396269508945568e-06, + "loss": 1.3295, + "step": 2021 + }, + { + "epoch": 2.0770416024653313, + "grad_norm": 3.0967671871185303, + "learning_rate": 1.7377236391320901e-06, + "loss": 1.3122, + "step": 2022 + }, + { + "epoch": 2.078068823831536, + "grad_norm": 3.228410005569458, + "learning_rate": 1.7358203273696233e-06, + "loss": 1.3766, + "step": 2023 + }, + { + "epoch": 2.07909604519774, + "grad_norm": 3.1973841190338135, + "learning_rate": 1.7339170156071566e-06, + "loss": 1.3325, + "step": 2024 + }, + { + "epoch": 2.0801232665639446, + "grad_norm": 3.131504535675049, + "learning_rate": 1.73201370384469e-06, + "loss": 1.3355, + "step": 2025 + }, + { + "epoch": 2.081150487930149, + "grad_norm": 3.132685661315918, + "learning_rate": 1.7301103920822231e-06, + "loss": 1.3446, + "step": 2026 + }, + { + "epoch": 2.0821777092963534, + "grad_norm": 3.2344162464141846, + "learning_rate": 1.7282070803197565e-06, + "loss": 1.3525, + "step": 2027 + }, + { + "epoch": 2.083204930662558, + "grad_norm": 3.1193928718566895, + "learning_rate": 1.7263037685572897e-06, + "loss": 1.3415, + "step": 2028 + }, + { + "epoch": 2.084232152028762, + "grad_norm": 3.164365291595459, + "learning_rate": 1.724400456794823e-06, + "loss": 1.3652, + "step": 2029 + }, + { + "epoch": 2.0852593733949667, + "grad_norm": 3.0825226306915283, + "learning_rate": 1.7224971450323564e-06, + "loss": 1.3242, + "step": 2030 + }, + { + "epoch": 2.0852593733949667, + "eval_loss": 1.8433457612991333, + "eval_runtime": 37.0921, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.086286594761171, + "grad_norm": 3.1651787757873535, + "learning_rate": 1.72059383326989e-06, + "loss": 1.333, + "step": 2031 + }, + { + "epoch": 2.0873138161273754, + "grad_norm": 3.295436382293701, + "learning_rate": 1.7186905215074231e-06, + "loss": 1.4064, + "step": 2032 + }, + { + "epoch": 2.08834103749358, + "grad_norm": 3.177628517150879, + "learning_rate": 1.7167872097449565e-06, + "loss": 1.3173, + "step": 2033 + }, + { + "epoch": 2.089368258859784, + "grad_norm": 3.112971544265747, + "learning_rate": 1.7148838979824896e-06, + "loss": 1.3401, + "step": 2034 + }, + { + "epoch": 2.0903954802259888, + "grad_norm": 3.150045871734619, + "learning_rate": 1.712980586220023e-06, + "loss": 1.3371, + "step": 2035 + }, + { + "epoch": 2.091422701592193, + "grad_norm": 3.0645205974578857, + "learning_rate": 1.7110772744575563e-06, + "loss": 1.3213, + "step": 2036 + }, + { + "epoch": 2.0924499229583975, + "grad_norm": 3.230185031890869, + "learning_rate": 1.7091739626950895e-06, + "loss": 1.3317, + "step": 2037 + }, + { + "epoch": 2.093477144324602, + "grad_norm": 3.1509249210357666, + "learning_rate": 1.7072706509326229e-06, + "loss": 1.3706, + "step": 2038 + }, + { + "epoch": 2.0945043656908062, + "grad_norm": 3.185981273651123, + "learning_rate": 1.7053673391701562e-06, + "loss": 1.3398, + "step": 2039 + }, + { + "epoch": 2.095531587057011, + "grad_norm": 3.000025510787964, + "learning_rate": 1.7034640274076894e-06, + "loss": 1.3197, + "step": 2040 + }, + { + "epoch": 2.095531587057011, + "eval_loss": 1.8413926362991333, + "eval_runtime": 37.0971, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 2040 + }, + { + "epoch": 2.0965588084232154, + "grad_norm": 3.156095027923584, + "learning_rate": 1.701560715645223e-06, + "loss": 1.3864, + "step": 2041 + }, + { + "epoch": 2.0975860297894195, + "grad_norm": 2.970560073852539, + "learning_rate": 1.6996574038827563e-06, + "loss": 1.3016, + "step": 2042 + }, + { + "epoch": 2.098613251155624, + "grad_norm": 3.141657829284668, + "learning_rate": 1.6977540921202895e-06, + "loss": 1.3488, + "step": 2043 + }, + { + "epoch": 2.0996404725218283, + "grad_norm": 3.125731945037842, + "learning_rate": 1.6958507803578228e-06, + "loss": 1.3275, + "step": 2044 + }, + { + "epoch": 2.100667693888033, + "grad_norm": 3.2703487873077393, + "learning_rate": 1.693947468595356e-06, + "loss": 1.3577, + "step": 2045 + }, + { + "epoch": 2.1016949152542375, + "grad_norm": 3.1855714321136475, + "learning_rate": 1.6920441568328893e-06, + "loss": 1.3671, + "step": 2046 + }, + { + "epoch": 2.1027221366204416, + "grad_norm": 3.0699303150177, + "learning_rate": 1.6901408450704227e-06, + "loss": 1.3066, + "step": 2047 + }, + { + "epoch": 2.103749357986646, + "grad_norm": 3.2417423725128174, + "learning_rate": 1.6882375333079559e-06, + "loss": 1.3274, + "step": 2048 + }, + { + "epoch": 2.1047765793528503, + "grad_norm": 3.265108108520508, + "learning_rate": 1.6863342215454892e-06, + "loss": 1.3924, + "step": 2049 + }, + { + "epoch": 2.105803800719055, + "grad_norm": 3.2078635692596436, + "learning_rate": 1.6844309097830226e-06, + "loss": 1.3782, + "step": 2050 + }, + { + "epoch": 2.105803800719055, + "eval_loss": 1.8441476821899414, + "eval_runtime": 37.1042, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.1068310220852595, + "grad_norm": 3.180485248565674, + "learning_rate": 1.682527598020556e-06, + "loss": 1.3613, + "step": 2051 + }, + { + "epoch": 2.1078582434514637, + "grad_norm": 3.110530376434326, + "learning_rate": 1.6806242862580893e-06, + "loss": 1.376, + "step": 2052 + }, + { + "epoch": 2.1088854648176683, + "grad_norm": 2.997999429702759, + "learning_rate": 1.6787209744956227e-06, + "loss": 1.3172, + "step": 2053 + }, + { + "epoch": 2.109912686183873, + "grad_norm": 3.0154783725738525, + "learning_rate": 1.6768176627331558e-06, + "loss": 1.3015, + "step": 2054 + }, + { + "epoch": 2.110939907550077, + "grad_norm": 2.992138624191284, + "learning_rate": 1.6749143509706892e-06, + "loss": 1.3517, + "step": 2055 + }, + { + "epoch": 2.1119671289162816, + "grad_norm": 3.0572457313537598, + "learning_rate": 1.6730110392082225e-06, + "loss": 1.3301, + "step": 2056 + }, + { + "epoch": 2.1129943502824857, + "grad_norm": 2.9924540519714355, + "learning_rate": 1.6711077274457557e-06, + "loss": 1.3534, + "step": 2057 + }, + { + "epoch": 2.1140215716486903, + "grad_norm": 2.9569015502929688, + "learning_rate": 1.669204415683289e-06, + "loss": 1.2972, + "step": 2058 + }, + { + "epoch": 2.115048793014895, + "grad_norm": 3.064371109008789, + "learning_rate": 1.6673011039208222e-06, + "loss": 1.3553, + "step": 2059 + }, + { + "epoch": 2.116076014381099, + "grad_norm": 3.0954525470733643, + "learning_rate": 1.6653977921583556e-06, + "loss": 1.3421, + "step": 2060 + }, + { + "epoch": 2.116076014381099, + "eval_loss": 1.8407723903656006, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 2060 + }, + { + "epoch": 2.1171032357473036, + "grad_norm": 3.2661712169647217, + "learning_rate": 1.663494480395889e-06, + "loss": 1.3582, + "step": 2061 + }, + { + "epoch": 2.118130457113508, + "grad_norm": 3.1663613319396973, + "learning_rate": 1.6615911686334225e-06, + "loss": 1.3203, + "step": 2062 + }, + { + "epoch": 2.1191576784797124, + "grad_norm": 3.0864570140838623, + "learning_rate": 1.6596878568709557e-06, + "loss": 1.3431, + "step": 2063 + }, + { + "epoch": 2.120184899845917, + "grad_norm": 3.1531989574432373, + "learning_rate": 1.657784545108489e-06, + "loss": 1.3275, + "step": 2064 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.2258596420288086, + "learning_rate": 1.6558812333460222e-06, + "loss": 1.3711, + "step": 2065 + }, + { + "epoch": 2.1222393425783257, + "grad_norm": 3.2881453037261963, + "learning_rate": 1.6539779215835555e-06, + "loss": 1.3688, + "step": 2066 + }, + { + "epoch": 2.12326656394453, + "grad_norm": 3.0734169483184814, + "learning_rate": 1.652074609821089e-06, + "loss": 1.3277, + "step": 2067 + }, + { + "epoch": 2.1242937853107344, + "grad_norm": 2.9943618774414062, + "learning_rate": 1.650171298058622e-06, + "loss": 1.3268, + "step": 2068 + }, + { + "epoch": 2.125321006676939, + "grad_norm": 3.034625291824341, + "learning_rate": 1.6482679862961554e-06, + "loss": 1.3265, + "step": 2069 + }, + { + "epoch": 2.126348228043143, + "grad_norm": 3.0360612869262695, + "learning_rate": 1.6463646745336886e-06, + "loss": 1.3256, + "step": 2070 + }, + { + "epoch": 2.126348228043143, + "eval_loss": 1.8443471193313599, + "eval_runtime": 37.0917, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.1273754494093478, + "grad_norm": 3.0917110443115234, + "learning_rate": 1.644461362771222e-06, + "loss": 1.3099, + "step": 2071 + }, + { + "epoch": 2.128402670775552, + "grad_norm": 3.1142187118530273, + "learning_rate": 1.6425580510087555e-06, + "loss": 1.3421, + "step": 2072 + }, + { + "epoch": 2.1294298921417565, + "grad_norm": 3.2128524780273438, + "learning_rate": 1.6406547392462889e-06, + "loss": 1.3123, + "step": 2073 + }, + { + "epoch": 2.130457113507961, + "grad_norm": 3.060520887374878, + "learning_rate": 1.638751427483822e-06, + "loss": 1.3069, + "step": 2074 + }, + { + "epoch": 2.131484334874165, + "grad_norm": 3.092120885848999, + "learning_rate": 1.6368481157213554e-06, + "loss": 1.3537, + "step": 2075 + }, + { + "epoch": 2.13251155624037, + "grad_norm": 3.2490074634552, + "learning_rate": 1.6349448039588885e-06, + "loss": 1.3324, + "step": 2076 + }, + { + "epoch": 2.1335387776065744, + "grad_norm": 3.2517645359039307, + "learning_rate": 1.6330414921964219e-06, + "loss": 1.385, + "step": 2077 + }, + { + "epoch": 2.1345659989727785, + "grad_norm": 3.1669421195983887, + "learning_rate": 1.6311381804339553e-06, + "loss": 1.3054, + "step": 2078 + }, + { + "epoch": 2.135593220338983, + "grad_norm": 3.2323193550109863, + "learning_rate": 1.6292348686714884e-06, + "loss": 1.3488, + "step": 2079 + }, + { + "epoch": 2.1366204417051873, + "grad_norm": 3.1058294773101807, + "learning_rate": 1.6273315569090218e-06, + "loss": 1.3256, + "step": 2080 + }, + { + "epoch": 2.1366204417051873, + "eval_loss": 1.8446909189224243, + "eval_runtime": 37.1231, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.137647663071392, + "grad_norm": 3.1662075519561768, + "learning_rate": 1.625428245146555e-06, + "loss": 1.4073, + "step": 2081 + }, + { + "epoch": 2.1386748844375965, + "grad_norm": 3.083482503890991, + "learning_rate": 1.6235249333840885e-06, + "loss": 1.3897, + "step": 2082 + }, + { + "epoch": 2.1397021058038006, + "grad_norm": 3.040388345718384, + "learning_rate": 1.6216216216216219e-06, + "loss": 1.349, + "step": 2083 + }, + { + "epoch": 2.140729327170005, + "grad_norm": 3.124114513397217, + "learning_rate": 1.6197183098591552e-06, + "loss": 1.3732, + "step": 2084 + }, + { + "epoch": 2.1417565485362093, + "grad_norm": 3.2662036418914795, + "learning_rate": 1.6178149980966884e-06, + "loss": 1.3522, + "step": 2085 + }, + { + "epoch": 2.142783769902414, + "grad_norm": 3.149472951889038, + "learning_rate": 1.6159116863342217e-06, + "loss": 1.3179, + "step": 2086 + }, + { + "epoch": 2.1438109912686185, + "grad_norm": 3.256852865219116, + "learning_rate": 1.6140083745717549e-06, + "loss": 1.3897, + "step": 2087 + }, + { + "epoch": 2.1448382126348227, + "grad_norm": 3.1227493286132812, + "learning_rate": 1.6121050628092882e-06, + "loss": 1.3104, + "step": 2088 + }, + { + "epoch": 2.1458654340010272, + "grad_norm": 3.3213047981262207, + "learning_rate": 1.6102017510468216e-06, + "loss": 1.3829, + "step": 2089 + }, + { + "epoch": 2.146892655367232, + "grad_norm": 3.160428047180176, + "learning_rate": 1.6082984392843548e-06, + "loss": 1.2939, + "step": 2090 + }, + { + "epoch": 2.146892655367232, + "eval_loss": 1.8534585237503052, + "eval_runtime": 37.3333, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.147919876733436, + "grad_norm": 3.224741220474243, + "learning_rate": 1.6063951275218881e-06, + "loss": 1.3418, + "step": 2091 + }, + { + "epoch": 2.1489470980996406, + "grad_norm": 3.3884503841400146, + "learning_rate": 1.6044918157594217e-06, + "loss": 1.395, + "step": 2092 + }, + { + "epoch": 2.1499743194658447, + "grad_norm": 3.1761722564697266, + "learning_rate": 1.6025885039969548e-06, + "loss": 1.3208, + "step": 2093 + }, + { + "epoch": 2.1510015408320493, + "grad_norm": 3.0476465225219727, + "learning_rate": 1.6006851922344882e-06, + "loss": 1.3465, + "step": 2094 + }, + { + "epoch": 2.152028762198254, + "grad_norm": 3.1662650108337402, + "learning_rate": 1.5987818804720216e-06, + "loss": 1.3645, + "step": 2095 + }, + { + "epoch": 2.153055983564458, + "grad_norm": 3.1860709190368652, + "learning_rate": 1.5968785687095547e-06, + "loss": 1.361, + "step": 2096 + }, + { + "epoch": 2.1540832049306626, + "grad_norm": 3.104501962661743, + "learning_rate": 1.594975256947088e-06, + "loss": 1.3135, + "step": 2097 + }, + { + "epoch": 2.1551104262968668, + "grad_norm": 3.067155599594116, + "learning_rate": 1.5930719451846212e-06, + "loss": 1.3693, + "step": 2098 + }, + { + "epoch": 2.1561376476630714, + "grad_norm": 3.195204019546509, + "learning_rate": 1.5911686334221546e-06, + "loss": 1.3143, + "step": 2099 + }, + { + "epoch": 2.157164869029276, + "grad_norm": 3.159278154373169, + "learning_rate": 1.589265321659688e-06, + "loss": 1.3607, + "step": 2100 + }, + { + "epoch": 2.157164869029276, + "eval_loss": 1.8502479791641235, + "eval_runtime": 37.4583, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.15819209039548, + "grad_norm": 2.973191261291504, + "learning_rate": 1.5873620098972211e-06, + "loss": 1.2776, + "step": 2101 + }, + { + "epoch": 2.1592193117616847, + "grad_norm": 3.0078418254852295, + "learning_rate": 1.5854586981347547e-06, + "loss": 1.3173, + "step": 2102 + }, + { + "epoch": 2.1602465331278893, + "grad_norm": 3.2115561962127686, + "learning_rate": 1.583555386372288e-06, + "loss": 1.3621, + "step": 2103 + }, + { + "epoch": 2.1612737544940934, + "grad_norm": 3.234210968017578, + "learning_rate": 1.5816520746098212e-06, + "loss": 1.374, + "step": 2104 + }, + { + "epoch": 2.162300975860298, + "grad_norm": 3.195405960083008, + "learning_rate": 1.5797487628473546e-06, + "loss": 1.3402, + "step": 2105 + }, + { + "epoch": 2.163328197226502, + "grad_norm": 3.016078233718872, + "learning_rate": 1.577845451084888e-06, + "loss": 1.3036, + "step": 2106 + }, + { + "epoch": 2.1643554185927067, + "grad_norm": 3.2147345542907715, + "learning_rate": 1.575942139322421e-06, + "loss": 1.3511, + "step": 2107 + }, + { + "epoch": 2.1653826399589113, + "grad_norm": 2.980320692062378, + "learning_rate": 1.5740388275599544e-06, + "loss": 1.2635, + "step": 2108 + }, + { + "epoch": 2.1664098613251155, + "grad_norm": 3.0693414211273193, + "learning_rate": 1.5721355157974876e-06, + "loss": 1.3435, + "step": 2109 + }, + { + "epoch": 2.16743708269132, + "grad_norm": 3.1445682048797607, + "learning_rate": 1.570232204035021e-06, + "loss": 1.3294, + "step": 2110 + }, + { + "epoch": 2.16743708269132, + "eval_loss": 1.8479558229446411, + "eval_runtime": 36.9467, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.168464304057524, + "grad_norm": 3.1309633255004883, + "learning_rate": 1.5683288922725543e-06, + "loss": 1.3312, + "step": 2111 + }, + { + "epoch": 2.169491525423729, + "grad_norm": 3.0906994342803955, + "learning_rate": 1.5664255805100879e-06, + "loss": 1.342, + "step": 2112 + }, + { + "epoch": 2.1705187467899334, + "grad_norm": 3.1987171173095703, + "learning_rate": 1.564522268747621e-06, + "loss": 1.3358, + "step": 2113 + }, + { + "epoch": 2.1715459681561375, + "grad_norm": 3.221889019012451, + "learning_rate": 1.5626189569851544e-06, + "loss": 1.3484, + "step": 2114 + }, + { + "epoch": 2.172573189522342, + "grad_norm": 3.1508629322052, + "learning_rate": 1.5607156452226875e-06, + "loss": 1.2785, + "step": 2115 + }, + { + "epoch": 2.1736004108885463, + "grad_norm": 3.115274667739868, + "learning_rate": 1.558812333460221e-06, + "loss": 1.3463, + "step": 2116 + }, + { + "epoch": 2.174627632254751, + "grad_norm": 3.0454249382019043, + "learning_rate": 1.5569090216977543e-06, + "loss": 1.3534, + "step": 2117 + }, + { + "epoch": 2.1756548536209555, + "grad_norm": 3.20078182220459, + "learning_rate": 1.5550057099352874e-06, + "loss": 1.3469, + "step": 2118 + }, + { + "epoch": 2.1766820749871596, + "grad_norm": 3.091568946838379, + "learning_rate": 1.5531023981728208e-06, + "loss": 1.3547, + "step": 2119 + }, + { + "epoch": 2.177709296353364, + "grad_norm": 3.0242369174957275, + "learning_rate": 1.5511990864103542e-06, + "loss": 1.3198, + "step": 2120 + }, + { + "epoch": 2.177709296353364, + "eval_loss": 1.8469057083129883, + "eval_runtime": 37.0881, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.1787365177195683, + "grad_norm": 3.1775739192962646, + "learning_rate": 1.5492957746478873e-06, + "loss": 1.3491, + "step": 2121 + }, + { + "epoch": 2.179763739085773, + "grad_norm": 3.239668846130371, + "learning_rate": 1.5473924628854209e-06, + "loss": 1.3641, + "step": 2122 + }, + { + "epoch": 2.1807909604519775, + "grad_norm": 3.194678544998169, + "learning_rate": 1.5454891511229542e-06, + "loss": 1.3317, + "step": 2123 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.129948377609253, + "learning_rate": 1.5435858393604874e-06, + "loss": 1.3405, + "step": 2124 + }, + { + "epoch": 2.1828454031843862, + "grad_norm": 3.2040436267852783, + "learning_rate": 1.5416825275980208e-06, + "loss": 1.3596, + "step": 2125 + }, + { + "epoch": 2.183872624550591, + "grad_norm": 3.2529456615448, + "learning_rate": 1.539779215835554e-06, + "loss": 1.3171, + "step": 2126 + }, + { + "epoch": 2.184899845916795, + "grad_norm": 3.2071499824523926, + "learning_rate": 1.5378759040730873e-06, + "loss": 1.318, + "step": 2127 + }, + { + "epoch": 2.1859270672829996, + "grad_norm": 3.2709555625915527, + "learning_rate": 1.5359725923106206e-06, + "loss": 1.3811, + "step": 2128 + }, + { + "epoch": 2.1869542886492037, + "grad_norm": 3.244680643081665, + "learning_rate": 1.5340692805481538e-06, + "loss": 1.3539, + "step": 2129 + }, + { + "epoch": 2.1879815100154083, + "grad_norm": 3.068683624267578, + "learning_rate": 1.5321659687856871e-06, + "loss": 1.3207, + "step": 2130 + }, + { + "epoch": 2.1879815100154083, + "eval_loss": 1.8478527069091797, + "eval_runtime": 37.138, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.189008731381613, + "grad_norm": 3.1829938888549805, + "learning_rate": 1.5302626570232205e-06, + "loss": 1.3127, + "step": 2131 + }, + { + "epoch": 2.190035952747817, + "grad_norm": 3.060347080230713, + "learning_rate": 1.5283593452607539e-06, + "loss": 1.287, + "step": 2132 + }, + { + "epoch": 2.1910631741140216, + "grad_norm": 3.3016905784606934, + "learning_rate": 1.5264560334982872e-06, + "loss": 1.3838, + "step": 2133 + }, + { + "epoch": 2.1920903954802258, + "grad_norm": 3.1945807933807373, + "learning_rate": 1.5245527217358206e-06, + "loss": 1.3311, + "step": 2134 + }, + { + "epoch": 2.1931176168464304, + "grad_norm": 3.182347059249878, + "learning_rate": 1.5226494099733537e-06, + "loss": 1.3544, + "step": 2135 + }, + { + "epoch": 2.194144838212635, + "grad_norm": 3.069206476211548, + "learning_rate": 1.520746098210887e-06, + "loss": 1.3148, + "step": 2136 + }, + { + "epoch": 2.195172059578839, + "grad_norm": 3.1632168292999268, + "learning_rate": 1.5188427864484205e-06, + "loss": 1.3857, + "step": 2137 + }, + { + "epoch": 2.1961992809450437, + "grad_norm": 3.055793523788452, + "learning_rate": 1.5169394746859536e-06, + "loss": 1.3116, + "step": 2138 + }, + { + "epoch": 2.1972265023112483, + "grad_norm": 3.1330678462982178, + "learning_rate": 1.515036162923487e-06, + "loss": 1.3214, + "step": 2139 + }, + { + "epoch": 2.1982537236774524, + "grad_norm": 3.118450880050659, + "learning_rate": 1.5131328511610201e-06, + "loss": 1.3479, + "step": 2140 + }, + { + "epoch": 2.1982537236774524, + "eval_loss": 1.8538312911987305, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.199280945043657, + "grad_norm": 3.249545097351074, + "learning_rate": 1.5112295393985535e-06, + "loss": 1.379, + "step": 2141 + }, + { + "epoch": 2.200308166409861, + "grad_norm": 3.1331286430358887, + "learning_rate": 1.509326227636087e-06, + "loss": 1.3621, + "step": 2142 + }, + { + "epoch": 2.2013353877760657, + "grad_norm": 3.150987148284912, + "learning_rate": 1.5074229158736204e-06, + "loss": 1.291, + "step": 2143 + }, + { + "epoch": 2.2023626091422703, + "grad_norm": 3.1020572185516357, + "learning_rate": 1.5055196041111536e-06, + "loss": 1.3381, + "step": 2144 + }, + { + "epoch": 2.2033898305084745, + "grad_norm": 3.129390239715576, + "learning_rate": 1.503616292348687e-06, + "loss": 1.3405, + "step": 2145 + }, + { + "epoch": 2.204417051874679, + "grad_norm": 3.2193870544433594, + "learning_rate": 1.50171298058622e-06, + "loss": 1.3744, + "step": 2146 + }, + { + "epoch": 2.205444273240883, + "grad_norm": 3.043828248977661, + "learning_rate": 1.4998096688237535e-06, + "loss": 1.2782, + "step": 2147 + }, + { + "epoch": 2.206471494607088, + "grad_norm": 3.2130300998687744, + "learning_rate": 1.4979063570612868e-06, + "loss": 1.3625, + "step": 2148 + }, + { + "epoch": 2.2074987159732924, + "grad_norm": 3.250370979309082, + "learning_rate": 1.49600304529882e-06, + "loss": 1.3396, + "step": 2149 + }, + { + "epoch": 2.2085259373394965, + "grad_norm": 3.0954599380493164, + "learning_rate": 1.4940997335363533e-06, + "loss": 1.3187, + "step": 2150 + }, + { + "epoch": 2.2085259373394965, + "eval_loss": 1.8469802141189575, + "eval_runtime": 37.2109, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 2150 + }, + { + "epoch": 2.209553158705701, + "grad_norm": 3.153545379638672, + "learning_rate": 1.4921964217738865e-06, + "loss": 1.3488, + "step": 2151 + }, + { + "epoch": 2.2105803800719057, + "grad_norm": 3.1206071376800537, + "learning_rate": 1.49029311001142e-06, + "loss": 1.336, + "step": 2152 + }, + { + "epoch": 2.21160760143811, + "grad_norm": 3.151784896850586, + "learning_rate": 1.4883897982489534e-06, + "loss": 1.3613, + "step": 2153 + }, + { + "epoch": 2.2126348228043144, + "grad_norm": 3.18965744972229, + "learning_rate": 1.4864864864864868e-06, + "loss": 1.3698, + "step": 2154 + }, + { + "epoch": 2.2136620441705186, + "grad_norm": 3.108281135559082, + "learning_rate": 1.48458317472402e-06, + "loss": 1.3287, + "step": 2155 + }, + { + "epoch": 2.214689265536723, + "grad_norm": 3.2309465408325195, + "learning_rate": 1.4826798629615533e-06, + "loss": 1.3425, + "step": 2156 + }, + { + "epoch": 2.2157164869029278, + "grad_norm": 3.0829577445983887, + "learning_rate": 1.4807765511990864e-06, + "loss": 1.2917, + "step": 2157 + }, + { + "epoch": 2.216743708269132, + "grad_norm": 3.1417033672332764, + "learning_rate": 1.4788732394366198e-06, + "loss": 1.3113, + "step": 2158 + }, + { + "epoch": 2.2177709296353365, + "grad_norm": 3.1996822357177734, + "learning_rate": 1.4769699276741532e-06, + "loss": 1.3248, + "step": 2159 + }, + { + "epoch": 2.2187981510015407, + "grad_norm": 3.262814998626709, + "learning_rate": 1.4750666159116863e-06, + "loss": 1.3186, + "step": 2160 + }, + { + "epoch": 2.2187981510015407, + "eval_loss": 1.844687819480896, + "eval_runtime": 37.2211, + "eval_samples_per_second": 26.866, + "eval_steps_per_second": 1.128, + "step": 2160 + }, + { + "epoch": 2.2198253723677452, + "grad_norm": 3.0836942195892334, + "learning_rate": 1.4731633041492197e-06, + "loss": 1.2935, + "step": 2161 + }, + { + "epoch": 2.22085259373395, + "grad_norm": 3.1720023155212402, + "learning_rate": 1.4712599923867528e-06, + "loss": 1.3622, + "step": 2162 + }, + { + "epoch": 2.221879815100154, + "grad_norm": 3.2829599380493164, + "learning_rate": 1.4693566806242864e-06, + "loss": 1.3794, + "step": 2163 + }, + { + "epoch": 2.2229070364663586, + "grad_norm": 3.045949935913086, + "learning_rate": 1.4674533688618198e-06, + "loss": 1.3054, + "step": 2164 + }, + { + "epoch": 2.2239342578325627, + "grad_norm": 3.1561572551727295, + "learning_rate": 1.4655500570993531e-06, + "loss": 1.3289, + "step": 2165 + }, + { + "epoch": 2.2249614791987673, + "grad_norm": 3.0877439975738525, + "learning_rate": 1.4636467453368863e-06, + "loss": 1.329, + "step": 2166 + }, + { + "epoch": 2.225988700564972, + "grad_norm": 3.1330180168151855, + "learning_rate": 1.4617434335744197e-06, + "loss": 1.3853, + "step": 2167 + }, + { + "epoch": 2.227015921931176, + "grad_norm": 3.231693744659424, + "learning_rate": 1.4598401218119528e-06, + "loss": 1.378, + "step": 2168 + }, + { + "epoch": 2.2280431432973806, + "grad_norm": 3.202930212020874, + "learning_rate": 1.4579368100494862e-06, + "loss": 1.3549, + "step": 2169 + }, + { + "epoch": 2.2290703646635848, + "grad_norm": 3.164461135864258, + "learning_rate": 1.4560334982870195e-06, + "loss": 1.3365, + "step": 2170 + }, + { + "epoch": 2.2290703646635848, + "eval_loss": 1.8483937978744507, + "eval_runtime": 37.1839, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.2300975860297894, + "grad_norm": 3.1866867542266846, + "learning_rate": 1.4541301865245527e-06, + "loss": 1.3503, + "step": 2171 + }, + { + "epoch": 2.231124807395994, + "grad_norm": 3.2202508449554443, + "learning_rate": 1.452226874762086e-06, + "loss": 1.354, + "step": 2172 + }, + { + "epoch": 2.232152028762198, + "grad_norm": 3.190744161605835, + "learning_rate": 1.4503235629996196e-06, + "loss": 1.3073, + "step": 2173 + }, + { + "epoch": 2.2331792501284027, + "grad_norm": 3.2158405780792236, + "learning_rate": 1.4484202512371528e-06, + "loss": 1.3731, + "step": 2174 + }, + { + "epoch": 2.2342064714946073, + "grad_norm": 3.0294954776763916, + "learning_rate": 1.4465169394746861e-06, + "loss": 1.2772, + "step": 2175 + }, + { + "epoch": 2.2352336928608114, + "grad_norm": 3.2871217727661133, + "learning_rate": 1.4446136277122195e-06, + "loss": 1.3888, + "step": 2176 + }, + { + "epoch": 2.236260914227016, + "grad_norm": 3.0494141578674316, + "learning_rate": 1.4427103159497526e-06, + "loss": 1.2991, + "step": 2177 + }, + { + "epoch": 2.23728813559322, + "grad_norm": 3.122699022293091, + "learning_rate": 1.440807004187286e-06, + "loss": 1.3186, + "step": 2178 + }, + { + "epoch": 2.2383153569594247, + "grad_norm": 3.119854211807251, + "learning_rate": 1.4389036924248192e-06, + "loss": 1.3421, + "step": 2179 + }, + { + "epoch": 2.2393425783256293, + "grad_norm": 3.2588119506835938, + "learning_rate": 1.4370003806623525e-06, + "loss": 1.3657, + "step": 2180 + }, + { + "epoch": 2.2393425783256293, + "eval_loss": 1.8473975658416748, + "eval_runtime": 37.1561, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.2403697996918335, + "grad_norm": 3.138901948928833, + "learning_rate": 1.4350970688998859e-06, + "loss": 1.3273, + "step": 2181 + }, + { + "epoch": 2.241397021058038, + "grad_norm": 3.2792508602142334, + "learning_rate": 1.433193757137419e-06, + "loss": 1.3763, + "step": 2182 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.065627098083496, + "learning_rate": 1.4312904453749526e-06, + "loss": 1.3292, + "step": 2183 + }, + { + "epoch": 2.243451463790447, + "grad_norm": 3.101377010345459, + "learning_rate": 1.429387133612486e-06, + "loss": 1.2957, + "step": 2184 + }, + { + "epoch": 2.2444786851566514, + "grad_norm": 3.140158176422119, + "learning_rate": 1.4274838218500191e-06, + "loss": 1.3406, + "step": 2185 + }, + { + "epoch": 2.2455059065228555, + "grad_norm": 3.2498116493225098, + "learning_rate": 1.4255805100875525e-06, + "loss": 1.3499, + "step": 2186 + }, + { + "epoch": 2.24653312788906, + "grad_norm": 3.2681965827941895, + "learning_rate": 1.4236771983250858e-06, + "loss": 1.407, + "step": 2187 + }, + { + "epoch": 2.2475603492552647, + "grad_norm": 3.2457778453826904, + "learning_rate": 1.421773886562619e-06, + "loss": 1.3278, + "step": 2188 + }, + { + "epoch": 2.248587570621469, + "grad_norm": 3.099942207336426, + "learning_rate": 1.4198705748001524e-06, + "loss": 1.3114, + "step": 2189 + }, + { + "epoch": 2.2496147919876734, + "grad_norm": 3.090271234512329, + "learning_rate": 1.4179672630376855e-06, + "loss": 1.3285, + "step": 2190 + }, + { + "epoch": 2.2496147919876734, + "eval_loss": 1.8479118347167969, + "eval_runtime": 37.2331, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.2506420133538776, + "grad_norm": 3.0344860553741455, + "learning_rate": 1.4160639512752189e-06, + "loss": 1.3104, + "step": 2191 + }, + { + "epoch": 2.251669234720082, + "grad_norm": 3.087860584259033, + "learning_rate": 1.4141606395127522e-06, + "loss": 1.3167, + "step": 2192 + }, + { + "epoch": 2.2526964560862868, + "grad_norm": 3.0974831581115723, + "learning_rate": 1.4122573277502858e-06, + "loss": 1.2926, + "step": 2193 + }, + { + "epoch": 2.253723677452491, + "grad_norm": 3.228492498397827, + "learning_rate": 1.410354015987819e-06, + "loss": 1.3541, + "step": 2194 + }, + { + "epoch": 2.2547508988186955, + "grad_norm": 3.1761956214904785, + "learning_rate": 1.4084507042253523e-06, + "loss": 1.3342, + "step": 2195 + }, + { + "epoch": 2.2557781201848996, + "grad_norm": 3.1890695095062256, + "learning_rate": 1.4065473924628855e-06, + "loss": 1.362, + "step": 2196 + }, + { + "epoch": 2.2568053415511042, + "grad_norm": 3.1543149948120117, + "learning_rate": 1.4046440807004188e-06, + "loss": 1.329, + "step": 2197 + }, + { + "epoch": 2.257832562917309, + "grad_norm": 3.2313342094421387, + "learning_rate": 1.4027407689379522e-06, + "loss": 1.3256, + "step": 2198 + }, + { + "epoch": 2.258859784283513, + "grad_norm": 3.130805253982544, + "learning_rate": 1.4008374571754854e-06, + "loss": 1.3047, + "step": 2199 + }, + { + "epoch": 2.2598870056497176, + "grad_norm": 3.0488216876983643, + "learning_rate": 1.3989341454130187e-06, + "loss": 1.3461, + "step": 2200 + }, + { + "epoch": 2.2598870056497176, + "eval_loss": 1.8517944812774658, + "eval_runtime": 37.4528, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 2200 + } + ], + "logging_steps": 1, + "max_steps": 2919, + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2200/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..1519f5698077f1a839cba1775dd60ae3dec27e89 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2300/trainer_state.json @@ -0,0 +1,17973 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.362609142270159, + "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.001027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7522, + "step": 1 + }, + { + "epoch": 0.002054442732408834, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7541, + "step": 2 + }, + { + "epoch": 0.0030816640986132513, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8031, + "step": 3 + }, + { + "epoch": 0.004108885464817668, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8062, + "step": 4 + }, + { + "epoch": 0.005136106831022085, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7733, + "step": 5 + }, + { + "epoch": 0.0061633281972265025, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7949, + "step": 6 + }, + { + "epoch": 0.007190549563430919, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7657, + "step": 7 + }, + { + "epoch": 0.008217770929635337, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.803, + "step": 8 + }, + { + "epoch": 0.009244992295839754, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7793, + "step": 9 + }, + { + "epoch": 0.01027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7935, + "step": 10 + }, + { + "epoch": 0.01027221366204417, + "eval_loss": 2.8114004135131836, + "eval_runtime": 37.8463, + "eval_samples_per_second": 26.423, + "eval_steps_per_second": 1.11, + "step": 10 + }, + { + "epoch": 0.011299435028248588, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7546, + "step": 11 + }, + { + "epoch": 0.012326656394453005, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7423, + "step": 12 + }, + { + "epoch": 0.01335387776065742, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8057, + "step": 13 + }, + { + "epoch": 0.014381099126861838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7663, + "step": 14 + }, + { + "epoch": 0.015408320493066256, + "grad_norm": 4.694424152374268, + "learning_rate": 1.7123287671232876e-08, + "loss": 2.7715, + "step": 15 + }, + { + "epoch": 0.016435541859270673, + "grad_norm": 4.905329704284668, + "learning_rate": 3.424657534246575e-08, + "loss": 2.751, + "step": 16 + }, + { + "epoch": 0.01746276322547509, + "grad_norm": 4.705279350280762, + "learning_rate": 5.136986301369863e-08, + "loss": 2.7186, + "step": 17 + }, + { + "epoch": 0.01848998459167951, + "grad_norm": 5.154810905456543, + "learning_rate": 6.84931506849315e-08, + "loss": 2.8073, + "step": 18 + }, + { + "epoch": 0.019517205957883924, + "grad_norm": 5.070403099060059, + "learning_rate": 8.561643835616439e-08, + "loss": 2.7237, + "step": 19 + }, + { + "epoch": 0.02054442732408834, + "grad_norm": 4.901423454284668, + "learning_rate": 1.0273972602739726e-07, + "loss": 2.7882, + "step": 20 + }, + { + "epoch": 0.02054442732408834, + "eval_loss": 2.810307502746582, + "eval_runtime": 37.65, + "eval_samples_per_second": 26.56, + "eval_steps_per_second": 1.116, + "step": 20 + }, + { + "epoch": 0.02157164869029276, + "grad_norm": 4.914156436920166, + "learning_rate": 1.1986301369863014e-07, + "loss": 2.8005, + "step": 21 + }, + { + "epoch": 0.022598870056497175, + "grad_norm": 4.937129974365234, + "learning_rate": 1.36986301369863e-07, + "loss": 2.7902, + "step": 22 + }, + { + "epoch": 0.02362609142270159, + "grad_norm": 5.0690202713012695, + "learning_rate": 1.541095890410959e-07, + "loss": 2.7474, + "step": 23 + }, + { + "epoch": 0.02465331278890601, + "grad_norm": 4.980844974517822, + "learning_rate": 1.7123287671232878e-07, + "loss": 2.7695, + "step": 24 + }, + { + "epoch": 0.025680534155110426, + "grad_norm": 4.8092451095581055, + "learning_rate": 1.8835616438356165e-07, + "loss": 2.7611, + "step": 25 + }, + { + "epoch": 0.02670775552131484, + "grad_norm": 4.765528202056885, + "learning_rate": 2.0547945205479452e-07, + "loss": 2.7823, + "step": 26 + }, + { + "epoch": 0.02773497688751926, + "grad_norm": 3.9759938716888428, + "learning_rate": 2.226027397260274e-07, + "loss": 2.7445, + "step": 27 + }, + { + "epoch": 0.028762198253723677, + "grad_norm": 4.023419380187988, + "learning_rate": 2.397260273972603e-07, + "loss": 2.7679, + "step": 28 + }, + { + "epoch": 0.029789419619928096, + "grad_norm": 4.161378383636475, + "learning_rate": 2.568493150684932e-07, + "loss": 2.7704, + "step": 29 + }, + { + "epoch": 0.030816640986132512, + "grad_norm": 3.8363842964172363, + "learning_rate": 2.73972602739726e-07, + "loss": 2.7598, + "step": 30 + }, + { + "epoch": 0.030816640986132512, + "eval_loss": 2.772540330886841, + "eval_runtime": 37.2624, + "eval_samples_per_second": 26.837, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03184386235233693, + "grad_norm": 3.718747615814209, + "learning_rate": 2.910958904109589e-07, + "loss": 2.7241, + "step": 31 + }, + { + "epoch": 0.03287108371854135, + "grad_norm": 3.555985927581787, + "learning_rate": 3.082191780821918e-07, + "loss": 2.7751, + "step": 32 + }, + { + "epoch": 0.03389830508474576, + "grad_norm": 3.6041972637176514, + "learning_rate": 3.2534246575342466e-07, + "loss": 2.7322, + "step": 33 + }, + { + "epoch": 0.03492552645095018, + "grad_norm": 3.4408438205718994, + "learning_rate": 3.4246575342465755e-07, + "loss": 2.727, + "step": 34 + }, + { + "epoch": 0.035952747817154594, + "grad_norm": 3.4169280529022217, + "learning_rate": 3.595890410958904e-07, + "loss": 2.7062, + "step": 35 + }, + { + "epoch": 0.03697996918335902, + "grad_norm": 3.3882925510406494, + "learning_rate": 3.767123287671233e-07, + "loss": 2.6271, + "step": 36 + }, + { + "epoch": 0.03800719054956343, + "grad_norm": 3.0491855144500732, + "learning_rate": 3.938356164383562e-07, + "loss": 2.7309, + "step": 37 + }, + { + "epoch": 0.03903441191576785, + "grad_norm": 2.8444902896881104, + "learning_rate": 4.1095890410958903e-07, + "loss": 2.647, + "step": 38 + }, + { + "epoch": 0.040061633281972264, + "grad_norm": 2.7438602447509766, + "learning_rate": 4.2808219178082193e-07, + "loss": 2.6478, + "step": 39 + }, + { + "epoch": 0.04108885464817668, + "grad_norm": 2.8886020183563232, + "learning_rate": 4.452054794520548e-07, + "loss": 2.7065, + "step": 40 + }, + { + "epoch": 0.04108885464817668, + "eval_loss": 2.7021424770355225, + "eval_runtime": 37.3227, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.042116076014381096, + "grad_norm": 2.8825268745422363, + "learning_rate": 4.6232876712328767e-07, + "loss": 2.6666, + "step": 41 + }, + { + "epoch": 0.04314329738058552, + "grad_norm": 2.8848748207092285, + "learning_rate": 4.794520547945206e-07, + "loss": 2.597, + "step": 42 + }, + { + "epoch": 0.044170518746789934, + "grad_norm": 2.9176979064941406, + "learning_rate": 4.965753424657534e-07, + "loss": 2.6638, + "step": 43 + }, + { + "epoch": 0.04519774011299435, + "grad_norm": 2.6903059482574463, + "learning_rate": 5.136986301369864e-07, + "loss": 2.6202, + "step": 44 + }, + { + "epoch": 0.046224961479198766, + "grad_norm": 2.5349855422973633, + "learning_rate": 5.308219178082192e-07, + "loss": 2.6466, + "step": 45 + }, + { + "epoch": 0.04725218284540318, + "grad_norm": 2.405383825302124, + "learning_rate": 5.47945205479452e-07, + "loss": 2.6291, + "step": 46 + }, + { + "epoch": 0.048279404211607604, + "grad_norm": 2.526106595993042, + "learning_rate": 5.65068493150685e-07, + "loss": 2.6718, + "step": 47 + }, + { + "epoch": 0.04930662557781202, + "grad_norm": 2.3981597423553467, + "learning_rate": 5.821917808219178e-07, + "loss": 2.592, + "step": 48 + }, + { + "epoch": 0.050333846944016436, + "grad_norm": 2.61842679977417, + "learning_rate": 5.993150684931507e-07, + "loss": 2.6039, + "step": 49 + }, + { + "epoch": 0.05136106831022085, + "grad_norm": 2.3001420497894287, + "learning_rate": 6.164383561643836e-07, + "loss": 2.5481, + "step": 50 + }, + { + "epoch": 0.05136106831022085, + "eval_loss": 2.6244444847106934, + "eval_runtime": 37.3248, + "eval_samples_per_second": 26.792, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05238828967642527, + "grad_norm": 2.12744140625, + "learning_rate": 6.335616438356165e-07, + "loss": 2.5336, + "step": 51 + }, + { + "epoch": 0.05341551104262968, + "grad_norm": 1.974413275718689, + "learning_rate": 6.506849315068493e-07, + "loss": 2.5882, + "step": 52 + }, + { + "epoch": 0.054442732408834106, + "grad_norm": 2.098527669906616, + "learning_rate": 6.678082191780823e-07, + "loss": 2.5073, + "step": 53 + }, + { + "epoch": 0.05546995377503852, + "grad_norm": 2.056288719177246, + "learning_rate": 6.849315068493151e-07, + "loss": 2.5612, + "step": 54 + }, + { + "epoch": 0.05649717514124294, + "grad_norm": 2.2291629314422607, + "learning_rate": 7.020547945205481e-07, + "loss": 2.527, + "step": 55 + }, + { + "epoch": 0.05752439650744735, + "grad_norm": 2.140164375305176, + "learning_rate": 7.191780821917808e-07, + "loss": 2.4972, + "step": 56 + }, + { + "epoch": 0.05855161787365177, + "grad_norm": 1.8682485818862915, + "learning_rate": 7.363013698630137e-07, + "loss": 2.5231, + "step": 57 + }, + { + "epoch": 0.05957883923985619, + "grad_norm": 1.9373308420181274, + "learning_rate": 7.534246575342466e-07, + "loss": 2.4509, + "step": 58 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.8535712957382202, + "learning_rate": 7.705479452054795e-07, + "loss": 2.5164, + "step": 59 + }, + { + "epoch": 0.061633281972265024, + "grad_norm": 1.7497329711914062, + "learning_rate": 7.876712328767124e-07, + "loss": 2.5177, + "step": 60 + }, + { + "epoch": 0.061633281972265024, + "eval_loss": 2.5608646869659424, + "eval_runtime": 37.3426, + "eval_samples_per_second": 26.779, + "eval_steps_per_second": 1.125, + "step": 60 + }, + { + "epoch": 0.06266050333846944, + "grad_norm": 1.8476629257202148, + "learning_rate": 8.047945205479453e-07, + "loss": 2.4838, + "step": 61 + }, + { + "epoch": 0.06368772470467386, + "grad_norm": 1.7130696773529053, + "learning_rate": 8.219178082191781e-07, + "loss": 2.473, + "step": 62 + }, + { + "epoch": 0.06471494607087827, + "grad_norm": 1.751081109046936, + "learning_rate": 8.39041095890411e-07, + "loss": 2.5152, + "step": 63 + }, + { + "epoch": 0.0657421674370827, + "grad_norm": 1.7758578062057495, + "learning_rate": 8.561643835616439e-07, + "loss": 2.5031, + "step": 64 + }, + { + "epoch": 0.0667693888032871, + "grad_norm": 1.9316363334655762, + "learning_rate": 8.732876712328768e-07, + "loss": 2.3695, + "step": 65 + }, + { + "epoch": 0.06779661016949153, + "grad_norm": 1.7765878438949585, + "learning_rate": 8.904109589041097e-07, + "loss": 2.4669, + "step": 66 + }, + { + "epoch": 0.06882383153569595, + "grad_norm": 1.7339810132980347, + "learning_rate": 9.075342465753426e-07, + "loss": 2.4112, + "step": 67 + }, + { + "epoch": 0.06985105290190036, + "grad_norm": 1.717781662940979, + "learning_rate": 9.246575342465753e-07, + "loss": 2.4677, + "step": 68 + }, + { + "epoch": 0.07087827426810478, + "grad_norm": 1.7711046934127808, + "learning_rate": 9.417808219178083e-07, + "loss": 2.4131, + "step": 69 + }, + { + "epoch": 0.07190549563430919, + "grad_norm": 1.708680510520935, + "learning_rate": 9.589041095890411e-07, + "loss": 2.5191, + "step": 70 + }, + { + "epoch": 0.07190549563430919, + "eval_loss": 2.4920735359191895, + "eval_runtime": 37.3501, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 70 + }, + { + "epoch": 0.07293271700051361, + "grad_norm": 1.9429680109024048, + "learning_rate": 9.76027397260274e-07, + "loss": 2.4099, + "step": 71 + }, + { + "epoch": 0.07395993836671803, + "grad_norm": 1.7411247491836548, + "learning_rate": 9.931506849315068e-07, + "loss": 2.3632, + "step": 72 + }, + { + "epoch": 0.07498715973292244, + "grad_norm": 1.7577983140945435, + "learning_rate": 1.0102739726027399e-06, + "loss": 2.4214, + "step": 73 + }, + { + "epoch": 0.07601438109912687, + "grad_norm": 2.025926351547241, + "learning_rate": 1.0273972602739727e-06, + "loss": 2.394, + "step": 74 + }, + { + "epoch": 0.07704160246533127, + "grad_norm": 2.4366860389709473, + "learning_rate": 1.0445205479452056e-06, + "loss": 2.3943, + "step": 75 + }, + { + "epoch": 0.0780688238315357, + "grad_norm": 2.437108278274536, + "learning_rate": 1.0616438356164384e-06, + "loss": 2.3844, + "step": 76 + }, + { + "epoch": 0.07909604519774012, + "grad_norm": 2.1352434158325195, + "learning_rate": 1.0787671232876712e-06, + "loss": 2.3871, + "step": 77 + }, + { + "epoch": 0.08012326656394453, + "grad_norm": 1.8895397186279297, + "learning_rate": 1.095890410958904e-06, + "loss": 2.3927, + "step": 78 + }, + { + "epoch": 0.08115048793014895, + "grad_norm": 2.2960093021392822, + "learning_rate": 1.1130136986301371e-06, + "loss": 2.3343, + "step": 79 + }, + { + "epoch": 0.08217770929635336, + "grad_norm": 2.0891849994659424, + "learning_rate": 1.13013698630137e-06, + "loss": 2.3732, + "step": 80 + }, + { + "epoch": 0.08217770929635336, + "eval_loss": 2.4189229011535645, + "eval_runtime": 37.3645, + "eval_samples_per_second": 26.763, + "eval_steps_per_second": 1.124, + "step": 80 + }, + { + "epoch": 0.08320493066255778, + "grad_norm": 2.450587034225464, + "learning_rate": 1.1472602739726028e-06, + "loss": 2.3163, + "step": 81 + }, + { + "epoch": 0.08423215202876219, + "grad_norm": 2.079949378967285, + "learning_rate": 1.1643835616438357e-06, + "loss": 2.3381, + "step": 82 + }, + { + "epoch": 0.08525937339496661, + "grad_norm": 2.160737991333008, + "learning_rate": 1.1815068493150685e-06, + "loss": 2.3376, + "step": 83 + }, + { + "epoch": 0.08628659476117104, + "grad_norm": 2.463890552520752, + "learning_rate": 1.1986301369863014e-06, + "loss": 2.308, + "step": 84 + }, + { + "epoch": 0.08731381612737545, + "grad_norm": 2.3923065662384033, + "learning_rate": 1.2157534246575344e-06, + "loss": 2.2682, + "step": 85 + }, + { + "epoch": 0.08834103749357987, + "grad_norm": 2.2096004486083984, + "learning_rate": 1.2328767123287673e-06, + "loss": 2.3256, + "step": 86 + }, + { + "epoch": 0.08936825885978428, + "grad_norm": 2.1402556896209717, + "learning_rate": 1.25e-06, + "loss": 2.2767, + "step": 87 + }, + { + "epoch": 0.0903954802259887, + "grad_norm": 2.500629186630249, + "learning_rate": 1.267123287671233e-06, + "loss": 2.2576, + "step": 88 + }, + { + "epoch": 0.09142270159219312, + "grad_norm": 2.0257675647735596, + "learning_rate": 1.284246575342466e-06, + "loss": 2.2722, + "step": 89 + }, + { + "epoch": 0.09244992295839753, + "grad_norm": 2.148428440093994, + "learning_rate": 1.3013698630136986e-06, + "loss": 2.2309, + "step": 90 + }, + { + "epoch": 0.09244992295839753, + "eval_loss": 2.3439993858337402, + "eval_runtime": 37.3803, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 90 + }, + { + "epoch": 0.09347714432460195, + "grad_norm": 2.3541946411132812, + "learning_rate": 1.3184931506849317e-06, + "loss": 2.2347, + "step": 91 + }, + { + "epoch": 0.09450436569080636, + "grad_norm": 2.6239383220672607, + "learning_rate": 1.3356164383561645e-06, + "loss": 2.2589, + "step": 92 + }, + { + "epoch": 0.09553158705701079, + "grad_norm": 2.1981804370880127, + "learning_rate": 1.3527397260273976e-06, + "loss": 2.2372, + "step": 93 + }, + { + "epoch": 0.09655880842321521, + "grad_norm": 2.3652358055114746, + "learning_rate": 1.3698630136986302e-06, + "loss": 2.2169, + "step": 94 + }, + { + "epoch": 0.09758602978941962, + "grad_norm": 2.2078158855438232, + "learning_rate": 1.386986301369863e-06, + "loss": 2.2761, + "step": 95 + }, + { + "epoch": 0.09861325115562404, + "grad_norm": 2.494685649871826, + "learning_rate": 1.4041095890410961e-06, + "loss": 2.1691, + "step": 96 + }, + { + "epoch": 0.09964047252182845, + "grad_norm": 2.3293051719665527, + "learning_rate": 1.421232876712329e-06, + "loss": 2.2246, + "step": 97 + }, + { + "epoch": 0.10066769388803287, + "grad_norm": 2.410687208175659, + "learning_rate": 1.4383561643835616e-06, + "loss": 2.1166, + "step": 98 + }, + { + "epoch": 0.1016949152542373, + "grad_norm": 2.5168960094451904, + "learning_rate": 1.4554794520547946e-06, + "loss": 2.1417, + "step": 99 + }, + { + "epoch": 0.1027221366204417, + "grad_norm": 2.178412914276123, + "learning_rate": 1.4726027397260275e-06, + "loss": 2.1638, + "step": 100 + }, + { + "epoch": 0.1027221366204417, + "eval_loss": 2.273622751235962, + "eval_runtime": 37.3892, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 100 + }, + { + "epoch": 0.10374935798664613, + "grad_norm": 2.8951706886291504, + "learning_rate": 1.4897260273972605e-06, + "loss": 2.1631, + "step": 101 + }, + { + "epoch": 0.10477657935285054, + "grad_norm": 2.103206157684326, + "learning_rate": 1.5068493150684932e-06, + "loss": 2.1419, + "step": 102 + }, + { + "epoch": 0.10580380071905496, + "grad_norm": 2.495969772338867, + "learning_rate": 1.5239726027397262e-06, + "loss": 2.1029, + "step": 103 + }, + { + "epoch": 0.10683102208525937, + "grad_norm": 2.5686826705932617, + "learning_rate": 1.541095890410959e-06, + "loss": 2.1235, + "step": 104 + }, + { + "epoch": 0.10785824345146379, + "grad_norm": 3.4710471630096436, + "learning_rate": 1.5582191780821921e-06, + "loss": 2.1832, + "step": 105 + }, + { + "epoch": 0.10888546481766821, + "grad_norm": 2.2563316822052, + "learning_rate": 1.5753424657534248e-06, + "loss": 2.1441, + "step": 106 + }, + { + "epoch": 0.10991268618387262, + "grad_norm": 3.860506534576416, + "learning_rate": 1.5924657534246576e-06, + "loss": 2.1764, + "step": 107 + }, + { + "epoch": 0.11093990755007704, + "grad_norm": 2.550135850906372, + "learning_rate": 1.6095890410958907e-06, + "loss": 2.1455, + "step": 108 + }, + { + "epoch": 0.11196712891628145, + "grad_norm": 3.168283700942993, + "learning_rate": 1.6267123287671235e-06, + "loss": 2.0246, + "step": 109 + }, + { + "epoch": 0.11299435028248588, + "grad_norm": 2.577069044113159, + "learning_rate": 1.6438356164383561e-06, + "loss": 2.0921, + "step": 110 + }, + { + "epoch": 0.11299435028248588, + "eval_loss": 2.2218997478485107, + "eval_runtime": 37.246, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 110 + }, + { + "epoch": 0.1140215716486903, + "grad_norm": 2.3515374660491943, + "learning_rate": 1.6609589041095892e-06, + "loss": 2.0209, + "step": 111 + }, + { + "epoch": 0.1150487930148947, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.678082191780822e-06, + "loss": 2.1165, + "step": 112 + }, + { + "epoch": 0.11607601438109913, + "grad_norm": 2.567336082458496, + "learning_rate": 1.695205479452055e-06, + "loss": 2.0529, + "step": 113 + }, + { + "epoch": 0.11710323574730354, + "grad_norm": 2.4126152992248535, + "learning_rate": 1.7123287671232877e-06, + "loss": 2.1028, + "step": 114 + }, + { + "epoch": 0.11813045711350796, + "grad_norm": 3.577528476715088, + "learning_rate": 1.7294520547945206e-06, + "loss": 2.0317, + "step": 115 + }, + { + "epoch": 0.11915767847971238, + "grad_norm": 2.9624195098876953, + "learning_rate": 1.7465753424657536e-06, + "loss": 2.1318, + "step": 116 + }, + { + "epoch": 0.12018489984591679, + "grad_norm": 2.789854049682617, + "learning_rate": 1.7636986301369865e-06, + "loss": 2.0822, + "step": 117 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.8168365955352783, + "learning_rate": 1.7808219178082193e-06, + "loss": 2.08, + "step": 118 + }, + { + "epoch": 0.12223934257832562, + "grad_norm": 2.7616257667541504, + "learning_rate": 1.7979452054794521e-06, + "loss": 2.0668, + "step": 119 + }, + { + "epoch": 0.12326656394453005, + "grad_norm": 2.5610315799713135, + "learning_rate": 1.8150684931506852e-06, + "loss": 2.1518, + "step": 120 + }, + { + "epoch": 0.12326656394453005, + "eval_loss": 2.184033155441284, + "eval_runtime": 37.3784, + "eval_samples_per_second": 26.753, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.12429378531073447, + "grad_norm": 2.817063093185425, + "learning_rate": 1.832191780821918e-06, + "loss": 2.0813, + "step": 121 + }, + { + "epoch": 0.12532100667693888, + "grad_norm": 2.987205982208252, + "learning_rate": 1.8493150684931507e-06, + "loss": 2.0421, + "step": 122 + }, + { + "epoch": 0.1263482280431433, + "grad_norm": 2.8480496406555176, + "learning_rate": 1.8664383561643837e-06, + "loss": 2.0018, + "step": 123 + }, + { + "epoch": 0.12737544940934772, + "grad_norm": 2.994274616241455, + "learning_rate": 1.8835616438356166e-06, + "loss": 2.0266, + "step": 124 + }, + { + "epoch": 0.12840267077555212, + "grad_norm": 2.64136004447937, + "learning_rate": 1.9006849315068496e-06, + "loss": 1.9873, + "step": 125 + }, + { + "epoch": 0.12942989214175654, + "grad_norm": 2.9426915645599365, + "learning_rate": 1.9178082191780823e-06, + "loss": 2.0566, + "step": 126 + }, + { + "epoch": 0.13045711350796096, + "grad_norm": 2.9018619060516357, + "learning_rate": 1.9349315068493153e-06, + "loss": 1.9674, + "step": 127 + }, + { + "epoch": 0.1314843348741654, + "grad_norm": 3.6921045780181885, + "learning_rate": 1.952054794520548e-06, + "loss": 2.0295, + "step": 128 + }, + { + "epoch": 0.1325115562403698, + "grad_norm": 2.4194657802581787, + "learning_rate": 1.969178082191781e-06, + "loss": 2.0639, + "step": 129 + }, + { + "epoch": 0.1335387776065742, + "grad_norm": 3.252349615097046, + "learning_rate": 1.9863013698630136e-06, + "loss": 2.0613, + "step": 130 + }, + { + "epoch": 0.1335387776065742, + "eval_loss": 2.153157949447632, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 130 + }, + { + "epoch": 0.13456599897277863, + "grad_norm": 2.9361023902893066, + "learning_rate": 2.0034246575342467e-06, + "loss": 1.9817, + "step": 131 + }, + { + "epoch": 0.13559322033898305, + "grad_norm": 2.834803581237793, + "learning_rate": 2.0205479452054797e-06, + "loss": 2.0539, + "step": 132 + }, + { + "epoch": 0.13662044170518747, + "grad_norm": 2.8814785480499268, + "learning_rate": 2.037671232876713e-06, + "loss": 2.0296, + "step": 133 + }, + { + "epoch": 0.1376476630713919, + "grad_norm": 2.647125482559204, + "learning_rate": 2.0547945205479454e-06, + "loss": 1.9739, + "step": 134 + }, + { + "epoch": 0.1386748844375963, + "grad_norm": 2.9608750343322754, + "learning_rate": 2.071917808219178e-06, + "loss": 2.0083, + "step": 135 + }, + { + "epoch": 0.1397021058038007, + "grad_norm": 2.8570032119750977, + "learning_rate": 2.089041095890411e-06, + "loss": 2.0216, + "step": 136 + }, + { + "epoch": 0.14072932717000514, + "grad_norm": 2.806358575820923, + "learning_rate": 2.106164383561644e-06, + "loss": 1.9457, + "step": 137 + }, + { + "epoch": 0.14175654853620956, + "grad_norm": 3.1003353595733643, + "learning_rate": 2.123287671232877e-06, + "loss": 2.0324, + "step": 138 + }, + { + "epoch": 0.14278376990241398, + "grad_norm": 2.679588556289673, + "learning_rate": 2.14041095890411e-06, + "loss": 1.9699, + "step": 139 + }, + { + "epoch": 0.14381099126861838, + "grad_norm": 2.9436194896698, + "learning_rate": 2.1575342465753425e-06, + "loss": 1.9824, + "step": 140 + }, + { + "epoch": 0.14381099126861838, + "eval_loss": 2.123953104019165, + "eval_runtime": 37.4214, + "eval_samples_per_second": 26.723, + "eval_steps_per_second": 1.122, + "step": 140 + }, + { + "epoch": 0.1448382126348228, + "grad_norm": 3.088301420211792, + "learning_rate": 2.1746575342465755e-06, + "loss": 1.9861, + "step": 141 + }, + { + "epoch": 0.14586543400102722, + "grad_norm": 3.011859178543091, + "learning_rate": 2.191780821917808e-06, + "loss": 1.9383, + "step": 142 + }, + { + "epoch": 0.14689265536723164, + "grad_norm": 2.930264472961426, + "learning_rate": 2.2089041095890412e-06, + "loss": 2.0674, + "step": 143 + }, + { + "epoch": 0.14791987673343607, + "grad_norm": 3.361328601837158, + "learning_rate": 2.2260273972602743e-06, + "loss": 1.9495, + "step": 144 + }, + { + "epoch": 0.14894709809964046, + "grad_norm": 2.7825112342834473, + "learning_rate": 2.243150684931507e-06, + "loss": 1.9335, + "step": 145 + }, + { + "epoch": 0.14997431946584489, + "grad_norm": 3.7215847969055176, + "learning_rate": 2.26027397260274e-06, + "loss": 1.9403, + "step": 146 + }, + { + "epoch": 0.1510015408320493, + "grad_norm": 2.5125460624694824, + "learning_rate": 2.2773972602739726e-06, + "loss": 1.9437, + "step": 147 + }, + { + "epoch": 0.15202876219825373, + "grad_norm": 3.0963492393493652, + "learning_rate": 2.2945205479452057e-06, + "loss": 1.9651, + "step": 148 + }, + { + "epoch": 0.15305598356445815, + "grad_norm": 2.9147121906280518, + "learning_rate": 2.3116438356164387e-06, + "loss": 2.0567, + "step": 149 + }, + { + "epoch": 0.15408320493066255, + "grad_norm": 3.5275802612304688, + "learning_rate": 2.3287671232876713e-06, + "loss": 1.9773, + "step": 150 + }, + { + "epoch": 0.15408320493066255, + "eval_loss": 2.112344264984131, + "eval_runtime": 37.4142, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 150 + }, + { + "epoch": 0.15511042629686697, + "grad_norm": 3.013786554336548, + "learning_rate": 2.3458904109589044e-06, + "loss": 1.8753, + "step": 151 + }, + { + "epoch": 0.1561376476630714, + "grad_norm": 3.433615207672119, + "learning_rate": 2.363013698630137e-06, + "loss": 1.9256, + "step": 152 + }, + { + "epoch": 0.15716486902927582, + "grad_norm": 3.1526033878326416, + "learning_rate": 2.38013698630137e-06, + "loss": 1.9815, + "step": 153 + }, + { + "epoch": 0.15819209039548024, + "grad_norm": 3.2099974155426025, + "learning_rate": 2.3972602739726027e-06, + "loss": 1.9628, + "step": 154 + }, + { + "epoch": 0.15921931176168463, + "grad_norm": 3.2713584899902344, + "learning_rate": 2.4143835616438358e-06, + "loss": 1.9128, + "step": 155 + }, + { + "epoch": 0.16024653312788906, + "grad_norm": 3.287649393081665, + "learning_rate": 2.431506849315069e-06, + "loss": 1.8764, + "step": 156 + }, + { + "epoch": 0.16127375449409348, + "grad_norm": 3.1439101696014404, + "learning_rate": 2.4486301369863015e-06, + "loss": 1.9428, + "step": 157 + }, + { + "epoch": 0.1623009758602979, + "grad_norm": 3.006457805633545, + "learning_rate": 2.4657534246575345e-06, + "loss": 1.9516, + "step": 158 + }, + { + "epoch": 0.1633281972265023, + "grad_norm": 2.9567787647247314, + "learning_rate": 2.482876712328767e-06, + "loss": 1.915, + "step": 159 + }, + { + "epoch": 0.16435541859270672, + "grad_norm": 3.2475216388702393, + "learning_rate": 2.5e-06, + "loss": 1.9451, + "step": 160 + }, + { + "epoch": 0.16435541859270672, + "eval_loss": 2.086299419403076, + "eval_runtime": 37.3939, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 160 + }, + { + "epoch": 0.16538263995891114, + "grad_norm": 2.9586498737335205, + "learning_rate": 2.5171232876712333e-06, + "loss": 1.9298, + "step": 161 + }, + { + "epoch": 0.16640986132511557, + "grad_norm": 3.066910982131958, + "learning_rate": 2.534246575342466e-06, + "loss": 1.9103, + "step": 162 + }, + { + "epoch": 0.16743708269132, + "grad_norm": 2.814741611480713, + "learning_rate": 2.551369863013699e-06, + "loss": 1.8678, + "step": 163 + }, + { + "epoch": 0.16846430405752438, + "grad_norm": 3.7239339351654053, + "learning_rate": 2.568493150684932e-06, + "loss": 1.9321, + "step": 164 + }, + { + "epoch": 0.1694915254237288, + "grad_norm": 2.829899787902832, + "learning_rate": 2.585616438356164e-06, + "loss": 1.9587, + "step": 165 + }, + { + "epoch": 0.17051874678993323, + "grad_norm": 3.826289176940918, + "learning_rate": 2.6027397260273973e-06, + "loss": 1.9288, + "step": 166 + }, + { + "epoch": 0.17154596815613765, + "grad_norm": 3.1028499603271484, + "learning_rate": 2.6198630136986303e-06, + "loss": 1.8685, + "step": 167 + }, + { + "epoch": 0.17257318952234207, + "grad_norm": 3.6923508644104004, + "learning_rate": 2.6369863013698634e-06, + "loss": 1.9276, + "step": 168 + }, + { + "epoch": 0.17360041088854647, + "grad_norm": 3.0577642917633057, + "learning_rate": 2.654109589041096e-06, + "loss": 1.9653, + "step": 169 + }, + { + "epoch": 0.1746276322547509, + "grad_norm": 3.363762617111206, + "learning_rate": 2.671232876712329e-06, + "loss": 1.9465, + "step": 170 + }, + { + "epoch": 0.1746276322547509, + "eval_loss": 2.0753531455993652, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 170 + }, + { + "epoch": 0.17565485362095531, + "grad_norm": 2.9063165187835693, + "learning_rate": 2.688356164383562e-06, + "loss": 1.8827, + "step": 171 + }, + { + "epoch": 0.17668207498715974, + "grad_norm": 3.4709181785583496, + "learning_rate": 2.705479452054795e-06, + "loss": 1.882, + "step": 172 + }, + { + "epoch": 0.17770929635336416, + "grad_norm": 3.395397663116455, + "learning_rate": 2.7226027397260274e-06, + "loss": 1.8711, + "step": 173 + }, + { + "epoch": 0.17873651771956856, + "grad_norm": 3.9973363876342773, + "learning_rate": 2.7397260273972604e-06, + "loss": 1.8223, + "step": 174 + }, + { + "epoch": 0.17976373908577298, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.947, + "step": 175 + }, + { + "epoch": 0.1807909604519774, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.9192, + "step": 176 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 3.84712290763855, + "learning_rate": 2.773972602739726e-06, + "loss": 1.8344, + "step": 177 + }, + { + "epoch": 0.18284540318438625, + "grad_norm": 3.487525463104248, + "learning_rate": 2.791095890410959e-06, + "loss": 1.9242, + "step": 178 + }, + { + "epoch": 0.18387262455059064, + "grad_norm": 3.1980319023132324, + "learning_rate": 2.8082191780821922e-06, + "loss": 1.7959, + "step": 179 + }, + { + "epoch": 0.18489984591679506, + "grad_norm": 3.56462025642395, + "learning_rate": 2.8253424657534253e-06, + "loss": 1.9668, + "step": 180 + }, + { + "epoch": 0.18489984591679506, + "eval_loss": 2.0606908798217773, + "eval_runtime": 37.3994, + "eval_samples_per_second": 26.738, + "eval_steps_per_second": 1.123, + "step": 180 + }, + { + "epoch": 0.1859270672829995, + "grad_norm": 3.613842248916626, + "learning_rate": 2.842465753424658e-06, + "loss": 1.925, + "step": 181 + }, + { + "epoch": 0.1869542886492039, + "grad_norm": 2.9755642414093018, + "learning_rate": 2.8595890410958905e-06, + "loss": 1.9251, + "step": 182 + }, + { + "epoch": 0.18798151001540833, + "grad_norm": 2.9678146839141846, + "learning_rate": 2.876712328767123e-06, + "loss": 1.9406, + "step": 183 + }, + { + "epoch": 0.18900873138161273, + "grad_norm": 3.3536393642425537, + "learning_rate": 2.8938356164383562e-06, + "loss": 1.8041, + "step": 184 + }, + { + "epoch": 0.19003595274781715, + "grad_norm": 2.714902877807617, + "learning_rate": 2.9109589041095893e-06, + "loss": 1.8213, + "step": 185 + }, + { + "epoch": 0.19106317411402157, + "grad_norm": 3.0686237812042236, + "learning_rate": 2.9280821917808223e-06, + "loss": 1.817, + "step": 186 + }, + { + "epoch": 0.192090395480226, + "grad_norm": 3.510448932647705, + "learning_rate": 2.945205479452055e-06, + "loss": 1.8317, + "step": 187 + }, + { + "epoch": 0.19311761684643042, + "grad_norm": 3.2060787677764893, + "learning_rate": 2.962328767123288e-06, + "loss": 1.894, + "step": 188 + }, + { + "epoch": 0.1941448382126348, + "grad_norm": 3.1756906509399414, + "learning_rate": 2.979452054794521e-06, + "loss": 1.8788, + "step": 189 + }, + { + "epoch": 0.19517205957883924, + "grad_norm": 3.059863805770874, + "learning_rate": 2.9965753424657533e-06, + "loss": 1.8198, + "step": 190 + }, + { + "epoch": 0.19517205957883924, + "eval_loss": 2.0506513118743896, + "eval_runtime": 37.4014, + "eval_samples_per_second": 26.737, + "eval_steps_per_second": 1.123, + "step": 190 + }, + { + "epoch": 0.19619928094504366, + "grad_norm": 3.193685531616211, + "learning_rate": 3.0136986301369864e-06, + "loss": 1.8818, + "step": 191 + }, + { + "epoch": 0.19722650231124808, + "grad_norm": 2.905402183532715, + "learning_rate": 3.0308219178082194e-06, + "loss": 1.8546, + "step": 192 + }, + { + "epoch": 0.1982537236774525, + "grad_norm": 3.423109531402588, + "learning_rate": 3.0479452054794525e-06, + "loss": 1.8142, + "step": 193 + }, + { + "epoch": 0.1992809450436569, + "grad_norm": 3.136094331741333, + "learning_rate": 3.065068493150685e-06, + "loss": 1.8553, + "step": 194 + }, + { + "epoch": 0.20030816640986132, + "grad_norm": 3.0746538639068604, + "learning_rate": 3.082191780821918e-06, + "loss": 1.8153, + "step": 195 + }, + { + "epoch": 0.20133538777606574, + "grad_norm": 2.9354681968688965, + "learning_rate": 3.099315068493151e-06, + "loss": 1.8301, + "step": 196 + }, + { + "epoch": 0.20236260914227017, + "grad_norm": 3.00426983833313, + "learning_rate": 3.1164383561643843e-06, + "loss": 1.8233, + "step": 197 + }, + { + "epoch": 0.2033898305084746, + "grad_norm": 3.2754054069519043, + "learning_rate": 3.1335616438356165e-06, + "loss": 1.787, + "step": 198 + }, + { + "epoch": 0.20441705187467898, + "grad_norm": 2.9854509830474854, + "learning_rate": 3.1506849315068495e-06, + "loss": 1.9341, + "step": 199 + }, + { + "epoch": 0.2054442732408834, + "grad_norm": 3.5655088424682617, + "learning_rate": 3.167808219178082e-06, + "loss": 1.7964, + "step": 200 + }, + { + "epoch": 0.2054442732408834, + "eval_loss": 2.0440263748168945, + "eval_runtime": 37.4054, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 200 + }, + { + "epoch": 0.20647149460708783, + "grad_norm": 3.132373571395874, + "learning_rate": 3.184931506849315e-06, + "loss": 1.8019, + "step": 201 + }, + { + "epoch": 0.20749871597329225, + "grad_norm": 3.3017780780792236, + "learning_rate": 3.2020547945205483e-06, + "loss": 1.7786, + "step": 202 + }, + { + "epoch": 0.20852593733949668, + "grad_norm": 3.009382486343384, + "learning_rate": 3.2191780821917813e-06, + "loss": 1.7639, + "step": 203 + }, + { + "epoch": 0.20955315870570107, + "grad_norm": 3.5784854888916016, + "learning_rate": 3.236301369863014e-06, + "loss": 1.76, + "step": 204 + }, + { + "epoch": 0.2105803800719055, + "grad_norm": 3.236858367919922, + "learning_rate": 3.253424657534247e-06, + "loss": 1.7295, + "step": 205 + }, + { + "epoch": 0.21160760143810992, + "grad_norm": 3.6157479286193848, + "learning_rate": 3.2705479452054796e-06, + "loss": 1.8287, + "step": 206 + }, + { + "epoch": 0.21263482280431434, + "grad_norm": 3.51790452003479, + "learning_rate": 3.2876712328767123e-06, + "loss": 1.8004, + "step": 207 + }, + { + "epoch": 0.21366204417051873, + "grad_norm": 3.0700764656066895, + "learning_rate": 3.3047945205479453e-06, + "loss": 1.7581, + "step": 208 + }, + { + "epoch": 0.21468926553672316, + "grad_norm": 3.2651591300964355, + "learning_rate": 3.3219178082191784e-06, + "loss": 1.7724, + "step": 209 + }, + { + "epoch": 0.21571648690292758, + "grad_norm": 2.9267404079437256, + "learning_rate": 3.3390410958904114e-06, + "loss": 1.8154, + "step": 210 + }, + { + "epoch": 0.21571648690292758, + "eval_loss": 2.0323731899261475, + "eval_runtime": 37.2156, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 210 + }, + { + "epoch": 0.216743708269132, + "grad_norm": 3.463411808013916, + "learning_rate": 3.356164383561644e-06, + "loss": 1.8309, + "step": 211 + }, + { + "epoch": 0.21777092963533642, + "grad_norm": 3.123573064804077, + "learning_rate": 3.373287671232877e-06, + "loss": 1.8793, + "step": 212 + }, + { + "epoch": 0.21879815100154082, + "grad_norm": 3.427051305770874, + "learning_rate": 3.39041095890411e-06, + "loss": 1.8259, + "step": 213 + }, + { + "epoch": 0.21982537236774524, + "grad_norm": 3.306366443634033, + "learning_rate": 3.4075342465753424e-06, + "loss": 1.8565, + "step": 214 + }, + { + "epoch": 0.22085259373394966, + "grad_norm": 3.557340383529663, + "learning_rate": 3.4246575342465754e-06, + "loss": 1.7531, + "step": 215 + }, + { + "epoch": 0.2218798151001541, + "grad_norm": 3.20525860786438, + "learning_rate": 3.4417808219178085e-06, + "loss": 1.8032, + "step": 216 + }, + { + "epoch": 0.2229070364663585, + "grad_norm": 3.2657511234283447, + "learning_rate": 3.458904109589041e-06, + "loss": 1.8026, + "step": 217 + }, + { + "epoch": 0.2239342578325629, + "grad_norm": 3.1869699954986572, + "learning_rate": 3.476027397260274e-06, + "loss": 1.814, + "step": 218 + }, + { + "epoch": 0.22496147919876733, + "grad_norm": 2.661536931991577, + "learning_rate": 3.4931506849315072e-06, + "loss": 1.8005, + "step": 219 + }, + { + "epoch": 0.22598870056497175, + "grad_norm": 3.1108739376068115, + "learning_rate": 3.5102739726027403e-06, + "loss": 1.8095, + "step": 220 + }, + { + "epoch": 0.22598870056497175, + "eval_loss": 2.015268325805664, + "eval_runtime": 37.2867, + "eval_samples_per_second": 26.819, + "eval_steps_per_second": 1.126, + "step": 220 + }, + { + "epoch": 0.22701592193117617, + "grad_norm": 2.8078837394714355, + "learning_rate": 3.527397260273973e-06, + "loss": 1.8016, + "step": 221 + }, + { + "epoch": 0.2280431432973806, + "grad_norm": 3.131399631500244, + "learning_rate": 3.5445205479452056e-06, + "loss": 1.7787, + "step": 222 + }, + { + "epoch": 0.229070364663585, + "grad_norm": 3.114422559738159, + "learning_rate": 3.5616438356164386e-06, + "loss": 1.7899, + "step": 223 + }, + { + "epoch": 0.2300975860297894, + "grad_norm": 3.040816068649292, + "learning_rate": 3.5787671232876712e-06, + "loss": 1.8601, + "step": 224 + }, + { + "epoch": 0.23112480739599384, + "grad_norm": 3.043100357055664, + "learning_rate": 3.5958904109589043e-06, + "loss": 1.8244, + "step": 225 + }, + { + "epoch": 0.23215202876219826, + "grad_norm": 3.4232981204986572, + "learning_rate": 3.6130136986301373e-06, + "loss": 1.7634, + "step": 226 + }, + { + "epoch": 0.23317925012840268, + "grad_norm": 2.8170571327209473, + "learning_rate": 3.6301369863013704e-06, + "loss": 1.7793, + "step": 227 + }, + { + "epoch": 0.23420647149460708, + "grad_norm": 3.9133572578430176, + "learning_rate": 3.647260273972603e-06, + "loss": 1.8239, + "step": 228 + }, + { + "epoch": 0.2352336928608115, + "grad_norm": 2.7053382396698, + "learning_rate": 3.664383561643836e-06, + "loss": 1.8209, + "step": 229 + }, + { + "epoch": 0.23626091422701592, + "grad_norm": 3.4366891384124756, + "learning_rate": 3.6815068493150687e-06, + "loss": 1.7539, + "step": 230 + }, + { + "epoch": 0.23626091422701592, + "eval_loss": 2.0176751613616943, + "eval_runtime": 37.3064, + "eval_samples_per_second": 26.805, + "eval_steps_per_second": 1.126, + "step": 230 + }, + { + "epoch": 0.23728813559322035, + "grad_norm": 2.592280149459839, + "learning_rate": 3.6986301369863014e-06, + "loss": 1.7858, + "step": 231 + }, + { + "epoch": 0.23831535695942477, + "grad_norm": 2.8370869159698486, + "learning_rate": 3.7157534246575344e-06, + "loss": 1.7414, + "step": 232 + }, + { + "epoch": 0.23934257832562916, + "grad_norm": 3.2120535373687744, + "learning_rate": 3.7328767123287675e-06, + "loss": 1.8467, + "step": 233 + }, + { + "epoch": 0.24036979969183359, + "grad_norm": 2.9314255714416504, + "learning_rate": 3.7500000000000005e-06, + "loss": 1.7602, + "step": 234 + }, + { + "epoch": 0.241397021058038, + "grad_norm": 3.056293249130249, + "learning_rate": 3.767123287671233e-06, + "loss": 1.7972, + "step": 235 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.5918917655944824, + "learning_rate": 3.784246575342466e-06, + "loss": 1.8274, + "step": 236 + }, + { + "epoch": 0.24345146379044685, + "grad_norm": 2.7911906242370605, + "learning_rate": 3.8013698630136993e-06, + "loss": 1.753, + "step": 237 + }, + { + "epoch": 0.24447868515665125, + "grad_norm": 2.810612916946411, + "learning_rate": 3.818493150684932e-06, + "loss": 1.7822, + "step": 238 + }, + { + "epoch": 0.24550590652285567, + "grad_norm": 2.8863744735717773, + "learning_rate": 3.8356164383561645e-06, + "loss": 1.7258, + "step": 239 + }, + { + "epoch": 0.2465331278890601, + "grad_norm": 3.002087116241455, + "learning_rate": 3.852739726027397e-06, + "loss": 1.8423, + "step": 240 + }, + { + "epoch": 0.2465331278890601, + "eval_loss": 2.020223617553711, + "eval_runtime": 37.3197, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 240 + }, + { + "epoch": 0.24756034925526452, + "grad_norm": 2.892317056655884, + "learning_rate": 3.869863013698631e-06, + "loss": 1.8181, + "step": 241 + }, + { + "epoch": 0.24858757062146894, + "grad_norm": 2.895620584487915, + "learning_rate": 3.886986301369863e-06, + "loss": 1.7538, + "step": 242 + }, + { + "epoch": 0.24961479198767333, + "grad_norm": 3.015660524368286, + "learning_rate": 3.904109589041096e-06, + "loss": 1.766, + "step": 243 + }, + { + "epoch": 0.25064201335387776, + "grad_norm": 2.7516112327575684, + "learning_rate": 3.921232876712329e-06, + "loss": 1.7324, + "step": 244 + }, + { + "epoch": 0.2516692347200822, + "grad_norm": 2.937777519226074, + "learning_rate": 3.938356164383562e-06, + "loss": 1.7468, + "step": 245 + }, + { + "epoch": 0.2526964560862866, + "grad_norm": 3.1706550121307373, + "learning_rate": 3.9554794520547955e-06, + "loss": 1.7035, + "step": 246 + }, + { + "epoch": 0.253723677452491, + "grad_norm": 2.7795372009277344, + "learning_rate": 3.972602739726027e-06, + "loss": 1.7912, + "step": 247 + }, + { + "epoch": 0.25475089881869545, + "grad_norm": 3.0375940799713135, + "learning_rate": 3.989726027397261e-06, + "loss": 1.7208, + "step": 248 + }, + { + "epoch": 0.25577812018489987, + "grad_norm": 2.863913059234619, + "learning_rate": 4.006849315068493e-06, + "loss": 1.7891, + "step": 249 + }, + { + "epoch": 0.25680534155110424, + "grad_norm": 3.02720046043396, + "learning_rate": 4.023972602739726e-06, + "loss": 1.7864, + "step": 250 + }, + { + "epoch": 0.25680534155110424, + "eval_loss": 1.9981943368911743, + "eval_runtime": 37.3019, + "eval_samples_per_second": 26.808, + "eval_steps_per_second": 1.126, + "step": 250 + }, + { + "epoch": 0.25783256291730866, + "grad_norm": 2.90144681930542, + "learning_rate": 4.0410958904109595e-06, + "loss": 1.7508, + "step": 251 + }, + { + "epoch": 0.2588597842835131, + "grad_norm": 2.7670342922210693, + "learning_rate": 4.058219178082192e-06, + "loss": 1.7668, + "step": 252 + }, + { + "epoch": 0.2598870056497175, + "grad_norm": 3.0309600830078125, + "learning_rate": 4.075342465753426e-06, + "loss": 1.7853, + "step": 253 + }, + { + "epoch": 0.26091422701592193, + "grad_norm": 2.995190143585205, + "learning_rate": 4.092465753424658e-06, + "loss": 1.7409, + "step": 254 + }, + { + "epoch": 0.26194144838212635, + "grad_norm": 3.127767324447632, + "learning_rate": 4.109589041095891e-06, + "loss": 1.743, + "step": 255 + }, + { + "epoch": 0.2629686697483308, + "grad_norm": 3.7104549407958984, + "learning_rate": 4.1267123287671235e-06, + "loss": 1.8149, + "step": 256 + }, + { + "epoch": 0.2639958911145352, + "grad_norm": 3.042569160461426, + "learning_rate": 4.143835616438356e-06, + "loss": 1.7544, + "step": 257 + }, + { + "epoch": 0.2650231124807396, + "grad_norm": 2.9337198734283447, + "learning_rate": 4.16095890410959e-06, + "loss": 1.823, + "step": 258 + }, + { + "epoch": 0.26605033384694404, + "grad_norm": 3.1853089332580566, + "learning_rate": 4.178082191780822e-06, + "loss": 1.8002, + "step": 259 + }, + { + "epoch": 0.2670775552131484, + "grad_norm": 2.9620680809020996, + "learning_rate": 4.195205479452055e-06, + "loss": 1.7401, + "step": 260 + }, + { + "epoch": 0.2670775552131484, + "eval_loss": 1.9844669103622437, + "eval_runtime": 37.3162, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 260 + }, + { + "epoch": 0.26810477657935283, + "grad_norm": 3.477985382080078, + "learning_rate": 4.212328767123288e-06, + "loss": 1.7456, + "step": 261 + }, + { + "epoch": 0.26913199794555726, + "grad_norm": 2.881277084350586, + "learning_rate": 4.229452054794521e-06, + "loss": 1.7427, + "step": 262 + }, + { + "epoch": 0.2701592193117617, + "grad_norm": 3.3313355445861816, + "learning_rate": 4.246575342465754e-06, + "loss": 1.7537, + "step": 263 + }, + { + "epoch": 0.2711864406779661, + "grad_norm": 2.947119951248169, + "learning_rate": 4.263698630136986e-06, + "loss": 1.7349, + "step": 264 + }, + { + "epoch": 0.2722136620441705, + "grad_norm": 3.0780889987945557, + "learning_rate": 4.28082191780822e-06, + "loss": 1.694, + "step": 265 + }, + { + "epoch": 0.27324088341037495, + "grad_norm": 3.3371143341064453, + "learning_rate": 4.297945205479452e-06, + "loss": 1.8489, + "step": 266 + }, + { + "epoch": 0.27426810477657937, + "grad_norm": 3.01694917678833, + "learning_rate": 4.315068493150685e-06, + "loss": 1.7478, + "step": 267 + }, + { + "epoch": 0.2752953261427838, + "grad_norm": 3.001850128173828, + "learning_rate": 4.3321917808219185e-06, + "loss": 1.7298, + "step": 268 + }, + { + "epoch": 0.2763225475089882, + "grad_norm": 2.9096696376800537, + "learning_rate": 4.349315068493151e-06, + "loss": 1.7542, + "step": 269 + }, + { + "epoch": 0.2773497688751926, + "grad_norm": 3.209601402282715, + "learning_rate": 4.3664383561643846e-06, + "loss": 1.7248, + "step": 270 + }, + { + "epoch": 0.2773497688751926, + "eval_loss": 1.9732507467269897, + "eval_runtime": 37.3151, + "eval_samples_per_second": 26.799, + "eval_steps_per_second": 1.126, + "step": 270 + }, + { + "epoch": 0.278376990241397, + "grad_norm": 2.9635558128356934, + "learning_rate": 4.383561643835616e-06, + "loss": 1.727, + "step": 271 + }, + { + "epoch": 0.2794042116076014, + "grad_norm": 3.143821954727173, + "learning_rate": 4.40068493150685e-06, + "loss": 1.8045, + "step": 272 + }, + { + "epoch": 0.28043143297380585, + "grad_norm": 3.4248714447021484, + "learning_rate": 4.4178082191780825e-06, + "loss": 1.7649, + "step": 273 + }, + { + "epoch": 0.2814586543400103, + "grad_norm": 2.902162790298462, + "learning_rate": 4.434931506849315e-06, + "loss": 1.7324, + "step": 274 + }, + { + "epoch": 0.2824858757062147, + "grad_norm": 2.8899712562561035, + "learning_rate": 4.4520547945205486e-06, + "loss": 1.6886, + "step": 275 + }, + { + "epoch": 0.2835130970724191, + "grad_norm": 3.2267258167266846, + "learning_rate": 4.469178082191781e-06, + "loss": 1.7442, + "step": 276 + }, + { + "epoch": 0.28454031843862354, + "grad_norm": 3.275735378265381, + "learning_rate": 4.486301369863014e-06, + "loss": 1.7334, + "step": 277 + }, + { + "epoch": 0.28556753980482796, + "grad_norm": 2.991509199142456, + "learning_rate": 4.503424657534247e-06, + "loss": 1.7853, + "step": 278 + }, + { + "epoch": 0.28659476117103233, + "grad_norm": 2.8380391597747803, + "learning_rate": 4.52054794520548e-06, + "loss": 1.6885, + "step": 279 + }, + { + "epoch": 0.28762198253723675, + "grad_norm": 2.909980297088623, + "learning_rate": 4.537671232876713e-06, + "loss": 1.7146, + "step": 280 + }, + { + "epoch": 0.28762198253723675, + "eval_loss": 1.9755980968475342, + "eval_runtime": 37.3524, + "eval_samples_per_second": 26.772, + "eval_steps_per_second": 1.124, + "step": 280 + }, + { + "epoch": 0.2886492039034412, + "grad_norm": 3.2689175605773926, + "learning_rate": 4.554794520547945e-06, + "loss": 1.7747, + "step": 281 + }, + { + "epoch": 0.2896764252696456, + "grad_norm": 2.8242290019989014, + "learning_rate": 4.571917808219179e-06, + "loss": 1.7657, + "step": 282 + }, + { + "epoch": 0.29070364663585, + "grad_norm": 3.03828501701355, + "learning_rate": 4.589041095890411e-06, + "loss": 1.7555, + "step": 283 + }, + { + "epoch": 0.29173086800205444, + "grad_norm": 2.714351177215576, + "learning_rate": 4.606164383561644e-06, + "loss": 1.7055, + "step": 284 + }, + { + "epoch": 0.29275808936825887, + "grad_norm": 2.9540348052978516, + "learning_rate": 4.6232876712328774e-06, + "loss": 1.7258, + "step": 285 + }, + { + "epoch": 0.2937853107344633, + "grad_norm": 3.107060670852661, + "learning_rate": 4.64041095890411e-06, + "loss": 1.7376, + "step": 286 + }, + { + "epoch": 0.2948125321006677, + "grad_norm": 3.3266639709472656, + "learning_rate": 4.657534246575343e-06, + "loss": 1.7063, + "step": 287 + }, + { + "epoch": 0.29583975346687214, + "grad_norm": 3.3113174438476562, + "learning_rate": 4.674657534246575e-06, + "loss": 1.7514, + "step": 288 + }, + { + "epoch": 0.2968669748330765, + "grad_norm": 3.313321352005005, + "learning_rate": 4.691780821917809e-06, + "loss": 1.73, + "step": 289 + }, + { + "epoch": 0.2978941961992809, + "grad_norm": 2.901522159576416, + "learning_rate": 4.7089041095890414e-06, + "loss": 1.7709, + "step": 290 + }, + { + "epoch": 0.2978941961992809, + "eval_loss": 1.9608837366104126, + "eval_runtime": 37.3687, + "eval_samples_per_second": 26.76, + "eval_steps_per_second": 1.124, + "step": 290 + }, + { + "epoch": 0.29892141756548535, + "grad_norm": 3.4208264350891113, + "learning_rate": 4.726027397260274e-06, + "loss": 1.73, + "step": 291 + }, + { + "epoch": 0.29994863893168977, + "grad_norm": 2.8982441425323486, + "learning_rate": 4.7431506849315075e-06, + "loss": 1.7416, + "step": 292 + }, + { + "epoch": 0.3009758602978942, + "grad_norm": 3.2743732929229736, + "learning_rate": 4.76027397260274e-06, + "loss": 1.7443, + "step": 293 + }, + { + "epoch": 0.3020030816640986, + "grad_norm": 3.028764486312866, + "learning_rate": 4.777397260273973e-06, + "loss": 1.6879, + "step": 294 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.3531785011291504, + "learning_rate": 4.7945205479452054e-06, + "loss": 1.7096, + "step": 295 + }, + { + "epoch": 0.30405752439650746, + "grad_norm": 2.5602540969848633, + "learning_rate": 4.811643835616439e-06, + "loss": 1.7125, + "step": 296 + }, + { + "epoch": 0.3050847457627119, + "grad_norm": 3.1823129653930664, + "learning_rate": 4.8287671232876716e-06, + "loss": 1.6531, + "step": 297 + }, + { + "epoch": 0.3061119671289163, + "grad_norm": 2.6700282096862793, + "learning_rate": 4.845890410958904e-06, + "loss": 1.6983, + "step": 298 + }, + { + "epoch": 0.3071391884951207, + "grad_norm": 2.7118732929229736, + "learning_rate": 4.863013698630138e-06, + "loss": 1.7056, + "step": 299 + }, + { + "epoch": 0.3081664098613251, + "grad_norm": 3.3493311405181885, + "learning_rate": 4.88013698630137e-06, + "loss": 1.6944, + "step": 300 + }, + { + "epoch": 0.3081664098613251, + "eval_loss": 1.9489911794662476, + "eval_runtime": 37.35, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 300 + }, + { + "epoch": 0.3091936312275295, + "grad_norm": 2.9436538219451904, + "learning_rate": 4.897260273972603e-06, + "loss": 1.6717, + "step": 301 + }, + { + "epoch": 0.31022085259373394, + "grad_norm": 2.894737482070923, + "learning_rate": 4.914383561643836e-06, + "loss": 1.7895, + "step": 302 + }, + { + "epoch": 0.31124807395993837, + "grad_norm": 2.8645665645599365, + "learning_rate": 4.931506849315069e-06, + "loss": 1.721, + "step": 303 + }, + { + "epoch": 0.3122752953261428, + "grad_norm": 3.050203323364258, + "learning_rate": 4.948630136986302e-06, + "loss": 1.7564, + "step": 304 + }, + { + "epoch": 0.3133025166923472, + "grad_norm": 3.146395206451416, + "learning_rate": 4.965753424657534e-06, + "loss": 1.7938, + "step": 305 + }, + { + "epoch": 0.31432973805855163, + "grad_norm": 3.1021251678466797, + "learning_rate": 4.982876712328768e-06, + "loss": 1.6682, + "step": 306 + }, + { + "epoch": 0.31535695942475606, + "grad_norm": 2.7301125526428223, + "learning_rate": 5e-06, + "loss": 1.6974, + "step": 307 + }, + { + "epoch": 0.3163841807909605, + "grad_norm": 2.7602250576019287, + "learning_rate": 4.998096688237534e-06, + "loss": 1.6523, + "step": 308 + }, + { + "epoch": 0.31741140215716485, + "grad_norm": 2.8149876594543457, + "learning_rate": 4.996193376475067e-06, + "loss": 1.7535, + "step": 309 + }, + { + "epoch": 0.31843862352336927, + "grad_norm": 2.54823637008667, + "learning_rate": 4.9942900647126005e-06, + "loss": 1.63, + "step": 310 + }, + { + "epoch": 0.31843862352336927, + "eval_loss": 1.928893804550171, + "eval_runtime": 37.2236, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3194658448895737, + "grad_norm": 2.7819766998291016, + "learning_rate": 4.992386752950134e-06, + "loss": 1.6461, + "step": 311 + }, + { + "epoch": 0.3204930662557781, + "grad_norm": 2.999209403991699, + "learning_rate": 4.990483441187667e-06, + "loss": 1.6861, + "step": 312 + }, + { + "epoch": 0.32152028762198254, + "grad_norm": 2.7288851737976074, + "learning_rate": 4.9885801294252e-06, + "loss": 1.7076, + "step": 313 + }, + { + "epoch": 0.32254750898818696, + "grad_norm": 2.8722457885742188, + "learning_rate": 4.986676817662733e-06, + "loss": 1.7079, + "step": 314 + }, + { + "epoch": 0.3235747303543914, + "grad_norm": 2.853344678878784, + "learning_rate": 4.9847735059002665e-06, + "loss": 1.6699, + "step": 315 + }, + { + "epoch": 0.3246019517205958, + "grad_norm": 2.7404720783233643, + "learning_rate": 4.9828701941378e-06, + "loss": 1.7423, + "step": 316 + }, + { + "epoch": 0.3256291730868002, + "grad_norm": 2.6062633991241455, + "learning_rate": 4.980966882375333e-06, + "loss": 1.6811, + "step": 317 + }, + { + "epoch": 0.3266563944530046, + "grad_norm": 2.834162712097168, + "learning_rate": 4.979063570612867e-06, + "loss": 1.6575, + "step": 318 + }, + { + "epoch": 0.327683615819209, + "grad_norm": 2.7741048336029053, + "learning_rate": 4.9771602588504e-06, + "loss": 1.6158, + "step": 319 + }, + { + "epoch": 0.32871083718541344, + "grad_norm": 2.6216888427734375, + "learning_rate": 4.975256947087933e-06, + "loss": 1.7295, + "step": 320 + }, + { + "epoch": 0.32871083718541344, + "eval_loss": 1.9309543371200562, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 320 + }, + { + "epoch": 0.32973805855161786, + "grad_norm": 2.6282548904418945, + "learning_rate": 4.973353635325467e-06, + "loss": 1.6266, + "step": 321 + }, + { + "epoch": 0.3307652799178223, + "grad_norm": 2.632124662399292, + "learning_rate": 4.971450323563e-06, + "loss": 1.6887, + "step": 322 + }, + { + "epoch": 0.3317925012840267, + "grad_norm": 2.5844857692718506, + "learning_rate": 4.969547011800533e-06, + "loss": 1.6287, + "step": 323 + }, + { + "epoch": 0.33281972265023113, + "grad_norm": 2.597620964050293, + "learning_rate": 4.967643700038067e-06, + "loss": 1.6876, + "step": 324 + }, + { + "epoch": 0.33384694401643555, + "grad_norm": 2.6529343128204346, + "learning_rate": 4.9657403882756e-06, + "loss": 1.6883, + "step": 325 + }, + { + "epoch": 0.33487416538264, + "grad_norm": 2.592874526977539, + "learning_rate": 4.9638370765131335e-06, + "loss": 1.653, + "step": 326 + }, + { + "epoch": 0.3359013867488444, + "grad_norm": 2.523489236831665, + "learning_rate": 4.961933764750667e-06, + "loss": 1.6501, + "step": 327 + }, + { + "epoch": 0.33692860811504877, + "grad_norm": 2.6721699237823486, + "learning_rate": 4.9600304529882e-06, + "loss": 1.7126, + "step": 328 + }, + { + "epoch": 0.3379558294812532, + "grad_norm": 2.513122797012329, + "learning_rate": 4.9581271412257335e-06, + "loss": 1.7268, + "step": 329 + }, + { + "epoch": 0.3389830508474576, + "grad_norm": 2.571118116378784, + "learning_rate": 4.956223829463266e-06, + "loss": 1.6862, + "step": 330 + }, + { + "epoch": 0.3389830508474576, + "eval_loss": 1.9501315355300903, + "eval_runtime": 37.3287, + "eval_samples_per_second": 26.789, + "eval_steps_per_second": 1.125, + "step": 330 + }, + { + "epoch": 0.34001027221366203, + "grad_norm": 2.5783371925354004, + "learning_rate": 4.954320517700799e-06, + "loss": 1.6041, + "step": 331 + }, + { + "epoch": 0.34103749357986646, + "grad_norm": 2.6829233169555664, + "learning_rate": 4.952417205938333e-06, + "loss": 1.7111, + "step": 332 + }, + { + "epoch": 0.3420647149460709, + "grad_norm": 2.6643247604370117, + "learning_rate": 4.950513894175866e-06, + "loss": 1.6797, + "step": 333 + }, + { + "epoch": 0.3430919363122753, + "grad_norm": 2.776308298110962, + "learning_rate": 4.9486105824133995e-06, + "loss": 1.6766, + "step": 334 + }, + { + "epoch": 0.3441191576784797, + "grad_norm": 2.722667932510376, + "learning_rate": 4.946707270650933e-06, + "loss": 1.6865, + "step": 335 + }, + { + "epoch": 0.34514637904468415, + "grad_norm": 2.7050845623016357, + "learning_rate": 4.944803958888466e-06, + "loss": 1.6967, + "step": 336 + }, + { + "epoch": 0.34617360041088857, + "grad_norm": 3.0009098052978516, + "learning_rate": 4.942900647126e-06, + "loss": 1.714, + "step": 337 + }, + { + "epoch": 0.34720082177709294, + "grad_norm": 2.793621301651001, + "learning_rate": 4.940997335363533e-06, + "loss": 1.6591, + "step": 338 + }, + { + "epoch": 0.34822804314329736, + "grad_norm": 2.844383478164673, + "learning_rate": 4.939094023601066e-06, + "loss": 1.611, + "step": 339 + }, + { + "epoch": 0.3492552645095018, + "grad_norm": 2.829378366470337, + "learning_rate": 4.9371907118386e-06, + "loss": 1.666, + "step": 340 + }, + { + "epoch": 0.3492552645095018, + "eval_loss": 1.96309494972229, + "eval_runtime": 37.3346, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 340 + }, + { + "epoch": 0.3502824858757062, + "grad_norm": 2.9702041149139404, + "learning_rate": 4.935287400076133e-06, + "loss": 1.6703, + "step": 341 + }, + { + "epoch": 0.35130970724191063, + "grad_norm": 2.7555935382843018, + "learning_rate": 4.933384088313666e-06, + "loss": 1.6879, + "step": 342 + }, + { + "epoch": 0.35233692860811505, + "grad_norm": 2.8968169689178467, + "learning_rate": 4.9314807765512e-06, + "loss": 1.6342, + "step": 343 + }, + { + "epoch": 0.3533641499743195, + "grad_norm": 3.010115385055542, + "learning_rate": 4.929577464788733e-06, + "loss": 1.5883, + "step": 344 + }, + { + "epoch": 0.3543913713405239, + "grad_norm": 2.6449785232543945, + "learning_rate": 4.9276741530262665e-06, + "loss": 1.654, + "step": 345 + }, + { + "epoch": 0.3554185927067283, + "grad_norm": 2.976109266281128, + "learning_rate": 4.9257708412638e-06, + "loss": 1.6483, + "step": 346 + }, + { + "epoch": 0.35644581407293274, + "grad_norm": 2.856144666671753, + "learning_rate": 4.923867529501332e-06, + "loss": 1.6536, + "step": 347 + }, + { + "epoch": 0.3574730354391371, + "grad_norm": 2.8142166137695312, + "learning_rate": 4.921964217738866e-06, + "loss": 1.6415, + "step": 348 + }, + { + "epoch": 0.35850025680534153, + "grad_norm": 2.70807147026062, + "learning_rate": 4.920060905976399e-06, + "loss": 1.7099, + "step": 349 + }, + { + "epoch": 0.35952747817154596, + "grad_norm": 2.7715506553649902, + "learning_rate": 4.9181575942139325e-06, + "loss": 1.6947, + "step": 350 + }, + { + "epoch": 0.35952747817154596, + "eval_loss": 1.9507063627243042, + "eval_runtime": 37.3094, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 350 + }, + { + "epoch": 0.3605546995377504, + "grad_norm": 2.8816769123077393, + "learning_rate": 4.916254282451466e-06, + "loss": 1.6906, + "step": 351 + }, + { + "epoch": 0.3615819209039548, + "grad_norm": 3.08797025680542, + "learning_rate": 4.914350970688999e-06, + "loss": 1.7387, + "step": 352 + }, + { + "epoch": 0.3626091422701592, + "grad_norm": 2.6018118858337402, + "learning_rate": 4.9124476589265325e-06, + "loss": 1.6319, + "step": 353 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7790255546569824, + "learning_rate": 4.910544347164066e-06, + "loss": 1.6357, + "step": 354 + }, + { + "epoch": 0.36466358500256807, + "grad_norm": 2.7360289096832275, + "learning_rate": 4.908641035401599e-06, + "loss": 1.7127, + "step": 355 + }, + { + "epoch": 0.3656908063687725, + "grad_norm": 2.6693854331970215, + "learning_rate": 4.906737723639133e-06, + "loss": 1.7747, + "step": 356 + }, + { + "epoch": 0.3667180277349769, + "grad_norm": 2.642868995666504, + "learning_rate": 4.904834411876665e-06, + "loss": 1.6401, + "step": 357 + }, + { + "epoch": 0.3677452491011813, + "grad_norm": 2.7230374813079834, + "learning_rate": 4.902931100114199e-06, + "loss": 1.6833, + "step": 358 + }, + { + "epoch": 0.3687724704673857, + "grad_norm": 2.499457359313965, + "learning_rate": 4.901027788351733e-06, + "loss": 1.6007, + "step": 359 + }, + { + "epoch": 0.3697996918335901, + "grad_norm": 2.7695276737213135, + "learning_rate": 4.899124476589266e-06, + "loss": 1.6486, + "step": 360 + }, + { + "epoch": 0.3697996918335901, + "eval_loss": 1.9434194564819336, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.37082691319979455, + "grad_norm": 2.5715794563293457, + "learning_rate": 4.8972211648267994e-06, + "loss": 1.6183, + "step": 361 + }, + { + "epoch": 0.371854134565999, + "grad_norm": 2.637859582901001, + "learning_rate": 4.895317853064333e-06, + "loss": 1.6648, + "step": 362 + }, + { + "epoch": 0.3728813559322034, + "grad_norm": 2.5601117610931396, + "learning_rate": 4.893414541301866e-06, + "loss": 1.6475, + "step": 363 + }, + { + "epoch": 0.3739085772984078, + "grad_norm": 2.429337739944458, + "learning_rate": 4.891511229539399e-06, + "loss": 1.6435, + "step": 364 + }, + { + "epoch": 0.37493579866461224, + "grad_norm": 2.6890885829925537, + "learning_rate": 4.889607917776932e-06, + "loss": 1.6577, + "step": 365 + }, + { + "epoch": 0.37596302003081666, + "grad_norm": 2.55492901802063, + "learning_rate": 4.887704606014465e-06, + "loss": 1.6523, + "step": 366 + }, + { + "epoch": 0.37699024139702103, + "grad_norm": 2.841362476348877, + "learning_rate": 4.885801294251999e-06, + "loss": 1.6742, + "step": 367 + }, + { + "epoch": 0.37801746276322545, + "grad_norm": 2.8041367530822754, + "learning_rate": 4.883897982489532e-06, + "loss": 1.5946, + "step": 368 + }, + { + "epoch": 0.3790446841294299, + "grad_norm": 2.7135822772979736, + "learning_rate": 4.8819946707270655e-06, + "loss": 1.6909, + "step": 369 + }, + { + "epoch": 0.3800719054956343, + "grad_norm": 2.9020211696624756, + "learning_rate": 4.880091358964599e-06, + "loss": 1.6804, + "step": 370 + }, + { + "epoch": 0.3800719054956343, + "eval_loss": 1.9353317022323608, + "eval_runtime": 37.3188, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 370 + }, + { + "epoch": 0.3810991268618387, + "grad_norm": 2.567310333251953, + "learning_rate": 4.878188047202132e-06, + "loss": 1.5885, + "step": 371 + }, + { + "epoch": 0.38212634822804314, + "grad_norm": 2.910834789276123, + "learning_rate": 4.876284735439666e-06, + "loss": 1.7681, + "step": 372 + }, + { + "epoch": 0.38315356959424757, + "grad_norm": 2.7507543563842773, + "learning_rate": 4.874381423677199e-06, + "loss": 1.7228, + "step": 373 + }, + { + "epoch": 0.384180790960452, + "grad_norm": 2.9230854511260986, + "learning_rate": 4.8724781119147315e-06, + "loss": 1.6491, + "step": 374 + }, + { + "epoch": 0.3852080123266564, + "grad_norm": 2.608980178833008, + "learning_rate": 4.870574800152265e-06, + "loss": 1.6757, + "step": 375 + }, + { + "epoch": 0.38623523369286084, + "grad_norm": 2.9557063579559326, + "learning_rate": 4.868671488389798e-06, + "loss": 1.607, + "step": 376 + }, + { + "epoch": 0.3872624550590652, + "grad_norm": 2.982243537902832, + "learning_rate": 4.8667681766273316e-06, + "loss": 1.6169, + "step": 377 + }, + { + "epoch": 0.3882896764252696, + "grad_norm": 2.775012254714966, + "learning_rate": 4.864864864864866e-06, + "loss": 1.5835, + "step": 378 + }, + { + "epoch": 0.38931689779147405, + "grad_norm": 2.9500741958618164, + "learning_rate": 4.862961553102399e-06, + "loss": 1.6504, + "step": 379 + }, + { + "epoch": 0.39034411915767847, + "grad_norm": 2.889702796936035, + "learning_rate": 4.8610582413399325e-06, + "loss": 1.6814, + "step": 380 + }, + { + "epoch": 0.39034411915767847, + "eval_loss": 1.9426418542861938, + "eval_runtime": 37.3276, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 380 + }, + { + "epoch": 0.3913713405238829, + "grad_norm": 2.5075154304504395, + "learning_rate": 4.859154929577465e-06, + "loss": 1.6368, + "step": 381 + }, + { + "epoch": 0.3923985618900873, + "grad_norm": 2.6461148262023926, + "learning_rate": 4.857251617814998e-06, + "loss": 1.6526, + "step": 382 + }, + { + "epoch": 0.39342578325629174, + "grad_norm": 2.681501626968384, + "learning_rate": 4.855348306052532e-06, + "loss": 1.6846, + "step": 383 + }, + { + "epoch": 0.39445300462249616, + "grad_norm": 2.557311773300171, + "learning_rate": 4.853444994290065e-06, + "loss": 1.6645, + "step": 384 + }, + { + "epoch": 0.3954802259887006, + "grad_norm": 2.6819186210632324, + "learning_rate": 4.8515416825275985e-06, + "loss": 1.6323, + "step": 385 + }, + { + "epoch": 0.396507447354905, + "grad_norm": 2.8803794384002686, + "learning_rate": 4.849638370765132e-06, + "loss": 1.6991, + "step": 386 + }, + { + "epoch": 0.3975346687211094, + "grad_norm": 2.7181358337402344, + "learning_rate": 4.847735059002665e-06, + "loss": 1.6256, + "step": 387 + }, + { + "epoch": 0.3985618900873138, + "grad_norm": 2.5668489933013916, + "learning_rate": 4.8458317472401985e-06, + "loss": 1.58, + "step": 388 + }, + { + "epoch": 0.3995891114535182, + "grad_norm": 2.708804130554199, + "learning_rate": 4.843928435477732e-06, + "loss": 1.67, + "step": 389 + }, + { + "epoch": 0.40061633281972264, + "grad_norm": 2.6565728187561035, + "learning_rate": 4.842025123715265e-06, + "loss": 1.622, + "step": 390 + }, + { + "epoch": 0.40061633281972264, + "eval_loss": 1.9529728889465332, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 390 + }, + { + "epoch": 0.40164355418592707, + "grad_norm": 2.8743324279785156, + "learning_rate": 4.840121811952798e-06, + "loss": 1.6618, + "step": 391 + }, + { + "epoch": 0.4026707755521315, + "grad_norm": 2.86134672164917, + "learning_rate": 4.838218500190331e-06, + "loss": 1.6544, + "step": 392 + }, + { + "epoch": 0.4036979969183359, + "grad_norm": 3.3492071628570557, + "learning_rate": 4.8363151884278645e-06, + "loss": 1.6683, + "step": 393 + }, + { + "epoch": 0.40472521828454033, + "grad_norm": 2.6630825996398926, + "learning_rate": 4.834411876665398e-06, + "loss": 1.6447, + "step": 394 + }, + { + "epoch": 0.40575243965074476, + "grad_norm": 2.820096492767334, + "learning_rate": 4.832508564902931e-06, + "loss": 1.6579, + "step": 395 + }, + { + "epoch": 0.4067796610169492, + "grad_norm": 3.076234817504883, + "learning_rate": 4.830605253140465e-06, + "loss": 1.6039, + "step": 396 + }, + { + "epoch": 0.40780688238315355, + "grad_norm": 2.667719841003418, + "learning_rate": 4.828701941377998e-06, + "loss": 1.6364, + "step": 397 + }, + { + "epoch": 0.40883410374935797, + "grad_norm": 3.2132771015167236, + "learning_rate": 4.826798629615531e-06, + "loss": 1.5906, + "step": 398 + }, + { + "epoch": 0.4098613251155624, + "grad_norm": 2.661484479904175, + "learning_rate": 4.824895317853065e-06, + "loss": 1.613, + "step": 399 + }, + { + "epoch": 0.4108885464817668, + "grad_norm": 3.002636671066284, + "learning_rate": 4.822992006090598e-06, + "loss": 1.6305, + "step": 400 + }, + { + "epoch": 0.4108885464817668, + "eval_loss": 1.9345510005950928, + "eval_runtime": 37.3369, + "eval_samples_per_second": 26.783, + "eval_steps_per_second": 1.125, + "step": 400 + }, + { + "epoch": 0.41191576784797124, + "grad_norm": 2.6405227184295654, + "learning_rate": 4.821088694328131e-06, + "loss": 1.5881, + "step": 401 + }, + { + "epoch": 0.41294298921417566, + "grad_norm": 3.006169080734253, + "learning_rate": 4.819185382565665e-06, + "loss": 1.6126, + "step": 402 + }, + { + "epoch": 0.4139702105803801, + "grad_norm": 2.7614858150482178, + "learning_rate": 4.817282070803198e-06, + "loss": 1.6695, + "step": 403 + }, + { + "epoch": 0.4149974319465845, + "grad_norm": 2.797260046005249, + "learning_rate": 4.8153787590407315e-06, + "loss": 1.6539, + "step": 404 + }, + { + "epoch": 0.41602465331278893, + "grad_norm": 2.8156301975250244, + "learning_rate": 4.813475447278265e-06, + "loss": 1.5845, + "step": 405 + }, + { + "epoch": 0.41705187467899335, + "grad_norm": 2.74100399017334, + "learning_rate": 4.811572135515798e-06, + "loss": 1.6699, + "step": 406 + }, + { + "epoch": 0.4180790960451977, + "grad_norm": 2.7785775661468506, + "learning_rate": 4.809668823753332e-06, + "loss": 1.5987, + "step": 407 + }, + { + "epoch": 0.41910631741140214, + "grad_norm": 2.663504123687744, + "learning_rate": 4.807765511990864e-06, + "loss": 1.5987, + "step": 408 + }, + { + "epoch": 0.42013353877760656, + "grad_norm": 2.903374433517456, + "learning_rate": 4.8058622002283975e-06, + "loss": 1.6302, + "step": 409 + }, + { + "epoch": 0.421160760143811, + "grad_norm": 2.696049690246582, + "learning_rate": 4.803958888465931e-06, + "loss": 1.671, + "step": 410 + }, + { + "epoch": 0.421160760143811, + "eval_loss": 1.9333878755569458, + "eval_runtime": 37.2472, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 410 + }, + { + "epoch": 0.4221879815100154, + "grad_norm": 2.486461877822876, + "learning_rate": 4.802055576703464e-06, + "loss": 1.6282, + "step": 411 + }, + { + "epoch": 0.42321520287621983, + "grad_norm": 2.6812057495117188, + "learning_rate": 4.8001522649409976e-06, + "loss": 1.7144, + "step": 412 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.8370249271392822, + "learning_rate": 4.798248953178531e-06, + "loss": 1.5979, + "step": 413 + }, + { + "epoch": 0.4252696456086287, + "grad_norm": 2.6073036193847656, + "learning_rate": 4.796345641416064e-06, + "loss": 1.5703, + "step": 414 + }, + { + "epoch": 0.4262968669748331, + "grad_norm": 2.5891220569610596, + "learning_rate": 4.794442329653598e-06, + "loss": 1.6212, + "step": 415 + }, + { + "epoch": 0.42732408834103747, + "grad_norm": 2.6296865940093994, + "learning_rate": 4.792539017891131e-06, + "loss": 1.6711, + "step": 416 + }, + { + "epoch": 0.4283513097072419, + "grad_norm": 2.573472499847412, + "learning_rate": 4.790635706128664e-06, + "loss": 1.5994, + "step": 417 + }, + { + "epoch": 0.4293785310734463, + "grad_norm": 2.734464168548584, + "learning_rate": 4.788732394366197e-06, + "loss": 1.6383, + "step": 418 + }, + { + "epoch": 0.43040575243965074, + "grad_norm": 2.6412851810455322, + "learning_rate": 4.786829082603731e-06, + "loss": 1.5646, + "step": 419 + }, + { + "epoch": 0.43143297380585516, + "grad_norm": 2.5065555572509766, + "learning_rate": 4.7849257708412645e-06, + "loss": 1.5636, + "step": 420 + }, + { + "epoch": 0.43143297380585516, + "eval_loss": 1.9357348680496216, + "eval_runtime": 37.372, + "eval_samples_per_second": 26.758, + "eval_steps_per_second": 1.124, + "step": 420 + }, + { + "epoch": 0.4324601951720596, + "grad_norm": 2.4889843463897705, + "learning_rate": 4.783022459078798e-06, + "loss": 1.6457, + "step": 421 + }, + { + "epoch": 0.433487416538264, + "grad_norm": 2.544801950454712, + "learning_rate": 4.781119147316331e-06, + "loss": 1.5897, + "step": 422 + }, + { + "epoch": 0.4345146379044684, + "grad_norm": 2.7834064960479736, + "learning_rate": 4.7792158355538645e-06, + "loss": 1.6174, + "step": 423 + }, + { + "epoch": 0.43554185927067285, + "grad_norm": 2.602952480316162, + "learning_rate": 4.777312523791398e-06, + "loss": 1.6152, + "step": 424 + }, + { + "epoch": 0.43656908063687727, + "grad_norm": 2.629746198654175, + "learning_rate": 4.77540921202893e-06, + "loss": 1.5991, + "step": 425 + }, + { + "epoch": 0.43759630200308164, + "grad_norm": 2.4987270832061768, + "learning_rate": 4.773505900266464e-06, + "loss": 1.6342, + "step": 426 + }, + { + "epoch": 0.43862352336928606, + "grad_norm": 2.607611894607544, + "learning_rate": 4.771602588503997e-06, + "loss": 1.6145, + "step": 427 + }, + { + "epoch": 0.4396507447354905, + "grad_norm": 2.5654876232147217, + "learning_rate": 4.7696992767415305e-06, + "loss": 1.6162, + "step": 428 + }, + { + "epoch": 0.4406779661016949, + "grad_norm": 2.5701870918273926, + "learning_rate": 4.767795964979064e-06, + "loss": 1.6516, + "step": 429 + }, + { + "epoch": 0.44170518746789933, + "grad_norm": 2.8381104469299316, + "learning_rate": 4.765892653216597e-06, + "loss": 1.659, + "step": 430 + }, + { + "epoch": 0.44170518746789933, + "eval_loss": 1.9259562492370605, + "eval_runtime": 37.3906, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 430 + }, + { + "epoch": 0.44273240883410375, + "grad_norm": 2.5777862071990967, + "learning_rate": 4.763989341454131e-06, + "loss": 1.6571, + "step": 431 + }, + { + "epoch": 0.4437596302003082, + "grad_norm": 2.6656861305236816, + "learning_rate": 4.762086029691664e-06, + "loss": 1.5786, + "step": 432 + }, + { + "epoch": 0.4447868515665126, + "grad_norm": 2.724851131439209, + "learning_rate": 4.760182717929197e-06, + "loss": 1.5916, + "step": 433 + }, + { + "epoch": 0.445814072932717, + "grad_norm": 2.532191514968872, + "learning_rate": 4.758279406166731e-06, + "loss": 1.608, + "step": 434 + }, + { + "epoch": 0.44684129429892144, + "grad_norm": 2.6252050399780273, + "learning_rate": 4.756376094404263e-06, + "loss": 1.5297, + "step": 435 + }, + { + "epoch": 0.4478685156651258, + "grad_norm": 2.674530029296875, + "learning_rate": 4.7544727826417966e-06, + "loss": 1.615, + "step": 436 + }, + { + "epoch": 0.44889573703133023, + "grad_norm": 2.9209136962890625, + "learning_rate": 4.75256947087933e-06, + "loss": 1.6749, + "step": 437 + }, + { + "epoch": 0.44992295839753466, + "grad_norm": 2.4959394931793213, + "learning_rate": 4.750666159116863e-06, + "loss": 1.6186, + "step": 438 + }, + { + "epoch": 0.4509501797637391, + "grad_norm": 2.6317503452301025, + "learning_rate": 4.7487628473543975e-06, + "loss": 1.5873, + "step": 439 + }, + { + "epoch": 0.4519774011299435, + "grad_norm": 2.6485610008239746, + "learning_rate": 4.746859535591931e-06, + "loss": 1.5451, + "step": 440 + }, + { + "epoch": 0.4519774011299435, + "eval_loss": 1.9235283136367798, + "eval_runtime": 37.367, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 440 + }, + { + "epoch": 0.4530046224961479, + "grad_norm": 2.491852045059204, + "learning_rate": 4.744956223829464e-06, + "loss": 1.6016, + "step": 441 + }, + { + "epoch": 0.45403184386235235, + "grad_norm": 2.5354514122009277, + "learning_rate": 4.743052912066997e-06, + "loss": 1.5795, + "step": 442 + }, + { + "epoch": 0.45505906522855677, + "grad_norm": 2.7756357192993164, + "learning_rate": 4.74114960030453e-06, + "loss": 1.6299, + "step": 443 + }, + { + "epoch": 0.4560862865947612, + "grad_norm": 2.638906955718994, + "learning_rate": 4.7392462885420635e-06, + "loss": 1.6429, + "step": 444 + }, + { + "epoch": 0.4571135079609656, + "grad_norm": 2.5858829021453857, + "learning_rate": 4.737342976779597e-06, + "loss": 1.5316, + "step": 445 + }, + { + "epoch": 0.45814072932717, + "grad_norm": 2.798567295074463, + "learning_rate": 4.73543966501713e-06, + "loss": 1.6166, + "step": 446 + }, + { + "epoch": 0.4591679506933744, + "grad_norm": 2.512143135070801, + "learning_rate": 4.7335363532546636e-06, + "loss": 1.5952, + "step": 447 + }, + { + "epoch": 0.4601951720595788, + "grad_norm": 2.7063655853271484, + "learning_rate": 4.731633041492197e-06, + "loss": 1.6051, + "step": 448 + }, + { + "epoch": 0.46122239342578325, + "grad_norm": 2.6358017921447754, + "learning_rate": 4.72972972972973e-06, + "loss": 1.6008, + "step": 449 + }, + { + "epoch": 0.4622496147919877, + "grad_norm": 2.75992751121521, + "learning_rate": 4.727826417967264e-06, + "loss": 1.644, + "step": 450 + }, + { + "epoch": 0.4622496147919877, + "eval_loss": 1.927345633506775, + "eval_runtime": 37.3707, + "eval_samples_per_second": 26.759, + "eval_steps_per_second": 1.124, + "step": 450 + }, + { + "epoch": 0.4632768361581921, + "grad_norm": 2.9103775024414062, + "learning_rate": 4.725923106204797e-06, + "loss": 1.6176, + "step": 451 + }, + { + "epoch": 0.4643040575243965, + "grad_norm": 2.627481460571289, + "learning_rate": 4.72401979444233e-06, + "loss": 1.586, + "step": 452 + }, + { + "epoch": 0.46533127889060094, + "grad_norm": 2.8432116508483887, + "learning_rate": 4.722116482679863e-06, + "loss": 1.6303, + "step": 453 + }, + { + "epoch": 0.46635850025680536, + "grad_norm": 2.8088958263397217, + "learning_rate": 4.720213170917396e-06, + "loss": 1.6163, + "step": 454 + }, + { + "epoch": 0.46738572162300973, + "grad_norm": 2.6040408611297607, + "learning_rate": 4.71830985915493e-06, + "loss": 1.6346, + "step": 455 + }, + { + "epoch": 0.46841294298921415, + "grad_norm": 2.8951191902160645, + "learning_rate": 4.716406547392463e-06, + "loss": 1.5859, + "step": 456 + }, + { + "epoch": 0.4694401643554186, + "grad_norm": 2.792473793029785, + "learning_rate": 4.714503235629996e-06, + "loss": 1.6441, + "step": 457 + }, + { + "epoch": 0.470467385721623, + "grad_norm": 2.6793878078460693, + "learning_rate": 4.71259992386753e-06, + "loss": 1.5699, + "step": 458 + }, + { + "epoch": 0.4714946070878274, + "grad_norm": 2.7223684787750244, + "learning_rate": 4.710696612105063e-06, + "loss": 1.6038, + "step": 459 + }, + { + "epoch": 0.47252182845403184, + "grad_norm": 2.8043856620788574, + "learning_rate": 4.708793300342596e-06, + "loss": 1.6018, + "step": 460 + }, + { + "epoch": 0.47252182845403184, + "eval_loss": 1.9213919639587402, + "eval_runtime": 37.3451, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 460 + }, + { + "epoch": 0.47354904982023627, + "grad_norm": 2.7657470703125, + "learning_rate": 4.70688998858013e-06, + "loss": 1.5896, + "step": 461 + }, + { + "epoch": 0.4745762711864407, + "grad_norm": 2.8417739868164062, + "learning_rate": 4.704986676817663e-06, + "loss": 1.5892, + "step": 462 + }, + { + "epoch": 0.4756034925526451, + "grad_norm": 2.5660197734832764, + "learning_rate": 4.7030833650551965e-06, + "loss": 1.615, + "step": 463 + }, + { + "epoch": 0.47663071391884954, + "grad_norm": 2.762880325317383, + "learning_rate": 4.70118005329273e-06, + "loss": 1.5747, + "step": 464 + }, + { + "epoch": 0.4776579352850539, + "grad_norm": 2.7191319465637207, + "learning_rate": 4.699276741530263e-06, + "loss": 1.6529, + "step": 465 + }, + { + "epoch": 0.4786851566512583, + "grad_norm": 2.4995453357696533, + "learning_rate": 4.697373429767797e-06, + "loss": 1.6285, + "step": 466 + }, + { + "epoch": 0.47971237801746275, + "grad_norm": 2.8300375938415527, + "learning_rate": 4.69547011800533e-06, + "loss": 1.6171, + "step": 467 + }, + { + "epoch": 0.48073959938366717, + "grad_norm": 2.6028811931610107, + "learning_rate": 4.693566806242863e-06, + "loss": 1.6014, + "step": 468 + }, + { + "epoch": 0.4817668207498716, + "grad_norm": 3.080156087875366, + "learning_rate": 4.691663494480397e-06, + "loss": 1.6055, + "step": 469 + }, + { + "epoch": 0.482794042116076, + "grad_norm": 2.6707916259765625, + "learning_rate": 4.689760182717929e-06, + "loss": 1.627, + "step": 470 + }, + { + "epoch": 0.482794042116076, + "eval_loss": 1.9015462398529053, + "eval_runtime": 37.3476, + "eval_samples_per_second": 26.776, + "eval_steps_per_second": 1.125, + "step": 470 + }, + { + "epoch": 0.48382126348228044, + "grad_norm": 2.6820614337921143, + "learning_rate": 4.6878568709554626e-06, + "loss": 1.5261, + "step": 471 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.841031789779663, + "learning_rate": 4.685953559192996e-06, + "loss": 1.5903, + "step": 472 + }, + { + "epoch": 0.4858757062146893, + "grad_norm": 2.470867156982422, + "learning_rate": 4.684050247430529e-06, + "loss": 1.6091, + "step": 473 + }, + { + "epoch": 0.4869029275808937, + "grad_norm": 2.4955153465270996, + "learning_rate": 4.682146935668063e-06, + "loss": 1.544, + "step": 474 + }, + { + "epoch": 0.4879301489470981, + "grad_norm": 2.9290695190429688, + "learning_rate": 4.680243623905596e-06, + "loss": 1.5893, + "step": 475 + }, + { + "epoch": 0.4889573703133025, + "grad_norm": 2.5902767181396484, + "learning_rate": 4.678340312143129e-06, + "loss": 1.6117, + "step": 476 + }, + { + "epoch": 0.4899845916795069, + "grad_norm": 2.6067540645599365, + "learning_rate": 4.676437000380663e-06, + "loss": 1.5703, + "step": 477 + }, + { + "epoch": 0.49101181304571134, + "grad_norm": 2.490105152130127, + "learning_rate": 4.674533688618196e-06, + "loss": 1.6931, + "step": 478 + }, + { + "epoch": 0.49203903441191577, + "grad_norm": 2.7736282348632812, + "learning_rate": 4.6726303768557295e-06, + "loss": 1.6204, + "step": 479 + }, + { + "epoch": 0.4930662557781202, + "grad_norm": 2.741196870803833, + "learning_rate": 4.670727065093263e-06, + "loss": 1.5331, + "step": 480 + }, + { + "epoch": 0.4930662557781202, + "eval_loss": 1.9095442295074463, + "eval_runtime": 37.3351, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 480 + }, + { + "epoch": 0.4940934771443246, + "grad_norm": 2.7624332904815674, + "learning_rate": 4.668823753330796e-06, + "loss": 1.5591, + "step": 481 + }, + { + "epoch": 0.49512069851052903, + "grad_norm": 2.86407470703125, + "learning_rate": 4.6669204415683295e-06, + "loss": 1.5844, + "step": 482 + }, + { + "epoch": 0.49614791987673346, + "grad_norm": 2.770030975341797, + "learning_rate": 4.665017129805863e-06, + "loss": 1.6515, + "step": 483 + }, + { + "epoch": 0.4971751412429379, + "grad_norm": 2.569862127304077, + "learning_rate": 4.663113818043396e-06, + "loss": 1.6228, + "step": 484 + }, + { + "epoch": 0.49820236260914225, + "grad_norm": 2.7176733016967773, + "learning_rate": 4.66121050628093e-06, + "loss": 1.5938, + "step": 485 + }, + { + "epoch": 0.49922958397534667, + "grad_norm": 2.4514758586883545, + "learning_rate": 4.659307194518463e-06, + "loss": 1.6307, + "step": 486 + }, + { + "epoch": 0.5002568053415511, + "grad_norm": 2.5439529418945312, + "learning_rate": 4.6574038827559955e-06, + "loss": 1.5855, + "step": 487 + }, + { + "epoch": 0.5012840267077555, + "grad_norm": 2.6395790576934814, + "learning_rate": 4.655500570993529e-06, + "loss": 1.6124, + "step": 488 + }, + { + "epoch": 0.50231124807396, + "grad_norm": 2.69413685798645, + "learning_rate": 4.653597259231062e-06, + "loss": 1.6358, + "step": 489 + }, + { + "epoch": 0.5033384694401644, + "grad_norm": 2.693129777908325, + "learning_rate": 4.651693947468596e-06, + "loss": 1.5218, + "step": 490 + }, + { + "epoch": 0.5033384694401644, + "eval_loss": 1.9201244115829468, + "eval_runtime": 37.3324, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 490 + }, + { + "epoch": 0.5043656908063687, + "grad_norm": 2.630596876144409, + "learning_rate": 4.649790635706129e-06, + "loss": 1.6719, + "step": 491 + }, + { + "epoch": 0.5053929121725732, + "grad_norm": 2.7230262756347656, + "learning_rate": 4.647887323943662e-06, + "loss": 1.5705, + "step": 492 + }, + { + "epoch": 0.5064201335387776, + "grad_norm": 2.529944896697998, + "learning_rate": 4.645984012181196e-06, + "loss": 1.5767, + "step": 493 + }, + { + "epoch": 0.507447354904982, + "grad_norm": 2.546849489212036, + "learning_rate": 4.644080700418729e-06, + "loss": 1.5039, + "step": 494 + }, + { + "epoch": 0.5084745762711864, + "grad_norm": 2.6698246002197266, + "learning_rate": 4.642177388656262e-06, + "loss": 1.6229, + "step": 495 + }, + { + "epoch": 0.5095017976373909, + "grad_norm": 2.7093148231506348, + "learning_rate": 4.640274076893796e-06, + "loss": 1.5797, + "step": 496 + }, + { + "epoch": 0.5105290190035953, + "grad_norm": 2.48629093170166, + "learning_rate": 4.638370765131328e-06, + "loss": 1.6366, + "step": 497 + }, + { + "epoch": 0.5115562403697997, + "grad_norm": 2.5818612575531006, + "learning_rate": 4.636467453368862e-06, + "loss": 1.5608, + "step": 498 + }, + { + "epoch": 0.5125834617360041, + "grad_norm": 2.6263651847839355, + "learning_rate": 4.634564141606396e-06, + "loss": 1.5518, + "step": 499 + }, + { + "epoch": 0.5136106831022085, + "grad_norm": 2.5724668502807617, + "learning_rate": 4.632660829843929e-06, + "loss": 1.6195, + "step": 500 + }, + { + "epoch": 0.5136106831022085, + "eval_loss": 1.9094111919403076, + "eval_runtime": 37.3483, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 500 + }, + { + "epoch": 0.514637904468413, + "grad_norm": 2.5028088092803955, + "learning_rate": 4.630757518081463e-06, + "loss": 1.5454, + "step": 501 + }, + { + "epoch": 0.5156651258346173, + "grad_norm": 2.588115692138672, + "learning_rate": 4.628854206318996e-06, + "loss": 1.6676, + "step": 502 + }, + { + "epoch": 0.5166923472008218, + "grad_norm": 2.9701285362243652, + "learning_rate": 4.626950894556529e-06, + "loss": 1.5822, + "step": 503 + }, + { + "epoch": 0.5177195685670262, + "grad_norm": 2.593867301940918, + "learning_rate": 4.625047582794062e-06, + "loss": 1.6091, + "step": 504 + }, + { + "epoch": 0.5187467899332306, + "grad_norm": 2.559642791748047, + "learning_rate": 4.623144271031595e-06, + "loss": 1.5359, + "step": 505 + }, + { + "epoch": 0.519774011299435, + "grad_norm": 2.4158153533935547, + "learning_rate": 4.6212409592691286e-06, + "loss": 1.5476, + "step": 506 + }, + { + "epoch": 0.5208012326656395, + "grad_norm": 2.666099786758423, + "learning_rate": 4.619337647506662e-06, + "loss": 1.6297, + "step": 507 + }, + { + "epoch": 0.5218284540318439, + "grad_norm": 2.7026472091674805, + "learning_rate": 4.617434335744195e-06, + "loss": 1.5891, + "step": 508 + }, + { + "epoch": 0.5228556753980482, + "grad_norm": 2.4591612815856934, + "learning_rate": 4.615531023981729e-06, + "loss": 1.5546, + "step": 509 + }, + { + "epoch": 0.5238828967642527, + "grad_norm": 2.870255708694458, + "learning_rate": 4.613627712219262e-06, + "loss": 1.5817, + "step": 510 + }, + { + "epoch": 0.5238828967642527, + "eval_loss": 1.910629153251648, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 510 + }, + { + "epoch": 0.5249101181304571, + "grad_norm": 2.653801679611206, + "learning_rate": 4.611724400456795e-06, + "loss": 1.586, + "step": 511 + }, + { + "epoch": 0.5259373394966615, + "grad_norm": 2.4235377311706543, + "learning_rate": 4.609821088694329e-06, + "loss": 1.5886, + "step": 512 + }, + { + "epoch": 0.5269645608628659, + "grad_norm": 2.8782477378845215, + "learning_rate": 4.607917776931862e-06, + "loss": 1.5634, + "step": 513 + }, + { + "epoch": 0.5279917822290704, + "grad_norm": 2.792876958847046, + "learning_rate": 4.606014465169395e-06, + "loss": 1.6073, + "step": 514 + }, + { + "epoch": 0.5290190035952748, + "grad_norm": 2.5579030513763428, + "learning_rate": 4.604111153406928e-06, + "loss": 1.6161, + "step": 515 + }, + { + "epoch": 0.5300462249614792, + "grad_norm": 2.6486189365386963, + "learning_rate": 4.602207841644461e-06, + "loss": 1.5297, + "step": 516 + }, + { + "epoch": 0.5310734463276836, + "grad_norm": 2.9047019481658936, + "learning_rate": 4.600304529881995e-06, + "loss": 1.626, + "step": 517 + }, + { + "epoch": 0.5321006676938881, + "grad_norm": 2.9164223670959473, + "learning_rate": 4.598401218119528e-06, + "loss": 1.6051, + "step": 518 + }, + { + "epoch": 0.5331278890600925, + "grad_norm": 2.544311046600342, + "learning_rate": 4.5964979063570614e-06, + "loss": 1.6247, + "step": 519 + }, + { + "epoch": 0.5341551104262968, + "grad_norm": 2.8632311820983887, + "learning_rate": 4.594594594594596e-06, + "loss": 1.5615, + "step": 520 + }, + { + "epoch": 0.5341551104262968, + "eval_loss": 1.9171669483184814, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 520 + }, + { + "epoch": 0.5351823317925013, + "grad_norm": 2.507697105407715, + "learning_rate": 4.592691282832128e-06, + "loss": 1.5929, + "step": 521 + }, + { + "epoch": 0.5362095531587057, + "grad_norm": 2.657752752304077, + "learning_rate": 4.5907879710696615e-06, + "loss": 1.6159, + "step": 522 + }, + { + "epoch": 0.5372367745249101, + "grad_norm": 2.32254695892334, + "learning_rate": 4.588884659307195e-06, + "loss": 1.5318, + "step": 523 + }, + { + "epoch": 0.5382639958911145, + "grad_norm": 2.5979015827178955, + "learning_rate": 4.586981347544728e-06, + "loss": 1.5999, + "step": 524 + }, + { + "epoch": 0.539291217257319, + "grad_norm": 2.839956521987915, + "learning_rate": 4.585078035782262e-06, + "loss": 1.547, + "step": 525 + }, + { + "epoch": 0.5403184386235234, + "grad_norm": 2.700131893157959, + "learning_rate": 4.583174724019795e-06, + "loss": 1.5831, + "step": 526 + }, + { + "epoch": 0.5413456599897278, + "grad_norm": 2.365183115005493, + "learning_rate": 4.581271412257328e-06, + "loss": 1.5357, + "step": 527 + }, + { + "epoch": 0.5423728813559322, + "grad_norm": 2.675567865371704, + "learning_rate": 4.579368100494862e-06, + "loss": 1.5212, + "step": 528 + }, + { + "epoch": 0.5434001027221366, + "grad_norm": 2.9096713066101074, + "learning_rate": 4.577464788732395e-06, + "loss": 1.6196, + "step": 529 + }, + { + "epoch": 0.544427324088341, + "grad_norm": 2.6186270713806152, + "learning_rate": 4.575561476969928e-06, + "loss": 1.5895, + "step": 530 + }, + { + "epoch": 0.544427324088341, + "eval_loss": 1.926998496055603, + "eval_runtime": 37.2248, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 530 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.7291345596313477, + "learning_rate": 4.573658165207461e-06, + "loss": 1.6335, + "step": 531 + }, + { + "epoch": 0.5464817668207499, + "grad_norm": 2.7084522247314453, + "learning_rate": 4.571754853444994e-06, + "loss": 1.5968, + "step": 532 + }, + { + "epoch": 0.5475089881869543, + "grad_norm": 2.699721336364746, + "learning_rate": 4.569851541682528e-06, + "loss": 1.5775, + "step": 533 + }, + { + "epoch": 0.5485362095531587, + "grad_norm": 2.602526903152466, + "learning_rate": 4.567948229920061e-06, + "loss": 1.5497, + "step": 534 + }, + { + "epoch": 0.5495634309193631, + "grad_norm": 2.597362518310547, + "learning_rate": 4.566044918157594e-06, + "loss": 1.5218, + "step": 535 + }, + { + "epoch": 0.5505906522855676, + "grad_norm": 2.6624414920806885, + "learning_rate": 4.564141606395128e-06, + "loss": 1.5553, + "step": 536 + }, + { + "epoch": 0.551617873651772, + "grad_norm": 2.750807285308838, + "learning_rate": 4.562238294632661e-06, + "loss": 1.6078, + "step": 537 + }, + { + "epoch": 0.5526450950179764, + "grad_norm": 2.7480952739715576, + "learning_rate": 4.5603349828701945e-06, + "loss": 1.5707, + "step": 538 + }, + { + "epoch": 0.5536723163841808, + "grad_norm": 2.6287965774536133, + "learning_rate": 4.558431671107728e-06, + "loss": 1.5961, + "step": 539 + }, + { + "epoch": 0.5546995377503852, + "grad_norm": 2.721311569213867, + "learning_rate": 4.556528359345261e-06, + "loss": 1.6335, + "step": 540 + }, + { + "epoch": 0.5546995377503852, + "eval_loss": 1.9290324449539185, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5557267591165896, + "grad_norm": 2.470325469970703, + "learning_rate": 4.5546250475827946e-06, + "loss": 1.5645, + "step": 541 + }, + { + "epoch": 0.556753980482794, + "grad_norm": 2.6089508533477783, + "learning_rate": 4.552721735820328e-06, + "loss": 1.5256, + "step": 542 + }, + { + "epoch": 0.5577812018489985, + "grad_norm": 2.541785478591919, + "learning_rate": 4.550818424057861e-06, + "loss": 1.5641, + "step": 543 + }, + { + "epoch": 0.5588084232152029, + "grad_norm": 2.6722488403320312, + "learning_rate": 4.548915112295395e-06, + "loss": 1.5708, + "step": 544 + }, + { + "epoch": 0.5598356445814073, + "grad_norm": 2.583695650100708, + "learning_rate": 4.547011800532928e-06, + "loss": 1.5799, + "step": 545 + }, + { + "epoch": 0.5608628659476117, + "grad_norm": 2.956885576248169, + "learning_rate": 4.545108488770461e-06, + "loss": 1.5345, + "step": 546 + }, + { + "epoch": 0.5618900873138162, + "grad_norm": 2.5534746646881104, + "learning_rate": 4.543205177007995e-06, + "loss": 1.636, + "step": 547 + }, + { + "epoch": 0.5629173086800205, + "grad_norm": 2.8133325576782227, + "learning_rate": 4.541301865245527e-06, + "loss": 1.5678, + "step": 548 + }, + { + "epoch": 0.5639445300462249, + "grad_norm": 2.5969228744506836, + "learning_rate": 4.539398553483061e-06, + "loss": 1.5975, + "step": 549 + }, + { + "epoch": 0.5649717514124294, + "grad_norm": 2.7594687938690186, + "learning_rate": 4.537495241720594e-06, + "loss": 1.5622, + "step": 550 + }, + { + "epoch": 0.5649717514124294, + "eval_loss": 1.8870681524276733, + "eval_runtime": 37.2094, + "eval_samples_per_second": 26.875, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.5659989727786338, + "grad_norm": 2.7413411140441895, + "learning_rate": 4.535591929958127e-06, + "loss": 1.6031, + "step": 551 + }, + { + "epoch": 0.5670261941448382, + "grad_norm": 2.6182661056518555, + "learning_rate": 4.533688618195661e-06, + "loss": 1.5828, + "step": 552 + }, + { + "epoch": 0.5680534155110426, + "grad_norm": 2.5016422271728516, + "learning_rate": 4.531785306433194e-06, + "loss": 1.5712, + "step": 553 + }, + { + "epoch": 0.5690806368772471, + "grad_norm": 2.7788732051849365, + "learning_rate": 4.529881994670727e-06, + "loss": 1.5472, + "step": 554 + }, + { + "epoch": 0.5701078582434514, + "grad_norm": 2.7602436542510986, + "learning_rate": 4.527978682908261e-06, + "loss": 1.5526, + "step": 555 + }, + { + "epoch": 0.5711350796096559, + "grad_norm": 2.5898544788360596, + "learning_rate": 4.526075371145794e-06, + "loss": 1.5644, + "step": 556 + }, + { + "epoch": 0.5721623009758603, + "grad_norm": 2.439344882965088, + "learning_rate": 4.5241720593833275e-06, + "loss": 1.5892, + "step": 557 + }, + { + "epoch": 0.5731895223420647, + "grad_norm": 2.6324563026428223, + "learning_rate": 4.52226874762086e-06, + "loss": 1.5729, + "step": 558 + }, + { + "epoch": 0.5742167437082691, + "grad_norm": 2.674042224884033, + "learning_rate": 4.520365435858393e-06, + "loss": 1.6016, + "step": 559 + }, + { + "epoch": 0.5752439650744735, + "grad_norm": 2.692619800567627, + "learning_rate": 4.518462124095928e-06, + "loss": 1.6229, + "step": 560 + }, + { + "epoch": 0.5752439650744735, + "eval_loss": 1.8689097166061401, + "eval_runtime": 37.2177, + "eval_samples_per_second": 26.869, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.576271186440678, + "grad_norm": 2.636924982070923, + "learning_rate": 4.516558812333461e-06, + "loss": 1.5748, + "step": 561 + }, + { + "epoch": 0.5772984078068824, + "grad_norm": 2.549285888671875, + "learning_rate": 4.514655500570994e-06, + "loss": 1.5656, + "step": 562 + }, + { + "epoch": 0.5783256291730868, + "grad_norm": 2.5873067378997803, + "learning_rate": 4.512752188808528e-06, + "loss": 1.631, + "step": 563 + }, + { + "epoch": 0.5793528505392912, + "grad_norm": 2.552121162414551, + "learning_rate": 4.510848877046061e-06, + "loss": 1.6334, + "step": 564 + }, + { + "epoch": 0.5803800719054957, + "grad_norm": 2.697615146636963, + "learning_rate": 4.5089455652835936e-06, + "loss": 1.5951, + "step": 565 + }, + { + "epoch": 0.5814072932717, + "grad_norm": 2.64585280418396, + "learning_rate": 4.507042253521127e-06, + "loss": 1.556, + "step": 566 + }, + { + "epoch": 0.5824345146379045, + "grad_norm": 2.6383707523345947, + "learning_rate": 4.50513894175866e-06, + "loss": 1.5184, + "step": 567 + }, + { + "epoch": 0.5834617360041089, + "grad_norm": 2.7571511268615723, + "learning_rate": 4.503235629996194e-06, + "loss": 1.5984, + "step": 568 + }, + { + "epoch": 0.5844889573703133, + "grad_norm": 2.6637237071990967, + "learning_rate": 4.501332318233727e-06, + "loss": 1.5989, + "step": 569 + }, + { + "epoch": 0.5855161787365177, + "grad_norm": 2.725754976272583, + "learning_rate": 4.49942900647126e-06, + "loss": 1.5594, + "step": 570 + }, + { + "epoch": 0.5855161787365177, + "eval_loss": 1.8722314834594727, + "eval_runtime": 37.1859, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 570 + }, + { + "epoch": 0.5865434001027221, + "grad_norm": 2.6657378673553467, + "learning_rate": 4.497525694708794e-06, + "loss": 1.5954, + "step": 571 + }, + { + "epoch": 0.5875706214689266, + "grad_norm": 2.78940486907959, + "learning_rate": 4.495622382946327e-06, + "loss": 1.592, + "step": 572 + }, + { + "epoch": 0.588597842835131, + "grad_norm": 2.580956220626831, + "learning_rate": 4.4937190711838605e-06, + "loss": 1.5444, + "step": 573 + }, + { + "epoch": 0.5896250642013354, + "grad_norm": 2.7783710956573486, + "learning_rate": 4.491815759421394e-06, + "loss": 1.6257, + "step": 574 + }, + { + "epoch": 0.5906522855675398, + "grad_norm": 2.9381537437438965, + "learning_rate": 4.489912447658926e-06, + "loss": 1.5489, + "step": 575 + }, + { + "epoch": 0.5916795069337443, + "grad_norm": 2.7915165424346924, + "learning_rate": 4.48800913589646e-06, + "loss": 1.5471, + "step": 576 + }, + { + "epoch": 0.5927067282999486, + "grad_norm": 2.5928280353546143, + "learning_rate": 4.486105824133993e-06, + "loss": 1.5864, + "step": 577 + }, + { + "epoch": 0.593733949666153, + "grad_norm": 2.844512462615967, + "learning_rate": 4.4842025123715264e-06, + "loss": 1.5841, + "step": 578 + }, + { + "epoch": 0.5947611710323575, + "grad_norm": 2.562718391418457, + "learning_rate": 4.48229920060906e-06, + "loss": 1.5476, + "step": 579 + }, + { + "epoch": 0.5957883923985619, + "grad_norm": 2.535759925842285, + "learning_rate": 4.480395888846594e-06, + "loss": 1.5274, + "step": 580 + }, + { + "epoch": 0.5957883923985619, + "eval_loss": 1.8881579637527466, + "eval_runtime": 37.2533, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.5968156137647663, + "grad_norm": 2.6940653324127197, + "learning_rate": 4.478492577084127e-06, + "loss": 1.6377, + "step": 581 + }, + { + "epoch": 0.5978428351309707, + "grad_norm": 2.4839298725128174, + "learning_rate": 4.47658926532166e-06, + "loss": 1.534, + "step": 582 + }, + { + "epoch": 0.5988700564971752, + "grad_norm": 2.5035688877105713, + "learning_rate": 4.474685953559193e-06, + "loss": 1.5764, + "step": 583 + }, + { + "epoch": 0.5998972778633795, + "grad_norm": 2.791907548904419, + "learning_rate": 4.472782641796727e-06, + "loss": 1.704, + "step": 584 + }, + { + "epoch": 0.600924499229584, + "grad_norm": 2.5384254455566406, + "learning_rate": 4.47087933003426e-06, + "loss": 1.5184, + "step": 585 + }, + { + "epoch": 0.6019517205957884, + "grad_norm": 2.4990930557250977, + "learning_rate": 4.468976018271793e-06, + "loss": 1.5335, + "step": 586 + }, + { + "epoch": 0.6029789419619929, + "grad_norm": 2.470872163772583, + "learning_rate": 4.467072706509327e-06, + "loss": 1.5464, + "step": 587 + }, + { + "epoch": 0.6040061633281972, + "grad_norm": 2.539036989212036, + "learning_rate": 4.46516939474686e-06, + "loss": 1.5793, + "step": 588 + }, + { + "epoch": 0.6050333846944016, + "grad_norm": 2.495309829711914, + "learning_rate": 4.463266082984393e-06, + "loss": 1.5312, + "step": 589 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.5937912464141846, + "learning_rate": 4.461362771221927e-06, + "loss": 1.6508, + "step": 590 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.8893816471099854, + "eval_runtime": 37.2539, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 590 + }, + { + "epoch": 0.6070878274268104, + "grad_norm": 2.53300404548645, + "learning_rate": 4.45945945945946e-06, + "loss": 1.5408, + "step": 591 + }, + { + "epoch": 0.6081150487930149, + "grad_norm": 2.5760276317596436, + "learning_rate": 4.457556147696993e-06, + "loss": 1.5575, + "step": 592 + }, + { + "epoch": 0.6091422701592193, + "grad_norm": 2.638353109359741, + "learning_rate": 4.455652835934526e-06, + "loss": 1.6286, + "step": 593 + }, + { + "epoch": 0.6101694915254238, + "grad_norm": 2.916550397872925, + "learning_rate": 4.453749524172059e-06, + "loss": 1.6334, + "step": 594 + }, + { + "epoch": 0.6111967128916281, + "grad_norm": 2.560878038406372, + "learning_rate": 4.451846212409593e-06, + "loss": 1.5806, + "step": 595 + }, + { + "epoch": 0.6122239342578326, + "grad_norm": 2.637882947921753, + "learning_rate": 4.449942900647126e-06, + "loss": 1.5791, + "step": 596 + }, + { + "epoch": 0.613251155624037, + "grad_norm": 2.752870798110962, + "learning_rate": 4.4480395888846595e-06, + "loss": 1.5117, + "step": 597 + }, + { + "epoch": 0.6142783769902413, + "grad_norm": 2.702087640762329, + "learning_rate": 4.446136277122193e-06, + "loss": 1.5783, + "step": 598 + }, + { + "epoch": 0.6153055983564458, + "grad_norm": 2.504819631576538, + "learning_rate": 4.444232965359726e-06, + "loss": 1.5136, + "step": 599 + }, + { + "epoch": 0.6163328197226502, + "grad_norm": 2.521670341491699, + "learning_rate": 4.4423296535972596e-06, + "loss": 1.5741, + "step": 600 + }, + { + "epoch": 0.6163328197226502, + "eval_loss": 1.8973714113235474, + "eval_runtime": 37.3167, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 600 + }, + { + "epoch": 0.6173600410888547, + "grad_norm": 2.927755355834961, + "learning_rate": 4.440426341834793e-06, + "loss": 1.5262, + "step": 601 + }, + { + "epoch": 0.618387262455059, + "grad_norm": 2.477053642272949, + "learning_rate": 4.438523030072326e-06, + "loss": 1.572, + "step": 602 + }, + { + "epoch": 0.6194144838212635, + "grad_norm": 2.6500253677368164, + "learning_rate": 4.43661971830986e-06, + "loss": 1.596, + "step": 603 + }, + { + "epoch": 0.6204417051874679, + "grad_norm": 2.5134825706481934, + "learning_rate": 4.434716406547393e-06, + "loss": 1.5432, + "step": 604 + }, + { + "epoch": 0.6214689265536724, + "grad_norm": 2.4142072200775146, + "learning_rate": 4.432813094784926e-06, + "loss": 1.5391, + "step": 605 + }, + { + "epoch": 0.6224961479198767, + "grad_norm": 2.752556800842285, + "learning_rate": 4.43090978302246e-06, + "loss": 1.6055, + "step": 606 + }, + { + "epoch": 0.6235233692860811, + "grad_norm": 2.5731008052825928, + "learning_rate": 4.429006471259993e-06, + "loss": 1.5648, + "step": 607 + }, + { + "epoch": 0.6245505906522856, + "grad_norm": 2.531656503677368, + "learning_rate": 4.4271031594975265e-06, + "loss": 1.5492, + "step": 608 + }, + { + "epoch": 0.6255778120184899, + "grad_norm": 2.6665475368499756, + "learning_rate": 4.425199847735059e-06, + "loss": 1.5194, + "step": 609 + }, + { + "epoch": 0.6266050333846944, + "grad_norm": 2.437051773071289, + "learning_rate": 4.423296535972592e-06, + "loss": 1.5271, + "step": 610 + }, + { + "epoch": 0.6266050333846944, + "eval_loss": 1.88953697681427, + "eval_runtime": 36.8703, + "eval_samples_per_second": 27.122, + "eval_steps_per_second": 1.139, + "step": 610 + }, + { + "epoch": 0.6276322547508988, + "grad_norm": 2.5932440757751465, + "learning_rate": 4.421393224210126e-06, + "loss": 1.5563, + "step": 611 + }, + { + "epoch": 0.6286594761171033, + "grad_norm": 2.589447021484375, + "learning_rate": 4.419489912447659e-06, + "loss": 1.5207, + "step": 612 + }, + { + "epoch": 0.6296866974833076, + "grad_norm": 2.570096731185913, + "learning_rate": 4.4175866006851924e-06, + "loss": 1.5995, + "step": 613 + }, + { + "epoch": 0.6307139188495121, + "grad_norm": 2.5356791019439697, + "learning_rate": 4.415683288922726e-06, + "loss": 1.5703, + "step": 614 + }, + { + "epoch": 0.6317411402157165, + "grad_norm": 2.705193281173706, + "learning_rate": 4.413779977160259e-06, + "loss": 1.5405, + "step": 615 + }, + { + "epoch": 0.632768361581921, + "grad_norm": 2.555168390274048, + "learning_rate": 4.4118766653977925e-06, + "loss": 1.5487, + "step": 616 + }, + { + "epoch": 0.6337955829481253, + "grad_norm": 2.804696559906006, + "learning_rate": 4.409973353635326e-06, + "loss": 1.6094, + "step": 617 + }, + { + "epoch": 0.6348228043143297, + "grad_norm": 2.6889376640319824, + "learning_rate": 4.408070041872859e-06, + "loss": 1.5656, + "step": 618 + }, + { + "epoch": 0.6358500256805342, + "grad_norm": 2.576746702194214, + "learning_rate": 4.406166730110393e-06, + "loss": 1.5244, + "step": 619 + }, + { + "epoch": 0.6368772470467385, + "grad_norm": 2.811156749725342, + "learning_rate": 4.404263418347925e-06, + "loss": 1.5153, + "step": 620 + }, + { + "epoch": 0.6368772470467385, + "eval_loss": 1.8949775695800781, + "eval_runtime": 36.978, + "eval_samples_per_second": 27.043, + "eval_steps_per_second": 1.136, + "step": 620 + }, + { + "epoch": 0.637904468412943, + "grad_norm": 2.6560428142547607, + "learning_rate": 4.402360106585459e-06, + "loss": 1.5347, + "step": 621 + }, + { + "epoch": 0.6389316897791474, + "grad_norm": 2.668100118637085, + "learning_rate": 4.400456794822993e-06, + "loss": 1.5236, + "step": 622 + }, + { + "epoch": 0.6399589111453519, + "grad_norm": 2.925978660583496, + "learning_rate": 4.398553483060526e-06, + "loss": 1.5494, + "step": 623 + }, + { + "epoch": 0.6409861325115562, + "grad_norm": 2.5964415073394775, + "learning_rate": 4.396650171298059e-06, + "loss": 1.522, + "step": 624 + }, + { + "epoch": 0.6420133538777607, + "grad_norm": 2.643888235092163, + "learning_rate": 4.394746859535593e-06, + "loss": 1.5332, + "step": 625 + }, + { + "epoch": 0.6430405752439651, + "grad_norm": 2.805192708969116, + "learning_rate": 4.392843547773125e-06, + "loss": 1.5031, + "step": 626 + }, + { + "epoch": 0.6440677966101694, + "grad_norm": 2.6862149238586426, + "learning_rate": 4.390940236010659e-06, + "loss": 1.5335, + "step": 627 + }, + { + "epoch": 0.6450950179763739, + "grad_norm": 2.5579185485839844, + "learning_rate": 4.389036924248192e-06, + "loss": 1.5466, + "step": 628 + }, + { + "epoch": 0.6461222393425783, + "grad_norm": 2.4550797939300537, + "learning_rate": 4.387133612485725e-06, + "loss": 1.5776, + "step": 629 + }, + { + "epoch": 0.6471494607087828, + "grad_norm": 2.6051297187805176, + "learning_rate": 4.385230300723259e-06, + "loss": 1.5435, + "step": 630 + }, + { + "epoch": 0.6471494607087828, + "eval_loss": 1.8860505819320679, + "eval_runtime": 36.989, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 630 + }, + { + "epoch": 0.6481766820749871, + "grad_norm": 2.3723294734954834, + "learning_rate": 4.383326988960792e-06, + "loss": 1.5413, + "step": 631 + }, + { + "epoch": 0.6492039034411916, + "grad_norm": 2.44370174407959, + "learning_rate": 4.3814236771983255e-06, + "loss": 1.5152, + "step": 632 + }, + { + "epoch": 0.650231124807396, + "grad_norm": 2.474081039428711, + "learning_rate": 4.379520365435859e-06, + "loss": 1.579, + "step": 633 + }, + { + "epoch": 0.6512583461736005, + "grad_norm": 2.837843179702759, + "learning_rate": 4.377617053673392e-06, + "loss": 1.559, + "step": 634 + }, + { + "epoch": 0.6522855675398048, + "grad_norm": 2.593322515487671, + "learning_rate": 4.3757137419109256e-06, + "loss": 1.552, + "step": 635 + }, + { + "epoch": 0.6533127889060092, + "grad_norm": 3.018051862716675, + "learning_rate": 4.373810430148459e-06, + "loss": 1.6231, + "step": 636 + }, + { + "epoch": 0.6543400102722137, + "grad_norm": 2.6338863372802734, + "learning_rate": 4.3719071183859914e-06, + "loss": 1.517, + "step": 637 + }, + { + "epoch": 0.655367231638418, + "grad_norm": 2.544140577316284, + "learning_rate": 4.370003806623525e-06, + "loss": 1.5763, + "step": 638 + }, + { + "epoch": 0.6563944530046225, + "grad_norm": 2.6375792026519775, + "learning_rate": 4.368100494861058e-06, + "loss": 1.5098, + "step": 639 + }, + { + "epoch": 0.6574216743708269, + "grad_norm": 2.5452051162719727, + "learning_rate": 4.3661971830985915e-06, + "loss": 1.5671, + "step": 640 + }, + { + "epoch": 0.6574216743708269, + "eval_loss": 1.881069540977478, + "eval_runtime": 36.9961, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 640 + }, + { + "epoch": 0.6584488957370314, + "grad_norm": 2.623411178588867, + "learning_rate": 4.364293871336126e-06, + "loss": 1.5215, + "step": 641 + }, + { + "epoch": 0.6594761171032357, + "grad_norm": 2.6149892807006836, + "learning_rate": 4.362390559573659e-06, + "loss": 1.5507, + "step": 642 + }, + { + "epoch": 0.6605033384694402, + "grad_norm": 2.5570266246795654, + "learning_rate": 4.3604872478111925e-06, + "loss": 1.5143, + "step": 643 + }, + { + "epoch": 0.6615305598356446, + "grad_norm": 2.4813005924224854, + "learning_rate": 4.358583936048725e-06, + "loss": 1.4518, + "step": 644 + }, + { + "epoch": 0.662557781201849, + "grad_norm": 3.0318472385406494, + "learning_rate": 4.356680624286258e-06, + "loss": 1.5812, + "step": 645 + }, + { + "epoch": 0.6635850025680534, + "grad_norm": 2.46936297416687, + "learning_rate": 4.354777312523792e-06, + "loss": 1.5396, + "step": 646 + }, + { + "epoch": 0.6646122239342578, + "grad_norm": 2.5295984745025635, + "learning_rate": 4.352874000761325e-06, + "loss": 1.586, + "step": 647 + }, + { + "epoch": 0.6656394453004623, + "grad_norm": 2.7826883792877197, + "learning_rate": 4.3509706889988584e-06, + "loss": 1.5339, + "step": 648 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.4273526668548584, + "learning_rate": 4.349067377236392e-06, + "loss": 1.598, + "step": 649 + }, + { + "epoch": 0.6676938880328711, + "grad_norm": 2.644308090209961, + "learning_rate": 4.347164065473925e-06, + "loss": 1.5889, + "step": 650 + }, + { + "epoch": 0.6676938880328711, + "eval_loss": 1.8861087560653687, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 650 + }, + { + "epoch": 0.6687211093990755, + "grad_norm": 2.827376127243042, + "learning_rate": 4.3452607537114585e-06, + "loss": 1.606, + "step": 651 + }, + { + "epoch": 0.66974833076528, + "grad_norm": 2.7463080883026123, + "learning_rate": 4.343357441948992e-06, + "loss": 1.5208, + "step": 652 + }, + { + "epoch": 0.6707755521314843, + "grad_norm": 2.6082730293273926, + "learning_rate": 4.341454130186525e-06, + "loss": 1.5597, + "step": 653 + }, + { + "epoch": 0.6718027734976888, + "grad_norm": 2.7131078243255615, + "learning_rate": 4.339550818424058e-06, + "loss": 1.526, + "step": 654 + }, + { + "epoch": 0.6728299948638932, + "grad_norm": 2.6635708808898926, + "learning_rate": 4.337647506661591e-06, + "loss": 1.5052, + "step": 655 + }, + { + "epoch": 0.6738572162300975, + "grad_norm": 2.5686581134796143, + "learning_rate": 4.3357441948991245e-06, + "loss": 1.5374, + "step": 656 + }, + { + "epoch": 0.674884437596302, + "grad_norm": 3.0532894134521484, + "learning_rate": 4.333840883136658e-06, + "loss": 1.5787, + "step": 657 + }, + { + "epoch": 0.6759116589625064, + "grad_norm": 2.6494507789611816, + "learning_rate": 4.331937571374191e-06, + "loss": 1.5121, + "step": 658 + }, + { + "epoch": 0.6769388803287109, + "grad_norm": 2.748142719268799, + "learning_rate": 4.3300342596117246e-06, + "loss": 1.5294, + "step": 659 + }, + { + "epoch": 0.6779661016949152, + "grad_norm": 2.837799310684204, + "learning_rate": 4.328130947849258e-06, + "loss": 1.5424, + "step": 660 + }, + { + "epoch": 0.6779661016949152, + "eval_loss": 1.8644919395446777, + "eval_runtime": 37.0096, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 660 + }, + { + "epoch": 0.6789933230611197, + "grad_norm": 2.7405757904052734, + "learning_rate": 4.326227636086791e-06, + "loss": 1.4949, + "step": 661 + }, + { + "epoch": 0.6800205444273241, + "grad_norm": 2.545466661453247, + "learning_rate": 4.324324324324325e-06, + "loss": 1.5445, + "step": 662 + }, + { + "epoch": 0.6810477657935285, + "grad_norm": 2.5715537071228027, + "learning_rate": 4.322421012561858e-06, + "loss": 1.5468, + "step": 663 + }, + { + "epoch": 0.6820749871597329, + "grad_norm": 2.7073662281036377, + "learning_rate": 4.320517700799391e-06, + "loss": 1.5293, + "step": 664 + }, + { + "epoch": 0.6831022085259374, + "grad_norm": 2.62264347076416, + "learning_rate": 4.318614389036925e-06, + "loss": 1.6258, + "step": 665 + }, + { + "epoch": 0.6841294298921418, + "grad_norm": 2.4506373405456543, + "learning_rate": 4.316711077274458e-06, + "loss": 1.5845, + "step": 666 + }, + { + "epoch": 0.6851566512583461, + "grad_norm": 2.849473237991333, + "learning_rate": 4.3148077655119915e-06, + "loss": 1.4959, + "step": 667 + }, + { + "epoch": 0.6861838726245506, + "grad_norm": 2.5865421295166016, + "learning_rate": 4.312904453749525e-06, + "loss": 1.599, + "step": 668 + }, + { + "epoch": 0.687211093990755, + "grad_norm": 2.4636387825012207, + "learning_rate": 4.311001141987058e-06, + "loss": 1.5495, + "step": 669 + }, + { + "epoch": 0.6882383153569595, + "grad_norm": 2.638047218322754, + "learning_rate": 4.3090978302245916e-06, + "loss": 1.5507, + "step": 670 + }, + { + "epoch": 0.6882383153569595, + "eval_loss": 1.8834826946258545, + "eval_runtime": 37.0335, + "eval_samples_per_second": 27.003, + "eval_steps_per_second": 1.134, + "step": 670 + }, + { + "epoch": 0.6892655367231638, + "grad_norm": 2.669201612472534, + "learning_rate": 4.307194518462124e-06, + "loss": 1.5494, + "step": 671 + }, + { + "epoch": 0.6902927580893683, + "grad_norm": 2.634634494781494, + "learning_rate": 4.3052912066996574e-06, + "loss": 1.6177, + "step": 672 + }, + { + "epoch": 0.6913199794555727, + "grad_norm": 2.5489139556884766, + "learning_rate": 4.303387894937191e-06, + "loss": 1.5876, + "step": 673 + }, + { + "epoch": 0.6923472008217771, + "grad_norm": 2.580073833465576, + "learning_rate": 4.301484583174724e-06, + "loss": 1.5106, + "step": 674 + }, + { + "epoch": 0.6933744221879815, + "grad_norm": 2.5126636028289795, + "learning_rate": 4.2995812714122575e-06, + "loss": 1.4988, + "step": 675 + }, + { + "epoch": 0.6944016435541859, + "grad_norm": 2.4630191326141357, + "learning_rate": 4.297677959649791e-06, + "loss": 1.5218, + "step": 676 + }, + { + "epoch": 0.6954288649203904, + "grad_norm": 2.8822433948516846, + "learning_rate": 4.295774647887324e-06, + "loss": 1.5592, + "step": 677 + }, + { + "epoch": 0.6964560862865947, + "grad_norm": 2.5498058795928955, + "learning_rate": 4.293871336124858e-06, + "loss": 1.5713, + "step": 678 + }, + { + "epoch": 0.6974833076527992, + "grad_norm": 2.6661157608032227, + "learning_rate": 4.291968024362391e-06, + "loss": 1.5431, + "step": 679 + }, + { + "epoch": 0.6985105290190036, + "grad_norm": 2.617971897125244, + "learning_rate": 4.290064712599924e-06, + "loss": 1.5792, + "step": 680 + }, + { + "epoch": 0.6985105290190036, + "eval_loss": 1.8702285289764404, + "eval_runtime": 37.0376, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 680 + }, + { + "epoch": 0.699537750385208, + "grad_norm": 2.5892348289489746, + "learning_rate": 4.288161400837458e-06, + "loss": 1.5626, + "step": 681 + }, + { + "epoch": 0.7005649717514124, + "grad_norm": 2.5460269451141357, + "learning_rate": 4.286258089074991e-06, + "loss": 1.4876, + "step": 682 + }, + { + "epoch": 0.7015921931176169, + "grad_norm": 2.601632595062256, + "learning_rate": 4.284354777312524e-06, + "loss": 1.5547, + "step": 683 + }, + { + "epoch": 0.7026194144838213, + "grad_norm": 2.6032373905181885, + "learning_rate": 4.282451465550058e-06, + "loss": 1.5133, + "step": 684 + }, + { + "epoch": 0.7036466358500256, + "grad_norm": 2.6283886432647705, + "learning_rate": 4.280548153787591e-06, + "loss": 1.5375, + "step": 685 + }, + { + "epoch": 0.7046738572162301, + "grad_norm": 2.5892159938812256, + "learning_rate": 4.2786448420251245e-06, + "loss": 1.5338, + "step": 686 + }, + { + "epoch": 0.7057010785824345, + "grad_norm": 2.518629789352417, + "learning_rate": 4.276741530262658e-06, + "loss": 1.5003, + "step": 687 + }, + { + "epoch": 0.706728299948639, + "grad_norm": 2.8705344200134277, + "learning_rate": 4.27483821850019e-06, + "loss": 1.5178, + "step": 688 + }, + { + "epoch": 0.7077555213148433, + "grad_norm": 2.7153639793395996, + "learning_rate": 4.272934906737724e-06, + "loss": 1.5914, + "step": 689 + }, + { + "epoch": 0.7087827426810478, + "grad_norm": 2.4585697650909424, + "learning_rate": 4.271031594975257e-06, + "loss": 1.5726, + "step": 690 + }, + { + "epoch": 0.7087827426810478, + "eval_loss": 1.8637797832489014, + "eval_runtime": 37.0465, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 690 + }, + { + "epoch": 0.7098099640472522, + "grad_norm": 2.6370503902435303, + "learning_rate": 4.2691282832127905e-06, + "loss": 1.5706, + "step": 691 + }, + { + "epoch": 0.7108371854134566, + "grad_norm": 2.70202898979187, + "learning_rate": 4.267224971450324e-06, + "loss": 1.5298, + "step": 692 + }, + { + "epoch": 0.711864406779661, + "grad_norm": 2.6149585247039795, + "learning_rate": 4.265321659687857e-06, + "loss": 1.477, + "step": 693 + }, + { + "epoch": 0.7128916281458655, + "grad_norm": 2.9561493396759033, + "learning_rate": 4.2634183479253906e-06, + "loss": 1.5189, + "step": 694 + }, + { + "epoch": 0.7139188495120699, + "grad_norm": 2.697680711746216, + "learning_rate": 4.261515036162924e-06, + "loss": 1.5001, + "step": 695 + }, + { + "epoch": 0.7149460708782742, + "grad_norm": 2.55727219581604, + "learning_rate": 4.259611724400457e-06, + "loss": 1.533, + "step": 696 + }, + { + "epoch": 0.7159732922444787, + "grad_norm": 2.854290723800659, + "learning_rate": 4.257708412637991e-06, + "loss": 1.5057, + "step": 697 + }, + { + "epoch": 0.7170005136106831, + "grad_norm": 2.73824405670166, + "learning_rate": 4.255805100875523e-06, + "loss": 1.4567, + "step": 698 + }, + { + "epoch": 0.7180277349768875, + "grad_norm": 2.730050802230835, + "learning_rate": 4.2539017891130565e-06, + "loss": 1.4873, + "step": 699 + }, + { + "epoch": 0.7190549563430919, + "grad_norm": 2.966660737991333, + "learning_rate": 4.25199847735059e-06, + "loss": 1.574, + "step": 700 + }, + { + "epoch": 0.7190549563430919, + "eval_loss": 1.8623589277267456, + "eval_runtime": 37.0761, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 700 + }, + { + "epoch": 0.7200821777092964, + "grad_norm": 2.7526276111602783, + "learning_rate": 4.250095165588123e-06, + "loss": 1.4831, + "step": 701 + }, + { + "epoch": 0.7211093990755008, + "grad_norm": 2.9722564220428467, + "learning_rate": 4.2481918538256575e-06, + "loss": 1.548, + "step": 702 + }, + { + "epoch": 0.7221366204417052, + "grad_norm": 2.679373025894165, + "learning_rate": 4.246288542063191e-06, + "loss": 1.5145, + "step": 703 + }, + { + "epoch": 0.7231638418079096, + "grad_norm": 2.733139753341675, + "learning_rate": 4.244385230300724e-06, + "loss": 1.5542, + "step": 704 + }, + { + "epoch": 0.724191063174114, + "grad_norm": 2.6549174785614014, + "learning_rate": 4.242481918538257e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.7252182845403184, + "grad_norm": 2.562958002090454, + "learning_rate": 4.24057860677579e-06, + "loss": 1.524, + "step": 706 + }, + { + "epoch": 0.7262455059065228, + "grad_norm": 2.641123056411743, + "learning_rate": 4.2386752950133234e-06, + "loss": 1.5723, + "step": 707 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 2.8128373622894287, + "learning_rate": 4.236771983250857e-06, + "loss": 1.5151, + "step": 708 + }, + { + "epoch": 0.7282999486389317, + "grad_norm": 2.676765203475952, + "learning_rate": 4.23486867148839e-06, + "loss": 1.5024, + "step": 709 + }, + { + "epoch": 0.7293271700051361, + "grad_norm": 2.917178153991699, + "learning_rate": 4.2329653597259235e-06, + "loss": 1.5326, + "step": 710 + }, + { + "epoch": 0.7293271700051361, + "eval_loss": 1.8676133155822754, + "eval_runtime": 36.8304, + "eval_samples_per_second": 27.152, + "eval_steps_per_second": 1.14, + "step": 710 + }, + { + "epoch": 0.7303543913713405, + "grad_norm": 2.8313796520233154, + "learning_rate": 4.231062047963457e-06, + "loss": 1.5623, + "step": 711 + }, + { + "epoch": 0.731381612737545, + "grad_norm": 2.61542010307312, + "learning_rate": 4.22915873620099e-06, + "loss": 1.5725, + "step": 712 + }, + { + "epoch": 0.7324088341037494, + "grad_norm": 2.5788609981536865, + "learning_rate": 4.227255424438524e-06, + "loss": 1.5381, + "step": 713 + }, + { + "epoch": 0.7334360554699538, + "grad_norm": 2.720306158065796, + "learning_rate": 4.225352112676057e-06, + "loss": 1.572, + "step": 714 + }, + { + "epoch": 0.7344632768361582, + "grad_norm": 2.780654191970825, + "learning_rate": 4.2234488009135895e-06, + "loss": 1.5084, + "step": 715 + }, + { + "epoch": 0.7354904982023626, + "grad_norm": 2.4448273181915283, + "learning_rate": 4.221545489151123e-06, + "loss": 1.4614, + "step": 716 + }, + { + "epoch": 0.736517719568567, + "grad_norm": 2.6579184532165527, + "learning_rate": 4.219642177388656e-06, + "loss": 1.5022, + "step": 717 + }, + { + "epoch": 0.7375449409347714, + "grad_norm": 2.7092721462249756, + "learning_rate": 4.2177388656261896e-06, + "loss": 1.5307, + "step": 718 + }, + { + "epoch": 0.7385721623009759, + "grad_norm": 2.65014910697937, + "learning_rate": 4.215835553863723e-06, + "loss": 1.4936, + "step": 719 + }, + { + "epoch": 0.7395993836671803, + "grad_norm": 2.7403628826141357, + "learning_rate": 4.213932242101256e-06, + "loss": 1.4458, + "step": 720 + }, + { + "epoch": 0.7395993836671803, + "eval_loss": 1.8778676986694336, + "eval_runtime": 36.9378, + "eval_samples_per_second": 27.073, + "eval_steps_per_second": 1.137, + "step": 720 + }, + { + "epoch": 0.7406266050333847, + "grad_norm": 2.3937759399414062, + "learning_rate": 4.21202893033879e-06, + "loss": 1.5086, + "step": 721 + }, + { + "epoch": 0.7416538263995891, + "grad_norm": 2.5786867141723633, + "learning_rate": 4.210125618576323e-06, + "loss": 1.5321, + "step": 722 + }, + { + "epoch": 0.7426810477657936, + "grad_norm": 2.607236862182617, + "learning_rate": 4.208222306813856e-06, + "loss": 1.5244, + "step": 723 + }, + { + "epoch": 0.743708269131998, + "grad_norm": 2.5461602210998535, + "learning_rate": 4.20631899505139e-06, + "loss": 1.5841, + "step": 724 + }, + { + "epoch": 0.7447354904982023, + "grad_norm": 2.5051612854003906, + "learning_rate": 4.204415683288923e-06, + "loss": 1.43, + "step": 725 + }, + { + "epoch": 0.7457627118644068, + "grad_norm": 2.692816734313965, + "learning_rate": 4.2025123715264565e-06, + "loss": 1.5439, + "step": 726 + }, + { + "epoch": 0.7467899332306112, + "grad_norm": 2.4827022552490234, + "learning_rate": 4.20060905976399e-06, + "loss": 1.4987, + "step": 727 + }, + { + "epoch": 0.7478171545968156, + "grad_norm": 2.7261135578155518, + "learning_rate": 4.198705748001523e-06, + "loss": 1.4573, + "step": 728 + }, + { + "epoch": 0.74884437596302, + "grad_norm": 2.5710837841033936, + "learning_rate": 4.1968024362390566e-06, + "loss": 1.564, + "step": 729 + }, + { + "epoch": 0.7498715973292245, + "grad_norm": 2.61181640625, + "learning_rate": 4.19489912447659e-06, + "loss": 1.5211, + "step": 730 + }, + { + "epoch": 0.7498715973292245, + "eval_loss": 1.8794434070587158, + "eval_runtime": 36.9676, + "eval_samples_per_second": 27.051, + "eval_steps_per_second": 1.136, + "step": 730 + }, + { + "epoch": 0.7508988186954288, + "grad_norm": 2.6946470737457275, + "learning_rate": 4.192995812714123e-06, + "loss": 1.5837, + "step": 731 + }, + { + "epoch": 0.7519260400616333, + "grad_norm": 2.5337390899658203, + "learning_rate": 4.191092500951656e-06, + "loss": 1.5251, + "step": 732 + }, + { + "epoch": 0.7529532614278377, + "grad_norm": 2.541532278060913, + "learning_rate": 4.189189189189189e-06, + "loss": 1.503, + "step": 733 + }, + { + "epoch": 0.7539804827940421, + "grad_norm": 2.639124631881714, + "learning_rate": 4.1872858774267225e-06, + "loss": 1.4963, + "step": 734 + }, + { + "epoch": 0.7550077041602465, + "grad_norm": 2.872861862182617, + "learning_rate": 4.185382565664256e-06, + "loss": 1.5632, + "step": 735 + }, + { + "epoch": 0.7560349255264509, + "grad_norm": 2.588980197906494, + "learning_rate": 4.183479253901789e-06, + "loss": 1.4998, + "step": 736 + }, + { + "epoch": 0.7570621468926554, + "grad_norm": 2.9199588298797607, + "learning_rate": 4.181575942139323e-06, + "loss": 1.466, + "step": 737 + }, + { + "epoch": 0.7580893682588598, + "grad_norm": 2.5677788257598877, + "learning_rate": 4.179672630376856e-06, + "loss": 1.5247, + "step": 738 + }, + { + "epoch": 0.7591165896250642, + "grad_norm": 2.481142520904541, + "learning_rate": 4.177769318614389e-06, + "loss": 1.4915, + "step": 739 + }, + { + "epoch": 0.7601438109912686, + "grad_norm": 2.7072012424468994, + "learning_rate": 4.175866006851923e-06, + "loss": 1.5783, + "step": 740 + }, + { + "epoch": 0.7601438109912686, + "eval_loss": 1.8817057609558105, + "eval_runtime": 36.9635, + "eval_samples_per_second": 27.054, + "eval_steps_per_second": 1.136, + "step": 740 + }, + { + "epoch": 0.7611710323574731, + "grad_norm": 2.5433032512664795, + "learning_rate": 4.173962695089456e-06, + "loss": 1.5457, + "step": 741 + }, + { + "epoch": 0.7621982537236774, + "grad_norm": 2.702077627182007, + "learning_rate": 4.1720593833269894e-06, + "loss": 1.5285, + "step": 742 + }, + { + "epoch": 0.7632254750898819, + "grad_norm": 2.6678309440612793, + "learning_rate": 4.170156071564523e-06, + "loss": 1.5381, + "step": 743 + }, + { + "epoch": 0.7642526964560863, + "grad_norm": 2.6097185611724854, + "learning_rate": 4.168252759802056e-06, + "loss": 1.6076, + "step": 744 + }, + { + "epoch": 0.7652799178222907, + "grad_norm": 2.5975704193115234, + "learning_rate": 4.1663494480395895e-06, + "loss": 1.515, + "step": 745 + }, + { + "epoch": 0.7663071391884951, + "grad_norm": 2.6858558654785156, + "learning_rate": 4.164446136277123e-06, + "loss": 1.5499, + "step": 746 + }, + { + "epoch": 0.7673343605546995, + "grad_norm": 2.565023899078369, + "learning_rate": 4.162542824514656e-06, + "loss": 1.5502, + "step": 747 + }, + { + "epoch": 0.768361581920904, + "grad_norm": 2.61715030670166, + "learning_rate": 4.16063951275219e-06, + "loss": 1.5271, + "step": 748 + }, + { + "epoch": 0.7693888032871083, + "grad_norm": 2.564382314682007, + "learning_rate": 4.158736200989722e-06, + "loss": 1.4756, + "step": 749 + }, + { + "epoch": 0.7704160246533128, + "grad_norm": 2.64457368850708, + "learning_rate": 4.1568328892272555e-06, + "loss": 1.5246, + "step": 750 + }, + { + "epoch": 0.7704160246533128, + "eval_loss": 1.8777766227722168, + "eval_runtime": 36.9737, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 750 + }, + { + "epoch": 0.7714432460195172, + "grad_norm": 2.773602247238159, + "learning_rate": 4.154929577464789e-06, + "loss": 1.5337, + "step": 751 + }, + { + "epoch": 0.7724704673857217, + "grad_norm": 2.5466997623443604, + "learning_rate": 4.153026265702322e-06, + "loss": 1.5269, + "step": 752 + }, + { + "epoch": 0.773497688751926, + "grad_norm": 2.702888250350952, + "learning_rate": 4.1511229539398556e-06, + "loss": 1.4636, + "step": 753 + }, + { + "epoch": 0.7745249101181304, + "grad_norm": 2.629251718521118, + "learning_rate": 4.149219642177389e-06, + "loss": 1.513, + "step": 754 + }, + { + "epoch": 0.7755521314843349, + "grad_norm": 2.52544903755188, + "learning_rate": 4.147316330414922e-06, + "loss": 1.5225, + "step": 755 + }, + { + "epoch": 0.7765793528505393, + "grad_norm": 2.4267778396606445, + "learning_rate": 4.145413018652456e-06, + "loss": 1.4377, + "step": 756 + }, + { + "epoch": 0.7776065742167437, + "grad_norm": 2.6061577796936035, + "learning_rate": 4.143509706889989e-06, + "loss": 1.5402, + "step": 757 + }, + { + "epoch": 0.7786337955829481, + "grad_norm": 2.6571085453033447, + "learning_rate": 4.141606395127522e-06, + "loss": 1.527, + "step": 758 + }, + { + "epoch": 0.7796610169491526, + "grad_norm": 2.48740291595459, + "learning_rate": 4.139703083365056e-06, + "loss": 1.517, + "step": 759 + }, + { + "epoch": 0.7806882383153569, + "grad_norm": 2.5044426918029785, + "learning_rate": 4.137799771602588e-06, + "loss": 1.5579, + "step": 760 + }, + { + "epoch": 0.7806882383153569, + "eval_loss": 1.8778951168060303, + "eval_runtime": 37.0045, + "eval_samples_per_second": 27.024, + "eval_steps_per_second": 1.135, + "step": 760 + }, + { + "epoch": 0.7817154596815614, + "grad_norm": 2.4728074073791504, + "learning_rate": 4.135896459840122e-06, + "loss": 1.5321, + "step": 761 + }, + { + "epoch": 0.7827426810477658, + "grad_norm": 2.6541659832000732, + "learning_rate": 4.133993148077656e-06, + "loss": 1.4866, + "step": 762 + }, + { + "epoch": 0.7837699024139703, + "grad_norm": 2.526716947555542, + "learning_rate": 4.132089836315189e-06, + "loss": 1.5035, + "step": 763 + }, + { + "epoch": 0.7847971237801746, + "grad_norm": 2.683306932449341, + "learning_rate": 4.1301865245527226e-06, + "loss": 1.5221, + "step": 764 + }, + { + "epoch": 0.785824345146379, + "grad_norm": 2.4980056285858154, + "learning_rate": 4.128283212790256e-06, + "loss": 1.4643, + "step": 765 + }, + { + "epoch": 0.7868515665125835, + "grad_norm": 2.3985488414764404, + "learning_rate": 4.1263799010277884e-06, + "loss": 1.4348, + "step": 766 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 2.600689649581909, + "learning_rate": 4.124476589265322e-06, + "loss": 1.5647, + "step": 767 + }, + { + "epoch": 0.7889060092449923, + "grad_norm": 2.5934395790100098, + "learning_rate": 4.122573277502855e-06, + "loss": 1.4887, + "step": 768 + }, + { + "epoch": 0.7899332306111967, + "grad_norm": 2.5209388732910156, + "learning_rate": 4.1206699657403885e-06, + "loss": 1.4893, + "step": 769 + }, + { + "epoch": 0.7909604519774012, + "grad_norm": 2.6656298637390137, + "learning_rate": 4.118766653977922e-06, + "loss": 1.5239, + "step": 770 + }, + { + "epoch": 0.7909604519774012, + "eval_loss": 1.8737632036209106, + "eval_runtime": 37.056, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 770 + }, + { + "epoch": 0.7919876733436055, + "grad_norm": 2.5147693157196045, + "learning_rate": 4.116863342215455e-06, + "loss": 1.5, + "step": 771 + }, + { + "epoch": 0.79301489470981, + "grad_norm": 2.4653773307800293, + "learning_rate": 4.114960030452989e-06, + "loss": 1.4802, + "step": 772 + }, + { + "epoch": 0.7940421160760144, + "grad_norm": 2.643160820007324, + "learning_rate": 4.113056718690522e-06, + "loss": 1.5086, + "step": 773 + }, + { + "epoch": 0.7950693374422187, + "grad_norm": 2.4873125553131104, + "learning_rate": 4.111153406928055e-06, + "loss": 1.4558, + "step": 774 + }, + { + "epoch": 0.7960965588084232, + "grad_norm": 2.541308879852295, + "learning_rate": 4.109250095165589e-06, + "loss": 1.5483, + "step": 775 + }, + { + "epoch": 0.7971237801746276, + "grad_norm": 2.701874256134033, + "learning_rate": 4.107346783403122e-06, + "loss": 1.5551, + "step": 776 + }, + { + "epoch": 0.7981510015408321, + "grad_norm": 2.457371234893799, + "learning_rate": 4.105443471640655e-06, + "loss": 1.4968, + "step": 777 + }, + { + "epoch": 0.7991782229070364, + "grad_norm": 2.5294973850250244, + "learning_rate": 4.103540159878188e-06, + "loss": 1.4756, + "step": 778 + }, + { + "epoch": 0.8002054442732409, + "grad_norm": 2.464932918548584, + "learning_rate": 4.101636848115721e-06, + "loss": 1.5215, + "step": 779 + }, + { + "epoch": 0.8012326656394453, + "grad_norm": 2.5919454097747803, + "learning_rate": 4.099733536353255e-06, + "loss": 1.4807, + "step": 780 + }, + { + "epoch": 0.8012326656394453, + "eval_loss": 1.875250220298767, + "eval_runtime": 37.035, + "eval_samples_per_second": 27.001, + "eval_steps_per_second": 1.134, + "step": 780 + }, + { + "epoch": 0.8022598870056498, + "grad_norm": 2.505749464035034, + "learning_rate": 4.097830224590788e-06, + "loss": 1.5224, + "step": 781 + }, + { + "epoch": 0.8032871083718541, + "grad_norm": 2.747262954711914, + "learning_rate": 4.095926912828322e-06, + "loss": 1.6051, + "step": 782 + }, + { + "epoch": 0.8043143297380585, + "grad_norm": 2.6147146224975586, + "learning_rate": 4.094023601065855e-06, + "loss": 1.565, + "step": 783 + }, + { + "epoch": 0.805341551104263, + "grad_norm": 2.5381031036376953, + "learning_rate": 4.092120289303388e-06, + "loss": 1.511, + "step": 784 + }, + { + "epoch": 0.8063687724704673, + "grad_norm": 2.6476190090179443, + "learning_rate": 4.0902169775409215e-06, + "loss": 1.5147, + "step": 785 + }, + { + "epoch": 0.8073959938366718, + "grad_norm": 2.486919641494751, + "learning_rate": 4.088313665778455e-06, + "loss": 1.4697, + "step": 786 + }, + { + "epoch": 0.8084232152028762, + "grad_norm": 2.7006654739379883, + "learning_rate": 4.086410354015988e-06, + "loss": 1.4965, + "step": 787 + }, + { + "epoch": 0.8094504365690807, + "grad_norm": 2.7587921619415283, + "learning_rate": 4.0845070422535216e-06, + "loss": 1.4781, + "step": 788 + }, + { + "epoch": 0.810477657935285, + "grad_norm": 2.64239501953125, + "learning_rate": 4.082603730491055e-06, + "loss": 1.4945, + "step": 789 + }, + { + "epoch": 0.8115048793014895, + "grad_norm": 2.659919261932373, + "learning_rate": 4.080700418728588e-06, + "loss": 1.5378, + "step": 790 + }, + { + "epoch": 0.8115048793014895, + "eval_loss": 1.881503939628601, + "eval_runtime": 37.2277, + "eval_samples_per_second": 26.862, + "eval_steps_per_second": 1.128, + "step": 790 + }, + { + "epoch": 0.8125321006676939, + "grad_norm": 2.820996046066284, + "learning_rate": 4.078797106966122e-06, + "loss": 1.5134, + "step": 791 + }, + { + "epoch": 0.8135593220338984, + "grad_norm": 2.4667489528656006, + "learning_rate": 4.076893795203655e-06, + "loss": 1.4165, + "step": 792 + }, + { + "epoch": 0.8145865434001027, + "grad_norm": 2.5528056621551514, + "learning_rate": 4.074990483441188e-06, + "loss": 1.481, + "step": 793 + }, + { + "epoch": 0.8156137647663071, + "grad_norm": 2.632922887802124, + "learning_rate": 4.073087171678721e-06, + "loss": 1.5502, + "step": 794 + }, + { + "epoch": 0.8166409861325116, + "grad_norm": 2.662536144256592, + "learning_rate": 4.071183859916254e-06, + "loss": 1.5302, + "step": 795 + }, + { + "epoch": 0.8176682074987159, + "grad_norm": 2.662454605102539, + "learning_rate": 4.069280548153788e-06, + "loss": 1.5131, + "step": 796 + }, + { + "epoch": 0.8186954288649204, + "grad_norm": 2.6387386322021484, + "learning_rate": 4.067377236391321e-06, + "loss": 1.4677, + "step": 797 + }, + { + "epoch": 0.8197226502311248, + "grad_norm": 2.8879177570343018, + "learning_rate": 4.065473924628854e-06, + "loss": 1.5106, + "step": 798 + }, + { + "epoch": 0.8207498715973293, + "grad_norm": 2.680610418319702, + "learning_rate": 4.063570612866388e-06, + "loss": 1.5389, + "step": 799 + }, + { + "epoch": 0.8217770929635336, + "grad_norm": 2.659590244293213, + "learning_rate": 4.061667301103921e-06, + "loss": 1.5254, + "step": 800 + }, + { + "epoch": 0.8217770929635336, + "eval_loss": 1.881851077079773, + "eval_runtime": 37.3385, + "eval_samples_per_second": 26.782, + "eval_steps_per_second": 1.125, + "step": 800 + }, + { + "epoch": 0.8228043143297381, + "grad_norm": 2.711195945739746, + "learning_rate": 4.0597639893414544e-06, + "loss": 1.5026, + "step": 801 + }, + { + "epoch": 0.8238315356959425, + "grad_norm": 2.6387939453125, + "learning_rate": 4.057860677578988e-06, + "loss": 1.5101, + "step": 802 + }, + { + "epoch": 0.8248587570621468, + "grad_norm": 2.507387638092041, + "learning_rate": 4.055957365816521e-06, + "loss": 1.5277, + "step": 803 + }, + { + "epoch": 0.8258859784283513, + "grad_norm": 2.610370397567749, + "learning_rate": 4.0540540540540545e-06, + "loss": 1.4695, + "step": 804 + }, + { + "epoch": 0.8269131997945557, + "grad_norm": 2.8345787525177, + "learning_rate": 4.052150742291588e-06, + "loss": 1.467, + "step": 805 + }, + { + "epoch": 0.8279404211607602, + "grad_norm": 2.5142805576324463, + "learning_rate": 4.050247430529121e-06, + "loss": 1.4637, + "step": 806 + }, + { + "epoch": 0.8289676425269645, + "grad_norm": 2.7722535133361816, + "learning_rate": 4.048344118766655e-06, + "loss": 1.5427, + "step": 807 + }, + { + "epoch": 0.829994863893169, + "grad_norm": 2.816789150238037, + "learning_rate": 4.046440807004188e-06, + "loss": 1.5422, + "step": 808 + }, + { + "epoch": 0.8310220852593734, + "grad_norm": 2.6485376358032227, + "learning_rate": 4.044537495241721e-06, + "loss": 1.5078, + "step": 809 + }, + { + "epoch": 0.8320493066255779, + "grad_norm": 2.4760921001434326, + "learning_rate": 4.042634183479255e-06, + "loss": 1.4827, + "step": 810 + }, + { + "epoch": 0.8320493066255779, + "eval_loss": 1.8709155321121216, + "eval_runtime": 36.9015, + "eval_samples_per_second": 27.099, + "eval_steps_per_second": 1.138, + "step": 810 + }, + { + "epoch": 0.8330765279917822, + "grad_norm": 2.658062219619751, + "learning_rate": 4.040730871716787e-06, + "loss": 1.5434, + "step": 811 + }, + { + "epoch": 0.8341037493579867, + "grad_norm": 2.5443618297576904, + "learning_rate": 4.038827559954321e-06, + "loss": 1.5165, + "step": 812 + }, + { + "epoch": 0.8351309707241911, + "grad_norm": 2.698432207107544, + "learning_rate": 4.036924248191854e-06, + "loss": 1.5553, + "step": 813 + }, + { + "epoch": 0.8361581920903954, + "grad_norm": 2.5598180294036865, + "learning_rate": 4.035020936429387e-06, + "loss": 1.5289, + "step": 814 + }, + { + "epoch": 0.8371854134565999, + "grad_norm": 2.7002663612365723, + "learning_rate": 4.033117624666921e-06, + "loss": 1.509, + "step": 815 + }, + { + "epoch": 0.8382126348228043, + "grad_norm": 2.7882776260375977, + "learning_rate": 4.031214312904454e-06, + "loss": 1.4787, + "step": 816 + }, + { + "epoch": 0.8392398561890088, + "grad_norm": 2.5583086013793945, + "learning_rate": 4.029311001141987e-06, + "loss": 1.4642, + "step": 817 + }, + { + "epoch": 0.8402670775552131, + "grad_norm": 2.584573268890381, + "learning_rate": 4.027407689379521e-06, + "loss": 1.4343, + "step": 818 + }, + { + "epoch": 0.8412942989214176, + "grad_norm": 2.5062575340270996, + "learning_rate": 4.025504377617054e-06, + "loss": 1.4975, + "step": 819 + }, + { + "epoch": 0.842321520287622, + "grad_norm": 2.3546457290649414, + "learning_rate": 4.0236010658545875e-06, + "loss": 1.4795, + "step": 820 + }, + { + "epoch": 0.842321520287622, + "eval_loss": 1.8716342449188232, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 0.8433487416538265, + "grad_norm": 2.513228178024292, + "learning_rate": 4.02169775409212e-06, + "loss": 1.5191, + "step": 821 + }, + { + "epoch": 0.8443759630200308, + "grad_norm": 2.6360220909118652, + "learning_rate": 4.019794442329653e-06, + "loss": 1.5739, + "step": 822 + }, + { + "epoch": 0.8454031843862352, + "grad_norm": 2.5351850986480713, + "learning_rate": 4.0178911305671876e-06, + "loss": 1.4487, + "step": 823 + }, + { + "epoch": 0.8464304057524397, + "grad_norm": 2.6693851947784424, + "learning_rate": 4.015987818804721e-06, + "loss": 1.5319, + "step": 824 + }, + { + "epoch": 0.847457627118644, + "grad_norm": 2.5210046768188477, + "learning_rate": 4.014084507042254e-06, + "loss": 1.4723, + "step": 825 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.5799994468688965, + "learning_rate": 4.012181195279788e-06, + "loss": 1.4409, + "step": 826 + }, + { + "epoch": 0.8495120698510529, + "grad_norm": 2.5206313133239746, + "learning_rate": 4.010277883517321e-06, + "loss": 1.5449, + "step": 827 + }, + { + "epoch": 0.8505392912172574, + "grad_norm": 2.480584144592285, + "learning_rate": 4.0083745717548535e-06, + "loss": 1.5306, + "step": 828 + }, + { + "epoch": 0.8515665125834617, + "grad_norm": 2.5229053497314453, + "learning_rate": 4.006471259992387e-06, + "loss": 1.5001, + "step": 829 + }, + { + "epoch": 0.8525937339496662, + "grad_norm": 2.53357195854187, + "learning_rate": 4.00456794822992e-06, + "loss": 1.5188, + "step": 830 + }, + { + "epoch": 0.8525937339496662, + "eval_loss": 1.8724256753921509, + "eval_runtime": 37.1114, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 830 + }, + { + "epoch": 0.8536209553158706, + "grad_norm": 2.6796159744262695, + "learning_rate": 4.002664636467454e-06, + "loss": 1.5577, + "step": 831 + }, + { + "epoch": 0.8546481766820749, + "grad_norm": 2.4717817306518555, + "learning_rate": 4.000761324704987e-06, + "loss": 1.5124, + "step": 832 + }, + { + "epoch": 0.8556753980482794, + "grad_norm": 2.649057626724243, + "learning_rate": 3.99885801294252e-06, + "loss": 1.4516, + "step": 833 + }, + { + "epoch": 0.8567026194144838, + "grad_norm": 2.592406749725342, + "learning_rate": 3.996954701180054e-06, + "loss": 1.5215, + "step": 834 + }, + { + "epoch": 0.8577298407806883, + "grad_norm": 2.46956205368042, + "learning_rate": 3.995051389417587e-06, + "loss": 1.4324, + "step": 835 + }, + { + "epoch": 0.8587570621468926, + "grad_norm": 2.8484537601470947, + "learning_rate": 3.9931480776551204e-06, + "loss": 1.4574, + "step": 836 + }, + { + "epoch": 0.8597842835130971, + "grad_norm": 2.6475489139556885, + "learning_rate": 3.991244765892654e-06, + "loss": 1.5581, + "step": 837 + }, + { + "epoch": 0.8608115048793015, + "grad_norm": 2.527153491973877, + "learning_rate": 3.989341454130186e-06, + "loss": 1.4713, + "step": 838 + }, + { + "epoch": 0.861838726245506, + "grad_norm": 2.6767020225524902, + "learning_rate": 3.98743814236772e-06, + "loss": 1.4459, + "step": 839 + }, + { + "epoch": 0.8628659476117103, + "grad_norm": 2.6406850814819336, + "learning_rate": 3.985534830605253e-06, + "loss": 1.4964, + "step": 840 + }, + { + "epoch": 0.8628659476117103, + "eval_loss": 1.886762261390686, + "eval_runtime": 37.1378, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 840 + }, + { + "epoch": 0.8638931689779148, + "grad_norm": 2.466372489929199, + "learning_rate": 3.983631518842786e-06, + "loss": 1.5136, + "step": 841 + }, + { + "epoch": 0.8649203903441192, + "grad_norm": 2.830249071121216, + "learning_rate": 3.98172820708032e-06, + "loss": 1.5372, + "step": 842 + }, + { + "epoch": 0.8659476117103235, + "grad_norm": 2.675889015197754, + "learning_rate": 3.979824895317854e-06, + "loss": 1.4881, + "step": 843 + }, + { + "epoch": 0.866974833076528, + "grad_norm": 2.7418212890625, + "learning_rate": 3.977921583555387e-06, + "loss": 1.4541, + "step": 844 + }, + { + "epoch": 0.8680020544427324, + "grad_norm": 2.572862386703491, + "learning_rate": 3.97601827179292e-06, + "loss": 1.5191, + "step": 845 + }, + { + "epoch": 0.8690292758089369, + "grad_norm": 2.6524274349212646, + "learning_rate": 3.974114960030453e-06, + "loss": 1.4647, + "step": 846 + }, + { + "epoch": 0.8700564971751412, + "grad_norm": 2.794415235519409, + "learning_rate": 3.9722116482679866e-06, + "loss": 1.504, + "step": 847 + }, + { + "epoch": 0.8710837185413457, + "grad_norm": 2.5947723388671875, + "learning_rate": 3.97030833650552e-06, + "loss": 1.492, + "step": 848 + }, + { + "epoch": 0.8721109399075501, + "grad_norm": 2.8539984226226807, + "learning_rate": 3.968405024743053e-06, + "loss": 1.5049, + "step": 849 + }, + { + "epoch": 0.8731381612737545, + "grad_norm": 2.522786855697632, + "learning_rate": 3.966501712980587e-06, + "loss": 1.4223, + "step": 850 + }, + { + "epoch": 0.8731381612737545, + "eval_loss": 1.8691684007644653, + "eval_runtime": 37.1287, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 850 + }, + { + "epoch": 0.8741653826399589, + "grad_norm": 2.6483359336853027, + "learning_rate": 3.96459840121812e-06, + "loss": 1.4815, + "step": 851 + }, + { + "epoch": 0.8751926040061633, + "grad_norm": 2.581429958343506, + "learning_rate": 3.962695089455653e-06, + "loss": 1.4837, + "step": 852 + }, + { + "epoch": 0.8762198253723678, + "grad_norm": 2.5967583656311035, + "learning_rate": 3.960791777693187e-06, + "loss": 1.5289, + "step": 853 + }, + { + "epoch": 0.8772470467385721, + "grad_norm": 2.7169225215911865, + "learning_rate": 3.95888846593072e-06, + "loss": 1.5173, + "step": 854 + }, + { + "epoch": 0.8782742681047766, + "grad_norm": 2.644656181335449, + "learning_rate": 3.956985154168253e-06, + "loss": 1.5095, + "step": 855 + }, + { + "epoch": 0.879301489470981, + "grad_norm": 2.6397511959075928, + "learning_rate": 3.955081842405786e-06, + "loss": 1.5422, + "step": 856 + }, + { + "epoch": 0.8803287108371854, + "grad_norm": 2.691540479660034, + "learning_rate": 3.953178530643319e-06, + "loss": 1.5703, + "step": 857 + }, + { + "epoch": 0.8813559322033898, + "grad_norm": 2.50526762008667, + "learning_rate": 3.951275218880853e-06, + "loss": 1.5885, + "step": 858 + }, + { + "epoch": 0.8823831535695943, + "grad_norm": 2.3846609592437744, + "learning_rate": 3.949371907118386e-06, + "loss": 1.4248, + "step": 859 + }, + { + "epoch": 0.8834103749357987, + "grad_norm": 2.7107760906219482, + "learning_rate": 3.9474685953559194e-06, + "loss": 1.5358, + "step": 860 + }, + { + "epoch": 0.8834103749357987, + "eval_loss": 1.876391887664795, + "eval_runtime": 37.1284, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.884437596302003, + "grad_norm": 2.6206252574920654, + "learning_rate": 3.945565283593453e-06, + "loss": 1.4907, + "step": 861 + }, + { + "epoch": 0.8854648176682075, + "grad_norm": 2.6641664505004883, + "learning_rate": 3.943661971830986e-06, + "loss": 1.4888, + "step": 862 + }, + { + "epoch": 0.8864920390344119, + "grad_norm": 2.7342002391815186, + "learning_rate": 3.9417586600685195e-06, + "loss": 1.5421, + "step": 863 + }, + { + "epoch": 0.8875192604006163, + "grad_norm": 2.7120890617370605, + "learning_rate": 3.939855348306053e-06, + "loss": 1.5154, + "step": 864 + }, + { + "epoch": 0.8885464817668207, + "grad_norm": 2.6350717544555664, + "learning_rate": 3.937952036543586e-06, + "loss": 1.5135, + "step": 865 + }, + { + "epoch": 0.8895737031330252, + "grad_norm": 2.570857524871826, + "learning_rate": 3.93604872478112e-06, + "loss": 1.5072, + "step": 866 + }, + { + "epoch": 0.8906009244992296, + "grad_norm": 2.855071544647217, + "learning_rate": 3.934145413018653e-06, + "loss": 1.5171, + "step": 867 + }, + { + "epoch": 0.891628145865434, + "grad_norm": 2.5592195987701416, + "learning_rate": 3.932242101256186e-06, + "loss": 1.4695, + "step": 868 + }, + { + "epoch": 0.8926553672316384, + "grad_norm": 2.6110830307006836, + "learning_rate": 3.93033878949372e-06, + "loss": 1.4802, + "step": 869 + }, + { + "epoch": 0.8936825885978429, + "grad_norm": 2.7121548652648926, + "learning_rate": 3.928435477731253e-06, + "loss": 1.4422, + "step": 870 + }, + { + "epoch": 0.8936825885978429, + "eval_loss": 1.8629941940307617, + "eval_runtime": 37.1091, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.8947098099640473, + "grad_norm": 2.7763099670410156, + "learning_rate": 3.9265321659687864e-06, + "loss": 1.5275, + "step": 871 + }, + { + "epoch": 0.8957370313302516, + "grad_norm": 2.596220016479492, + "learning_rate": 3.924628854206319e-06, + "loss": 1.5374, + "step": 872 + }, + { + "epoch": 0.8967642526964561, + "grad_norm": 2.575990915298462, + "learning_rate": 3.922725542443852e-06, + "loss": 1.5071, + "step": 873 + }, + { + "epoch": 0.8977914740626605, + "grad_norm": 2.7052223682403564, + "learning_rate": 3.920822230681386e-06, + "loss": 1.4656, + "step": 874 + }, + { + "epoch": 0.8988186954288649, + "grad_norm": 2.550267219543457, + "learning_rate": 3.918918918918919e-06, + "loss": 1.4837, + "step": 875 + }, + { + "epoch": 0.8998459167950693, + "grad_norm": 2.6948060989379883, + "learning_rate": 3.917015607156452e-06, + "loss": 1.5156, + "step": 876 + }, + { + "epoch": 0.9008731381612738, + "grad_norm": 2.575596809387207, + "learning_rate": 3.915112295393986e-06, + "loss": 1.4881, + "step": 877 + }, + { + "epoch": 0.9019003595274782, + "grad_norm": 2.9835500717163086, + "learning_rate": 3.913208983631519e-06, + "loss": 1.5729, + "step": 878 + }, + { + "epoch": 0.9029275808936826, + "grad_norm": 2.5961520671844482, + "learning_rate": 3.9113056718690525e-06, + "loss": 1.5469, + "step": 879 + }, + { + "epoch": 0.903954802259887, + "grad_norm": 2.518260955810547, + "learning_rate": 3.909402360106586e-06, + "loss": 1.4704, + "step": 880 + }, + { + "epoch": 0.903954802259887, + "eval_loss": 1.8570330142974854, + "eval_runtime": 37.1293, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 880 + }, + { + "epoch": 0.9049820236260914, + "grad_norm": 2.740879535675049, + "learning_rate": 3.907499048344119e-06, + "loss": 1.5092, + "step": 881 + }, + { + "epoch": 0.9060092449922958, + "grad_norm": 2.5227973461151123, + "learning_rate": 3.905595736581652e-06, + "loss": 1.5252, + "step": 882 + }, + { + "epoch": 0.9070364663585002, + "grad_norm": 2.483125925064087, + "learning_rate": 3.903692424819186e-06, + "loss": 1.5216, + "step": 883 + }, + { + "epoch": 0.9080636877247047, + "grad_norm": 2.4979538917541504, + "learning_rate": 3.901789113056719e-06, + "loss": 1.4734, + "step": 884 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.707669496536255, + "learning_rate": 3.899885801294253e-06, + "loss": 1.5347, + "step": 885 + }, + { + "epoch": 0.9101181304571135, + "grad_norm": 2.6548409461975098, + "learning_rate": 3.897982489531786e-06, + "loss": 1.5079, + "step": 886 + }, + { + "epoch": 0.9111453518233179, + "grad_norm": 2.6678848266601562, + "learning_rate": 3.896079177769319e-06, + "loss": 1.5659, + "step": 887 + }, + { + "epoch": 0.9121725731895224, + "grad_norm": 2.570976734161377, + "learning_rate": 3.894175866006853e-06, + "loss": 1.5281, + "step": 888 + }, + { + "epoch": 0.9131997945557268, + "grad_norm": 2.5939226150512695, + "learning_rate": 3.892272554244385e-06, + "loss": 1.56, + "step": 889 + }, + { + "epoch": 0.9142270159219312, + "grad_norm": 2.4577109813690186, + "learning_rate": 3.890369242481919e-06, + "loss": 1.4702, + "step": 890 + }, + { + "epoch": 0.9142270159219312, + "eval_loss": 1.8534926176071167, + "eval_runtime": 37.3353, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 890 + }, + { + "epoch": 0.9152542372881356, + "grad_norm": 2.5012643337249756, + "learning_rate": 3.888465930719452e-06, + "loss": 1.5411, + "step": 891 + }, + { + "epoch": 0.91628145865434, + "grad_norm": 2.470905065536499, + "learning_rate": 3.886562618956985e-06, + "loss": 1.4766, + "step": 892 + }, + { + "epoch": 0.9173086800205444, + "grad_norm": 2.4610393047332764, + "learning_rate": 3.884659307194519e-06, + "loss": 1.4925, + "step": 893 + }, + { + "epoch": 0.9183359013867488, + "grad_norm": 2.573151111602783, + "learning_rate": 3.882755995432052e-06, + "loss": 1.4389, + "step": 894 + }, + { + "epoch": 0.9193631227529533, + "grad_norm": 2.628934144973755, + "learning_rate": 3.8808526836695854e-06, + "loss": 1.5099, + "step": 895 + }, + { + "epoch": 0.9203903441191577, + "grad_norm": 2.639920949935913, + "learning_rate": 3.878949371907119e-06, + "loss": 1.4724, + "step": 896 + }, + { + "epoch": 0.9214175654853621, + "grad_norm": 2.4775230884552, + "learning_rate": 3.877046060144652e-06, + "loss": 1.4657, + "step": 897 + }, + { + "epoch": 0.9224447868515665, + "grad_norm": 2.5242490768432617, + "learning_rate": 3.8751427483821855e-06, + "loss": 1.459, + "step": 898 + }, + { + "epoch": 0.923472008217771, + "grad_norm": 2.4444870948791504, + "learning_rate": 3.873239436619718e-06, + "loss": 1.494, + "step": 899 + }, + { + "epoch": 0.9244992295839753, + "grad_norm": 2.4326930046081543, + "learning_rate": 3.871336124857251e-06, + "loss": 1.4449, + "step": 900 + }, + { + "epoch": 0.9244992295839753, + "eval_loss": 1.8844366073608398, + "eval_runtime": 37.432, + "eval_samples_per_second": 26.715, + "eval_steps_per_second": 1.122, + "step": 900 + }, + { + "epoch": 0.9255264509501797, + "grad_norm": 2.621652603149414, + "learning_rate": 3.869432813094785e-06, + "loss": 1.5055, + "step": 901 + }, + { + "epoch": 0.9265536723163842, + "grad_norm": 2.5909814834594727, + "learning_rate": 3.867529501332318e-06, + "loss": 1.5176, + "step": 902 + }, + { + "epoch": 0.9275808936825886, + "grad_norm": 2.512760877609253, + "learning_rate": 3.8656261895698515e-06, + "loss": 1.5027, + "step": 903 + }, + { + "epoch": 0.928608115048793, + "grad_norm": 2.6954784393310547, + "learning_rate": 3.863722877807386e-06, + "loss": 1.4912, + "step": 904 + }, + { + "epoch": 0.9296353364149974, + "grad_norm": 2.481905221939087, + "learning_rate": 3.861819566044919e-06, + "loss": 1.529, + "step": 905 + }, + { + "epoch": 0.9306625577812019, + "grad_norm": 2.612758159637451, + "learning_rate": 3.859916254282452e-06, + "loss": 1.5658, + "step": 906 + }, + { + "epoch": 0.9316897791474062, + "grad_norm": 2.523171901702881, + "learning_rate": 3.858012942519985e-06, + "loss": 1.4804, + "step": 907 + }, + { + "epoch": 0.9327170005136107, + "grad_norm": 2.5053422451019287, + "learning_rate": 3.856109630757518e-06, + "loss": 1.5007, + "step": 908 + }, + { + "epoch": 0.9337442218798151, + "grad_norm": 2.6110904216766357, + "learning_rate": 3.854206318995052e-06, + "loss": 1.4758, + "step": 909 + }, + { + "epoch": 0.9347714432460195, + "grad_norm": 2.7004992961883545, + "learning_rate": 3.852303007232585e-06, + "loss": 1.5255, + "step": 910 + }, + { + "epoch": 0.9347714432460195, + "eval_loss": 1.892087697982788, + "eval_runtime": 36.9356, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 910 + }, + { + "epoch": 0.9357986646122239, + "grad_norm": 2.617180109024048, + "learning_rate": 3.850399695470118e-06, + "loss": 1.4563, + "step": 911 + }, + { + "epoch": 0.9368258859784283, + "grad_norm": 2.5294559001922607, + "learning_rate": 3.848496383707652e-06, + "loss": 1.4767, + "step": 912 + }, + { + "epoch": 0.9378531073446328, + "grad_norm": 2.594238042831421, + "learning_rate": 3.846593071945185e-06, + "loss": 1.4617, + "step": 913 + }, + { + "epoch": 0.9388803287108372, + "grad_norm": 2.6076996326446533, + "learning_rate": 3.8446897601827185e-06, + "loss": 1.446, + "step": 914 + }, + { + "epoch": 0.9399075500770416, + "grad_norm": 2.6047134399414062, + "learning_rate": 3.842786448420252e-06, + "loss": 1.5698, + "step": 915 + }, + { + "epoch": 0.940934771443246, + "grad_norm": 2.5670995712280273, + "learning_rate": 3.840883136657785e-06, + "loss": 1.5803, + "step": 916 + }, + { + "epoch": 0.9419619928094505, + "grad_norm": 2.6153712272644043, + "learning_rate": 3.838979824895318e-06, + "loss": 1.5565, + "step": 917 + }, + { + "epoch": 0.9429892141756548, + "grad_norm": 2.6707098484039307, + "learning_rate": 3.837076513132851e-06, + "loss": 1.4417, + "step": 918 + }, + { + "epoch": 0.9440164355418593, + "grad_norm": 2.591336488723755, + "learning_rate": 3.8351732013703844e-06, + "loss": 1.5159, + "step": 919 + }, + { + "epoch": 0.9450436569080637, + "grad_norm": 2.4689993858337402, + "learning_rate": 3.833269889607918e-06, + "loss": 1.4735, + "step": 920 + }, + { + "epoch": 0.9450436569080637, + "eval_loss": 1.875109076499939, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 920 + }, + { + "epoch": 0.9460708782742681, + "grad_norm": 2.739758253097534, + "learning_rate": 3.831366577845451e-06, + "loss": 1.4566, + "step": 921 + }, + { + "epoch": 0.9470980996404725, + "grad_norm": 2.65278697013855, + "learning_rate": 3.8294632660829845e-06, + "loss": 1.4545, + "step": 922 + }, + { + "epoch": 0.9481253210066769, + "grad_norm": 2.678321123123169, + "learning_rate": 3.827559954320518e-06, + "loss": 1.4864, + "step": 923 + }, + { + "epoch": 0.9491525423728814, + "grad_norm": 2.7516839504241943, + "learning_rate": 3.825656642558051e-06, + "loss": 1.5031, + "step": 924 + }, + { + "epoch": 0.9501797637390857, + "grad_norm": 2.6125247478485107, + "learning_rate": 3.823753330795585e-06, + "loss": 1.521, + "step": 925 + }, + { + "epoch": 0.9512069851052902, + "grad_norm": 2.5134871006011963, + "learning_rate": 3.821850019033118e-06, + "loss": 1.5019, + "step": 926 + }, + { + "epoch": 0.9522342064714946, + "grad_norm": 2.5274763107299805, + "learning_rate": 3.819946707270651e-06, + "loss": 1.4358, + "step": 927 + }, + { + "epoch": 0.9532614278376991, + "grad_norm": 2.480628728866577, + "learning_rate": 3.818043395508185e-06, + "loss": 1.483, + "step": 928 + }, + { + "epoch": 0.9542886492039034, + "grad_norm": 2.6450936794281006, + "learning_rate": 3.816140083745718e-06, + "loss": 1.4892, + "step": 929 + }, + { + "epoch": 0.9553158705701078, + "grad_norm": 2.6758296489715576, + "learning_rate": 3.8142367719832514e-06, + "loss": 1.5701, + "step": 930 + }, + { + "epoch": 0.9553158705701078, + "eval_loss": 1.8651176691055298, + "eval_runtime": 37.0553, + "eval_samples_per_second": 26.987, + "eval_steps_per_second": 1.133, + "step": 930 + }, + { + "epoch": 0.9563430919363123, + "grad_norm": 2.6887295246124268, + "learning_rate": 3.8123334602207844e-06, + "loss": 1.4596, + "step": 931 + }, + { + "epoch": 0.9573703133025167, + "grad_norm": 2.577507734298706, + "learning_rate": 3.8104301484583177e-06, + "loss": 1.4664, + "step": 932 + }, + { + "epoch": 0.9583975346687211, + "grad_norm": 2.6431517601013184, + "learning_rate": 3.808526836695851e-06, + "loss": 1.4715, + "step": 933 + }, + { + "epoch": 0.9594247560349255, + "grad_norm": 2.8451743125915527, + "learning_rate": 3.8066235249333845e-06, + "loss": 1.5096, + "step": 934 + }, + { + "epoch": 0.96045197740113, + "grad_norm": 2.5525193214416504, + "learning_rate": 3.804720213170918e-06, + "loss": 1.4975, + "step": 935 + }, + { + "epoch": 0.9614791987673343, + "grad_norm": 2.6353704929351807, + "learning_rate": 3.8028169014084508e-06, + "loss": 1.4979, + "step": 936 + }, + { + "epoch": 0.9625064201335388, + "grad_norm": 2.923781156539917, + "learning_rate": 3.800913589645984e-06, + "loss": 1.5118, + "step": 937 + }, + { + "epoch": 0.9635336414997432, + "grad_norm": 2.7363650798797607, + "learning_rate": 3.7990102778835175e-06, + "loss": 1.5016, + "step": 938 + }, + { + "epoch": 0.9645608628659477, + "grad_norm": 2.7126598358154297, + "learning_rate": 3.797106966121051e-06, + "loss": 1.5022, + "step": 939 + }, + { + "epoch": 0.965588084232152, + "grad_norm": 3.0102460384368896, + "learning_rate": 3.7952036543585842e-06, + "loss": 1.5638, + "step": 940 + }, + { + "epoch": 0.965588084232152, + "eval_loss": 1.8475635051727295, + "eval_runtime": 37.0871, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 940 + }, + { + "epoch": 0.9666153055983564, + "grad_norm": 2.5455925464630127, + "learning_rate": 3.793300342596117e-06, + "loss": 1.5218, + "step": 941 + }, + { + "epoch": 0.9676425269645609, + "grad_norm": 2.5489683151245117, + "learning_rate": 3.7913970308336505e-06, + "loss": 1.5375, + "step": 942 + }, + { + "epoch": 0.9686697483307652, + "grad_norm": 2.672626256942749, + "learning_rate": 3.789493719071184e-06, + "loss": 1.5217, + "step": 943 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.498378038406372, + "learning_rate": 3.7875904073087177e-06, + "loss": 1.475, + "step": 944 + }, + { + "epoch": 0.9707241910631741, + "grad_norm": 2.515190362930298, + "learning_rate": 3.785687095546251e-06, + "loss": 1.4341, + "step": 945 + }, + { + "epoch": 0.9717514124293786, + "grad_norm": 2.7561733722686768, + "learning_rate": 3.7837837837837844e-06, + "loss": 1.4838, + "step": 946 + }, + { + "epoch": 0.9727786337955829, + "grad_norm": 2.544776201248169, + "learning_rate": 3.7818804720213178e-06, + "loss": 1.4539, + "step": 947 + }, + { + "epoch": 0.9738058551617874, + "grad_norm": 2.744417428970337, + "learning_rate": 3.7799771602588507e-06, + "loss": 1.4296, + "step": 948 + }, + { + "epoch": 0.9748330765279918, + "grad_norm": 2.7759811878204346, + "learning_rate": 3.778073848496384e-06, + "loss": 1.4314, + "step": 949 + }, + { + "epoch": 0.9758602978941961, + "grad_norm": 2.5204875469207764, + "learning_rate": 3.7761705367339174e-06, + "loss": 1.495, + "step": 950 + }, + { + "epoch": 0.9758602978941961, + "eval_loss": 1.8491986989974976, + "eval_runtime": 37.1266, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 0.9768875192604006, + "grad_norm": 2.5156631469726562, + "learning_rate": 3.7742672249714508e-06, + "loss": 1.5228, + "step": 951 + }, + { + "epoch": 0.977914740626605, + "grad_norm": 2.469834089279175, + "learning_rate": 3.772363913208984e-06, + "loss": 1.5188, + "step": 952 + }, + { + "epoch": 0.9789419619928095, + "grad_norm": 2.4933393001556396, + "learning_rate": 3.770460601446517e-06, + "loss": 1.4818, + "step": 953 + }, + { + "epoch": 0.9799691833590138, + "grad_norm": 2.5382790565490723, + "learning_rate": 3.7685572896840504e-06, + "loss": 1.5107, + "step": 954 + }, + { + "epoch": 0.9809964047252183, + "grad_norm": 2.6962392330169678, + "learning_rate": 3.766653977921584e-06, + "loss": 1.4534, + "step": 955 + }, + { + "epoch": 0.9820236260914227, + "grad_norm": 2.5411603450775146, + "learning_rate": 3.764750666159117e-06, + "loss": 1.3942, + "step": 956 + }, + { + "epoch": 0.9830508474576272, + "grad_norm": 2.7101845741271973, + "learning_rate": 3.7628473543966505e-06, + "loss": 1.5003, + "step": 957 + }, + { + "epoch": 0.9840780688238315, + "grad_norm": 2.5780858993530273, + "learning_rate": 3.7609440426341835e-06, + "loss": 1.4508, + "step": 958 + }, + { + "epoch": 0.9851052901900359, + "grad_norm": 2.814004421234131, + "learning_rate": 3.759040730871717e-06, + "loss": 1.5435, + "step": 959 + }, + { + "epoch": 0.9861325115562404, + "grad_norm": 2.5701088905334473, + "learning_rate": 3.75713741910925e-06, + "loss": 1.4386, + "step": 960 + }, + { + "epoch": 0.9861325115562404, + "eval_loss": 1.8556313514709473, + "eval_runtime": 37.1375, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 0.9871597329224447, + "grad_norm": 2.407271146774292, + "learning_rate": 3.7552341073467836e-06, + "loss": 1.4338, + "step": 961 + }, + { + "epoch": 0.9881869542886492, + "grad_norm": 2.433256149291992, + "learning_rate": 3.753330795584317e-06, + "loss": 1.47, + "step": 962 + }, + { + "epoch": 0.9892141756548536, + "grad_norm": 2.679339647293091, + "learning_rate": 3.75142748382185e-06, + "loss": 1.5459, + "step": 963 + }, + { + "epoch": 0.9902413970210581, + "grad_norm": 2.325840473175049, + "learning_rate": 3.749524172059384e-06, + "loss": 1.4385, + "step": 964 + }, + { + "epoch": 0.9912686183872624, + "grad_norm": 2.7060739994049072, + "learning_rate": 3.747620860296917e-06, + "loss": 1.4938, + "step": 965 + }, + { + "epoch": 0.9922958397534669, + "grad_norm": 2.589808464050293, + "learning_rate": 3.7457175485344504e-06, + "loss": 1.5258, + "step": 966 + }, + { + "epoch": 0.9933230611196713, + "grad_norm": 2.4876766204833984, + "learning_rate": 3.7438142367719837e-06, + "loss": 1.5361, + "step": 967 + }, + { + "epoch": 0.9943502824858758, + "grad_norm": 2.7370550632476807, + "learning_rate": 3.741910925009517e-06, + "loss": 1.4999, + "step": 968 + }, + { + "epoch": 0.9953775038520801, + "grad_norm": 2.6552398204803467, + "learning_rate": 3.7400076132470505e-06, + "loss": 1.4612, + "step": 969 + }, + { + "epoch": 0.9964047252182845, + "grad_norm": 2.5229146480560303, + "learning_rate": 3.7381043014845834e-06, + "loss": 1.4794, + "step": 970 + }, + { + "epoch": 0.9964047252182845, + "eval_loss": 1.8614000082015991, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 0.997431946584489, + "grad_norm": 2.5604071617126465, + "learning_rate": 3.7362009897221168e-06, + "loss": 1.5078, + "step": 971 + }, + { + "epoch": 0.9984591679506933, + "grad_norm": 2.614192485809326, + "learning_rate": 3.73429767795965e-06, + "loss": 1.4849, + "step": 972 + }, + { + "epoch": 0.9994863893168978, + "grad_norm": 2.5245072841644287, + "learning_rate": 3.7323943661971835e-06, + "loss": 1.4993, + "step": 973 + }, + { + "epoch": 1.0005136106831023, + "grad_norm": 2.5631163120269775, + "learning_rate": 3.730491054434717e-06, + "loss": 1.4136, + "step": 974 + }, + { + "epoch": 1.0015408320493067, + "grad_norm": 2.625396728515625, + "learning_rate": 3.72858774267225e-06, + "loss": 1.3623, + "step": 975 + }, + { + "epoch": 1.002568053415511, + "grad_norm": 2.7067251205444336, + "learning_rate": 3.726684430909783e-06, + "loss": 1.4527, + "step": 976 + }, + { + "epoch": 1.0035952747817154, + "grad_norm": 2.5352942943573, + "learning_rate": 3.7247811191473165e-06, + "loss": 1.474, + "step": 977 + }, + { + "epoch": 1.00462249614792, + "grad_norm": 2.818143367767334, + "learning_rate": 3.72287780738485e-06, + "loss": 1.4852, + "step": 978 + }, + { + "epoch": 1.0056497175141244, + "grad_norm": 2.8269577026367188, + "learning_rate": 3.7209744956223832e-06, + "loss": 1.4667, + "step": 979 + }, + { + "epoch": 1.0066769388803287, + "grad_norm": 2.837454319000244, + "learning_rate": 3.719071183859916e-06, + "loss": 1.4728, + "step": 980 + }, + { + "epoch": 1.0066769388803287, + "eval_loss": 1.8701857328414917, + "eval_runtime": 37.2859, + "eval_samples_per_second": 26.82, + "eval_steps_per_second": 1.126, + "step": 980 + }, + { + "epoch": 1.007704160246533, + "grad_norm": 2.6601309776306152, + "learning_rate": 3.7171678720974495e-06, + "loss": 1.3917, + "step": 981 + }, + { + "epoch": 1.0087313816127375, + "grad_norm": 2.761413097381592, + "learning_rate": 3.715264560334983e-06, + "loss": 1.5049, + "step": 982 + }, + { + "epoch": 1.009758602978942, + "grad_norm": 2.681429147720337, + "learning_rate": 3.7133612485725163e-06, + "loss": 1.4579, + "step": 983 + }, + { + "epoch": 1.0107858243451464, + "grad_norm": 2.693094253540039, + "learning_rate": 3.71145793681005e-06, + "loss": 1.4766, + "step": 984 + }, + { + "epoch": 1.0118130457113508, + "grad_norm": 2.589165210723877, + "learning_rate": 3.7095546250475834e-06, + "loss": 1.4011, + "step": 985 + }, + { + "epoch": 1.0128402670775551, + "grad_norm": 2.738140821456909, + "learning_rate": 3.7076513132851168e-06, + "loss": 1.5109, + "step": 986 + }, + { + "epoch": 1.0138674884437597, + "grad_norm": 2.7863996028900146, + "learning_rate": 3.7057480015226497e-06, + "loss": 1.4344, + "step": 987 + }, + { + "epoch": 1.014894709809964, + "grad_norm": 2.6011335849761963, + "learning_rate": 3.703844689760183e-06, + "loss": 1.3981, + "step": 988 + }, + { + "epoch": 1.0159219311761685, + "grad_norm": 2.5715689659118652, + "learning_rate": 3.7019413779977164e-06, + "loss": 1.4876, + "step": 989 + }, + { + "epoch": 1.0169491525423728, + "grad_norm": 2.6804113388061523, + "learning_rate": 3.70003806623525e-06, + "loss": 1.4576, + "step": 990 + }, + { + "epoch": 1.0169491525423728, + "eval_loss": 1.8802876472473145, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 990 + }, + { + "epoch": 1.0179763739085772, + "grad_norm": 2.737604856491089, + "learning_rate": 3.698134754472783e-06, + "loss": 1.4399, + "step": 991 + }, + { + "epoch": 1.0190035952747818, + "grad_norm": 2.658388137817383, + "learning_rate": 3.696231442710316e-06, + "loss": 1.5035, + "step": 992 + }, + { + "epoch": 1.0200308166409862, + "grad_norm": 2.7263925075531006, + "learning_rate": 3.6943281309478495e-06, + "loss": 1.4469, + "step": 993 + }, + { + "epoch": 1.0210580380071905, + "grad_norm": 2.7885773181915283, + "learning_rate": 3.692424819185383e-06, + "loss": 1.4641, + "step": 994 + }, + { + "epoch": 1.022085259373395, + "grad_norm": 2.8929779529571533, + "learning_rate": 3.690521507422916e-06, + "loss": 1.4727, + "step": 995 + }, + { + "epoch": 1.0231124807395995, + "grad_norm": 2.6605303287506104, + "learning_rate": 3.6886181956604496e-06, + "loss": 1.4153, + "step": 996 + }, + { + "epoch": 1.0241397021058039, + "grad_norm": 2.8678219318389893, + "learning_rate": 3.686714883897983e-06, + "loss": 1.4199, + "step": 997 + }, + { + "epoch": 1.0251669234720082, + "grad_norm": 2.884453058242798, + "learning_rate": 3.684811572135516e-06, + "loss": 1.4909, + "step": 998 + }, + { + "epoch": 1.0261941448382126, + "grad_norm": 2.6681578159332275, + "learning_rate": 3.6829082603730492e-06, + "loss": 1.4903, + "step": 999 + }, + { + "epoch": 1.027221366204417, + "grad_norm": 2.8013756275177, + "learning_rate": 3.6810049486105826e-06, + "loss": 1.449, + "step": 1000 + }, + { + "epoch": 1.027221366204417, + "eval_loss": 1.8735578060150146, + "eval_runtime": 37.4416, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1000 + }, + { + "epoch": 1.0282485875706215, + "grad_norm": 2.787750244140625, + "learning_rate": 3.679101636848116e-06, + "loss": 1.4808, + "step": 1001 + }, + { + "epoch": 1.029275808936826, + "grad_norm": 2.666874647140503, + "learning_rate": 3.6771983250856493e-06, + "loss": 1.4739, + "step": 1002 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.720586061477661, + "learning_rate": 3.6752950133231822e-06, + "loss": 1.4439, + "step": 1003 + }, + { + "epoch": 1.0313302516692346, + "grad_norm": 2.707771062850952, + "learning_rate": 3.6733917015607156e-06, + "loss": 1.4765, + "step": 1004 + }, + { + "epoch": 1.0323574730354392, + "grad_norm": 2.658092737197876, + "learning_rate": 3.6714883897982494e-06, + "loss": 1.4822, + "step": 1005 + }, + { + "epoch": 1.0333846944016436, + "grad_norm": 2.66788911819458, + "learning_rate": 3.6695850780357828e-06, + "loss": 1.4237, + "step": 1006 + }, + { + "epoch": 1.034411915767848, + "grad_norm": 2.638956308364868, + "learning_rate": 3.667681766273316e-06, + "loss": 1.4168, + "step": 1007 + }, + { + "epoch": 1.0354391371340523, + "grad_norm": 2.739384412765503, + "learning_rate": 3.6657784545108495e-06, + "loss": 1.4338, + "step": 1008 + }, + { + "epoch": 1.0364663585002567, + "grad_norm": 2.599773406982422, + "learning_rate": 3.6638751427483824e-06, + "loss": 1.5188, + "step": 1009 + }, + { + "epoch": 1.0374935798664613, + "grad_norm": 2.705876588821411, + "learning_rate": 3.6619718309859158e-06, + "loss": 1.468, + "step": 1010 + }, + { + "epoch": 1.0374935798664613, + "eval_loss": 1.8751709461212158, + "eval_runtime": 36.9681, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 1010 + }, + { + "epoch": 1.0385208012326657, + "grad_norm": 2.712103843688965, + "learning_rate": 3.660068519223449e-06, + "loss": 1.508, + "step": 1011 + }, + { + "epoch": 1.03954802259887, + "grad_norm": 2.6621742248535156, + "learning_rate": 3.6581652074609825e-06, + "loss": 1.464, + "step": 1012 + }, + { + "epoch": 1.0405752439650744, + "grad_norm": 2.6141257286071777, + "learning_rate": 3.656261895698516e-06, + "loss": 1.4429, + "step": 1013 + }, + { + "epoch": 1.041602465331279, + "grad_norm": 2.5651750564575195, + "learning_rate": 3.6543585839360492e-06, + "loss": 1.4254, + "step": 1014 + }, + { + "epoch": 1.0426296866974833, + "grad_norm": 2.598686933517456, + "learning_rate": 3.652455272173582e-06, + "loss": 1.436, + "step": 1015 + }, + { + "epoch": 1.0436569080636877, + "grad_norm": 2.559375524520874, + "learning_rate": 3.6505519604111155e-06, + "loss": 1.4359, + "step": 1016 + }, + { + "epoch": 1.044684129429892, + "grad_norm": 2.57920503616333, + "learning_rate": 3.648648648648649e-06, + "loss": 1.4524, + "step": 1017 + }, + { + "epoch": 1.0457113507960965, + "grad_norm": 2.6629042625427246, + "learning_rate": 3.6467453368861823e-06, + "loss": 1.4901, + "step": 1018 + }, + { + "epoch": 1.046738572162301, + "grad_norm": 2.6846625804901123, + "learning_rate": 3.6448420251237156e-06, + "loss": 1.4345, + "step": 1019 + }, + { + "epoch": 1.0477657935285054, + "grad_norm": 2.7288694381713867, + "learning_rate": 3.6429387133612486e-06, + "loss": 1.4632, + "step": 1020 + }, + { + "epoch": 1.0477657935285054, + "eval_loss": 1.878115177154541, + "eval_runtime": 37.0627, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.0487930148947098, + "grad_norm": 2.6689510345458984, + "learning_rate": 3.641035401598782e-06, + "loss": 1.4934, + "step": 1021 + }, + { + "epoch": 1.0498202362609141, + "grad_norm": 2.6762588024139404, + "learning_rate": 3.6391320898363153e-06, + "loss": 1.4284, + "step": 1022 + }, + { + "epoch": 1.0508474576271187, + "grad_norm": 2.7875330448150635, + "learning_rate": 3.6372287780738487e-06, + "loss": 1.4717, + "step": 1023 + }, + { + "epoch": 1.051874678993323, + "grad_norm": 2.9239425659179688, + "learning_rate": 3.635325466311382e-06, + "loss": 1.4504, + "step": 1024 + }, + { + "epoch": 1.0529019003595275, + "grad_norm": 2.700913429260254, + "learning_rate": 3.633422154548916e-06, + "loss": 1.4701, + "step": 1025 + }, + { + "epoch": 1.0539291217257318, + "grad_norm": 2.693148612976074, + "learning_rate": 3.6315188427864487e-06, + "loss": 1.4889, + "step": 1026 + }, + { + "epoch": 1.0549563430919364, + "grad_norm": 2.6215498447418213, + "learning_rate": 3.629615531023982e-06, + "loss": 1.4586, + "step": 1027 + }, + { + "epoch": 1.0559835644581408, + "grad_norm": 2.704425573348999, + "learning_rate": 3.6277122192615155e-06, + "loss": 1.4182, + "step": 1028 + }, + { + "epoch": 1.0570107858243452, + "grad_norm": 2.691352128982544, + "learning_rate": 3.625808907499049e-06, + "loss": 1.468, + "step": 1029 + }, + { + "epoch": 1.0580380071905495, + "grad_norm": 2.769744634628296, + "learning_rate": 3.623905595736582e-06, + "loss": 1.4375, + "step": 1030 + }, + { + "epoch": 1.0580380071905495, + "eval_loss": 1.8686742782592773, + "eval_runtime": 37.1278, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.059065228556754, + "grad_norm": 2.6012840270996094, + "learning_rate": 3.6220022839741156e-06, + "loss": 1.4334, + "step": 1031 + }, + { + "epoch": 1.0600924499229585, + "grad_norm": 2.641416072845459, + "learning_rate": 3.6200989722116485e-06, + "loss": 1.4186, + "step": 1032 + }, + { + "epoch": 1.0611196712891628, + "grad_norm": 2.821176052093506, + "learning_rate": 3.618195660449182e-06, + "loss": 1.4438, + "step": 1033 + }, + { + "epoch": 1.0621468926553672, + "grad_norm": 2.719303607940674, + "learning_rate": 3.6162923486867152e-06, + "loss": 1.4519, + "step": 1034 + }, + { + "epoch": 1.0631741140215716, + "grad_norm": 2.76882004737854, + "learning_rate": 3.6143890369242486e-06, + "loss": 1.3941, + "step": 1035 + }, + { + "epoch": 1.064201335387776, + "grad_norm": 2.601137399673462, + "learning_rate": 3.612485725161782e-06, + "loss": 1.4167, + "step": 1036 + }, + { + "epoch": 1.0652285567539805, + "grad_norm": 2.691140651702881, + "learning_rate": 3.610582413399315e-06, + "loss": 1.4361, + "step": 1037 + }, + { + "epoch": 1.066255778120185, + "grad_norm": 2.5779640674591064, + "learning_rate": 3.6086791016368482e-06, + "loss": 1.4237, + "step": 1038 + }, + { + "epoch": 1.0672829994863893, + "grad_norm": 2.6535801887512207, + "learning_rate": 3.6067757898743816e-06, + "loss": 1.436, + "step": 1039 + }, + { + "epoch": 1.0683102208525936, + "grad_norm": 2.8365395069122314, + "learning_rate": 3.604872478111915e-06, + "loss": 1.5067, + "step": 1040 + }, + { + "epoch": 1.0683102208525936, + "eval_loss": 1.8686083555221558, + "eval_runtime": 37.1495, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 1040 + }, + { + "epoch": 1.0693374422187982, + "grad_norm": 2.704638957977295, + "learning_rate": 3.6029691663494483e-06, + "loss": 1.5105, + "step": 1041 + }, + { + "epoch": 1.0703646635850026, + "grad_norm": 2.6670243740081787, + "learning_rate": 3.6010658545869813e-06, + "loss": 1.451, + "step": 1042 + }, + { + "epoch": 1.071391884951207, + "grad_norm": 2.8808884620666504, + "learning_rate": 3.5991625428245146e-06, + "loss": 1.4795, + "step": 1043 + }, + { + "epoch": 1.0724191063174113, + "grad_norm": 2.800075054168701, + "learning_rate": 3.597259231062048e-06, + "loss": 1.4758, + "step": 1044 + }, + { + "epoch": 1.073446327683616, + "grad_norm": 2.4989686012268066, + "learning_rate": 3.5953559192995818e-06, + "loss": 1.42, + "step": 1045 + }, + { + "epoch": 1.0744735490498203, + "grad_norm": 2.7234370708465576, + "learning_rate": 3.593452607537115e-06, + "loss": 1.4549, + "step": 1046 + }, + { + "epoch": 1.0755007704160247, + "grad_norm": 2.7219443321228027, + "learning_rate": 3.5915492957746485e-06, + "loss": 1.5245, + "step": 1047 + }, + { + "epoch": 1.076527991782229, + "grad_norm": 2.6963443756103516, + "learning_rate": 3.589645984012182e-06, + "loss": 1.4472, + "step": 1048 + }, + { + "epoch": 1.0775552131484334, + "grad_norm": 2.7101387977600098, + "learning_rate": 3.587742672249715e-06, + "loss": 1.4659, + "step": 1049 + }, + { + "epoch": 1.078582434514638, + "grad_norm": 2.7673099040985107, + "learning_rate": 3.585839360487248e-06, + "loss": 1.4284, + "step": 1050 + }, + { + "epoch": 1.078582434514638, + "eval_loss": 1.879286289215088, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.13, + "step": 1050 + }, + { + "epoch": 1.0796096558808423, + "grad_norm": 2.81614351272583, + "learning_rate": 3.5839360487247815e-06, + "loss": 1.4251, + "step": 1051 + }, + { + "epoch": 1.0806368772470467, + "grad_norm": 2.770350933074951, + "learning_rate": 3.582032736962315e-06, + "loss": 1.4372, + "step": 1052 + }, + { + "epoch": 1.081664098613251, + "grad_norm": 2.8099257946014404, + "learning_rate": 3.5801294251998483e-06, + "loss": 1.495, + "step": 1053 + }, + { + "epoch": 1.0826913199794557, + "grad_norm": 2.582378387451172, + "learning_rate": 3.578226113437381e-06, + "loss": 1.4005, + "step": 1054 + }, + { + "epoch": 1.08371854134566, + "grad_norm": 2.6215474605560303, + "learning_rate": 3.5763228016749146e-06, + "loss": 1.4428, + "step": 1055 + }, + { + "epoch": 1.0847457627118644, + "grad_norm": 2.7204151153564453, + "learning_rate": 3.574419489912448e-06, + "loss": 1.5197, + "step": 1056 + }, + { + "epoch": 1.0857729840780688, + "grad_norm": 2.6563117504119873, + "learning_rate": 3.5725161781499813e-06, + "loss": 1.457, + "step": 1057 + }, + { + "epoch": 1.0868002054442731, + "grad_norm": 2.559816837310791, + "learning_rate": 3.5706128663875147e-06, + "loss": 1.4376, + "step": 1058 + }, + { + "epoch": 1.0878274268104777, + "grad_norm": 2.6742372512817383, + "learning_rate": 3.5687095546250476e-06, + "loss": 1.4626, + "step": 1059 + }, + { + "epoch": 1.088854648176682, + "grad_norm": 2.8052852153778076, + "learning_rate": 3.566806242862581e-06, + "loss": 1.4785, + "step": 1060 + }, + { + "epoch": 1.088854648176682, + "eval_loss": 1.8775676488876343, + "eval_runtime": 37.1523, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1060 + }, + { + "epoch": 1.0898818695428865, + "grad_norm": 2.687411308288574, + "learning_rate": 3.5649029311001143e-06, + "loss": 1.4491, + "step": 1061 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.662604808807373, + "learning_rate": 3.5629996193376477e-06, + "loss": 1.4274, + "step": 1062 + }, + { + "epoch": 1.0919363122752954, + "grad_norm": 2.619046211242676, + "learning_rate": 3.561096307575181e-06, + "loss": 1.4449, + "step": 1063 + }, + { + "epoch": 1.0929635336414998, + "grad_norm": 2.672553539276123, + "learning_rate": 3.559192995812714e-06, + "loss": 1.4719, + "step": 1064 + }, + { + "epoch": 1.0939907550077042, + "grad_norm": 2.868760824203491, + "learning_rate": 3.557289684050248e-06, + "loss": 1.4955, + "step": 1065 + }, + { + "epoch": 1.0950179763739085, + "grad_norm": 2.7635276317596436, + "learning_rate": 3.555386372287781e-06, + "loss": 1.4765, + "step": 1066 + }, + { + "epoch": 1.0960451977401129, + "grad_norm": 2.883657455444336, + "learning_rate": 3.5534830605253145e-06, + "loss": 1.4923, + "step": 1067 + }, + { + "epoch": 1.0970724191063175, + "grad_norm": 2.7595503330230713, + "learning_rate": 3.551579748762848e-06, + "loss": 1.4527, + "step": 1068 + }, + { + "epoch": 1.0980996404725218, + "grad_norm": 2.8310179710388184, + "learning_rate": 3.5496764370003812e-06, + "loss": 1.4485, + "step": 1069 + }, + { + "epoch": 1.0991268618387262, + "grad_norm": 2.699432134628296, + "learning_rate": 3.5477731252379146e-06, + "loss": 1.4322, + "step": 1070 + }, + { + "epoch": 1.0991268618387262, + "eval_loss": 1.8706557750701904, + "eval_runtime": 37.1509, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1070 + }, + { + "epoch": 1.1001540832049306, + "grad_norm": 2.6249091625213623, + "learning_rate": 3.5458698134754475e-06, + "loss": 1.4031, + "step": 1071 + }, + { + "epoch": 1.1011813045711352, + "grad_norm": 2.6882944107055664, + "learning_rate": 3.543966501712981e-06, + "loss": 1.4075, + "step": 1072 + }, + { + "epoch": 1.1022085259373395, + "grad_norm": 2.6578996181488037, + "learning_rate": 3.5420631899505142e-06, + "loss": 1.4256, + "step": 1073 + }, + { + "epoch": 1.103235747303544, + "grad_norm": 2.8124637603759766, + "learning_rate": 3.5401598781880476e-06, + "loss": 1.5263, + "step": 1074 + }, + { + "epoch": 1.1042629686697483, + "grad_norm": 2.6022191047668457, + "learning_rate": 3.538256566425581e-06, + "loss": 1.3482, + "step": 1075 + }, + { + "epoch": 1.1052901900359529, + "grad_norm": 2.753009080886841, + "learning_rate": 3.536353254663114e-06, + "loss": 1.483, + "step": 1076 + }, + { + "epoch": 1.1063174114021572, + "grad_norm": 2.575666904449463, + "learning_rate": 3.5344499429006473e-06, + "loss": 1.4325, + "step": 1077 + }, + { + "epoch": 1.1073446327683616, + "grad_norm": 2.594136953353882, + "learning_rate": 3.5325466311381806e-06, + "loss": 1.4187, + "step": 1078 + }, + { + "epoch": 1.108371854134566, + "grad_norm": 2.807074785232544, + "learning_rate": 3.530643319375714e-06, + "loss": 1.4763, + "step": 1079 + }, + { + "epoch": 1.1093990755007703, + "grad_norm": 2.5865330696105957, + "learning_rate": 3.5287400076132474e-06, + "loss": 1.3896, + "step": 1080 + }, + { + "epoch": 1.1093990755007703, + "eval_loss": 1.8865792751312256, + "eval_runtime": 37.1514, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1080 + }, + { + "epoch": 1.110426296866975, + "grad_norm": 2.715419054031372, + "learning_rate": 3.5268366958507803e-06, + "loss": 1.4259, + "step": 1081 + }, + { + "epoch": 1.1114535182331793, + "grad_norm": 2.6837334632873535, + "learning_rate": 3.5249333840883137e-06, + "loss": 1.4581, + "step": 1082 + }, + { + "epoch": 1.1124807395993837, + "grad_norm": 2.722128391265869, + "learning_rate": 3.523030072325847e-06, + "loss": 1.4486, + "step": 1083 + }, + { + "epoch": 1.113507960965588, + "grad_norm": 2.7390329837799072, + "learning_rate": 3.5211267605633804e-06, + "loss": 1.4225, + "step": 1084 + }, + { + "epoch": 1.1145351823317924, + "grad_norm": 2.740271806716919, + "learning_rate": 3.519223448800914e-06, + "loss": 1.4274, + "step": 1085 + }, + { + "epoch": 1.115562403697997, + "grad_norm": 2.8321313858032227, + "learning_rate": 3.5173201370384475e-06, + "loss": 1.4337, + "step": 1086 + }, + { + "epoch": 1.1165896250642013, + "grad_norm": 2.740826368331909, + "learning_rate": 3.515416825275981e-06, + "loss": 1.4331, + "step": 1087 + }, + { + "epoch": 1.1176168464304057, + "grad_norm": 2.7382850646972656, + "learning_rate": 3.513513513513514e-06, + "loss": 1.4495, + "step": 1088 + }, + { + "epoch": 1.11864406779661, + "grad_norm": 2.7677972316741943, + "learning_rate": 3.511610201751047e-06, + "loss": 1.4628, + "step": 1089 + }, + { + "epoch": 1.1196712891628147, + "grad_norm": 2.8850181102752686, + "learning_rate": 3.5097068899885806e-06, + "loss": 1.3739, + "step": 1090 + }, + { + "epoch": 1.1196712891628147, + "eval_loss": 1.8794182538986206, + "eval_runtime": 37.4388, + "eval_samples_per_second": 26.71, + "eval_steps_per_second": 1.122, + "step": 1090 + }, + { + "epoch": 1.120698510529019, + "grad_norm": 2.73730731010437, + "learning_rate": 3.507803578226114e-06, + "loss": 1.4444, + "step": 1091 + }, + { + "epoch": 1.1217257318952234, + "grad_norm": 3.027599811553955, + "learning_rate": 3.5059002664636473e-06, + "loss": 1.5123, + "step": 1092 + }, + { + "epoch": 1.1227529532614278, + "grad_norm": 2.613658905029297, + "learning_rate": 3.5039969547011802e-06, + "loss": 1.3798, + "step": 1093 + }, + { + "epoch": 1.1237801746276324, + "grad_norm": 2.6396987438201904, + "learning_rate": 3.5020936429387136e-06, + "loss": 1.4357, + "step": 1094 + }, + { + "epoch": 1.1248073959938367, + "grad_norm": 2.625540018081665, + "learning_rate": 3.500190331176247e-06, + "loss": 1.4354, + "step": 1095 + }, + { + "epoch": 1.125834617360041, + "grad_norm": 2.8413279056549072, + "learning_rate": 3.4982870194137803e-06, + "loss": 1.4846, + "step": 1096 + }, + { + "epoch": 1.1268618387262455, + "grad_norm": 2.7898473739624023, + "learning_rate": 3.4963837076513137e-06, + "loss": 1.421, + "step": 1097 + }, + { + "epoch": 1.1278890600924498, + "grad_norm": 2.6385879516601562, + "learning_rate": 3.4944803958888466e-06, + "loss": 1.4241, + "step": 1098 + }, + { + "epoch": 1.1289162814586544, + "grad_norm": 2.677098512649536, + "learning_rate": 3.49257708412638e-06, + "loss": 1.3671, + "step": 1099 + }, + { + "epoch": 1.1299435028248588, + "grad_norm": 2.8582913875579834, + "learning_rate": 3.4906737723639133e-06, + "loss": 1.4886, + "step": 1100 + }, + { + "epoch": 1.1299435028248588, + "eval_loss": 1.8696506023406982, + "eval_runtime": 37.462, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 1100 + }, + { + "epoch": 1.1309707241910631, + "grad_norm": 2.7157933712005615, + "learning_rate": 3.4887704606014467e-06, + "loss": 1.4964, + "step": 1101 + }, + { + "epoch": 1.1319979455572675, + "grad_norm": 2.6703319549560547, + "learning_rate": 3.48686714883898e-06, + "loss": 1.4509, + "step": 1102 + }, + { + "epoch": 1.133025166923472, + "grad_norm": 2.6307899951934814, + "learning_rate": 3.484963837076513e-06, + "loss": 1.485, + "step": 1103 + }, + { + "epoch": 1.1340523882896765, + "grad_norm": 2.8333141803741455, + "learning_rate": 3.4830605253140464e-06, + "loss": 1.432, + "step": 1104 + }, + { + "epoch": 1.1350796096558808, + "grad_norm": 2.545804023742676, + "learning_rate": 3.4811572135515797e-06, + "loss": 1.3799, + "step": 1105 + }, + { + "epoch": 1.1361068310220852, + "grad_norm": 2.7070441246032715, + "learning_rate": 3.4792539017891135e-06, + "loss": 1.4193, + "step": 1106 + }, + { + "epoch": 1.1371340523882898, + "grad_norm": 2.697140693664551, + "learning_rate": 3.477350590026647e-06, + "loss": 1.398, + "step": 1107 + }, + { + "epoch": 1.1381612737544942, + "grad_norm": 2.784795045852661, + "learning_rate": 3.4754472782641802e-06, + "loss": 1.4708, + "step": 1108 + }, + { + "epoch": 1.1391884951206985, + "grad_norm": 2.8518831729888916, + "learning_rate": 3.4735439665017136e-06, + "loss": 1.4707, + "step": 1109 + }, + { + "epoch": 1.140215716486903, + "grad_norm": 3.0481607913970947, + "learning_rate": 3.4716406547392465e-06, + "loss": 1.4385, + "step": 1110 + }, + { + "epoch": 1.140215716486903, + "eval_loss": 1.8367345333099365, + "eval_runtime": 36.9539, + "eval_samples_per_second": 27.061, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.1412429378531073, + "grad_norm": 2.803053140640259, + "learning_rate": 3.46973734297678e-06, + "loss": 1.4361, + "step": 1111 + }, + { + "epoch": 1.1422701592193119, + "grad_norm": 2.762638568878174, + "learning_rate": 3.4678340312143133e-06, + "loss": 1.4522, + "step": 1112 + }, + { + "epoch": 1.1432973805855162, + "grad_norm": 2.9458558559417725, + "learning_rate": 3.4659307194518466e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.1443246019517206, + "grad_norm": 2.774721622467041, + "learning_rate": 3.46402740768938e-06, + "loss": 1.4723, + "step": 1114 + }, + { + "epoch": 1.145351823317925, + "grad_norm": 2.6556880474090576, + "learning_rate": 3.462124095926913e-06, + "loss": 1.4073, + "step": 1115 + }, + { + "epoch": 1.1463790446841293, + "grad_norm": 2.7482454776763916, + "learning_rate": 3.4602207841644463e-06, + "loss": 1.4022, + "step": 1116 + }, + { + "epoch": 1.147406266050334, + "grad_norm": 2.627567768096924, + "learning_rate": 3.4583174724019797e-06, + "loss": 1.4428, + "step": 1117 + }, + { + "epoch": 1.1484334874165383, + "grad_norm": 2.750666856765747, + "learning_rate": 3.456414160639513e-06, + "loss": 1.4972, + "step": 1118 + }, + { + "epoch": 1.1494607087827426, + "grad_norm": 2.842184543609619, + "learning_rate": 3.4545108488770464e-06, + "loss": 1.4753, + "step": 1119 + }, + { + "epoch": 1.150487930148947, + "grad_norm": 2.829437732696533, + "learning_rate": 3.4526075371145793e-06, + "loss": 1.4525, + "step": 1120 + }, + { + "epoch": 1.150487930148947, + "eval_loss": 1.8498426675796509, + "eval_runtime": 37.0234, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 1120 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.7674920558929443, + "learning_rate": 3.4507042253521127e-06, + "loss": 1.4516, + "step": 1121 + }, + { + "epoch": 1.152542372881356, + "grad_norm": 2.726806879043579, + "learning_rate": 3.448800913589646e-06, + "loss": 1.5113, + "step": 1122 + }, + { + "epoch": 1.1535695942475603, + "grad_norm": 2.7698283195495605, + "learning_rate": 3.4468976018271794e-06, + "loss": 1.4295, + "step": 1123 + }, + { + "epoch": 1.1545968156137647, + "grad_norm": 2.6224706172943115, + "learning_rate": 3.4449942900647128e-06, + "loss": 1.5179, + "step": 1124 + }, + { + "epoch": 1.1556240369799693, + "grad_norm": 2.7266266345977783, + "learning_rate": 3.443090978302246e-06, + "loss": 1.4784, + "step": 1125 + }, + { + "epoch": 1.1566512583461737, + "grad_norm": 2.631751298904419, + "learning_rate": 3.44118766653978e-06, + "loss": 1.4659, + "step": 1126 + }, + { + "epoch": 1.157678479712378, + "grad_norm": 2.7077064514160156, + "learning_rate": 3.439284354777313e-06, + "loss": 1.4654, + "step": 1127 + }, + { + "epoch": 1.1587057010785824, + "grad_norm": 2.735102415084839, + "learning_rate": 3.4373810430148462e-06, + "loss": 1.4604, + "step": 1128 + }, + { + "epoch": 1.1597329224447868, + "grad_norm": 2.9314727783203125, + "learning_rate": 3.4354777312523796e-06, + "loss": 1.3964, + "step": 1129 + }, + { + "epoch": 1.1607601438109914, + "grad_norm": 2.658930778503418, + "learning_rate": 3.433574419489913e-06, + "loss": 1.4331, + "step": 1130 + }, + { + "epoch": 1.1607601438109914, + "eval_loss": 1.8545702695846558, + "eval_runtime": 37.0493, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1130 + }, + { + "epoch": 1.1617873651771957, + "grad_norm": 2.8146262168884277, + "learning_rate": 3.4316711077274463e-06, + "loss": 1.4711, + "step": 1131 + }, + { + "epoch": 1.1628145865434, + "grad_norm": 2.6918272972106934, + "learning_rate": 3.4297677959649792e-06, + "loss": 1.4327, + "step": 1132 + }, + { + "epoch": 1.1638418079096045, + "grad_norm": 2.680551528930664, + "learning_rate": 3.4278644842025126e-06, + "loss": 1.4886, + "step": 1133 + }, + { + "epoch": 1.1648690292758088, + "grad_norm": 2.617286443710327, + "learning_rate": 3.425961172440046e-06, + "loss": 1.4167, + "step": 1134 + }, + { + "epoch": 1.1658962506420134, + "grad_norm": 2.6772518157958984, + "learning_rate": 3.4240578606775793e-06, + "loss": 1.5274, + "step": 1135 + }, + { + "epoch": 1.1669234720082178, + "grad_norm": 2.640723705291748, + "learning_rate": 3.4221545489151127e-06, + "loss": 1.4398, + "step": 1136 + }, + { + "epoch": 1.1679506933744221, + "grad_norm": 2.7104315757751465, + "learning_rate": 3.4202512371526456e-06, + "loss": 1.4367, + "step": 1137 + }, + { + "epoch": 1.1689779147406265, + "grad_norm": 2.831645965576172, + "learning_rate": 3.418347925390179e-06, + "loss": 1.5329, + "step": 1138 + }, + { + "epoch": 1.170005136106831, + "grad_norm": 2.7174274921417236, + "learning_rate": 3.4164446136277124e-06, + "loss": 1.481, + "step": 1139 + }, + { + "epoch": 1.1710323574730355, + "grad_norm": 2.6496286392211914, + "learning_rate": 3.4145413018652457e-06, + "loss": 1.4489, + "step": 1140 + }, + { + "epoch": 1.1710323574730355, + "eval_loss": 1.8524384498596191, + "eval_runtime": 37.1347, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 1140 + }, + { + "epoch": 1.1720595788392398, + "grad_norm": 2.610058069229126, + "learning_rate": 3.412637990102779e-06, + "loss": 1.4755, + "step": 1141 + }, + { + "epoch": 1.1730868002054442, + "grad_norm": 2.642070770263672, + "learning_rate": 3.4107346783403125e-06, + "loss": 1.4981, + "step": 1142 + }, + { + "epoch": 1.1741140215716488, + "grad_norm": 2.57493257522583, + "learning_rate": 3.4088313665778454e-06, + "loss": 1.4039, + "step": 1143 + }, + { + "epoch": 1.1751412429378532, + "grad_norm": 2.5919628143310547, + "learning_rate": 3.4069280548153788e-06, + "loss": 1.4391, + "step": 1144 + }, + { + "epoch": 1.1761684643040575, + "grad_norm": 2.698862314224243, + "learning_rate": 3.405024743052912e-06, + "loss": 1.4198, + "step": 1145 + }, + { + "epoch": 1.177195685670262, + "grad_norm": 2.6387715339660645, + "learning_rate": 3.403121431290446e-06, + "loss": 1.4373, + "step": 1146 + }, + { + "epoch": 1.1782229070364663, + "grad_norm": 2.764643430709839, + "learning_rate": 3.4012181195279793e-06, + "loss": 1.4348, + "step": 1147 + }, + { + "epoch": 1.1792501284026708, + "grad_norm": 2.721038579940796, + "learning_rate": 3.3993148077655126e-06, + "loss": 1.4232, + "step": 1148 + }, + { + "epoch": 1.1802773497688752, + "grad_norm": 2.8942360877990723, + "learning_rate": 3.3974114960030456e-06, + "loss": 1.4405, + "step": 1149 + }, + { + "epoch": 1.1813045711350796, + "grad_norm": 2.687194585800171, + "learning_rate": 3.395508184240579e-06, + "loss": 1.3854, + "step": 1150 + }, + { + "epoch": 1.1813045711350796, + "eval_loss": 1.8701989650726318, + "eval_runtime": 37.132, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.182331792501284, + "grad_norm": 2.7137982845306396, + "learning_rate": 3.3936048724781123e-06, + "loss": 1.4174, + "step": 1151 + }, + { + "epoch": 1.1833590138674885, + "grad_norm": 2.8916432857513428, + "learning_rate": 3.3917015607156457e-06, + "loss": 1.5043, + "step": 1152 + }, + { + "epoch": 1.184386235233693, + "grad_norm": 2.6625723838806152, + "learning_rate": 3.389798248953179e-06, + "loss": 1.3832, + "step": 1153 + }, + { + "epoch": 1.1854134565998973, + "grad_norm": 2.8155555725097656, + "learning_rate": 3.387894937190712e-06, + "loss": 1.4098, + "step": 1154 + }, + { + "epoch": 1.1864406779661016, + "grad_norm": 2.634751796722412, + "learning_rate": 3.3859916254282453e-06, + "loss": 1.4522, + "step": 1155 + }, + { + "epoch": 1.1874678993323062, + "grad_norm": 2.7085063457489014, + "learning_rate": 3.3840883136657787e-06, + "loss": 1.4539, + "step": 1156 + }, + { + "epoch": 1.1884951206985106, + "grad_norm": 2.65588116645813, + "learning_rate": 3.382185001903312e-06, + "loss": 1.5084, + "step": 1157 + }, + { + "epoch": 1.189522342064715, + "grad_norm": 2.582489252090454, + "learning_rate": 3.3802816901408454e-06, + "loss": 1.4218, + "step": 1158 + }, + { + "epoch": 1.1905495634309193, + "grad_norm": 2.6364378929138184, + "learning_rate": 3.3783783783783788e-06, + "loss": 1.4522, + "step": 1159 + }, + { + "epoch": 1.1915767847971237, + "grad_norm": 2.6329915523529053, + "learning_rate": 3.3764750666159117e-06, + "loss": 1.442, + "step": 1160 + }, + { + "epoch": 1.1915767847971237, + "eval_loss": 1.8593298196792603, + "eval_runtime": 37.1143, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1160 + }, + { + "epoch": 1.1926040061633283, + "grad_norm": 2.610050916671753, + "learning_rate": 3.374571754853445e-06, + "loss": 1.4397, + "step": 1161 + }, + { + "epoch": 1.1936312275295327, + "grad_norm": 2.5503945350646973, + "learning_rate": 3.3726684430909784e-06, + "loss": 1.4211, + "step": 1162 + }, + { + "epoch": 1.194658448895737, + "grad_norm": 2.6648080348968506, + "learning_rate": 3.370765131328512e-06, + "loss": 1.3938, + "step": 1163 + }, + { + "epoch": 1.1956856702619414, + "grad_norm": 2.6705355644226074, + "learning_rate": 3.368861819566045e-06, + "loss": 1.4107, + "step": 1164 + }, + { + "epoch": 1.1967128916281458, + "grad_norm": 2.7582640647888184, + "learning_rate": 3.366958507803578e-06, + "loss": 1.4327, + "step": 1165 + }, + { + "epoch": 1.1977401129943503, + "grad_norm": 2.7768008708953857, + "learning_rate": 3.365055196041112e-06, + "loss": 1.454, + "step": 1166 + }, + { + "epoch": 1.1987673343605547, + "grad_norm": 2.628997325897217, + "learning_rate": 3.3631518842786452e-06, + "loss": 1.3988, + "step": 1167 + }, + { + "epoch": 1.199794555726759, + "grad_norm": 2.637505531311035, + "learning_rate": 3.3612485725161786e-06, + "loss": 1.4766, + "step": 1168 + }, + { + "epoch": 1.2008217770929634, + "grad_norm": 2.6935489177703857, + "learning_rate": 3.359345260753712e-06, + "loss": 1.4063, + "step": 1169 + }, + { + "epoch": 1.201848998459168, + "grad_norm": 2.7827870845794678, + "learning_rate": 3.3574419489912453e-06, + "loss": 1.4882, + "step": 1170 + }, + { + "epoch": 1.201848998459168, + "eval_loss": 1.8575458526611328, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.2028762198253724, + "grad_norm": 2.789053201675415, + "learning_rate": 3.3555386372287783e-06, + "loss": 1.4699, + "step": 1171 + }, + { + "epoch": 1.2039034411915768, + "grad_norm": 3.053184747695923, + "learning_rate": 3.3536353254663116e-06, + "loss": 1.4993, + "step": 1172 + }, + { + "epoch": 1.2049306625577811, + "grad_norm": 2.6963112354278564, + "learning_rate": 3.351732013703845e-06, + "loss": 1.3985, + "step": 1173 + }, + { + "epoch": 1.2059578839239857, + "grad_norm": 2.689101219177246, + "learning_rate": 3.3498287019413784e-06, + "loss": 1.4464, + "step": 1174 + }, + { + "epoch": 1.20698510529019, + "grad_norm": 2.7748899459838867, + "learning_rate": 3.3479253901789117e-06, + "loss": 1.4554, + "step": 1175 + }, + { + "epoch": 1.2080123266563945, + "grad_norm": 2.8673830032348633, + "learning_rate": 3.346022078416445e-06, + "loss": 1.4796, + "step": 1176 + }, + { + "epoch": 1.2090395480225988, + "grad_norm": 2.7719533443450928, + "learning_rate": 3.344118766653978e-06, + "loss": 1.3716, + "step": 1177 + }, + { + "epoch": 1.2100667693888032, + "grad_norm": 2.71771240234375, + "learning_rate": 3.3422154548915114e-06, + "loss": 1.4554, + "step": 1178 + }, + { + "epoch": 1.2110939907550078, + "grad_norm": 2.7095048427581787, + "learning_rate": 3.3403121431290448e-06, + "loss": 1.4233, + "step": 1179 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 2.631638526916504, + "learning_rate": 3.338408831366578e-06, + "loss": 1.4581, + "step": 1180 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.855039119720459, + "eval_runtime": 37.1083, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1180 + }, + { + "epoch": 1.2131484334874165, + "grad_norm": 2.657257318496704, + "learning_rate": 3.3365055196041115e-06, + "loss": 1.3864, + "step": 1181 + }, + { + "epoch": 1.214175654853621, + "grad_norm": 2.894169569015503, + "learning_rate": 3.3346022078416444e-06, + "loss": 1.4673, + "step": 1182 + }, + { + "epoch": 1.2152028762198253, + "grad_norm": 2.852644920349121, + "learning_rate": 3.3326988960791778e-06, + "loss": 1.5026, + "step": 1183 + }, + { + "epoch": 1.2162300975860298, + "grad_norm": 2.720878839492798, + "learning_rate": 3.330795584316711e-06, + "loss": 1.4281, + "step": 1184 + }, + { + "epoch": 1.2172573189522342, + "grad_norm": 2.6779472827911377, + "learning_rate": 3.3288922725542445e-06, + "loss": 1.3559, + "step": 1185 + }, + { + "epoch": 1.2182845403184386, + "grad_norm": 2.836782217025757, + "learning_rate": 3.326988960791778e-06, + "loss": 1.4185, + "step": 1186 + }, + { + "epoch": 1.219311761684643, + "grad_norm": 2.9181265830993652, + "learning_rate": 3.3250856490293117e-06, + "loss": 1.4379, + "step": 1187 + }, + { + "epoch": 1.2203389830508475, + "grad_norm": 2.7315778732299805, + "learning_rate": 3.323182337266845e-06, + "loss": 1.4824, + "step": 1188 + }, + { + "epoch": 1.221366204417052, + "grad_norm": 2.8483448028564453, + "learning_rate": 3.321279025504378e-06, + "loss": 1.4766, + "step": 1189 + }, + { + "epoch": 1.2223934257832563, + "grad_norm": 2.736323118209839, + "learning_rate": 3.3193757137419113e-06, + "loss": 1.4647, + "step": 1190 + }, + { + "epoch": 1.2223934257832563, + "eval_loss": 1.8581864833831787, + "eval_runtime": 37.3401, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.2234206471494606, + "grad_norm": 2.8863027095794678, + "learning_rate": 3.3174724019794447e-06, + "loss": 1.4533, + "step": 1191 + }, + { + "epoch": 1.2244478685156652, + "grad_norm": 2.951228380203247, + "learning_rate": 3.315569090216978e-06, + "loss": 1.495, + "step": 1192 + }, + { + "epoch": 1.2254750898818696, + "grad_norm": 2.7486696243286133, + "learning_rate": 3.3136657784545114e-06, + "loss": 1.4082, + "step": 1193 + }, + { + "epoch": 1.226502311248074, + "grad_norm": 2.9105849266052246, + "learning_rate": 3.3117624666920443e-06, + "loss": 1.4988, + "step": 1194 + }, + { + "epoch": 1.2275295326142783, + "grad_norm": 2.8165998458862305, + "learning_rate": 3.3098591549295777e-06, + "loss": 1.4162, + "step": 1195 + }, + { + "epoch": 1.2285567539804827, + "grad_norm": 2.8999876976013184, + "learning_rate": 3.307955843167111e-06, + "loss": 1.3909, + "step": 1196 + }, + { + "epoch": 1.2295839753466873, + "grad_norm": 2.8325014114379883, + "learning_rate": 3.3060525314046444e-06, + "loss": 1.4625, + "step": 1197 + }, + { + "epoch": 1.2306111967128917, + "grad_norm": 2.6553618907928467, + "learning_rate": 3.304149219642178e-06, + "loss": 1.4453, + "step": 1198 + }, + { + "epoch": 1.231638418079096, + "grad_norm": 2.584757089614868, + "learning_rate": 3.3022459078797107e-06, + "loss": 1.4412, + "step": 1199 + }, + { + "epoch": 1.2326656394453004, + "grad_norm": 2.68215274810791, + "learning_rate": 3.300342596117244e-06, + "loss": 1.4511, + "step": 1200 + }, + { + "epoch": 1.2326656394453004, + "eval_loss": 1.849502682685852, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 1200 + }, + { + "epoch": 1.2336928608115048, + "grad_norm": 2.735494375228882, + "learning_rate": 3.2984392843547775e-06, + "loss": 1.4433, + "step": 1201 + }, + { + "epoch": 1.2347200821777093, + "grad_norm": 2.6691815853118896, + "learning_rate": 3.296535972592311e-06, + "loss": 1.4033, + "step": 1202 + }, + { + "epoch": 1.2357473035439137, + "grad_norm": 2.6378355026245117, + "learning_rate": 3.294632660829844e-06, + "loss": 1.378, + "step": 1203 + }, + { + "epoch": 1.236774524910118, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4227, + "step": 1204 + }, + { + "epoch": 1.2378017462763227, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4204, + "step": 1205 + }, + { + "epoch": 1.238828967642527, + "grad_norm": 2.786475896835327, + "learning_rate": 3.2908260373049105e-06, + "loss": 1.4165, + "step": 1206 + }, + { + "epoch": 1.2398561890087314, + "grad_norm": 2.936825752258301, + "learning_rate": 3.288922725542444e-06, + "loss": 1.4712, + "step": 1207 + }, + { + "epoch": 1.2408834103749358, + "grad_norm": 2.7255940437316895, + "learning_rate": 3.2870194137799776e-06, + "loss": 1.4177, + "step": 1208 + }, + { + "epoch": 1.2419106317411401, + "grad_norm": 2.6526482105255127, + "learning_rate": 3.285116102017511e-06, + "loss": 1.3975, + "step": 1209 + }, + { + "epoch": 1.2429378531073447, + "grad_norm": 2.7539143562316895, + "learning_rate": 3.2832127902550444e-06, + "loss": 1.4717, + "step": 1210 + }, + { + "epoch": 1.2429378531073447, + "eval_loss": 1.8445062637329102, + "eval_runtime": 36.9889, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 1210 + }, + { + "epoch": 1.243965074473549, + "grad_norm": 2.907252073287964, + "learning_rate": 3.2813094784925777e-06, + "loss": 1.4484, + "step": 1211 + }, + { + "epoch": 1.2449922958397535, + "grad_norm": 2.7312281131744385, + "learning_rate": 3.2794061667301107e-06, + "loss": 1.3863, + "step": 1212 + }, + { + "epoch": 1.2460195172059578, + "grad_norm": 2.739184617996216, + "learning_rate": 3.277502854967644e-06, + "loss": 1.4302, + "step": 1213 + }, + { + "epoch": 1.2470467385721622, + "grad_norm": 2.798630475997925, + "learning_rate": 3.2755995432051774e-06, + "loss": 1.4134, + "step": 1214 + }, + { + "epoch": 1.2480739599383668, + "grad_norm": 2.835686683654785, + "learning_rate": 3.2736962314427107e-06, + "loss": 1.4258, + "step": 1215 + }, + { + "epoch": 1.2491011813045712, + "grad_norm": 2.7404251098632812, + "learning_rate": 3.271792919680244e-06, + "loss": 1.4347, + "step": 1216 + }, + { + "epoch": 1.2501284026707755, + "grad_norm": 2.8362114429473877, + "learning_rate": 3.269889607917777e-06, + "loss": 1.4909, + "step": 1217 + }, + { + "epoch": 1.25115562403698, + "grad_norm": 2.842283248901367, + "learning_rate": 3.2679862961553104e-06, + "loss": 1.4065, + "step": 1218 + }, + { + "epoch": 1.2521828454031843, + "grad_norm": 2.717536449432373, + "learning_rate": 3.2660829843928438e-06, + "loss": 1.4446, + "step": 1219 + }, + { + "epoch": 1.2532100667693888, + "grad_norm": 2.687009572982788, + "learning_rate": 3.264179672630377e-06, + "loss": 1.4826, + "step": 1220 + }, + { + "epoch": 1.2532100667693888, + "eval_loss": 1.8576964139938354, + "eval_runtime": 37.0759, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1220 + }, + { + "epoch": 1.2542372881355932, + "grad_norm": 2.7234137058258057, + "learning_rate": 3.2622763608679105e-06, + "loss": 1.4503, + "step": 1221 + }, + { + "epoch": 1.2552645095017976, + "grad_norm": 2.733876943588257, + "learning_rate": 3.2603730491054434e-06, + "loss": 1.4076, + "step": 1222 + }, + { + "epoch": 1.2562917308680022, + "grad_norm": 2.6480257511138916, + "learning_rate": 3.258469737342977e-06, + "loss": 1.422, + "step": 1223 + }, + { + "epoch": 1.2573189522342065, + "grad_norm": 2.7006945610046387, + "learning_rate": 3.25656642558051e-06, + "loss": 1.3975, + "step": 1224 + }, + { + "epoch": 1.258346173600411, + "grad_norm": 2.6176297664642334, + "learning_rate": 3.2546631138180435e-06, + "loss": 1.4138, + "step": 1225 + }, + { + "epoch": 1.2593733949666153, + "grad_norm": 2.717742443084717, + "learning_rate": 3.252759802055577e-06, + "loss": 1.4092, + "step": 1226 + }, + { + "epoch": 1.2604006163328196, + "grad_norm": 2.693643093109131, + "learning_rate": 3.25085649029311e-06, + "loss": 1.4221, + "step": 1227 + }, + { + "epoch": 1.2614278376990242, + "grad_norm": 2.709026575088501, + "learning_rate": 3.248953178530644e-06, + "loss": 1.4474, + "step": 1228 + }, + { + "epoch": 1.2624550590652286, + "grad_norm": 2.628399610519409, + "learning_rate": 3.247049866768177e-06, + "loss": 1.4075, + "step": 1229 + }, + { + "epoch": 1.263482280431433, + "grad_norm": 2.681164503097534, + "learning_rate": 3.2451465550057103e-06, + "loss": 1.3683, + "step": 1230 + }, + { + "epoch": 1.263482280431433, + "eval_loss": 1.8543845415115356, + "eval_runtime": 37.1602, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1230 + }, + { + "epoch": 1.2645095017976373, + "grad_norm": 2.6538004875183105, + "learning_rate": 3.2432432432432437e-06, + "loss": 1.4214, + "step": 1231 + }, + { + "epoch": 1.2655367231638417, + "grad_norm": 2.714805841445923, + "learning_rate": 3.241339931480777e-06, + "loss": 1.4617, + "step": 1232 + }, + { + "epoch": 1.2665639445300463, + "grad_norm": 2.743016481399536, + "learning_rate": 3.2394366197183104e-06, + "loss": 1.4556, + "step": 1233 + }, + { + "epoch": 1.2675911658962506, + "grad_norm": 2.618269681930542, + "learning_rate": 3.2375333079558434e-06, + "loss": 1.419, + "step": 1234 + }, + { + "epoch": 1.268618387262455, + "grad_norm": 2.672124147415161, + "learning_rate": 3.2356299961933767e-06, + "loss": 1.4477, + "step": 1235 + }, + { + "epoch": 1.2696456086286596, + "grad_norm": 2.831037998199463, + "learning_rate": 3.23372668443091e-06, + "loss": 1.4552, + "step": 1236 + }, + { + "epoch": 1.270672829994864, + "grad_norm": 2.713193893432617, + "learning_rate": 3.2318233726684435e-06, + "loss": 1.4653, + "step": 1237 + }, + { + "epoch": 1.2717000513610683, + "grad_norm": 2.645097017288208, + "learning_rate": 3.229920060905977e-06, + "loss": 1.4087, + "step": 1238 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.741934061050415, + "learning_rate": 3.2280167491435098e-06, + "loss": 1.4155, + "step": 1239 + }, + { + "epoch": 1.273754494093477, + "grad_norm": 2.5912556648254395, + "learning_rate": 3.226113437381043e-06, + "loss": 1.4214, + "step": 1240 + }, + { + "epoch": 1.273754494093477, + "eval_loss": 1.8524503707885742, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1240 + }, + { + "epoch": 1.2747817154596817, + "grad_norm": 2.5774216651916504, + "learning_rate": 3.2242101256185765e-06, + "loss": 1.4309, + "step": 1241 + }, + { + "epoch": 1.275808936825886, + "grad_norm": 2.7143211364746094, + "learning_rate": 3.22230681385611e-06, + "loss": 1.4928, + "step": 1242 + }, + { + "epoch": 1.2768361581920904, + "grad_norm": 2.8101789951324463, + "learning_rate": 3.220403502093643e-06, + "loss": 1.4325, + "step": 1243 + }, + { + "epoch": 1.2778633795582948, + "grad_norm": 2.704641580581665, + "learning_rate": 3.218500190331176e-06, + "loss": 1.4729, + "step": 1244 + }, + { + "epoch": 1.2788906009244991, + "grad_norm": 2.649195671081543, + "learning_rate": 3.2165968785687095e-06, + "loss": 1.4186, + "step": 1245 + }, + { + "epoch": 1.2799178222907037, + "grad_norm": 2.7863168716430664, + "learning_rate": 3.214693566806243e-06, + "loss": 1.4474, + "step": 1246 + }, + { + "epoch": 1.280945043656908, + "grad_norm": 2.693589687347412, + "learning_rate": 3.2127902550437762e-06, + "loss": 1.4293, + "step": 1247 + }, + { + "epoch": 1.2819722650231125, + "grad_norm": 2.7287509441375732, + "learning_rate": 3.21088694328131e-06, + "loss": 1.4305, + "step": 1248 + }, + { + "epoch": 1.2829994863893168, + "grad_norm": 2.5521676540374756, + "learning_rate": 3.2089836315188434e-06, + "loss": 1.4203, + "step": 1249 + }, + { + "epoch": 1.2840267077555212, + "grad_norm": 2.745425224304199, + "learning_rate": 3.2070803197563767e-06, + "loss": 1.4234, + "step": 1250 + }, + { + "epoch": 1.2840267077555212, + "eval_loss": 1.8586679697036743, + "eval_runtime": 37.1173, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1250 + }, + { + "epoch": 1.2850539291217258, + "grad_norm": 2.687504768371582, + "learning_rate": 3.2051770079939097e-06, + "loss": 1.443, + "step": 1251 + }, + { + "epoch": 1.2860811504879301, + "grad_norm": 2.7841391563415527, + "learning_rate": 3.203273696231443e-06, + "loss": 1.4722, + "step": 1252 + }, + { + "epoch": 1.2871083718541345, + "grad_norm": 2.7715537548065186, + "learning_rate": 3.2013703844689764e-06, + "loss": 1.4063, + "step": 1253 + }, + { + "epoch": 1.288135593220339, + "grad_norm": 2.8183677196502686, + "learning_rate": 3.1994670727065098e-06, + "loss": 1.4534, + "step": 1254 + }, + { + "epoch": 1.2891628145865435, + "grad_norm": 2.795215606689453, + "learning_rate": 3.197563760944043e-06, + "loss": 1.4779, + "step": 1255 + }, + { + "epoch": 1.2901900359527478, + "grad_norm": 2.7606441974639893, + "learning_rate": 3.195660449181576e-06, + "loss": 1.4668, + "step": 1256 + }, + { + "epoch": 1.2912172573189522, + "grad_norm": 2.629718065261841, + "learning_rate": 3.1937571374191094e-06, + "loss": 1.4114, + "step": 1257 + }, + { + "epoch": 1.2922444786851566, + "grad_norm": 2.776845932006836, + "learning_rate": 3.191853825656643e-06, + "loss": 1.4348, + "step": 1258 + }, + { + "epoch": 1.2932717000513612, + "grad_norm": 2.672982931137085, + "learning_rate": 3.189950513894176e-06, + "loss": 1.4274, + "step": 1259 + }, + { + "epoch": 1.2942989214175655, + "grad_norm": 2.5834884643554688, + "learning_rate": 3.1880472021317095e-06, + "loss": 1.4282, + "step": 1260 + }, + { + "epoch": 1.2942989214175655, + "eval_loss": 1.8541877269744873, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1260 + }, + { + "epoch": 1.29532614278377, + "grad_norm": 2.752054452896118, + "learning_rate": 3.1861438903692425e-06, + "loss": 1.3797, + "step": 1261 + }, + { + "epoch": 1.2963533641499743, + "grad_norm": 2.723689556121826, + "learning_rate": 3.184240578606776e-06, + "loss": 1.4137, + "step": 1262 + }, + { + "epoch": 1.2973805855161786, + "grad_norm": 2.806439161300659, + "learning_rate": 3.182337266844309e-06, + "loss": 1.4396, + "step": 1263 + }, + { + "epoch": 1.2984078068823832, + "grad_norm": 2.8140552043914795, + "learning_rate": 3.1804339550818426e-06, + "loss": 1.4677, + "step": 1264 + }, + { + "epoch": 1.2994350282485876, + "grad_norm": 2.78153920173645, + "learning_rate": 3.178530643319376e-06, + "loss": 1.433, + "step": 1265 + }, + { + "epoch": 1.300462249614792, + "grad_norm": 2.6926140785217285, + "learning_rate": 3.176627331556909e-06, + "loss": 1.3901, + "step": 1266 + }, + { + "epoch": 1.3014894709809965, + "grad_norm": 2.636629581451416, + "learning_rate": 3.1747240197944422e-06, + "loss": 1.4331, + "step": 1267 + }, + { + "epoch": 1.3025166923472007, + "grad_norm": 2.625173330307007, + "learning_rate": 3.172820708031976e-06, + "loss": 1.3918, + "step": 1268 + }, + { + "epoch": 1.3035439137134053, + "grad_norm": 2.745666980743408, + "learning_rate": 3.1709173962695094e-06, + "loss": 1.4669, + "step": 1269 + }, + { + "epoch": 1.3045711350796096, + "grad_norm": 2.8781471252441406, + "learning_rate": 3.1690140845070427e-06, + "loss": 1.4586, + "step": 1270 + }, + { + "epoch": 1.3045711350796096, + "eval_loss": 1.8759416341781616, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1270 + }, + { + "epoch": 1.305598356445814, + "grad_norm": 2.832078456878662, + "learning_rate": 3.167110772744576e-06, + "loss": 1.4385, + "step": 1271 + }, + { + "epoch": 1.3066255778120186, + "grad_norm": 2.7153384685516357, + "learning_rate": 3.1652074609821095e-06, + "loss": 1.427, + "step": 1272 + }, + { + "epoch": 1.307652799178223, + "grad_norm": 2.6645848751068115, + "learning_rate": 3.1633041492196424e-06, + "loss": 1.4147, + "step": 1273 + }, + { + "epoch": 1.3086800205444273, + "grad_norm": 2.7231268882751465, + "learning_rate": 3.1614008374571758e-06, + "loss": 1.436, + "step": 1274 + }, + { + "epoch": 1.3097072419106317, + "grad_norm": 2.8131442070007324, + "learning_rate": 3.159497525694709e-06, + "loss": 1.4357, + "step": 1275 + }, + { + "epoch": 1.310734463276836, + "grad_norm": 2.6366660594940186, + "learning_rate": 3.1575942139322425e-06, + "loss": 1.4089, + "step": 1276 + }, + { + "epoch": 1.3117616846430407, + "grad_norm": 2.7849857807159424, + "learning_rate": 3.155690902169776e-06, + "loss": 1.4486, + "step": 1277 + }, + { + "epoch": 1.312788906009245, + "grad_norm": 2.771298408508301, + "learning_rate": 3.1537875904073088e-06, + "loss": 1.4085, + "step": 1278 + }, + { + "epoch": 1.3138161273754494, + "grad_norm": 2.7618181705474854, + "learning_rate": 3.151884278644842e-06, + "loss": 1.4272, + "step": 1279 + }, + { + "epoch": 1.3148433487416538, + "grad_norm": 2.6741080284118652, + "learning_rate": 3.1499809668823755e-06, + "loss": 1.4135, + "step": 1280 + }, + { + "epoch": 1.3148433487416538, + "eval_loss": 1.8567897081375122, + "eval_runtime": 37.1429, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1280 + }, + { + "epoch": 1.3158705701078581, + "grad_norm": 2.6671698093414307, + "learning_rate": 3.148077655119909e-06, + "loss": 1.4598, + "step": 1281 + }, + { + "epoch": 1.3168977914740627, + "grad_norm": 2.6641244888305664, + "learning_rate": 3.1461743433574422e-06, + "loss": 1.3998, + "step": 1282 + }, + { + "epoch": 1.317925012840267, + "grad_norm": 2.747417449951172, + "learning_rate": 3.144271031594975e-06, + "loss": 1.4602, + "step": 1283 + }, + { + "epoch": 1.3189522342064715, + "grad_norm": 2.708984136581421, + "learning_rate": 3.1423677198325085e-06, + "loss": 1.4562, + "step": 1284 + }, + { + "epoch": 1.319979455572676, + "grad_norm": 2.5973010063171387, + "learning_rate": 3.140464408070042e-06, + "loss": 1.4121, + "step": 1285 + }, + { + "epoch": 1.3210066769388804, + "grad_norm": 2.6334621906280518, + "learning_rate": 3.1385610963075753e-06, + "loss": 1.4049, + "step": 1286 + }, + { + "epoch": 1.3220338983050848, + "grad_norm": 2.772655963897705, + "learning_rate": 3.1366577845451086e-06, + "loss": 1.3953, + "step": 1287 + }, + { + "epoch": 1.3230611196712891, + "grad_norm": 2.5006916522979736, + "learning_rate": 3.134754472782642e-06, + "loss": 1.3842, + "step": 1288 + }, + { + "epoch": 1.3240883410374935, + "grad_norm": 2.6052703857421875, + "learning_rate": 3.1328511610201758e-06, + "loss": 1.3801, + "step": 1289 + }, + { + "epoch": 1.325115562403698, + "grad_norm": 2.703693151473999, + "learning_rate": 3.1309478492577087e-06, + "loss": 1.3672, + "step": 1290 + }, + { + "epoch": 1.325115562403698, + "eval_loss": 1.8470649719238281, + "eval_runtime": 37.2714, + "eval_samples_per_second": 26.83, + "eval_steps_per_second": 1.127, + "step": 1290 + }, + { + "epoch": 1.3261427837699025, + "grad_norm": 2.6864330768585205, + "learning_rate": 3.129044537495242e-06, + "loss": 1.4252, + "step": 1291 + }, + { + "epoch": 1.3271700051361068, + "grad_norm": 2.7768399715423584, + "learning_rate": 3.1271412257327754e-06, + "loss": 1.4483, + "step": 1292 + }, + { + "epoch": 1.3281972265023112, + "grad_norm": 2.6658599376678467, + "learning_rate": 3.125237913970309e-06, + "loss": 1.4059, + "step": 1293 + }, + { + "epoch": 1.3292244478685156, + "grad_norm": 2.754718065261841, + "learning_rate": 3.123334602207842e-06, + "loss": 1.3787, + "step": 1294 + }, + { + "epoch": 1.3302516692347202, + "grad_norm": 2.7994842529296875, + "learning_rate": 3.121431290445375e-06, + "loss": 1.4623, + "step": 1295 + }, + { + "epoch": 1.3312788906009245, + "grad_norm": 2.805941343307495, + "learning_rate": 3.1195279786829085e-06, + "loss": 1.5186, + "step": 1296 + }, + { + "epoch": 1.332306111967129, + "grad_norm": 2.6829605102539062, + "learning_rate": 3.117624666920442e-06, + "loss": 1.4052, + "step": 1297 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.7221593856811523, + "learning_rate": 3.115721355157975e-06, + "loss": 1.4031, + "step": 1298 + }, + { + "epoch": 1.3343605546995376, + "grad_norm": 2.579638719558716, + "learning_rate": 3.1138180433955085e-06, + "loss": 1.3627, + "step": 1299 + }, + { + "epoch": 1.3353877760657422, + "grad_norm": 2.861189842224121, + "learning_rate": 3.1119147316330415e-06, + "loss": 1.4405, + "step": 1300 + }, + { + "epoch": 1.3353877760657422, + "eval_loss": 1.8476718664169312, + "eval_runtime": 37.4298, + "eval_samples_per_second": 26.717, + "eval_steps_per_second": 1.122, + "step": 1300 + }, + { + "epoch": 1.3364149974319466, + "grad_norm": 2.8968658447265625, + "learning_rate": 3.110011419870575e-06, + "loss": 1.4774, + "step": 1301 + }, + { + "epoch": 1.337442218798151, + "grad_norm": 2.8708558082580566, + "learning_rate": 3.1081081081081082e-06, + "loss": 1.4085, + "step": 1302 + }, + { + "epoch": 1.3384694401643555, + "grad_norm": 2.802922487258911, + "learning_rate": 3.1062047963456416e-06, + "loss": 1.4486, + "step": 1303 + }, + { + "epoch": 1.33949666153056, + "grad_norm": 2.9320907592773438, + "learning_rate": 3.104301484583175e-06, + "loss": 1.4932, + "step": 1304 + }, + { + "epoch": 1.3405238828967643, + "grad_norm": 2.7708749771118164, + "learning_rate": 3.1023981728207083e-06, + "loss": 1.4601, + "step": 1305 + }, + { + "epoch": 1.3415511042629686, + "grad_norm": 2.7464756965637207, + "learning_rate": 3.1004948610582412e-06, + "loss": 1.4245, + "step": 1306 + }, + { + "epoch": 1.342578325629173, + "grad_norm": 2.767235517501831, + "learning_rate": 3.0985915492957746e-06, + "loss": 1.4199, + "step": 1307 + }, + { + "epoch": 1.3436055469953776, + "grad_norm": 2.6810076236724854, + "learning_rate": 3.096688237533308e-06, + "loss": 1.3851, + "step": 1308 + }, + { + "epoch": 1.344632768361582, + "grad_norm": 2.6390531063079834, + "learning_rate": 3.0947849257708418e-06, + "loss": 1.4688, + "step": 1309 + }, + { + "epoch": 1.3456599897277863, + "grad_norm": 2.729029893875122, + "learning_rate": 3.092881614008375e-06, + "loss": 1.4248, + "step": 1310 + }, + { + "epoch": 1.3456599897277863, + "eval_loss": 1.854402780532837, + "eval_runtime": 36.9357, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.3466872110939907, + "grad_norm": 2.8098297119140625, + "learning_rate": 3.0909783022459085e-06, + "loss": 1.4761, + "step": 1311 + }, + { + "epoch": 1.347714432460195, + "grad_norm": 2.8076772689819336, + "learning_rate": 3.0890749904834414e-06, + "loss": 1.4119, + "step": 1312 + }, + { + "epoch": 1.3487416538263997, + "grad_norm": 2.8214125633239746, + "learning_rate": 3.0871716787209748e-06, + "loss": 1.4693, + "step": 1313 + }, + { + "epoch": 1.349768875192604, + "grad_norm": 2.830083131790161, + "learning_rate": 3.085268366958508e-06, + "loss": 1.438, + "step": 1314 + }, + { + "epoch": 1.3507960965588084, + "grad_norm": 2.7671611309051514, + "learning_rate": 3.0833650551960415e-06, + "loss": 1.4417, + "step": 1315 + }, + { + "epoch": 1.3518233179250128, + "grad_norm": 2.6464383602142334, + "learning_rate": 3.081461743433575e-06, + "loss": 1.4096, + "step": 1316 + }, + { + "epoch": 1.3528505392912171, + "grad_norm": 2.651646137237549, + "learning_rate": 3.079558431671108e-06, + "loss": 1.4264, + "step": 1317 + }, + { + "epoch": 1.3538777606574217, + "grad_norm": 2.6222681999206543, + "learning_rate": 3.077655119908641e-06, + "loss": 1.3836, + "step": 1318 + }, + { + "epoch": 1.354904982023626, + "grad_norm": 2.5674147605895996, + "learning_rate": 3.0757518081461745e-06, + "loss": 1.4206, + "step": 1319 + }, + { + "epoch": 1.3559322033898304, + "grad_norm": 2.7767961025238037, + "learning_rate": 3.073848496383708e-06, + "loss": 1.4348, + "step": 1320 + }, + { + "epoch": 1.3559322033898304, + "eval_loss": 1.8607728481292725, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 1320 + }, + { + "epoch": 1.356959424756035, + "grad_norm": 2.7811837196350098, + "learning_rate": 3.0719451846212413e-06, + "loss": 1.4082, + "step": 1321 + }, + { + "epoch": 1.3579866461222394, + "grad_norm": 2.5633227825164795, + "learning_rate": 3.0700418728587746e-06, + "loss": 1.4352, + "step": 1322 + }, + { + "epoch": 1.3590138674884438, + "grad_norm": 2.7205374240875244, + "learning_rate": 3.0681385610963076e-06, + "loss": 1.405, + "step": 1323 + }, + { + "epoch": 1.3600410888546481, + "grad_norm": 2.82409930229187, + "learning_rate": 3.066235249333841e-06, + "loss": 1.448, + "step": 1324 + }, + { + "epoch": 1.3610683102208525, + "grad_norm": 2.7932820320129395, + "learning_rate": 3.0643319375713743e-06, + "loss": 1.4785, + "step": 1325 + }, + { + "epoch": 1.362095531587057, + "grad_norm": 2.68669056892395, + "learning_rate": 3.0624286258089076e-06, + "loss": 1.4232, + "step": 1326 + }, + { + "epoch": 1.3631227529532615, + "grad_norm": 2.6492018699645996, + "learning_rate": 3.060525314046441e-06, + "loss": 1.4052, + "step": 1327 + }, + { + "epoch": 1.3641499743194658, + "grad_norm": 2.7677114009857178, + "learning_rate": 3.058622002283974e-06, + "loss": 1.4426, + "step": 1328 + }, + { + "epoch": 1.3651771956856702, + "grad_norm": 2.7124569416046143, + "learning_rate": 3.0567186905215077e-06, + "loss": 1.4631, + "step": 1329 + }, + { + "epoch": 1.3662044170518746, + "grad_norm": 2.7084693908691406, + "learning_rate": 3.054815378759041e-06, + "loss": 1.4656, + "step": 1330 + }, + { + "epoch": 1.3662044170518746, + "eval_loss": 1.8457012176513672, + "eval_runtime": 37.0141, + "eval_samples_per_second": 27.017, + "eval_steps_per_second": 1.135, + "step": 1330 + }, + { + "epoch": 1.3672316384180792, + "grad_norm": 2.7612688541412354, + "learning_rate": 3.0529120669965745e-06, + "loss": 1.455, + "step": 1331 + }, + { + "epoch": 1.3682588597842835, + "grad_norm": 2.5172924995422363, + "learning_rate": 3.051008755234108e-06, + "loss": 1.3915, + "step": 1332 + }, + { + "epoch": 1.3692860811504879, + "grad_norm": 2.7225120067596436, + "learning_rate": 3.049105443471641e-06, + "loss": 1.4025, + "step": 1333 + }, + { + "epoch": 1.3703133025166925, + "grad_norm": 2.835470676422119, + "learning_rate": 3.0472021317091745e-06, + "loss": 1.5423, + "step": 1334 + }, + { + "epoch": 1.3713405238828968, + "grad_norm": 2.6247284412384033, + "learning_rate": 3.0452988199467075e-06, + "loss": 1.4562, + "step": 1335 + }, + { + "epoch": 1.3723677452491012, + "grad_norm": 2.6346960067749023, + "learning_rate": 3.043395508184241e-06, + "loss": 1.4566, + "step": 1336 + }, + { + "epoch": 1.3733949666153056, + "grad_norm": 2.6278724670410156, + "learning_rate": 3.041492196421774e-06, + "loss": 1.403, + "step": 1337 + }, + { + "epoch": 1.37442218798151, + "grad_norm": 2.8066110610961914, + "learning_rate": 3.0395888846593076e-06, + "loss": 1.4433, + "step": 1338 + }, + { + "epoch": 1.3754494093477145, + "grad_norm": 2.6565256118774414, + "learning_rate": 3.037685572896841e-06, + "loss": 1.4177, + "step": 1339 + }, + { + "epoch": 1.376476630713919, + "grad_norm": 2.6529884338378906, + "learning_rate": 3.035782261134374e-06, + "loss": 1.4378, + "step": 1340 + }, + { + "epoch": 1.376476630713919, + "eval_loss": 1.8618167638778687, + "eval_runtime": 37.025, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1340 + }, + { + "epoch": 1.3775038520801233, + "grad_norm": 2.820838689804077, + "learning_rate": 3.0338789493719072e-06, + "loss": 1.4471, + "step": 1341 + }, + { + "epoch": 1.3785310734463276, + "grad_norm": 2.6786739826202393, + "learning_rate": 3.0319756376094406e-06, + "loss": 1.4393, + "step": 1342 + }, + { + "epoch": 1.379558294812532, + "grad_norm": 2.7962610721588135, + "learning_rate": 3.030072325846974e-06, + "loss": 1.4407, + "step": 1343 + }, + { + "epoch": 1.3805855161787366, + "grad_norm": 2.739959716796875, + "learning_rate": 3.0281690140845073e-06, + "loss": 1.4059, + "step": 1344 + }, + { + "epoch": 1.381612737544941, + "grad_norm": 2.7812039852142334, + "learning_rate": 3.0262657023220403e-06, + "loss": 1.4449, + "step": 1345 + }, + { + "epoch": 1.3826399589111453, + "grad_norm": 2.749671459197998, + "learning_rate": 3.0243623905595736e-06, + "loss": 1.4459, + "step": 1346 + }, + { + "epoch": 1.3836671802773497, + "grad_norm": 2.7494630813598633, + "learning_rate": 3.022459078797107e-06, + "loss": 1.4204, + "step": 1347 + }, + { + "epoch": 1.384694401643554, + "grad_norm": 2.7188141345977783, + "learning_rate": 3.0205557670346404e-06, + "loss": 1.4249, + "step": 1348 + }, + { + "epoch": 1.3857216230097587, + "grad_norm": 2.712442636489868, + "learning_rate": 3.018652455272174e-06, + "loss": 1.3975, + "step": 1349 + }, + { + "epoch": 1.386748844375963, + "grad_norm": 2.9888060092926025, + "learning_rate": 3.0167491435097075e-06, + "loss": 1.5258, + "step": 1350 + }, + { + "epoch": 1.386748844375963, + "eval_loss": 1.8591758012771606, + "eval_runtime": 37.0078, + "eval_samples_per_second": 27.021, + "eval_steps_per_second": 1.135, + "step": 1350 + }, + { + "epoch": 1.3877760657421674, + "grad_norm": 2.789698600769043, + "learning_rate": 3.014845831747241e-06, + "loss": 1.4305, + "step": 1351 + }, + { + "epoch": 1.388803287108372, + "grad_norm": 2.747840166091919, + "learning_rate": 3.012942519984774e-06, + "loss": 1.4031, + "step": 1352 + }, + { + "epoch": 1.3898305084745763, + "grad_norm": 2.7078518867492676, + "learning_rate": 3.011039208222307e-06, + "loss": 1.4139, + "step": 1353 + }, + { + "epoch": 1.3908577298407807, + "grad_norm": 2.615677833557129, + "learning_rate": 3.0091358964598405e-06, + "loss": 1.3728, + "step": 1354 + }, + { + "epoch": 1.391884951206985, + "grad_norm": 2.7485580444335938, + "learning_rate": 3.007232584697374e-06, + "loss": 1.4272, + "step": 1355 + }, + { + "epoch": 1.3929121725731894, + "grad_norm": 2.7168128490448, + "learning_rate": 3.0053292729349073e-06, + "loss": 1.4353, + "step": 1356 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 2.7295777797698975, + "learning_rate": 3.00342596117244e-06, + "loss": 1.4324, + "step": 1357 + }, + { + "epoch": 1.3949666153055984, + "grad_norm": 2.8791208267211914, + "learning_rate": 3.0015226494099736e-06, + "loss": 1.4303, + "step": 1358 + }, + { + "epoch": 1.3959938366718028, + "grad_norm": 2.647028684616089, + "learning_rate": 2.999619337647507e-06, + "loss": 1.402, + "step": 1359 + }, + { + "epoch": 1.3970210580380071, + "grad_norm": 2.869901180267334, + "learning_rate": 2.9977160258850403e-06, + "loss": 1.404, + "step": 1360 + }, + { + "epoch": 1.3970210580380071, + "eval_loss": 1.8588491678237915, + "eval_runtime": 37.0159, + "eval_samples_per_second": 27.015, + "eval_steps_per_second": 1.135, + "step": 1360 + }, + { + "epoch": 1.3980482794042115, + "grad_norm": 2.7664403915405273, + "learning_rate": 2.9958127141225736e-06, + "loss": 1.3895, + "step": 1361 + }, + { + "epoch": 1.399075500770416, + "grad_norm": 2.8101043701171875, + "learning_rate": 2.9939094023601066e-06, + "loss": 1.4629, + "step": 1362 + }, + { + "epoch": 1.4001027221366205, + "grad_norm": 2.7114098072052, + "learning_rate": 2.99200609059764e-06, + "loss": 1.4365, + "step": 1363 + }, + { + "epoch": 1.4011299435028248, + "grad_norm": 2.6739253997802734, + "learning_rate": 2.9901027788351733e-06, + "loss": 1.382, + "step": 1364 + }, + { + "epoch": 1.4021571648690292, + "grad_norm": 2.8102970123291016, + "learning_rate": 2.9881994670727067e-06, + "loss": 1.392, + "step": 1365 + }, + { + "epoch": 1.4031843862352336, + "grad_norm": 2.645071506500244, + "learning_rate": 2.98629615531024e-06, + "loss": 1.4157, + "step": 1366 + }, + { + "epoch": 1.4042116076014381, + "grad_norm": 2.6110310554504395, + "learning_rate": 2.984392843547773e-06, + "loss": 1.3996, + "step": 1367 + }, + { + "epoch": 1.4052388289676425, + "grad_norm": 2.784090042114258, + "learning_rate": 2.9824895317853063e-06, + "loss": 1.4178, + "step": 1368 + }, + { + "epoch": 1.4062660503338469, + "grad_norm": 2.7293829917907715, + "learning_rate": 2.98058622002284e-06, + "loss": 1.4193, + "step": 1369 + }, + { + "epoch": 1.4072932717000515, + "grad_norm": 2.6924493312835693, + "learning_rate": 2.9786829082603735e-06, + "loss": 1.476, + "step": 1370 + }, + { + "epoch": 1.4072932717000515, + "eval_loss": 1.8642805814743042, + "eval_runtime": 37.0452, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1370 + }, + { + "epoch": 1.4083204930662558, + "grad_norm": 2.7714414596557617, + "learning_rate": 2.976779596497907e-06, + "loss": 1.4574, + "step": 1371 + }, + { + "epoch": 1.4093477144324602, + "grad_norm": 2.7013490200042725, + "learning_rate": 2.97487628473544e-06, + "loss": 1.4718, + "step": 1372 + }, + { + "epoch": 1.4103749357986646, + "grad_norm": 2.9120547771453857, + "learning_rate": 2.9729729729729736e-06, + "loss": 1.4298, + "step": 1373 + }, + { + "epoch": 1.411402157164869, + "grad_norm": 2.9127988815307617, + "learning_rate": 2.9710696612105065e-06, + "loss": 1.4703, + "step": 1374 + }, + { + "epoch": 1.4124293785310735, + "grad_norm": 2.696772575378418, + "learning_rate": 2.96916634944804e-06, + "loss": 1.412, + "step": 1375 + }, + { + "epoch": 1.413456599897278, + "grad_norm": 2.8330435752868652, + "learning_rate": 2.9672630376855732e-06, + "loss": 1.4731, + "step": 1376 + }, + { + "epoch": 1.4144838212634823, + "grad_norm": 2.7354564666748047, + "learning_rate": 2.9653597259231066e-06, + "loss": 1.4342, + "step": 1377 + }, + { + "epoch": 1.4155110426296866, + "grad_norm": 2.7386012077331543, + "learning_rate": 2.96345641416064e-06, + "loss": 1.3935, + "step": 1378 + }, + { + "epoch": 1.416538263995891, + "grad_norm": 2.5881052017211914, + "learning_rate": 2.961553102398173e-06, + "loss": 1.3282, + "step": 1379 + }, + { + "epoch": 1.4175654853620956, + "grad_norm": 2.805558443069458, + "learning_rate": 2.9596497906357063e-06, + "loss": 1.4691, + "step": 1380 + }, + { + "epoch": 1.4175654853620956, + "eval_loss": 1.84714937210083, + "eval_runtime": 37.0349, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1380 + }, + { + "epoch": 1.4185927067283, + "grad_norm": 2.8639137744903564, + "learning_rate": 2.9577464788732396e-06, + "loss": 1.4745, + "step": 1381 + }, + { + "epoch": 1.4196199280945043, + "grad_norm": 2.6688785552978516, + "learning_rate": 2.955843167110773e-06, + "loss": 1.4016, + "step": 1382 + }, + { + "epoch": 1.420647149460709, + "grad_norm": 2.6851978302001953, + "learning_rate": 2.9539398553483064e-06, + "loss": 1.4205, + "step": 1383 + }, + { + "epoch": 1.4216743708269133, + "grad_norm": 2.7381017208099365, + "learning_rate": 2.9520365435858393e-06, + "loss": 1.4315, + "step": 1384 + }, + { + "epoch": 1.4227015921931176, + "grad_norm": 2.9249424934387207, + "learning_rate": 2.9501332318233727e-06, + "loss": 1.49, + "step": 1385 + }, + { + "epoch": 1.423728813559322, + "grad_norm": 2.765490770339966, + "learning_rate": 2.948229920060906e-06, + "loss": 1.4359, + "step": 1386 + }, + { + "epoch": 1.4247560349255264, + "grad_norm": 2.578561544418335, + "learning_rate": 2.9463266082984394e-06, + "loss": 1.4032, + "step": 1387 + }, + { + "epoch": 1.425783256291731, + "grad_norm": 2.6673238277435303, + "learning_rate": 2.9444232965359727e-06, + "loss": 1.4144, + "step": 1388 + }, + { + "epoch": 1.4268104776579353, + "grad_norm": 2.737516164779663, + "learning_rate": 2.9425199847735057e-06, + "loss": 1.4375, + "step": 1389 + }, + { + "epoch": 1.4278376990241397, + "grad_norm": 2.656177282333374, + "learning_rate": 2.94061667301104e-06, + "loss": 1.4007, + "step": 1390 + }, + { + "epoch": 1.4278376990241397, + "eval_loss": 1.8427128791809082, + "eval_runtime": 37.1322, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1390 + }, + { + "epoch": 1.428864920390344, + "grad_norm": 2.6011996269226074, + "learning_rate": 2.938713361248573e-06, + "loss": 1.339, + "step": 1391 + }, + { + "epoch": 1.4298921417565484, + "grad_norm": 2.6374619007110596, + "learning_rate": 2.936810049486106e-06, + "loss": 1.4092, + "step": 1392 + }, + { + "epoch": 1.430919363122753, + "grad_norm": 2.6879520416259766, + "learning_rate": 2.9349067377236396e-06, + "loss": 1.4235, + "step": 1393 + }, + { + "epoch": 1.4319465844889574, + "grad_norm": 2.8400580883026123, + "learning_rate": 2.933003425961173e-06, + "loss": 1.381, + "step": 1394 + }, + { + "epoch": 1.4329738058551618, + "grad_norm": 2.742767810821533, + "learning_rate": 2.9311001141987063e-06, + "loss": 1.4055, + "step": 1395 + }, + { + "epoch": 1.4340010272213661, + "grad_norm": 2.734239101409912, + "learning_rate": 2.9291968024362392e-06, + "loss": 1.3997, + "step": 1396 + }, + { + "epoch": 1.4350282485875705, + "grad_norm": 2.6337532997131348, + "learning_rate": 2.9272934906737726e-06, + "loss": 1.3825, + "step": 1397 + }, + { + "epoch": 1.436055469953775, + "grad_norm": 2.668790578842163, + "learning_rate": 2.925390178911306e-06, + "loss": 1.3918, + "step": 1398 + }, + { + "epoch": 1.4370826913199795, + "grad_norm": 2.8307507038116455, + "learning_rate": 2.9234868671488393e-06, + "loss": 1.4183, + "step": 1399 + }, + { + "epoch": 1.4381099126861838, + "grad_norm": 2.6661787033081055, + "learning_rate": 2.9215835553863727e-06, + "loss": 1.4225, + "step": 1400 + }, + { + "epoch": 1.4381099126861838, + "eval_loss": 1.8518847227096558, + "eval_runtime": 37.3672, + "eval_samples_per_second": 26.761, + "eval_steps_per_second": 1.124, + "step": 1400 + }, + { + "epoch": 1.4391371340523884, + "grad_norm": 2.698025703430176, + "learning_rate": 2.9196802436239056e-06, + "loss": 1.4115, + "step": 1401 + }, + { + "epoch": 1.4401643554185928, + "grad_norm": 2.6792502403259277, + "learning_rate": 2.917776931861439e-06, + "loss": 1.3869, + "step": 1402 + }, + { + "epoch": 1.4411915767847971, + "grad_norm": 2.693514347076416, + "learning_rate": 2.9158736200989723e-06, + "loss": 1.3875, + "step": 1403 + }, + { + "epoch": 1.4422187981510015, + "grad_norm": 2.6383583545684814, + "learning_rate": 2.9139703083365057e-06, + "loss": 1.3942, + "step": 1404 + }, + { + "epoch": 1.4432460195172059, + "grad_norm": 2.6992499828338623, + "learning_rate": 2.912066996574039e-06, + "loss": 1.4243, + "step": 1405 + }, + { + "epoch": 1.4442732408834105, + "grad_norm": 2.7901880741119385, + "learning_rate": 2.910163684811572e-06, + "loss": 1.3559, + "step": 1406 + }, + { + "epoch": 1.4453004622496148, + "grad_norm": 2.9373207092285156, + "learning_rate": 2.9082603730491054e-06, + "loss": 1.4734, + "step": 1407 + }, + { + "epoch": 1.4463276836158192, + "grad_norm": 2.8224740028381348, + "learning_rate": 2.9063570612866387e-06, + "loss": 1.4474, + "step": 1408 + }, + { + "epoch": 1.4473549049820236, + "grad_norm": 2.9534835815429688, + "learning_rate": 2.904453749524172e-06, + "loss": 1.3691, + "step": 1409 + }, + { + "epoch": 1.448382126348228, + "grad_norm": 2.8153076171875, + "learning_rate": 2.902550437761706e-06, + "loss": 1.3954, + "step": 1410 + }, + { + "epoch": 1.448382126348228, + "eval_loss": 1.842891812324524, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1410 + }, + { + "epoch": 1.4494093477144325, + "grad_norm": 2.8490853309631348, + "learning_rate": 2.9006471259992392e-06, + "loss": 1.4114, + "step": 1411 + }, + { + "epoch": 1.450436569080637, + "grad_norm": 2.7475955486297607, + "learning_rate": 2.8987438142367726e-06, + "loss": 1.4068, + "step": 1412 + }, + { + "epoch": 1.4514637904468413, + "grad_norm": 2.7209911346435547, + "learning_rate": 2.8968405024743055e-06, + "loss": 1.4658, + "step": 1413 + }, + { + "epoch": 1.4524910118130456, + "grad_norm": 2.578934907913208, + "learning_rate": 2.894937190711839e-06, + "loss": 1.3686, + "step": 1414 + }, + { + "epoch": 1.45351823317925, + "grad_norm": 2.722381114959717, + "learning_rate": 2.8930338789493723e-06, + "loss": 1.4337, + "step": 1415 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.702178478240967, + "learning_rate": 2.8911305671869056e-06, + "loss": 1.4163, + "step": 1416 + }, + { + "epoch": 1.455572675911659, + "grad_norm": 2.6742405891418457, + "learning_rate": 2.889227255424439e-06, + "loss": 1.3585, + "step": 1417 + }, + { + "epoch": 1.4565998972778633, + "grad_norm": 2.6732680797576904, + "learning_rate": 2.887323943661972e-06, + "loss": 1.406, + "step": 1418 + }, + { + "epoch": 1.457627118644068, + "grad_norm": 2.698864459991455, + "learning_rate": 2.8854206318995053e-06, + "loss": 1.3872, + "step": 1419 + }, + { + "epoch": 1.4586543400102723, + "grad_norm": 2.7534351348876953, + "learning_rate": 2.8835173201370386e-06, + "loss": 1.4055, + "step": 1420 + }, + { + "epoch": 1.4586543400102723, + "eval_loss": 1.8564038276672363, + "eval_runtime": 37.0708, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.4596815613764766, + "grad_norm": 2.9538612365722656, + "learning_rate": 2.881614008374572e-06, + "loss": 1.4596, + "step": 1421 + }, + { + "epoch": 1.460708782742681, + "grad_norm": 2.829641103744507, + "learning_rate": 2.8797106966121054e-06, + "loss": 1.4265, + "step": 1422 + }, + { + "epoch": 1.4617360041088854, + "grad_norm": 2.6814422607421875, + "learning_rate": 2.8778073848496383e-06, + "loss": 1.4112, + "step": 1423 + }, + { + "epoch": 1.46276322547509, + "grad_norm": 2.7880561351776123, + "learning_rate": 2.8759040730871717e-06, + "loss": 1.4435, + "step": 1424 + }, + { + "epoch": 1.4637904468412943, + "grad_norm": 2.676196336746216, + "learning_rate": 2.874000761324705e-06, + "loss": 1.4104, + "step": 1425 + }, + { + "epoch": 1.4648176682074987, + "grad_norm": 2.7417874336242676, + "learning_rate": 2.8720974495622384e-06, + "loss": 1.4335, + "step": 1426 + }, + { + "epoch": 1.465844889573703, + "grad_norm": 2.7870116233825684, + "learning_rate": 2.8701941377997718e-06, + "loss": 1.4226, + "step": 1427 + }, + { + "epoch": 1.4668721109399074, + "grad_norm": 2.765194892883301, + "learning_rate": 2.8682908260373047e-06, + "loss": 1.4582, + "step": 1428 + }, + { + "epoch": 1.467899332306112, + "grad_norm": 2.608175277709961, + "learning_rate": 2.866387514274838e-06, + "loss": 1.4285, + "step": 1429 + }, + { + "epoch": 1.4689265536723164, + "grad_norm": 2.7573461532592773, + "learning_rate": 2.864484202512372e-06, + "loss": 1.4395, + "step": 1430 + }, + { + "epoch": 1.4689265536723164, + "eval_loss": 1.8600319623947144, + "eval_runtime": 37.1192, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1430 + }, + { + "epoch": 1.4699537750385208, + "grad_norm": 2.635175943374634, + "learning_rate": 2.8625808907499052e-06, + "loss": 1.402, + "step": 1431 + }, + { + "epoch": 1.4709809964047253, + "grad_norm": 2.7318811416625977, + "learning_rate": 2.8606775789874386e-06, + "loss": 1.4352, + "step": 1432 + }, + { + "epoch": 1.4720082177709297, + "grad_norm": 2.783669948577881, + "learning_rate": 2.858774267224972e-06, + "loss": 1.4052, + "step": 1433 + }, + { + "epoch": 1.473035439137134, + "grad_norm": 2.7327511310577393, + "learning_rate": 2.8568709554625053e-06, + "loss": 1.4209, + "step": 1434 + }, + { + "epoch": 1.4740626605033385, + "grad_norm": 2.76542329788208, + "learning_rate": 2.8549676437000382e-06, + "loss": 1.4753, + "step": 1435 + }, + { + "epoch": 1.4750898818695428, + "grad_norm": 2.8031232357025146, + "learning_rate": 2.8530643319375716e-06, + "loss": 1.4399, + "step": 1436 + }, + { + "epoch": 1.4761171032357474, + "grad_norm": 2.737053155899048, + "learning_rate": 2.851161020175105e-06, + "loss": 1.4085, + "step": 1437 + }, + { + "epoch": 1.4771443246019518, + "grad_norm": 2.609987735748291, + "learning_rate": 2.8492577084126383e-06, + "loss": 1.3809, + "step": 1438 + }, + { + "epoch": 1.4781715459681561, + "grad_norm": 2.7482123374938965, + "learning_rate": 2.8473543966501717e-06, + "loss": 1.4234, + "step": 1439 + }, + { + "epoch": 1.4791987673343605, + "grad_norm": 2.80330753326416, + "learning_rate": 2.8454510848877046e-06, + "loss": 1.4665, + "step": 1440 + }, + { + "epoch": 1.4791987673343605, + "eval_loss": 1.8637592792510986, + "eval_runtime": 37.1066, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 1440 + }, + { + "epoch": 1.4802259887005649, + "grad_norm": 2.698328733444214, + "learning_rate": 2.843547773125238e-06, + "loss": 1.4224, + "step": 1441 + }, + { + "epoch": 1.4812532100667695, + "grad_norm": 2.663471221923828, + "learning_rate": 2.8416444613627714e-06, + "loss": 1.4205, + "step": 1442 + }, + { + "epoch": 1.4822804314329738, + "grad_norm": 2.7955918312072754, + "learning_rate": 2.8397411496003047e-06, + "loss": 1.3694, + "step": 1443 + }, + { + "epoch": 1.4833076527991782, + "grad_norm": 2.7320404052734375, + "learning_rate": 2.837837837837838e-06, + "loss": 1.4208, + "step": 1444 + }, + { + "epoch": 1.4843348741653826, + "grad_norm": 2.7786989212036133, + "learning_rate": 2.835934526075371e-06, + "loss": 1.4392, + "step": 1445 + }, + { + "epoch": 1.485362095531587, + "grad_norm": 2.6282734870910645, + "learning_rate": 2.8340312143129044e-06, + "loss": 1.3866, + "step": 1446 + }, + { + "epoch": 1.4863893168977915, + "grad_norm": 2.615072011947632, + "learning_rate": 2.8321279025504377e-06, + "loss": 1.3695, + "step": 1447 + }, + { + "epoch": 1.487416538263996, + "grad_norm": 2.884423017501831, + "learning_rate": 2.830224590787971e-06, + "loss": 1.4867, + "step": 1448 + }, + { + "epoch": 1.4884437596302003, + "grad_norm": 2.7745864391326904, + "learning_rate": 2.8283212790255045e-06, + "loss": 1.42, + "step": 1449 + }, + { + "epoch": 1.4894709809964048, + "grad_norm": 2.718130111694336, + "learning_rate": 2.8264179672630383e-06, + "loss": 1.4665, + "step": 1450 + }, + { + "epoch": 1.4894709809964048, + "eval_loss": 1.8556917905807495, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1450 + }, + { + "epoch": 1.4904982023626092, + "grad_norm": 2.7502355575561523, + "learning_rate": 2.8245146555005716e-06, + "loss": 1.3946, + "step": 1451 + }, + { + "epoch": 1.4915254237288136, + "grad_norm": 2.635500192642212, + "learning_rate": 2.8226113437381046e-06, + "loss": 1.3612, + "step": 1452 + }, + { + "epoch": 1.492552645095018, + "grad_norm": 2.617408275604248, + "learning_rate": 2.820708031975638e-06, + "loss": 1.4146, + "step": 1453 + }, + { + "epoch": 1.4935798664612223, + "grad_norm": 2.7706692218780518, + "learning_rate": 2.8188047202131713e-06, + "loss": 1.4474, + "step": 1454 + }, + { + "epoch": 1.494607087827427, + "grad_norm": 2.7609055042266846, + "learning_rate": 2.8169014084507046e-06, + "loss": 1.4618, + "step": 1455 + }, + { + "epoch": 1.4956343091936313, + "grad_norm": 2.757258892059326, + "learning_rate": 2.814998096688238e-06, + "loss": 1.4131, + "step": 1456 + }, + { + "epoch": 1.4966615305598356, + "grad_norm": 2.5547049045562744, + "learning_rate": 2.813094784925771e-06, + "loss": 1.3692, + "step": 1457 + }, + { + "epoch": 1.49768875192604, + "grad_norm": 2.817617177963257, + "learning_rate": 2.8111914731633043e-06, + "loss": 1.4664, + "step": 1458 + }, + { + "epoch": 1.4987159732922444, + "grad_norm": 2.6999337673187256, + "learning_rate": 2.8092881614008377e-06, + "loss": 1.4126, + "step": 1459 + }, + { + "epoch": 1.499743194658449, + "grad_norm": 2.571727991104126, + "learning_rate": 2.807384849638371e-06, + "loss": 1.3828, + "step": 1460 + }, + { + "epoch": 1.499743194658449, + "eval_loss": 1.8413318395614624, + "eval_runtime": 37.1803, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1460 + }, + { + "epoch": 1.5007704160246533, + "grad_norm": 2.615567684173584, + "learning_rate": 2.8054815378759044e-06, + "loss": 1.4336, + "step": 1461 + }, + { + "epoch": 1.5017976373908577, + "grad_norm": 2.696854591369629, + "learning_rate": 2.8035782261134378e-06, + "loss": 1.4015, + "step": 1462 + }, + { + "epoch": 1.5028248587570623, + "grad_norm": 2.7395496368408203, + "learning_rate": 2.8016749143509707e-06, + "loss": 1.4009, + "step": 1463 + }, + { + "epoch": 1.5038520801232664, + "grad_norm": 2.8281755447387695, + "learning_rate": 2.799771602588504e-06, + "loss": 1.4069, + "step": 1464 + }, + { + "epoch": 1.504879301489471, + "grad_norm": 2.641793727874756, + "learning_rate": 2.7978682908260374e-06, + "loss": 1.4341, + "step": 1465 + }, + { + "epoch": 1.5059065228556754, + "grad_norm": 2.8969004154205322, + "learning_rate": 2.795964979063571e-06, + "loss": 1.4493, + "step": 1466 + }, + { + "epoch": 1.5069337442218798, + "grad_norm": 2.758255958557129, + "learning_rate": 2.794061667301104e-06, + "loss": 1.423, + "step": 1467 + }, + { + "epoch": 1.5079609655880843, + "grad_norm": 2.7737441062927246, + "learning_rate": 2.792158355538637e-06, + "loss": 1.43, + "step": 1468 + }, + { + "epoch": 1.5089881869542885, + "grad_norm": 2.798095226287842, + "learning_rate": 2.7902550437761705e-06, + "loss": 1.4276, + "step": 1469 + }, + { + "epoch": 1.510015408320493, + "grad_norm": 2.760437488555908, + "learning_rate": 2.7883517320137042e-06, + "loss": 1.3927, + "step": 1470 + }, + { + "epoch": 1.510015408320493, + "eval_loss": 1.8517959117889404, + "eval_runtime": 37.163, + "eval_samples_per_second": 26.908, + "eval_steps_per_second": 1.13, + "step": 1470 + }, + { + "epoch": 1.5110426296866974, + "grad_norm": 2.689023733139038, + "learning_rate": 2.7864484202512376e-06, + "loss": 1.387, + "step": 1471 + }, + { + "epoch": 1.5120698510529018, + "grad_norm": 2.876424551010132, + "learning_rate": 2.784545108488771e-06, + "loss": 1.4183, + "step": 1472 + }, + { + "epoch": 1.5130970724191064, + "grad_norm": 2.789368152618408, + "learning_rate": 2.7826417967263043e-06, + "loss": 1.4239, + "step": 1473 + }, + { + "epoch": 1.5141242937853108, + "grad_norm": 2.945103883743286, + "learning_rate": 2.7807384849638373e-06, + "loss": 1.458, + "step": 1474 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.694589614868164, + "learning_rate": 2.7788351732013706e-06, + "loss": 1.4145, + "step": 1475 + }, + { + "epoch": 1.5161787365177197, + "grad_norm": 2.7276177406311035, + "learning_rate": 2.776931861438904e-06, + "loss": 1.4063, + "step": 1476 + }, + { + "epoch": 1.5172059578839239, + "grad_norm": 2.694197177886963, + "learning_rate": 2.7750285496764374e-06, + "loss": 1.4125, + "step": 1477 + }, + { + "epoch": 1.5182331792501285, + "grad_norm": 2.832883834838867, + "learning_rate": 2.7731252379139707e-06, + "loss": 1.4176, + "step": 1478 + }, + { + "epoch": 1.5192604006163328, + "grad_norm": 2.67136549949646, + "learning_rate": 2.771221926151504e-06, + "loss": 1.3633, + "step": 1479 + }, + { + "epoch": 1.5202876219825372, + "grad_norm": 2.9618499279022217, + "learning_rate": 2.769318614389037e-06, + "loss": 1.4283, + "step": 1480 + }, + { + "epoch": 1.5202876219825372, + "eval_loss": 1.8486552238464355, + "eval_runtime": 37.1808, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1480 + }, + { + "epoch": 1.5213148433487418, + "grad_norm": 2.91403865814209, + "learning_rate": 2.7674153026265704e-06, + "loss": 1.4254, + "step": 1481 + }, + { + "epoch": 1.522342064714946, + "grad_norm": 2.9121804237365723, + "learning_rate": 2.7655119908641037e-06, + "loss": 1.4251, + "step": 1482 + }, + { + "epoch": 1.5233692860811505, + "grad_norm": 2.7484965324401855, + "learning_rate": 2.763608679101637e-06, + "loss": 1.4141, + "step": 1483 + }, + { + "epoch": 1.5243965074473549, + "grad_norm": 2.913203716278076, + "learning_rate": 2.7617053673391705e-06, + "loss": 1.4516, + "step": 1484 + }, + { + "epoch": 1.5254237288135593, + "grad_norm": 2.7844395637512207, + "learning_rate": 2.7598020555767034e-06, + "loss": 1.45, + "step": 1485 + }, + { + "epoch": 1.5264509501797638, + "grad_norm": 2.8250396251678467, + "learning_rate": 2.7578987438142368e-06, + "loss": 1.3746, + "step": 1486 + }, + { + "epoch": 1.5274781715459682, + "grad_norm": 2.749629497528076, + "learning_rate": 2.75599543205177e-06, + "loss": 1.433, + "step": 1487 + }, + { + "epoch": 1.5285053929121726, + "grad_norm": 2.7559521198272705, + "learning_rate": 2.7540921202893035e-06, + "loss": 1.452, + "step": 1488 + }, + { + "epoch": 1.529532614278377, + "grad_norm": 2.825148344039917, + "learning_rate": 2.752188808526837e-06, + "loss": 1.427, + "step": 1489 + }, + { + "epoch": 1.5305598356445813, + "grad_norm": 2.645327091217041, + "learning_rate": 2.75028549676437e-06, + "loss": 1.4089, + "step": 1490 + }, + { + "epoch": 1.5305598356445813, + "eval_loss": 1.8497828245162964, + "eval_runtime": 37.3756, + "eval_samples_per_second": 26.755, + "eval_steps_per_second": 1.124, + "step": 1490 + }, + { + "epoch": 1.531587057010786, + "grad_norm": 2.753156900405884, + "learning_rate": 2.7483821850019036e-06, + "loss": 1.4163, + "step": 1491 + }, + { + "epoch": 1.5326142783769903, + "grad_norm": 2.8842933177948, + "learning_rate": 2.746478873239437e-06, + "loss": 1.4041, + "step": 1492 + }, + { + "epoch": 1.5336414997431946, + "grad_norm": 2.7529265880584717, + "learning_rate": 2.7445755614769703e-06, + "loss": 1.41, + "step": 1493 + }, + { + "epoch": 1.5346687211093992, + "grad_norm": 2.6907176971435547, + "learning_rate": 2.7426722497145037e-06, + "loss": 1.3841, + "step": 1494 + }, + { + "epoch": 1.5356959424756034, + "grad_norm": 2.793073892593384, + "learning_rate": 2.740768937952037e-06, + "loss": 1.3885, + "step": 1495 + }, + { + "epoch": 1.536723163841808, + "grad_norm": 2.7895116806030273, + "learning_rate": 2.7388656261895704e-06, + "loss": 1.4065, + "step": 1496 + }, + { + "epoch": 1.5377503852080123, + "grad_norm": 2.903615713119507, + "learning_rate": 2.7369623144271033e-06, + "loss": 1.4812, + "step": 1497 + }, + { + "epoch": 1.5387776065742167, + "grad_norm": 2.7081339359283447, + "learning_rate": 2.7350590026646367e-06, + "loss": 1.403, + "step": 1498 + }, + { + "epoch": 1.5398048279404213, + "grad_norm": 2.75414776802063, + "learning_rate": 2.73315569090217e-06, + "loss": 1.3629, + "step": 1499 + }, + { + "epoch": 1.5408320493066254, + "grad_norm": 2.795982599258423, + "learning_rate": 2.7312523791397034e-06, + "loss": 1.4551, + "step": 1500 + }, + { + "epoch": 1.5408320493066254, + "eval_loss": 1.8510199785232544, + "eval_runtime": 37.3806, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 1500 + }, + { + "epoch": 1.54185927067283, + "grad_norm": 2.894488573074341, + "learning_rate": 2.7293490673772368e-06, + "loss": 1.4794, + "step": 1501 + }, + { + "epoch": 1.5428864920390344, + "grad_norm": 2.7982053756713867, + "learning_rate": 2.7274457556147697e-06, + "loss": 1.4365, + "step": 1502 + }, + { + "epoch": 1.5439137134052388, + "grad_norm": 2.913367509841919, + "learning_rate": 2.725542443852303e-06, + "loss": 1.4132, + "step": 1503 + }, + { + "epoch": 1.5449409347714433, + "grad_norm": 2.6658358573913574, + "learning_rate": 2.7236391320898365e-06, + "loss": 1.4288, + "step": 1504 + }, + { + "epoch": 1.5459681561376477, + "grad_norm": 2.710700035095215, + "learning_rate": 2.72173582032737e-06, + "loss": 1.3936, + "step": 1505 + }, + { + "epoch": 1.546995377503852, + "grad_norm": 2.7881481647491455, + "learning_rate": 2.719832508564903e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.5480225988700564, + "grad_norm": 2.7621042728424072, + "learning_rate": 2.717929196802436e-06, + "loss": 1.4218, + "step": 1507 + }, + { + "epoch": 1.5490498202362608, + "grad_norm": 2.7504117488861084, + "learning_rate": 2.7160258850399695e-06, + "loss": 1.3918, + "step": 1508 + }, + { + "epoch": 1.5500770416024654, + "grad_norm": 2.719702959060669, + "learning_rate": 2.714122573277503e-06, + "loss": 1.4154, + "step": 1509 + }, + { + "epoch": 1.5511042629686698, + "grad_norm": 2.6698062419891357, + "learning_rate": 2.712219261515036e-06, + "loss": 1.4084, + "step": 1510 + }, + { + "epoch": 1.5511042629686698, + "eval_loss": 1.8468451499938965, + "eval_runtime": 36.9179, + "eval_samples_per_second": 27.087, + "eval_steps_per_second": 1.138, + "step": 1510 + }, + { + "epoch": 1.5521314843348741, + "grad_norm": 2.816978693008423, + "learning_rate": 2.71031594975257e-06, + "loss": 1.4195, + "step": 1511 + }, + { + "epoch": 1.5531587057010787, + "grad_norm": 2.8047056198120117, + "learning_rate": 2.7084126379901034e-06, + "loss": 1.4395, + "step": 1512 + }, + { + "epoch": 1.5541859270672829, + "grad_norm": 2.8461363315582275, + "learning_rate": 2.7065093262276367e-06, + "loss": 1.4472, + "step": 1513 + }, + { + "epoch": 1.5552131484334875, + "grad_norm": 2.712890863418579, + "learning_rate": 2.7046060144651697e-06, + "loss": 1.4538, + "step": 1514 + }, + { + "epoch": 1.5562403697996918, + "grad_norm": 2.844165802001953, + "learning_rate": 2.702702702702703e-06, + "loss": 1.4719, + "step": 1515 + }, + { + "epoch": 1.5572675911658962, + "grad_norm": 2.629138231277466, + "learning_rate": 2.7007993909402364e-06, + "loss": 1.3441, + "step": 1516 + }, + { + "epoch": 1.5582948125321008, + "grad_norm": 2.7072019577026367, + "learning_rate": 2.6988960791777697e-06, + "loss": 1.4692, + "step": 1517 + }, + { + "epoch": 1.559322033898305, + "grad_norm": 2.739567279815674, + "learning_rate": 2.696992767415303e-06, + "loss": 1.4215, + "step": 1518 + }, + { + "epoch": 1.5603492552645095, + "grad_norm": 2.7846791744232178, + "learning_rate": 2.695089455652836e-06, + "loss": 1.4151, + "step": 1519 + }, + { + "epoch": 1.5613764766307139, + "grad_norm": 2.7417876720428467, + "learning_rate": 2.6931861438903694e-06, + "loss": 1.3797, + "step": 1520 + }, + { + "epoch": 1.5613764766307139, + "eval_loss": 1.847011923789978, + "eval_runtime": 37.0386, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1520 + }, + { + "epoch": 1.5624036979969183, + "grad_norm": 2.743271827697754, + "learning_rate": 2.6912828321279028e-06, + "loss": 1.4372, + "step": 1521 + }, + { + "epoch": 1.5634309193631228, + "grad_norm": 2.8119285106658936, + "learning_rate": 2.689379520365436e-06, + "loss": 1.4075, + "step": 1522 + }, + { + "epoch": 1.5644581407293272, + "grad_norm": 2.7443952560424805, + "learning_rate": 2.6874762086029695e-06, + "loss": 1.4442, + "step": 1523 + }, + { + "epoch": 1.5654853620955316, + "grad_norm": 2.7869346141815186, + "learning_rate": 2.6855728968405024e-06, + "loss": 1.3948, + "step": 1524 + }, + { + "epoch": 1.5665125834617362, + "grad_norm": 2.750185251235962, + "learning_rate": 2.683669585078036e-06, + "loss": 1.4313, + "step": 1525 + }, + { + "epoch": 1.5675398048279403, + "grad_norm": 2.787389039993286, + "learning_rate": 2.681766273315569e-06, + "loss": 1.3748, + "step": 1526 + }, + { + "epoch": 1.568567026194145, + "grad_norm": 2.6675543785095215, + "learning_rate": 2.6798629615531025e-06, + "loss": 1.4089, + "step": 1527 + }, + { + "epoch": 1.5695942475603493, + "grad_norm": 2.6468663215637207, + "learning_rate": 2.677959649790636e-06, + "loss": 1.4148, + "step": 1528 + }, + { + "epoch": 1.5706214689265536, + "grad_norm": 2.7241196632385254, + "learning_rate": 2.676056338028169e-06, + "loss": 1.4498, + "step": 1529 + }, + { + "epoch": 1.5716486902927582, + "grad_norm": 2.705486536026001, + "learning_rate": 2.674153026265702e-06, + "loss": 1.4324, + "step": 1530 + }, + { + "epoch": 1.5716486902927582, + "eval_loss": 1.8467438220977783, + "eval_runtime": 37.0576, + "eval_samples_per_second": 26.985, + "eval_steps_per_second": 1.133, + "step": 1530 + }, + { + "epoch": 1.5726759116589624, + "grad_norm": 2.805429458618164, + "learning_rate": 2.672249714503236e-06, + "loss": 1.4475, + "step": 1531 + }, + { + "epoch": 1.573703133025167, + "grad_norm": 2.7074034214019775, + "learning_rate": 2.6703464027407693e-06, + "loss": 1.3997, + "step": 1532 + }, + { + "epoch": 1.5747303543913713, + "grad_norm": 2.737571954727173, + "learning_rate": 2.6684430909783027e-06, + "loss": 1.4238, + "step": 1533 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.703036069869995, + "learning_rate": 2.666539779215836e-06, + "loss": 1.4107, + "step": 1534 + }, + { + "epoch": 1.5767847971237803, + "grad_norm": 2.7987451553344727, + "learning_rate": 2.6646364674533694e-06, + "loss": 1.3927, + "step": 1535 + }, + { + "epoch": 1.5778120184899846, + "grad_norm": 2.622497081756592, + "learning_rate": 2.6627331556909024e-06, + "loss": 1.3215, + "step": 1536 + }, + { + "epoch": 1.578839239856189, + "grad_norm": 2.5890607833862305, + "learning_rate": 2.6608298439284357e-06, + "loss": 1.3994, + "step": 1537 + }, + { + "epoch": 1.5798664612223934, + "grad_norm": 2.6438405513763428, + "learning_rate": 2.658926532165969e-06, + "loss": 1.4042, + "step": 1538 + }, + { + "epoch": 1.5808936825885977, + "grad_norm": 2.690793752670288, + "learning_rate": 2.6570232204035024e-06, + "loss": 1.3673, + "step": 1539 + }, + { + "epoch": 1.5819209039548023, + "grad_norm": 2.7485456466674805, + "learning_rate": 2.655119908641036e-06, + "loss": 1.3979, + "step": 1540 + }, + { + "epoch": 1.5819209039548023, + "eval_loss": 1.852581262588501, + "eval_runtime": 37.0688, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1540 + }, + { + "epoch": 1.5829481253210067, + "grad_norm": 2.8531415462493896, + "learning_rate": 2.6532165968785687e-06, + "loss": 1.4094, + "step": 1541 + }, + { + "epoch": 1.583975346687211, + "grad_norm": 2.638291358947754, + "learning_rate": 2.651313285116102e-06, + "loss": 1.3708, + "step": 1542 + }, + { + "epoch": 1.5850025680534157, + "grad_norm": 2.717550754547119, + "learning_rate": 2.6494099733536355e-06, + "loss": 1.3821, + "step": 1543 + }, + { + "epoch": 1.5860297894196198, + "grad_norm": 2.91959810256958, + "learning_rate": 2.647506661591169e-06, + "loss": 1.5532, + "step": 1544 + }, + { + "epoch": 1.5870570107858244, + "grad_norm": 2.7281882762908936, + "learning_rate": 2.645603349828702e-06, + "loss": 1.4361, + "step": 1545 + }, + { + "epoch": 1.5880842321520288, + "grad_norm": 2.732283592224121, + "learning_rate": 2.643700038066235e-06, + "loss": 1.3669, + "step": 1546 + }, + { + "epoch": 1.5891114535182331, + "grad_norm": 2.731055498123169, + "learning_rate": 2.6417967263037685e-06, + "loss": 1.4223, + "step": 1547 + }, + { + "epoch": 1.5901386748844377, + "grad_norm": 2.831334352493286, + "learning_rate": 2.639893414541302e-06, + "loss": 1.4454, + "step": 1548 + }, + { + "epoch": 1.5911658962506419, + "grad_norm": 2.920548677444458, + "learning_rate": 2.6379901027788352e-06, + "loss": 1.4686, + "step": 1549 + }, + { + "epoch": 1.5921931176168465, + "grad_norm": 2.6671719551086426, + "learning_rate": 2.6360867910163686e-06, + "loss": 1.398, + "step": 1550 + }, + { + "epoch": 1.5921931176168465, + "eval_loss": 1.857794165611267, + "eval_runtime": 37.0817, + "eval_samples_per_second": 26.967, + "eval_steps_per_second": 1.133, + "step": 1550 + }, + { + "epoch": 1.5932203389830508, + "grad_norm": 2.685175657272339, + "learning_rate": 2.6341834792539024e-06, + "loss": 1.4176, + "step": 1551 + }, + { + "epoch": 1.5942475603492552, + "grad_norm": 2.7412822246551514, + "learning_rate": 2.6322801674914357e-06, + "loss": 1.4195, + "step": 1552 + }, + { + "epoch": 1.5952747817154598, + "grad_norm": 2.7461724281311035, + "learning_rate": 2.6303768557289687e-06, + "loss": 1.4081, + "step": 1553 + }, + { + "epoch": 1.5963020030816641, + "grad_norm": 2.6494853496551514, + "learning_rate": 2.628473543966502e-06, + "loss": 1.3759, + "step": 1554 + }, + { + "epoch": 1.5973292244478685, + "grad_norm": 2.736752986907959, + "learning_rate": 2.6265702322040354e-06, + "loss": 1.4415, + "step": 1555 + }, + { + "epoch": 1.5983564458140729, + "grad_norm": 2.733452796936035, + "learning_rate": 2.6246669204415688e-06, + "loss": 1.3867, + "step": 1556 + }, + { + "epoch": 1.5993836671802772, + "grad_norm": 2.692986249923706, + "learning_rate": 2.622763608679102e-06, + "loss": 1.3811, + "step": 1557 + }, + { + "epoch": 1.6004108885464818, + "grad_norm": 2.580052614212036, + "learning_rate": 2.620860296916635e-06, + "loss": 1.399, + "step": 1558 + }, + { + "epoch": 1.6014381099126862, + "grad_norm": 2.651503086090088, + "learning_rate": 2.6189569851541684e-06, + "loss": 1.34, + "step": 1559 + }, + { + "epoch": 1.6024653312788906, + "grad_norm": 2.8157193660736084, + "learning_rate": 2.617053673391702e-06, + "loss": 1.4437, + "step": 1560 + }, + { + "epoch": 1.6024653312788906, + "eval_loss": 1.8588496446609497, + "eval_runtime": 37.0786, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 1560 + }, + { + "epoch": 1.6034925526450952, + "grad_norm": 2.893589973449707, + "learning_rate": 2.615150361629235e-06, + "loss": 1.4775, + "step": 1561 + }, + { + "epoch": 1.6045197740112993, + "grad_norm": 2.719688653945923, + "learning_rate": 2.6132470498667685e-06, + "loss": 1.432, + "step": 1562 + }, + { + "epoch": 1.605546995377504, + "grad_norm": 2.5626275539398193, + "learning_rate": 2.6113437381043015e-06, + "loss": 1.4118, + "step": 1563 + }, + { + "epoch": 1.6065742167437083, + "grad_norm": 2.6879632472991943, + "learning_rate": 2.609440426341835e-06, + "loss": 1.398, + "step": 1564 + }, + { + "epoch": 1.6076014381099126, + "grad_norm": 2.890155792236328, + "learning_rate": 2.607537114579368e-06, + "loss": 1.4336, + "step": 1565 + }, + { + "epoch": 1.6086286594761172, + "grad_norm": 2.8333349227905273, + "learning_rate": 2.6056338028169015e-06, + "loss": 1.4148, + "step": 1566 + }, + { + "epoch": 1.6096558808423214, + "grad_norm": 2.729215383529663, + "learning_rate": 2.603730491054435e-06, + "loss": 1.3831, + "step": 1567 + }, + { + "epoch": 1.610683102208526, + "grad_norm": 2.800276279449463, + "learning_rate": 2.601827179291968e-06, + "loss": 1.4315, + "step": 1568 + }, + { + "epoch": 1.6117103235747303, + "grad_norm": 2.7900125980377197, + "learning_rate": 2.599923867529501e-06, + "loss": 1.4257, + "step": 1569 + }, + { + "epoch": 1.6127375449409347, + "grad_norm": 2.7753336429595947, + "learning_rate": 2.5980205557670346e-06, + "loss": 1.4089, + "step": 1570 + }, + { + "epoch": 1.6127375449409347, + "eval_loss": 1.8533762693405151, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1570 + }, + { + "epoch": 1.6137647663071393, + "grad_norm": 2.9164156913757324, + "learning_rate": 2.5961172440045684e-06, + "loss": 1.4467, + "step": 1571 + }, + { + "epoch": 1.6147919876733436, + "grad_norm": 2.747676134109497, + "learning_rate": 2.5942139322421017e-06, + "loss": 1.4521, + "step": 1572 + }, + { + "epoch": 1.615819209039548, + "grad_norm": 2.6693432331085205, + "learning_rate": 2.592310620479635e-06, + "loss": 1.3982, + "step": 1573 + }, + { + "epoch": 1.6168464304057526, + "grad_norm": 2.5721216201782227, + "learning_rate": 2.5904073087171684e-06, + "loss": 1.3657, + "step": 1574 + }, + { + "epoch": 1.6178736517719567, + "grad_norm": 2.706143617630005, + "learning_rate": 2.5885039969547014e-06, + "loss": 1.4315, + "step": 1575 + }, + { + "epoch": 1.6189008731381613, + "grad_norm": 2.59006404876709, + "learning_rate": 2.5866006851922347e-06, + "loss": 1.372, + "step": 1576 + }, + { + "epoch": 1.6199280945043657, + "grad_norm": 2.7117512226104736, + "learning_rate": 2.584697373429768e-06, + "loss": 1.3834, + "step": 1577 + }, + { + "epoch": 1.62095531587057, + "grad_norm": 2.718029737472534, + "learning_rate": 2.5827940616673015e-06, + "loss": 1.3978, + "step": 1578 + }, + { + "epoch": 1.6219825372367747, + "grad_norm": 2.7660539150238037, + "learning_rate": 2.580890749904835e-06, + "loss": 1.4505, + "step": 1579 + }, + { + "epoch": 1.6230097586029788, + "grad_norm": 2.7671236991882324, + "learning_rate": 2.5789874381423678e-06, + "loss": 1.3951, + "step": 1580 + }, + { + "epoch": 1.6230097586029788, + "eval_loss": 1.8412485122680664, + "eval_runtime": 37.1162, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.6240369799691834, + "grad_norm": 2.6504807472229004, + "learning_rate": 2.577084126379901e-06, + "loss": 1.3957, + "step": 1581 + }, + { + "epoch": 1.6250642013353878, + "grad_norm": 2.688166618347168, + "learning_rate": 2.5751808146174345e-06, + "loss": 1.3686, + "step": 1582 + }, + { + "epoch": 1.6260914227015921, + "grad_norm": 2.7034637928009033, + "learning_rate": 2.573277502854968e-06, + "loss": 1.4201, + "step": 1583 + }, + { + "epoch": 1.6271186440677967, + "grad_norm": 2.6097683906555176, + "learning_rate": 2.5713741910925012e-06, + "loss": 1.3881, + "step": 1584 + }, + { + "epoch": 1.6281458654340009, + "grad_norm": 2.6197433471679688, + "learning_rate": 2.569470879330034e-06, + "loss": 1.3962, + "step": 1585 + }, + { + "epoch": 1.6291730868002054, + "grad_norm": 3.0436367988586426, + "learning_rate": 2.5675675675675675e-06, + "loss": 1.4714, + "step": 1586 + }, + { + "epoch": 1.6302003081664098, + "grad_norm": 2.708744764328003, + "learning_rate": 2.565664255805101e-06, + "loss": 1.4007, + "step": 1587 + }, + { + "epoch": 1.6312275295326142, + "grad_norm": 2.708279848098755, + "learning_rate": 2.5637609440426343e-06, + "loss": 1.4284, + "step": 1588 + }, + { + "epoch": 1.6322547508988188, + "grad_norm": 2.7955429553985596, + "learning_rate": 2.5618576322801676e-06, + "loss": 1.3736, + "step": 1589 + }, + { + "epoch": 1.6332819722650231, + "grad_norm": 2.890211820602417, + "learning_rate": 2.5599543205177006e-06, + "loss": 1.4671, + "step": 1590 + }, + { + "epoch": 1.6332819722650231, + "eval_loss": 1.8467649221420288, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 1590 + }, + { + "epoch": 1.6343091936312275, + "grad_norm": 2.698066234588623, + "learning_rate": 2.558051008755234e-06, + "loss": 1.4229, + "step": 1591 + }, + { + "epoch": 1.635336414997432, + "grad_norm": 2.8393352031707764, + "learning_rate": 2.5561476969927677e-06, + "loss": 1.469, + "step": 1592 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.589914321899414, + "learning_rate": 2.554244385230301e-06, + "loss": 1.3853, + "step": 1593 + }, + { + "epoch": 1.6373908577298408, + "grad_norm": 2.630443572998047, + "learning_rate": 2.5523410734678344e-06, + "loss": 1.4176, + "step": 1594 + }, + { + "epoch": 1.6384180790960452, + "grad_norm": 2.784212112426758, + "learning_rate": 2.550437761705368e-06, + "loss": 1.344, + "step": 1595 + }, + { + "epoch": 1.6394453004622496, + "grad_norm": 2.736009359359741, + "learning_rate": 2.548534449942901e-06, + "loss": 1.3867, + "step": 1596 + }, + { + "epoch": 1.6404725218284542, + "grad_norm": 2.9134042263031006, + "learning_rate": 2.546631138180434e-06, + "loss": 1.4214, + "step": 1597 + }, + { + "epoch": 1.6414997431946583, + "grad_norm": 2.995575428009033, + "learning_rate": 2.5447278264179675e-06, + "loss": 1.409, + "step": 1598 + }, + { + "epoch": 1.642526964560863, + "grad_norm": 2.744499683380127, + "learning_rate": 2.542824514655501e-06, + "loss": 1.3727, + "step": 1599 + }, + { + "epoch": 1.6435541859270673, + "grad_norm": 2.835494041442871, + "learning_rate": 2.540921202893034e-06, + "loss": 1.4636, + "step": 1600 + }, + { + "epoch": 1.6435541859270673, + "eval_loss": 1.8454105854034424, + "eval_runtime": 37.4381, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1600 + }, + { + "epoch": 1.6445814072932716, + "grad_norm": 2.817537307739258, + "learning_rate": 2.5390178911305675e-06, + "loss": 1.4416, + "step": 1601 + }, + { + "epoch": 1.6456086286594762, + "grad_norm": 2.868093252182007, + "learning_rate": 2.5371145793681005e-06, + "loss": 1.4187, + "step": 1602 + }, + { + "epoch": 1.6466358500256806, + "grad_norm": 2.861865758895874, + "learning_rate": 2.535211267605634e-06, + "loss": 1.4067, + "step": 1603 + }, + { + "epoch": 1.647663071391885, + "grad_norm": 2.7015087604522705, + "learning_rate": 2.533307955843167e-06, + "loss": 1.3528, + "step": 1604 + }, + { + "epoch": 1.6486902927580893, + "grad_norm": 2.7585439682006836, + "learning_rate": 2.5314046440807006e-06, + "loss": 1.4157, + "step": 1605 + }, + { + "epoch": 1.6497175141242937, + "grad_norm": 2.853614330291748, + "learning_rate": 2.529501332318234e-06, + "loss": 1.3932, + "step": 1606 + }, + { + "epoch": 1.6507447354904983, + "grad_norm": 2.9736239910125732, + "learning_rate": 2.5275980205557673e-06, + "loss": 1.4511, + "step": 1607 + }, + { + "epoch": 1.6517719568567026, + "grad_norm": 2.955389976501465, + "learning_rate": 2.5256947087933002e-06, + "loss": 1.4777, + "step": 1608 + }, + { + "epoch": 1.652799178222907, + "grad_norm": 2.9508774280548096, + "learning_rate": 2.5237913970308336e-06, + "loss": 1.4523, + "step": 1609 + }, + { + "epoch": 1.6538263995891116, + "grad_norm": 2.88675856590271, + "learning_rate": 2.521888085268367e-06, + "loss": 1.4064, + "step": 1610 + }, + { + "epoch": 1.6538263995891116, + "eval_loss": 1.8569127321243286, + "eval_runtime": 36.9819, + "eval_samples_per_second": 27.04, + "eval_steps_per_second": 1.136, + "step": 1610 + }, + { + "epoch": 1.6548536209553157, + "grad_norm": 2.8439440727233887, + "learning_rate": 2.5199847735059003e-06, + "loss": 1.4591, + "step": 1611 + }, + { + "epoch": 1.6558808423215203, + "grad_norm": 2.9314420223236084, + "learning_rate": 2.518081461743434e-06, + "loss": 1.39, + "step": 1612 + }, + { + "epoch": 1.6569080636877247, + "grad_norm": 2.6517245769500732, + "learning_rate": 2.5161781499809675e-06, + "loss": 1.4234, + "step": 1613 + }, + { + "epoch": 1.657935285053929, + "grad_norm": 2.5690908432006836, + "learning_rate": 2.5142748382185004e-06, + "loss": 1.3762, + "step": 1614 + }, + { + "epoch": 1.6589625064201337, + "grad_norm": 2.808892011642456, + "learning_rate": 2.5123715264560338e-06, + "loss": 1.3727, + "step": 1615 + }, + { + "epoch": 1.6599897277863378, + "grad_norm": 2.777846336364746, + "learning_rate": 2.510468214693567e-06, + "loss": 1.3941, + "step": 1616 + }, + { + "epoch": 1.6610169491525424, + "grad_norm": 2.8298423290252686, + "learning_rate": 2.5085649029311005e-06, + "loss": 1.3455, + "step": 1617 + }, + { + "epoch": 1.6620441705187468, + "grad_norm": 2.6464784145355225, + "learning_rate": 2.506661591168634e-06, + "loss": 1.3061, + "step": 1618 + }, + { + "epoch": 1.6630713918849511, + "grad_norm": 2.7149951457977295, + "learning_rate": 2.504758279406167e-06, + "loss": 1.3995, + "step": 1619 + }, + { + "epoch": 1.6640986132511557, + "grad_norm": 3.0328569412231445, + "learning_rate": 2.5028549676437e-06, + "loss": 1.4343, + "step": 1620 + }, + { + "epoch": 1.6640986132511557, + "eval_loss": 1.8629075288772583, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1620 + }, + { + "epoch": 1.66512583461736, + "grad_norm": 2.8838441371917725, + "learning_rate": 2.5009516558812335e-06, + "loss": 1.3817, + "step": 1621 + }, + { + "epoch": 1.6661530559835644, + "grad_norm": 2.843242645263672, + "learning_rate": 2.499048344118767e-06, + "loss": 1.4322, + "step": 1622 + }, + { + "epoch": 1.667180277349769, + "grad_norm": 2.642946720123291, + "learning_rate": 2.4971450323563002e-06, + "loss": 1.4702, + "step": 1623 + }, + { + "epoch": 1.6682074987159732, + "grad_norm": 2.666194200515747, + "learning_rate": 2.4952417205938336e-06, + "loss": 1.3808, + "step": 1624 + }, + { + "epoch": 1.6692347200821778, + "grad_norm": 2.7515366077423096, + "learning_rate": 2.4933384088313666e-06, + "loss": 1.3892, + "step": 1625 + }, + { + "epoch": 1.6702619414483821, + "grad_norm": 2.7051661014556885, + "learning_rate": 2.4914350970689e-06, + "loss": 1.3809, + "step": 1626 + }, + { + "epoch": 1.6712891628145865, + "grad_norm": 2.7311975955963135, + "learning_rate": 2.4895317853064337e-06, + "loss": 1.3952, + "step": 1627 + }, + { + "epoch": 1.672316384180791, + "grad_norm": 2.8365893363952637, + "learning_rate": 2.4876284735439666e-06, + "loss": 1.4505, + "step": 1628 + }, + { + "epoch": 1.6733436055469952, + "grad_norm": 2.644507884979248, + "learning_rate": 2.4857251617815e-06, + "loss": 1.4114, + "step": 1629 + }, + { + "epoch": 1.6743708269131998, + "grad_norm": 2.7952826023101807, + "learning_rate": 2.4838218500190334e-06, + "loss": 1.4223, + "step": 1630 + }, + { + "epoch": 1.6743708269131998, + "eval_loss": 1.8577097654342651, + "eval_runtime": 37.1645, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 1630 + }, + { + "epoch": 1.6753980482794042, + "grad_norm": 2.787752866744995, + "learning_rate": 2.4819185382565667e-06, + "loss": 1.4022, + "step": 1631 + }, + { + "epoch": 1.6764252696456086, + "grad_norm": 2.6963443756103516, + "learning_rate": 2.4800152264941e-06, + "loss": 1.3974, + "step": 1632 + }, + { + "epoch": 1.6774524910118132, + "grad_norm": 2.6753835678100586, + "learning_rate": 2.478111914731633e-06, + "loss": 1.3639, + "step": 1633 + }, + { + "epoch": 1.6784797123780173, + "grad_norm": 2.7086808681488037, + "learning_rate": 2.4762086029691664e-06, + "loss": 1.3987, + "step": 1634 + }, + { + "epoch": 1.6795069337442219, + "grad_norm": 3.2228872776031494, + "learning_rate": 2.4743052912066998e-06, + "loss": 1.5202, + "step": 1635 + }, + { + "epoch": 1.6805341551104263, + "grad_norm": 2.8254599571228027, + "learning_rate": 2.472401979444233e-06, + "loss": 1.3781, + "step": 1636 + }, + { + "epoch": 1.6815613764766306, + "grad_norm": 2.9406354427337646, + "learning_rate": 2.4704986676817665e-06, + "loss": 1.4686, + "step": 1637 + }, + { + "epoch": 1.6825885978428352, + "grad_norm": 2.8287105560302734, + "learning_rate": 2.4685953559193e-06, + "loss": 1.4026, + "step": 1638 + }, + { + "epoch": 1.6836158192090396, + "grad_norm": 2.923633337020874, + "learning_rate": 2.466692044156833e-06, + "loss": 1.4493, + "step": 1639 + }, + { + "epoch": 1.684643040575244, + "grad_norm": 2.9239513874053955, + "learning_rate": 2.4647887323943666e-06, + "loss": 1.3878, + "step": 1640 + }, + { + "epoch": 1.684643040575244, + "eval_loss": 1.853607416152954, + "eval_runtime": 37.1661, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1640 + }, + { + "epoch": 1.6856702619414485, + "grad_norm": 2.699998617172241, + "learning_rate": 2.4628854206319e-06, + "loss": 1.3637, + "step": 1641 + }, + { + "epoch": 1.6866974833076527, + "grad_norm": 2.901700258255005, + "learning_rate": 2.460982108869433e-06, + "loss": 1.3554, + "step": 1642 + }, + { + "epoch": 1.6877247046738573, + "grad_norm": 3.011157512664795, + "learning_rate": 2.4590787971069662e-06, + "loss": 1.4487, + "step": 1643 + }, + { + "epoch": 1.6887519260400616, + "grad_norm": 2.672830581665039, + "learning_rate": 2.4571754853444996e-06, + "loss": 1.3906, + "step": 1644 + }, + { + "epoch": 1.689779147406266, + "grad_norm": 2.7201497554779053, + "learning_rate": 2.455272173582033e-06, + "loss": 1.3747, + "step": 1645 + }, + { + "epoch": 1.6908063687724706, + "grad_norm": 2.979395627975464, + "learning_rate": 2.4533688618195663e-06, + "loss": 1.3895, + "step": 1646 + }, + { + "epoch": 1.6918335901386747, + "grad_norm": 2.8048512935638428, + "learning_rate": 2.4514655500570997e-06, + "loss": 1.3271, + "step": 1647 + }, + { + "epoch": 1.6928608115048793, + "grad_norm": 2.6457700729370117, + "learning_rate": 2.449562238294633e-06, + "loss": 1.3679, + "step": 1648 + }, + { + "epoch": 1.6938880328710837, + "grad_norm": 2.732454299926758, + "learning_rate": 2.4476589265321664e-06, + "loss": 1.3332, + "step": 1649 + }, + { + "epoch": 1.694915254237288, + "grad_norm": 3.190871238708496, + "learning_rate": 2.4457556147696993e-06, + "loss": 1.4437, + "step": 1650 + }, + { + "epoch": 1.694915254237288, + "eval_loss": 1.8642219305038452, + "eval_runtime": 37.1865, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 1650 + }, + { + "epoch": 1.6959424756034926, + "grad_norm": 2.85784912109375, + "learning_rate": 2.4438523030072327e-06, + "loss": 1.4121, + "step": 1651 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.766545534133911, + "learning_rate": 2.441948991244766e-06, + "loss": 1.4231, + "step": 1652 + }, + { + "epoch": 1.6979969183359014, + "grad_norm": 2.8962197303771973, + "learning_rate": 2.4400456794822994e-06, + "loss": 1.4146, + "step": 1653 + }, + { + "epoch": 1.6990241397021058, + "grad_norm": 2.907231569290161, + "learning_rate": 2.438142367719833e-06, + "loss": 1.4375, + "step": 1654 + }, + { + "epoch": 1.7000513610683101, + "grad_norm": 2.8746557235717773, + "learning_rate": 2.4362390559573657e-06, + "loss": 1.4002, + "step": 1655 + }, + { + "epoch": 1.7010785824345147, + "grad_norm": 2.741013765335083, + "learning_rate": 2.434335744194899e-06, + "loss": 1.3865, + "step": 1656 + }, + { + "epoch": 1.702105803800719, + "grad_norm": 3.118908643722534, + "learning_rate": 2.432432432432433e-06, + "loss": 1.4412, + "step": 1657 + }, + { + "epoch": 1.7031330251669234, + "grad_norm": 2.789179563522339, + "learning_rate": 2.4305291206699662e-06, + "loss": 1.3446, + "step": 1658 + }, + { + "epoch": 1.704160246533128, + "grad_norm": 2.9241371154785156, + "learning_rate": 2.428625808907499e-06, + "loss": 1.4263, + "step": 1659 + }, + { + "epoch": 1.7051874678993322, + "grad_norm": 2.8632888793945312, + "learning_rate": 2.4267224971450325e-06, + "loss": 1.4413, + "step": 1660 + }, + { + "epoch": 1.7051874678993322, + "eval_loss": 1.85648512840271, + "eval_runtime": 37.199, + "eval_samples_per_second": 26.882, + "eval_steps_per_second": 1.129, + "step": 1660 + }, + { + "epoch": 1.7062146892655368, + "grad_norm": 2.948183059692383, + "learning_rate": 2.424819185382566e-06, + "loss": 1.4058, + "step": 1661 + }, + { + "epoch": 1.7072419106317411, + "grad_norm": 3.04127836227417, + "learning_rate": 2.4229158736200993e-06, + "loss": 1.3582, + "step": 1662 + }, + { + "epoch": 1.7082691319979455, + "grad_norm": 2.6956393718719482, + "learning_rate": 2.4210125618576326e-06, + "loss": 1.3705, + "step": 1663 + }, + { + "epoch": 1.70929635336415, + "grad_norm": 2.7690250873565674, + "learning_rate": 2.4191092500951656e-06, + "loss": 1.3869, + "step": 1664 + }, + { + "epoch": 1.7103235747303542, + "grad_norm": 2.851231813430786, + "learning_rate": 2.417205938332699e-06, + "loss": 1.4198, + "step": 1665 + }, + { + "epoch": 1.7113507960965588, + "grad_norm": 2.9241318702697754, + "learning_rate": 2.4153026265702323e-06, + "loss": 1.3441, + "step": 1666 + }, + { + "epoch": 1.7123780174627632, + "grad_norm": 2.7888636589050293, + "learning_rate": 2.4133993148077657e-06, + "loss": 1.3765, + "step": 1667 + }, + { + "epoch": 1.7134052388289676, + "grad_norm": 2.891366481781006, + "learning_rate": 2.411496003045299e-06, + "loss": 1.4393, + "step": 1668 + }, + { + "epoch": 1.7144324601951721, + "grad_norm": 2.8162693977355957, + "learning_rate": 2.4095926912828324e-06, + "loss": 1.3811, + "step": 1669 + }, + { + "epoch": 1.7154596815613765, + "grad_norm": 2.6741931438446045, + "learning_rate": 2.4076893795203658e-06, + "loss": 1.404, + "step": 1670 + }, + { + "epoch": 1.7154596815613765, + "eval_loss": 1.8440946340560913, + "eval_runtime": 37.1273, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1670 + }, + { + "epoch": 1.7164869029275809, + "grad_norm": 2.834951400756836, + "learning_rate": 2.405786067757899e-06, + "loss": 1.377, + "step": 1671 + }, + { + "epoch": 1.7175141242937855, + "grad_norm": 2.846113681793213, + "learning_rate": 2.403882755995432e-06, + "loss": 1.4689, + "step": 1672 + }, + { + "epoch": 1.7185413456599896, + "grad_norm": 2.6862854957580566, + "learning_rate": 2.4019794442329654e-06, + "loss": 1.3979, + "step": 1673 + }, + { + "epoch": 1.7195685670261942, + "grad_norm": 2.7214972972869873, + "learning_rate": 2.4000761324704988e-06, + "loss": 1.3738, + "step": 1674 + }, + { + "epoch": 1.7205957883923986, + "grad_norm": 2.7037723064422607, + "learning_rate": 2.398172820708032e-06, + "loss": 1.4184, + "step": 1675 + }, + { + "epoch": 1.721623009758603, + "grad_norm": 2.7854912281036377, + "learning_rate": 2.3962695089455655e-06, + "loss": 1.3854, + "step": 1676 + }, + { + "epoch": 1.7226502311248075, + "grad_norm": 2.640059471130371, + "learning_rate": 2.3943661971830984e-06, + "loss": 1.3656, + "step": 1677 + }, + { + "epoch": 1.7236774524910117, + "grad_norm": 2.6955630779266357, + "learning_rate": 2.3924628854206322e-06, + "loss": 1.3764, + "step": 1678 + }, + { + "epoch": 1.7247046738572163, + "grad_norm": 3.018975257873535, + "learning_rate": 2.3905595736581656e-06, + "loss": 1.3922, + "step": 1679 + }, + { + "epoch": 1.7257318952234206, + "grad_norm": 2.7709803581237793, + "learning_rate": 2.388656261895699e-06, + "loss": 1.4288, + "step": 1680 + }, + { + "epoch": 1.7257318952234206, + "eval_loss": 1.8474006652832031, + "eval_runtime": 37.1233, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 1680 + }, + { + "epoch": 1.726759116589625, + "grad_norm": 2.8898143768310547, + "learning_rate": 2.386752950133232e-06, + "loss": 1.4143, + "step": 1681 + }, + { + "epoch": 1.7277863379558296, + "grad_norm": 2.9092485904693604, + "learning_rate": 2.3848496383707653e-06, + "loss": 1.4097, + "step": 1682 + }, + { + "epoch": 1.7288135593220337, + "grad_norm": 2.886662483215332, + "learning_rate": 2.3829463266082986e-06, + "loss": 1.3968, + "step": 1683 + }, + { + "epoch": 1.7298407806882383, + "grad_norm": 2.7398223876953125, + "learning_rate": 2.381043014845832e-06, + "loss": 1.4198, + "step": 1684 + }, + { + "epoch": 1.7308680020544427, + "grad_norm": 2.626732587814331, + "learning_rate": 2.3791397030833653e-06, + "loss": 1.4116, + "step": 1685 + }, + { + "epoch": 1.731895223420647, + "grad_norm": 2.6434433460235596, + "learning_rate": 2.3772363913208983e-06, + "loss": 1.3857, + "step": 1686 + }, + { + "epoch": 1.7329224447868516, + "grad_norm": 2.83876633644104, + "learning_rate": 2.3753330795584316e-06, + "loss": 1.3672, + "step": 1687 + }, + { + "epoch": 1.733949666153056, + "grad_norm": 2.925100564956665, + "learning_rate": 2.3734297677959654e-06, + "loss": 1.3957, + "step": 1688 + }, + { + "epoch": 1.7349768875192604, + "grad_norm": 2.833876371383667, + "learning_rate": 2.3715264560334984e-06, + "loss": 1.442, + "step": 1689 + }, + { + "epoch": 1.736004108885465, + "grad_norm": 2.8355112075805664, + "learning_rate": 2.3696231442710317e-06, + "loss": 1.4153, + "step": 1690 + }, + { + "epoch": 1.736004108885465, + "eval_loss": 1.8604445457458496, + "eval_runtime": 37.2018, + "eval_samples_per_second": 26.88, + "eval_steps_per_second": 1.129, + "step": 1690 + }, + { + "epoch": 1.7370313302516691, + "grad_norm": 2.7500364780426025, + "learning_rate": 2.367719832508565e-06, + "loss": 1.4165, + "step": 1691 + }, + { + "epoch": 1.7380585516178737, + "grad_norm": 2.87259578704834, + "learning_rate": 2.3658165207460985e-06, + "loss": 1.4431, + "step": 1692 + }, + { + "epoch": 1.739085772984078, + "grad_norm": 2.899538993835449, + "learning_rate": 2.363913208983632e-06, + "loss": 1.4134, + "step": 1693 + }, + { + "epoch": 1.7401129943502824, + "grad_norm": 2.799593687057495, + "learning_rate": 2.362009897221165e-06, + "loss": 1.4249, + "step": 1694 + }, + { + "epoch": 1.741140215716487, + "grad_norm": 2.8966968059539795, + "learning_rate": 2.360106585458698e-06, + "loss": 1.4184, + "step": 1695 + }, + { + "epoch": 1.7421674370826912, + "grad_norm": 2.6944682598114014, + "learning_rate": 2.3582032736962315e-06, + "loss": 1.3678, + "step": 1696 + }, + { + "epoch": 1.7431946584488958, + "grad_norm": 2.819845199584961, + "learning_rate": 2.356299961933765e-06, + "loss": 1.407, + "step": 1697 + }, + { + "epoch": 1.7442218798151001, + "grad_norm": 2.6873581409454346, + "learning_rate": 2.354396650171298e-06, + "loss": 1.3771, + "step": 1698 + }, + { + "epoch": 1.7452491011813045, + "grad_norm": 2.6941304206848145, + "learning_rate": 2.3524933384088316e-06, + "loss": 1.3743, + "step": 1699 + }, + { + "epoch": 1.746276322547509, + "grad_norm": 2.925926685333252, + "learning_rate": 2.350590026646365e-06, + "loss": 1.4549, + "step": 1700 + }, + { + "epoch": 1.746276322547509, + "eval_loss": 1.8619909286499023, + "eval_runtime": 37.3963, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 1700 + }, + { + "epoch": 1.7473035439137135, + "grad_norm": 2.7275006771087646, + "learning_rate": 2.3486867148838983e-06, + "loss": 1.385, + "step": 1701 + }, + { + "epoch": 1.7483307652799178, + "grad_norm": 2.7096338272094727, + "learning_rate": 2.3467834031214317e-06, + "loss": 1.3732, + "step": 1702 + }, + { + "epoch": 1.7493579866461222, + "grad_norm": 2.718862295150757, + "learning_rate": 2.3448800913589646e-06, + "loss": 1.3889, + "step": 1703 + }, + { + "epoch": 1.7503852080123266, + "grad_norm": 2.8386173248291016, + "learning_rate": 2.342976779596498e-06, + "loss": 1.4083, + "step": 1704 + }, + { + "epoch": 1.7514124293785311, + "grad_norm": 2.868790864944458, + "learning_rate": 2.3410734678340313e-06, + "loss": 1.3958, + "step": 1705 + }, + { + "epoch": 1.7524396507447355, + "grad_norm": 2.869119882583618, + "learning_rate": 2.3391701560715647e-06, + "loss": 1.4344, + "step": 1706 + }, + { + "epoch": 1.7534668721109399, + "grad_norm": 2.702211380004883, + "learning_rate": 2.337266844309098e-06, + "loss": 1.3928, + "step": 1707 + }, + { + "epoch": 1.7544940934771445, + "grad_norm": 2.6727139949798584, + "learning_rate": 2.3353635325466314e-06, + "loss": 1.3872, + "step": 1708 + }, + { + "epoch": 1.7555213148433486, + "grad_norm": 2.875605583190918, + "learning_rate": 2.3334602207841648e-06, + "loss": 1.4097, + "step": 1709 + }, + { + "epoch": 1.7565485362095532, + "grad_norm": 2.7349722385406494, + "learning_rate": 2.331556909021698e-06, + "loss": 1.4041, + "step": 1710 + }, + { + "epoch": 1.7565485362095532, + "eval_loss": 1.8533939123153687, + "eval_runtime": 36.9293, + "eval_samples_per_second": 27.079, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 2.7475039958953857, + "learning_rate": 2.3296535972592315e-06, + "loss": 1.4339, + "step": 1711 + }, + { + "epoch": 1.758602978941962, + "grad_norm": 2.705641746520996, + "learning_rate": 2.3277502854967644e-06, + "loss": 1.3763, + "step": 1712 + }, + { + "epoch": 1.7596302003081665, + "grad_norm": 2.77583646774292, + "learning_rate": 2.325846973734298e-06, + "loss": 1.4423, + "step": 1713 + }, + { + "epoch": 1.7606574216743707, + "grad_norm": 2.662750482559204, + "learning_rate": 2.323943661971831e-06, + "loss": 1.3864, + "step": 1714 + }, + { + "epoch": 1.7616846430405753, + "grad_norm": 2.769313335418701, + "learning_rate": 2.3220403502093645e-06, + "loss": 1.4272, + "step": 1715 + }, + { + "epoch": 1.7627118644067796, + "grad_norm": 2.7852976322174072, + "learning_rate": 2.320137038446898e-06, + "loss": 1.4398, + "step": 1716 + }, + { + "epoch": 1.763739085772984, + "grad_norm": 2.7452852725982666, + "learning_rate": 2.318233726684431e-06, + "loss": 1.4504, + "step": 1717 + }, + { + "epoch": 1.7647663071391886, + "grad_norm": 2.762258291244507, + "learning_rate": 2.3163304149219646e-06, + "loss": 1.4325, + "step": 1718 + }, + { + "epoch": 1.765793528505393, + "grad_norm": 2.7612709999084473, + "learning_rate": 2.314427103159498e-06, + "loss": 1.4027, + "step": 1719 + }, + { + "epoch": 1.7668207498715973, + "grad_norm": 2.7290871143341064, + "learning_rate": 2.312523791397031e-06, + "loss": 1.3919, + "step": 1720 + }, + { + "epoch": 1.7668207498715973, + "eval_loss": 1.8454608917236328, + "eval_runtime": 37.0509, + "eval_samples_per_second": 26.99, + "eval_steps_per_second": 1.134, + "step": 1720 + }, + { + "epoch": 1.767847971237802, + "grad_norm": 2.5960779190063477, + "learning_rate": 2.3106204796345643e-06, + "loss": 1.3632, + "step": 1721 + }, + { + "epoch": 1.768875192604006, + "grad_norm": 2.7832717895507812, + "learning_rate": 2.3087171678720976e-06, + "loss": 1.3899, + "step": 1722 + }, + { + "epoch": 1.7699024139702106, + "grad_norm": 2.8227648735046387, + "learning_rate": 2.306813856109631e-06, + "loss": 1.4517, + "step": 1723 + }, + { + "epoch": 1.770929635336415, + "grad_norm": 2.880059242248535, + "learning_rate": 2.3049105443471644e-06, + "loss": 1.4104, + "step": 1724 + }, + { + "epoch": 1.7719568567026194, + "grad_norm": 2.6653501987457275, + "learning_rate": 2.3030072325846973e-06, + "loss": 1.3908, + "step": 1725 + }, + { + "epoch": 1.772984078068824, + "grad_norm": 2.844914197921753, + "learning_rate": 2.3011039208222307e-06, + "loss": 1.3781, + "step": 1726 + }, + { + "epoch": 1.774011299435028, + "grad_norm": 2.8133163452148438, + "learning_rate": 2.299200609059764e-06, + "loss": 1.3775, + "step": 1727 + }, + { + "epoch": 1.7750385208012327, + "grad_norm": 2.7466955184936523, + "learning_rate": 2.297297297297298e-06, + "loss": 1.3926, + "step": 1728 + }, + { + "epoch": 1.776065742167437, + "grad_norm": 2.6311862468719482, + "learning_rate": 2.2953939855348308e-06, + "loss": 1.3486, + "step": 1729 + }, + { + "epoch": 1.7770929635336414, + "grad_norm": 2.7905211448669434, + "learning_rate": 2.293490673772364e-06, + "loss": 1.4369, + "step": 1730 + }, + { + "epoch": 1.7770929635336414, + "eval_loss": 1.8418751955032349, + "eval_runtime": 37.0448, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1730 + }, + { + "epoch": 1.778120184899846, + "grad_norm": 2.862295627593994, + "learning_rate": 2.2915873620098975e-06, + "loss": 1.4209, + "step": 1731 + }, + { + "epoch": 1.7791474062660502, + "grad_norm": 2.765862464904785, + "learning_rate": 2.289684050247431e-06, + "loss": 1.4135, + "step": 1732 + }, + { + "epoch": 1.7801746276322548, + "grad_norm": 2.7296290397644043, + "learning_rate": 2.287780738484964e-06, + "loss": 1.3814, + "step": 1733 + }, + { + "epoch": 1.7812018489984591, + "grad_norm": 2.7937533855438232, + "learning_rate": 2.285877426722497e-06, + "loss": 1.4203, + "step": 1734 + }, + { + "epoch": 1.7822290703646635, + "grad_norm": 2.7680230140686035, + "learning_rate": 2.2839741149600305e-06, + "loss": 1.4354, + "step": 1735 + }, + { + "epoch": 1.783256291730868, + "grad_norm": 2.8785109519958496, + "learning_rate": 2.282070803197564e-06, + "loss": 1.4338, + "step": 1736 + }, + { + "epoch": 1.7842835130970724, + "grad_norm": 2.747436046600342, + "learning_rate": 2.2801674914350972e-06, + "loss": 1.4087, + "step": 1737 + }, + { + "epoch": 1.7853107344632768, + "grad_norm": 2.6810362339019775, + "learning_rate": 2.2782641796726306e-06, + "loss": 1.3993, + "step": 1738 + }, + { + "epoch": 1.7863379558294814, + "grad_norm": 2.7365734577178955, + "learning_rate": 2.276360867910164e-06, + "loss": 1.4235, + "step": 1739 + }, + { + "epoch": 1.7873651771956856, + "grad_norm": 2.6946303844451904, + "learning_rate": 2.2744575561476973e-06, + "loss": 1.3626, + "step": 1740 + }, + { + "epoch": 1.7873651771956856, + "eval_loss": 1.8456190824508667, + "eval_runtime": 37.069, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1740 + }, + { + "epoch": 1.7883923985618901, + "grad_norm": 2.756920337677002, + "learning_rate": 2.2725542443852307e-06, + "loss": 1.396, + "step": 1741 + }, + { + "epoch": 1.7894196199280945, + "grad_norm": 2.765428066253662, + "learning_rate": 2.2706509326227636e-06, + "loss": 1.4374, + "step": 1742 + }, + { + "epoch": 1.7904468412942989, + "grad_norm": 2.8224971294403076, + "learning_rate": 2.268747620860297e-06, + "loss": 1.4113, + "step": 1743 + }, + { + "epoch": 1.7914740626605035, + "grad_norm": 2.90537428855896, + "learning_rate": 2.2668443090978303e-06, + "loss": 1.4124, + "step": 1744 + }, + { + "epoch": 1.7925012840267076, + "grad_norm": 2.712862491607666, + "learning_rate": 2.2649409973353637e-06, + "loss": 1.4256, + "step": 1745 + }, + { + "epoch": 1.7935285053929122, + "grad_norm": 2.688558340072632, + "learning_rate": 2.263037685572897e-06, + "loss": 1.3842, + "step": 1746 + }, + { + "epoch": 1.7945557267591166, + "grad_norm": 2.6504769325256348, + "learning_rate": 2.26113437381043e-06, + "loss": 1.3703, + "step": 1747 + }, + { + "epoch": 1.795582948125321, + "grad_norm": 3.0093724727630615, + "learning_rate": 2.259231062047964e-06, + "loss": 1.4106, + "step": 1748 + }, + { + "epoch": 1.7966101694915255, + "grad_norm": 2.727687120437622, + "learning_rate": 2.257327750285497e-06, + "loss": 1.4286, + "step": 1749 + }, + { + "epoch": 1.7976373908577299, + "grad_norm": 2.679896116256714, + "learning_rate": 2.2554244385230305e-06, + "loss": 1.3937, + "step": 1750 + }, + { + "epoch": 1.7976373908577299, + "eval_loss": 1.8520207405090332, + "eval_runtime": 37.062, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1750 + }, + { + "epoch": 1.7986646122239343, + "grad_norm": 2.8119945526123047, + "learning_rate": 2.2535211267605635e-06, + "loss": 1.3935, + "step": 1751 + }, + { + "epoch": 1.7996918335901386, + "grad_norm": 2.8470423221588135, + "learning_rate": 2.251617814998097e-06, + "loss": 1.4138, + "step": 1752 + }, + { + "epoch": 1.800719054956343, + "grad_norm": 2.6278114318847656, + "learning_rate": 2.24971450323563e-06, + "loss": 1.328, + "step": 1753 + }, + { + "epoch": 1.8017462763225476, + "grad_norm": 2.861860990524292, + "learning_rate": 2.2478111914731636e-06, + "loss": 1.416, + "step": 1754 + }, + { + "epoch": 1.802773497688752, + "grad_norm": 2.559736728668213, + "learning_rate": 2.245907879710697e-06, + "loss": 1.2891, + "step": 1755 + }, + { + "epoch": 1.8038007190549563, + "grad_norm": 2.8461058139801025, + "learning_rate": 2.24400456794823e-06, + "loss": 1.4194, + "step": 1756 + }, + { + "epoch": 1.804827940421161, + "grad_norm": 2.7069878578186035, + "learning_rate": 2.2421012561857632e-06, + "loss": 1.382, + "step": 1757 + }, + { + "epoch": 1.805855161787365, + "grad_norm": 2.727435827255249, + "learning_rate": 2.240197944423297e-06, + "loss": 1.3714, + "step": 1758 + }, + { + "epoch": 1.8068823831535696, + "grad_norm": 2.7298061847686768, + "learning_rate": 2.23829463266083e-06, + "loss": 1.3834, + "step": 1759 + }, + { + "epoch": 1.807909604519774, + "grad_norm": 2.7419843673706055, + "learning_rate": 2.2363913208983633e-06, + "loss": 1.3687, + "step": 1760 + }, + { + "epoch": 1.807909604519774, + "eval_loss": 1.8575356006622314, + "eval_runtime": 37.0563, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1760 + }, + { + "epoch": 1.8089368258859784, + "grad_norm": 2.8058817386627197, + "learning_rate": 2.2344880091358967e-06, + "loss": 1.415, + "step": 1761 + }, + { + "epoch": 1.809964047252183, + "grad_norm": 2.7287003993988037, + "learning_rate": 2.23258469737343e-06, + "loss": 1.3778, + "step": 1762 + }, + { + "epoch": 1.810991268618387, + "grad_norm": 2.7943124771118164, + "learning_rate": 2.2306813856109634e-06, + "loss": 1.433, + "step": 1763 + }, + { + "epoch": 1.8120184899845917, + "grad_norm": 2.7254626750946045, + "learning_rate": 2.2287780738484963e-06, + "loss": 1.3822, + "step": 1764 + }, + { + "epoch": 1.813045711350796, + "grad_norm": 2.799793004989624, + "learning_rate": 2.2268747620860297e-06, + "loss": 1.4595, + "step": 1765 + }, + { + "epoch": 1.8140729327170004, + "grad_norm": 2.79420804977417, + "learning_rate": 2.224971450323563e-06, + "loss": 1.3949, + "step": 1766 + }, + { + "epoch": 1.815100154083205, + "grad_norm": 2.861016035079956, + "learning_rate": 2.2230681385610964e-06, + "loss": 1.3301, + "step": 1767 + }, + { + "epoch": 1.8161273754494094, + "grad_norm": 2.9270451068878174, + "learning_rate": 2.2211648267986298e-06, + "loss": 1.3895, + "step": 1768 + }, + { + "epoch": 1.8171545968156138, + "grad_norm": 2.709031105041504, + "learning_rate": 2.219261515036163e-06, + "loss": 1.3757, + "step": 1769 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 2.7763030529022217, + "learning_rate": 2.2173582032736965e-06, + "loss": 1.4135, + "step": 1770 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.8521159887313843, + "eval_runtime": 37.1265, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 1.8192090395480225, + "grad_norm": 2.97048282623291, + "learning_rate": 2.21545489151123e-06, + "loss": 1.4345, + "step": 1771 + }, + { + "epoch": 1.820236260914227, + "grad_norm": 2.8238744735717773, + "learning_rate": 2.2135515797487632e-06, + "loss": 1.3948, + "step": 1772 + }, + { + "epoch": 1.8212634822804314, + "grad_norm": 2.6430611610412598, + "learning_rate": 2.211648267986296e-06, + "loss": 1.3878, + "step": 1773 + }, + { + "epoch": 1.8222907036466358, + "grad_norm": 2.602883815765381, + "learning_rate": 2.2097449562238295e-06, + "loss": 1.3978, + "step": 1774 + }, + { + "epoch": 1.8233179250128404, + "grad_norm": 2.707813262939453, + "learning_rate": 2.207841644461363e-06, + "loss": 1.3937, + "step": 1775 + }, + { + "epoch": 1.8243451463790445, + "grad_norm": 2.662684679031372, + "learning_rate": 2.2059383326988963e-06, + "loss": 1.4381, + "step": 1776 + }, + { + "epoch": 1.8253723677452491, + "grad_norm": 2.9407193660736084, + "learning_rate": 2.2040350209364296e-06, + "loss": 1.4059, + "step": 1777 + }, + { + "epoch": 1.8263995891114535, + "grad_norm": 2.6678171157836914, + "learning_rate": 2.2021317091739626e-06, + "loss": 1.3718, + "step": 1778 + }, + { + "epoch": 1.8274268104776579, + "grad_norm": 2.7536516189575195, + "learning_rate": 2.2002283974114963e-06, + "loss": 1.3998, + "step": 1779 + }, + { + "epoch": 1.8284540318438625, + "grad_norm": 2.7340128421783447, + "learning_rate": 2.1983250856490297e-06, + "loss": 1.3299, + "step": 1780 + }, + { + "epoch": 1.8284540318438625, + "eval_loss": 1.854264736175537, + "eval_runtime": 37.0874, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1780 + }, + { + "epoch": 1.8294812532100666, + "grad_norm": 2.844285726547241, + "learning_rate": 2.1964217738865626e-06, + "loss": 1.3746, + "step": 1781 + }, + { + "epoch": 1.8305084745762712, + "grad_norm": 2.9005022048950195, + "learning_rate": 2.194518462124096e-06, + "loss": 1.4087, + "step": 1782 + }, + { + "epoch": 1.8315356959424756, + "grad_norm": 2.8825247287750244, + "learning_rate": 2.1926151503616294e-06, + "loss": 1.4458, + "step": 1783 + }, + { + "epoch": 1.83256291730868, + "grad_norm": 2.712778091430664, + "learning_rate": 2.1907118385991627e-06, + "loss": 1.414, + "step": 1784 + }, + { + "epoch": 1.8335901386748845, + "grad_norm": 2.7036001682281494, + "learning_rate": 2.188808526836696e-06, + "loss": 1.3632, + "step": 1785 + }, + { + "epoch": 1.8346173600410889, + "grad_norm": 2.704606771469116, + "learning_rate": 2.1869052150742295e-06, + "loss": 1.3992, + "step": 1786 + }, + { + "epoch": 1.8356445814072933, + "grad_norm": 2.7564282417297363, + "learning_rate": 2.1850019033117624e-06, + "loss": 1.3331, + "step": 1787 + }, + { + "epoch": 1.8366718027734978, + "grad_norm": 2.898256540298462, + "learning_rate": 2.1830985915492958e-06, + "loss": 1.3881, + "step": 1788 + }, + { + "epoch": 1.837699024139702, + "grad_norm": 2.7897069454193115, + "learning_rate": 2.1811952797868295e-06, + "loss": 1.4081, + "step": 1789 + }, + { + "epoch": 1.8387262455059066, + "grad_norm": 2.7293193340301514, + "learning_rate": 2.1792919680243625e-06, + "loss": 1.3705, + "step": 1790 + }, + { + "epoch": 1.8387262455059066, + "eval_loss": 1.8483723402023315, + "eval_runtime": 37.4058, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 1790 + }, + { + "epoch": 1.839753466872111, + "grad_norm": 2.9257752895355225, + "learning_rate": 2.177388656261896e-06, + "loss": 1.4071, + "step": 1791 + }, + { + "epoch": 1.8407806882383153, + "grad_norm": 3.016033172607422, + "learning_rate": 2.1754853444994292e-06, + "loss": 1.4774, + "step": 1792 + }, + { + "epoch": 1.84180790960452, + "grad_norm": 2.74198842048645, + "learning_rate": 2.1735820327369626e-06, + "loss": 1.301, + "step": 1793 + }, + { + "epoch": 1.842835130970724, + "grad_norm": 2.809215545654297, + "learning_rate": 2.171678720974496e-06, + "loss": 1.4192, + "step": 1794 + }, + { + "epoch": 1.8438623523369286, + "grad_norm": 2.673997402191162, + "learning_rate": 2.169775409212029e-06, + "loss": 1.4182, + "step": 1795 + }, + { + "epoch": 1.844889573703133, + "grad_norm": 2.9519944190979004, + "learning_rate": 2.1678720974495622e-06, + "loss": 1.4253, + "step": 1796 + }, + { + "epoch": 1.8459167950693374, + "grad_norm": 2.7683424949645996, + "learning_rate": 2.1659687856870956e-06, + "loss": 1.4162, + "step": 1797 + }, + { + "epoch": 1.846944016435542, + "grad_norm": 2.861701726913452, + "learning_rate": 2.164065473924629e-06, + "loss": 1.4167, + "step": 1798 + }, + { + "epoch": 1.8479712378017463, + "grad_norm": 2.62581467628479, + "learning_rate": 2.1621621621621623e-06, + "loss": 1.3568, + "step": 1799 + }, + { + "epoch": 1.8489984591679507, + "grad_norm": 2.782197952270508, + "learning_rate": 2.1602588503996957e-06, + "loss": 1.4184, + "step": 1800 + }, + { + "epoch": 1.8489984591679507, + "eval_loss": 1.8527827262878418, + "eval_runtime": 37.4743, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 1.850025680534155, + "grad_norm": 2.7289304733276367, + "learning_rate": 2.158355538637229e-06, + "loss": 1.3979, + "step": 1801 + }, + { + "epoch": 1.8510529019003594, + "grad_norm": 2.7153542041778564, + "learning_rate": 2.1564522268747624e-06, + "loss": 1.3863, + "step": 1802 + }, + { + "epoch": 1.852080123266564, + "grad_norm": 2.884798765182495, + "learning_rate": 2.1545489151122958e-06, + "loss": 1.4656, + "step": 1803 + }, + { + "epoch": 1.8531073446327684, + "grad_norm": 2.6483845710754395, + "learning_rate": 2.1526456033498287e-06, + "loss": 1.4118, + "step": 1804 + }, + { + "epoch": 1.8541345659989728, + "grad_norm": 2.7074198722839355, + "learning_rate": 2.150742291587362e-06, + "loss": 1.41, + "step": 1805 + }, + { + "epoch": 1.8551617873651773, + "grad_norm": 2.734137535095215, + "learning_rate": 2.1488389798248954e-06, + "loss": 1.3968, + "step": 1806 + }, + { + "epoch": 1.8561890087313815, + "grad_norm": 2.7855992317199707, + "learning_rate": 2.146935668062429e-06, + "loss": 1.427, + "step": 1807 + }, + { + "epoch": 1.857216230097586, + "grad_norm": 2.7769339084625244, + "learning_rate": 2.145032356299962e-06, + "loss": 1.3476, + "step": 1808 + }, + { + "epoch": 1.8582434514637904, + "grad_norm": 2.6978964805603027, + "learning_rate": 2.1431290445374955e-06, + "loss": 1.4047, + "step": 1809 + }, + { + "epoch": 1.8592706728299948, + "grad_norm": 2.806438446044922, + "learning_rate": 2.141225732775029e-06, + "loss": 1.3693, + "step": 1810 + }, + { + "epoch": 1.8592706728299948, + "eval_loss": 1.8606462478637695, + "eval_runtime": 36.9431, + "eval_samples_per_second": 27.069, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 1.8602978941961994, + "grad_norm": 2.886394500732422, + "learning_rate": 2.1393224210125623e-06, + "loss": 1.4022, + "step": 1811 + }, + { + "epoch": 1.8613251155624035, + "grad_norm": 2.7849435806274414, + "learning_rate": 2.137419109250095e-06, + "loss": 1.3907, + "step": 1812 + }, + { + "epoch": 1.8623523369286081, + "grad_norm": 2.85810923576355, + "learning_rate": 2.1355157974876286e-06, + "loss": 1.4451, + "step": 1813 + }, + { + "epoch": 1.8633795582948125, + "grad_norm": 2.8717806339263916, + "learning_rate": 2.133612485725162e-06, + "loss": 1.4303, + "step": 1814 + }, + { + "epoch": 1.8644067796610169, + "grad_norm": 2.6955783367156982, + "learning_rate": 2.1317091739626953e-06, + "loss": 1.3454, + "step": 1815 + }, + { + "epoch": 1.8654340010272215, + "grad_norm": 2.835347890853882, + "learning_rate": 2.1298058622002286e-06, + "loss": 1.3671, + "step": 1816 + }, + { + "epoch": 1.8664612223934258, + "grad_norm": 2.804835796356201, + "learning_rate": 2.1279025504377616e-06, + "loss": 1.388, + "step": 1817 + }, + { + "epoch": 1.8674884437596302, + "grad_norm": 2.69130539894104, + "learning_rate": 2.125999238675295e-06, + "loss": 1.3815, + "step": 1818 + }, + { + "epoch": 1.8685156651258348, + "grad_norm": 2.856457233428955, + "learning_rate": 2.1240959269128287e-06, + "loss": 1.4323, + "step": 1819 + }, + { + "epoch": 1.869542886492039, + "grad_norm": 2.8100039958953857, + "learning_rate": 2.122192615150362e-06, + "loss": 1.3942, + "step": 1820 + }, + { + "epoch": 1.869542886492039, + "eval_loss": 1.860397219657898, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 1820 + }, + { + "epoch": 1.8705701078582435, + "grad_norm": 2.6977529525756836, + "learning_rate": 2.120289303387895e-06, + "loss": 1.3371, + "step": 1821 + }, + { + "epoch": 1.8715973292244479, + "grad_norm": 2.8403844833374023, + "learning_rate": 2.1183859916254284e-06, + "loss": 1.3776, + "step": 1822 + }, + { + "epoch": 1.8726245505906522, + "grad_norm": 2.829756498336792, + "learning_rate": 2.1164826798629618e-06, + "loss": 1.4229, + "step": 1823 + }, + { + "epoch": 1.8736517719568568, + "grad_norm": 2.6810450553894043, + "learning_rate": 2.114579368100495e-06, + "loss": 1.3757, + "step": 1824 + }, + { + "epoch": 1.874678993323061, + "grad_norm": 2.988982677459717, + "learning_rate": 2.1126760563380285e-06, + "loss": 1.3671, + "step": 1825 + }, + { + "epoch": 1.8757062146892656, + "grad_norm": 2.661953926086426, + "learning_rate": 2.1107727445755614e-06, + "loss": 1.3382, + "step": 1826 + }, + { + "epoch": 1.87673343605547, + "grad_norm": 2.7993199825286865, + "learning_rate": 2.1088694328130948e-06, + "loss": 1.4621, + "step": 1827 + }, + { + "epoch": 1.8777606574216743, + "grad_norm": 2.681609869003296, + "learning_rate": 2.106966121050628e-06, + "loss": 1.3596, + "step": 1828 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.988734722137451, + "learning_rate": 2.1050628092881615e-06, + "loss": 1.4269, + "step": 1829 + }, + { + "epoch": 1.879815100154083, + "grad_norm": 2.867772102355957, + "learning_rate": 2.103159497525695e-06, + "loss": 1.4618, + "step": 1830 + }, + { + "epoch": 1.879815100154083, + "eval_loss": 1.8595913648605347, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1830 + }, + { + "epoch": 1.8808423215202876, + "grad_norm": 2.7212493419647217, + "learning_rate": 2.1012561857632282e-06, + "loss": 1.3738, + "step": 1831 + }, + { + "epoch": 1.881869542886492, + "grad_norm": 2.7997097969055176, + "learning_rate": 2.0993528740007616e-06, + "loss": 1.4268, + "step": 1832 + }, + { + "epoch": 1.8828967642526964, + "grad_norm": 2.8695220947265625, + "learning_rate": 2.097449562238295e-06, + "loss": 1.4397, + "step": 1833 + }, + { + "epoch": 1.883923985618901, + "grad_norm": 2.681015729904175, + "learning_rate": 2.095546250475828e-06, + "loss": 1.3873, + "step": 1834 + }, + { + "epoch": 1.8849512069851053, + "grad_norm": 2.787360429763794, + "learning_rate": 2.0936429387133613e-06, + "loss": 1.3944, + "step": 1835 + }, + { + "epoch": 1.8859784283513097, + "grad_norm": 2.830164670944214, + "learning_rate": 2.0917396269508946e-06, + "loss": 1.4485, + "step": 1836 + }, + { + "epoch": 1.8870056497175143, + "grad_norm": 2.9803659915924072, + "learning_rate": 2.089836315188428e-06, + "loss": 1.4259, + "step": 1837 + }, + { + "epoch": 1.8880328710837184, + "grad_norm": 2.846390962600708, + "learning_rate": 2.0879330034259614e-06, + "loss": 1.4371, + "step": 1838 + }, + { + "epoch": 1.889060092449923, + "grad_norm": 2.636085033416748, + "learning_rate": 2.0860296916634947e-06, + "loss": 1.3679, + "step": 1839 + }, + { + "epoch": 1.8900873138161274, + "grad_norm": 2.806547164916992, + "learning_rate": 2.084126379901028e-06, + "loss": 1.4662, + "step": 1840 + }, + { + "epoch": 1.8900873138161274, + "eval_loss": 1.8502073287963867, + "eval_runtime": 37.1321, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 1.8911145351823317, + "grad_norm": 2.8658089637756348, + "learning_rate": 2.0822230681385614e-06, + "loss": 1.3593, + "step": 1841 + }, + { + "epoch": 1.8921417565485363, + "grad_norm": 2.87060284614563, + "learning_rate": 2.080319756376095e-06, + "loss": 1.4189, + "step": 1842 + }, + { + "epoch": 1.8931689779147405, + "grad_norm": 2.7965638637542725, + "learning_rate": 2.0784164446136277e-06, + "loss": 1.4208, + "step": 1843 + }, + { + "epoch": 1.894196199280945, + "grad_norm": 2.5999045372009277, + "learning_rate": 2.076513132851161e-06, + "loss": 1.375, + "step": 1844 + }, + { + "epoch": 1.8952234206471494, + "grad_norm": 2.833935260772705, + "learning_rate": 2.0746098210886945e-06, + "loss": 1.4137, + "step": 1845 + }, + { + "epoch": 1.8962506420133538, + "grad_norm": 2.9809396266937256, + "learning_rate": 2.072706509326228e-06, + "loss": 1.3874, + "step": 1846 + }, + { + "epoch": 1.8972778633795584, + "grad_norm": 2.9341282844543457, + "learning_rate": 2.070803197563761e-06, + "loss": 1.3953, + "step": 1847 + }, + { + "epoch": 1.8983050847457628, + "grad_norm": 2.989488124847412, + "learning_rate": 2.068899885801294e-06, + "loss": 1.3913, + "step": 1848 + }, + { + "epoch": 1.8993323061119671, + "grad_norm": 2.8224217891693115, + "learning_rate": 2.066996574038828e-06, + "loss": 1.4032, + "step": 1849 + }, + { + "epoch": 1.9003595274781715, + "grad_norm": 2.8850255012512207, + "learning_rate": 2.0650932622763613e-06, + "loss": 1.4402, + "step": 1850 + }, + { + "epoch": 1.9003595274781715, + "eval_loss": 1.8440051078796387, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1850 + }, + { + "epoch": 1.9013867488443759, + "grad_norm": 2.927288293838501, + "learning_rate": 2.0631899505138942e-06, + "loss": 1.3215, + "step": 1851 + }, + { + "epoch": 1.9024139702105805, + "grad_norm": 2.7680368423461914, + "learning_rate": 2.0612866387514276e-06, + "loss": 1.3636, + "step": 1852 + }, + { + "epoch": 1.9034411915767848, + "grad_norm": 2.8515689373016357, + "learning_rate": 2.059383326988961e-06, + "loss": 1.4049, + "step": 1853 + }, + { + "epoch": 1.9044684129429892, + "grad_norm": 2.7973034381866455, + "learning_rate": 2.0574800152264943e-06, + "loss": 1.3876, + "step": 1854 + }, + { + "epoch": 1.9054956343091938, + "grad_norm": 2.7860023975372314, + "learning_rate": 2.0555767034640277e-06, + "loss": 1.3687, + "step": 1855 + }, + { + "epoch": 1.906522855675398, + "grad_norm": 2.807511568069458, + "learning_rate": 2.053673391701561e-06, + "loss": 1.3382, + "step": 1856 + }, + { + "epoch": 1.9075500770416025, + "grad_norm": 2.9342591762542725, + "learning_rate": 2.051770079939094e-06, + "loss": 1.4025, + "step": 1857 + }, + { + "epoch": 1.9085772984078069, + "grad_norm": 2.85943865776062, + "learning_rate": 2.0498667681766273e-06, + "loss": 1.3626, + "step": 1858 + }, + { + "epoch": 1.9096045197740112, + "grad_norm": 2.8514609336853027, + "learning_rate": 2.047963456414161e-06, + "loss": 1.3675, + "step": 1859 + }, + { + "epoch": 1.9106317411402158, + "grad_norm": 2.8177332878112793, + "learning_rate": 2.046060144651694e-06, + "loss": 1.3968, + "step": 1860 + }, + { + "epoch": 1.9106317411402158, + "eval_loss": 1.8451731204986572, + "eval_runtime": 37.1047, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1860 + }, + { + "epoch": 1.91165896250642, + "grad_norm": 2.74876070022583, + "learning_rate": 2.0441568328892274e-06, + "loss": 1.4022, + "step": 1861 + }, + { + "epoch": 1.9126861838726246, + "grad_norm": 2.6633286476135254, + "learning_rate": 2.0422535211267608e-06, + "loss": 1.4004, + "step": 1862 + }, + { + "epoch": 1.913713405238829, + "grad_norm": 2.6457369327545166, + "learning_rate": 2.040350209364294e-06, + "loss": 1.3767, + "step": 1863 + }, + { + "epoch": 1.9147406266050333, + "grad_norm": 2.7011449337005615, + "learning_rate": 2.0384468976018275e-06, + "loss": 1.373, + "step": 1864 + }, + { + "epoch": 1.915767847971238, + "grad_norm": 2.676337480545044, + "learning_rate": 2.0365435858393604e-06, + "loss": 1.3713, + "step": 1865 + }, + { + "epoch": 1.9167950693374423, + "grad_norm": 2.7401342391967773, + "learning_rate": 2.034640274076894e-06, + "loss": 1.3644, + "step": 1866 + }, + { + "epoch": 1.9178222907036466, + "grad_norm": 2.7799527645111084, + "learning_rate": 2.032736962314427e-06, + "loss": 1.433, + "step": 1867 + }, + { + "epoch": 1.9188495120698512, + "grad_norm": 2.7372307777404785, + "learning_rate": 2.0308336505519605e-06, + "loss": 1.3781, + "step": 1868 + }, + { + "epoch": 1.9198767334360554, + "grad_norm": 2.986297845840454, + "learning_rate": 2.028930338789494e-06, + "loss": 1.4406, + "step": 1869 + }, + { + "epoch": 1.92090395480226, + "grad_norm": 2.815706491470337, + "learning_rate": 2.0270270270270273e-06, + "loss": 1.3555, + "step": 1870 + }, + { + "epoch": 1.92090395480226, + "eval_loss": 1.8498139381408691, + "eval_runtime": 37.1146, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1870 + }, + { + "epoch": 1.9219311761684643, + "grad_norm": 2.734189987182617, + "learning_rate": 2.0251237152645606e-06, + "loss": 1.373, + "step": 1871 + }, + { + "epoch": 1.9229583975346687, + "grad_norm": 2.7900359630584717, + "learning_rate": 2.023220403502094e-06, + "loss": 1.3623, + "step": 1872 + }, + { + "epoch": 1.9239856189008733, + "grad_norm": 2.8590710163116455, + "learning_rate": 2.0213170917396273e-06, + "loss": 1.3424, + "step": 1873 + }, + { + "epoch": 1.9250128402670774, + "grad_norm": 2.880814790725708, + "learning_rate": 2.0194137799771603e-06, + "loss": 1.3683, + "step": 1874 + }, + { + "epoch": 1.926040061633282, + "grad_norm": 2.7775630950927734, + "learning_rate": 2.0175104682146937e-06, + "loss": 1.3879, + "step": 1875 + }, + { + "epoch": 1.9270672829994864, + "grad_norm": 2.713557481765747, + "learning_rate": 2.015607156452227e-06, + "loss": 1.3216, + "step": 1876 + }, + { + "epoch": 1.9280945043656907, + "grad_norm": 2.8787734508514404, + "learning_rate": 2.0137038446897604e-06, + "loss": 1.3659, + "step": 1877 + }, + { + "epoch": 1.9291217257318953, + "grad_norm": 2.8224005699157715, + "learning_rate": 2.0118005329272937e-06, + "loss": 1.3572, + "step": 1878 + }, + { + "epoch": 1.9301489470980995, + "grad_norm": 2.865316390991211, + "learning_rate": 2.0098972211648267e-06, + "loss": 1.3817, + "step": 1879 + }, + { + "epoch": 1.931176168464304, + "grad_norm": 2.7488853931427, + "learning_rate": 2.0079939094023605e-06, + "loss": 1.3766, + "step": 1880 + }, + { + "epoch": 1.931176168464304, + "eval_loss": 1.8444081544876099, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1880 + }, + { + "epoch": 1.9322033898305084, + "grad_norm": 2.902949333190918, + "learning_rate": 2.006090597639894e-06, + "loss": 1.4183, + "step": 1881 + }, + { + "epoch": 1.9332306111967128, + "grad_norm": 2.8154122829437256, + "learning_rate": 2.0041872858774268e-06, + "loss": 1.4525, + "step": 1882 + }, + { + "epoch": 1.9342578325629174, + "grad_norm": 2.8343429565429688, + "learning_rate": 2.00228397411496e-06, + "loss": 1.4099, + "step": 1883 + }, + { + "epoch": 1.9352850539291218, + "grad_norm": 2.863067388534546, + "learning_rate": 2.0003806623524935e-06, + "loss": 1.3973, + "step": 1884 + }, + { + "epoch": 1.9363122752953261, + "grad_norm": 2.784085988998413, + "learning_rate": 1.998477350590027e-06, + "loss": 1.3117, + "step": 1885 + }, + { + "epoch": 1.9373394966615307, + "grad_norm": 2.7667958736419678, + "learning_rate": 1.9965740388275602e-06, + "loss": 1.4056, + "step": 1886 + }, + { + "epoch": 1.9383667180277349, + "grad_norm": 2.747432231903076, + "learning_rate": 1.994670727065093e-06, + "loss": 1.3927, + "step": 1887 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9162001609802246, + "learning_rate": 1.9927674153026265e-06, + "loss": 1.3453, + "step": 1888 + }, + { + "epoch": 1.9404211607601438, + "grad_norm": 3.0909414291381836, + "learning_rate": 1.99086410354016e-06, + "loss": 1.4148, + "step": 1889 + }, + { + "epoch": 1.9414483821263482, + "grad_norm": 2.774672746658325, + "learning_rate": 1.9889607917776937e-06, + "loss": 1.3762, + "step": 1890 + }, + { + "epoch": 1.9414483821263482, + "eval_loss": 1.848209023475647, + "eval_runtime": 37.2784, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 1.9424756034925528, + "grad_norm": 2.7985880374908447, + "learning_rate": 1.9870574800152266e-06, + "loss": 1.3908, + "step": 1891 + }, + { + "epoch": 1.943502824858757, + "grad_norm": 2.749664783477783, + "learning_rate": 1.98515416825276e-06, + "loss": 1.4088, + "step": 1892 + }, + { + "epoch": 1.9445300462249615, + "grad_norm": 2.770726442337036, + "learning_rate": 1.9832508564902933e-06, + "loss": 1.4054, + "step": 1893 + }, + { + "epoch": 1.9455572675911659, + "grad_norm": 2.634516477584839, + "learning_rate": 1.9813475447278267e-06, + "loss": 1.3247, + "step": 1894 + }, + { + "epoch": 1.9465844889573702, + "grad_norm": 2.886481285095215, + "learning_rate": 1.97944423296536e-06, + "loss": 1.4025, + "step": 1895 + }, + { + "epoch": 1.9476117103235748, + "grad_norm": 2.8149733543395996, + "learning_rate": 1.977540921202893e-06, + "loss": 1.3745, + "step": 1896 + }, + { + "epoch": 1.9486389316897792, + "grad_norm": 2.643970251083374, + "learning_rate": 1.9756376094404264e-06, + "loss": 1.3581, + "step": 1897 + }, + { + "epoch": 1.9496661530559836, + "grad_norm": 2.7485244274139404, + "learning_rate": 1.9737342976779597e-06, + "loss": 1.3426, + "step": 1898 + }, + { + "epoch": 1.950693374422188, + "grad_norm": 2.930513858795166, + "learning_rate": 1.971830985915493e-06, + "loss": 1.4194, + "step": 1899 + }, + { + "epoch": 1.9517205957883923, + "grad_norm": 2.7637267112731934, + "learning_rate": 1.9699276741530264e-06, + "loss": 1.3908, + "step": 1900 + }, + { + "epoch": 1.9517205957883923, + "eval_loss": 1.8510984182357788, + "eval_runtime": 37.4089, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1900 + }, + { + "epoch": 1.9527478171545969, + "grad_norm": 2.8946938514709473, + "learning_rate": 1.96802436239056e-06, + "loss": 1.3892, + "step": 1901 + }, + { + "epoch": 1.9537750385208013, + "grad_norm": 2.7064263820648193, + "learning_rate": 1.966121050628093e-06, + "loss": 1.3628, + "step": 1902 + }, + { + "epoch": 1.9548022598870056, + "grad_norm": 2.8588147163391113, + "learning_rate": 1.9642177388656265e-06, + "loss": 1.4097, + "step": 1903 + }, + { + "epoch": 1.9558294812532102, + "grad_norm": 2.7746825218200684, + "learning_rate": 1.9623144271031595e-06, + "loss": 1.4255, + "step": 1904 + }, + { + "epoch": 1.9568567026194144, + "grad_norm": 2.872587203979492, + "learning_rate": 1.960411115340693e-06, + "loss": 1.4308, + "step": 1905 + }, + { + "epoch": 1.957883923985619, + "grad_norm": 2.714769124984741, + "learning_rate": 1.958507803578226e-06, + "loss": 1.3847, + "step": 1906 + }, + { + "epoch": 1.9589111453518233, + "grad_norm": 2.7117342948913574, + "learning_rate": 1.9566044918157596e-06, + "loss": 1.406, + "step": 1907 + }, + { + "epoch": 1.9599383667180277, + "grad_norm": 2.754995346069336, + "learning_rate": 1.954701180053293e-06, + "loss": 1.4216, + "step": 1908 + }, + { + "epoch": 1.9609655880842323, + "grad_norm": 2.6597256660461426, + "learning_rate": 1.952797868290826e-06, + "loss": 1.3748, + "step": 1909 + }, + { + "epoch": 1.9619928094504364, + "grad_norm": 2.7966320514678955, + "learning_rate": 1.9508945565283596e-06, + "loss": 1.4237, + "step": 1910 + }, + { + "epoch": 1.9619928094504364, + "eval_loss": 1.8453261852264404, + "eval_runtime": 36.9326, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1910 + }, + { + "epoch": 1.963020030816641, + "grad_norm": 2.7504498958587646, + "learning_rate": 1.948991244765893e-06, + "loss": 1.3832, + "step": 1911 + }, + { + "epoch": 1.9640472521828454, + "grad_norm": 2.7729601860046387, + "learning_rate": 1.9470879330034264e-06, + "loss": 1.4176, + "step": 1912 + }, + { + "epoch": 1.9650744735490497, + "grad_norm": 2.6513381004333496, + "learning_rate": 1.9451846212409593e-06, + "loss": 1.3524, + "step": 1913 + }, + { + "epoch": 1.9661016949152543, + "grad_norm": 2.8147811889648438, + "learning_rate": 1.9432813094784927e-06, + "loss": 1.4119, + "step": 1914 + }, + { + "epoch": 1.9671289162814587, + "grad_norm": 2.7253763675689697, + "learning_rate": 1.941377997716026e-06, + "loss": 1.3458, + "step": 1915 + }, + { + "epoch": 1.968156137647663, + "grad_norm": 2.6741955280303955, + "learning_rate": 1.9394746859535594e-06, + "loss": 1.3764, + "step": 1916 + }, + { + "epoch": 1.9691833590138677, + "grad_norm": 2.785856008529663, + "learning_rate": 1.9375713741910928e-06, + "loss": 1.3484, + "step": 1917 + }, + { + "epoch": 1.9702105803800718, + "grad_norm": 2.743168592453003, + "learning_rate": 1.9356680624286257e-06, + "loss": 1.3779, + "step": 1918 + }, + { + "epoch": 1.9712378017462764, + "grad_norm": 2.827883243560791, + "learning_rate": 1.933764750666159e-06, + "loss": 1.3246, + "step": 1919 + }, + { + "epoch": 1.9722650231124808, + "grad_norm": 2.7379672527313232, + "learning_rate": 1.931861438903693e-06, + "loss": 1.383, + "step": 1920 + }, + { + "epoch": 1.9722650231124808, + "eval_loss": 1.8433737754821777, + "eval_runtime": 37.0421, + "eval_samples_per_second": 26.996, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 1.9732922444786851, + "grad_norm": 2.7514965534210205, + "learning_rate": 1.929958127141226e-06, + "loss": 1.3513, + "step": 1921 + }, + { + "epoch": 1.9743194658448897, + "grad_norm": 2.6764447689056396, + "learning_rate": 1.928054815378759e-06, + "loss": 1.3718, + "step": 1922 + }, + { + "epoch": 1.9753466872110939, + "grad_norm": 2.8592004776000977, + "learning_rate": 1.9261515036162925e-06, + "loss": 1.4479, + "step": 1923 + }, + { + "epoch": 1.9763739085772984, + "grad_norm": 2.8035528659820557, + "learning_rate": 1.924248191853826e-06, + "loss": 1.4222, + "step": 1924 + }, + { + "epoch": 1.9774011299435028, + "grad_norm": 2.7685303688049316, + "learning_rate": 1.9223448800913592e-06, + "loss": 1.3939, + "step": 1925 + }, + { + "epoch": 1.9784283513097072, + "grad_norm": 2.6679201126098633, + "learning_rate": 1.9204415683288926e-06, + "loss": 1.385, + "step": 1926 + }, + { + "epoch": 1.9794555726759118, + "grad_norm": 2.7546098232269287, + "learning_rate": 1.9185382565664255e-06, + "loss": 1.4134, + "step": 1927 + }, + { + "epoch": 1.980482794042116, + "grad_norm": 2.794991970062256, + "learning_rate": 1.916634944803959e-06, + "loss": 1.4212, + "step": 1928 + }, + { + "epoch": 1.9815100154083205, + "grad_norm": 2.767075538635254, + "learning_rate": 1.9147316330414923e-06, + "loss": 1.3979, + "step": 1929 + }, + { + "epoch": 1.9825372367745249, + "grad_norm": 2.7712795734405518, + "learning_rate": 1.9128283212790256e-06, + "loss": 1.3827, + "step": 1930 + }, + { + "epoch": 1.9825372367745249, + "eval_loss": 1.8437753915786743, + "eval_runtime": 37.0885, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1930 + }, + { + "epoch": 1.9835644581407292, + "grad_norm": 2.791165351867676, + "learning_rate": 1.910925009516559e-06, + "loss": 1.4328, + "step": 1931 + }, + { + "epoch": 1.9845916795069338, + "grad_norm": 2.7170767784118652, + "learning_rate": 1.9090216977540924e-06, + "loss": 1.3958, + "step": 1932 + }, + { + "epoch": 1.9856189008731382, + "grad_norm": 2.8529574871063232, + "learning_rate": 1.9071183859916257e-06, + "loss": 1.4379, + "step": 1933 + }, + { + "epoch": 1.9866461222393426, + "grad_norm": 2.8424859046936035, + "learning_rate": 1.9052150742291589e-06, + "loss": 1.4066, + "step": 1934 + }, + { + "epoch": 1.9876733436055471, + "grad_norm": 2.8055100440979004, + "learning_rate": 1.9033117624666922e-06, + "loss": 1.4, + "step": 1935 + }, + { + "epoch": 1.9887005649717513, + "grad_norm": 2.73415470123291, + "learning_rate": 1.9014084507042254e-06, + "loss": 1.3652, + "step": 1936 + }, + { + "epoch": 1.9897277863379559, + "grad_norm": 2.7907590866088867, + "learning_rate": 1.8995051389417587e-06, + "loss": 1.3475, + "step": 1937 + }, + { + "epoch": 1.9907550077041603, + "grad_norm": 2.6994218826293945, + "learning_rate": 1.8976018271792921e-06, + "loss": 1.3428, + "step": 1938 + }, + { + "epoch": 1.9917822290703646, + "grad_norm": 2.8237955570220947, + "learning_rate": 1.8956985154168253e-06, + "loss": 1.3918, + "step": 1939 + }, + { + "epoch": 1.9928094504365692, + "grad_norm": 2.779270648956299, + "learning_rate": 1.8937952036543588e-06, + "loss": 1.3726, + "step": 1940 + }, + { + "epoch": 1.9928094504365692, + "eval_loss": 1.8435344696044922, + "eval_runtime": 37.0797, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1940 + }, + { + "epoch": 1.9938366718027734, + "grad_norm": 2.7808098793029785, + "learning_rate": 1.8918918918918922e-06, + "loss": 1.3368, + "step": 1941 + }, + { + "epoch": 1.994863893168978, + "grad_norm": 2.7355751991271973, + "learning_rate": 1.8899885801294253e-06, + "loss": 1.39, + "step": 1942 + }, + { + "epoch": 1.9958911145351823, + "grad_norm": 2.907468318939209, + "learning_rate": 1.8880852683669587e-06, + "loss": 1.399, + "step": 1943 + }, + { + "epoch": 1.9969183359013867, + "grad_norm": 2.8597185611724854, + "learning_rate": 1.886181956604492e-06, + "loss": 1.3731, + "step": 1944 + }, + { + "epoch": 1.9979455572675913, + "grad_norm": 2.8638176918029785, + "learning_rate": 1.8842786448420252e-06, + "loss": 1.3696, + "step": 1945 + }, + { + "epoch": 1.9989727786337956, + "grad_norm": 2.880169153213501, + "learning_rate": 1.8823753330795586e-06, + "loss": 1.3971, + "step": 1946 + }, + { + "epoch": 2.0, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.8804720213170917e-06, + "loss": 1.3026, + "step": 1947 + }, + { + "epoch": 2.0010272213662046, + "grad_norm": 2.9506170749664307, + "learning_rate": 1.878568709554625e-06, + "loss": 1.3298, + "step": 1948 + }, + { + "epoch": 2.0020544427324087, + "grad_norm": 2.8743488788604736, + "learning_rate": 1.8766653977921585e-06, + "loss": 1.3808, + "step": 1949 + }, + { + "epoch": 2.0030816640986133, + "grad_norm": 2.8676979541778564, + "learning_rate": 1.874762086029692e-06, + "loss": 1.3394, + "step": 1950 + }, + { + "epoch": 2.0030816640986133, + "eval_loss": 1.8473349809646606, + "eval_runtime": 37.093, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1950 + }, + { + "epoch": 2.0041088854648175, + "grad_norm": 3.1344428062438965, + "learning_rate": 1.8728587742672252e-06, + "loss": 1.3774, + "step": 1951 + }, + { + "epoch": 2.005136106831022, + "grad_norm": 3.1567065715789795, + "learning_rate": 1.8709554625047585e-06, + "loss": 1.3863, + "step": 1952 + }, + { + "epoch": 2.0061633281972266, + "grad_norm": 3.109293222427368, + "learning_rate": 1.8690521507422917e-06, + "loss": 1.3196, + "step": 1953 + }, + { + "epoch": 2.007190549563431, + "grad_norm": 3.1035029888153076, + "learning_rate": 1.867148838979825e-06, + "loss": 1.3473, + "step": 1954 + }, + { + "epoch": 2.0082177709296354, + "grad_norm": 3.02809476852417, + "learning_rate": 1.8652455272173584e-06, + "loss": 1.3718, + "step": 1955 + }, + { + "epoch": 2.00924499229584, + "grad_norm": 2.918850898742676, + "learning_rate": 1.8633422154548916e-06, + "loss": 1.3794, + "step": 1956 + }, + { + "epoch": 2.010272213662044, + "grad_norm": 2.89725923538208, + "learning_rate": 1.861438903692425e-06, + "loss": 1.2861, + "step": 1957 + }, + { + "epoch": 2.0112994350282487, + "grad_norm": 3.051868200302124, + "learning_rate": 1.859535591929958e-06, + "loss": 1.3122, + "step": 1958 + }, + { + "epoch": 2.012326656394453, + "grad_norm": 3.1091904640197754, + "learning_rate": 1.8576322801674915e-06, + "loss": 1.3941, + "step": 1959 + }, + { + "epoch": 2.0133538777606574, + "grad_norm": 3.046663761138916, + "learning_rate": 1.855728968405025e-06, + "loss": 1.3293, + "step": 1960 + }, + { + "epoch": 2.0133538777606574, + "eval_loss": 1.851180911064148, + "eval_runtime": 37.0842, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1960 + }, + { + "epoch": 2.014381099126862, + "grad_norm": 3.2143752574920654, + "learning_rate": 1.8538256566425584e-06, + "loss": 1.3431, + "step": 1961 + }, + { + "epoch": 2.015408320493066, + "grad_norm": 3.162942409515381, + "learning_rate": 1.8519223448800915e-06, + "loss": 1.3883, + "step": 1962 + }, + { + "epoch": 2.0164355418592708, + "grad_norm": 3.1271183490753174, + "learning_rate": 1.850019033117625e-06, + "loss": 1.3295, + "step": 1963 + }, + { + "epoch": 2.017462763225475, + "grad_norm": 3.07658314704895, + "learning_rate": 1.848115721355158e-06, + "loss": 1.3323, + "step": 1964 + }, + { + "epoch": 2.0184899845916795, + "grad_norm": 3.075946569442749, + "learning_rate": 1.8462124095926914e-06, + "loss": 1.3524, + "step": 1965 + }, + { + "epoch": 2.019517205957884, + "grad_norm": 3.0359086990356445, + "learning_rate": 1.8443090978302248e-06, + "loss": 1.3578, + "step": 1966 + }, + { + "epoch": 2.0205444273240882, + "grad_norm": 3.1957743167877197, + "learning_rate": 1.842405786067758e-06, + "loss": 1.3909, + "step": 1967 + }, + { + "epoch": 2.021571648690293, + "grad_norm": 3.002462863922119, + "learning_rate": 1.8405024743052913e-06, + "loss": 1.3466, + "step": 1968 + }, + { + "epoch": 2.022598870056497, + "grad_norm": 3.0507266521453857, + "learning_rate": 1.8385991625428247e-06, + "loss": 1.3717, + "step": 1969 + }, + { + "epoch": 2.0236260914227016, + "grad_norm": 3.0719404220581055, + "learning_rate": 1.8366958507803578e-06, + "loss": 1.313, + "step": 1970 + }, + { + "epoch": 2.0236260914227016, + "eval_loss": 1.8496923446655273, + "eval_runtime": 37.1112, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 1970 + }, + { + "epoch": 2.024653312788906, + "grad_norm": 3.068819522857666, + "learning_rate": 1.8347925390178914e-06, + "loss": 1.3614, + "step": 1971 + }, + { + "epoch": 2.0256805341551103, + "grad_norm": 3.005855083465576, + "learning_rate": 1.8328892272554247e-06, + "loss": 1.3007, + "step": 1972 + }, + { + "epoch": 2.026707755521315, + "grad_norm": 3.068032741546631, + "learning_rate": 1.8309859154929579e-06, + "loss": 1.3644, + "step": 1973 + }, + { + "epoch": 2.0277349768875195, + "grad_norm": 3.066153049468994, + "learning_rate": 1.8290826037304913e-06, + "loss": 1.3518, + "step": 1974 + }, + { + "epoch": 2.0287621982537236, + "grad_norm": 3.019429922103882, + "learning_rate": 1.8271792919680246e-06, + "loss": 1.3208, + "step": 1975 + }, + { + "epoch": 2.029789419619928, + "grad_norm": 3.2104897499084473, + "learning_rate": 1.8252759802055578e-06, + "loss": 1.329, + "step": 1976 + }, + { + "epoch": 2.0308166409861323, + "grad_norm": 3.272581100463867, + "learning_rate": 1.8233726684430911e-06, + "loss": 1.3665, + "step": 1977 + }, + { + "epoch": 2.031843862352337, + "grad_norm": 3.2155191898345947, + "learning_rate": 1.8214693566806243e-06, + "loss": 1.3178, + "step": 1978 + }, + { + "epoch": 2.0328710837185415, + "grad_norm": 3.130934238433838, + "learning_rate": 1.8195660449181576e-06, + "loss": 1.3492, + "step": 1979 + }, + { + "epoch": 2.0338983050847457, + "grad_norm": 3.211571216583252, + "learning_rate": 1.817662733155691e-06, + "loss": 1.3477, + "step": 1980 + }, + { + "epoch": 2.0338983050847457, + "eval_loss": 1.8516111373901367, + "eval_runtime": 37.1099, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 1980 + }, + { + "epoch": 2.0349255264509503, + "grad_norm": 2.9799704551696777, + "learning_rate": 1.8157594213932244e-06, + "loss": 1.3469, + "step": 1981 + }, + { + "epoch": 2.0359527478171544, + "grad_norm": 3.0847010612487793, + "learning_rate": 1.8138561096307577e-06, + "loss": 1.3587, + "step": 1982 + }, + { + "epoch": 2.036979969183359, + "grad_norm": 3.0739681720733643, + "learning_rate": 1.811952797868291e-06, + "loss": 1.3053, + "step": 1983 + }, + { + "epoch": 2.0380071905495636, + "grad_norm": 3.0315258502960205, + "learning_rate": 1.8100494861058242e-06, + "loss": 1.3381, + "step": 1984 + }, + { + "epoch": 2.0390344119157677, + "grad_norm": 2.9987258911132812, + "learning_rate": 1.8081461743433576e-06, + "loss": 1.3545, + "step": 1985 + }, + { + "epoch": 2.0400616332819723, + "grad_norm": 2.9383034706115723, + "learning_rate": 1.806242862580891e-06, + "loss": 1.3085, + "step": 1986 + }, + { + "epoch": 2.0410888546481765, + "grad_norm": 3.064450740814209, + "learning_rate": 1.8043395508184241e-06, + "loss": 1.3572, + "step": 1987 + }, + { + "epoch": 2.042116076014381, + "grad_norm": 3.102689027786255, + "learning_rate": 1.8024362390559575e-06, + "loss": 1.3453, + "step": 1988 + }, + { + "epoch": 2.0431432973805856, + "grad_norm": 3.0369763374328613, + "learning_rate": 1.8005329272934906e-06, + "loss": 1.3123, + "step": 1989 + }, + { + "epoch": 2.04417051874679, + "grad_norm": 3.2758162021636963, + "learning_rate": 1.798629615531024e-06, + "loss": 1.4068, + "step": 1990 + }, + { + "epoch": 2.04417051874679, + "eval_loss": 1.8500858545303345, + "eval_runtime": 37.334, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 1990 + }, + { + "epoch": 2.0451977401129944, + "grad_norm": 3.1648876667022705, + "learning_rate": 1.7967263037685576e-06, + "loss": 1.3328, + "step": 1991 + }, + { + "epoch": 2.046224961479199, + "grad_norm": 3.104123830795288, + "learning_rate": 1.794822992006091e-06, + "loss": 1.3297, + "step": 1992 + }, + { + "epoch": 2.047252182845403, + "grad_norm": 3.198483943939209, + "learning_rate": 1.792919680243624e-06, + "loss": 1.3851, + "step": 1993 + }, + { + "epoch": 2.0482794042116077, + "grad_norm": 3.212456226348877, + "learning_rate": 1.7910163684811574e-06, + "loss": 1.3062, + "step": 1994 + }, + { + "epoch": 2.049306625577812, + "grad_norm": 3.058353900909424, + "learning_rate": 1.7891130567186906e-06, + "loss": 1.3347, + "step": 1995 + }, + { + "epoch": 2.0503338469440164, + "grad_norm": 2.963183879852295, + "learning_rate": 1.787209744956224e-06, + "loss": 1.3163, + "step": 1996 + }, + { + "epoch": 2.051361068310221, + "grad_norm": 3.1013290882110596, + "learning_rate": 1.7853064331937573e-06, + "loss": 1.3377, + "step": 1997 + }, + { + "epoch": 2.052388289676425, + "grad_norm": 3.068808078765869, + "learning_rate": 1.7834031214312905e-06, + "loss": 1.3772, + "step": 1998 + }, + { + "epoch": 2.0534155110426298, + "grad_norm": 3.0556447505950928, + "learning_rate": 1.7814998096688238e-06, + "loss": 1.3113, + "step": 1999 + }, + { + "epoch": 2.054442732408834, + "grad_norm": 3.115323543548584, + "learning_rate": 1.779596497906357e-06, + "loss": 1.3499, + "step": 2000 + }, + { + "epoch": 2.054442732408834, + "eval_loss": 1.8396267890930176, + "eval_runtime": 37.4474, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 2000 + }, + { + "epoch": 2.0554699537750385, + "grad_norm": 3.133096694946289, + "learning_rate": 1.7776931861438906e-06, + "loss": 1.3938, + "step": 2001 + }, + { + "epoch": 2.056497175141243, + "grad_norm": 3.259174108505249, + "learning_rate": 1.775789874381424e-06, + "loss": 1.382, + "step": 2002 + }, + { + "epoch": 2.0575243965074472, + "grad_norm": 3.17948317527771, + "learning_rate": 1.7738865626189573e-06, + "loss": 1.3643, + "step": 2003 + }, + { + "epoch": 2.058551617873652, + "grad_norm": 3.0918309688568115, + "learning_rate": 1.7719832508564904e-06, + "loss": 1.3067, + "step": 2004 + }, + { + "epoch": 2.0595788392398564, + "grad_norm": 2.9949889183044434, + "learning_rate": 1.7700799390940238e-06, + "loss": 1.3323, + "step": 2005 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.044935464859009, + "learning_rate": 1.768176627331557e-06, + "loss": 1.359, + "step": 2006 + }, + { + "epoch": 2.061633281972265, + "grad_norm": 2.9965357780456543, + "learning_rate": 1.7662733155690903e-06, + "loss": 1.3456, + "step": 2007 + }, + { + "epoch": 2.0626605033384693, + "grad_norm": 3.092984676361084, + "learning_rate": 1.7643700038066237e-06, + "loss": 1.3831, + "step": 2008 + }, + { + "epoch": 2.063687724704674, + "grad_norm": 3.004263162612915, + "learning_rate": 1.7624666920441568e-06, + "loss": 1.3087, + "step": 2009 + }, + { + "epoch": 2.0647149460708785, + "grad_norm": 3.0408692359924316, + "learning_rate": 1.7605633802816902e-06, + "loss": 1.3058, + "step": 2010 + }, + { + "epoch": 2.0647149460708785, + "eval_loss": 1.8413820266723633, + "eval_runtime": 36.9354, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.0657421674370826, + "grad_norm": 3.155033826828003, + "learning_rate": 1.7586600685192238e-06, + "loss": 1.394, + "step": 2011 + }, + { + "epoch": 2.066769388803287, + "grad_norm": 3.3466567993164062, + "learning_rate": 1.756756756756757e-06, + "loss": 1.401, + "step": 2012 + }, + { + "epoch": 2.0677966101694913, + "grad_norm": 3.219165802001953, + "learning_rate": 1.7548534449942903e-06, + "loss": 1.3564, + "step": 2013 + }, + { + "epoch": 2.068823831535696, + "grad_norm": 3.1733009815216064, + "learning_rate": 1.7529501332318236e-06, + "loss": 1.3446, + "step": 2014 + }, + { + "epoch": 2.0698510529019005, + "grad_norm": 3.1392745971679688, + "learning_rate": 1.7510468214693568e-06, + "loss": 1.3212, + "step": 2015 + }, + { + "epoch": 2.0708782742681047, + "grad_norm": 2.9845967292785645, + "learning_rate": 1.7491435097068902e-06, + "loss": 1.3117, + "step": 2016 + }, + { + "epoch": 2.0719054956343093, + "grad_norm": 3.2182602882385254, + "learning_rate": 1.7472401979444233e-06, + "loss": 1.3539, + "step": 2017 + }, + { + "epoch": 2.0729327170005134, + "grad_norm": 3.0809013843536377, + "learning_rate": 1.7453368861819567e-06, + "loss": 1.2891, + "step": 2018 + }, + { + "epoch": 2.073959938366718, + "grad_norm": 3.0690958499908447, + "learning_rate": 1.74343357441949e-06, + "loss": 1.3381, + "step": 2019 + }, + { + "epoch": 2.0749871597329226, + "grad_norm": 3.091454267501831, + "learning_rate": 1.7415302626570232e-06, + "loss": 1.3874, + "step": 2020 + }, + { + "epoch": 2.0749871597329226, + "eval_loss": 1.8516818284988403, + "eval_runtime": 37.0619, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.0760143810991267, + "grad_norm": 3.039327621459961, + "learning_rate": 1.7396269508945568e-06, + "loss": 1.3295, + "step": 2021 + }, + { + "epoch": 2.0770416024653313, + "grad_norm": 3.0967671871185303, + "learning_rate": 1.7377236391320901e-06, + "loss": 1.3122, + "step": 2022 + }, + { + "epoch": 2.078068823831536, + "grad_norm": 3.228410005569458, + "learning_rate": 1.7358203273696233e-06, + "loss": 1.3766, + "step": 2023 + }, + { + "epoch": 2.07909604519774, + "grad_norm": 3.1973841190338135, + "learning_rate": 1.7339170156071566e-06, + "loss": 1.3325, + "step": 2024 + }, + { + "epoch": 2.0801232665639446, + "grad_norm": 3.131504535675049, + "learning_rate": 1.73201370384469e-06, + "loss": 1.3355, + "step": 2025 + }, + { + "epoch": 2.081150487930149, + "grad_norm": 3.132685661315918, + "learning_rate": 1.7301103920822231e-06, + "loss": 1.3446, + "step": 2026 + }, + { + "epoch": 2.0821777092963534, + "grad_norm": 3.2344162464141846, + "learning_rate": 1.7282070803197565e-06, + "loss": 1.3525, + "step": 2027 + }, + { + "epoch": 2.083204930662558, + "grad_norm": 3.1193928718566895, + "learning_rate": 1.7263037685572897e-06, + "loss": 1.3415, + "step": 2028 + }, + { + "epoch": 2.084232152028762, + "grad_norm": 3.164365291595459, + "learning_rate": 1.724400456794823e-06, + "loss": 1.3652, + "step": 2029 + }, + { + "epoch": 2.0852593733949667, + "grad_norm": 3.0825226306915283, + "learning_rate": 1.7224971450323564e-06, + "loss": 1.3242, + "step": 2030 + }, + { + "epoch": 2.0852593733949667, + "eval_loss": 1.8433457612991333, + "eval_runtime": 37.0921, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.086286594761171, + "grad_norm": 3.1651787757873535, + "learning_rate": 1.72059383326989e-06, + "loss": 1.333, + "step": 2031 + }, + { + "epoch": 2.0873138161273754, + "grad_norm": 3.295436382293701, + "learning_rate": 1.7186905215074231e-06, + "loss": 1.4064, + "step": 2032 + }, + { + "epoch": 2.08834103749358, + "grad_norm": 3.177628517150879, + "learning_rate": 1.7167872097449565e-06, + "loss": 1.3173, + "step": 2033 + }, + { + "epoch": 2.089368258859784, + "grad_norm": 3.112971544265747, + "learning_rate": 1.7148838979824896e-06, + "loss": 1.3401, + "step": 2034 + }, + { + "epoch": 2.0903954802259888, + "grad_norm": 3.150045871734619, + "learning_rate": 1.712980586220023e-06, + "loss": 1.3371, + "step": 2035 + }, + { + "epoch": 2.091422701592193, + "grad_norm": 3.0645205974578857, + "learning_rate": 1.7110772744575563e-06, + "loss": 1.3213, + "step": 2036 + }, + { + "epoch": 2.0924499229583975, + "grad_norm": 3.230185031890869, + "learning_rate": 1.7091739626950895e-06, + "loss": 1.3317, + "step": 2037 + }, + { + "epoch": 2.093477144324602, + "grad_norm": 3.1509249210357666, + "learning_rate": 1.7072706509326229e-06, + "loss": 1.3706, + "step": 2038 + }, + { + "epoch": 2.0945043656908062, + "grad_norm": 3.185981273651123, + "learning_rate": 1.7053673391701562e-06, + "loss": 1.3398, + "step": 2039 + }, + { + "epoch": 2.095531587057011, + "grad_norm": 3.000025510787964, + "learning_rate": 1.7034640274076894e-06, + "loss": 1.3197, + "step": 2040 + }, + { + "epoch": 2.095531587057011, + "eval_loss": 1.8413926362991333, + "eval_runtime": 37.0971, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 2040 + }, + { + "epoch": 2.0965588084232154, + "grad_norm": 3.156095027923584, + "learning_rate": 1.701560715645223e-06, + "loss": 1.3864, + "step": 2041 + }, + { + "epoch": 2.0975860297894195, + "grad_norm": 2.970560073852539, + "learning_rate": 1.6996574038827563e-06, + "loss": 1.3016, + "step": 2042 + }, + { + "epoch": 2.098613251155624, + "grad_norm": 3.141657829284668, + "learning_rate": 1.6977540921202895e-06, + "loss": 1.3488, + "step": 2043 + }, + { + "epoch": 2.0996404725218283, + "grad_norm": 3.125731945037842, + "learning_rate": 1.6958507803578228e-06, + "loss": 1.3275, + "step": 2044 + }, + { + "epoch": 2.100667693888033, + "grad_norm": 3.2703487873077393, + "learning_rate": 1.693947468595356e-06, + "loss": 1.3577, + "step": 2045 + }, + { + "epoch": 2.1016949152542375, + "grad_norm": 3.1855714321136475, + "learning_rate": 1.6920441568328893e-06, + "loss": 1.3671, + "step": 2046 + }, + { + "epoch": 2.1027221366204416, + "grad_norm": 3.0699303150177, + "learning_rate": 1.6901408450704227e-06, + "loss": 1.3066, + "step": 2047 + }, + { + "epoch": 2.103749357986646, + "grad_norm": 3.2417423725128174, + "learning_rate": 1.6882375333079559e-06, + "loss": 1.3274, + "step": 2048 + }, + { + "epoch": 2.1047765793528503, + "grad_norm": 3.265108108520508, + "learning_rate": 1.6863342215454892e-06, + "loss": 1.3924, + "step": 2049 + }, + { + "epoch": 2.105803800719055, + "grad_norm": 3.2078635692596436, + "learning_rate": 1.6844309097830226e-06, + "loss": 1.3782, + "step": 2050 + }, + { + "epoch": 2.105803800719055, + "eval_loss": 1.8441476821899414, + "eval_runtime": 37.1042, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.1068310220852595, + "grad_norm": 3.180485248565674, + "learning_rate": 1.682527598020556e-06, + "loss": 1.3613, + "step": 2051 + }, + { + "epoch": 2.1078582434514637, + "grad_norm": 3.110530376434326, + "learning_rate": 1.6806242862580893e-06, + "loss": 1.376, + "step": 2052 + }, + { + "epoch": 2.1088854648176683, + "grad_norm": 2.997999429702759, + "learning_rate": 1.6787209744956227e-06, + "loss": 1.3172, + "step": 2053 + }, + { + "epoch": 2.109912686183873, + "grad_norm": 3.0154783725738525, + "learning_rate": 1.6768176627331558e-06, + "loss": 1.3015, + "step": 2054 + }, + { + "epoch": 2.110939907550077, + "grad_norm": 2.992138624191284, + "learning_rate": 1.6749143509706892e-06, + "loss": 1.3517, + "step": 2055 + }, + { + "epoch": 2.1119671289162816, + "grad_norm": 3.0572457313537598, + "learning_rate": 1.6730110392082225e-06, + "loss": 1.3301, + "step": 2056 + }, + { + "epoch": 2.1129943502824857, + "grad_norm": 2.9924540519714355, + "learning_rate": 1.6711077274457557e-06, + "loss": 1.3534, + "step": 2057 + }, + { + "epoch": 2.1140215716486903, + "grad_norm": 2.9569015502929688, + "learning_rate": 1.669204415683289e-06, + "loss": 1.2972, + "step": 2058 + }, + { + "epoch": 2.115048793014895, + "grad_norm": 3.064371109008789, + "learning_rate": 1.6673011039208222e-06, + "loss": 1.3553, + "step": 2059 + }, + { + "epoch": 2.116076014381099, + "grad_norm": 3.0954525470733643, + "learning_rate": 1.6653977921583556e-06, + "loss": 1.3421, + "step": 2060 + }, + { + "epoch": 2.116076014381099, + "eval_loss": 1.8407723903656006, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 2060 + }, + { + "epoch": 2.1171032357473036, + "grad_norm": 3.2661712169647217, + "learning_rate": 1.663494480395889e-06, + "loss": 1.3582, + "step": 2061 + }, + { + "epoch": 2.118130457113508, + "grad_norm": 3.1663613319396973, + "learning_rate": 1.6615911686334225e-06, + "loss": 1.3203, + "step": 2062 + }, + { + "epoch": 2.1191576784797124, + "grad_norm": 3.0864570140838623, + "learning_rate": 1.6596878568709557e-06, + "loss": 1.3431, + "step": 2063 + }, + { + "epoch": 2.120184899845917, + "grad_norm": 3.1531989574432373, + "learning_rate": 1.657784545108489e-06, + "loss": 1.3275, + "step": 2064 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.2258596420288086, + "learning_rate": 1.6558812333460222e-06, + "loss": 1.3711, + "step": 2065 + }, + { + "epoch": 2.1222393425783257, + "grad_norm": 3.2881453037261963, + "learning_rate": 1.6539779215835555e-06, + "loss": 1.3688, + "step": 2066 + }, + { + "epoch": 2.12326656394453, + "grad_norm": 3.0734169483184814, + "learning_rate": 1.652074609821089e-06, + "loss": 1.3277, + "step": 2067 + }, + { + "epoch": 2.1242937853107344, + "grad_norm": 2.9943618774414062, + "learning_rate": 1.650171298058622e-06, + "loss": 1.3268, + "step": 2068 + }, + { + "epoch": 2.125321006676939, + "grad_norm": 3.034625291824341, + "learning_rate": 1.6482679862961554e-06, + "loss": 1.3265, + "step": 2069 + }, + { + "epoch": 2.126348228043143, + "grad_norm": 3.0360612869262695, + "learning_rate": 1.6463646745336886e-06, + "loss": 1.3256, + "step": 2070 + }, + { + "epoch": 2.126348228043143, + "eval_loss": 1.8443471193313599, + "eval_runtime": 37.0917, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.1273754494093478, + "grad_norm": 3.0917110443115234, + "learning_rate": 1.644461362771222e-06, + "loss": 1.3099, + "step": 2071 + }, + { + "epoch": 2.128402670775552, + "grad_norm": 3.1142187118530273, + "learning_rate": 1.6425580510087555e-06, + "loss": 1.3421, + "step": 2072 + }, + { + "epoch": 2.1294298921417565, + "grad_norm": 3.2128524780273438, + "learning_rate": 1.6406547392462889e-06, + "loss": 1.3123, + "step": 2073 + }, + { + "epoch": 2.130457113507961, + "grad_norm": 3.060520887374878, + "learning_rate": 1.638751427483822e-06, + "loss": 1.3069, + "step": 2074 + }, + { + "epoch": 2.131484334874165, + "grad_norm": 3.092120885848999, + "learning_rate": 1.6368481157213554e-06, + "loss": 1.3537, + "step": 2075 + }, + { + "epoch": 2.13251155624037, + "grad_norm": 3.2490074634552, + "learning_rate": 1.6349448039588885e-06, + "loss": 1.3324, + "step": 2076 + }, + { + "epoch": 2.1335387776065744, + "grad_norm": 3.2517645359039307, + "learning_rate": 1.6330414921964219e-06, + "loss": 1.385, + "step": 2077 + }, + { + "epoch": 2.1345659989727785, + "grad_norm": 3.1669421195983887, + "learning_rate": 1.6311381804339553e-06, + "loss": 1.3054, + "step": 2078 + }, + { + "epoch": 2.135593220338983, + "grad_norm": 3.2323193550109863, + "learning_rate": 1.6292348686714884e-06, + "loss": 1.3488, + "step": 2079 + }, + { + "epoch": 2.1366204417051873, + "grad_norm": 3.1058294773101807, + "learning_rate": 1.6273315569090218e-06, + "loss": 1.3256, + "step": 2080 + }, + { + "epoch": 2.1366204417051873, + "eval_loss": 1.8446909189224243, + "eval_runtime": 37.1231, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.137647663071392, + "grad_norm": 3.1662075519561768, + "learning_rate": 1.625428245146555e-06, + "loss": 1.4073, + "step": 2081 + }, + { + "epoch": 2.1386748844375965, + "grad_norm": 3.083482503890991, + "learning_rate": 1.6235249333840885e-06, + "loss": 1.3897, + "step": 2082 + }, + { + "epoch": 2.1397021058038006, + "grad_norm": 3.040388345718384, + "learning_rate": 1.6216216216216219e-06, + "loss": 1.349, + "step": 2083 + }, + { + "epoch": 2.140729327170005, + "grad_norm": 3.124114513397217, + "learning_rate": 1.6197183098591552e-06, + "loss": 1.3732, + "step": 2084 + }, + { + "epoch": 2.1417565485362093, + "grad_norm": 3.2662036418914795, + "learning_rate": 1.6178149980966884e-06, + "loss": 1.3522, + "step": 2085 + }, + { + "epoch": 2.142783769902414, + "grad_norm": 3.149472951889038, + "learning_rate": 1.6159116863342217e-06, + "loss": 1.3179, + "step": 2086 + }, + { + "epoch": 2.1438109912686185, + "grad_norm": 3.256852865219116, + "learning_rate": 1.6140083745717549e-06, + "loss": 1.3897, + "step": 2087 + }, + { + "epoch": 2.1448382126348227, + "grad_norm": 3.1227493286132812, + "learning_rate": 1.6121050628092882e-06, + "loss": 1.3104, + "step": 2088 + }, + { + "epoch": 2.1458654340010272, + "grad_norm": 3.3213047981262207, + "learning_rate": 1.6102017510468216e-06, + "loss": 1.3829, + "step": 2089 + }, + { + "epoch": 2.146892655367232, + "grad_norm": 3.160428047180176, + "learning_rate": 1.6082984392843548e-06, + "loss": 1.2939, + "step": 2090 + }, + { + "epoch": 2.146892655367232, + "eval_loss": 1.8534585237503052, + "eval_runtime": 37.3333, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.147919876733436, + "grad_norm": 3.224741220474243, + "learning_rate": 1.6063951275218881e-06, + "loss": 1.3418, + "step": 2091 + }, + { + "epoch": 2.1489470980996406, + "grad_norm": 3.3884503841400146, + "learning_rate": 1.6044918157594217e-06, + "loss": 1.395, + "step": 2092 + }, + { + "epoch": 2.1499743194658447, + "grad_norm": 3.1761722564697266, + "learning_rate": 1.6025885039969548e-06, + "loss": 1.3208, + "step": 2093 + }, + { + "epoch": 2.1510015408320493, + "grad_norm": 3.0476465225219727, + "learning_rate": 1.6006851922344882e-06, + "loss": 1.3465, + "step": 2094 + }, + { + "epoch": 2.152028762198254, + "grad_norm": 3.1662650108337402, + "learning_rate": 1.5987818804720216e-06, + "loss": 1.3645, + "step": 2095 + }, + { + "epoch": 2.153055983564458, + "grad_norm": 3.1860709190368652, + "learning_rate": 1.5968785687095547e-06, + "loss": 1.361, + "step": 2096 + }, + { + "epoch": 2.1540832049306626, + "grad_norm": 3.104501962661743, + "learning_rate": 1.594975256947088e-06, + "loss": 1.3135, + "step": 2097 + }, + { + "epoch": 2.1551104262968668, + "grad_norm": 3.067155599594116, + "learning_rate": 1.5930719451846212e-06, + "loss": 1.3693, + "step": 2098 + }, + { + "epoch": 2.1561376476630714, + "grad_norm": 3.195204019546509, + "learning_rate": 1.5911686334221546e-06, + "loss": 1.3143, + "step": 2099 + }, + { + "epoch": 2.157164869029276, + "grad_norm": 3.159278154373169, + "learning_rate": 1.589265321659688e-06, + "loss": 1.3607, + "step": 2100 + }, + { + "epoch": 2.157164869029276, + "eval_loss": 1.8502479791641235, + "eval_runtime": 37.4583, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.15819209039548, + "grad_norm": 2.973191261291504, + "learning_rate": 1.5873620098972211e-06, + "loss": 1.2776, + "step": 2101 + }, + { + "epoch": 2.1592193117616847, + "grad_norm": 3.0078418254852295, + "learning_rate": 1.5854586981347547e-06, + "loss": 1.3173, + "step": 2102 + }, + { + "epoch": 2.1602465331278893, + "grad_norm": 3.2115561962127686, + "learning_rate": 1.583555386372288e-06, + "loss": 1.3621, + "step": 2103 + }, + { + "epoch": 2.1612737544940934, + "grad_norm": 3.234210968017578, + "learning_rate": 1.5816520746098212e-06, + "loss": 1.374, + "step": 2104 + }, + { + "epoch": 2.162300975860298, + "grad_norm": 3.195405960083008, + "learning_rate": 1.5797487628473546e-06, + "loss": 1.3402, + "step": 2105 + }, + { + "epoch": 2.163328197226502, + "grad_norm": 3.016078233718872, + "learning_rate": 1.577845451084888e-06, + "loss": 1.3036, + "step": 2106 + }, + { + "epoch": 2.1643554185927067, + "grad_norm": 3.2147345542907715, + "learning_rate": 1.575942139322421e-06, + "loss": 1.3511, + "step": 2107 + }, + { + "epoch": 2.1653826399589113, + "grad_norm": 2.980320692062378, + "learning_rate": 1.5740388275599544e-06, + "loss": 1.2635, + "step": 2108 + }, + { + "epoch": 2.1664098613251155, + "grad_norm": 3.0693414211273193, + "learning_rate": 1.5721355157974876e-06, + "loss": 1.3435, + "step": 2109 + }, + { + "epoch": 2.16743708269132, + "grad_norm": 3.1445682048797607, + "learning_rate": 1.570232204035021e-06, + "loss": 1.3294, + "step": 2110 + }, + { + "epoch": 2.16743708269132, + "eval_loss": 1.8479558229446411, + "eval_runtime": 36.9467, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.168464304057524, + "grad_norm": 3.1309633255004883, + "learning_rate": 1.5683288922725543e-06, + "loss": 1.3312, + "step": 2111 + }, + { + "epoch": 2.169491525423729, + "grad_norm": 3.0906994342803955, + "learning_rate": 1.5664255805100879e-06, + "loss": 1.342, + "step": 2112 + }, + { + "epoch": 2.1705187467899334, + "grad_norm": 3.1987171173095703, + "learning_rate": 1.564522268747621e-06, + "loss": 1.3358, + "step": 2113 + }, + { + "epoch": 2.1715459681561375, + "grad_norm": 3.221889019012451, + "learning_rate": 1.5626189569851544e-06, + "loss": 1.3484, + "step": 2114 + }, + { + "epoch": 2.172573189522342, + "grad_norm": 3.1508629322052, + "learning_rate": 1.5607156452226875e-06, + "loss": 1.2785, + "step": 2115 + }, + { + "epoch": 2.1736004108885463, + "grad_norm": 3.115274667739868, + "learning_rate": 1.558812333460221e-06, + "loss": 1.3463, + "step": 2116 + }, + { + "epoch": 2.174627632254751, + "grad_norm": 3.0454249382019043, + "learning_rate": 1.5569090216977543e-06, + "loss": 1.3534, + "step": 2117 + }, + { + "epoch": 2.1756548536209555, + "grad_norm": 3.20078182220459, + "learning_rate": 1.5550057099352874e-06, + "loss": 1.3469, + "step": 2118 + }, + { + "epoch": 2.1766820749871596, + "grad_norm": 3.091568946838379, + "learning_rate": 1.5531023981728208e-06, + "loss": 1.3547, + "step": 2119 + }, + { + "epoch": 2.177709296353364, + "grad_norm": 3.0242369174957275, + "learning_rate": 1.5511990864103542e-06, + "loss": 1.3198, + "step": 2120 + }, + { + "epoch": 2.177709296353364, + "eval_loss": 1.8469057083129883, + "eval_runtime": 37.0881, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.1787365177195683, + "grad_norm": 3.1775739192962646, + "learning_rate": 1.5492957746478873e-06, + "loss": 1.3491, + "step": 2121 + }, + { + "epoch": 2.179763739085773, + "grad_norm": 3.239668846130371, + "learning_rate": 1.5473924628854209e-06, + "loss": 1.3641, + "step": 2122 + }, + { + "epoch": 2.1807909604519775, + "grad_norm": 3.194678544998169, + "learning_rate": 1.5454891511229542e-06, + "loss": 1.3317, + "step": 2123 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.129948377609253, + "learning_rate": 1.5435858393604874e-06, + "loss": 1.3405, + "step": 2124 + }, + { + "epoch": 2.1828454031843862, + "grad_norm": 3.2040436267852783, + "learning_rate": 1.5416825275980208e-06, + "loss": 1.3596, + "step": 2125 + }, + { + "epoch": 2.183872624550591, + "grad_norm": 3.2529456615448, + "learning_rate": 1.539779215835554e-06, + "loss": 1.3171, + "step": 2126 + }, + { + "epoch": 2.184899845916795, + "grad_norm": 3.2071499824523926, + "learning_rate": 1.5378759040730873e-06, + "loss": 1.318, + "step": 2127 + }, + { + "epoch": 2.1859270672829996, + "grad_norm": 3.2709555625915527, + "learning_rate": 1.5359725923106206e-06, + "loss": 1.3811, + "step": 2128 + }, + { + "epoch": 2.1869542886492037, + "grad_norm": 3.244680643081665, + "learning_rate": 1.5340692805481538e-06, + "loss": 1.3539, + "step": 2129 + }, + { + "epoch": 2.1879815100154083, + "grad_norm": 3.068683624267578, + "learning_rate": 1.5321659687856871e-06, + "loss": 1.3207, + "step": 2130 + }, + { + "epoch": 2.1879815100154083, + "eval_loss": 1.8478527069091797, + "eval_runtime": 37.138, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.189008731381613, + "grad_norm": 3.1829938888549805, + "learning_rate": 1.5302626570232205e-06, + "loss": 1.3127, + "step": 2131 + }, + { + "epoch": 2.190035952747817, + "grad_norm": 3.060347080230713, + "learning_rate": 1.5283593452607539e-06, + "loss": 1.287, + "step": 2132 + }, + { + "epoch": 2.1910631741140216, + "grad_norm": 3.3016905784606934, + "learning_rate": 1.5264560334982872e-06, + "loss": 1.3838, + "step": 2133 + }, + { + "epoch": 2.1920903954802258, + "grad_norm": 3.1945807933807373, + "learning_rate": 1.5245527217358206e-06, + "loss": 1.3311, + "step": 2134 + }, + { + "epoch": 2.1931176168464304, + "grad_norm": 3.182347059249878, + "learning_rate": 1.5226494099733537e-06, + "loss": 1.3544, + "step": 2135 + }, + { + "epoch": 2.194144838212635, + "grad_norm": 3.069206476211548, + "learning_rate": 1.520746098210887e-06, + "loss": 1.3148, + "step": 2136 + }, + { + "epoch": 2.195172059578839, + "grad_norm": 3.1632168292999268, + "learning_rate": 1.5188427864484205e-06, + "loss": 1.3857, + "step": 2137 + }, + { + "epoch": 2.1961992809450437, + "grad_norm": 3.055793523788452, + "learning_rate": 1.5169394746859536e-06, + "loss": 1.3116, + "step": 2138 + }, + { + "epoch": 2.1972265023112483, + "grad_norm": 3.1330678462982178, + "learning_rate": 1.515036162923487e-06, + "loss": 1.3214, + "step": 2139 + }, + { + "epoch": 2.1982537236774524, + "grad_norm": 3.118450880050659, + "learning_rate": 1.5131328511610201e-06, + "loss": 1.3479, + "step": 2140 + }, + { + "epoch": 2.1982537236774524, + "eval_loss": 1.8538312911987305, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.199280945043657, + "grad_norm": 3.249545097351074, + "learning_rate": 1.5112295393985535e-06, + "loss": 1.379, + "step": 2141 + }, + { + "epoch": 2.200308166409861, + "grad_norm": 3.1331286430358887, + "learning_rate": 1.509326227636087e-06, + "loss": 1.3621, + "step": 2142 + }, + { + "epoch": 2.2013353877760657, + "grad_norm": 3.150987148284912, + "learning_rate": 1.5074229158736204e-06, + "loss": 1.291, + "step": 2143 + }, + { + "epoch": 2.2023626091422703, + "grad_norm": 3.1020572185516357, + "learning_rate": 1.5055196041111536e-06, + "loss": 1.3381, + "step": 2144 + }, + { + "epoch": 2.2033898305084745, + "grad_norm": 3.129390239715576, + "learning_rate": 1.503616292348687e-06, + "loss": 1.3405, + "step": 2145 + }, + { + "epoch": 2.204417051874679, + "grad_norm": 3.2193870544433594, + "learning_rate": 1.50171298058622e-06, + "loss": 1.3744, + "step": 2146 + }, + { + "epoch": 2.205444273240883, + "grad_norm": 3.043828248977661, + "learning_rate": 1.4998096688237535e-06, + "loss": 1.2782, + "step": 2147 + }, + { + "epoch": 2.206471494607088, + "grad_norm": 3.2130300998687744, + "learning_rate": 1.4979063570612868e-06, + "loss": 1.3625, + "step": 2148 + }, + { + "epoch": 2.2074987159732924, + "grad_norm": 3.250370979309082, + "learning_rate": 1.49600304529882e-06, + "loss": 1.3396, + "step": 2149 + }, + { + "epoch": 2.2085259373394965, + "grad_norm": 3.0954599380493164, + "learning_rate": 1.4940997335363533e-06, + "loss": 1.3187, + "step": 2150 + }, + { + "epoch": 2.2085259373394965, + "eval_loss": 1.8469802141189575, + "eval_runtime": 37.2109, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 2150 + }, + { + "epoch": 2.209553158705701, + "grad_norm": 3.153545379638672, + "learning_rate": 1.4921964217738865e-06, + "loss": 1.3488, + "step": 2151 + }, + { + "epoch": 2.2105803800719057, + "grad_norm": 3.1206071376800537, + "learning_rate": 1.49029311001142e-06, + "loss": 1.336, + "step": 2152 + }, + { + "epoch": 2.21160760143811, + "grad_norm": 3.151784896850586, + "learning_rate": 1.4883897982489534e-06, + "loss": 1.3613, + "step": 2153 + }, + { + "epoch": 2.2126348228043144, + "grad_norm": 3.18965744972229, + "learning_rate": 1.4864864864864868e-06, + "loss": 1.3698, + "step": 2154 + }, + { + "epoch": 2.2136620441705186, + "grad_norm": 3.108281135559082, + "learning_rate": 1.48458317472402e-06, + "loss": 1.3287, + "step": 2155 + }, + { + "epoch": 2.214689265536723, + "grad_norm": 3.2309465408325195, + "learning_rate": 1.4826798629615533e-06, + "loss": 1.3425, + "step": 2156 + }, + { + "epoch": 2.2157164869029278, + "grad_norm": 3.0829577445983887, + "learning_rate": 1.4807765511990864e-06, + "loss": 1.2917, + "step": 2157 + }, + { + "epoch": 2.216743708269132, + "grad_norm": 3.1417033672332764, + "learning_rate": 1.4788732394366198e-06, + "loss": 1.3113, + "step": 2158 + }, + { + "epoch": 2.2177709296353365, + "grad_norm": 3.1996822357177734, + "learning_rate": 1.4769699276741532e-06, + "loss": 1.3248, + "step": 2159 + }, + { + "epoch": 2.2187981510015407, + "grad_norm": 3.262814998626709, + "learning_rate": 1.4750666159116863e-06, + "loss": 1.3186, + "step": 2160 + }, + { + "epoch": 2.2187981510015407, + "eval_loss": 1.844687819480896, + "eval_runtime": 37.2211, + "eval_samples_per_second": 26.866, + "eval_steps_per_second": 1.128, + "step": 2160 + }, + { + "epoch": 2.2198253723677452, + "grad_norm": 3.0836942195892334, + "learning_rate": 1.4731633041492197e-06, + "loss": 1.2935, + "step": 2161 + }, + { + "epoch": 2.22085259373395, + "grad_norm": 3.1720023155212402, + "learning_rate": 1.4712599923867528e-06, + "loss": 1.3622, + "step": 2162 + }, + { + "epoch": 2.221879815100154, + "grad_norm": 3.2829599380493164, + "learning_rate": 1.4693566806242864e-06, + "loss": 1.3794, + "step": 2163 + }, + { + "epoch": 2.2229070364663586, + "grad_norm": 3.045949935913086, + "learning_rate": 1.4674533688618198e-06, + "loss": 1.3054, + "step": 2164 + }, + { + "epoch": 2.2239342578325627, + "grad_norm": 3.1561572551727295, + "learning_rate": 1.4655500570993531e-06, + "loss": 1.3289, + "step": 2165 + }, + { + "epoch": 2.2249614791987673, + "grad_norm": 3.0877439975738525, + "learning_rate": 1.4636467453368863e-06, + "loss": 1.329, + "step": 2166 + }, + { + "epoch": 2.225988700564972, + "grad_norm": 3.1330180168151855, + "learning_rate": 1.4617434335744197e-06, + "loss": 1.3853, + "step": 2167 + }, + { + "epoch": 2.227015921931176, + "grad_norm": 3.231693744659424, + "learning_rate": 1.4598401218119528e-06, + "loss": 1.378, + "step": 2168 + }, + { + "epoch": 2.2280431432973806, + "grad_norm": 3.202930212020874, + "learning_rate": 1.4579368100494862e-06, + "loss": 1.3549, + "step": 2169 + }, + { + "epoch": 2.2290703646635848, + "grad_norm": 3.164461135864258, + "learning_rate": 1.4560334982870195e-06, + "loss": 1.3365, + "step": 2170 + }, + { + "epoch": 2.2290703646635848, + "eval_loss": 1.8483937978744507, + "eval_runtime": 37.1839, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.2300975860297894, + "grad_norm": 3.1866867542266846, + "learning_rate": 1.4541301865245527e-06, + "loss": 1.3503, + "step": 2171 + }, + { + "epoch": 2.231124807395994, + "grad_norm": 3.2202508449554443, + "learning_rate": 1.452226874762086e-06, + "loss": 1.354, + "step": 2172 + }, + { + "epoch": 2.232152028762198, + "grad_norm": 3.190744161605835, + "learning_rate": 1.4503235629996196e-06, + "loss": 1.3073, + "step": 2173 + }, + { + "epoch": 2.2331792501284027, + "grad_norm": 3.2158405780792236, + "learning_rate": 1.4484202512371528e-06, + "loss": 1.3731, + "step": 2174 + }, + { + "epoch": 2.2342064714946073, + "grad_norm": 3.0294954776763916, + "learning_rate": 1.4465169394746861e-06, + "loss": 1.2772, + "step": 2175 + }, + { + "epoch": 2.2352336928608114, + "grad_norm": 3.2871217727661133, + "learning_rate": 1.4446136277122195e-06, + "loss": 1.3888, + "step": 2176 + }, + { + "epoch": 2.236260914227016, + "grad_norm": 3.0494141578674316, + "learning_rate": 1.4427103159497526e-06, + "loss": 1.2991, + "step": 2177 + }, + { + "epoch": 2.23728813559322, + "grad_norm": 3.122699022293091, + "learning_rate": 1.440807004187286e-06, + "loss": 1.3186, + "step": 2178 + }, + { + "epoch": 2.2383153569594247, + "grad_norm": 3.119854211807251, + "learning_rate": 1.4389036924248192e-06, + "loss": 1.3421, + "step": 2179 + }, + { + "epoch": 2.2393425783256293, + "grad_norm": 3.2588119506835938, + "learning_rate": 1.4370003806623525e-06, + "loss": 1.3657, + "step": 2180 + }, + { + "epoch": 2.2393425783256293, + "eval_loss": 1.8473975658416748, + "eval_runtime": 37.1561, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.2403697996918335, + "grad_norm": 3.138901948928833, + "learning_rate": 1.4350970688998859e-06, + "loss": 1.3273, + "step": 2181 + }, + { + "epoch": 2.241397021058038, + "grad_norm": 3.2792508602142334, + "learning_rate": 1.433193757137419e-06, + "loss": 1.3763, + "step": 2182 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.065627098083496, + "learning_rate": 1.4312904453749526e-06, + "loss": 1.3292, + "step": 2183 + }, + { + "epoch": 2.243451463790447, + "grad_norm": 3.101377010345459, + "learning_rate": 1.429387133612486e-06, + "loss": 1.2957, + "step": 2184 + }, + { + "epoch": 2.2444786851566514, + "grad_norm": 3.140158176422119, + "learning_rate": 1.4274838218500191e-06, + "loss": 1.3406, + "step": 2185 + }, + { + "epoch": 2.2455059065228555, + "grad_norm": 3.2498116493225098, + "learning_rate": 1.4255805100875525e-06, + "loss": 1.3499, + "step": 2186 + }, + { + "epoch": 2.24653312788906, + "grad_norm": 3.2681965827941895, + "learning_rate": 1.4236771983250858e-06, + "loss": 1.407, + "step": 2187 + }, + { + "epoch": 2.2475603492552647, + "grad_norm": 3.2457778453826904, + "learning_rate": 1.421773886562619e-06, + "loss": 1.3278, + "step": 2188 + }, + { + "epoch": 2.248587570621469, + "grad_norm": 3.099942207336426, + "learning_rate": 1.4198705748001524e-06, + "loss": 1.3114, + "step": 2189 + }, + { + "epoch": 2.2496147919876734, + "grad_norm": 3.090271234512329, + "learning_rate": 1.4179672630376855e-06, + "loss": 1.3285, + "step": 2190 + }, + { + "epoch": 2.2496147919876734, + "eval_loss": 1.8479118347167969, + "eval_runtime": 37.2331, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.2506420133538776, + "grad_norm": 3.0344860553741455, + "learning_rate": 1.4160639512752189e-06, + "loss": 1.3104, + "step": 2191 + }, + { + "epoch": 2.251669234720082, + "grad_norm": 3.087860584259033, + "learning_rate": 1.4141606395127522e-06, + "loss": 1.3167, + "step": 2192 + }, + { + "epoch": 2.2526964560862868, + "grad_norm": 3.0974831581115723, + "learning_rate": 1.4122573277502858e-06, + "loss": 1.2926, + "step": 2193 + }, + { + "epoch": 2.253723677452491, + "grad_norm": 3.228492498397827, + "learning_rate": 1.410354015987819e-06, + "loss": 1.3541, + "step": 2194 + }, + { + "epoch": 2.2547508988186955, + "grad_norm": 3.1761956214904785, + "learning_rate": 1.4084507042253523e-06, + "loss": 1.3342, + "step": 2195 + }, + { + "epoch": 2.2557781201848996, + "grad_norm": 3.1890695095062256, + "learning_rate": 1.4065473924628855e-06, + "loss": 1.362, + "step": 2196 + }, + { + "epoch": 2.2568053415511042, + "grad_norm": 3.1543149948120117, + "learning_rate": 1.4046440807004188e-06, + "loss": 1.329, + "step": 2197 + }, + { + "epoch": 2.257832562917309, + "grad_norm": 3.2313342094421387, + "learning_rate": 1.4027407689379522e-06, + "loss": 1.3256, + "step": 2198 + }, + { + "epoch": 2.258859784283513, + "grad_norm": 3.130805253982544, + "learning_rate": 1.4008374571754854e-06, + "loss": 1.3047, + "step": 2199 + }, + { + "epoch": 2.2598870056497176, + "grad_norm": 3.0488216876983643, + "learning_rate": 1.3989341454130187e-06, + "loss": 1.3461, + "step": 2200 + }, + { + "epoch": 2.2598870056497176, + "eval_loss": 1.8517944812774658, + "eval_runtime": 37.4528, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 2200 + }, + { + "epoch": 2.260914227015922, + "grad_norm": 3.0915918350219727, + "learning_rate": 1.397030833650552e-06, + "loss": 1.3391, + "step": 2201 + }, + { + "epoch": 2.2619414483821263, + "grad_norm": 3.156053066253662, + "learning_rate": 1.3951275218880852e-06, + "loss": 1.3525, + "step": 2202 + }, + { + "epoch": 2.262968669748331, + "grad_norm": 3.4035868644714355, + "learning_rate": 1.3932242101256188e-06, + "loss": 1.4122, + "step": 2203 + }, + { + "epoch": 2.263995891114535, + "grad_norm": 2.9787867069244385, + "learning_rate": 1.3913208983631522e-06, + "loss": 1.3096, + "step": 2204 + }, + { + "epoch": 2.2650231124807396, + "grad_norm": 3.154271364212036, + "learning_rate": 1.3894175866006853e-06, + "loss": 1.3421, + "step": 2205 + }, + { + "epoch": 2.266050333846944, + "grad_norm": 3.1899831295013428, + "learning_rate": 1.3875142748382187e-06, + "loss": 1.2949, + "step": 2206 + }, + { + "epoch": 2.2670775552131484, + "grad_norm": 3.126694679260254, + "learning_rate": 1.385610963075752e-06, + "loss": 1.2898, + "step": 2207 + }, + { + "epoch": 2.268104776579353, + "grad_norm": 3.149989128112793, + "learning_rate": 1.3837076513132852e-06, + "loss": 1.3369, + "step": 2208 + }, + { + "epoch": 2.269131997945557, + "grad_norm": 3.250481128692627, + "learning_rate": 1.3818043395508186e-06, + "loss": 1.3605, + "step": 2209 + }, + { + "epoch": 2.2701592193117617, + "grad_norm": 3.2549924850463867, + "learning_rate": 1.3799010277883517e-06, + "loss": 1.3428, + "step": 2210 + }, + { + "epoch": 2.2701592193117617, + "eval_loss": 1.8562498092651367, + "eval_runtime": 37.0713, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 2210 + }, + { + "epoch": 2.2711864406779663, + "grad_norm": 3.1369433403015137, + "learning_rate": 1.377997716025885e-06, + "loss": 1.333, + "step": 2211 + }, + { + "epoch": 2.2722136620441704, + "grad_norm": 3.2522025108337402, + "learning_rate": 1.3760944042634184e-06, + "loss": 1.3496, + "step": 2212 + }, + { + "epoch": 2.273240883410375, + "grad_norm": 3.225821018218994, + "learning_rate": 1.3741910925009518e-06, + "loss": 1.3716, + "step": 2213 + }, + { + "epoch": 2.2742681047765796, + "grad_norm": 3.1488454341888428, + "learning_rate": 1.3722877807384852e-06, + "loss": 1.3508, + "step": 2214 + }, + { + "epoch": 2.2752953261427837, + "grad_norm": 3.258491039276123, + "learning_rate": 1.3703844689760185e-06, + "loss": 1.3565, + "step": 2215 + }, + { + "epoch": 2.2763225475089883, + "grad_norm": 3.1634750366210938, + "learning_rate": 1.3684811572135517e-06, + "loss": 1.3269, + "step": 2216 + }, + { + "epoch": 2.2773497688751925, + "grad_norm": 3.1604928970336914, + "learning_rate": 1.366577845451085e-06, + "loss": 1.2832, + "step": 2217 + }, + { + "epoch": 2.278376990241397, + "grad_norm": 3.2551920413970947, + "learning_rate": 1.3646745336886184e-06, + "loss": 1.3567, + "step": 2218 + }, + { + "epoch": 2.279404211607601, + "grad_norm": 3.392820119857788, + "learning_rate": 1.3627712219261515e-06, + "loss": 1.3601, + "step": 2219 + }, + { + "epoch": 2.280431432973806, + "grad_norm": 3.057724952697754, + "learning_rate": 1.360867910163685e-06, + "loss": 1.3138, + "step": 2220 + }, + { + "epoch": 2.280431432973806, + "eval_loss": 1.848382830619812, + "eval_runtime": 37.2255, + "eval_samples_per_second": 26.863, + "eval_steps_per_second": 1.128, + "step": 2220 + }, + { + "epoch": 2.2814586543400104, + "grad_norm": 3.2199649810791016, + "learning_rate": 1.358964598401218e-06, + "loss": 1.3338, + "step": 2221 + }, + { + "epoch": 2.2824858757062145, + "grad_norm": 3.252066135406494, + "learning_rate": 1.3570612866387514e-06, + "loss": 1.3413, + "step": 2222 + }, + { + "epoch": 2.283513097072419, + "grad_norm": 3.038052558898926, + "learning_rate": 1.355157974876285e-06, + "loss": 1.3268, + "step": 2223 + }, + { + "epoch": 2.2845403184386237, + "grad_norm": 3.2332992553710938, + "learning_rate": 1.3532546631138184e-06, + "loss": 1.3866, + "step": 2224 + }, + { + "epoch": 2.285567539804828, + "grad_norm": 3.14321231842041, + "learning_rate": 1.3513513513513515e-06, + "loss": 1.3466, + "step": 2225 + }, + { + "epoch": 2.2865947611710324, + "grad_norm": 3.0714094638824463, + "learning_rate": 1.3494480395888849e-06, + "loss": 1.331, + "step": 2226 + }, + { + "epoch": 2.2876219825372366, + "grad_norm": 3.1167569160461426, + "learning_rate": 1.347544727826418e-06, + "loss": 1.3242, + "step": 2227 + }, + { + "epoch": 2.288649203903441, + "grad_norm": 3.072951316833496, + "learning_rate": 1.3456414160639514e-06, + "loss": 1.3139, + "step": 2228 + }, + { + "epoch": 2.2896764252696458, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.3524, + "step": 2229 + }, + { + "epoch": 2.29070364663585, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.334, + "step": 2230 + }, + { + "epoch": 2.29070364663585, + "eval_loss": 1.848712682723999, + "eval_runtime": 37.2687, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2230 + }, + { + "epoch": 2.2917308680020545, + "grad_norm": 3.056426525115967, + "learning_rate": 1.341834792539018e-06, + "loss": 1.3344, + "step": 2231 + }, + { + "epoch": 2.2927580893682586, + "grad_norm": 3.2412500381469727, + "learning_rate": 1.3399314807765513e-06, + "loss": 1.3416, + "step": 2232 + }, + { + "epoch": 2.2937853107344632, + "grad_norm": 3.178572416305542, + "learning_rate": 1.3380281690140844e-06, + "loss": 1.3472, + "step": 2233 + }, + { + "epoch": 2.294812532100668, + "grad_norm": 3.2567203044891357, + "learning_rate": 1.336124857251618e-06, + "loss": 1.311, + "step": 2234 + }, + { + "epoch": 2.295839753466872, + "grad_norm": 3.2018561363220215, + "learning_rate": 1.3342215454891513e-06, + "loss": 1.301, + "step": 2235 + }, + { + "epoch": 2.2968669748330766, + "grad_norm": 3.0722200870513916, + "learning_rate": 1.3323182337266847e-06, + "loss": 1.3325, + "step": 2236 + }, + { + "epoch": 2.297894196199281, + "grad_norm": 3.2170779705047607, + "learning_rate": 1.3304149219642179e-06, + "loss": 1.3752, + "step": 2237 + }, + { + "epoch": 2.2989214175654853, + "grad_norm": 3.1309115886688232, + "learning_rate": 1.3285116102017512e-06, + "loss": 1.3261, + "step": 2238 + }, + { + "epoch": 2.29994863893169, + "grad_norm": 3.1463022232055664, + "learning_rate": 1.3266082984392844e-06, + "loss": 1.3316, + "step": 2239 + }, + { + "epoch": 2.300975860297894, + "grad_norm": 3.174701690673828, + "learning_rate": 1.3247049866768177e-06, + "loss": 1.3148, + "step": 2240 + }, + { + "epoch": 2.300975860297894, + "eval_loss": 1.8436999320983887, + "eval_runtime": 37.2577, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 2240 + }, + { + "epoch": 2.3020030816640986, + "grad_norm": 3.18149471282959, + "learning_rate": 1.322801674914351e-06, + "loss": 1.312, + "step": 2241 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.213361978530884, + "learning_rate": 1.3208983631518843e-06, + "loss": 1.3406, + "step": 2242 + }, + { + "epoch": 2.3040575243965074, + "grad_norm": 3.128354787826538, + "learning_rate": 1.3189950513894176e-06, + "loss": 1.3286, + "step": 2243 + }, + { + "epoch": 2.305084745762712, + "grad_norm": 3.210330009460449, + "learning_rate": 1.3170917396269512e-06, + "loss": 1.3406, + "step": 2244 + }, + { + "epoch": 2.306111967128916, + "grad_norm": 3.187161445617676, + "learning_rate": 1.3151884278644843e-06, + "loss": 1.3621, + "step": 2245 + }, + { + "epoch": 2.3071391884951207, + "grad_norm": 3.108806848526001, + "learning_rate": 1.3132851161020177e-06, + "loss": 1.3094, + "step": 2246 + }, + { + "epoch": 2.3081664098613253, + "grad_norm": 3.336510181427002, + "learning_rate": 1.311381804339551e-06, + "loss": 1.3998, + "step": 2247 + }, + { + "epoch": 2.3091936312275294, + "grad_norm": 3.159184455871582, + "learning_rate": 1.3094784925770842e-06, + "loss": 1.3082, + "step": 2248 + }, + { + "epoch": 2.310220852593734, + "grad_norm": 3.128321409225464, + "learning_rate": 1.3075751808146176e-06, + "loss": 1.3494, + "step": 2249 + }, + { + "epoch": 2.3112480739599386, + "grad_norm": 3.122690439224243, + "learning_rate": 1.3056718690521507e-06, + "loss": 1.3064, + "step": 2250 + }, + { + "epoch": 2.3112480739599386, + "eval_loss": 1.848455786705017, + "eval_runtime": 37.2647, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 2250 + }, + { + "epoch": 2.3122752953261427, + "grad_norm": 3.1642186641693115, + "learning_rate": 1.303768557289684e-06, + "loss": 1.3134, + "step": 2251 + }, + { + "epoch": 2.3133025166923473, + "grad_norm": 3.1710383892059326, + "learning_rate": 1.3018652455272175e-06, + "loss": 1.3126, + "step": 2252 + }, + { + "epoch": 2.3143297380585515, + "grad_norm": 3.1513214111328125, + "learning_rate": 1.2999619337647506e-06, + "loss": 1.3218, + "step": 2253 + }, + { + "epoch": 2.315356959424756, + "grad_norm": 3.1538784503936768, + "learning_rate": 1.2980586220022842e-06, + "loss": 1.3473, + "step": 2254 + }, + { + "epoch": 2.3163841807909606, + "grad_norm": 3.2409846782684326, + "learning_rate": 1.2961553102398175e-06, + "loss": 1.371, + "step": 2255 + }, + { + "epoch": 2.317411402157165, + "grad_norm": 3.3224987983703613, + "learning_rate": 1.2942519984773507e-06, + "loss": 1.3506, + "step": 2256 + }, + { + "epoch": 2.3184386235233694, + "grad_norm": 3.1684653759002686, + "learning_rate": 1.292348686714884e-06, + "loss": 1.3127, + "step": 2257 + }, + { + "epoch": 2.3194658448895735, + "grad_norm": 3.1784262657165527, + "learning_rate": 1.2904453749524174e-06, + "loss": 1.3541, + "step": 2258 + }, + { + "epoch": 2.320493066255778, + "grad_norm": 3.1121561527252197, + "learning_rate": 1.2885420631899506e-06, + "loss": 1.2843, + "step": 2259 + }, + { + "epoch": 2.3215202876219827, + "grad_norm": 3.200490951538086, + "learning_rate": 1.286638751427484e-06, + "loss": 1.3191, + "step": 2260 + }, + { + "epoch": 2.3215202876219827, + "eval_loss": 1.8469436168670654, + "eval_runtime": 37.2695, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2260 + }, + { + "epoch": 2.322547508988187, + "grad_norm": 3.143329620361328, + "learning_rate": 1.284735439665017e-06, + "loss": 1.3454, + "step": 2261 + }, + { + "epoch": 2.3235747303543914, + "grad_norm": 3.1955461502075195, + "learning_rate": 1.2828321279025504e-06, + "loss": 1.3505, + "step": 2262 + }, + { + "epoch": 2.324601951720596, + "grad_norm": 3.3297348022460938, + "learning_rate": 1.2809288161400838e-06, + "loss": 1.3374, + "step": 2263 + }, + { + "epoch": 2.3256291730868, + "grad_norm": 3.144385814666748, + "learning_rate": 1.279025504377617e-06, + "loss": 1.3198, + "step": 2264 + }, + { + "epoch": 2.3266563944530048, + "grad_norm": 3.2952210903167725, + "learning_rate": 1.2771221926151505e-06, + "loss": 1.3744, + "step": 2265 + }, + { + "epoch": 2.327683615819209, + "grad_norm": 3.212287187576294, + "learning_rate": 1.275218880852684e-06, + "loss": 1.3173, + "step": 2266 + }, + { + "epoch": 2.3287108371854135, + "grad_norm": 3.235272169113159, + "learning_rate": 1.273315569090217e-06, + "loss": 1.3231, + "step": 2267 + }, + { + "epoch": 2.3297380585516176, + "grad_norm": 3.2961113452911377, + "learning_rate": 1.2714122573277504e-06, + "loss": 1.3909, + "step": 2268 + }, + { + "epoch": 2.3307652799178222, + "grad_norm": 3.13678240776062, + "learning_rate": 1.2695089455652838e-06, + "loss": 1.3911, + "step": 2269 + }, + { + "epoch": 2.331792501284027, + "grad_norm": 3.166139841079712, + "learning_rate": 1.267605633802817e-06, + "loss": 1.3496, + "step": 2270 + }, + { + "epoch": 2.331792501284027, + "eval_loss": 1.8424044847488403, + "eval_runtime": 37.128, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 2270 + }, + { + "epoch": 2.332819722650231, + "grad_norm": 3.20859432220459, + "learning_rate": 1.2657023220403503e-06, + "loss": 1.3402, + "step": 2271 + }, + { + "epoch": 2.3338469440164356, + "grad_norm": 3.18941593170166, + "learning_rate": 1.2637990102778836e-06, + "loss": 1.3272, + "step": 2272 + }, + { + "epoch": 2.33487416538264, + "grad_norm": 3.2382333278656006, + "learning_rate": 1.2618956985154168e-06, + "loss": 1.3242, + "step": 2273 + }, + { + "epoch": 2.3359013867488443, + "grad_norm": 3.2496516704559326, + "learning_rate": 1.2599923867529502e-06, + "loss": 1.3675, + "step": 2274 + }, + { + "epoch": 2.336928608115049, + "grad_norm": 3.1282081604003906, + "learning_rate": 1.2580890749904837e-06, + "loss": 1.3343, + "step": 2275 + }, + { + "epoch": 2.337955829481253, + "grad_norm": 3.34047794342041, + "learning_rate": 1.2561857632280169e-06, + "loss": 1.3651, + "step": 2276 + }, + { + "epoch": 2.3389830508474576, + "grad_norm": 3.3369057178497314, + "learning_rate": 1.2542824514655502e-06, + "loss": 1.3944, + "step": 2277 + }, + { + "epoch": 2.340010272213662, + "grad_norm": 3.282226085662842, + "learning_rate": 1.2523791397030834e-06, + "loss": 1.3595, + "step": 2278 + }, + { + "epoch": 2.3410374935798663, + "grad_norm": 3.111595630645752, + "learning_rate": 1.2504758279406168e-06, + "loss": 1.3325, + "step": 2279 + }, + { + "epoch": 2.342064714946071, + "grad_norm": 3.306988477706909, + "learning_rate": 1.2485725161781501e-06, + "loss": 1.3595, + "step": 2280 + }, + { + "epoch": 2.342064714946071, + "eval_loss": 1.8465285301208496, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 2280 + }, + { + "epoch": 2.343091936312275, + "grad_norm": 3.4089088439941406, + "learning_rate": 1.2466692044156833e-06, + "loss": 1.4337, + "step": 2281 + }, + { + "epoch": 2.3441191576784797, + "grad_norm": 3.1354665756225586, + "learning_rate": 1.2447658926532168e-06, + "loss": 1.2996, + "step": 2282 + }, + { + "epoch": 2.3451463790446843, + "grad_norm": 3.067383289337158, + "learning_rate": 1.24286258089075e-06, + "loss": 1.3446, + "step": 2283 + }, + { + "epoch": 2.3461736004108884, + "grad_norm": 3.162349224090576, + "learning_rate": 1.2409592691282834e-06, + "loss": 1.3068, + "step": 2284 + }, + { + "epoch": 2.347200821777093, + "grad_norm": 3.2846035957336426, + "learning_rate": 1.2390559573658165e-06, + "loss": 1.3496, + "step": 2285 + }, + { + "epoch": 2.3482280431432976, + "grad_norm": 3.290997266769409, + "learning_rate": 1.2371526456033499e-06, + "loss": 1.3471, + "step": 2286 + }, + { + "epoch": 2.3492552645095017, + "grad_norm": 3.1327056884765625, + "learning_rate": 1.2352493338408832e-06, + "loss": 1.3108, + "step": 2287 + }, + { + "epoch": 2.3502824858757063, + "grad_norm": 3.295562744140625, + "learning_rate": 1.2333460220784166e-06, + "loss": 1.4226, + "step": 2288 + }, + { + "epoch": 2.3513097072419105, + "grad_norm": 3.3461313247680664, + "learning_rate": 1.23144271031595e-06, + "loss": 1.3494, + "step": 2289 + }, + { + "epoch": 2.352336928608115, + "grad_norm": 3.4178872108459473, + "learning_rate": 1.2295393985534831e-06, + "loss": 1.3876, + "step": 2290 + }, + { + "epoch": 2.352336928608115, + "eval_loss": 1.855657935142517, + "eval_runtime": 37.2052, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 2290 + }, + { + "epoch": 2.3533641499743196, + "grad_norm": 3.0909383296966553, + "learning_rate": 1.2276360867910165e-06, + "loss": 1.2625, + "step": 2291 + }, + { + "epoch": 2.354391371340524, + "grad_norm": 3.2940239906311035, + "learning_rate": 1.2257327750285498e-06, + "loss": 1.3836, + "step": 2292 + }, + { + "epoch": 2.3554185927067284, + "grad_norm": 3.166198253631592, + "learning_rate": 1.2238294632660832e-06, + "loss": 1.3014, + "step": 2293 + }, + { + "epoch": 2.3564458140729325, + "grad_norm": 3.1372265815734863, + "learning_rate": 1.2219261515036164e-06, + "loss": 1.3133, + "step": 2294 + }, + { + "epoch": 2.357473035439137, + "grad_norm": 3.2904369831085205, + "learning_rate": 1.2200228397411497e-06, + "loss": 1.3578, + "step": 2295 + }, + { + "epoch": 2.3585002568053417, + "grad_norm": 3.1268763542175293, + "learning_rate": 1.2181195279786829e-06, + "loss": 1.391, + "step": 2296 + }, + { + "epoch": 2.359527478171546, + "grad_norm": 3.1357498168945312, + "learning_rate": 1.2162162162162164e-06, + "loss": 1.327, + "step": 2297 + }, + { + "epoch": 2.3605546995377504, + "grad_norm": 3.1766469478607178, + "learning_rate": 1.2143129044537496e-06, + "loss": 1.3146, + "step": 2298 + }, + { + "epoch": 2.361581920903955, + "grad_norm": 3.1096513271331787, + "learning_rate": 1.212409592691283e-06, + "loss": 1.3235, + "step": 2299 + }, + { + "epoch": 2.362609142270159, + "grad_norm": 3.161182403564453, + "learning_rate": 1.2105062809288163e-06, + "loss": 1.3411, + "step": 2300 + }, + { + "epoch": 2.362609142270159, + "eval_loss": 1.852128267288208, + "eval_runtime": 37.4484, + "eval_samples_per_second": 26.703, + "eval_steps_per_second": 1.122, + "step": 2300 + } + ], + "logging_steps": 1, + "max_steps": 2919, + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/latest b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/latest new file mode 100644 index 0000000000000000000000000000000000000000..98f8bed9a5485ee900d9931cc06950de69499848 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/latest @@ -0,0 +1 @@ +global_step2500 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..0b782d4602b361f7577abb52b710a8c0e088588d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/tokenizer_config.json @@ -0,0 +1,2078 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..106fa922289b08177880b8127da096c96da7e77f --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/trainer_state.json @@ -0,0 +1,19533 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.5680534155110424, + "eval_steps": 10, + "global_step": 2500, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.001027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7522, + "step": 1 + }, + { + "epoch": 0.002054442732408834, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7541, + "step": 2 + }, + { + "epoch": 0.0030816640986132513, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8031, + "step": 3 + }, + { + "epoch": 0.004108885464817668, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8062, + "step": 4 + }, + { + "epoch": 0.005136106831022085, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7733, + "step": 5 + }, + { + "epoch": 0.0061633281972265025, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7949, + "step": 6 + }, + { + "epoch": 0.007190549563430919, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7657, + "step": 7 + }, + { + "epoch": 0.008217770929635337, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.803, + "step": 8 + }, + { + "epoch": 0.009244992295839754, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7793, + "step": 9 + }, + { + "epoch": 0.01027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7935, + "step": 10 + }, + { + "epoch": 0.01027221366204417, + "eval_loss": 2.8114004135131836, + "eval_runtime": 37.8463, + "eval_samples_per_second": 26.423, + "eval_steps_per_second": 1.11, + "step": 10 + }, + { + "epoch": 0.011299435028248588, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7546, + "step": 11 + }, + { + "epoch": 0.012326656394453005, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7423, + "step": 12 + }, + { + "epoch": 0.01335387776065742, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8057, + "step": 13 + }, + { + "epoch": 0.014381099126861838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7663, + "step": 14 + }, + { + "epoch": 0.015408320493066256, + "grad_norm": 4.694424152374268, + "learning_rate": 1.7123287671232876e-08, + "loss": 2.7715, + "step": 15 + }, + { + "epoch": 0.016435541859270673, + "grad_norm": 4.905329704284668, + "learning_rate": 3.424657534246575e-08, + "loss": 2.751, + "step": 16 + }, + { + "epoch": 0.01746276322547509, + "grad_norm": 4.705279350280762, + "learning_rate": 5.136986301369863e-08, + "loss": 2.7186, + "step": 17 + }, + { + "epoch": 0.01848998459167951, + "grad_norm": 5.154810905456543, + "learning_rate": 6.84931506849315e-08, + "loss": 2.8073, + "step": 18 + }, + { + "epoch": 0.019517205957883924, + "grad_norm": 5.070403099060059, + "learning_rate": 8.561643835616439e-08, + "loss": 2.7237, + "step": 19 + }, + { + "epoch": 0.02054442732408834, + "grad_norm": 4.901423454284668, + "learning_rate": 1.0273972602739726e-07, + "loss": 2.7882, + "step": 20 + }, + { + "epoch": 0.02054442732408834, + "eval_loss": 2.810307502746582, + "eval_runtime": 37.65, + "eval_samples_per_second": 26.56, + "eval_steps_per_second": 1.116, + "step": 20 + }, + { + "epoch": 0.02157164869029276, + "grad_norm": 4.914156436920166, + "learning_rate": 1.1986301369863014e-07, + "loss": 2.8005, + "step": 21 + }, + { + "epoch": 0.022598870056497175, + "grad_norm": 4.937129974365234, + "learning_rate": 1.36986301369863e-07, + "loss": 2.7902, + "step": 22 + }, + { + "epoch": 0.02362609142270159, + "grad_norm": 5.0690202713012695, + "learning_rate": 1.541095890410959e-07, + "loss": 2.7474, + "step": 23 + }, + { + "epoch": 0.02465331278890601, + "grad_norm": 4.980844974517822, + "learning_rate": 1.7123287671232878e-07, + "loss": 2.7695, + "step": 24 + }, + { + "epoch": 0.025680534155110426, + "grad_norm": 4.8092451095581055, + "learning_rate": 1.8835616438356165e-07, + "loss": 2.7611, + "step": 25 + }, + { + "epoch": 0.02670775552131484, + "grad_norm": 4.765528202056885, + "learning_rate": 2.0547945205479452e-07, + "loss": 2.7823, + "step": 26 + }, + { + "epoch": 0.02773497688751926, + "grad_norm": 3.9759938716888428, + "learning_rate": 2.226027397260274e-07, + "loss": 2.7445, + "step": 27 + }, + { + "epoch": 0.028762198253723677, + "grad_norm": 4.023419380187988, + "learning_rate": 2.397260273972603e-07, + "loss": 2.7679, + "step": 28 + }, + { + "epoch": 0.029789419619928096, + "grad_norm": 4.161378383636475, + "learning_rate": 2.568493150684932e-07, + "loss": 2.7704, + "step": 29 + }, + { + "epoch": 0.030816640986132512, + "grad_norm": 3.8363842964172363, + "learning_rate": 2.73972602739726e-07, + "loss": 2.7598, + "step": 30 + }, + { + "epoch": 0.030816640986132512, + "eval_loss": 2.772540330886841, + "eval_runtime": 37.2624, + "eval_samples_per_second": 26.837, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03184386235233693, + "grad_norm": 3.718747615814209, + "learning_rate": 2.910958904109589e-07, + "loss": 2.7241, + "step": 31 + }, + { + "epoch": 0.03287108371854135, + "grad_norm": 3.555985927581787, + "learning_rate": 3.082191780821918e-07, + "loss": 2.7751, + "step": 32 + }, + { + "epoch": 0.03389830508474576, + "grad_norm": 3.6041972637176514, + "learning_rate": 3.2534246575342466e-07, + "loss": 2.7322, + "step": 33 + }, + { + "epoch": 0.03492552645095018, + "grad_norm": 3.4408438205718994, + "learning_rate": 3.4246575342465755e-07, + "loss": 2.727, + "step": 34 + }, + { + "epoch": 0.035952747817154594, + "grad_norm": 3.4169280529022217, + "learning_rate": 3.595890410958904e-07, + "loss": 2.7062, + "step": 35 + }, + { + "epoch": 0.03697996918335902, + "grad_norm": 3.3882925510406494, + "learning_rate": 3.767123287671233e-07, + "loss": 2.6271, + "step": 36 + }, + { + "epoch": 0.03800719054956343, + "grad_norm": 3.0491855144500732, + "learning_rate": 3.938356164383562e-07, + "loss": 2.7309, + "step": 37 + }, + { + "epoch": 0.03903441191576785, + "grad_norm": 2.8444902896881104, + "learning_rate": 4.1095890410958903e-07, + "loss": 2.647, + "step": 38 + }, + { + "epoch": 0.040061633281972264, + "grad_norm": 2.7438602447509766, + "learning_rate": 4.2808219178082193e-07, + "loss": 2.6478, + "step": 39 + }, + { + "epoch": 0.04108885464817668, + "grad_norm": 2.8886020183563232, + "learning_rate": 4.452054794520548e-07, + "loss": 2.7065, + "step": 40 + }, + { + "epoch": 0.04108885464817668, + "eval_loss": 2.7021424770355225, + "eval_runtime": 37.3227, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.042116076014381096, + "grad_norm": 2.8825268745422363, + "learning_rate": 4.6232876712328767e-07, + "loss": 2.6666, + "step": 41 + }, + { + "epoch": 0.04314329738058552, + "grad_norm": 2.8848748207092285, + "learning_rate": 4.794520547945206e-07, + "loss": 2.597, + "step": 42 + }, + { + "epoch": 0.044170518746789934, + "grad_norm": 2.9176979064941406, + "learning_rate": 4.965753424657534e-07, + "loss": 2.6638, + "step": 43 + }, + { + "epoch": 0.04519774011299435, + "grad_norm": 2.6903059482574463, + "learning_rate": 5.136986301369864e-07, + "loss": 2.6202, + "step": 44 + }, + { + "epoch": 0.046224961479198766, + "grad_norm": 2.5349855422973633, + "learning_rate": 5.308219178082192e-07, + "loss": 2.6466, + "step": 45 + }, + { + "epoch": 0.04725218284540318, + "grad_norm": 2.405383825302124, + "learning_rate": 5.47945205479452e-07, + "loss": 2.6291, + "step": 46 + }, + { + "epoch": 0.048279404211607604, + "grad_norm": 2.526106595993042, + "learning_rate": 5.65068493150685e-07, + "loss": 2.6718, + "step": 47 + }, + { + "epoch": 0.04930662557781202, + "grad_norm": 2.3981597423553467, + "learning_rate": 5.821917808219178e-07, + "loss": 2.592, + "step": 48 + }, + { + "epoch": 0.050333846944016436, + "grad_norm": 2.61842679977417, + "learning_rate": 5.993150684931507e-07, + "loss": 2.6039, + "step": 49 + }, + { + "epoch": 0.05136106831022085, + "grad_norm": 2.3001420497894287, + "learning_rate": 6.164383561643836e-07, + "loss": 2.5481, + "step": 50 + }, + { + "epoch": 0.05136106831022085, + "eval_loss": 2.6244444847106934, + "eval_runtime": 37.3248, + "eval_samples_per_second": 26.792, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05238828967642527, + "grad_norm": 2.12744140625, + "learning_rate": 6.335616438356165e-07, + "loss": 2.5336, + "step": 51 + }, + { + "epoch": 0.05341551104262968, + "grad_norm": 1.974413275718689, + "learning_rate": 6.506849315068493e-07, + "loss": 2.5882, + "step": 52 + }, + { + "epoch": 0.054442732408834106, + "grad_norm": 2.098527669906616, + "learning_rate": 6.678082191780823e-07, + "loss": 2.5073, + "step": 53 + }, + { + "epoch": 0.05546995377503852, + "grad_norm": 2.056288719177246, + "learning_rate": 6.849315068493151e-07, + "loss": 2.5612, + "step": 54 + }, + { + "epoch": 0.05649717514124294, + "grad_norm": 2.2291629314422607, + "learning_rate": 7.020547945205481e-07, + "loss": 2.527, + "step": 55 + }, + { + "epoch": 0.05752439650744735, + "grad_norm": 2.140164375305176, + "learning_rate": 7.191780821917808e-07, + "loss": 2.4972, + "step": 56 + }, + { + "epoch": 0.05855161787365177, + "grad_norm": 1.8682485818862915, + "learning_rate": 7.363013698630137e-07, + "loss": 2.5231, + "step": 57 + }, + { + "epoch": 0.05957883923985619, + "grad_norm": 1.9373308420181274, + "learning_rate": 7.534246575342466e-07, + "loss": 2.4509, + "step": 58 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.8535712957382202, + "learning_rate": 7.705479452054795e-07, + "loss": 2.5164, + "step": 59 + }, + { + "epoch": 0.061633281972265024, + "grad_norm": 1.7497329711914062, + "learning_rate": 7.876712328767124e-07, + "loss": 2.5177, + "step": 60 + }, + { + "epoch": 0.061633281972265024, + "eval_loss": 2.5608646869659424, + "eval_runtime": 37.3426, + "eval_samples_per_second": 26.779, + "eval_steps_per_second": 1.125, + "step": 60 + }, + { + "epoch": 0.06266050333846944, + "grad_norm": 1.8476629257202148, + "learning_rate": 8.047945205479453e-07, + "loss": 2.4838, + "step": 61 + }, + { + "epoch": 0.06368772470467386, + "grad_norm": 1.7130696773529053, + "learning_rate": 8.219178082191781e-07, + "loss": 2.473, + "step": 62 + }, + { + "epoch": 0.06471494607087827, + "grad_norm": 1.751081109046936, + "learning_rate": 8.39041095890411e-07, + "loss": 2.5152, + "step": 63 + }, + { + "epoch": 0.0657421674370827, + "grad_norm": 1.7758578062057495, + "learning_rate": 8.561643835616439e-07, + "loss": 2.5031, + "step": 64 + }, + { + "epoch": 0.0667693888032871, + "grad_norm": 1.9316363334655762, + "learning_rate": 8.732876712328768e-07, + "loss": 2.3695, + "step": 65 + }, + { + "epoch": 0.06779661016949153, + "grad_norm": 1.7765878438949585, + "learning_rate": 8.904109589041097e-07, + "loss": 2.4669, + "step": 66 + }, + { + "epoch": 0.06882383153569595, + "grad_norm": 1.7339810132980347, + "learning_rate": 9.075342465753426e-07, + "loss": 2.4112, + "step": 67 + }, + { + "epoch": 0.06985105290190036, + "grad_norm": 1.717781662940979, + "learning_rate": 9.246575342465753e-07, + "loss": 2.4677, + "step": 68 + }, + { + "epoch": 0.07087827426810478, + "grad_norm": 1.7711046934127808, + "learning_rate": 9.417808219178083e-07, + "loss": 2.4131, + "step": 69 + }, + { + "epoch": 0.07190549563430919, + "grad_norm": 1.708680510520935, + "learning_rate": 9.589041095890411e-07, + "loss": 2.5191, + "step": 70 + }, + { + "epoch": 0.07190549563430919, + "eval_loss": 2.4920735359191895, + "eval_runtime": 37.3501, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 70 + }, + { + "epoch": 0.07293271700051361, + "grad_norm": 1.9429680109024048, + "learning_rate": 9.76027397260274e-07, + "loss": 2.4099, + "step": 71 + }, + { + "epoch": 0.07395993836671803, + "grad_norm": 1.7411247491836548, + "learning_rate": 9.931506849315068e-07, + "loss": 2.3632, + "step": 72 + }, + { + "epoch": 0.07498715973292244, + "grad_norm": 1.7577983140945435, + "learning_rate": 1.0102739726027399e-06, + "loss": 2.4214, + "step": 73 + }, + { + "epoch": 0.07601438109912687, + "grad_norm": 2.025926351547241, + "learning_rate": 1.0273972602739727e-06, + "loss": 2.394, + "step": 74 + }, + { + "epoch": 0.07704160246533127, + "grad_norm": 2.4366860389709473, + "learning_rate": 1.0445205479452056e-06, + "loss": 2.3943, + "step": 75 + }, + { + "epoch": 0.0780688238315357, + "grad_norm": 2.437108278274536, + "learning_rate": 1.0616438356164384e-06, + "loss": 2.3844, + "step": 76 + }, + { + "epoch": 0.07909604519774012, + "grad_norm": 2.1352434158325195, + "learning_rate": 1.0787671232876712e-06, + "loss": 2.3871, + "step": 77 + }, + { + "epoch": 0.08012326656394453, + "grad_norm": 1.8895397186279297, + "learning_rate": 1.095890410958904e-06, + "loss": 2.3927, + "step": 78 + }, + { + "epoch": 0.08115048793014895, + "grad_norm": 2.2960093021392822, + "learning_rate": 1.1130136986301371e-06, + "loss": 2.3343, + "step": 79 + }, + { + "epoch": 0.08217770929635336, + "grad_norm": 2.0891849994659424, + "learning_rate": 1.13013698630137e-06, + "loss": 2.3732, + "step": 80 + }, + { + "epoch": 0.08217770929635336, + "eval_loss": 2.4189229011535645, + "eval_runtime": 37.3645, + "eval_samples_per_second": 26.763, + "eval_steps_per_second": 1.124, + "step": 80 + }, + { + "epoch": 0.08320493066255778, + "grad_norm": 2.450587034225464, + "learning_rate": 1.1472602739726028e-06, + "loss": 2.3163, + "step": 81 + }, + { + "epoch": 0.08423215202876219, + "grad_norm": 2.079949378967285, + "learning_rate": 1.1643835616438357e-06, + "loss": 2.3381, + "step": 82 + }, + { + "epoch": 0.08525937339496661, + "grad_norm": 2.160737991333008, + "learning_rate": 1.1815068493150685e-06, + "loss": 2.3376, + "step": 83 + }, + { + "epoch": 0.08628659476117104, + "grad_norm": 2.463890552520752, + "learning_rate": 1.1986301369863014e-06, + "loss": 2.308, + "step": 84 + }, + { + "epoch": 0.08731381612737545, + "grad_norm": 2.3923065662384033, + "learning_rate": 1.2157534246575344e-06, + "loss": 2.2682, + "step": 85 + }, + { + "epoch": 0.08834103749357987, + "grad_norm": 2.2096004486083984, + "learning_rate": 1.2328767123287673e-06, + "loss": 2.3256, + "step": 86 + }, + { + "epoch": 0.08936825885978428, + "grad_norm": 2.1402556896209717, + "learning_rate": 1.25e-06, + "loss": 2.2767, + "step": 87 + }, + { + "epoch": 0.0903954802259887, + "grad_norm": 2.500629186630249, + "learning_rate": 1.267123287671233e-06, + "loss": 2.2576, + "step": 88 + }, + { + "epoch": 0.09142270159219312, + "grad_norm": 2.0257675647735596, + "learning_rate": 1.284246575342466e-06, + "loss": 2.2722, + "step": 89 + }, + { + "epoch": 0.09244992295839753, + "grad_norm": 2.148428440093994, + "learning_rate": 1.3013698630136986e-06, + "loss": 2.2309, + "step": 90 + }, + { + "epoch": 0.09244992295839753, + "eval_loss": 2.3439993858337402, + "eval_runtime": 37.3803, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 90 + }, + { + "epoch": 0.09347714432460195, + "grad_norm": 2.3541946411132812, + "learning_rate": 1.3184931506849317e-06, + "loss": 2.2347, + "step": 91 + }, + { + "epoch": 0.09450436569080636, + "grad_norm": 2.6239383220672607, + "learning_rate": 1.3356164383561645e-06, + "loss": 2.2589, + "step": 92 + }, + { + "epoch": 0.09553158705701079, + "grad_norm": 2.1981804370880127, + "learning_rate": 1.3527397260273976e-06, + "loss": 2.2372, + "step": 93 + }, + { + "epoch": 0.09655880842321521, + "grad_norm": 2.3652358055114746, + "learning_rate": 1.3698630136986302e-06, + "loss": 2.2169, + "step": 94 + }, + { + "epoch": 0.09758602978941962, + "grad_norm": 2.2078158855438232, + "learning_rate": 1.386986301369863e-06, + "loss": 2.2761, + "step": 95 + }, + { + "epoch": 0.09861325115562404, + "grad_norm": 2.494685649871826, + "learning_rate": 1.4041095890410961e-06, + "loss": 2.1691, + "step": 96 + }, + { + "epoch": 0.09964047252182845, + "grad_norm": 2.3293051719665527, + "learning_rate": 1.421232876712329e-06, + "loss": 2.2246, + "step": 97 + }, + { + "epoch": 0.10066769388803287, + "grad_norm": 2.410687208175659, + "learning_rate": 1.4383561643835616e-06, + "loss": 2.1166, + "step": 98 + }, + { + "epoch": 0.1016949152542373, + "grad_norm": 2.5168960094451904, + "learning_rate": 1.4554794520547946e-06, + "loss": 2.1417, + "step": 99 + }, + { + "epoch": 0.1027221366204417, + "grad_norm": 2.178412914276123, + "learning_rate": 1.4726027397260275e-06, + "loss": 2.1638, + "step": 100 + }, + { + "epoch": 0.1027221366204417, + "eval_loss": 2.273622751235962, + "eval_runtime": 37.3892, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 100 + }, + { + "epoch": 0.10374935798664613, + "grad_norm": 2.8951706886291504, + "learning_rate": 1.4897260273972605e-06, + "loss": 2.1631, + "step": 101 + }, + { + "epoch": 0.10477657935285054, + "grad_norm": 2.103206157684326, + "learning_rate": 1.5068493150684932e-06, + "loss": 2.1419, + "step": 102 + }, + { + "epoch": 0.10580380071905496, + "grad_norm": 2.495969772338867, + "learning_rate": 1.5239726027397262e-06, + "loss": 2.1029, + "step": 103 + }, + { + "epoch": 0.10683102208525937, + "grad_norm": 2.5686826705932617, + "learning_rate": 1.541095890410959e-06, + "loss": 2.1235, + "step": 104 + }, + { + "epoch": 0.10785824345146379, + "grad_norm": 3.4710471630096436, + "learning_rate": 1.5582191780821921e-06, + "loss": 2.1832, + "step": 105 + }, + { + "epoch": 0.10888546481766821, + "grad_norm": 2.2563316822052, + "learning_rate": 1.5753424657534248e-06, + "loss": 2.1441, + "step": 106 + }, + { + "epoch": 0.10991268618387262, + "grad_norm": 3.860506534576416, + "learning_rate": 1.5924657534246576e-06, + "loss": 2.1764, + "step": 107 + }, + { + "epoch": 0.11093990755007704, + "grad_norm": 2.550135850906372, + "learning_rate": 1.6095890410958907e-06, + "loss": 2.1455, + "step": 108 + }, + { + "epoch": 0.11196712891628145, + "grad_norm": 3.168283700942993, + "learning_rate": 1.6267123287671235e-06, + "loss": 2.0246, + "step": 109 + }, + { + "epoch": 0.11299435028248588, + "grad_norm": 2.577069044113159, + "learning_rate": 1.6438356164383561e-06, + "loss": 2.0921, + "step": 110 + }, + { + "epoch": 0.11299435028248588, + "eval_loss": 2.2218997478485107, + "eval_runtime": 37.246, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 110 + }, + { + "epoch": 0.1140215716486903, + "grad_norm": 2.3515374660491943, + "learning_rate": 1.6609589041095892e-06, + "loss": 2.0209, + "step": 111 + }, + { + "epoch": 0.1150487930148947, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.678082191780822e-06, + "loss": 2.1165, + "step": 112 + }, + { + "epoch": 0.11607601438109913, + "grad_norm": 2.567336082458496, + "learning_rate": 1.695205479452055e-06, + "loss": 2.0529, + "step": 113 + }, + { + "epoch": 0.11710323574730354, + "grad_norm": 2.4126152992248535, + "learning_rate": 1.7123287671232877e-06, + "loss": 2.1028, + "step": 114 + }, + { + "epoch": 0.11813045711350796, + "grad_norm": 3.577528476715088, + "learning_rate": 1.7294520547945206e-06, + "loss": 2.0317, + "step": 115 + }, + { + "epoch": 0.11915767847971238, + "grad_norm": 2.9624195098876953, + "learning_rate": 1.7465753424657536e-06, + "loss": 2.1318, + "step": 116 + }, + { + "epoch": 0.12018489984591679, + "grad_norm": 2.789854049682617, + "learning_rate": 1.7636986301369865e-06, + "loss": 2.0822, + "step": 117 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.8168365955352783, + "learning_rate": 1.7808219178082193e-06, + "loss": 2.08, + "step": 118 + }, + { + "epoch": 0.12223934257832562, + "grad_norm": 2.7616257667541504, + "learning_rate": 1.7979452054794521e-06, + "loss": 2.0668, + "step": 119 + }, + { + "epoch": 0.12326656394453005, + "grad_norm": 2.5610315799713135, + "learning_rate": 1.8150684931506852e-06, + "loss": 2.1518, + "step": 120 + }, + { + "epoch": 0.12326656394453005, + "eval_loss": 2.184033155441284, + "eval_runtime": 37.3784, + "eval_samples_per_second": 26.753, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.12429378531073447, + "grad_norm": 2.817063093185425, + "learning_rate": 1.832191780821918e-06, + "loss": 2.0813, + "step": 121 + }, + { + "epoch": 0.12532100667693888, + "grad_norm": 2.987205982208252, + "learning_rate": 1.8493150684931507e-06, + "loss": 2.0421, + "step": 122 + }, + { + "epoch": 0.1263482280431433, + "grad_norm": 2.8480496406555176, + "learning_rate": 1.8664383561643837e-06, + "loss": 2.0018, + "step": 123 + }, + { + "epoch": 0.12737544940934772, + "grad_norm": 2.994274616241455, + "learning_rate": 1.8835616438356166e-06, + "loss": 2.0266, + "step": 124 + }, + { + "epoch": 0.12840267077555212, + "grad_norm": 2.64136004447937, + "learning_rate": 1.9006849315068496e-06, + "loss": 1.9873, + "step": 125 + }, + { + "epoch": 0.12942989214175654, + "grad_norm": 2.9426915645599365, + "learning_rate": 1.9178082191780823e-06, + "loss": 2.0566, + "step": 126 + }, + { + "epoch": 0.13045711350796096, + "grad_norm": 2.9018619060516357, + "learning_rate": 1.9349315068493153e-06, + "loss": 1.9674, + "step": 127 + }, + { + "epoch": 0.1314843348741654, + "grad_norm": 3.6921045780181885, + "learning_rate": 1.952054794520548e-06, + "loss": 2.0295, + "step": 128 + }, + { + "epoch": 0.1325115562403698, + "grad_norm": 2.4194657802581787, + "learning_rate": 1.969178082191781e-06, + "loss": 2.0639, + "step": 129 + }, + { + "epoch": 0.1335387776065742, + "grad_norm": 3.252349615097046, + "learning_rate": 1.9863013698630136e-06, + "loss": 2.0613, + "step": 130 + }, + { + "epoch": 0.1335387776065742, + "eval_loss": 2.153157949447632, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 130 + }, + { + "epoch": 0.13456599897277863, + "grad_norm": 2.9361023902893066, + "learning_rate": 2.0034246575342467e-06, + "loss": 1.9817, + "step": 131 + }, + { + "epoch": 0.13559322033898305, + "grad_norm": 2.834803581237793, + "learning_rate": 2.0205479452054797e-06, + "loss": 2.0539, + "step": 132 + }, + { + "epoch": 0.13662044170518747, + "grad_norm": 2.8814785480499268, + "learning_rate": 2.037671232876713e-06, + "loss": 2.0296, + "step": 133 + }, + { + "epoch": 0.1376476630713919, + "grad_norm": 2.647125482559204, + "learning_rate": 2.0547945205479454e-06, + "loss": 1.9739, + "step": 134 + }, + { + "epoch": 0.1386748844375963, + "grad_norm": 2.9608750343322754, + "learning_rate": 2.071917808219178e-06, + "loss": 2.0083, + "step": 135 + }, + { + "epoch": 0.1397021058038007, + "grad_norm": 2.8570032119750977, + "learning_rate": 2.089041095890411e-06, + "loss": 2.0216, + "step": 136 + }, + { + "epoch": 0.14072932717000514, + "grad_norm": 2.806358575820923, + "learning_rate": 2.106164383561644e-06, + "loss": 1.9457, + "step": 137 + }, + { + "epoch": 0.14175654853620956, + "grad_norm": 3.1003353595733643, + "learning_rate": 2.123287671232877e-06, + "loss": 2.0324, + "step": 138 + }, + { + "epoch": 0.14278376990241398, + "grad_norm": 2.679588556289673, + "learning_rate": 2.14041095890411e-06, + "loss": 1.9699, + "step": 139 + }, + { + "epoch": 0.14381099126861838, + "grad_norm": 2.9436194896698, + "learning_rate": 2.1575342465753425e-06, + "loss": 1.9824, + "step": 140 + }, + { + "epoch": 0.14381099126861838, + "eval_loss": 2.123953104019165, + "eval_runtime": 37.4214, + "eval_samples_per_second": 26.723, + "eval_steps_per_second": 1.122, + "step": 140 + }, + { + "epoch": 0.1448382126348228, + "grad_norm": 3.088301420211792, + "learning_rate": 2.1746575342465755e-06, + "loss": 1.9861, + "step": 141 + }, + { + "epoch": 0.14586543400102722, + "grad_norm": 3.011859178543091, + "learning_rate": 2.191780821917808e-06, + "loss": 1.9383, + "step": 142 + }, + { + "epoch": 0.14689265536723164, + "grad_norm": 2.930264472961426, + "learning_rate": 2.2089041095890412e-06, + "loss": 2.0674, + "step": 143 + }, + { + "epoch": 0.14791987673343607, + "grad_norm": 3.361328601837158, + "learning_rate": 2.2260273972602743e-06, + "loss": 1.9495, + "step": 144 + }, + { + "epoch": 0.14894709809964046, + "grad_norm": 2.7825112342834473, + "learning_rate": 2.243150684931507e-06, + "loss": 1.9335, + "step": 145 + }, + { + "epoch": 0.14997431946584489, + "grad_norm": 3.7215847969055176, + "learning_rate": 2.26027397260274e-06, + "loss": 1.9403, + "step": 146 + }, + { + "epoch": 0.1510015408320493, + "grad_norm": 2.5125460624694824, + "learning_rate": 2.2773972602739726e-06, + "loss": 1.9437, + "step": 147 + }, + { + "epoch": 0.15202876219825373, + "grad_norm": 3.0963492393493652, + "learning_rate": 2.2945205479452057e-06, + "loss": 1.9651, + "step": 148 + }, + { + "epoch": 0.15305598356445815, + "grad_norm": 2.9147121906280518, + "learning_rate": 2.3116438356164387e-06, + "loss": 2.0567, + "step": 149 + }, + { + "epoch": 0.15408320493066255, + "grad_norm": 3.5275802612304688, + "learning_rate": 2.3287671232876713e-06, + "loss": 1.9773, + "step": 150 + }, + { + "epoch": 0.15408320493066255, + "eval_loss": 2.112344264984131, + "eval_runtime": 37.4142, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 150 + }, + { + "epoch": 0.15511042629686697, + "grad_norm": 3.013786554336548, + "learning_rate": 2.3458904109589044e-06, + "loss": 1.8753, + "step": 151 + }, + { + "epoch": 0.1561376476630714, + "grad_norm": 3.433615207672119, + "learning_rate": 2.363013698630137e-06, + "loss": 1.9256, + "step": 152 + }, + { + "epoch": 0.15716486902927582, + "grad_norm": 3.1526033878326416, + "learning_rate": 2.38013698630137e-06, + "loss": 1.9815, + "step": 153 + }, + { + "epoch": 0.15819209039548024, + "grad_norm": 3.2099974155426025, + "learning_rate": 2.3972602739726027e-06, + "loss": 1.9628, + "step": 154 + }, + { + "epoch": 0.15921931176168463, + "grad_norm": 3.2713584899902344, + "learning_rate": 2.4143835616438358e-06, + "loss": 1.9128, + "step": 155 + }, + { + "epoch": 0.16024653312788906, + "grad_norm": 3.287649393081665, + "learning_rate": 2.431506849315069e-06, + "loss": 1.8764, + "step": 156 + }, + { + "epoch": 0.16127375449409348, + "grad_norm": 3.1439101696014404, + "learning_rate": 2.4486301369863015e-06, + "loss": 1.9428, + "step": 157 + }, + { + "epoch": 0.1623009758602979, + "grad_norm": 3.006457805633545, + "learning_rate": 2.4657534246575345e-06, + "loss": 1.9516, + "step": 158 + }, + { + "epoch": 0.1633281972265023, + "grad_norm": 2.9567787647247314, + "learning_rate": 2.482876712328767e-06, + "loss": 1.915, + "step": 159 + }, + { + "epoch": 0.16435541859270672, + "grad_norm": 3.2475216388702393, + "learning_rate": 2.5e-06, + "loss": 1.9451, + "step": 160 + }, + { + "epoch": 0.16435541859270672, + "eval_loss": 2.086299419403076, + "eval_runtime": 37.3939, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 160 + }, + { + "epoch": 0.16538263995891114, + "grad_norm": 2.9586498737335205, + "learning_rate": 2.5171232876712333e-06, + "loss": 1.9298, + "step": 161 + }, + { + "epoch": 0.16640986132511557, + "grad_norm": 3.066910982131958, + "learning_rate": 2.534246575342466e-06, + "loss": 1.9103, + "step": 162 + }, + { + "epoch": 0.16743708269132, + "grad_norm": 2.814741611480713, + "learning_rate": 2.551369863013699e-06, + "loss": 1.8678, + "step": 163 + }, + { + "epoch": 0.16846430405752438, + "grad_norm": 3.7239339351654053, + "learning_rate": 2.568493150684932e-06, + "loss": 1.9321, + "step": 164 + }, + { + "epoch": 0.1694915254237288, + "grad_norm": 2.829899787902832, + "learning_rate": 2.585616438356164e-06, + "loss": 1.9587, + "step": 165 + }, + { + "epoch": 0.17051874678993323, + "grad_norm": 3.826289176940918, + "learning_rate": 2.6027397260273973e-06, + "loss": 1.9288, + "step": 166 + }, + { + "epoch": 0.17154596815613765, + "grad_norm": 3.1028499603271484, + "learning_rate": 2.6198630136986303e-06, + "loss": 1.8685, + "step": 167 + }, + { + "epoch": 0.17257318952234207, + "grad_norm": 3.6923508644104004, + "learning_rate": 2.6369863013698634e-06, + "loss": 1.9276, + "step": 168 + }, + { + "epoch": 0.17360041088854647, + "grad_norm": 3.0577642917633057, + "learning_rate": 2.654109589041096e-06, + "loss": 1.9653, + "step": 169 + }, + { + "epoch": 0.1746276322547509, + "grad_norm": 3.363762617111206, + "learning_rate": 2.671232876712329e-06, + "loss": 1.9465, + "step": 170 + }, + { + "epoch": 0.1746276322547509, + "eval_loss": 2.0753531455993652, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 170 + }, + { + "epoch": 0.17565485362095531, + "grad_norm": 2.9063165187835693, + "learning_rate": 2.688356164383562e-06, + "loss": 1.8827, + "step": 171 + }, + { + "epoch": 0.17668207498715974, + "grad_norm": 3.4709181785583496, + "learning_rate": 2.705479452054795e-06, + "loss": 1.882, + "step": 172 + }, + { + "epoch": 0.17770929635336416, + "grad_norm": 3.395397663116455, + "learning_rate": 2.7226027397260274e-06, + "loss": 1.8711, + "step": 173 + }, + { + "epoch": 0.17873651771956856, + "grad_norm": 3.9973363876342773, + "learning_rate": 2.7397260273972604e-06, + "loss": 1.8223, + "step": 174 + }, + { + "epoch": 0.17976373908577298, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.947, + "step": 175 + }, + { + "epoch": 0.1807909604519774, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.9192, + "step": 176 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 3.84712290763855, + "learning_rate": 2.773972602739726e-06, + "loss": 1.8344, + "step": 177 + }, + { + "epoch": 0.18284540318438625, + "grad_norm": 3.487525463104248, + "learning_rate": 2.791095890410959e-06, + "loss": 1.9242, + "step": 178 + }, + { + "epoch": 0.18387262455059064, + "grad_norm": 3.1980319023132324, + "learning_rate": 2.8082191780821922e-06, + "loss": 1.7959, + "step": 179 + }, + { + "epoch": 0.18489984591679506, + "grad_norm": 3.56462025642395, + "learning_rate": 2.8253424657534253e-06, + "loss": 1.9668, + "step": 180 + }, + { + "epoch": 0.18489984591679506, + "eval_loss": 2.0606908798217773, + "eval_runtime": 37.3994, + "eval_samples_per_second": 26.738, + "eval_steps_per_second": 1.123, + "step": 180 + }, + { + "epoch": 0.1859270672829995, + "grad_norm": 3.613842248916626, + "learning_rate": 2.842465753424658e-06, + "loss": 1.925, + "step": 181 + }, + { + "epoch": 0.1869542886492039, + "grad_norm": 2.9755642414093018, + "learning_rate": 2.8595890410958905e-06, + "loss": 1.9251, + "step": 182 + }, + { + "epoch": 0.18798151001540833, + "grad_norm": 2.9678146839141846, + "learning_rate": 2.876712328767123e-06, + "loss": 1.9406, + "step": 183 + }, + { + "epoch": 0.18900873138161273, + "grad_norm": 3.3536393642425537, + "learning_rate": 2.8938356164383562e-06, + "loss": 1.8041, + "step": 184 + }, + { + "epoch": 0.19003595274781715, + "grad_norm": 2.714902877807617, + "learning_rate": 2.9109589041095893e-06, + "loss": 1.8213, + "step": 185 + }, + { + "epoch": 0.19106317411402157, + "grad_norm": 3.0686237812042236, + "learning_rate": 2.9280821917808223e-06, + "loss": 1.817, + "step": 186 + }, + { + "epoch": 0.192090395480226, + "grad_norm": 3.510448932647705, + "learning_rate": 2.945205479452055e-06, + "loss": 1.8317, + "step": 187 + }, + { + "epoch": 0.19311761684643042, + "grad_norm": 3.2060787677764893, + "learning_rate": 2.962328767123288e-06, + "loss": 1.894, + "step": 188 + }, + { + "epoch": 0.1941448382126348, + "grad_norm": 3.1756906509399414, + "learning_rate": 2.979452054794521e-06, + "loss": 1.8788, + "step": 189 + }, + { + "epoch": 0.19517205957883924, + "grad_norm": 3.059863805770874, + "learning_rate": 2.9965753424657533e-06, + "loss": 1.8198, + "step": 190 + }, + { + "epoch": 0.19517205957883924, + "eval_loss": 2.0506513118743896, + "eval_runtime": 37.4014, + "eval_samples_per_second": 26.737, + "eval_steps_per_second": 1.123, + "step": 190 + }, + { + "epoch": 0.19619928094504366, + "grad_norm": 3.193685531616211, + "learning_rate": 3.0136986301369864e-06, + "loss": 1.8818, + "step": 191 + }, + { + "epoch": 0.19722650231124808, + "grad_norm": 2.905402183532715, + "learning_rate": 3.0308219178082194e-06, + "loss": 1.8546, + "step": 192 + }, + { + "epoch": 0.1982537236774525, + "grad_norm": 3.423109531402588, + "learning_rate": 3.0479452054794525e-06, + "loss": 1.8142, + "step": 193 + }, + { + "epoch": 0.1992809450436569, + "grad_norm": 3.136094331741333, + "learning_rate": 3.065068493150685e-06, + "loss": 1.8553, + "step": 194 + }, + { + "epoch": 0.20030816640986132, + "grad_norm": 3.0746538639068604, + "learning_rate": 3.082191780821918e-06, + "loss": 1.8153, + "step": 195 + }, + { + "epoch": 0.20133538777606574, + "grad_norm": 2.9354681968688965, + "learning_rate": 3.099315068493151e-06, + "loss": 1.8301, + "step": 196 + }, + { + "epoch": 0.20236260914227017, + "grad_norm": 3.00426983833313, + "learning_rate": 3.1164383561643843e-06, + "loss": 1.8233, + "step": 197 + }, + { + "epoch": 0.2033898305084746, + "grad_norm": 3.2754054069519043, + "learning_rate": 3.1335616438356165e-06, + "loss": 1.787, + "step": 198 + }, + { + "epoch": 0.20441705187467898, + "grad_norm": 2.9854509830474854, + "learning_rate": 3.1506849315068495e-06, + "loss": 1.9341, + "step": 199 + }, + { + "epoch": 0.2054442732408834, + "grad_norm": 3.5655088424682617, + "learning_rate": 3.167808219178082e-06, + "loss": 1.7964, + "step": 200 + }, + { + "epoch": 0.2054442732408834, + "eval_loss": 2.0440263748168945, + "eval_runtime": 37.4054, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 200 + }, + { + "epoch": 0.20647149460708783, + "grad_norm": 3.132373571395874, + "learning_rate": 3.184931506849315e-06, + "loss": 1.8019, + "step": 201 + }, + { + "epoch": 0.20749871597329225, + "grad_norm": 3.3017780780792236, + "learning_rate": 3.2020547945205483e-06, + "loss": 1.7786, + "step": 202 + }, + { + "epoch": 0.20852593733949668, + "grad_norm": 3.009382486343384, + "learning_rate": 3.2191780821917813e-06, + "loss": 1.7639, + "step": 203 + }, + { + "epoch": 0.20955315870570107, + "grad_norm": 3.5784854888916016, + "learning_rate": 3.236301369863014e-06, + "loss": 1.76, + "step": 204 + }, + { + "epoch": 0.2105803800719055, + "grad_norm": 3.236858367919922, + "learning_rate": 3.253424657534247e-06, + "loss": 1.7295, + "step": 205 + }, + { + "epoch": 0.21160760143810992, + "grad_norm": 3.6157479286193848, + "learning_rate": 3.2705479452054796e-06, + "loss": 1.8287, + "step": 206 + }, + { + "epoch": 0.21263482280431434, + "grad_norm": 3.51790452003479, + "learning_rate": 3.2876712328767123e-06, + "loss": 1.8004, + "step": 207 + }, + { + "epoch": 0.21366204417051873, + "grad_norm": 3.0700764656066895, + "learning_rate": 3.3047945205479453e-06, + "loss": 1.7581, + "step": 208 + }, + { + "epoch": 0.21468926553672316, + "grad_norm": 3.2651591300964355, + "learning_rate": 3.3219178082191784e-06, + "loss": 1.7724, + "step": 209 + }, + { + "epoch": 0.21571648690292758, + "grad_norm": 2.9267404079437256, + "learning_rate": 3.3390410958904114e-06, + "loss": 1.8154, + "step": 210 + }, + { + "epoch": 0.21571648690292758, + "eval_loss": 2.0323731899261475, + "eval_runtime": 37.2156, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 210 + }, + { + "epoch": 0.216743708269132, + "grad_norm": 3.463411808013916, + "learning_rate": 3.356164383561644e-06, + "loss": 1.8309, + "step": 211 + }, + { + "epoch": 0.21777092963533642, + "grad_norm": 3.123573064804077, + "learning_rate": 3.373287671232877e-06, + "loss": 1.8793, + "step": 212 + }, + { + "epoch": 0.21879815100154082, + "grad_norm": 3.427051305770874, + "learning_rate": 3.39041095890411e-06, + "loss": 1.8259, + "step": 213 + }, + { + "epoch": 0.21982537236774524, + "grad_norm": 3.306366443634033, + "learning_rate": 3.4075342465753424e-06, + "loss": 1.8565, + "step": 214 + }, + { + "epoch": 0.22085259373394966, + "grad_norm": 3.557340383529663, + "learning_rate": 3.4246575342465754e-06, + "loss": 1.7531, + "step": 215 + }, + { + "epoch": 0.2218798151001541, + "grad_norm": 3.20525860786438, + "learning_rate": 3.4417808219178085e-06, + "loss": 1.8032, + "step": 216 + }, + { + "epoch": 0.2229070364663585, + "grad_norm": 3.2657511234283447, + "learning_rate": 3.458904109589041e-06, + "loss": 1.8026, + "step": 217 + }, + { + "epoch": 0.2239342578325629, + "grad_norm": 3.1869699954986572, + "learning_rate": 3.476027397260274e-06, + "loss": 1.814, + "step": 218 + }, + { + "epoch": 0.22496147919876733, + "grad_norm": 2.661536931991577, + "learning_rate": 3.4931506849315072e-06, + "loss": 1.8005, + "step": 219 + }, + { + "epoch": 0.22598870056497175, + "grad_norm": 3.1108739376068115, + "learning_rate": 3.5102739726027403e-06, + "loss": 1.8095, + "step": 220 + }, + { + "epoch": 0.22598870056497175, + "eval_loss": 2.015268325805664, + "eval_runtime": 37.2867, + "eval_samples_per_second": 26.819, + "eval_steps_per_second": 1.126, + "step": 220 + }, + { + "epoch": 0.22701592193117617, + "grad_norm": 2.8078837394714355, + "learning_rate": 3.527397260273973e-06, + "loss": 1.8016, + "step": 221 + }, + { + "epoch": 0.2280431432973806, + "grad_norm": 3.131399631500244, + "learning_rate": 3.5445205479452056e-06, + "loss": 1.7787, + "step": 222 + }, + { + "epoch": 0.229070364663585, + "grad_norm": 3.114422559738159, + "learning_rate": 3.5616438356164386e-06, + "loss": 1.7899, + "step": 223 + }, + { + "epoch": 0.2300975860297894, + "grad_norm": 3.040816068649292, + "learning_rate": 3.5787671232876712e-06, + "loss": 1.8601, + "step": 224 + }, + { + "epoch": 0.23112480739599384, + "grad_norm": 3.043100357055664, + "learning_rate": 3.5958904109589043e-06, + "loss": 1.8244, + "step": 225 + }, + { + "epoch": 0.23215202876219826, + "grad_norm": 3.4232981204986572, + "learning_rate": 3.6130136986301373e-06, + "loss": 1.7634, + "step": 226 + }, + { + "epoch": 0.23317925012840268, + "grad_norm": 2.8170571327209473, + "learning_rate": 3.6301369863013704e-06, + "loss": 1.7793, + "step": 227 + }, + { + "epoch": 0.23420647149460708, + "grad_norm": 3.9133572578430176, + "learning_rate": 3.647260273972603e-06, + "loss": 1.8239, + "step": 228 + }, + { + "epoch": 0.2352336928608115, + "grad_norm": 2.7053382396698, + "learning_rate": 3.664383561643836e-06, + "loss": 1.8209, + "step": 229 + }, + { + "epoch": 0.23626091422701592, + "grad_norm": 3.4366891384124756, + "learning_rate": 3.6815068493150687e-06, + "loss": 1.7539, + "step": 230 + }, + { + "epoch": 0.23626091422701592, + "eval_loss": 2.0176751613616943, + "eval_runtime": 37.3064, + "eval_samples_per_second": 26.805, + "eval_steps_per_second": 1.126, + "step": 230 + }, + { + "epoch": 0.23728813559322035, + "grad_norm": 2.592280149459839, + "learning_rate": 3.6986301369863014e-06, + "loss": 1.7858, + "step": 231 + }, + { + "epoch": 0.23831535695942477, + "grad_norm": 2.8370869159698486, + "learning_rate": 3.7157534246575344e-06, + "loss": 1.7414, + "step": 232 + }, + { + "epoch": 0.23934257832562916, + "grad_norm": 3.2120535373687744, + "learning_rate": 3.7328767123287675e-06, + "loss": 1.8467, + "step": 233 + }, + { + "epoch": 0.24036979969183359, + "grad_norm": 2.9314255714416504, + "learning_rate": 3.7500000000000005e-06, + "loss": 1.7602, + "step": 234 + }, + { + "epoch": 0.241397021058038, + "grad_norm": 3.056293249130249, + "learning_rate": 3.767123287671233e-06, + "loss": 1.7972, + "step": 235 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.5918917655944824, + "learning_rate": 3.784246575342466e-06, + "loss": 1.8274, + "step": 236 + }, + { + "epoch": 0.24345146379044685, + "grad_norm": 2.7911906242370605, + "learning_rate": 3.8013698630136993e-06, + "loss": 1.753, + "step": 237 + }, + { + "epoch": 0.24447868515665125, + "grad_norm": 2.810612916946411, + "learning_rate": 3.818493150684932e-06, + "loss": 1.7822, + "step": 238 + }, + { + "epoch": 0.24550590652285567, + "grad_norm": 2.8863744735717773, + "learning_rate": 3.8356164383561645e-06, + "loss": 1.7258, + "step": 239 + }, + { + "epoch": 0.2465331278890601, + "grad_norm": 3.002087116241455, + "learning_rate": 3.852739726027397e-06, + "loss": 1.8423, + "step": 240 + }, + { + "epoch": 0.2465331278890601, + "eval_loss": 2.020223617553711, + "eval_runtime": 37.3197, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 240 + }, + { + "epoch": 0.24756034925526452, + "grad_norm": 2.892317056655884, + "learning_rate": 3.869863013698631e-06, + "loss": 1.8181, + "step": 241 + }, + { + "epoch": 0.24858757062146894, + "grad_norm": 2.895620584487915, + "learning_rate": 3.886986301369863e-06, + "loss": 1.7538, + "step": 242 + }, + { + "epoch": 0.24961479198767333, + "grad_norm": 3.015660524368286, + "learning_rate": 3.904109589041096e-06, + "loss": 1.766, + "step": 243 + }, + { + "epoch": 0.25064201335387776, + "grad_norm": 2.7516112327575684, + "learning_rate": 3.921232876712329e-06, + "loss": 1.7324, + "step": 244 + }, + { + "epoch": 0.2516692347200822, + "grad_norm": 2.937777519226074, + "learning_rate": 3.938356164383562e-06, + "loss": 1.7468, + "step": 245 + }, + { + "epoch": 0.2526964560862866, + "grad_norm": 3.1706550121307373, + "learning_rate": 3.9554794520547955e-06, + "loss": 1.7035, + "step": 246 + }, + { + "epoch": 0.253723677452491, + "grad_norm": 2.7795372009277344, + "learning_rate": 3.972602739726027e-06, + "loss": 1.7912, + "step": 247 + }, + { + "epoch": 0.25475089881869545, + "grad_norm": 3.0375940799713135, + "learning_rate": 3.989726027397261e-06, + "loss": 1.7208, + "step": 248 + }, + { + "epoch": 0.25577812018489987, + "grad_norm": 2.863913059234619, + "learning_rate": 4.006849315068493e-06, + "loss": 1.7891, + "step": 249 + }, + { + "epoch": 0.25680534155110424, + "grad_norm": 3.02720046043396, + "learning_rate": 4.023972602739726e-06, + "loss": 1.7864, + "step": 250 + }, + { + "epoch": 0.25680534155110424, + "eval_loss": 1.9981943368911743, + "eval_runtime": 37.3019, + "eval_samples_per_second": 26.808, + "eval_steps_per_second": 1.126, + "step": 250 + }, + { + "epoch": 0.25783256291730866, + "grad_norm": 2.90144681930542, + "learning_rate": 4.0410958904109595e-06, + "loss": 1.7508, + "step": 251 + }, + { + "epoch": 0.2588597842835131, + "grad_norm": 2.7670342922210693, + "learning_rate": 4.058219178082192e-06, + "loss": 1.7668, + "step": 252 + }, + { + "epoch": 0.2598870056497175, + "grad_norm": 3.0309600830078125, + "learning_rate": 4.075342465753426e-06, + "loss": 1.7853, + "step": 253 + }, + { + "epoch": 0.26091422701592193, + "grad_norm": 2.995190143585205, + "learning_rate": 4.092465753424658e-06, + "loss": 1.7409, + "step": 254 + }, + { + "epoch": 0.26194144838212635, + "grad_norm": 3.127767324447632, + "learning_rate": 4.109589041095891e-06, + "loss": 1.743, + "step": 255 + }, + { + "epoch": 0.2629686697483308, + "grad_norm": 3.7104549407958984, + "learning_rate": 4.1267123287671235e-06, + "loss": 1.8149, + "step": 256 + }, + { + "epoch": 0.2639958911145352, + "grad_norm": 3.042569160461426, + "learning_rate": 4.143835616438356e-06, + "loss": 1.7544, + "step": 257 + }, + { + "epoch": 0.2650231124807396, + "grad_norm": 2.9337198734283447, + "learning_rate": 4.16095890410959e-06, + "loss": 1.823, + "step": 258 + }, + { + "epoch": 0.26605033384694404, + "grad_norm": 3.1853089332580566, + "learning_rate": 4.178082191780822e-06, + "loss": 1.8002, + "step": 259 + }, + { + "epoch": 0.2670775552131484, + "grad_norm": 2.9620680809020996, + "learning_rate": 4.195205479452055e-06, + "loss": 1.7401, + "step": 260 + }, + { + "epoch": 0.2670775552131484, + "eval_loss": 1.9844669103622437, + "eval_runtime": 37.3162, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 260 + }, + { + "epoch": 0.26810477657935283, + "grad_norm": 3.477985382080078, + "learning_rate": 4.212328767123288e-06, + "loss": 1.7456, + "step": 261 + }, + { + "epoch": 0.26913199794555726, + "grad_norm": 2.881277084350586, + "learning_rate": 4.229452054794521e-06, + "loss": 1.7427, + "step": 262 + }, + { + "epoch": 0.2701592193117617, + "grad_norm": 3.3313355445861816, + "learning_rate": 4.246575342465754e-06, + "loss": 1.7537, + "step": 263 + }, + { + "epoch": 0.2711864406779661, + "grad_norm": 2.947119951248169, + "learning_rate": 4.263698630136986e-06, + "loss": 1.7349, + "step": 264 + }, + { + "epoch": 0.2722136620441705, + "grad_norm": 3.0780889987945557, + "learning_rate": 4.28082191780822e-06, + "loss": 1.694, + "step": 265 + }, + { + "epoch": 0.27324088341037495, + "grad_norm": 3.3371143341064453, + "learning_rate": 4.297945205479452e-06, + "loss": 1.8489, + "step": 266 + }, + { + "epoch": 0.27426810477657937, + "grad_norm": 3.01694917678833, + "learning_rate": 4.315068493150685e-06, + "loss": 1.7478, + "step": 267 + }, + { + "epoch": 0.2752953261427838, + "grad_norm": 3.001850128173828, + "learning_rate": 4.3321917808219185e-06, + "loss": 1.7298, + "step": 268 + }, + { + "epoch": 0.2763225475089882, + "grad_norm": 2.9096696376800537, + "learning_rate": 4.349315068493151e-06, + "loss": 1.7542, + "step": 269 + }, + { + "epoch": 0.2773497688751926, + "grad_norm": 3.209601402282715, + "learning_rate": 4.3664383561643846e-06, + "loss": 1.7248, + "step": 270 + }, + { + "epoch": 0.2773497688751926, + "eval_loss": 1.9732507467269897, + "eval_runtime": 37.3151, + "eval_samples_per_second": 26.799, + "eval_steps_per_second": 1.126, + "step": 270 + }, + { + "epoch": 0.278376990241397, + "grad_norm": 2.9635558128356934, + "learning_rate": 4.383561643835616e-06, + "loss": 1.727, + "step": 271 + }, + { + "epoch": 0.2794042116076014, + "grad_norm": 3.143821954727173, + "learning_rate": 4.40068493150685e-06, + "loss": 1.8045, + "step": 272 + }, + { + "epoch": 0.28043143297380585, + "grad_norm": 3.4248714447021484, + "learning_rate": 4.4178082191780825e-06, + "loss": 1.7649, + "step": 273 + }, + { + "epoch": 0.2814586543400103, + "grad_norm": 2.902162790298462, + "learning_rate": 4.434931506849315e-06, + "loss": 1.7324, + "step": 274 + }, + { + "epoch": 0.2824858757062147, + "grad_norm": 2.8899712562561035, + "learning_rate": 4.4520547945205486e-06, + "loss": 1.6886, + "step": 275 + }, + { + "epoch": 0.2835130970724191, + "grad_norm": 3.2267258167266846, + "learning_rate": 4.469178082191781e-06, + "loss": 1.7442, + "step": 276 + }, + { + "epoch": 0.28454031843862354, + "grad_norm": 3.275735378265381, + "learning_rate": 4.486301369863014e-06, + "loss": 1.7334, + "step": 277 + }, + { + "epoch": 0.28556753980482796, + "grad_norm": 2.991509199142456, + "learning_rate": 4.503424657534247e-06, + "loss": 1.7853, + "step": 278 + }, + { + "epoch": 0.28659476117103233, + "grad_norm": 2.8380391597747803, + "learning_rate": 4.52054794520548e-06, + "loss": 1.6885, + "step": 279 + }, + { + "epoch": 0.28762198253723675, + "grad_norm": 2.909980297088623, + "learning_rate": 4.537671232876713e-06, + "loss": 1.7146, + "step": 280 + }, + { + "epoch": 0.28762198253723675, + "eval_loss": 1.9755980968475342, + "eval_runtime": 37.3524, + "eval_samples_per_second": 26.772, + "eval_steps_per_second": 1.124, + "step": 280 + }, + { + "epoch": 0.2886492039034412, + "grad_norm": 3.2689175605773926, + "learning_rate": 4.554794520547945e-06, + "loss": 1.7747, + "step": 281 + }, + { + "epoch": 0.2896764252696456, + "grad_norm": 2.8242290019989014, + "learning_rate": 4.571917808219179e-06, + "loss": 1.7657, + "step": 282 + }, + { + "epoch": 0.29070364663585, + "grad_norm": 3.03828501701355, + "learning_rate": 4.589041095890411e-06, + "loss": 1.7555, + "step": 283 + }, + { + "epoch": 0.29173086800205444, + "grad_norm": 2.714351177215576, + "learning_rate": 4.606164383561644e-06, + "loss": 1.7055, + "step": 284 + }, + { + "epoch": 0.29275808936825887, + "grad_norm": 2.9540348052978516, + "learning_rate": 4.6232876712328774e-06, + "loss": 1.7258, + "step": 285 + }, + { + "epoch": 0.2937853107344633, + "grad_norm": 3.107060670852661, + "learning_rate": 4.64041095890411e-06, + "loss": 1.7376, + "step": 286 + }, + { + "epoch": 0.2948125321006677, + "grad_norm": 3.3266639709472656, + "learning_rate": 4.657534246575343e-06, + "loss": 1.7063, + "step": 287 + }, + { + "epoch": 0.29583975346687214, + "grad_norm": 3.3113174438476562, + "learning_rate": 4.674657534246575e-06, + "loss": 1.7514, + "step": 288 + }, + { + "epoch": 0.2968669748330765, + "grad_norm": 3.313321352005005, + "learning_rate": 4.691780821917809e-06, + "loss": 1.73, + "step": 289 + }, + { + "epoch": 0.2978941961992809, + "grad_norm": 2.901522159576416, + "learning_rate": 4.7089041095890414e-06, + "loss": 1.7709, + "step": 290 + }, + { + "epoch": 0.2978941961992809, + "eval_loss": 1.9608837366104126, + "eval_runtime": 37.3687, + "eval_samples_per_second": 26.76, + "eval_steps_per_second": 1.124, + "step": 290 + }, + { + "epoch": 0.29892141756548535, + "grad_norm": 3.4208264350891113, + "learning_rate": 4.726027397260274e-06, + "loss": 1.73, + "step": 291 + }, + { + "epoch": 0.29994863893168977, + "grad_norm": 2.8982441425323486, + "learning_rate": 4.7431506849315075e-06, + "loss": 1.7416, + "step": 292 + }, + { + "epoch": 0.3009758602978942, + "grad_norm": 3.2743732929229736, + "learning_rate": 4.76027397260274e-06, + "loss": 1.7443, + "step": 293 + }, + { + "epoch": 0.3020030816640986, + "grad_norm": 3.028764486312866, + "learning_rate": 4.777397260273973e-06, + "loss": 1.6879, + "step": 294 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.3531785011291504, + "learning_rate": 4.7945205479452054e-06, + "loss": 1.7096, + "step": 295 + }, + { + "epoch": 0.30405752439650746, + "grad_norm": 2.5602540969848633, + "learning_rate": 4.811643835616439e-06, + "loss": 1.7125, + "step": 296 + }, + { + "epoch": 0.3050847457627119, + "grad_norm": 3.1823129653930664, + "learning_rate": 4.8287671232876716e-06, + "loss": 1.6531, + "step": 297 + }, + { + "epoch": 0.3061119671289163, + "grad_norm": 2.6700282096862793, + "learning_rate": 4.845890410958904e-06, + "loss": 1.6983, + "step": 298 + }, + { + "epoch": 0.3071391884951207, + "grad_norm": 2.7118732929229736, + "learning_rate": 4.863013698630138e-06, + "loss": 1.7056, + "step": 299 + }, + { + "epoch": 0.3081664098613251, + "grad_norm": 3.3493311405181885, + "learning_rate": 4.88013698630137e-06, + "loss": 1.6944, + "step": 300 + }, + { + "epoch": 0.3081664098613251, + "eval_loss": 1.9489911794662476, + "eval_runtime": 37.35, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 300 + }, + { + "epoch": 0.3091936312275295, + "grad_norm": 2.9436538219451904, + "learning_rate": 4.897260273972603e-06, + "loss": 1.6717, + "step": 301 + }, + { + "epoch": 0.31022085259373394, + "grad_norm": 2.894737482070923, + "learning_rate": 4.914383561643836e-06, + "loss": 1.7895, + "step": 302 + }, + { + "epoch": 0.31124807395993837, + "grad_norm": 2.8645665645599365, + "learning_rate": 4.931506849315069e-06, + "loss": 1.721, + "step": 303 + }, + { + "epoch": 0.3122752953261428, + "grad_norm": 3.050203323364258, + "learning_rate": 4.948630136986302e-06, + "loss": 1.7564, + "step": 304 + }, + { + "epoch": 0.3133025166923472, + "grad_norm": 3.146395206451416, + "learning_rate": 4.965753424657534e-06, + "loss": 1.7938, + "step": 305 + }, + { + "epoch": 0.31432973805855163, + "grad_norm": 3.1021251678466797, + "learning_rate": 4.982876712328768e-06, + "loss": 1.6682, + "step": 306 + }, + { + "epoch": 0.31535695942475606, + "grad_norm": 2.7301125526428223, + "learning_rate": 5e-06, + "loss": 1.6974, + "step": 307 + }, + { + "epoch": 0.3163841807909605, + "grad_norm": 2.7602250576019287, + "learning_rate": 4.998096688237534e-06, + "loss": 1.6523, + "step": 308 + }, + { + "epoch": 0.31741140215716485, + "grad_norm": 2.8149876594543457, + "learning_rate": 4.996193376475067e-06, + "loss": 1.7535, + "step": 309 + }, + { + "epoch": 0.31843862352336927, + "grad_norm": 2.54823637008667, + "learning_rate": 4.9942900647126005e-06, + "loss": 1.63, + "step": 310 + }, + { + "epoch": 0.31843862352336927, + "eval_loss": 1.928893804550171, + "eval_runtime": 37.2236, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3194658448895737, + "grad_norm": 2.7819766998291016, + "learning_rate": 4.992386752950134e-06, + "loss": 1.6461, + "step": 311 + }, + { + "epoch": 0.3204930662557781, + "grad_norm": 2.999209403991699, + "learning_rate": 4.990483441187667e-06, + "loss": 1.6861, + "step": 312 + }, + { + "epoch": 0.32152028762198254, + "grad_norm": 2.7288851737976074, + "learning_rate": 4.9885801294252e-06, + "loss": 1.7076, + "step": 313 + }, + { + "epoch": 0.32254750898818696, + "grad_norm": 2.8722457885742188, + "learning_rate": 4.986676817662733e-06, + "loss": 1.7079, + "step": 314 + }, + { + "epoch": 0.3235747303543914, + "grad_norm": 2.853344678878784, + "learning_rate": 4.9847735059002665e-06, + "loss": 1.6699, + "step": 315 + }, + { + "epoch": 0.3246019517205958, + "grad_norm": 2.7404720783233643, + "learning_rate": 4.9828701941378e-06, + "loss": 1.7423, + "step": 316 + }, + { + "epoch": 0.3256291730868002, + "grad_norm": 2.6062633991241455, + "learning_rate": 4.980966882375333e-06, + "loss": 1.6811, + "step": 317 + }, + { + "epoch": 0.3266563944530046, + "grad_norm": 2.834162712097168, + "learning_rate": 4.979063570612867e-06, + "loss": 1.6575, + "step": 318 + }, + { + "epoch": 0.327683615819209, + "grad_norm": 2.7741048336029053, + "learning_rate": 4.9771602588504e-06, + "loss": 1.6158, + "step": 319 + }, + { + "epoch": 0.32871083718541344, + "grad_norm": 2.6216888427734375, + "learning_rate": 4.975256947087933e-06, + "loss": 1.7295, + "step": 320 + }, + { + "epoch": 0.32871083718541344, + "eval_loss": 1.9309543371200562, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 320 + }, + { + "epoch": 0.32973805855161786, + "grad_norm": 2.6282548904418945, + "learning_rate": 4.973353635325467e-06, + "loss": 1.6266, + "step": 321 + }, + { + "epoch": 0.3307652799178223, + "grad_norm": 2.632124662399292, + "learning_rate": 4.971450323563e-06, + "loss": 1.6887, + "step": 322 + }, + { + "epoch": 0.3317925012840267, + "grad_norm": 2.5844857692718506, + "learning_rate": 4.969547011800533e-06, + "loss": 1.6287, + "step": 323 + }, + { + "epoch": 0.33281972265023113, + "grad_norm": 2.597620964050293, + "learning_rate": 4.967643700038067e-06, + "loss": 1.6876, + "step": 324 + }, + { + "epoch": 0.33384694401643555, + "grad_norm": 2.6529343128204346, + "learning_rate": 4.9657403882756e-06, + "loss": 1.6883, + "step": 325 + }, + { + "epoch": 0.33487416538264, + "grad_norm": 2.592874526977539, + "learning_rate": 4.9638370765131335e-06, + "loss": 1.653, + "step": 326 + }, + { + "epoch": 0.3359013867488444, + "grad_norm": 2.523489236831665, + "learning_rate": 4.961933764750667e-06, + "loss": 1.6501, + "step": 327 + }, + { + "epoch": 0.33692860811504877, + "grad_norm": 2.6721699237823486, + "learning_rate": 4.9600304529882e-06, + "loss": 1.7126, + "step": 328 + }, + { + "epoch": 0.3379558294812532, + "grad_norm": 2.513122797012329, + "learning_rate": 4.9581271412257335e-06, + "loss": 1.7268, + "step": 329 + }, + { + "epoch": 0.3389830508474576, + "grad_norm": 2.571118116378784, + "learning_rate": 4.956223829463266e-06, + "loss": 1.6862, + "step": 330 + }, + { + "epoch": 0.3389830508474576, + "eval_loss": 1.9501315355300903, + "eval_runtime": 37.3287, + "eval_samples_per_second": 26.789, + "eval_steps_per_second": 1.125, + "step": 330 + }, + { + "epoch": 0.34001027221366203, + "grad_norm": 2.5783371925354004, + "learning_rate": 4.954320517700799e-06, + "loss": 1.6041, + "step": 331 + }, + { + "epoch": 0.34103749357986646, + "grad_norm": 2.6829233169555664, + "learning_rate": 4.952417205938333e-06, + "loss": 1.7111, + "step": 332 + }, + { + "epoch": 0.3420647149460709, + "grad_norm": 2.6643247604370117, + "learning_rate": 4.950513894175866e-06, + "loss": 1.6797, + "step": 333 + }, + { + "epoch": 0.3430919363122753, + "grad_norm": 2.776308298110962, + "learning_rate": 4.9486105824133995e-06, + "loss": 1.6766, + "step": 334 + }, + { + "epoch": 0.3441191576784797, + "grad_norm": 2.722667932510376, + "learning_rate": 4.946707270650933e-06, + "loss": 1.6865, + "step": 335 + }, + { + "epoch": 0.34514637904468415, + "grad_norm": 2.7050845623016357, + "learning_rate": 4.944803958888466e-06, + "loss": 1.6967, + "step": 336 + }, + { + "epoch": 0.34617360041088857, + "grad_norm": 3.0009098052978516, + "learning_rate": 4.942900647126e-06, + "loss": 1.714, + "step": 337 + }, + { + "epoch": 0.34720082177709294, + "grad_norm": 2.793621301651001, + "learning_rate": 4.940997335363533e-06, + "loss": 1.6591, + "step": 338 + }, + { + "epoch": 0.34822804314329736, + "grad_norm": 2.844383478164673, + "learning_rate": 4.939094023601066e-06, + "loss": 1.611, + "step": 339 + }, + { + "epoch": 0.3492552645095018, + "grad_norm": 2.829378366470337, + "learning_rate": 4.9371907118386e-06, + "loss": 1.666, + "step": 340 + }, + { + "epoch": 0.3492552645095018, + "eval_loss": 1.96309494972229, + "eval_runtime": 37.3346, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 340 + }, + { + "epoch": 0.3502824858757062, + "grad_norm": 2.9702041149139404, + "learning_rate": 4.935287400076133e-06, + "loss": 1.6703, + "step": 341 + }, + { + "epoch": 0.35130970724191063, + "grad_norm": 2.7555935382843018, + "learning_rate": 4.933384088313666e-06, + "loss": 1.6879, + "step": 342 + }, + { + "epoch": 0.35233692860811505, + "grad_norm": 2.8968169689178467, + "learning_rate": 4.9314807765512e-06, + "loss": 1.6342, + "step": 343 + }, + { + "epoch": 0.3533641499743195, + "grad_norm": 3.010115385055542, + "learning_rate": 4.929577464788733e-06, + "loss": 1.5883, + "step": 344 + }, + { + "epoch": 0.3543913713405239, + "grad_norm": 2.6449785232543945, + "learning_rate": 4.9276741530262665e-06, + "loss": 1.654, + "step": 345 + }, + { + "epoch": 0.3554185927067283, + "grad_norm": 2.976109266281128, + "learning_rate": 4.9257708412638e-06, + "loss": 1.6483, + "step": 346 + }, + { + "epoch": 0.35644581407293274, + "grad_norm": 2.856144666671753, + "learning_rate": 4.923867529501332e-06, + "loss": 1.6536, + "step": 347 + }, + { + "epoch": 0.3574730354391371, + "grad_norm": 2.8142166137695312, + "learning_rate": 4.921964217738866e-06, + "loss": 1.6415, + "step": 348 + }, + { + "epoch": 0.35850025680534153, + "grad_norm": 2.70807147026062, + "learning_rate": 4.920060905976399e-06, + "loss": 1.7099, + "step": 349 + }, + { + "epoch": 0.35952747817154596, + "grad_norm": 2.7715506553649902, + "learning_rate": 4.9181575942139325e-06, + "loss": 1.6947, + "step": 350 + }, + { + "epoch": 0.35952747817154596, + "eval_loss": 1.9507063627243042, + "eval_runtime": 37.3094, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 350 + }, + { + "epoch": 0.3605546995377504, + "grad_norm": 2.8816769123077393, + "learning_rate": 4.916254282451466e-06, + "loss": 1.6906, + "step": 351 + }, + { + "epoch": 0.3615819209039548, + "grad_norm": 3.08797025680542, + "learning_rate": 4.914350970688999e-06, + "loss": 1.7387, + "step": 352 + }, + { + "epoch": 0.3626091422701592, + "grad_norm": 2.6018118858337402, + "learning_rate": 4.9124476589265325e-06, + "loss": 1.6319, + "step": 353 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7790255546569824, + "learning_rate": 4.910544347164066e-06, + "loss": 1.6357, + "step": 354 + }, + { + "epoch": 0.36466358500256807, + "grad_norm": 2.7360289096832275, + "learning_rate": 4.908641035401599e-06, + "loss": 1.7127, + "step": 355 + }, + { + "epoch": 0.3656908063687725, + "grad_norm": 2.6693854331970215, + "learning_rate": 4.906737723639133e-06, + "loss": 1.7747, + "step": 356 + }, + { + "epoch": 0.3667180277349769, + "grad_norm": 2.642868995666504, + "learning_rate": 4.904834411876665e-06, + "loss": 1.6401, + "step": 357 + }, + { + "epoch": 0.3677452491011813, + "grad_norm": 2.7230374813079834, + "learning_rate": 4.902931100114199e-06, + "loss": 1.6833, + "step": 358 + }, + { + "epoch": 0.3687724704673857, + "grad_norm": 2.499457359313965, + "learning_rate": 4.901027788351733e-06, + "loss": 1.6007, + "step": 359 + }, + { + "epoch": 0.3697996918335901, + "grad_norm": 2.7695276737213135, + "learning_rate": 4.899124476589266e-06, + "loss": 1.6486, + "step": 360 + }, + { + "epoch": 0.3697996918335901, + "eval_loss": 1.9434194564819336, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.37082691319979455, + "grad_norm": 2.5715794563293457, + "learning_rate": 4.8972211648267994e-06, + "loss": 1.6183, + "step": 361 + }, + { + "epoch": 0.371854134565999, + "grad_norm": 2.637859582901001, + "learning_rate": 4.895317853064333e-06, + "loss": 1.6648, + "step": 362 + }, + { + "epoch": 0.3728813559322034, + "grad_norm": 2.5601117610931396, + "learning_rate": 4.893414541301866e-06, + "loss": 1.6475, + "step": 363 + }, + { + "epoch": 0.3739085772984078, + "grad_norm": 2.429337739944458, + "learning_rate": 4.891511229539399e-06, + "loss": 1.6435, + "step": 364 + }, + { + "epoch": 0.37493579866461224, + "grad_norm": 2.6890885829925537, + "learning_rate": 4.889607917776932e-06, + "loss": 1.6577, + "step": 365 + }, + { + "epoch": 0.37596302003081666, + "grad_norm": 2.55492901802063, + "learning_rate": 4.887704606014465e-06, + "loss": 1.6523, + "step": 366 + }, + { + "epoch": 0.37699024139702103, + "grad_norm": 2.841362476348877, + "learning_rate": 4.885801294251999e-06, + "loss": 1.6742, + "step": 367 + }, + { + "epoch": 0.37801746276322545, + "grad_norm": 2.8041367530822754, + "learning_rate": 4.883897982489532e-06, + "loss": 1.5946, + "step": 368 + }, + { + "epoch": 0.3790446841294299, + "grad_norm": 2.7135822772979736, + "learning_rate": 4.8819946707270655e-06, + "loss": 1.6909, + "step": 369 + }, + { + "epoch": 0.3800719054956343, + "grad_norm": 2.9020211696624756, + "learning_rate": 4.880091358964599e-06, + "loss": 1.6804, + "step": 370 + }, + { + "epoch": 0.3800719054956343, + "eval_loss": 1.9353317022323608, + "eval_runtime": 37.3188, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 370 + }, + { + "epoch": 0.3810991268618387, + "grad_norm": 2.567310333251953, + "learning_rate": 4.878188047202132e-06, + "loss": 1.5885, + "step": 371 + }, + { + "epoch": 0.38212634822804314, + "grad_norm": 2.910834789276123, + "learning_rate": 4.876284735439666e-06, + "loss": 1.7681, + "step": 372 + }, + { + "epoch": 0.38315356959424757, + "grad_norm": 2.7507543563842773, + "learning_rate": 4.874381423677199e-06, + "loss": 1.7228, + "step": 373 + }, + { + "epoch": 0.384180790960452, + "grad_norm": 2.9230854511260986, + "learning_rate": 4.8724781119147315e-06, + "loss": 1.6491, + "step": 374 + }, + { + "epoch": 0.3852080123266564, + "grad_norm": 2.608980178833008, + "learning_rate": 4.870574800152265e-06, + "loss": 1.6757, + "step": 375 + }, + { + "epoch": 0.38623523369286084, + "grad_norm": 2.9557063579559326, + "learning_rate": 4.868671488389798e-06, + "loss": 1.607, + "step": 376 + }, + { + "epoch": 0.3872624550590652, + "grad_norm": 2.982243537902832, + "learning_rate": 4.8667681766273316e-06, + "loss": 1.6169, + "step": 377 + }, + { + "epoch": 0.3882896764252696, + "grad_norm": 2.775012254714966, + "learning_rate": 4.864864864864866e-06, + "loss": 1.5835, + "step": 378 + }, + { + "epoch": 0.38931689779147405, + "grad_norm": 2.9500741958618164, + "learning_rate": 4.862961553102399e-06, + "loss": 1.6504, + "step": 379 + }, + { + "epoch": 0.39034411915767847, + "grad_norm": 2.889702796936035, + "learning_rate": 4.8610582413399325e-06, + "loss": 1.6814, + "step": 380 + }, + { + "epoch": 0.39034411915767847, + "eval_loss": 1.9426418542861938, + "eval_runtime": 37.3276, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 380 + }, + { + "epoch": 0.3913713405238829, + "grad_norm": 2.5075154304504395, + "learning_rate": 4.859154929577465e-06, + "loss": 1.6368, + "step": 381 + }, + { + "epoch": 0.3923985618900873, + "grad_norm": 2.6461148262023926, + "learning_rate": 4.857251617814998e-06, + "loss": 1.6526, + "step": 382 + }, + { + "epoch": 0.39342578325629174, + "grad_norm": 2.681501626968384, + "learning_rate": 4.855348306052532e-06, + "loss": 1.6846, + "step": 383 + }, + { + "epoch": 0.39445300462249616, + "grad_norm": 2.557311773300171, + "learning_rate": 4.853444994290065e-06, + "loss": 1.6645, + "step": 384 + }, + { + "epoch": 0.3954802259887006, + "grad_norm": 2.6819186210632324, + "learning_rate": 4.8515416825275985e-06, + "loss": 1.6323, + "step": 385 + }, + { + "epoch": 0.396507447354905, + "grad_norm": 2.8803794384002686, + "learning_rate": 4.849638370765132e-06, + "loss": 1.6991, + "step": 386 + }, + { + "epoch": 0.3975346687211094, + "grad_norm": 2.7181358337402344, + "learning_rate": 4.847735059002665e-06, + "loss": 1.6256, + "step": 387 + }, + { + "epoch": 0.3985618900873138, + "grad_norm": 2.5668489933013916, + "learning_rate": 4.8458317472401985e-06, + "loss": 1.58, + "step": 388 + }, + { + "epoch": 0.3995891114535182, + "grad_norm": 2.708804130554199, + "learning_rate": 4.843928435477732e-06, + "loss": 1.67, + "step": 389 + }, + { + "epoch": 0.40061633281972264, + "grad_norm": 2.6565728187561035, + "learning_rate": 4.842025123715265e-06, + "loss": 1.622, + "step": 390 + }, + { + "epoch": 0.40061633281972264, + "eval_loss": 1.9529728889465332, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 390 + }, + { + "epoch": 0.40164355418592707, + "grad_norm": 2.8743324279785156, + "learning_rate": 4.840121811952798e-06, + "loss": 1.6618, + "step": 391 + }, + { + "epoch": 0.4026707755521315, + "grad_norm": 2.86134672164917, + "learning_rate": 4.838218500190331e-06, + "loss": 1.6544, + "step": 392 + }, + { + "epoch": 0.4036979969183359, + "grad_norm": 3.3492071628570557, + "learning_rate": 4.8363151884278645e-06, + "loss": 1.6683, + "step": 393 + }, + { + "epoch": 0.40472521828454033, + "grad_norm": 2.6630825996398926, + "learning_rate": 4.834411876665398e-06, + "loss": 1.6447, + "step": 394 + }, + { + "epoch": 0.40575243965074476, + "grad_norm": 2.820096492767334, + "learning_rate": 4.832508564902931e-06, + "loss": 1.6579, + "step": 395 + }, + { + "epoch": 0.4067796610169492, + "grad_norm": 3.076234817504883, + "learning_rate": 4.830605253140465e-06, + "loss": 1.6039, + "step": 396 + }, + { + "epoch": 0.40780688238315355, + "grad_norm": 2.667719841003418, + "learning_rate": 4.828701941377998e-06, + "loss": 1.6364, + "step": 397 + }, + { + "epoch": 0.40883410374935797, + "grad_norm": 3.2132771015167236, + "learning_rate": 4.826798629615531e-06, + "loss": 1.5906, + "step": 398 + }, + { + "epoch": 0.4098613251155624, + "grad_norm": 2.661484479904175, + "learning_rate": 4.824895317853065e-06, + "loss": 1.613, + "step": 399 + }, + { + "epoch": 0.4108885464817668, + "grad_norm": 3.002636671066284, + "learning_rate": 4.822992006090598e-06, + "loss": 1.6305, + "step": 400 + }, + { + "epoch": 0.4108885464817668, + "eval_loss": 1.9345510005950928, + "eval_runtime": 37.3369, + "eval_samples_per_second": 26.783, + "eval_steps_per_second": 1.125, + "step": 400 + }, + { + "epoch": 0.41191576784797124, + "grad_norm": 2.6405227184295654, + "learning_rate": 4.821088694328131e-06, + "loss": 1.5881, + "step": 401 + }, + { + "epoch": 0.41294298921417566, + "grad_norm": 3.006169080734253, + "learning_rate": 4.819185382565665e-06, + "loss": 1.6126, + "step": 402 + }, + { + "epoch": 0.4139702105803801, + "grad_norm": 2.7614858150482178, + "learning_rate": 4.817282070803198e-06, + "loss": 1.6695, + "step": 403 + }, + { + "epoch": 0.4149974319465845, + "grad_norm": 2.797260046005249, + "learning_rate": 4.8153787590407315e-06, + "loss": 1.6539, + "step": 404 + }, + { + "epoch": 0.41602465331278893, + "grad_norm": 2.8156301975250244, + "learning_rate": 4.813475447278265e-06, + "loss": 1.5845, + "step": 405 + }, + { + "epoch": 0.41705187467899335, + "grad_norm": 2.74100399017334, + "learning_rate": 4.811572135515798e-06, + "loss": 1.6699, + "step": 406 + }, + { + "epoch": 0.4180790960451977, + "grad_norm": 2.7785775661468506, + "learning_rate": 4.809668823753332e-06, + "loss": 1.5987, + "step": 407 + }, + { + "epoch": 0.41910631741140214, + "grad_norm": 2.663504123687744, + "learning_rate": 4.807765511990864e-06, + "loss": 1.5987, + "step": 408 + }, + { + "epoch": 0.42013353877760656, + "grad_norm": 2.903374433517456, + "learning_rate": 4.8058622002283975e-06, + "loss": 1.6302, + "step": 409 + }, + { + "epoch": 0.421160760143811, + "grad_norm": 2.696049690246582, + "learning_rate": 4.803958888465931e-06, + "loss": 1.671, + "step": 410 + }, + { + "epoch": 0.421160760143811, + "eval_loss": 1.9333878755569458, + "eval_runtime": 37.2472, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 410 + }, + { + "epoch": 0.4221879815100154, + "grad_norm": 2.486461877822876, + "learning_rate": 4.802055576703464e-06, + "loss": 1.6282, + "step": 411 + }, + { + "epoch": 0.42321520287621983, + "grad_norm": 2.6812057495117188, + "learning_rate": 4.8001522649409976e-06, + "loss": 1.7144, + "step": 412 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.8370249271392822, + "learning_rate": 4.798248953178531e-06, + "loss": 1.5979, + "step": 413 + }, + { + "epoch": 0.4252696456086287, + "grad_norm": 2.6073036193847656, + "learning_rate": 4.796345641416064e-06, + "loss": 1.5703, + "step": 414 + }, + { + "epoch": 0.4262968669748331, + "grad_norm": 2.5891220569610596, + "learning_rate": 4.794442329653598e-06, + "loss": 1.6212, + "step": 415 + }, + { + "epoch": 0.42732408834103747, + "grad_norm": 2.6296865940093994, + "learning_rate": 4.792539017891131e-06, + "loss": 1.6711, + "step": 416 + }, + { + "epoch": 0.4283513097072419, + "grad_norm": 2.573472499847412, + "learning_rate": 4.790635706128664e-06, + "loss": 1.5994, + "step": 417 + }, + { + "epoch": 0.4293785310734463, + "grad_norm": 2.734464168548584, + "learning_rate": 4.788732394366197e-06, + "loss": 1.6383, + "step": 418 + }, + { + "epoch": 0.43040575243965074, + "grad_norm": 2.6412851810455322, + "learning_rate": 4.786829082603731e-06, + "loss": 1.5646, + "step": 419 + }, + { + "epoch": 0.43143297380585516, + "grad_norm": 2.5065555572509766, + "learning_rate": 4.7849257708412645e-06, + "loss": 1.5636, + "step": 420 + }, + { + "epoch": 0.43143297380585516, + "eval_loss": 1.9357348680496216, + "eval_runtime": 37.372, + "eval_samples_per_second": 26.758, + "eval_steps_per_second": 1.124, + "step": 420 + }, + { + "epoch": 0.4324601951720596, + "grad_norm": 2.4889843463897705, + "learning_rate": 4.783022459078798e-06, + "loss": 1.6457, + "step": 421 + }, + { + "epoch": 0.433487416538264, + "grad_norm": 2.544801950454712, + "learning_rate": 4.781119147316331e-06, + "loss": 1.5897, + "step": 422 + }, + { + "epoch": 0.4345146379044684, + "grad_norm": 2.7834064960479736, + "learning_rate": 4.7792158355538645e-06, + "loss": 1.6174, + "step": 423 + }, + { + "epoch": 0.43554185927067285, + "grad_norm": 2.602952480316162, + "learning_rate": 4.777312523791398e-06, + "loss": 1.6152, + "step": 424 + }, + { + "epoch": 0.43656908063687727, + "grad_norm": 2.629746198654175, + "learning_rate": 4.77540921202893e-06, + "loss": 1.5991, + "step": 425 + }, + { + "epoch": 0.43759630200308164, + "grad_norm": 2.4987270832061768, + "learning_rate": 4.773505900266464e-06, + "loss": 1.6342, + "step": 426 + }, + { + "epoch": 0.43862352336928606, + "grad_norm": 2.607611894607544, + "learning_rate": 4.771602588503997e-06, + "loss": 1.6145, + "step": 427 + }, + { + "epoch": 0.4396507447354905, + "grad_norm": 2.5654876232147217, + "learning_rate": 4.7696992767415305e-06, + "loss": 1.6162, + "step": 428 + }, + { + "epoch": 0.4406779661016949, + "grad_norm": 2.5701870918273926, + "learning_rate": 4.767795964979064e-06, + "loss": 1.6516, + "step": 429 + }, + { + "epoch": 0.44170518746789933, + "grad_norm": 2.8381104469299316, + "learning_rate": 4.765892653216597e-06, + "loss": 1.659, + "step": 430 + }, + { + "epoch": 0.44170518746789933, + "eval_loss": 1.9259562492370605, + "eval_runtime": 37.3906, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 430 + }, + { + "epoch": 0.44273240883410375, + "grad_norm": 2.5777862071990967, + "learning_rate": 4.763989341454131e-06, + "loss": 1.6571, + "step": 431 + }, + { + "epoch": 0.4437596302003082, + "grad_norm": 2.6656861305236816, + "learning_rate": 4.762086029691664e-06, + "loss": 1.5786, + "step": 432 + }, + { + "epoch": 0.4447868515665126, + "grad_norm": 2.724851131439209, + "learning_rate": 4.760182717929197e-06, + "loss": 1.5916, + "step": 433 + }, + { + "epoch": 0.445814072932717, + "grad_norm": 2.532191514968872, + "learning_rate": 4.758279406166731e-06, + "loss": 1.608, + "step": 434 + }, + { + "epoch": 0.44684129429892144, + "grad_norm": 2.6252050399780273, + "learning_rate": 4.756376094404263e-06, + "loss": 1.5297, + "step": 435 + }, + { + "epoch": 0.4478685156651258, + "grad_norm": 2.674530029296875, + "learning_rate": 4.7544727826417966e-06, + "loss": 1.615, + "step": 436 + }, + { + "epoch": 0.44889573703133023, + "grad_norm": 2.9209136962890625, + "learning_rate": 4.75256947087933e-06, + "loss": 1.6749, + "step": 437 + }, + { + "epoch": 0.44992295839753466, + "grad_norm": 2.4959394931793213, + "learning_rate": 4.750666159116863e-06, + "loss": 1.6186, + "step": 438 + }, + { + "epoch": 0.4509501797637391, + "grad_norm": 2.6317503452301025, + "learning_rate": 4.7487628473543975e-06, + "loss": 1.5873, + "step": 439 + }, + { + "epoch": 0.4519774011299435, + "grad_norm": 2.6485610008239746, + "learning_rate": 4.746859535591931e-06, + "loss": 1.5451, + "step": 440 + }, + { + "epoch": 0.4519774011299435, + "eval_loss": 1.9235283136367798, + "eval_runtime": 37.367, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 440 + }, + { + "epoch": 0.4530046224961479, + "grad_norm": 2.491852045059204, + "learning_rate": 4.744956223829464e-06, + "loss": 1.6016, + "step": 441 + }, + { + "epoch": 0.45403184386235235, + "grad_norm": 2.5354514122009277, + "learning_rate": 4.743052912066997e-06, + "loss": 1.5795, + "step": 442 + }, + { + "epoch": 0.45505906522855677, + "grad_norm": 2.7756357192993164, + "learning_rate": 4.74114960030453e-06, + "loss": 1.6299, + "step": 443 + }, + { + "epoch": 0.4560862865947612, + "grad_norm": 2.638906955718994, + "learning_rate": 4.7392462885420635e-06, + "loss": 1.6429, + "step": 444 + }, + { + "epoch": 0.4571135079609656, + "grad_norm": 2.5858829021453857, + "learning_rate": 4.737342976779597e-06, + "loss": 1.5316, + "step": 445 + }, + { + "epoch": 0.45814072932717, + "grad_norm": 2.798567295074463, + "learning_rate": 4.73543966501713e-06, + "loss": 1.6166, + "step": 446 + }, + { + "epoch": 0.4591679506933744, + "grad_norm": 2.512143135070801, + "learning_rate": 4.7335363532546636e-06, + "loss": 1.5952, + "step": 447 + }, + { + "epoch": 0.4601951720595788, + "grad_norm": 2.7063655853271484, + "learning_rate": 4.731633041492197e-06, + "loss": 1.6051, + "step": 448 + }, + { + "epoch": 0.46122239342578325, + "grad_norm": 2.6358017921447754, + "learning_rate": 4.72972972972973e-06, + "loss": 1.6008, + "step": 449 + }, + { + "epoch": 0.4622496147919877, + "grad_norm": 2.75992751121521, + "learning_rate": 4.727826417967264e-06, + "loss": 1.644, + "step": 450 + }, + { + "epoch": 0.4622496147919877, + "eval_loss": 1.927345633506775, + "eval_runtime": 37.3707, + "eval_samples_per_second": 26.759, + "eval_steps_per_second": 1.124, + "step": 450 + }, + { + "epoch": 0.4632768361581921, + "grad_norm": 2.9103775024414062, + "learning_rate": 4.725923106204797e-06, + "loss": 1.6176, + "step": 451 + }, + { + "epoch": 0.4643040575243965, + "grad_norm": 2.627481460571289, + "learning_rate": 4.72401979444233e-06, + "loss": 1.586, + "step": 452 + }, + { + "epoch": 0.46533127889060094, + "grad_norm": 2.8432116508483887, + "learning_rate": 4.722116482679863e-06, + "loss": 1.6303, + "step": 453 + }, + { + "epoch": 0.46635850025680536, + "grad_norm": 2.8088958263397217, + "learning_rate": 4.720213170917396e-06, + "loss": 1.6163, + "step": 454 + }, + { + "epoch": 0.46738572162300973, + "grad_norm": 2.6040408611297607, + "learning_rate": 4.71830985915493e-06, + "loss": 1.6346, + "step": 455 + }, + { + "epoch": 0.46841294298921415, + "grad_norm": 2.8951191902160645, + "learning_rate": 4.716406547392463e-06, + "loss": 1.5859, + "step": 456 + }, + { + "epoch": 0.4694401643554186, + "grad_norm": 2.792473793029785, + "learning_rate": 4.714503235629996e-06, + "loss": 1.6441, + "step": 457 + }, + { + "epoch": 0.470467385721623, + "grad_norm": 2.6793878078460693, + "learning_rate": 4.71259992386753e-06, + "loss": 1.5699, + "step": 458 + }, + { + "epoch": 0.4714946070878274, + "grad_norm": 2.7223684787750244, + "learning_rate": 4.710696612105063e-06, + "loss": 1.6038, + "step": 459 + }, + { + "epoch": 0.47252182845403184, + "grad_norm": 2.8043856620788574, + "learning_rate": 4.708793300342596e-06, + "loss": 1.6018, + "step": 460 + }, + { + "epoch": 0.47252182845403184, + "eval_loss": 1.9213919639587402, + "eval_runtime": 37.3451, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 460 + }, + { + "epoch": 0.47354904982023627, + "grad_norm": 2.7657470703125, + "learning_rate": 4.70688998858013e-06, + "loss": 1.5896, + "step": 461 + }, + { + "epoch": 0.4745762711864407, + "grad_norm": 2.8417739868164062, + "learning_rate": 4.704986676817663e-06, + "loss": 1.5892, + "step": 462 + }, + { + "epoch": 0.4756034925526451, + "grad_norm": 2.5660197734832764, + "learning_rate": 4.7030833650551965e-06, + "loss": 1.615, + "step": 463 + }, + { + "epoch": 0.47663071391884954, + "grad_norm": 2.762880325317383, + "learning_rate": 4.70118005329273e-06, + "loss": 1.5747, + "step": 464 + }, + { + "epoch": 0.4776579352850539, + "grad_norm": 2.7191319465637207, + "learning_rate": 4.699276741530263e-06, + "loss": 1.6529, + "step": 465 + }, + { + "epoch": 0.4786851566512583, + "grad_norm": 2.4995453357696533, + "learning_rate": 4.697373429767797e-06, + "loss": 1.6285, + "step": 466 + }, + { + "epoch": 0.47971237801746275, + "grad_norm": 2.8300375938415527, + "learning_rate": 4.69547011800533e-06, + "loss": 1.6171, + "step": 467 + }, + { + "epoch": 0.48073959938366717, + "grad_norm": 2.6028811931610107, + "learning_rate": 4.693566806242863e-06, + "loss": 1.6014, + "step": 468 + }, + { + "epoch": 0.4817668207498716, + "grad_norm": 3.080156087875366, + "learning_rate": 4.691663494480397e-06, + "loss": 1.6055, + "step": 469 + }, + { + "epoch": 0.482794042116076, + "grad_norm": 2.6707916259765625, + "learning_rate": 4.689760182717929e-06, + "loss": 1.627, + "step": 470 + }, + { + "epoch": 0.482794042116076, + "eval_loss": 1.9015462398529053, + "eval_runtime": 37.3476, + "eval_samples_per_second": 26.776, + "eval_steps_per_second": 1.125, + "step": 470 + }, + { + "epoch": 0.48382126348228044, + "grad_norm": 2.6820614337921143, + "learning_rate": 4.6878568709554626e-06, + "loss": 1.5261, + "step": 471 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.841031789779663, + "learning_rate": 4.685953559192996e-06, + "loss": 1.5903, + "step": 472 + }, + { + "epoch": 0.4858757062146893, + "grad_norm": 2.470867156982422, + "learning_rate": 4.684050247430529e-06, + "loss": 1.6091, + "step": 473 + }, + { + "epoch": 0.4869029275808937, + "grad_norm": 2.4955153465270996, + "learning_rate": 4.682146935668063e-06, + "loss": 1.544, + "step": 474 + }, + { + "epoch": 0.4879301489470981, + "grad_norm": 2.9290695190429688, + "learning_rate": 4.680243623905596e-06, + "loss": 1.5893, + "step": 475 + }, + { + "epoch": 0.4889573703133025, + "grad_norm": 2.5902767181396484, + "learning_rate": 4.678340312143129e-06, + "loss": 1.6117, + "step": 476 + }, + { + "epoch": 0.4899845916795069, + "grad_norm": 2.6067540645599365, + "learning_rate": 4.676437000380663e-06, + "loss": 1.5703, + "step": 477 + }, + { + "epoch": 0.49101181304571134, + "grad_norm": 2.490105152130127, + "learning_rate": 4.674533688618196e-06, + "loss": 1.6931, + "step": 478 + }, + { + "epoch": 0.49203903441191577, + "grad_norm": 2.7736282348632812, + "learning_rate": 4.6726303768557295e-06, + "loss": 1.6204, + "step": 479 + }, + { + "epoch": 0.4930662557781202, + "grad_norm": 2.741196870803833, + "learning_rate": 4.670727065093263e-06, + "loss": 1.5331, + "step": 480 + }, + { + "epoch": 0.4930662557781202, + "eval_loss": 1.9095442295074463, + "eval_runtime": 37.3351, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 480 + }, + { + "epoch": 0.4940934771443246, + "grad_norm": 2.7624332904815674, + "learning_rate": 4.668823753330796e-06, + "loss": 1.5591, + "step": 481 + }, + { + "epoch": 0.49512069851052903, + "grad_norm": 2.86407470703125, + "learning_rate": 4.6669204415683295e-06, + "loss": 1.5844, + "step": 482 + }, + { + "epoch": 0.49614791987673346, + "grad_norm": 2.770030975341797, + "learning_rate": 4.665017129805863e-06, + "loss": 1.6515, + "step": 483 + }, + { + "epoch": 0.4971751412429379, + "grad_norm": 2.569862127304077, + "learning_rate": 4.663113818043396e-06, + "loss": 1.6228, + "step": 484 + }, + { + "epoch": 0.49820236260914225, + "grad_norm": 2.7176733016967773, + "learning_rate": 4.66121050628093e-06, + "loss": 1.5938, + "step": 485 + }, + { + "epoch": 0.49922958397534667, + "grad_norm": 2.4514758586883545, + "learning_rate": 4.659307194518463e-06, + "loss": 1.6307, + "step": 486 + }, + { + "epoch": 0.5002568053415511, + "grad_norm": 2.5439529418945312, + "learning_rate": 4.6574038827559955e-06, + "loss": 1.5855, + "step": 487 + }, + { + "epoch": 0.5012840267077555, + "grad_norm": 2.6395790576934814, + "learning_rate": 4.655500570993529e-06, + "loss": 1.6124, + "step": 488 + }, + { + "epoch": 0.50231124807396, + "grad_norm": 2.69413685798645, + "learning_rate": 4.653597259231062e-06, + "loss": 1.6358, + "step": 489 + }, + { + "epoch": 0.5033384694401644, + "grad_norm": 2.693129777908325, + "learning_rate": 4.651693947468596e-06, + "loss": 1.5218, + "step": 490 + }, + { + "epoch": 0.5033384694401644, + "eval_loss": 1.9201244115829468, + "eval_runtime": 37.3324, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 490 + }, + { + "epoch": 0.5043656908063687, + "grad_norm": 2.630596876144409, + "learning_rate": 4.649790635706129e-06, + "loss": 1.6719, + "step": 491 + }, + { + "epoch": 0.5053929121725732, + "grad_norm": 2.7230262756347656, + "learning_rate": 4.647887323943662e-06, + "loss": 1.5705, + "step": 492 + }, + { + "epoch": 0.5064201335387776, + "grad_norm": 2.529944896697998, + "learning_rate": 4.645984012181196e-06, + "loss": 1.5767, + "step": 493 + }, + { + "epoch": 0.507447354904982, + "grad_norm": 2.546849489212036, + "learning_rate": 4.644080700418729e-06, + "loss": 1.5039, + "step": 494 + }, + { + "epoch": 0.5084745762711864, + "grad_norm": 2.6698246002197266, + "learning_rate": 4.642177388656262e-06, + "loss": 1.6229, + "step": 495 + }, + { + "epoch": 0.5095017976373909, + "grad_norm": 2.7093148231506348, + "learning_rate": 4.640274076893796e-06, + "loss": 1.5797, + "step": 496 + }, + { + "epoch": 0.5105290190035953, + "grad_norm": 2.48629093170166, + "learning_rate": 4.638370765131328e-06, + "loss": 1.6366, + "step": 497 + }, + { + "epoch": 0.5115562403697997, + "grad_norm": 2.5818612575531006, + "learning_rate": 4.636467453368862e-06, + "loss": 1.5608, + "step": 498 + }, + { + "epoch": 0.5125834617360041, + "grad_norm": 2.6263651847839355, + "learning_rate": 4.634564141606396e-06, + "loss": 1.5518, + "step": 499 + }, + { + "epoch": 0.5136106831022085, + "grad_norm": 2.5724668502807617, + "learning_rate": 4.632660829843929e-06, + "loss": 1.6195, + "step": 500 + }, + { + "epoch": 0.5136106831022085, + "eval_loss": 1.9094111919403076, + "eval_runtime": 37.3483, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 500 + }, + { + "epoch": 0.514637904468413, + "grad_norm": 2.5028088092803955, + "learning_rate": 4.630757518081463e-06, + "loss": 1.5454, + "step": 501 + }, + { + "epoch": 0.5156651258346173, + "grad_norm": 2.588115692138672, + "learning_rate": 4.628854206318996e-06, + "loss": 1.6676, + "step": 502 + }, + { + "epoch": 0.5166923472008218, + "grad_norm": 2.9701285362243652, + "learning_rate": 4.626950894556529e-06, + "loss": 1.5822, + "step": 503 + }, + { + "epoch": 0.5177195685670262, + "grad_norm": 2.593867301940918, + "learning_rate": 4.625047582794062e-06, + "loss": 1.6091, + "step": 504 + }, + { + "epoch": 0.5187467899332306, + "grad_norm": 2.559642791748047, + "learning_rate": 4.623144271031595e-06, + "loss": 1.5359, + "step": 505 + }, + { + "epoch": 0.519774011299435, + "grad_norm": 2.4158153533935547, + "learning_rate": 4.6212409592691286e-06, + "loss": 1.5476, + "step": 506 + }, + { + "epoch": 0.5208012326656395, + "grad_norm": 2.666099786758423, + "learning_rate": 4.619337647506662e-06, + "loss": 1.6297, + "step": 507 + }, + { + "epoch": 0.5218284540318439, + "grad_norm": 2.7026472091674805, + "learning_rate": 4.617434335744195e-06, + "loss": 1.5891, + "step": 508 + }, + { + "epoch": 0.5228556753980482, + "grad_norm": 2.4591612815856934, + "learning_rate": 4.615531023981729e-06, + "loss": 1.5546, + "step": 509 + }, + { + "epoch": 0.5238828967642527, + "grad_norm": 2.870255708694458, + "learning_rate": 4.613627712219262e-06, + "loss": 1.5817, + "step": 510 + }, + { + "epoch": 0.5238828967642527, + "eval_loss": 1.910629153251648, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 510 + }, + { + "epoch": 0.5249101181304571, + "grad_norm": 2.653801679611206, + "learning_rate": 4.611724400456795e-06, + "loss": 1.586, + "step": 511 + }, + { + "epoch": 0.5259373394966615, + "grad_norm": 2.4235377311706543, + "learning_rate": 4.609821088694329e-06, + "loss": 1.5886, + "step": 512 + }, + { + "epoch": 0.5269645608628659, + "grad_norm": 2.8782477378845215, + "learning_rate": 4.607917776931862e-06, + "loss": 1.5634, + "step": 513 + }, + { + "epoch": 0.5279917822290704, + "grad_norm": 2.792876958847046, + "learning_rate": 4.606014465169395e-06, + "loss": 1.6073, + "step": 514 + }, + { + "epoch": 0.5290190035952748, + "grad_norm": 2.5579030513763428, + "learning_rate": 4.604111153406928e-06, + "loss": 1.6161, + "step": 515 + }, + { + "epoch": 0.5300462249614792, + "grad_norm": 2.6486189365386963, + "learning_rate": 4.602207841644461e-06, + "loss": 1.5297, + "step": 516 + }, + { + "epoch": 0.5310734463276836, + "grad_norm": 2.9047019481658936, + "learning_rate": 4.600304529881995e-06, + "loss": 1.626, + "step": 517 + }, + { + "epoch": 0.5321006676938881, + "grad_norm": 2.9164223670959473, + "learning_rate": 4.598401218119528e-06, + "loss": 1.6051, + "step": 518 + }, + { + "epoch": 0.5331278890600925, + "grad_norm": 2.544311046600342, + "learning_rate": 4.5964979063570614e-06, + "loss": 1.6247, + "step": 519 + }, + { + "epoch": 0.5341551104262968, + "grad_norm": 2.8632311820983887, + "learning_rate": 4.594594594594596e-06, + "loss": 1.5615, + "step": 520 + }, + { + "epoch": 0.5341551104262968, + "eval_loss": 1.9171669483184814, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 520 + }, + { + "epoch": 0.5351823317925013, + "grad_norm": 2.507697105407715, + "learning_rate": 4.592691282832128e-06, + "loss": 1.5929, + "step": 521 + }, + { + "epoch": 0.5362095531587057, + "grad_norm": 2.657752752304077, + "learning_rate": 4.5907879710696615e-06, + "loss": 1.6159, + "step": 522 + }, + { + "epoch": 0.5372367745249101, + "grad_norm": 2.32254695892334, + "learning_rate": 4.588884659307195e-06, + "loss": 1.5318, + "step": 523 + }, + { + "epoch": 0.5382639958911145, + "grad_norm": 2.5979015827178955, + "learning_rate": 4.586981347544728e-06, + "loss": 1.5999, + "step": 524 + }, + { + "epoch": 0.539291217257319, + "grad_norm": 2.839956521987915, + "learning_rate": 4.585078035782262e-06, + "loss": 1.547, + "step": 525 + }, + { + "epoch": 0.5403184386235234, + "grad_norm": 2.700131893157959, + "learning_rate": 4.583174724019795e-06, + "loss": 1.5831, + "step": 526 + }, + { + "epoch": 0.5413456599897278, + "grad_norm": 2.365183115005493, + "learning_rate": 4.581271412257328e-06, + "loss": 1.5357, + "step": 527 + }, + { + "epoch": 0.5423728813559322, + "grad_norm": 2.675567865371704, + "learning_rate": 4.579368100494862e-06, + "loss": 1.5212, + "step": 528 + }, + { + "epoch": 0.5434001027221366, + "grad_norm": 2.9096713066101074, + "learning_rate": 4.577464788732395e-06, + "loss": 1.6196, + "step": 529 + }, + { + "epoch": 0.544427324088341, + "grad_norm": 2.6186270713806152, + "learning_rate": 4.575561476969928e-06, + "loss": 1.5895, + "step": 530 + }, + { + "epoch": 0.544427324088341, + "eval_loss": 1.926998496055603, + "eval_runtime": 37.2248, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 530 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.7291345596313477, + "learning_rate": 4.573658165207461e-06, + "loss": 1.6335, + "step": 531 + }, + { + "epoch": 0.5464817668207499, + "grad_norm": 2.7084522247314453, + "learning_rate": 4.571754853444994e-06, + "loss": 1.5968, + "step": 532 + }, + { + "epoch": 0.5475089881869543, + "grad_norm": 2.699721336364746, + "learning_rate": 4.569851541682528e-06, + "loss": 1.5775, + "step": 533 + }, + { + "epoch": 0.5485362095531587, + "grad_norm": 2.602526903152466, + "learning_rate": 4.567948229920061e-06, + "loss": 1.5497, + "step": 534 + }, + { + "epoch": 0.5495634309193631, + "grad_norm": 2.597362518310547, + "learning_rate": 4.566044918157594e-06, + "loss": 1.5218, + "step": 535 + }, + { + "epoch": 0.5505906522855676, + "grad_norm": 2.6624414920806885, + "learning_rate": 4.564141606395128e-06, + "loss": 1.5553, + "step": 536 + }, + { + "epoch": 0.551617873651772, + "grad_norm": 2.750807285308838, + "learning_rate": 4.562238294632661e-06, + "loss": 1.6078, + "step": 537 + }, + { + "epoch": 0.5526450950179764, + "grad_norm": 2.7480952739715576, + "learning_rate": 4.5603349828701945e-06, + "loss": 1.5707, + "step": 538 + }, + { + "epoch": 0.5536723163841808, + "grad_norm": 2.6287965774536133, + "learning_rate": 4.558431671107728e-06, + "loss": 1.5961, + "step": 539 + }, + { + "epoch": 0.5546995377503852, + "grad_norm": 2.721311569213867, + "learning_rate": 4.556528359345261e-06, + "loss": 1.6335, + "step": 540 + }, + { + "epoch": 0.5546995377503852, + "eval_loss": 1.9290324449539185, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5557267591165896, + "grad_norm": 2.470325469970703, + "learning_rate": 4.5546250475827946e-06, + "loss": 1.5645, + "step": 541 + }, + { + "epoch": 0.556753980482794, + "grad_norm": 2.6089508533477783, + "learning_rate": 4.552721735820328e-06, + "loss": 1.5256, + "step": 542 + }, + { + "epoch": 0.5577812018489985, + "grad_norm": 2.541785478591919, + "learning_rate": 4.550818424057861e-06, + "loss": 1.5641, + "step": 543 + }, + { + "epoch": 0.5588084232152029, + "grad_norm": 2.6722488403320312, + "learning_rate": 4.548915112295395e-06, + "loss": 1.5708, + "step": 544 + }, + { + "epoch": 0.5598356445814073, + "grad_norm": 2.583695650100708, + "learning_rate": 4.547011800532928e-06, + "loss": 1.5799, + "step": 545 + }, + { + "epoch": 0.5608628659476117, + "grad_norm": 2.956885576248169, + "learning_rate": 4.545108488770461e-06, + "loss": 1.5345, + "step": 546 + }, + { + "epoch": 0.5618900873138162, + "grad_norm": 2.5534746646881104, + "learning_rate": 4.543205177007995e-06, + "loss": 1.636, + "step": 547 + }, + { + "epoch": 0.5629173086800205, + "grad_norm": 2.8133325576782227, + "learning_rate": 4.541301865245527e-06, + "loss": 1.5678, + "step": 548 + }, + { + "epoch": 0.5639445300462249, + "grad_norm": 2.5969228744506836, + "learning_rate": 4.539398553483061e-06, + "loss": 1.5975, + "step": 549 + }, + { + "epoch": 0.5649717514124294, + "grad_norm": 2.7594687938690186, + "learning_rate": 4.537495241720594e-06, + "loss": 1.5622, + "step": 550 + }, + { + "epoch": 0.5649717514124294, + "eval_loss": 1.8870681524276733, + "eval_runtime": 37.2094, + "eval_samples_per_second": 26.875, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.5659989727786338, + "grad_norm": 2.7413411140441895, + "learning_rate": 4.535591929958127e-06, + "loss": 1.6031, + "step": 551 + }, + { + "epoch": 0.5670261941448382, + "grad_norm": 2.6182661056518555, + "learning_rate": 4.533688618195661e-06, + "loss": 1.5828, + "step": 552 + }, + { + "epoch": 0.5680534155110426, + "grad_norm": 2.5016422271728516, + "learning_rate": 4.531785306433194e-06, + "loss": 1.5712, + "step": 553 + }, + { + "epoch": 0.5690806368772471, + "grad_norm": 2.7788732051849365, + "learning_rate": 4.529881994670727e-06, + "loss": 1.5472, + "step": 554 + }, + { + "epoch": 0.5701078582434514, + "grad_norm": 2.7602436542510986, + "learning_rate": 4.527978682908261e-06, + "loss": 1.5526, + "step": 555 + }, + { + "epoch": 0.5711350796096559, + "grad_norm": 2.5898544788360596, + "learning_rate": 4.526075371145794e-06, + "loss": 1.5644, + "step": 556 + }, + { + "epoch": 0.5721623009758603, + "grad_norm": 2.439344882965088, + "learning_rate": 4.5241720593833275e-06, + "loss": 1.5892, + "step": 557 + }, + { + "epoch": 0.5731895223420647, + "grad_norm": 2.6324563026428223, + "learning_rate": 4.52226874762086e-06, + "loss": 1.5729, + "step": 558 + }, + { + "epoch": 0.5742167437082691, + "grad_norm": 2.674042224884033, + "learning_rate": 4.520365435858393e-06, + "loss": 1.6016, + "step": 559 + }, + { + "epoch": 0.5752439650744735, + "grad_norm": 2.692619800567627, + "learning_rate": 4.518462124095928e-06, + "loss": 1.6229, + "step": 560 + }, + { + "epoch": 0.5752439650744735, + "eval_loss": 1.8689097166061401, + "eval_runtime": 37.2177, + "eval_samples_per_second": 26.869, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.576271186440678, + "grad_norm": 2.636924982070923, + "learning_rate": 4.516558812333461e-06, + "loss": 1.5748, + "step": 561 + }, + { + "epoch": 0.5772984078068824, + "grad_norm": 2.549285888671875, + "learning_rate": 4.514655500570994e-06, + "loss": 1.5656, + "step": 562 + }, + { + "epoch": 0.5783256291730868, + "grad_norm": 2.5873067378997803, + "learning_rate": 4.512752188808528e-06, + "loss": 1.631, + "step": 563 + }, + { + "epoch": 0.5793528505392912, + "grad_norm": 2.552121162414551, + "learning_rate": 4.510848877046061e-06, + "loss": 1.6334, + "step": 564 + }, + { + "epoch": 0.5803800719054957, + "grad_norm": 2.697615146636963, + "learning_rate": 4.5089455652835936e-06, + "loss": 1.5951, + "step": 565 + }, + { + "epoch": 0.5814072932717, + "grad_norm": 2.64585280418396, + "learning_rate": 4.507042253521127e-06, + "loss": 1.556, + "step": 566 + }, + { + "epoch": 0.5824345146379045, + "grad_norm": 2.6383707523345947, + "learning_rate": 4.50513894175866e-06, + "loss": 1.5184, + "step": 567 + }, + { + "epoch": 0.5834617360041089, + "grad_norm": 2.7571511268615723, + "learning_rate": 4.503235629996194e-06, + "loss": 1.5984, + "step": 568 + }, + { + "epoch": 0.5844889573703133, + "grad_norm": 2.6637237071990967, + "learning_rate": 4.501332318233727e-06, + "loss": 1.5989, + "step": 569 + }, + { + "epoch": 0.5855161787365177, + "grad_norm": 2.725754976272583, + "learning_rate": 4.49942900647126e-06, + "loss": 1.5594, + "step": 570 + }, + { + "epoch": 0.5855161787365177, + "eval_loss": 1.8722314834594727, + "eval_runtime": 37.1859, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 570 + }, + { + "epoch": 0.5865434001027221, + "grad_norm": 2.6657378673553467, + "learning_rate": 4.497525694708794e-06, + "loss": 1.5954, + "step": 571 + }, + { + "epoch": 0.5875706214689266, + "grad_norm": 2.78940486907959, + "learning_rate": 4.495622382946327e-06, + "loss": 1.592, + "step": 572 + }, + { + "epoch": 0.588597842835131, + "grad_norm": 2.580956220626831, + "learning_rate": 4.4937190711838605e-06, + "loss": 1.5444, + "step": 573 + }, + { + "epoch": 0.5896250642013354, + "grad_norm": 2.7783710956573486, + "learning_rate": 4.491815759421394e-06, + "loss": 1.6257, + "step": 574 + }, + { + "epoch": 0.5906522855675398, + "grad_norm": 2.9381537437438965, + "learning_rate": 4.489912447658926e-06, + "loss": 1.5489, + "step": 575 + }, + { + "epoch": 0.5916795069337443, + "grad_norm": 2.7915165424346924, + "learning_rate": 4.48800913589646e-06, + "loss": 1.5471, + "step": 576 + }, + { + "epoch": 0.5927067282999486, + "grad_norm": 2.5928280353546143, + "learning_rate": 4.486105824133993e-06, + "loss": 1.5864, + "step": 577 + }, + { + "epoch": 0.593733949666153, + "grad_norm": 2.844512462615967, + "learning_rate": 4.4842025123715264e-06, + "loss": 1.5841, + "step": 578 + }, + { + "epoch": 0.5947611710323575, + "grad_norm": 2.562718391418457, + "learning_rate": 4.48229920060906e-06, + "loss": 1.5476, + "step": 579 + }, + { + "epoch": 0.5957883923985619, + "grad_norm": 2.535759925842285, + "learning_rate": 4.480395888846594e-06, + "loss": 1.5274, + "step": 580 + }, + { + "epoch": 0.5957883923985619, + "eval_loss": 1.8881579637527466, + "eval_runtime": 37.2533, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.5968156137647663, + "grad_norm": 2.6940653324127197, + "learning_rate": 4.478492577084127e-06, + "loss": 1.6377, + "step": 581 + }, + { + "epoch": 0.5978428351309707, + "grad_norm": 2.4839298725128174, + "learning_rate": 4.47658926532166e-06, + "loss": 1.534, + "step": 582 + }, + { + "epoch": 0.5988700564971752, + "grad_norm": 2.5035688877105713, + "learning_rate": 4.474685953559193e-06, + "loss": 1.5764, + "step": 583 + }, + { + "epoch": 0.5998972778633795, + "grad_norm": 2.791907548904419, + "learning_rate": 4.472782641796727e-06, + "loss": 1.704, + "step": 584 + }, + { + "epoch": 0.600924499229584, + "grad_norm": 2.5384254455566406, + "learning_rate": 4.47087933003426e-06, + "loss": 1.5184, + "step": 585 + }, + { + "epoch": 0.6019517205957884, + "grad_norm": 2.4990930557250977, + "learning_rate": 4.468976018271793e-06, + "loss": 1.5335, + "step": 586 + }, + { + "epoch": 0.6029789419619929, + "grad_norm": 2.470872163772583, + "learning_rate": 4.467072706509327e-06, + "loss": 1.5464, + "step": 587 + }, + { + "epoch": 0.6040061633281972, + "grad_norm": 2.539036989212036, + "learning_rate": 4.46516939474686e-06, + "loss": 1.5793, + "step": 588 + }, + { + "epoch": 0.6050333846944016, + "grad_norm": 2.495309829711914, + "learning_rate": 4.463266082984393e-06, + "loss": 1.5312, + "step": 589 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.5937912464141846, + "learning_rate": 4.461362771221927e-06, + "loss": 1.6508, + "step": 590 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.8893816471099854, + "eval_runtime": 37.2539, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 590 + }, + { + "epoch": 0.6070878274268104, + "grad_norm": 2.53300404548645, + "learning_rate": 4.45945945945946e-06, + "loss": 1.5408, + "step": 591 + }, + { + "epoch": 0.6081150487930149, + "grad_norm": 2.5760276317596436, + "learning_rate": 4.457556147696993e-06, + "loss": 1.5575, + "step": 592 + }, + { + "epoch": 0.6091422701592193, + "grad_norm": 2.638353109359741, + "learning_rate": 4.455652835934526e-06, + "loss": 1.6286, + "step": 593 + }, + { + "epoch": 0.6101694915254238, + "grad_norm": 2.916550397872925, + "learning_rate": 4.453749524172059e-06, + "loss": 1.6334, + "step": 594 + }, + { + "epoch": 0.6111967128916281, + "grad_norm": 2.560878038406372, + "learning_rate": 4.451846212409593e-06, + "loss": 1.5806, + "step": 595 + }, + { + "epoch": 0.6122239342578326, + "grad_norm": 2.637882947921753, + "learning_rate": 4.449942900647126e-06, + "loss": 1.5791, + "step": 596 + }, + { + "epoch": 0.613251155624037, + "grad_norm": 2.752870798110962, + "learning_rate": 4.4480395888846595e-06, + "loss": 1.5117, + "step": 597 + }, + { + "epoch": 0.6142783769902413, + "grad_norm": 2.702087640762329, + "learning_rate": 4.446136277122193e-06, + "loss": 1.5783, + "step": 598 + }, + { + "epoch": 0.6153055983564458, + "grad_norm": 2.504819631576538, + "learning_rate": 4.444232965359726e-06, + "loss": 1.5136, + "step": 599 + }, + { + "epoch": 0.6163328197226502, + "grad_norm": 2.521670341491699, + "learning_rate": 4.4423296535972596e-06, + "loss": 1.5741, + "step": 600 + }, + { + "epoch": 0.6163328197226502, + "eval_loss": 1.8973714113235474, + "eval_runtime": 37.3167, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 600 + }, + { + "epoch": 0.6173600410888547, + "grad_norm": 2.927755355834961, + "learning_rate": 4.440426341834793e-06, + "loss": 1.5262, + "step": 601 + }, + { + "epoch": 0.618387262455059, + "grad_norm": 2.477053642272949, + "learning_rate": 4.438523030072326e-06, + "loss": 1.572, + "step": 602 + }, + { + "epoch": 0.6194144838212635, + "grad_norm": 2.6500253677368164, + "learning_rate": 4.43661971830986e-06, + "loss": 1.596, + "step": 603 + }, + { + "epoch": 0.6204417051874679, + "grad_norm": 2.5134825706481934, + "learning_rate": 4.434716406547393e-06, + "loss": 1.5432, + "step": 604 + }, + { + "epoch": 0.6214689265536724, + "grad_norm": 2.4142072200775146, + "learning_rate": 4.432813094784926e-06, + "loss": 1.5391, + "step": 605 + }, + { + "epoch": 0.6224961479198767, + "grad_norm": 2.752556800842285, + "learning_rate": 4.43090978302246e-06, + "loss": 1.6055, + "step": 606 + }, + { + "epoch": 0.6235233692860811, + "grad_norm": 2.5731008052825928, + "learning_rate": 4.429006471259993e-06, + "loss": 1.5648, + "step": 607 + }, + { + "epoch": 0.6245505906522856, + "grad_norm": 2.531656503677368, + "learning_rate": 4.4271031594975265e-06, + "loss": 1.5492, + "step": 608 + }, + { + "epoch": 0.6255778120184899, + "grad_norm": 2.6665475368499756, + "learning_rate": 4.425199847735059e-06, + "loss": 1.5194, + "step": 609 + }, + { + "epoch": 0.6266050333846944, + "grad_norm": 2.437051773071289, + "learning_rate": 4.423296535972592e-06, + "loss": 1.5271, + "step": 610 + }, + { + "epoch": 0.6266050333846944, + "eval_loss": 1.88953697681427, + "eval_runtime": 36.8703, + "eval_samples_per_second": 27.122, + "eval_steps_per_second": 1.139, + "step": 610 + }, + { + "epoch": 0.6276322547508988, + "grad_norm": 2.5932440757751465, + "learning_rate": 4.421393224210126e-06, + "loss": 1.5563, + "step": 611 + }, + { + "epoch": 0.6286594761171033, + "grad_norm": 2.589447021484375, + "learning_rate": 4.419489912447659e-06, + "loss": 1.5207, + "step": 612 + }, + { + "epoch": 0.6296866974833076, + "grad_norm": 2.570096731185913, + "learning_rate": 4.4175866006851924e-06, + "loss": 1.5995, + "step": 613 + }, + { + "epoch": 0.6307139188495121, + "grad_norm": 2.5356791019439697, + "learning_rate": 4.415683288922726e-06, + "loss": 1.5703, + "step": 614 + }, + { + "epoch": 0.6317411402157165, + "grad_norm": 2.705193281173706, + "learning_rate": 4.413779977160259e-06, + "loss": 1.5405, + "step": 615 + }, + { + "epoch": 0.632768361581921, + "grad_norm": 2.555168390274048, + "learning_rate": 4.4118766653977925e-06, + "loss": 1.5487, + "step": 616 + }, + { + "epoch": 0.6337955829481253, + "grad_norm": 2.804696559906006, + "learning_rate": 4.409973353635326e-06, + "loss": 1.6094, + "step": 617 + }, + { + "epoch": 0.6348228043143297, + "grad_norm": 2.6889376640319824, + "learning_rate": 4.408070041872859e-06, + "loss": 1.5656, + "step": 618 + }, + { + "epoch": 0.6358500256805342, + "grad_norm": 2.576746702194214, + "learning_rate": 4.406166730110393e-06, + "loss": 1.5244, + "step": 619 + }, + { + "epoch": 0.6368772470467385, + "grad_norm": 2.811156749725342, + "learning_rate": 4.404263418347925e-06, + "loss": 1.5153, + "step": 620 + }, + { + "epoch": 0.6368772470467385, + "eval_loss": 1.8949775695800781, + "eval_runtime": 36.978, + "eval_samples_per_second": 27.043, + "eval_steps_per_second": 1.136, + "step": 620 + }, + { + "epoch": 0.637904468412943, + "grad_norm": 2.6560428142547607, + "learning_rate": 4.402360106585459e-06, + "loss": 1.5347, + "step": 621 + }, + { + "epoch": 0.6389316897791474, + "grad_norm": 2.668100118637085, + "learning_rate": 4.400456794822993e-06, + "loss": 1.5236, + "step": 622 + }, + { + "epoch": 0.6399589111453519, + "grad_norm": 2.925978660583496, + "learning_rate": 4.398553483060526e-06, + "loss": 1.5494, + "step": 623 + }, + { + "epoch": 0.6409861325115562, + "grad_norm": 2.5964415073394775, + "learning_rate": 4.396650171298059e-06, + "loss": 1.522, + "step": 624 + }, + { + "epoch": 0.6420133538777607, + "grad_norm": 2.643888235092163, + "learning_rate": 4.394746859535593e-06, + "loss": 1.5332, + "step": 625 + }, + { + "epoch": 0.6430405752439651, + "grad_norm": 2.805192708969116, + "learning_rate": 4.392843547773125e-06, + "loss": 1.5031, + "step": 626 + }, + { + "epoch": 0.6440677966101694, + "grad_norm": 2.6862149238586426, + "learning_rate": 4.390940236010659e-06, + "loss": 1.5335, + "step": 627 + }, + { + "epoch": 0.6450950179763739, + "grad_norm": 2.5579185485839844, + "learning_rate": 4.389036924248192e-06, + "loss": 1.5466, + "step": 628 + }, + { + "epoch": 0.6461222393425783, + "grad_norm": 2.4550797939300537, + "learning_rate": 4.387133612485725e-06, + "loss": 1.5776, + "step": 629 + }, + { + "epoch": 0.6471494607087828, + "grad_norm": 2.6051297187805176, + "learning_rate": 4.385230300723259e-06, + "loss": 1.5435, + "step": 630 + }, + { + "epoch": 0.6471494607087828, + "eval_loss": 1.8860505819320679, + "eval_runtime": 36.989, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 630 + }, + { + "epoch": 0.6481766820749871, + "grad_norm": 2.3723294734954834, + "learning_rate": 4.383326988960792e-06, + "loss": 1.5413, + "step": 631 + }, + { + "epoch": 0.6492039034411916, + "grad_norm": 2.44370174407959, + "learning_rate": 4.3814236771983255e-06, + "loss": 1.5152, + "step": 632 + }, + { + "epoch": 0.650231124807396, + "grad_norm": 2.474081039428711, + "learning_rate": 4.379520365435859e-06, + "loss": 1.579, + "step": 633 + }, + { + "epoch": 0.6512583461736005, + "grad_norm": 2.837843179702759, + "learning_rate": 4.377617053673392e-06, + "loss": 1.559, + "step": 634 + }, + { + "epoch": 0.6522855675398048, + "grad_norm": 2.593322515487671, + "learning_rate": 4.3757137419109256e-06, + "loss": 1.552, + "step": 635 + }, + { + "epoch": 0.6533127889060092, + "grad_norm": 3.018051862716675, + "learning_rate": 4.373810430148459e-06, + "loss": 1.6231, + "step": 636 + }, + { + "epoch": 0.6543400102722137, + "grad_norm": 2.6338863372802734, + "learning_rate": 4.3719071183859914e-06, + "loss": 1.517, + "step": 637 + }, + { + "epoch": 0.655367231638418, + "grad_norm": 2.544140577316284, + "learning_rate": 4.370003806623525e-06, + "loss": 1.5763, + "step": 638 + }, + { + "epoch": 0.6563944530046225, + "grad_norm": 2.6375792026519775, + "learning_rate": 4.368100494861058e-06, + "loss": 1.5098, + "step": 639 + }, + { + "epoch": 0.6574216743708269, + "grad_norm": 2.5452051162719727, + "learning_rate": 4.3661971830985915e-06, + "loss": 1.5671, + "step": 640 + }, + { + "epoch": 0.6574216743708269, + "eval_loss": 1.881069540977478, + "eval_runtime": 36.9961, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 640 + }, + { + "epoch": 0.6584488957370314, + "grad_norm": 2.623411178588867, + "learning_rate": 4.364293871336126e-06, + "loss": 1.5215, + "step": 641 + }, + { + "epoch": 0.6594761171032357, + "grad_norm": 2.6149892807006836, + "learning_rate": 4.362390559573659e-06, + "loss": 1.5507, + "step": 642 + }, + { + "epoch": 0.6605033384694402, + "grad_norm": 2.5570266246795654, + "learning_rate": 4.3604872478111925e-06, + "loss": 1.5143, + "step": 643 + }, + { + "epoch": 0.6615305598356446, + "grad_norm": 2.4813005924224854, + "learning_rate": 4.358583936048725e-06, + "loss": 1.4518, + "step": 644 + }, + { + "epoch": 0.662557781201849, + "grad_norm": 3.0318472385406494, + "learning_rate": 4.356680624286258e-06, + "loss": 1.5812, + "step": 645 + }, + { + "epoch": 0.6635850025680534, + "grad_norm": 2.46936297416687, + "learning_rate": 4.354777312523792e-06, + "loss": 1.5396, + "step": 646 + }, + { + "epoch": 0.6646122239342578, + "grad_norm": 2.5295984745025635, + "learning_rate": 4.352874000761325e-06, + "loss": 1.586, + "step": 647 + }, + { + "epoch": 0.6656394453004623, + "grad_norm": 2.7826883792877197, + "learning_rate": 4.3509706889988584e-06, + "loss": 1.5339, + "step": 648 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.4273526668548584, + "learning_rate": 4.349067377236392e-06, + "loss": 1.598, + "step": 649 + }, + { + "epoch": 0.6676938880328711, + "grad_norm": 2.644308090209961, + "learning_rate": 4.347164065473925e-06, + "loss": 1.5889, + "step": 650 + }, + { + "epoch": 0.6676938880328711, + "eval_loss": 1.8861087560653687, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 650 + }, + { + "epoch": 0.6687211093990755, + "grad_norm": 2.827376127243042, + "learning_rate": 4.3452607537114585e-06, + "loss": 1.606, + "step": 651 + }, + { + "epoch": 0.66974833076528, + "grad_norm": 2.7463080883026123, + "learning_rate": 4.343357441948992e-06, + "loss": 1.5208, + "step": 652 + }, + { + "epoch": 0.6707755521314843, + "grad_norm": 2.6082730293273926, + "learning_rate": 4.341454130186525e-06, + "loss": 1.5597, + "step": 653 + }, + { + "epoch": 0.6718027734976888, + "grad_norm": 2.7131078243255615, + "learning_rate": 4.339550818424058e-06, + "loss": 1.526, + "step": 654 + }, + { + "epoch": 0.6728299948638932, + "grad_norm": 2.6635708808898926, + "learning_rate": 4.337647506661591e-06, + "loss": 1.5052, + "step": 655 + }, + { + "epoch": 0.6738572162300975, + "grad_norm": 2.5686581134796143, + "learning_rate": 4.3357441948991245e-06, + "loss": 1.5374, + "step": 656 + }, + { + "epoch": 0.674884437596302, + "grad_norm": 3.0532894134521484, + "learning_rate": 4.333840883136658e-06, + "loss": 1.5787, + "step": 657 + }, + { + "epoch": 0.6759116589625064, + "grad_norm": 2.6494507789611816, + "learning_rate": 4.331937571374191e-06, + "loss": 1.5121, + "step": 658 + }, + { + "epoch": 0.6769388803287109, + "grad_norm": 2.748142719268799, + "learning_rate": 4.3300342596117246e-06, + "loss": 1.5294, + "step": 659 + }, + { + "epoch": 0.6779661016949152, + "grad_norm": 2.837799310684204, + "learning_rate": 4.328130947849258e-06, + "loss": 1.5424, + "step": 660 + }, + { + "epoch": 0.6779661016949152, + "eval_loss": 1.8644919395446777, + "eval_runtime": 37.0096, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 660 + }, + { + "epoch": 0.6789933230611197, + "grad_norm": 2.7405757904052734, + "learning_rate": 4.326227636086791e-06, + "loss": 1.4949, + "step": 661 + }, + { + "epoch": 0.6800205444273241, + "grad_norm": 2.545466661453247, + "learning_rate": 4.324324324324325e-06, + "loss": 1.5445, + "step": 662 + }, + { + "epoch": 0.6810477657935285, + "grad_norm": 2.5715537071228027, + "learning_rate": 4.322421012561858e-06, + "loss": 1.5468, + "step": 663 + }, + { + "epoch": 0.6820749871597329, + "grad_norm": 2.7073662281036377, + "learning_rate": 4.320517700799391e-06, + "loss": 1.5293, + "step": 664 + }, + { + "epoch": 0.6831022085259374, + "grad_norm": 2.62264347076416, + "learning_rate": 4.318614389036925e-06, + "loss": 1.6258, + "step": 665 + }, + { + "epoch": 0.6841294298921418, + "grad_norm": 2.4506373405456543, + "learning_rate": 4.316711077274458e-06, + "loss": 1.5845, + "step": 666 + }, + { + "epoch": 0.6851566512583461, + "grad_norm": 2.849473237991333, + "learning_rate": 4.3148077655119915e-06, + "loss": 1.4959, + "step": 667 + }, + { + "epoch": 0.6861838726245506, + "grad_norm": 2.5865421295166016, + "learning_rate": 4.312904453749525e-06, + "loss": 1.599, + "step": 668 + }, + { + "epoch": 0.687211093990755, + "grad_norm": 2.4636387825012207, + "learning_rate": 4.311001141987058e-06, + "loss": 1.5495, + "step": 669 + }, + { + "epoch": 0.6882383153569595, + "grad_norm": 2.638047218322754, + "learning_rate": 4.3090978302245916e-06, + "loss": 1.5507, + "step": 670 + }, + { + "epoch": 0.6882383153569595, + "eval_loss": 1.8834826946258545, + "eval_runtime": 37.0335, + "eval_samples_per_second": 27.003, + "eval_steps_per_second": 1.134, + "step": 670 + }, + { + "epoch": 0.6892655367231638, + "grad_norm": 2.669201612472534, + "learning_rate": 4.307194518462124e-06, + "loss": 1.5494, + "step": 671 + }, + { + "epoch": 0.6902927580893683, + "grad_norm": 2.634634494781494, + "learning_rate": 4.3052912066996574e-06, + "loss": 1.6177, + "step": 672 + }, + { + "epoch": 0.6913199794555727, + "grad_norm": 2.5489139556884766, + "learning_rate": 4.303387894937191e-06, + "loss": 1.5876, + "step": 673 + }, + { + "epoch": 0.6923472008217771, + "grad_norm": 2.580073833465576, + "learning_rate": 4.301484583174724e-06, + "loss": 1.5106, + "step": 674 + }, + { + "epoch": 0.6933744221879815, + "grad_norm": 2.5126636028289795, + "learning_rate": 4.2995812714122575e-06, + "loss": 1.4988, + "step": 675 + }, + { + "epoch": 0.6944016435541859, + "grad_norm": 2.4630191326141357, + "learning_rate": 4.297677959649791e-06, + "loss": 1.5218, + "step": 676 + }, + { + "epoch": 0.6954288649203904, + "grad_norm": 2.8822433948516846, + "learning_rate": 4.295774647887324e-06, + "loss": 1.5592, + "step": 677 + }, + { + "epoch": 0.6964560862865947, + "grad_norm": 2.5498058795928955, + "learning_rate": 4.293871336124858e-06, + "loss": 1.5713, + "step": 678 + }, + { + "epoch": 0.6974833076527992, + "grad_norm": 2.6661157608032227, + "learning_rate": 4.291968024362391e-06, + "loss": 1.5431, + "step": 679 + }, + { + "epoch": 0.6985105290190036, + "grad_norm": 2.617971897125244, + "learning_rate": 4.290064712599924e-06, + "loss": 1.5792, + "step": 680 + }, + { + "epoch": 0.6985105290190036, + "eval_loss": 1.8702285289764404, + "eval_runtime": 37.0376, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 680 + }, + { + "epoch": 0.699537750385208, + "grad_norm": 2.5892348289489746, + "learning_rate": 4.288161400837458e-06, + "loss": 1.5626, + "step": 681 + }, + { + "epoch": 0.7005649717514124, + "grad_norm": 2.5460269451141357, + "learning_rate": 4.286258089074991e-06, + "loss": 1.4876, + "step": 682 + }, + { + "epoch": 0.7015921931176169, + "grad_norm": 2.601632595062256, + "learning_rate": 4.284354777312524e-06, + "loss": 1.5547, + "step": 683 + }, + { + "epoch": 0.7026194144838213, + "grad_norm": 2.6032373905181885, + "learning_rate": 4.282451465550058e-06, + "loss": 1.5133, + "step": 684 + }, + { + "epoch": 0.7036466358500256, + "grad_norm": 2.6283886432647705, + "learning_rate": 4.280548153787591e-06, + "loss": 1.5375, + "step": 685 + }, + { + "epoch": 0.7046738572162301, + "grad_norm": 2.5892159938812256, + "learning_rate": 4.2786448420251245e-06, + "loss": 1.5338, + "step": 686 + }, + { + "epoch": 0.7057010785824345, + "grad_norm": 2.518629789352417, + "learning_rate": 4.276741530262658e-06, + "loss": 1.5003, + "step": 687 + }, + { + "epoch": 0.706728299948639, + "grad_norm": 2.8705344200134277, + "learning_rate": 4.27483821850019e-06, + "loss": 1.5178, + "step": 688 + }, + { + "epoch": 0.7077555213148433, + "grad_norm": 2.7153639793395996, + "learning_rate": 4.272934906737724e-06, + "loss": 1.5914, + "step": 689 + }, + { + "epoch": 0.7087827426810478, + "grad_norm": 2.4585697650909424, + "learning_rate": 4.271031594975257e-06, + "loss": 1.5726, + "step": 690 + }, + { + "epoch": 0.7087827426810478, + "eval_loss": 1.8637797832489014, + "eval_runtime": 37.0465, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 690 + }, + { + "epoch": 0.7098099640472522, + "grad_norm": 2.6370503902435303, + "learning_rate": 4.2691282832127905e-06, + "loss": 1.5706, + "step": 691 + }, + { + "epoch": 0.7108371854134566, + "grad_norm": 2.70202898979187, + "learning_rate": 4.267224971450324e-06, + "loss": 1.5298, + "step": 692 + }, + { + "epoch": 0.711864406779661, + "grad_norm": 2.6149585247039795, + "learning_rate": 4.265321659687857e-06, + "loss": 1.477, + "step": 693 + }, + { + "epoch": 0.7128916281458655, + "grad_norm": 2.9561493396759033, + "learning_rate": 4.2634183479253906e-06, + "loss": 1.5189, + "step": 694 + }, + { + "epoch": 0.7139188495120699, + "grad_norm": 2.697680711746216, + "learning_rate": 4.261515036162924e-06, + "loss": 1.5001, + "step": 695 + }, + { + "epoch": 0.7149460708782742, + "grad_norm": 2.55727219581604, + "learning_rate": 4.259611724400457e-06, + "loss": 1.533, + "step": 696 + }, + { + "epoch": 0.7159732922444787, + "grad_norm": 2.854290723800659, + "learning_rate": 4.257708412637991e-06, + "loss": 1.5057, + "step": 697 + }, + { + "epoch": 0.7170005136106831, + "grad_norm": 2.73824405670166, + "learning_rate": 4.255805100875523e-06, + "loss": 1.4567, + "step": 698 + }, + { + "epoch": 0.7180277349768875, + "grad_norm": 2.730050802230835, + "learning_rate": 4.2539017891130565e-06, + "loss": 1.4873, + "step": 699 + }, + { + "epoch": 0.7190549563430919, + "grad_norm": 2.966660737991333, + "learning_rate": 4.25199847735059e-06, + "loss": 1.574, + "step": 700 + }, + { + "epoch": 0.7190549563430919, + "eval_loss": 1.8623589277267456, + "eval_runtime": 37.0761, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 700 + }, + { + "epoch": 0.7200821777092964, + "grad_norm": 2.7526276111602783, + "learning_rate": 4.250095165588123e-06, + "loss": 1.4831, + "step": 701 + }, + { + "epoch": 0.7211093990755008, + "grad_norm": 2.9722564220428467, + "learning_rate": 4.2481918538256575e-06, + "loss": 1.548, + "step": 702 + }, + { + "epoch": 0.7221366204417052, + "grad_norm": 2.679373025894165, + "learning_rate": 4.246288542063191e-06, + "loss": 1.5145, + "step": 703 + }, + { + "epoch": 0.7231638418079096, + "grad_norm": 2.733139753341675, + "learning_rate": 4.244385230300724e-06, + "loss": 1.5542, + "step": 704 + }, + { + "epoch": 0.724191063174114, + "grad_norm": 2.6549174785614014, + "learning_rate": 4.242481918538257e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.7252182845403184, + "grad_norm": 2.562958002090454, + "learning_rate": 4.24057860677579e-06, + "loss": 1.524, + "step": 706 + }, + { + "epoch": 0.7262455059065228, + "grad_norm": 2.641123056411743, + "learning_rate": 4.2386752950133234e-06, + "loss": 1.5723, + "step": 707 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 2.8128373622894287, + "learning_rate": 4.236771983250857e-06, + "loss": 1.5151, + "step": 708 + }, + { + "epoch": 0.7282999486389317, + "grad_norm": 2.676765203475952, + "learning_rate": 4.23486867148839e-06, + "loss": 1.5024, + "step": 709 + }, + { + "epoch": 0.7293271700051361, + "grad_norm": 2.917178153991699, + "learning_rate": 4.2329653597259235e-06, + "loss": 1.5326, + "step": 710 + }, + { + "epoch": 0.7293271700051361, + "eval_loss": 1.8676133155822754, + "eval_runtime": 36.8304, + "eval_samples_per_second": 27.152, + "eval_steps_per_second": 1.14, + "step": 710 + }, + { + "epoch": 0.7303543913713405, + "grad_norm": 2.8313796520233154, + "learning_rate": 4.231062047963457e-06, + "loss": 1.5623, + "step": 711 + }, + { + "epoch": 0.731381612737545, + "grad_norm": 2.61542010307312, + "learning_rate": 4.22915873620099e-06, + "loss": 1.5725, + "step": 712 + }, + { + "epoch": 0.7324088341037494, + "grad_norm": 2.5788609981536865, + "learning_rate": 4.227255424438524e-06, + "loss": 1.5381, + "step": 713 + }, + { + "epoch": 0.7334360554699538, + "grad_norm": 2.720306158065796, + "learning_rate": 4.225352112676057e-06, + "loss": 1.572, + "step": 714 + }, + { + "epoch": 0.7344632768361582, + "grad_norm": 2.780654191970825, + "learning_rate": 4.2234488009135895e-06, + "loss": 1.5084, + "step": 715 + }, + { + "epoch": 0.7354904982023626, + "grad_norm": 2.4448273181915283, + "learning_rate": 4.221545489151123e-06, + "loss": 1.4614, + "step": 716 + }, + { + "epoch": 0.736517719568567, + "grad_norm": 2.6579184532165527, + "learning_rate": 4.219642177388656e-06, + "loss": 1.5022, + "step": 717 + }, + { + "epoch": 0.7375449409347714, + "grad_norm": 2.7092721462249756, + "learning_rate": 4.2177388656261896e-06, + "loss": 1.5307, + "step": 718 + }, + { + "epoch": 0.7385721623009759, + "grad_norm": 2.65014910697937, + "learning_rate": 4.215835553863723e-06, + "loss": 1.4936, + "step": 719 + }, + { + "epoch": 0.7395993836671803, + "grad_norm": 2.7403628826141357, + "learning_rate": 4.213932242101256e-06, + "loss": 1.4458, + "step": 720 + }, + { + "epoch": 0.7395993836671803, + "eval_loss": 1.8778676986694336, + "eval_runtime": 36.9378, + "eval_samples_per_second": 27.073, + "eval_steps_per_second": 1.137, + "step": 720 + }, + { + "epoch": 0.7406266050333847, + "grad_norm": 2.3937759399414062, + "learning_rate": 4.21202893033879e-06, + "loss": 1.5086, + "step": 721 + }, + { + "epoch": 0.7416538263995891, + "grad_norm": 2.5786867141723633, + "learning_rate": 4.210125618576323e-06, + "loss": 1.5321, + "step": 722 + }, + { + "epoch": 0.7426810477657936, + "grad_norm": 2.607236862182617, + "learning_rate": 4.208222306813856e-06, + "loss": 1.5244, + "step": 723 + }, + { + "epoch": 0.743708269131998, + "grad_norm": 2.5461602210998535, + "learning_rate": 4.20631899505139e-06, + "loss": 1.5841, + "step": 724 + }, + { + "epoch": 0.7447354904982023, + "grad_norm": 2.5051612854003906, + "learning_rate": 4.204415683288923e-06, + "loss": 1.43, + "step": 725 + }, + { + "epoch": 0.7457627118644068, + "grad_norm": 2.692816734313965, + "learning_rate": 4.2025123715264565e-06, + "loss": 1.5439, + "step": 726 + }, + { + "epoch": 0.7467899332306112, + "grad_norm": 2.4827022552490234, + "learning_rate": 4.20060905976399e-06, + "loss": 1.4987, + "step": 727 + }, + { + "epoch": 0.7478171545968156, + "grad_norm": 2.7261135578155518, + "learning_rate": 4.198705748001523e-06, + "loss": 1.4573, + "step": 728 + }, + { + "epoch": 0.74884437596302, + "grad_norm": 2.5710837841033936, + "learning_rate": 4.1968024362390566e-06, + "loss": 1.564, + "step": 729 + }, + { + "epoch": 0.7498715973292245, + "grad_norm": 2.61181640625, + "learning_rate": 4.19489912447659e-06, + "loss": 1.5211, + "step": 730 + }, + { + "epoch": 0.7498715973292245, + "eval_loss": 1.8794434070587158, + "eval_runtime": 36.9676, + "eval_samples_per_second": 27.051, + "eval_steps_per_second": 1.136, + "step": 730 + }, + { + "epoch": 0.7508988186954288, + "grad_norm": 2.6946470737457275, + "learning_rate": 4.192995812714123e-06, + "loss": 1.5837, + "step": 731 + }, + { + "epoch": 0.7519260400616333, + "grad_norm": 2.5337390899658203, + "learning_rate": 4.191092500951656e-06, + "loss": 1.5251, + "step": 732 + }, + { + "epoch": 0.7529532614278377, + "grad_norm": 2.541532278060913, + "learning_rate": 4.189189189189189e-06, + "loss": 1.503, + "step": 733 + }, + { + "epoch": 0.7539804827940421, + "grad_norm": 2.639124631881714, + "learning_rate": 4.1872858774267225e-06, + "loss": 1.4963, + "step": 734 + }, + { + "epoch": 0.7550077041602465, + "grad_norm": 2.872861862182617, + "learning_rate": 4.185382565664256e-06, + "loss": 1.5632, + "step": 735 + }, + { + "epoch": 0.7560349255264509, + "grad_norm": 2.588980197906494, + "learning_rate": 4.183479253901789e-06, + "loss": 1.4998, + "step": 736 + }, + { + "epoch": 0.7570621468926554, + "grad_norm": 2.9199588298797607, + "learning_rate": 4.181575942139323e-06, + "loss": 1.466, + "step": 737 + }, + { + "epoch": 0.7580893682588598, + "grad_norm": 2.5677788257598877, + "learning_rate": 4.179672630376856e-06, + "loss": 1.5247, + "step": 738 + }, + { + "epoch": 0.7591165896250642, + "grad_norm": 2.481142520904541, + "learning_rate": 4.177769318614389e-06, + "loss": 1.4915, + "step": 739 + }, + { + "epoch": 0.7601438109912686, + "grad_norm": 2.7072012424468994, + "learning_rate": 4.175866006851923e-06, + "loss": 1.5783, + "step": 740 + }, + { + "epoch": 0.7601438109912686, + "eval_loss": 1.8817057609558105, + "eval_runtime": 36.9635, + "eval_samples_per_second": 27.054, + "eval_steps_per_second": 1.136, + "step": 740 + }, + { + "epoch": 0.7611710323574731, + "grad_norm": 2.5433032512664795, + "learning_rate": 4.173962695089456e-06, + "loss": 1.5457, + "step": 741 + }, + { + "epoch": 0.7621982537236774, + "grad_norm": 2.702077627182007, + "learning_rate": 4.1720593833269894e-06, + "loss": 1.5285, + "step": 742 + }, + { + "epoch": 0.7632254750898819, + "grad_norm": 2.6678309440612793, + "learning_rate": 4.170156071564523e-06, + "loss": 1.5381, + "step": 743 + }, + { + "epoch": 0.7642526964560863, + "grad_norm": 2.6097185611724854, + "learning_rate": 4.168252759802056e-06, + "loss": 1.6076, + "step": 744 + }, + { + "epoch": 0.7652799178222907, + "grad_norm": 2.5975704193115234, + "learning_rate": 4.1663494480395895e-06, + "loss": 1.515, + "step": 745 + }, + { + "epoch": 0.7663071391884951, + "grad_norm": 2.6858558654785156, + "learning_rate": 4.164446136277123e-06, + "loss": 1.5499, + "step": 746 + }, + { + "epoch": 0.7673343605546995, + "grad_norm": 2.565023899078369, + "learning_rate": 4.162542824514656e-06, + "loss": 1.5502, + "step": 747 + }, + { + "epoch": 0.768361581920904, + "grad_norm": 2.61715030670166, + "learning_rate": 4.16063951275219e-06, + "loss": 1.5271, + "step": 748 + }, + { + "epoch": 0.7693888032871083, + "grad_norm": 2.564382314682007, + "learning_rate": 4.158736200989722e-06, + "loss": 1.4756, + "step": 749 + }, + { + "epoch": 0.7704160246533128, + "grad_norm": 2.64457368850708, + "learning_rate": 4.1568328892272555e-06, + "loss": 1.5246, + "step": 750 + }, + { + "epoch": 0.7704160246533128, + "eval_loss": 1.8777766227722168, + "eval_runtime": 36.9737, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 750 + }, + { + "epoch": 0.7714432460195172, + "grad_norm": 2.773602247238159, + "learning_rate": 4.154929577464789e-06, + "loss": 1.5337, + "step": 751 + }, + { + "epoch": 0.7724704673857217, + "grad_norm": 2.5466997623443604, + "learning_rate": 4.153026265702322e-06, + "loss": 1.5269, + "step": 752 + }, + { + "epoch": 0.773497688751926, + "grad_norm": 2.702888250350952, + "learning_rate": 4.1511229539398556e-06, + "loss": 1.4636, + "step": 753 + }, + { + "epoch": 0.7745249101181304, + "grad_norm": 2.629251718521118, + "learning_rate": 4.149219642177389e-06, + "loss": 1.513, + "step": 754 + }, + { + "epoch": 0.7755521314843349, + "grad_norm": 2.52544903755188, + "learning_rate": 4.147316330414922e-06, + "loss": 1.5225, + "step": 755 + }, + { + "epoch": 0.7765793528505393, + "grad_norm": 2.4267778396606445, + "learning_rate": 4.145413018652456e-06, + "loss": 1.4377, + "step": 756 + }, + { + "epoch": 0.7776065742167437, + "grad_norm": 2.6061577796936035, + "learning_rate": 4.143509706889989e-06, + "loss": 1.5402, + "step": 757 + }, + { + "epoch": 0.7786337955829481, + "grad_norm": 2.6571085453033447, + "learning_rate": 4.141606395127522e-06, + "loss": 1.527, + "step": 758 + }, + { + "epoch": 0.7796610169491526, + "grad_norm": 2.48740291595459, + "learning_rate": 4.139703083365056e-06, + "loss": 1.517, + "step": 759 + }, + { + "epoch": 0.7806882383153569, + "grad_norm": 2.5044426918029785, + "learning_rate": 4.137799771602588e-06, + "loss": 1.5579, + "step": 760 + }, + { + "epoch": 0.7806882383153569, + "eval_loss": 1.8778951168060303, + "eval_runtime": 37.0045, + "eval_samples_per_second": 27.024, + "eval_steps_per_second": 1.135, + "step": 760 + }, + { + "epoch": 0.7817154596815614, + "grad_norm": 2.4728074073791504, + "learning_rate": 4.135896459840122e-06, + "loss": 1.5321, + "step": 761 + }, + { + "epoch": 0.7827426810477658, + "grad_norm": 2.6541659832000732, + "learning_rate": 4.133993148077656e-06, + "loss": 1.4866, + "step": 762 + }, + { + "epoch": 0.7837699024139703, + "grad_norm": 2.526716947555542, + "learning_rate": 4.132089836315189e-06, + "loss": 1.5035, + "step": 763 + }, + { + "epoch": 0.7847971237801746, + "grad_norm": 2.683306932449341, + "learning_rate": 4.1301865245527226e-06, + "loss": 1.5221, + "step": 764 + }, + { + "epoch": 0.785824345146379, + "grad_norm": 2.4980056285858154, + "learning_rate": 4.128283212790256e-06, + "loss": 1.4643, + "step": 765 + }, + { + "epoch": 0.7868515665125835, + "grad_norm": 2.3985488414764404, + "learning_rate": 4.1263799010277884e-06, + "loss": 1.4348, + "step": 766 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 2.600689649581909, + "learning_rate": 4.124476589265322e-06, + "loss": 1.5647, + "step": 767 + }, + { + "epoch": 0.7889060092449923, + "grad_norm": 2.5934395790100098, + "learning_rate": 4.122573277502855e-06, + "loss": 1.4887, + "step": 768 + }, + { + "epoch": 0.7899332306111967, + "grad_norm": 2.5209388732910156, + "learning_rate": 4.1206699657403885e-06, + "loss": 1.4893, + "step": 769 + }, + { + "epoch": 0.7909604519774012, + "grad_norm": 2.6656298637390137, + "learning_rate": 4.118766653977922e-06, + "loss": 1.5239, + "step": 770 + }, + { + "epoch": 0.7909604519774012, + "eval_loss": 1.8737632036209106, + "eval_runtime": 37.056, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 770 + }, + { + "epoch": 0.7919876733436055, + "grad_norm": 2.5147693157196045, + "learning_rate": 4.116863342215455e-06, + "loss": 1.5, + "step": 771 + }, + { + "epoch": 0.79301489470981, + "grad_norm": 2.4653773307800293, + "learning_rate": 4.114960030452989e-06, + "loss": 1.4802, + "step": 772 + }, + { + "epoch": 0.7940421160760144, + "grad_norm": 2.643160820007324, + "learning_rate": 4.113056718690522e-06, + "loss": 1.5086, + "step": 773 + }, + { + "epoch": 0.7950693374422187, + "grad_norm": 2.4873125553131104, + "learning_rate": 4.111153406928055e-06, + "loss": 1.4558, + "step": 774 + }, + { + "epoch": 0.7960965588084232, + "grad_norm": 2.541308879852295, + "learning_rate": 4.109250095165589e-06, + "loss": 1.5483, + "step": 775 + }, + { + "epoch": 0.7971237801746276, + "grad_norm": 2.701874256134033, + "learning_rate": 4.107346783403122e-06, + "loss": 1.5551, + "step": 776 + }, + { + "epoch": 0.7981510015408321, + "grad_norm": 2.457371234893799, + "learning_rate": 4.105443471640655e-06, + "loss": 1.4968, + "step": 777 + }, + { + "epoch": 0.7991782229070364, + "grad_norm": 2.5294973850250244, + "learning_rate": 4.103540159878188e-06, + "loss": 1.4756, + "step": 778 + }, + { + "epoch": 0.8002054442732409, + "grad_norm": 2.464932918548584, + "learning_rate": 4.101636848115721e-06, + "loss": 1.5215, + "step": 779 + }, + { + "epoch": 0.8012326656394453, + "grad_norm": 2.5919454097747803, + "learning_rate": 4.099733536353255e-06, + "loss": 1.4807, + "step": 780 + }, + { + "epoch": 0.8012326656394453, + "eval_loss": 1.875250220298767, + "eval_runtime": 37.035, + "eval_samples_per_second": 27.001, + "eval_steps_per_second": 1.134, + "step": 780 + }, + { + "epoch": 0.8022598870056498, + "grad_norm": 2.505749464035034, + "learning_rate": 4.097830224590788e-06, + "loss": 1.5224, + "step": 781 + }, + { + "epoch": 0.8032871083718541, + "grad_norm": 2.747262954711914, + "learning_rate": 4.095926912828322e-06, + "loss": 1.6051, + "step": 782 + }, + { + "epoch": 0.8043143297380585, + "grad_norm": 2.6147146224975586, + "learning_rate": 4.094023601065855e-06, + "loss": 1.565, + "step": 783 + }, + { + "epoch": 0.805341551104263, + "grad_norm": 2.5381031036376953, + "learning_rate": 4.092120289303388e-06, + "loss": 1.511, + "step": 784 + }, + { + "epoch": 0.8063687724704673, + "grad_norm": 2.6476190090179443, + "learning_rate": 4.0902169775409215e-06, + "loss": 1.5147, + "step": 785 + }, + { + "epoch": 0.8073959938366718, + "grad_norm": 2.486919641494751, + "learning_rate": 4.088313665778455e-06, + "loss": 1.4697, + "step": 786 + }, + { + "epoch": 0.8084232152028762, + "grad_norm": 2.7006654739379883, + "learning_rate": 4.086410354015988e-06, + "loss": 1.4965, + "step": 787 + }, + { + "epoch": 0.8094504365690807, + "grad_norm": 2.7587921619415283, + "learning_rate": 4.0845070422535216e-06, + "loss": 1.4781, + "step": 788 + }, + { + "epoch": 0.810477657935285, + "grad_norm": 2.64239501953125, + "learning_rate": 4.082603730491055e-06, + "loss": 1.4945, + "step": 789 + }, + { + "epoch": 0.8115048793014895, + "grad_norm": 2.659919261932373, + "learning_rate": 4.080700418728588e-06, + "loss": 1.5378, + "step": 790 + }, + { + "epoch": 0.8115048793014895, + "eval_loss": 1.881503939628601, + "eval_runtime": 37.2277, + "eval_samples_per_second": 26.862, + "eval_steps_per_second": 1.128, + "step": 790 + }, + { + "epoch": 0.8125321006676939, + "grad_norm": 2.820996046066284, + "learning_rate": 4.078797106966122e-06, + "loss": 1.5134, + "step": 791 + }, + { + "epoch": 0.8135593220338984, + "grad_norm": 2.4667489528656006, + "learning_rate": 4.076893795203655e-06, + "loss": 1.4165, + "step": 792 + }, + { + "epoch": 0.8145865434001027, + "grad_norm": 2.5528056621551514, + "learning_rate": 4.074990483441188e-06, + "loss": 1.481, + "step": 793 + }, + { + "epoch": 0.8156137647663071, + "grad_norm": 2.632922887802124, + "learning_rate": 4.073087171678721e-06, + "loss": 1.5502, + "step": 794 + }, + { + "epoch": 0.8166409861325116, + "grad_norm": 2.662536144256592, + "learning_rate": 4.071183859916254e-06, + "loss": 1.5302, + "step": 795 + }, + { + "epoch": 0.8176682074987159, + "grad_norm": 2.662454605102539, + "learning_rate": 4.069280548153788e-06, + "loss": 1.5131, + "step": 796 + }, + { + "epoch": 0.8186954288649204, + "grad_norm": 2.6387386322021484, + "learning_rate": 4.067377236391321e-06, + "loss": 1.4677, + "step": 797 + }, + { + "epoch": 0.8197226502311248, + "grad_norm": 2.8879177570343018, + "learning_rate": 4.065473924628854e-06, + "loss": 1.5106, + "step": 798 + }, + { + "epoch": 0.8207498715973293, + "grad_norm": 2.680610418319702, + "learning_rate": 4.063570612866388e-06, + "loss": 1.5389, + "step": 799 + }, + { + "epoch": 0.8217770929635336, + "grad_norm": 2.659590244293213, + "learning_rate": 4.061667301103921e-06, + "loss": 1.5254, + "step": 800 + }, + { + "epoch": 0.8217770929635336, + "eval_loss": 1.881851077079773, + "eval_runtime": 37.3385, + "eval_samples_per_second": 26.782, + "eval_steps_per_second": 1.125, + "step": 800 + }, + { + "epoch": 0.8228043143297381, + "grad_norm": 2.711195945739746, + "learning_rate": 4.0597639893414544e-06, + "loss": 1.5026, + "step": 801 + }, + { + "epoch": 0.8238315356959425, + "grad_norm": 2.6387939453125, + "learning_rate": 4.057860677578988e-06, + "loss": 1.5101, + "step": 802 + }, + { + "epoch": 0.8248587570621468, + "grad_norm": 2.507387638092041, + "learning_rate": 4.055957365816521e-06, + "loss": 1.5277, + "step": 803 + }, + { + "epoch": 0.8258859784283513, + "grad_norm": 2.610370397567749, + "learning_rate": 4.0540540540540545e-06, + "loss": 1.4695, + "step": 804 + }, + { + "epoch": 0.8269131997945557, + "grad_norm": 2.8345787525177, + "learning_rate": 4.052150742291588e-06, + "loss": 1.467, + "step": 805 + }, + { + "epoch": 0.8279404211607602, + "grad_norm": 2.5142805576324463, + "learning_rate": 4.050247430529121e-06, + "loss": 1.4637, + "step": 806 + }, + { + "epoch": 0.8289676425269645, + "grad_norm": 2.7722535133361816, + "learning_rate": 4.048344118766655e-06, + "loss": 1.5427, + "step": 807 + }, + { + "epoch": 0.829994863893169, + "grad_norm": 2.816789150238037, + "learning_rate": 4.046440807004188e-06, + "loss": 1.5422, + "step": 808 + }, + { + "epoch": 0.8310220852593734, + "grad_norm": 2.6485376358032227, + "learning_rate": 4.044537495241721e-06, + "loss": 1.5078, + "step": 809 + }, + { + "epoch": 0.8320493066255779, + "grad_norm": 2.4760921001434326, + "learning_rate": 4.042634183479255e-06, + "loss": 1.4827, + "step": 810 + }, + { + "epoch": 0.8320493066255779, + "eval_loss": 1.8709155321121216, + "eval_runtime": 36.9015, + "eval_samples_per_second": 27.099, + "eval_steps_per_second": 1.138, + "step": 810 + }, + { + "epoch": 0.8330765279917822, + "grad_norm": 2.658062219619751, + "learning_rate": 4.040730871716787e-06, + "loss": 1.5434, + "step": 811 + }, + { + "epoch": 0.8341037493579867, + "grad_norm": 2.5443618297576904, + "learning_rate": 4.038827559954321e-06, + "loss": 1.5165, + "step": 812 + }, + { + "epoch": 0.8351309707241911, + "grad_norm": 2.698432207107544, + "learning_rate": 4.036924248191854e-06, + "loss": 1.5553, + "step": 813 + }, + { + "epoch": 0.8361581920903954, + "grad_norm": 2.5598180294036865, + "learning_rate": 4.035020936429387e-06, + "loss": 1.5289, + "step": 814 + }, + { + "epoch": 0.8371854134565999, + "grad_norm": 2.7002663612365723, + "learning_rate": 4.033117624666921e-06, + "loss": 1.509, + "step": 815 + }, + { + "epoch": 0.8382126348228043, + "grad_norm": 2.7882776260375977, + "learning_rate": 4.031214312904454e-06, + "loss": 1.4787, + "step": 816 + }, + { + "epoch": 0.8392398561890088, + "grad_norm": 2.5583086013793945, + "learning_rate": 4.029311001141987e-06, + "loss": 1.4642, + "step": 817 + }, + { + "epoch": 0.8402670775552131, + "grad_norm": 2.584573268890381, + "learning_rate": 4.027407689379521e-06, + "loss": 1.4343, + "step": 818 + }, + { + "epoch": 0.8412942989214176, + "grad_norm": 2.5062575340270996, + "learning_rate": 4.025504377617054e-06, + "loss": 1.4975, + "step": 819 + }, + { + "epoch": 0.842321520287622, + "grad_norm": 2.3546457290649414, + "learning_rate": 4.0236010658545875e-06, + "loss": 1.4795, + "step": 820 + }, + { + "epoch": 0.842321520287622, + "eval_loss": 1.8716342449188232, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 0.8433487416538265, + "grad_norm": 2.513228178024292, + "learning_rate": 4.02169775409212e-06, + "loss": 1.5191, + "step": 821 + }, + { + "epoch": 0.8443759630200308, + "grad_norm": 2.6360220909118652, + "learning_rate": 4.019794442329653e-06, + "loss": 1.5739, + "step": 822 + }, + { + "epoch": 0.8454031843862352, + "grad_norm": 2.5351850986480713, + "learning_rate": 4.0178911305671876e-06, + "loss": 1.4487, + "step": 823 + }, + { + "epoch": 0.8464304057524397, + "grad_norm": 2.6693851947784424, + "learning_rate": 4.015987818804721e-06, + "loss": 1.5319, + "step": 824 + }, + { + "epoch": 0.847457627118644, + "grad_norm": 2.5210046768188477, + "learning_rate": 4.014084507042254e-06, + "loss": 1.4723, + "step": 825 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.5799994468688965, + "learning_rate": 4.012181195279788e-06, + "loss": 1.4409, + "step": 826 + }, + { + "epoch": 0.8495120698510529, + "grad_norm": 2.5206313133239746, + "learning_rate": 4.010277883517321e-06, + "loss": 1.5449, + "step": 827 + }, + { + "epoch": 0.8505392912172574, + "grad_norm": 2.480584144592285, + "learning_rate": 4.0083745717548535e-06, + "loss": 1.5306, + "step": 828 + }, + { + "epoch": 0.8515665125834617, + "grad_norm": 2.5229053497314453, + "learning_rate": 4.006471259992387e-06, + "loss": 1.5001, + "step": 829 + }, + { + "epoch": 0.8525937339496662, + "grad_norm": 2.53357195854187, + "learning_rate": 4.00456794822992e-06, + "loss": 1.5188, + "step": 830 + }, + { + "epoch": 0.8525937339496662, + "eval_loss": 1.8724256753921509, + "eval_runtime": 37.1114, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 830 + }, + { + "epoch": 0.8536209553158706, + "grad_norm": 2.6796159744262695, + "learning_rate": 4.002664636467454e-06, + "loss": 1.5577, + "step": 831 + }, + { + "epoch": 0.8546481766820749, + "grad_norm": 2.4717817306518555, + "learning_rate": 4.000761324704987e-06, + "loss": 1.5124, + "step": 832 + }, + { + "epoch": 0.8556753980482794, + "grad_norm": 2.649057626724243, + "learning_rate": 3.99885801294252e-06, + "loss": 1.4516, + "step": 833 + }, + { + "epoch": 0.8567026194144838, + "grad_norm": 2.592406749725342, + "learning_rate": 3.996954701180054e-06, + "loss": 1.5215, + "step": 834 + }, + { + "epoch": 0.8577298407806883, + "grad_norm": 2.46956205368042, + "learning_rate": 3.995051389417587e-06, + "loss": 1.4324, + "step": 835 + }, + { + "epoch": 0.8587570621468926, + "grad_norm": 2.8484537601470947, + "learning_rate": 3.9931480776551204e-06, + "loss": 1.4574, + "step": 836 + }, + { + "epoch": 0.8597842835130971, + "grad_norm": 2.6475489139556885, + "learning_rate": 3.991244765892654e-06, + "loss": 1.5581, + "step": 837 + }, + { + "epoch": 0.8608115048793015, + "grad_norm": 2.527153491973877, + "learning_rate": 3.989341454130186e-06, + "loss": 1.4713, + "step": 838 + }, + { + "epoch": 0.861838726245506, + "grad_norm": 2.6767020225524902, + "learning_rate": 3.98743814236772e-06, + "loss": 1.4459, + "step": 839 + }, + { + "epoch": 0.8628659476117103, + "grad_norm": 2.6406850814819336, + "learning_rate": 3.985534830605253e-06, + "loss": 1.4964, + "step": 840 + }, + { + "epoch": 0.8628659476117103, + "eval_loss": 1.886762261390686, + "eval_runtime": 37.1378, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 840 + }, + { + "epoch": 0.8638931689779148, + "grad_norm": 2.466372489929199, + "learning_rate": 3.983631518842786e-06, + "loss": 1.5136, + "step": 841 + }, + { + "epoch": 0.8649203903441192, + "grad_norm": 2.830249071121216, + "learning_rate": 3.98172820708032e-06, + "loss": 1.5372, + "step": 842 + }, + { + "epoch": 0.8659476117103235, + "grad_norm": 2.675889015197754, + "learning_rate": 3.979824895317854e-06, + "loss": 1.4881, + "step": 843 + }, + { + "epoch": 0.866974833076528, + "grad_norm": 2.7418212890625, + "learning_rate": 3.977921583555387e-06, + "loss": 1.4541, + "step": 844 + }, + { + "epoch": 0.8680020544427324, + "grad_norm": 2.572862386703491, + "learning_rate": 3.97601827179292e-06, + "loss": 1.5191, + "step": 845 + }, + { + "epoch": 0.8690292758089369, + "grad_norm": 2.6524274349212646, + "learning_rate": 3.974114960030453e-06, + "loss": 1.4647, + "step": 846 + }, + { + "epoch": 0.8700564971751412, + "grad_norm": 2.794415235519409, + "learning_rate": 3.9722116482679866e-06, + "loss": 1.504, + "step": 847 + }, + { + "epoch": 0.8710837185413457, + "grad_norm": 2.5947723388671875, + "learning_rate": 3.97030833650552e-06, + "loss": 1.492, + "step": 848 + }, + { + "epoch": 0.8721109399075501, + "grad_norm": 2.8539984226226807, + "learning_rate": 3.968405024743053e-06, + "loss": 1.5049, + "step": 849 + }, + { + "epoch": 0.8731381612737545, + "grad_norm": 2.522786855697632, + "learning_rate": 3.966501712980587e-06, + "loss": 1.4223, + "step": 850 + }, + { + "epoch": 0.8731381612737545, + "eval_loss": 1.8691684007644653, + "eval_runtime": 37.1287, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 850 + }, + { + "epoch": 0.8741653826399589, + "grad_norm": 2.6483359336853027, + "learning_rate": 3.96459840121812e-06, + "loss": 1.4815, + "step": 851 + }, + { + "epoch": 0.8751926040061633, + "grad_norm": 2.581429958343506, + "learning_rate": 3.962695089455653e-06, + "loss": 1.4837, + "step": 852 + }, + { + "epoch": 0.8762198253723678, + "grad_norm": 2.5967583656311035, + "learning_rate": 3.960791777693187e-06, + "loss": 1.5289, + "step": 853 + }, + { + "epoch": 0.8772470467385721, + "grad_norm": 2.7169225215911865, + "learning_rate": 3.95888846593072e-06, + "loss": 1.5173, + "step": 854 + }, + { + "epoch": 0.8782742681047766, + "grad_norm": 2.644656181335449, + "learning_rate": 3.956985154168253e-06, + "loss": 1.5095, + "step": 855 + }, + { + "epoch": 0.879301489470981, + "grad_norm": 2.6397511959075928, + "learning_rate": 3.955081842405786e-06, + "loss": 1.5422, + "step": 856 + }, + { + "epoch": 0.8803287108371854, + "grad_norm": 2.691540479660034, + "learning_rate": 3.953178530643319e-06, + "loss": 1.5703, + "step": 857 + }, + { + "epoch": 0.8813559322033898, + "grad_norm": 2.50526762008667, + "learning_rate": 3.951275218880853e-06, + "loss": 1.5885, + "step": 858 + }, + { + "epoch": 0.8823831535695943, + "grad_norm": 2.3846609592437744, + "learning_rate": 3.949371907118386e-06, + "loss": 1.4248, + "step": 859 + }, + { + "epoch": 0.8834103749357987, + "grad_norm": 2.7107760906219482, + "learning_rate": 3.9474685953559194e-06, + "loss": 1.5358, + "step": 860 + }, + { + "epoch": 0.8834103749357987, + "eval_loss": 1.876391887664795, + "eval_runtime": 37.1284, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.884437596302003, + "grad_norm": 2.6206252574920654, + "learning_rate": 3.945565283593453e-06, + "loss": 1.4907, + "step": 861 + }, + { + "epoch": 0.8854648176682075, + "grad_norm": 2.6641664505004883, + "learning_rate": 3.943661971830986e-06, + "loss": 1.4888, + "step": 862 + }, + { + "epoch": 0.8864920390344119, + "grad_norm": 2.7342002391815186, + "learning_rate": 3.9417586600685195e-06, + "loss": 1.5421, + "step": 863 + }, + { + "epoch": 0.8875192604006163, + "grad_norm": 2.7120890617370605, + "learning_rate": 3.939855348306053e-06, + "loss": 1.5154, + "step": 864 + }, + { + "epoch": 0.8885464817668207, + "grad_norm": 2.6350717544555664, + "learning_rate": 3.937952036543586e-06, + "loss": 1.5135, + "step": 865 + }, + { + "epoch": 0.8895737031330252, + "grad_norm": 2.570857524871826, + "learning_rate": 3.93604872478112e-06, + "loss": 1.5072, + "step": 866 + }, + { + "epoch": 0.8906009244992296, + "grad_norm": 2.855071544647217, + "learning_rate": 3.934145413018653e-06, + "loss": 1.5171, + "step": 867 + }, + { + "epoch": 0.891628145865434, + "grad_norm": 2.5592195987701416, + "learning_rate": 3.932242101256186e-06, + "loss": 1.4695, + "step": 868 + }, + { + "epoch": 0.8926553672316384, + "grad_norm": 2.6110830307006836, + "learning_rate": 3.93033878949372e-06, + "loss": 1.4802, + "step": 869 + }, + { + "epoch": 0.8936825885978429, + "grad_norm": 2.7121548652648926, + "learning_rate": 3.928435477731253e-06, + "loss": 1.4422, + "step": 870 + }, + { + "epoch": 0.8936825885978429, + "eval_loss": 1.8629941940307617, + "eval_runtime": 37.1091, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.8947098099640473, + "grad_norm": 2.7763099670410156, + "learning_rate": 3.9265321659687864e-06, + "loss": 1.5275, + "step": 871 + }, + { + "epoch": 0.8957370313302516, + "grad_norm": 2.596220016479492, + "learning_rate": 3.924628854206319e-06, + "loss": 1.5374, + "step": 872 + }, + { + "epoch": 0.8967642526964561, + "grad_norm": 2.575990915298462, + "learning_rate": 3.922725542443852e-06, + "loss": 1.5071, + "step": 873 + }, + { + "epoch": 0.8977914740626605, + "grad_norm": 2.7052223682403564, + "learning_rate": 3.920822230681386e-06, + "loss": 1.4656, + "step": 874 + }, + { + "epoch": 0.8988186954288649, + "grad_norm": 2.550267219543457, + "learning_rate": 3.918918918918919e-06, + "loss": 1.4837, + "step": 875 + }, + { + "epoch": 0.8998459167950693, + "grad_norm": 2.6948060989379883, + "learning_rate": 3.917015607156452e-06, + "loss": 1.5156, + "step": 876 + }, + { + "epoch": 0.9008731381612738, + "grad_norm": 2.575596809387207, + "learning_rate": 3.915112295393986e-06, + "loss": 1.4881, + "step": 877 + }, + { + "epoch": 0.9019003595274782, + "grad_norm": 2.9835500717163086, + "learning_rate": 3.913208983631519e-06, + "loss": 1.5729, + "step": 878 + }, + { + "epoch": 0.9029275808936826, + "grad_norm": 2.5961520671844482, + "learning_rate": 3.9113056718690525e-06, + "loss": 1.5469, + "step": 879 + }, + { + "epoch": 0.903954802259887, + "grad_norm": 2.518260955810547, + "learning_rate": 3.909402360106586e-06, + "loss": 1.4704, + "step": 880 + }, + { + "epoch": 0.903954802259887, + "eval_loss": 1.8570330142974854, + "eval_runtime": 37.1293, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 880 + }, + { + "epoch": 0.9049820236260914, + "grad_norm": 2.740879535675049, + "learning_rate": 3.907499048344119e-06, + "loss": 1.5092, + "step": 881 + }, + { + "epoch": 0.9060092449922958, + "grad_norm": 2.5227973461151123, + "learning_rate": 3.905595736581652e-06, + "loss": 1.5252, + "step": 882 + }, + { + "epoch": 0.9070364663585002, + "grad_norm": 2.483125925064087, + "learning_rate": 3.903692424819186e-06, + "loss": 1.5216, + "step": 883 + }, + { + "epoch": 0.9080636877247047, + "grad_norm": 2.4979538917541504, + "learning_rate": 3.901789113056719e-06, + "loss": 1.4734, + "step": 884 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.707669496536255, + "learning_rate": 3.899885801294253e-06, + "loss": 1.5347, + "step": 885 + }, + { + "epoch": 0.9101181304571135, + "grad_norm": 2.6548409461975098, + "learning_rate": 3.897982489531786e-06, + "loss": 1.5079, + "step": 886 + }, + { + "epoch": 0.9111453518233179, + "grad_norm": 2.6678848266601562, + "learning_rate": 3.896079177769319e-06, + "loss": 1.5659, + "step": 887 + }, + { + "epoch": 0.9121725731895224, + "grad_norm": 2.570976734161377, + "learning_rate": 3.894175866006853e-06, + "loss": 1.5281, + "step": 888 + }, + { + "epoch": 0.9131997945557268, + "grad_norm": 2.5939226150512695, + "learning_rate": 3.892272554244385e-06, + "loss": 1.56, + "step": 889 + }, + { + "epoch": 0.9142270159219312, + "grad_norm": 2.4577109813690186, + "learning_rate": 3.890369242481919e-06, + "loss": 1.4702, + "step": 890 + }, + { + "epoch": 0.9142270159219312, + "eval_loss": 1.8534926176071167, + "eval_runtime": 37.3353, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 890 + }, + { + "epoch": 0.9152542372881356, + "grad_norm": 2.5012643337249756, + "learning_rate": 3.888465930719452e-06, + "loss": 1.5411, + "step": 891 + }, + { + "epoch": 0.91628145865434, + "grad_norm": 2.470905065536499, + "learning_rate": 3.886562618956985e-06, + "loss": 1.4766, + "step": 892 + }, + { + "epoch": 0.9173086800205444, + "grad_norm": 2.4610393047332764, + "learning_rate": 3.884659307194519e-06, + "loss": 1.4925, + "step": 893 + }, + { + "epoch": 0.9183359013867488, + "grad_norm": 2.573151111602783, + "learning_rate": 3.882755995432052e-06, + "loss": 1.4389, + "step": 894 + }, + { + "epoch": 0.9193631227529533, + "grad_norm": 2.628934144973755, + "learning_rate": 3.8808526836695854e-06, + "loss": 1.5099, + "step": 895 + }, + { + "epoch": 0.9203903441191577, + "grad_norm": 2.639920949935913, + "learning_rate": 3.878949371907119e-06, + "loss": 1.4724, + "step": 896 + }, + { + "epoch": 0.9214175654853621, + "grad_norm": 2.4775230884552, + "learning_rate": 3.877046060144652e-06, + "loss": 1.4657, + "step": 897 + }, + { + "epoch": 0.9224447868515665, + "grad_norm": 2.5242490768432617, + "learning_rate": 3.8751427483821855e-06, + "loss": 1.459, + "step": 898 + }, + { + "epoch": 0.923472008217771, + "grad_norm": 2.4444870948791504, + "learning_rate": 3.873239436619718e-06, + "loss": 1.494, + "step": 899 + }, + { + "epoch": 0.9244992295839753, + "grad_norm": 2.4326930046081543, + "learning_rate": 3.871336124857251e-06, + "loss": 1.4449, + "step": 900 + }, + { + "epoch": 0.9244992295839753, + "eval_loss": 1.8844366073608398, + "eval_runtime": 37.432, + "eval_samples_per_second": 26.715, + "eval_steps_per_second": 1.122, + "step": 900 + }, + { + "epoch": 0.9255264509501797, + "grad_norm": 2.621652603149414, + "learning_rate": 3.869432813094785e-06, + "loss": 1.5055, + "step": 901 + }, + { + "epoch": 0.9265536723163842, + "grad_norm": 2.5909814834594727, + "learning_rate": 3.867529501332318e-06, + "loss": 1.5176, + "step": 902 + }, + { + "epoch": 0.9275808936825886, + "grad_norm": 2.512760877609253, + "learning_rate": 3.8656261895698515e-06, + "loss": 1.5027, + "step": 903 + }, + { + "epoch": 0.928608115048793, + "grad_norm": 2.6954784393310547, + "learning_rate": 3.863722877807386e-06, + "loss": 1.4912, + "step": 904 + }, + { + "epoch": 0.9296353364149974, + "grad_norm": 2.481905221939087, + "learning_rate": 3.861819566044919e-06, + "loss": 1.529, + "step": 905 + }, + { + "epoch": 0.9306625577812019, + "grad_norm": 2.612758159637451, + "learning_rate": 3.859916254282452e-06, + "loss": 1.5658, + "step": 906 + }, + { + "epoch": 0.9316897791474062, + "grad_norm": 2.523171901702881, + "learning_rate": 3.858012942519985e-06, + "loss": 1.4804, + "step": 907 + }, + { + "epoch": 0.9327170005136107, + "grad_norm": 2.5053422451019287, + "learning_rate": 3.856109630757518e-06, + "loss": 1.5007, + "step": 908 + }, + { + "epoch": 0.9337442218798151, + "grad_norm": 2.6110904216766357, + "learning_rate": 3.854206318995052e-06, + "loss": 1.4758, + "step": 909 + }, + { + "epoch": 0.9347714432460195, + "grad_norm": 2.7004992961883545, + "learning_rate": 3.852303007232585e-06, + "loss": 1.5255, + "step": 910 + }, + { + "epoch": 0.9347714432460195, + "eval_loss": 1.892087697982788, + "eval_runtime": 36.9356, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 910 + }, + { + "epoch": 0.9357986646122239, + "grad_norm": 2.617180109024048, + "learning_rate": 3.850399695470118e-06, + "loss": 1.4563, + "step": 911 + }, + { + "epoch": 0.9368258859784283, + "grad_norm": 2.5294559001922607, + "learning_rate": 3.848496383707652e-06, + "loss": 1.4767, + "step": 912 + }, + { + "epoch": 0.9378531073446328, + "grad_norm": 2.594238042831421, + "learning_rate": 3.846593071945185e-06, + "loss": 1.4617, + "step": 913 + }, + { + "epoch": 0.9388803287108372, + "grad_norm": 2.6076996326446533, + "learning_rate": 3.8446897601827185e-06, + "loss": 1.446, + "step": 914 + }, + { + "epoch": 0.9399075500770416, + "grad_norm": 2.6047134399414062, + "learning_rate": 3.842786448420252e-06, + "loss": 1.5698, + "step": 915 + }, + { + "epoch": 0.940934771443246, + "grad_norm": 2.5670995712280273, + "learning_rate": 3.840883136657785e-06, + "loss": 1.5803, + "step": 916 + }, + { + "epoch": 0.9419619928094505, + "grad_norm": 2.6153712272644043, + "learning_rate": 3.838979824895318e-06, + "loss": 1.5565, + "step": 917 + }, + { + "epoch": 0.9429892141756548, + "grad_norm": 2.6707098484039307, + "learning_rate": 3.837076513132851e-06, + "loss": 1.4417, + "step": 918 + }, + { + "epoch": 0.9440164355418593, + "grad_norm": 2.591336488723755, + "learning_rate": 3.8351732013703844e-06, + "loss": 1.5159, + "step": 919 + }, + { + "epoch": 0.9450436569080637, + "grad_norm": 2.4689993858337402, + "learning_rate": 3.833269889607918e-06, + "loss": 1.4735, + "step": 920 + }, + { + "epoch": 0.9450436569080637, + "eval_loss": 1.875109076499939, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 920 + }, + { + "epoch": 0.9460708782742681, + "grad_norm": 2.739758253097534, + "learning_rate": 3.831366577845451e-06, + "loss": 1.4566, + "step": 921 + }, + { + "epoch": 0.9470980996404725, + "grad_norm": 2.65278697013855, + "learning_rate": 3.8294632660829845e-06, + "loss": 1.4545, + "step": 922 + }, + { + "epoch": 0.9481253210066769, + "grad_norm": 2.678321123123169, + "learning_rate": 3.827559954320518e-06, + "loss": 1.4864, + "step": 923 + }, + { + "epoch": 0.9491525423728814, + "grad_norm": 2.7516839504241943, + "learning_rate": 3.825656642558051e-06, + "loss": 1.5031, + "step": 924 + }, + { + "epoch": 0.9501797637390857, + "grad_norm": 2.6125247478485107, + "learning_rate": 3.823753330795585e-06, + "loss": 1.521, + "step": 925 + }, + { + "epoch": 0.9512069851052902, + "grad_norm": 2.5134871006011963, + "learning_rate": 3.821850019033118e-06, + "loss": 1.5019, + "step": 926 + }, + { + "epoch": 0.9522342064714946, + "grad_norm": 2.5274763107299805, + "learning_rate": 3.819946707270651e-06, + "loss": 1.4358, + "step": 927 + }, + { + "epoch": 0.9532614278376991, + "grad_norm": 2.480628728866577, + "learning_rate": 3.818043395508185e-06, + "loss": 1.483, + "step": 928 + }, + { + "epoch": 0.9542886492039034, + "grad_norm": 2.6450936794281006, + "learning_rate": 3.816140083745718e-06, + "loss": 1.4892, + "step": 929 + }, + { + "epoch": 0.9553158705701078, + "grad_norm": 2.6758296489715576, + "learning_rate": 3.8142367719832514e-06, + "loss": 1.5701, + "step": 930 + }, + { + "epoch": 0.9553158705701078, + "eval_loss": 1.8651176691055298, + "eval_runtime": 37.0553, + "eval_samples_per_second": 26.987, + "eval_steps_per_second": 1.133, + "step": 930 + }, + { + "epoch": 0.9563430919363123, + "grad_norm": 2.6887295246124268, + "learning_rate": 3.8123334602207844e-06, + "loss": 1.4596, + "step": 931 + }, + { + "epoch": 0.9573703133025167, + "grad_norm": 2.577507734298706, + "learning_rate": 3.8104301484583177e-06, + "loss": 1.4664, + "step": 932 + }, + { + "epoch": 0.9583975346687211, + "grad_norm": 2.6431517601013184, + "learning_rate": 3.808526836695851e-06, + "loss": 1.4715, + "step": 933 + }, + { + "epoch": 0.9594247560349255, + "grad_norm": 2.8451743125915527, + "learning_rate": 3.8066235249333845e-06, + "loss": 1.5096, + "step": 934 + }, + { + "epoch": 0.96045197740113, + "grad_norm": 2.5525193214416504, + "learning_rate": 3.804720213170918e-06, + "loss": 1.4975, + "step": 935 + }, + { + "epoch": 0.9614791987673343, + "grad_norm": 2.6353704929351807, + "learning_rate": 3.8028169014084508e-06, + "loss": 1.4979, + "step": 936 + }, + { + "epoch": 0.9625064201335388, + "grad_norm": 2.923781156539917, + "learning_rate": 3.800913589645984e-06, + "loss": 1.5118, + "step": 937 + }, + { + "epoch": 0.9635336414997432, + "grad_norm": 2.7363650798797607, + "learning_rate": 3.7990102778835175e-06, + "loss": 1.5016, + "step": 938 + }, + { + "epoch": 0.9645608628659477, + "grad_norm": 2.7126598358154297, + "learning_rate": 3.797106966121051e-06, + "loss": 1.5022, + "step": 939 + }, + { + "epoch": 0.965588084232152, + "grad_norm": 3.0102460384368896, + "learning_rate": 3.7952036543585842e-06, + "loss": 1.5638, + "step": 940 + }, + { + "epoch": 0.965588084232152, + "eval_loss": 1.8475635051727295, + "eval_runtime": 37.0871, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 940 + }, + { + "epoch": 0.9666153055983564, + "grad_norm": 2.5455925464630127, + "learning_rate": 3.793300342596117e-06, + "loss": 1.5218, + "step": 941 + }, + { + "epoch": 0.9676425269645609, + "grad_norm": 2.5489683151245117, + "learning_rate": 3.7913970308336505e-06, + "loss": 1.5375, + "step": 942 + }, + { + "epoch": 0.9686697483307652, + "grad_norm": 2.672626256942749, + "learning_rate": 3.789493719071184e-06, + "loss": 1.5217, + "step": 943 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.498378038406372, + "learning_rate": 3.7875904073087177e-06, + "loss": 1.475, + "step": 944 + }, + { + "epoch": 0.9707241910631741, + "grad_norm": 2.515190362930298, + "learning_rate": 3.785687095546251e-06, + "loss": 1.4341, + "step": 945 + }, + { + "epoch": 0.9717514124293786, + "grad_norm": 2.7561733722686768, + "learning_rate": 3.7837837837837844e-06, + "loss": 1.4838, + "step": 946 + }, + { + "epoch": 0.9727786337955829, + "grad_norm": 2.544776201248169, + "learning_rate": 3.7818804720213178e-06, + "loss": 1.4539, + "step": 947 + }, + { + "epoch": 0.9738058551617874, + "grad_norm": 2.744417428970337, + "learning_rate": 3.7799771602588507e-06, + "loss": 1.4296, + "step": 948 + }, + { + "epoch": 0.9748330765279918, + "grad_norm": 2.7759811878204346, + "learning_rate": 3.778073848496384e-06, + "loss": 1.4314, + "step": 949 + }, + { + "epoch": 0.9758602978941961, + "grad_norm": 2.5204875469207764, + "learning_rate": 3.7761705367339174e-06, + "loss": 1.495, + "step": 950 + }, + { + "epoch": 0.9758602978941961, + "eval_loss": 1.8491986989974976, + "eval_runtime": 37.1266, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 0.9768875192604006, + "grad_norm": 2.5156631469726562, + "learning_rate": 3.7742672249714508e-06, + "loss": 1.5228, + "step": 951 + }, + { + "epoch": 0.977914740626605, + "grad_norm": 2.469834089279175, + "learning_rate": 3.772363913208984e-06, + "loss": 1.5188, + "step": 952 + }, + { + "epoch": 0.9789419619928095, + "grad_norm": 2.4933393001556396, + "learning_rate": 3.770460601446517e-06, + "loss": 1.4818, + "step": 953 + }, + { + "epoch": 0.9799691833590138, + "grad_norm": 2.5382790565490723, + "learning_rate": 3.7685572896840504e-06, + "loss": 1.5107, + "step": 954 + }, + { + "epoch": 0.9809964047252183, + "grad_norm": 2.6962392330169678, + "learning_rate": 3.766653977921584e-06, + "loss": 1.4534, + "step": 955 + }, + { + "epoch": 0.9820236260914227, + "grad_norm": 2.5411603450775146, + "learning_rate": 3.764750666159117e-06, + "loss": 1.3942, + "step": 956 + }, + { + "epoch": 0.9830508474576272, + "grad_norm": 2.7101845741271973, + "learning_rate": 3.7628473543966505e-06, + "loss": 1.5003, + "step": 957 + }, + { + "epoch": 0.9840780688238315, + "grad_norm": 2.5780858993530273, + "learning_rate": 3.7609440426341835e-06, + "loss": 1.4508, + "step": 958 + }, + { + "epoch": 0.9851052901900359, + "grad_norm": 2.814004421234131, + "learning_rate": 3.759040730871717e-06, + "loss": 1.5435, + "step": 959 + }, + { + "epoch": 0.9861325115562404, + "grad_norm": 2.5701088905334473, + "learning_rate": 3.75713741910925e-06, + "loss": 1.4386, + "step": 960 + }, + { + "epoch": 0.9861325115562404, + "eval_loss": 1.8556313514709473, + "eval_runtime": 37.1375, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 0.9871597329224447, + "grad_norm": 2.407271146774292, + "learning_rate": 3.7552341073467836e-06, + "loss": 1.4338, + "step": 961 + }, + { + "epoch": 0.9881869542886492, + "grad_norm": 2.433256149291992, + "learning_rate": 3.753330795584317e-06, + "loss": 1.47, + "step": 962 + }, + { + "epoch": 0.9892141756548536, + "grad_norm": 2.679339647293091, + "learning_rate": 3.75142748382185e-06, + "loss": 1.5459, + "step": 963 + }, + { + "epoch": 0.9902413970210581, + "grad_norm": 2.325840473175049, + "learning_rate": 3.749524172059384e-06, + "loss": 1.4385, + "step": 964 + }, + { + "epoch": 0.9912686183872624, + "grad_norm": 2.7060739994049072, + "learning_rate": 3.747620860296917e-06, + "loss": 1.4938, + "step": 965 + }, + { + "epoch": 0.9922958397534669, + "grad_norm": 2.589808464050293, + "learning_rate": 3.7457175485344504e-06, + "loss": 1.5258, + "step": 966 + }, + { + "epoch": 0.9933230611196713, + "grad_norm": 2.4876766204833984, + "learning_rate": 3.7438142367719837e-06, + "loss": 1.5361, + "step": 967 + }, + { + "epoch": 0.9943502824858758, + "grad_norm": 2.7370550632476807, + "learning_rate": 3.741910925009517e-06, + "loss": 1.4999, + "step": 968 + }, + { + "epoch": 0.9953775038520801, + "grad_norm": 2.6552398204803467, + "learning_rate": 3.7400076132470505e-06, + "loss": 1.4612, + "step": 969 + }, + { + "epoch": 0.9964047252182845, + "grad_norm": 2.5229146480560303, + "learning_rate": 3.7381043014845834e-06, + "loss": 1.4794, + "step": 970 + }, + { + "epoch": 0.9964047252182845, + "eval_loss": 1.8614000082015991, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 0.997431946584489, + "grad_norm": 2.5604071617126465, + "learning_rate": 3.7362009897221168e-06, + "loss": 1.5078, + "step": 971 + }, + { + "epoch": 0.9984591679506933, + "grad_norm": 2.614192485809326, + "learning_rate": 3.73429767795965e-06, + "loss": 1.4849, + "step": 972 + }, + { + "epoch": 0.9994863893168978, + "grad_norm": 2.5245072841644287, + "learning_rate": 3.7323943661971835e-06, + "loss": 1.4993, + "step": 973 + }, + { + "epoch": 1.0005136106831023, + "grad_norm": 2.5631163120269775, + "learning_rate": 3.730491054434717e-06, + "loss": 1.4136, + "step": 974 + }, + { + "epoch": 1.0015408320493067, + "grad_norm": 2.625396728515625, + "learning_rate": 3.72858774267225e-06, + "loss": 1.3623, + "step": 975 + }, + { + "epoch": 1.002568053415511, + "grad_norm": 2.7067251205444336, + "learning_rate": 3.726684430909783e-06, + "loss": 1.4527, + "step": 976 + }, + { + "epoch": 1.0035952747817154, + "grad_norm": 2.5352942943573, + "learning_rate": 3.7247811191473165e-06, + "loss": 1.474, + "step": 977 + }, + { + "epoch": 1.00462249614792, + "grad_norm": 2.818143367767334, + "learning_rate": 3.72287780738485e-06, + "loss": 1.4852, + "step": 978 + }, + { + "epoch": 1.0056497175141244, + "grad_norm": 2.8269577026367188, + "learning_rate": 3.7209744956223832e-06, + "loss": 1.4667, + "step": 979 + }, + { + "epoch": 1.0066769388803287, + "grad_norm": 2.837454319000244, + "learning_rate": 3.719071183859916e-06, + "loss": 1.4728, + "step": 980 + }, + { + "epoch": 1.0066769388803287, + "eval_loss": 1.8701857328414917, + "eval_runtime": 37.2859, + "eval_samples_per_second": 26.82, + "eval_steps_per_second": 1.126, + "step": 980 + }, + { + "epoch": 1.007704160246533, + "grad_norm": 2.6601309776306152, + "learning_rate": 3.7171678720974495e-06, + "loss": 1.3917, + "step": 981 + }, + { + "epoch": 1.0087313816127375, + "grad_norm": 2.761413097381592, + "learning_rate": 3.715264560334983e-06, + "loss": 1.5049, + "step": 982 + }, + { + "epoch": 1.009758602978942, + "grad_norm": 2.681429147720337, + "learning_rate": 3.7133612485725163e-06, + "loss": 1.4579, + "step": 983 + }, + { + "epoch": 1.0107858243451464, + "grad_norm": 2.693094253540039, + "learning_rate": 3.71145793681005e-06, + "loss": 1.4766, + "step": 984 + }, + { + "epoch": 1.0118130457113508, + "grad_norm": 2.589165210723877, + "learning_rate": 3.7095546250475834e-06, + "loss": 1.4011, + "step": 985 + }, + { + "epoch": 1.0128402670775551, + "grad_norm": 2.738140821456909, + "learning_rate": 3.7076513132851168e-06, + "loss": 1.5109, + "step": 986 + }, + { + "epoch": 1.0138674884437597, + "grad_norm": 2.7863996028900146, + "learning_rate": 3.7057480015226497e-06, + "loss": 1.4344, + "step": 987 + }, + { + "epoch": 1.014894709809964, + "grad_norm": 2.6011335849761963, + "learning_rate": 3.703844689760183e-06, + "loss": 1.3981, + "step": 988 + }, + { + "epoch": 1.0159219311761685, + "grad_norm": 2.5715689659118652, + "learning_rate": 3.7019413779977164e-06, + "loss": 1.4876, + "step": 989 + }, + { + "epoch": 1.0169491525423728, + "grad_norm": 2.6804113388061523, + "learning_rate": 3.70003806623525e-06, + "loss": 1.4576, + "step": 990 + }, + { + "epoch": 1.0169491525423728, + "eval_loss": 1.8802876472473145, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 990 + }, + { + "epoch": 1.0179763739085772, + "grad_norm": 2.737604856491089, + "learning_rate": 3.698134754472783e-06, + "loss": 1.4399, + "step": 991 + }, + { + "epoch": 1.0190035952747818, + "grad_norm": 2.658388137817383, + "learning_rate": 3.696231442710316e-06, + "loss": 1.5035, + "step": 992 + }, + { + "epoch": 1.0200308166409862, + "grad_norm": 2.7263925075531006, + "learning_rate": 3.6943281309478495e-06, + "loss": 1.4469, + "step": 993 + }, + { + "epoch": 1.0210580380071905, + "grad_norm": 2.7885773181915283, + "learning_rate": 3.692424819185383e-06, + "loss": 1.4641, + "step": 994 + }, + { + "epoch": 1.022085259373395, + "grad_norm": 2.8929779529571533, + "learning_rate": 3.690521507422916e-06, + "loss": 1.4727, + "step": 995 + }, + { + "epoch": 1.0231124807395995, + "grad_norm": 2.6605303287506104, + "learning_rate": 3.6886181956604496e-06, + "loss": 1.4153, + "step": 996 + }, + { + "epoch": 1.0241397021058039, + "grad_norm": 2.8678219318389893, + "learning_rate": 3.686714883897983e-06, + "loss": 1.4199, + "step": 997 + }, + { + "epoch": 1.0251669234720082, + "grad_norm": 2.884453058242798, + "learning_rate": 3.684811572135516e-06, + "loss": 1.4909, + "step": 998 + }, + { + "epoch": 1.0261941448382126, + "grad_norm": 2.6681578159332275, + "learning_rate": 3.6829082603730492e-06, + "loss": 1.4903, + "step": 999 + }, + { + "epoch": 1.027221366204417, + "grad_norm": 2.8013756275177, + "learning_rate": 3.6810049486105826e-06, + "loss": 1.449, + "step": 1000 + }, + { + "epoch": 1.027221366204417, + "eval_loss": 1.8735578060150146, + "eval_runtime": 37.4416, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1000 + }, + { + "epoch": 1.0282485875706215, + "grad_norm": 2.787750244140625, + "learning_rate": 3.679101636848116e-06, + "loss": 1.4808, + "step": 1001 + }, + { + "epoch": 1.029275808936826, + "grad_norm": 2.666874647140503, + "learning_rate": 3.6771983250856493e-06, + "loss": 1.4739, + "step": 1002 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.720586061477661, + "learning_rate": 3.6752950133231822e-06, + "loss": 1.4439, + "step": 1003 + }, + { + "epoch": 1.0313302516692346, + "grad_norm": 2.707771062850952, + "learning_rate": 3.6733917015607156e-06, + "loss": 1.4765, + "step": 1004 + }, + { + "epoch": 1.0323574730354392, + "grad_norm": 2.658092737197876, + "learning_rate": 3.6714883897982494e-06, + "loss": 1.4822, + "step": 1005 + }, + { + "epoch": 1.0333846944016436, + "grad_norm": 2.66788911819458, + "learning_rate": 3.6695850780357828e-06, + "loss": 1.4237, + "step": 1006 + }, + { + "epoch": 1.034411915767848, + "grad_norm": 2.638956308364868, + "learning_rate": 3.667681766273316e-06, + "loss": 1.4168, + "step": 1007 + }, + { + "epoch": 1.0354391371340523, + "grad_norm": 2.739384412765503, + "learning_rate": 3.6657784545108495e-06, + "loss": 1.4338, + "step": 1008 + }, + { + "epoch": 1.0364663585002567, + "grad_norm": 2.599773406982422, + "learning_rate": 3.6638751427483824e-06, + "loss": 1.5188, + "step": 1009 + }, + { + "epoch": 1.0374935798664613, + "grad_norm": 2.705876588821411, + "learning_rate": 3.6619718309859158e-06, + "loss": 1.468, + "step": 1010 + }, + { + "epoch": 1.0374935798664613, + "eval_loss": 1.8751709461212158, + "eval_runtime": 36.9681, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 1010 + }, + { + "epoch": 1.0385208012326657, + "grad_norm": 2.712103843688965, + "learning_rate": 3.660068519223449e-06, + "loss": 1.508, + "step": 1011 + }, + { + "epoch": 1.03954802259887, + "grad_norm": 2.6621742248535156, + "learning_rate": 3.6581652074609825e-06, + "loss": 1.464, + "step": 1012 + }, + { + "epoch": 1.0405752439650744, + "grad_norm": 2.6141257286071777, + "learning_rate": 3.656261895698516e-06, + "loss": 1.4429, + "step": 1013 + }, + { + "epoch": 1.041602465331279, + "grad_norm": 2.5651750564575195, + "learning_rate": 3.6543585839360492e-06, + "loss": 1.4254, + "step": 1014 + }, + { + "epoch": 1.0426296866974833, + "grad_norm": 2.598686933517456, + "learning_rate": 3.652455272173582e-06, + "loss": 1.436, + "step": 1015 + }, + { + "epoch": 1.0436569080636877, + "grad_norm": 2.559375524520874, + "learning_rate": 3.6505519604111155e-06, + "loss": 1.4359, + "step": 1016 + }, + { + "epoch": 1.044684129429892, + "grad_norm": 2.57920503616333, + "learning_rate": 3.648648648648649e-06, + "loss": 1.4524, + "step": 1017 + }, + { + "epoch": 1.0457113507960965, + "grad_norm": 2.6629042625427246, + "learning_rate": 3.6467453368861823e-06, + "loss": 1.4901, + "step": 1018 + }, + { + "epoch": 1.046738572162301, + "grad_norm": 2.6846625804901123, + "learning_rate": 3.6448420251237156e-06, + "loss": 1.4345, + "step": 1019 + }, + { + "epoch": 1.0477657935285054, + "grad_norm": 2.7288694381713867, + "learning_rate": 3.6429387133612486e-06, + "loss": 1.4632, + "step": 1020 + }, + { + "epoch": 1.0477657935285054, + "eval_loss": 1.878115177154541, + "eval_runtime": 37.0627, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.0487930148947098, + "grad_norm": 2.6689510345458984, + "learning_rate": 3.641035401598782e-06, + "loss": 1.4934, + "step": 1021 + }, + { + "epoch": 1.0498202362609141, + "grad_norm": 2.6762588024139404, + "learning_rate": 3.6391320898363153e-06, + "loss": 1.4284, + "step": 1022 + }, + { + "epoch": 1.0508474576271187, + "grad_norm": 2.7875330448150635, + "learning_rate": 3.6372287780738487e-06, + "loss": 1.4717, + "step": 1023 + }, + { + "epoch": 1.051874678993323, + "grad_norm": 2.9239425659179688, + "learning_rate": 3.635325466311382e-06, + "loss": 1.4504, + "step": 1024 + }, + { + "epoch": 1.0529019003595275, + "grad_norm": 2.700913429260254, + "learning_rate": 3.633422154548916e-06, + "loss": 1.4701, + "step": 1025 + }, + { + "epoch": 1.0539291217257318, + "grad_norm": 2.693148612976074, + "learning_rate": 3.6315188427864487e-06, + "loss": 1.4889, + "step": 1026 + }, + { + "epoch": 1.0549563430919364, + "grad_norm": 2.6215498447418213, + "learning_rate": 3.629615531023982e-06, + "loss": 1.4586, + "step": 1027 + }, + { + "epoch": 1.0559835644581408, + "grad_norm": 2.704425573348999, + "learning_rate": 3.6277122192615155e-06, + "loss": 1.4182, + "step": 1028 + }, + { + "epoch": 1.0570107858243452, + "grad_norm": 2.691352128982544, + "learning_rate": 3.625808907499049e-06, + "loss": 1.468, + "step": 1029 + }, + { + "epoch": 1.0580380071905495, + "grad_norm": 2.769744634628296, + "learning_rate": 3.623905595736582e-06, + "loss": 1.4375, + "step": 1030 + }, + { + "epoch": 1.0580380071905495, + "eval_loss": 1.8686742782592773, + "eval_runtime": 37.1278, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.059065228556754, + "grad_norm": 2.6012840270996094, + "learning_rate": 3.6220022839741156e-06, + "loss": 1.4334, + "step": 1031 + }, + { + "epoch": 1.0600924499229585, + "grad_norm": 2.641416072845459, + "learning_rate": 3.6200989722116485e-06, + "loss": 1.4186, + "step": 1032 + }, + { + "epoch": 1.0611196712891628, + "grad_norm": 2.821176052093506, + "learning_rate": 3.618195660449182e-06, + "loss": 1.4438, + "step": 1033 + }, + { + "epoch": 1.0621468926553672, + "grad_norm": 2.719303607940674, + "learning_rate": 3.6162923486867152e-06, + "loss": 1.4519, + "step": 1034 + }, + { + "epoch": 1.0631741140215716, + "grad_norm": 2.76882004737854, + "learning_rate": 3.6143890369242486e-06, + "loss": 1.3941, + "step": 1035 + }, + { + "epoch": 1.064201335387776, + "grad_norm": 2.601137399673462, + "learning_rate": 3.612485725161782e-06, + "loss": 1.4167, + "step": 1036 + }, + { + "epoch": 1.0652285567539805, + "grad_norm": 2.691140651702881, + "learning_rate": 3.610582413399315e-06, + "loss": 1.4361, + "step": 1037 + }, + { + "epoch": 1.066255778120185, + "grad_norm": 2.5779640674591064, + "learning_rate": 3.6086791016368482e-06, + "loss": 1.4237, + "step": 1038 + }, + { + "epoch": 1.0672829994863893, + "grad_norm": 2.6535801887512207, + "learning_rate": 3.6067757898743816e-06, + "loss": 1.436, + "step": 1039 + }, + { + "epoch": 1.0683102208525936, + "grad_norm": 2.8365395069122314, + "learning_rate": 3.604872478111915e-06, + "loss": 1.5067, + "step": 1040 + }, + { + "epoch": 1.0683102208525936, + "eval_loss": 1.8686083555221558, + "eval_runtime": 37.1495, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 1040 + }, + { + "epoch": 1.0693374422187982, + "grad_norm": 2.704638957977295, + "learning_rate": 3.6029691663494483e-06, + "loss": 1.5105, + "step": 1041 + }, + { + "epoch": 1.0703646635850026, + "grad_norm": 2.6670243740081787, + "learning_rate": 3.6010658545869813e-06, + "loss": 1.451, + "step": 1042 + }, + { + "epoch": 1.071391884951207, + "grad_norm": 2.8808884620666504, + "learning_rate": 3.5991625428245146e-06, + "loss": 1.4795, + "step": 1043 + }, + { + "epoch": 1.0724191063174113, + "grad_norm": 2.800075054168701, + "learning_rate": 3.597259231062048e-06, + "loss": 1.4758, + "step": 1044 + }, + { + "epoch": 1.073446327683616, + "grad_norm": 2.4989686012268066, + "learning_rate": 3.5953559192995818e-06, + "loss": 1.42, + "step": 1045 + }, + { + "epoch": 1.0744735490498203, + "grad_norm": 2.7234370708465576, + "learning_rate": 3.593452607537115e-06, + "loss": 1.4549, + "step": 1046 + }, + { + "epoch": 1.0755007704160247, + "grad_norm": 2.7219443321228027, + "learning_rate": 3.5915492957746485e-06, + "loss": 1.5245, + "step": 1047 + }, + { + "epoch": 1.076527991782229, + "grad_norm": 2.6963443756103516, + "learning_rate": 3.589645984012182e-06, + "loss": 1.4472, + "step": 1048 + }, + { + "epoch": 1.0775552131484334, + "grad_norm": 2.7101387977600098, + "learning_rate": 3.587742672249715e-06, + "loss": 1.4659, + "step": 1049 + }, + { + "epoch": 1.078582434514638, + "grad_norm": 2.7673099040985107, + "learning_rate": 3.585839360487248e-06, + "loss": 1.4284, + "step": 1050 + }, + { + "epoch": 1.078582434514638, + "eval_loss": 1.879286289215088, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.13, + "step": 1050 + }, + { + "epoch": 1.0796096558808423, + "grad_norm": 2.81614351272583, + "learning_rate": 3.5839360487247815e-06, + "loss": 1.4251, + "step": 1051 + }, + { + "epoch": 1.0806368772470467, + "grad_norm": 2.770350933074951, + "learning_rate": 3.582032736962315e-06, + "loss": 1.4372, + "step": 1052 + }, + { + "epoch": 1.081664098613251, + "grad_norm": 2.8099257946014404, + "learning_rate": 3.5801294251998483e-06, + "loss": 1.495, + "step": 1053 + }, + { + "epoch": 1.0826913199794557, + "grad_norm": 2.582378387451172, + "learning_rate": 3.578226113437381e-06, + "loss": 1.4005, + "step": 1054 + }, + { + "epoch": 1.08371854134566, + "grad_norm": 2.6215474605560303, + "learning_rate": 3.5763228016749146e-06, + "loss": 1.4428, + "step": 1055 + }, + { + "epoch": 1.0847457627118644, + "grad_norm": 2.7204151153564453, + "learning_rate": 3.574419489912448e-06, + "loss": 1.5197, + "step": 1056 + }, + { + "epoch": 1.0857729840780688, + "grad_norm": 2.6563117504119873, + "learning_rate": 3.5725161781499813e-06, + "loss": 1.457, + "step": 1057 + }, + { + "epoch": 1.0868002054442731, + "grad_norm": 2.559816837310791, + "learning_rate": 3.5706128663875147e-06, + "loss": 1.4376, + "step": 1058 + }, + { + "epoch": 1.0878274268104777, + "grad_norm": 2.6742372512817383, + "learning_rate": 3.5687095546250476e-06, + "loss": 1.4626, + "step": 1059 + }, + { + "epoch": 1.088854648176682, + "grad_norm": 2.8052852153778076, + "learning_rate": 3.566806242862581e-06, + "loss": 1.4785, + "step": 1060 + }, + { + "epoch": 1.088854648176682, + "eval_loss": 1.8775676488876343, + "eval_runtime": 37.1523, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1060 + }, + { + "epoch": 1.0898818695428865, + "grad_norm": 2.687411308288574, + "learning_rate": 3.5649029311001143e-06, + "loss": 1.4491, + "step": 1061 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.662604808807373, + "learning_rate": 3.5629996193376477e-06, + "loss": 1.4274, + "step": 1062 + }, + { + "epoch": 1.0919363122752954, + "grad_norm": 2.619046211242676, + "learning_rate": 3.561096307575181e-06, + "loss": 1.4449, + "step": 1063 + }, + { + "epoch": 1.0929635336414998, + "grad_norm": 2.672553539276123, + "learning_rate": 3.559192995812714e-06, + "loss": 1.4719, + "step": 1064 + }, + { + "epoch": 1.0939907550077042, + "grad_norm": 2.868760824203491, + "learning_rate": 3.557289684050248e-06, + "loss": 1.4955, + "step": 1065 + }, + { + "epoch": 1.0950179763739085, + "grad_norm": 2.7635276317596436, + "learning_rate": 3.555386372287781e-06, + "loss": 1.4765, + "step": 1066 + }, + { + "epoch": 1.0960451977401129, + "grad_norm": 2.883657455444336, + "learning_rate": 3.5534830605253145e-06, + "loss": 1.4923, + "step": 1067 + }, + { + "epoch": 1.0970724191063175, + "grad_norm": 2.7595503330230713, + "learning_rate": 3.551579748762848e-06, + "loss": 1.4527, + "step": 1068 + }, + { + "epoch": 1.0980996404725218, + "grad_norm": 2.8310179710388184, + "learning_rate": 3.5496764370003812e-06, + "loss": 1.4485, + "step": 1069 + }, + { + "epoch": 1.0991268618387262, + "grad_norm": 2.699432134628296, + "learning_rate": 3.5477731252379146e-06, + "loss": 1.4322, + "step": 1070 + }, + { + "epoch": 1.0991268618387262, + "eval_loss": 1.8706557750701904, + "eval_runtime": 37.1509, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1070 + }, + { + "epoch": 1.1001540832049306, + "grad_norm": 2.6249091625213623, + "learning_rate": 3.5458698134754475e-06, + "loss": 1.4031, + "step": 1071 + }, + { + "epoch": 1.1011813045711352, + "grad_norm": 2.6882944107055664, + "learning_rate": 3.543966501712981e-06, + "loss": 1.4075, + "step": 1072 + }, + { + "epoch": 1.1022085259373395, + "grad_norm": 2.6578996181488037, + "learning_rate": 3.5420631899505142e-06, + "loss": 1.4256, + "step": 1073 + }, + { + "epoch": 1.103235747303544, + "grad_norm": 2.8124637603759766, + "learning_rate": 3.5401598781880476e-06, + "loss": 1.5263, + "step": 1074 + }, + { + "epoch": 1.1042629686697483, + "grad_norm": 2.6022191047668457, + "learning_rate": 3.538256566425581e-06, + "loss": 1.3482, + "step": 1075 + }, + { + "epoch": 1.1052901900359529, + "grad_norm": 2.753009080886841, + "learning_rate": 3.536353254663114e-06, + "loss": 1.483, + "step": 1076 + }, + { + "epoch": 1.1063174114021572, + "grad_norm": 2.575666904449463, + "learning_rate": 3.5344499429006473e-06, + "loss": 1.4325, + "step": 1077 + }, + { + "epoch": 1.1073446327683616, + "grad_norm": 2.594136953353882, + "learning_rate": 3.5325466311381806e-06, + "loss": 1.4187, + "step": 1078 + }, + { + "epoch": 1.108371854134566, + "grad_norm": 2.807074785232544, + "learning_rate": 3.530643319375714e-06, + "loss": 1.4763, + "step": 1079 + }, + { + "epoch": 1.1093990755007703, + "grad_norm": 2.5865330696105957, + "learning_rate": 3.5287400076132474e-06, + "loss": 1.3896, + "step": 1080 + }, + { + "epoch": 1.1093990755007703, + "eval_loss": 1.8865792751312256, + "eval_runtime": 37.1514, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1080 + }, + { + "epoch": 1.110426296866975, + "grad_norm": 2.715419054031372, + "learning_rate": 3.5268366958507803e-06, + "loss": 1.4259, + "step": 1081 + }, + { + "epoch": 1.1114535182331793, + "grad_norm": 2.6837334632873535, + "learning_rate": 3.5249333840883137e-06, + "loss": 1.4581, + "step": 1082 + }, + { + "epoch": 1.1124807395993837, + "grad_norm": 2.722128391265869, + "learning_rate": 3.523030072325847e-06, + "loss": 1.4486, + "step": 1083 + }, + { + "epoch": 1.113507960965588, + "grad_norm": 2.7390329837799072, + "learning_rate": 3.5211267605633804e-06, + "loss": 1.4225, + "step": 1084 + }, + { + "epoch": 1.1145351823317924, + "grad_norm": 2.740271806716919, + "learning_rate": 3.519223448800914e-06, + "loss": 1.4274, + "step": 1085 + }, + { + "epoch": 1.115562403697997, + "grad_norm": 2.8321313858032227, + "learning_rate": 3.5173201370384475e-06, + "loss": 1.4337, + "step": 1086 + }, + { + "epoch": 1.1165896250642013, + "grad_norm": 2.740826368331909, + "learning_rate": 3.515416825275981e-06, + "loss": 1.4331, + "step": 1087 + }, + { + "epoch": 1.1176168464304057, + "grad_norm": 2.7382850646972656, + "learning_rate": 3.513513513513514e-06, + "loss": 1.4495, + "step": 1088 + }, + { + "epoch": 1.11864406779661, + "grad_norm": 2.7677972316741943, + "learning_rate": 3.511610201751047e-06, + "loss": 1.4628, + "step": 1089 + }, + { + "epoch": 1.1196712891628147, + "grad_norm": 2.8850181102752686, + "learning_rate": 3.5097068899885806e-06, + "loss": 1.3739, + "step": 1090 + }, + { + "epoch": 1.1196712891628147, + "eval_loss": 1.8794182538986206, + "eval_runtime": 37.4388, + "eval_samples_per_second": 26.71, + "eval_steps_per_second": 1.122, + "step": 1090 + }, + { + "epoch": 1.120698510529019, + "grad_norm": 2.73730731010437, + "learning_rate": 3.507803578226114e-06, + "loss": 1.4444, + "step": 1091 + }, + { + "epoch": 1.1217257318952234, + "grad_norm": 3.027599811553955, + "learning_rate": 3.5059002664636473e-06, + "loss": 1.5123, + "step": 1092 + }, + { + "epoch": 1.1227529532614278, + "grad_norm": 2.613658905029297, + "learning_rate": 3.5039969547011802e-06, + "loss": 1.3798, + "step": 1093 + }, + { + "epoch": 1.1237801746276324, + "grad_norm": 2.6396987438201904, + "learning_rate": 3.5020936429387136e-06, + "loss": 1.4357, + "step": 1094 + }, + { + "epoch": 1.1248073959938367, + "grad_norm": 2.625540018081665, + "learning_rate": 3.500190331176247e-06, + "loss": 1.4354, + "step": 1095 + }, + { + "epoch": 1.125834617360041, + "grad_norm": 2.8413279056549072, + "learning_rate": 3.4982870194137803e-06, + "loss": 1.4846, + "step": 1096 + }, + { + "epoch": 1.1268618387262455, + "grad_norm": 2.7898473739624023, + "learning_rate": 3.4963837076513137e-06, + "loss": 1.421, + "step": 1097 + }, + { + "epoch": 1.1278890600924498, + "grad_norm": 2.6385879516601562, + "learning_rate": 3.4944803958888466e-06, + "loss": 1.4241, + "step": 1098 + }, + { + "epoch": 1.1289162814586544, + "grad_norm": 2.677098512649536, + "learning_rate": 3.49257708412638e-06, + "loss": 1.3671, + "step": 1099 + }, + { + "epoch": 1.1299435028248588, + "grad_norm": 2.8582913875579834, + "learning_rate": 3.4906737723639133e-06, + "loss": 1.4886, + "step": 1100 + }, + { + "epoch": 1.1299435028248588, + "eval_loss": 1.8696506023406982, + "eval_runtime": 37.462, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 1100 + }, + { + "epoch": 1.1309707241910631, + "grad_norm": 2.7157933712005615, + "learning_rate": 3.4887704606014467e-06, + "loss": 1.4964, + "step": 1101 + }, + { + "epoch": 1.1319979455572675, + "grad_norm": 2.6703319549560547, + "learning_rate": 3.48686714883898e-06, + "loss": 1.4509, + "step": 1102 + }, + { + "epoch": 1.133025166923472, + "grad_norm": 2.6307899951934814, + "learning_rate": 3.484963837076513e-06, + "loss": 1.485, + "step": 1103 + }, + { + "epoch": 1.1340523882896765, + "grad_norm": 2.8333141803741455, + "learning_rate": 3.4830605253140464e-06, + "loss": 1.432, + "step": 1104 + }, + { + "epoch": 1.1350796096558808, + "grad_norm": 2.545804023742676, + "learning_rate": 3.4811572135515797e-06, + "loss": 1.3799, + "step": 1105 + }, + { + "epoch": 1.1361068310220852, + "grad_norm": 2.7070441246032715, + "learning_rate": 3.4792539017891135e-06, + "loss": 1.4193, + "step": 1106 + }, + { + "epoch": 1.1371340523882898, + "grad_norm": 2.697140693664551, + "learning_rate": 3.477350590026647e-06, + "loss": 1.398, + "step": 1107 + }, + { + "epoch": 1.1381612737544942, + "grad_norm": 2.784795045852661, + "learning_rate": 3.4754472782641802e-06, + "loss": 1.4708, + "step": 1108 + }, + { + "epoch": 1.1391884951206985, + "grad_norm": 2.8518831729888916, + "learning_rate": 3.4735439665017136e-06, + "loss": 1.4707, + "step": 1109 + }, + { + "epoch": 1.140215716486903, + "grad_norm": 3.0481607913970947, + "learning_rate": 3.4716406547392465e-06, + "loss": 1.4385, + "step": 1110 + }, + { + "epoch": 1.140215716486903, + "eval_loss": 1.8367345333099365, + "eval_runtime": 36.9539, + "eval_samples_per_second": 27.061, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.1412429378531073, + "grad_norm": 2.803053140640259, + "learning_rate": 3.46973734297678e-06, + "loss": 1.4361, + "step": 1111 + }, + { + "epoch": 1.1422701592193119, + "grad_norm": 2.762638568878174, + "learning_rate": 3.4678340312143133e-06, + "loss": 1.4522, + "step": 1112 + }, + { + "epoch": 1.1432973805855162, + "grad_norm": 2.9458558559417725, + "learning_rate": 3.4659307194518466e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.1443246019517206, + "grad_norm": 2.774721622467041, + "learning_rate": 3.46402740768938e-06, + "loss": 1.4723, + "step": 1114 + }, + { + "epoch": 1.145351823317925, + "grad_norm": 2.6556880474090576, + "learning_rate": 3.462124095926913e-06, + "loss": 1.4073, + "step": 1115 + }, + { + "epoch": 1.1463790446841293, + "grad_norm": 2.7482454776763916, + "learning_rate": 3.4602207841644463e-06, + "loss": 1.4022, + "step": 1116 + }, + { + "epoch": 1.147406266050334, + "grad_norm": 2.627567768096924, + "learning_rate": 3.4583174724019797e-06, + "loss": 1.4428, + "step": 1117 + }, + { + "epoch": 1.1484334874165383, + "grad_norm": 2.750666856765747, + "learning_rate": 3.456414160639513e-06, + "loss": 1.4972, + "step": 1118 + }, + { + "epoch": 1.1494607087827426, + "grad_norm": 2.842184543609619, + "learning_rate": 3.4545108488770464e-06, + "loss": 1.4753, + "step": 1119 + }, + { + "epoch": 1.150487930148947, + "grad_norm": 2.829437732696533, + "learning_rate": 3.4526075371145793e-06, + "loss": 1.4525, + "step": 1120 + }, + { + "epoch": 1.150487930148947, + "eval_loss": 1.8498426675796509, + "eval_runtime": 37.0234, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 1120 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.7674920558929443, + "learning_rate": 3.4507042253521127e-06, + "loss": 1.4516, + "step": 1121 + }, + { + "epoch": 1.152542372881356, + "grad_norm": 2.726806879043579, + "learning_rate": 3.448800913589646e-06, + "loss": 1.5113, + "step": 1122 + }, + { + "epoch": 1.1535695942475603, + "grad_norm": 2.7698283195495605, + "learning_rate": 3.4468976018271794e-06, + "loss": 1.4295, + "step": 1123 + }, + { + "epoch": 1.1545968156137647, + "grad_norm": 2.6224706172943115, + "learning_rate": 3.4449942900647128e-06, + "loss": 1.5179, + "step": 1124 + }, + { + "epoch": 1.1556240369799693, + "grad_norm": 2.7266266345977783, + "learning_rate": 3.443090978302246e-06, + "loss": 1.4784, + "step": 1125 + }, + { + "epoch": 1.1566512583461737, + "grad_norm": 2.631751298904419, + "learning_rate": 3.44118766653978e-06, + "loss": 1.4659, + "step": 1126 + }, + { + "epoch": 1.157678479712378, + "grad_norm": 2.7077064514160156, + "learning_rate": 3.439284354777313e-06, + "loss": 1.4654, + "step": 1127 + }, + { + "epoch": 1.1587057010785824, + "grad_norm": 2.735102415084839, + "learning_rate": 3.4373810430148462e-06, + "loss": 1.4604, + "step": 1128 + }, + { + "epoch": 1.1597329224447868, + "grad_norm": 2.9314727783203125, + "learning_rate": 3.4354777312523796e-06, + "loss": 1.3964, + "step": 1129 + }, + { + "epoch": 1.1607601438109914, + "grad_norm": 2.658930778503418, + "learning_rate": 3.433574419489913e-06, + "loss": 1.4331, + "step": 1130 + }, + { + "epoch": 1.1607601438109914, + "eval_loss": 1.8545702695846558, + "eval_runtime": 37.0493, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1130 + }, + { + "epoch": 1.1617873651771957, + "grad_norm": 2.8146262168884277, + "learning_rate": 3.4316711077274463e-06, + "loss": 1.4711, + "step": 1131 + }, + { + "epoch": 1.1628145865434, + "grad_norm": 2.6918272972106934, + "learning_rate": 3.4297677959649792e-06, + "loss": 1.4327, + "step": 1132 + }, + { + "epoch": 1.1638418079096045, + "grad_norm": 2.680551528930664, + "learning_rate": 3.4278644842025126e-06, + "loss": 1.4886, + "step": 1133 + }, + { + "epoch": 1.1648690292758088, + "grad_norm": 2.617286443710327, + "learning_rate": 3.425961172440046e-06, + "loss": 1.4167, + "step": 1134 + }, + { + "epoch": 1.1658962506420134, + "grad_norm": 2.6772518157958984, + "learning_rate": 3.4240578606775793e-06, + "loss": 1.5274, + "step": 1135 + }, + { + "epoch": 1.1669234720082178, + "grad_norm": 2.640723705291748, + "learning_rate": 3.4221545489151127e-06, + "loss": 1.4398, + "step": 1136 + }, + { + "epoch": 1.1679506933744221, + "grad_norm": 2.7104315757751465, + "learning_rate": 3.4202512371526456e-06, + "loss": 1.4367, + "step": 1137 + }, + { + "epoch": 1.1689779147406265, + "grad_norm": 2.831645965576172, + "learning_rate": 3.418347925390179e-06, + "loss": 1.5329, + "step": 1138 + }, + { + "epoch": 1.170005136106831, + "grad_norm": 2.7174274921417236, + "learning_rate": 3.4164446136277124e-06, + "loss": 1.481, + "step": 1139 + }, + { + "epoch": 1.1710323574730355, + "grad_norm": 2.6496286392211914, + "learning_rate": 3.4145413018652457e-06, + "loss": 1.4489, + "step": 1140 + }, + { + "epoch": 1.1710323574730355, + "eval_loss": 1.8524384498596191, + "eval_runtime": 37.1347, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 1140 + }, + { + "epoch": 1.1720595788392398, + "grad_norm": 2.610058069229126, + "learning_rate": 3.412637990102779e-06, + "loss": 1.4755, + "step": 1141 + }, + { + "epoch": 1.1730868002054442, + "grad_norm": 2.642070770263672, + "learning_rate": 3.4107346783403125e-06, + "loss": 1.4981, + "step": 1142 + }, + { + "epoch": 1.1741140215716488, + "grad_norm": 2.57493257522583, + "learning_rate": 3.4088313665778454e-06, + "loss": 1.4039, + "step": 1143 + }, + { + "epoch": 1.1751412429378532, + "grad_norm": 2.5919628143310547, + "learning_rate": 3.4069280548153788e-06, + "loss": 1.4391, + "step": 1144 + }, + { + "epoch": 1.1761684643040575, + "grad_norm": 2.698862314224243, + "learning_rate": 3.405024743052912e-06, + "loss": 1.4198, + "step": 1145 + }, + { + "epoch": 1.177195685670262, + "grad_norm": 2.6387715339660645, + "learning_rate": 3.403121431290446e-06, + "loss": 1.4373, + "step": 1146 + }, + { + "epoch": 1.1782229070364663, + "grad_norm": 2.764643430709839, + "learning_rate": 3.4012181195279793e-06, + "loss": 1.4348, + "step": 1147 + }, + { + "epoch": 1.1792501284026708, + "grad_norm": 2.721038579940796, + "learning_rate": 3.3993148077655126e-06, + "loss": 1.4232, + "step": 1148 + }, + { + "epoch": 1.1802773497688752, + "grad_norm": 2.8942360877990723, + "learning_rate": 3.3974114960030456e-06, + "loss": 1.4405, + "step": 1149 + }, + { + "epoch": 1.1813045711350796, + "grad_norm": 2.687194585800171, + "learning_rate": 3.395508184240579e-06, + "loss": 1.3854, + "step": 1150 + }, + { + "epoch": 1.1813045711350796, + "eval_loss": 1.8701989650726318, + "eval_runtime": 37.132, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.182331792501284, + "grad_norm": 2.7137982845306396, + "learning_rate": 3.3936048724781123e-06, + "loss": 1.4174, + "step": 1151 + }, + { + "epoch": 1.1833590138674885, + "grad_norm": 2.8916432857513428, + "learning_rate": 3.3917015607156457e-06, + "loss": 1.5043, + "step": 1152 + }, + { + "epoch": 1.184386235233693, + "grad_norm": 2.6625723838806152, + "learning_rate": 3.389798248953179e-06, + "loss": 1.3832, + "step": 1153 + }, + { + "epoch": 1.1854134565998973, + "grad_norm": 2.8155555725097656, + "learning_rate": 3.387894937190712e-06, + "loss": 1.4098, + "step": 1154 + }, + { + "epoch": 1.1864406779661016, + "grad_norm": 2.634751796722412, + "learning_rate": 3.3859916254282453e-06, + "loss": 1.4522, + "step": 1155 + }, + { + "epoch": 1.1874678993323062, + "grad_norm": 2.7085063457489014, + "learning_rate": 3.3840883136657787e-06, + "loss": 1.4539, + "step": 1156 + }, + { + "epoch": 1.1884951206985106, + "grad_norm": 2.65588116645813, + "learning_rate": 3.382185001903312e-06, + "loss": 1.5084, + "step": 1157 + }, + { + "epoch": 1.189522342064715, + "grad_norm": 2.582489252090454, + "learning_rate": 3.3802816901408454e-06, + "loss": 1.4218, + "step": 1158 + }, + { + "epoch": 1.1905495634309193, + "grad_norm": 2.6364378929138184, + "learning_rate": 3.3783783783783788e-06, + "loss": 1.4522, + "step": 1159 + }, + { + "epoch": 1.1915767847971237, + "grad_norm": 2.6329915523529053, + "learning_rate": 3.3764750666159117e-06, + "loss": 1.442, + "step": 1160 + }, + { + "epoch": 1.1915767847971237, + "eval_loss": 1.8593298196792603, + "eval_runtime": 37.1143, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1160 + }, + { + "epoch": 1.1926040061633283, + "grad_norm": 2.610050916671753, + "learning_rate": 3.374571754853445e-06, + "loss": 1.4397, + "step": 1161 + }, + { + "epoch": 1.1936312275295327, + "grad_norm": 2.5503945350646973, + "learning_rate": 3.3726684430909784e-06, + "loss": 1.4211, + "step": 1162 + }, + { + "epoch": 1.194658448895737, + "grad_norm": 2.6648080348968506, + "learning_rate": 3.370765131328512e-06, + "loss": 1.3938, + "step": 1163 + }, + { + "epoch": 1.1956856702619414, + "grad_norm": 2.6705355644226074, + "learning_rate": 3.368861819566045e-06, + "loss": 1.4107, + "step": 1164 + }, + { + "epoch": 1.1967128916281458, + "grad_norm": 2.7582640647888184, + "learning_rate": 3.366958507803578e-06, + "loss": 1.4327, + "step": 1165 + }, + { + "epoch": 1.1977401129943503, + "grad_norm": 2.7768008708953857, + "learning_rate": 3.365055196041112e-06, + "loss": 1.454, + "step": 1166 + }, + { + "epoch": 1.1987673343605547, + "grad_norm": 2.628997325897217, + "learning_rate": 3.3631518842786452e-06, + "loss": 1.3988, + "step": 1167 + }, + { + "epoch": 1.199794555726759, + "grad_norm": 2.637505531311035, + "learning_rate": 3.3612485725161786e-06, + "loss": 1.4766, + "step": 1168 + }, + { + "epoch": 1.2008217770929634, + "grad_norm": 2.6935489177703857, + "learning_rate": 3.359345260753712e-06, + "loss": 1.4063, + "step": 1169 + }, + { + "epoch": 1.201848998459168, + "grad_norm": 2.7827870845794678, + "learning_rate": 3.3574419489912453e-06, + "loss": 1.4882, + "step": 1170 + }, + { + "epoch": 1.201848998459168, + "eval_loss": 1.8575458526611328, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.2028762198253724, + "grad_norm": 2.789053201675415, + "learning_rate": 3.3555386372287783e-06, + "loss": 1.4699, + "step": 1171 + }, + { + "epoch": 1.2039034411915768, + "grad_norm": 3.053184747695923, + "learning_rate": 3.3536353254663116e-06, + "loss": 1.4993, + "step": 1172 + }, + { + "epoch": 1.2049306625577811, + "grad_norm": 2.6963112354278564, + "learning_rate": 3.351732013703845e-06, + "loss": 1.3985, + "step": 1173 + }, + { + "epoch": 1.2059578839239857, + "grad_norm": 2.689101219177246, + "learning_rate": 3.3498287019413784e-06, + "loss": 1.4464, + "step": 1174 + }, + { + "epoch": 1.20698510529019, + "grad_norm": 2.7748899459838867, + "learning_rate": 3.3479253901789117e-06, + "loss": 1.4554, + "step": 1175 + }, + { + "epoch": 1.2080123266563945, + "grad_norm": 2.8673830032348633, + "learning_rate": 3.346022078416445e-06, + "loss": 1.4796, + "step": 1176 + }, + { + "epoch": 1.2090395480225988, + "grad_norm": 2.7719533443450928, + "learning_rate": 3.344118766653978e-06, + "loss": 1.3716, + "step": 1177 + }, + { + "epoch": 1.2100667693888032, + "grad_norm": 2.71771240234375, + "learning_rate": 3.3422154548915114e-06, + "loss": 1.4554, + "step": 1178 + }, + { + "epoch": 1.2110939907550078, + "grad_norm": 2.7095048427581787, + "learning_rate": 3.3403121431290448e-06, + "loss": 1.4233, + "step": 1179 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 2.631638526916504, + "learning_rate": 3.338408831366578e-06, + "loss": 1.4581, + "step": 1180 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.855039119720459, + "eval_runtime": 37.1083, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1180 + }, + { + "epoch": 1.2131484334874165, + "grad_norm": 2.657257318496704, + "learning_rate": 3.3365055196041115e-06, + "loss": 1.3864, + "step": 1181 + }, + { + "epoch": 1.214175654853621, + "grad_norm": 2.894169569015503, + "learning_rate": 3.3346022078416444e-06, + "loss": 1.4673, + "step": 1182 + }, + { + "epoch": 1.2152028762198253, + "grad_norm": 2.852644920349121, + "learning_rate": 3.3326988960791778e-06, + "loss": 1.5026, + "step": 1183 + }, + { + "epoch": 1.2162300975860298, + "grad_norm": 2.720878839492798, + "learning_rate": 3.330795584316711e-06, + "loss": 1.4281, + "step": 1184 + }, + { + "epoch": 1.2172573189522342, + "grad_norm": 2.6779472827911377, + "learning_rate": 3.3288922725542445e-06, + "loss": 1.3559, + "step": 1185 + }, + { + "epoch": 1.2182845403184386, + "grad_norm": 2.836782217025757, + "learning_rate": 3.326988960791778e-06, + "loss": 1.4185, + "step": 1186 + }, + { + "epoch": 1.219311761684643, + "grad_norm": 2.9181265830993652, + "learning_rate": 3.3250856490293117e-06, + "loss": 1.4379, + "step": 1187 + }, + { + "epoch": 1.2203389830508475, + "grad_norm": 2.7315778732299805, + "learning_rate": 3.323182337266845e-06, + "loss": 1.4824, + "step": 1188 + }, + { + "epoch": 1.221366204417052, + "grad_norm": 2.8483448028564453, + "learning_rate": 3.321279025504378e-06, + "loss": 1.4766, + "step": 1189 + }, + { + "epoch": 1.2223934257832563, + "grad_norm": 2.736323118209839, + "learning_rate": 3.3193757137419113e-06, + "loss": 1.4647, + "step": 1190 + }, + { + "epoch": 1.2223934257832563, + "eval_loss": 1.8581864833831787, + "eval_runtime": 37.3401, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.2234206471494606, + "grad_norm": 2.8863027095794678, + "learning_rate": 3.3174724019794447e-06, + "loss": 1.4533, + "step": 1191 + }, + { + "epoch": 1.2244478685156652, + "grad_norm": 2.951228380203247, + "learning_rate": 3.315569090216978e-06, + "loss": 1.495, + "step": 1192 + }, + { + "epoch": 1.2254750898818696, + "grad_norm": 2.7486696243286133, + "learning_rate": 3.3136657784545114e-06, + "loss": 1.4082, + "step": 1193 + }, + { + "epoch": 1.226502311248074, + "grad_norm": 2.9105849266052246, + "learning_rate": 3.3117624666920443e-06, + "loss": 1.4988, + "step": 1194 + }, + { + "epoch": 1.2275295326142783, + "grad_norm": 2.8165998458862305, + "learning_rate": 3.3098591549295777e-06, + "loss": 1.4162, + "step": 1195 + }, + { + "epoch": 1.2285567539804827, + "grad_norm": 2.8999876976013184, + "learning_rate": 3.307955843167111e-06, + "loss": 1.3909, + "step": 1196 + }, + { + "epoch": 1.2295839753466873, + "grad_norm": 2.8325014114379883, + "learning_rate": 3.3060525314046444e-06, + "loss": 1.4625, + "step": 1197 + }, + { + "epoch": 1.2306111967128917, + "grad_norm": 2.6553618907928467, + "learning_rate": 3.304149219642178e-06, + "loss": 1.4453, + "step": 1198 + }, + { + "epoch": 1.231638418079096, + "grad_norm": 2.584757089614868, + "learning_rate": 3.3022459078797107e-06, + "loss": 1.4412, + "step": 1199 + }, + { + "epoch": 1.2326656394453004, + "grad_norm": 2.68215274810791, + "learning_rate": 3.300342596117244e-06, + "loss": 1.4511, + "step": 1200 + }, + { + "epoch": 1.2326656394453004, + "eval_loss": 1.849502682685852, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 1200 + }, + { + "epoch": 1.2336928608115048, + "grad_norm": 2.735494375228882, + "learning_rate": 3.2984392843547775e-06, + "loss": 1.4433, + "step": 1201 + }, + { + "epoch": 1.2347200821777093, + "grad_norm": 2.6691815853118896, + "learning_rate": 3.296535972592311e-06, + "loss": 1.4033, + "step": 1202 + }, + { + "epoch": 1.2357473035439137, + "grad_norm": 2.6378355026245117, + "learning_rate": 3.294632660829844e-06, + "loss": 1.378, + "step": 1203 + }, + { + "epoch": 1.236774524910118, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4227, + "step": 1204 + }, + { + "epoch": 1.2378017462763227, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4204, + "step": 1205 + }, + { + "epoch": 1.238828967642527, + "grad_norm": 2.786475896835327, + "learning_rate": 3.2908260373049105e-06, + "loss": 1.4165, + "step": 1206 + }, + { + "epoch": 1.2398561890087314, + "grad_norm": 2.936825752258301, + "learning_rate": 3.288922725542444e-06, + "loss": 1.4712, + "step": 1207 + }, + { + "epoch": 1.2408834103749358, + "grad_norm": 2.7255940437316895, + "learning_rate": 3.2870194137799776e-06, + "loss": 1.4177, + "step": 1208 + }, + { + "epoch": 1.2419106317411401, + "grad_norm": 2.6526482105255127, + "learning_rate": 3.285116102017511e-06, + "loss": 1.3975, + "step": 1209 + }, + { + "epoch": 1.2429378531073447, + "grad_norm": 2.7539143562316895, + "learning_rate": 3.2832127902550444e-06, + "loss": 1.4717, + "step": 1210 + }, + { + "epoch": 1.2429378531073447, + "eval_loss": 1.8445062637329102, + "eval_runtime": 36.9889, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 1210 + }, + { + "epoch": 1.243965074473549, + "grad_norm": 2.907252073287964, + "learning_rate": 3.2813094784925777e-06, + "loss": 1.4484, + "step": 1211 + }, + { + "epoch": 1.2449922958397535, + "grad_norm": 2.7312281131744385, + "learning_rate": 3.2794061667301107e-06, + "loss": 1.3863, + "step": 1212 + }, + { + "epoch": 1.2460195172059578, + "grad_norm": 2.739184617996216, + "learning_rate": 3.277502854967644e-06, + "loss": 1.4302, + "step": 1213 + }, + { + "epoch": 1.2470467385721622, + "grad_norm": 2.798630475997925, + "learning_rate": 3.2755995432051774e-06, + "loss": 1.4134, + "step": 1214 + }, + { + "epoch": 1.2480739599383668, + "grad_norm": 2.835686683654785, + "learning_rate": 3.2736962314427107e-06, + "loss": 1.4258, + "step": 1215 + }, + { + "epoch": 1.2491011813045712, + "grad_norm": 2.7404251098632812, + "learning_rate": 3.271792919680244e-06, + "loss": 1.4347, + "step": 1216 + }, + { + "epoch": 1.2501284026707755, + "grad_norm": 2.8362114429473877, + "learning_rate": 3.269889607917777e-06, + "loss": 1.4909, + "step": 1217 + }, + { + "epoch": 1.25115562403698, + "grad_norm": 2.842283248901367, + "learning_rate": 3.2679862961553104e-06, + "loss": 1.4065, + "step": 1218 + }, + { + "epoch": 1.2521828454031843, + "grad_norm": 2.717536449432373, + "learning_rate": 3.2660829843928438e-06, + "loss": 1.4446, + "step": 1219 + }, + { + "epoch": 1.2532100667693888, + "grad_norm": 2.687009572982788, + "learning_rate": 3.264179672630377e-06, + "loss": 1.4826, + "step": 1220 + }, + { + "epoch": 1.2532100667693888, + "eval_loss": 1.8576964139938354, + "eval_runtime": 37.0759, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1220 + }, + { + "epoch": 1.2542372881355932, + "grad_norm": 2.7234137058258057, + "learning_rate": 3.2622763608679105e-06, + "loss": 1.4503, + "step": 1221 + }, + { + "epoch": 1.2552645095017976, + "grad_norm": 2.733876943588257, + "learning_rate": 3.2603730491054434e-06, + "loss": 1.4076, + "step": 1222 + }, + { + "epoch": 1.2562917308680022, + "grad_norm": 2.6480257511138916, + "learning_rate": 3.258469737342977e-06, + "loss": 1.422, + "step": 1223 + }, + { + "epoch": 1.2573189522342065, + "grad_norm": 2.7006945610046387, + "learning_rate": 3.25656642558051e-06, + "loss": 1.3975, + "step": 1224 + }, + { + "epoch": 1.258346173600411, + "grad_norm": 2.6176297664642334, + "learning_rate": 3.2546631138180435e-06, + "loss": 1.4138, + "step": 1225 + }, + { + "epoch": 1.2593733949666153, + "grad_norm": 2.717742443084717, + "learning_rate": 3.252759802055577e-06, + "loss": 1.4092, + "step": 1226 + }, + { + "epoch": 1.2604006163328196, + "grad_norm": 2.693643093109131, + "learning_rate": 3.25085649029311e-06, + "loss": 1.4221, + "step": 1227 + }, + { + "epoch": 1.2614278376990242, + "grad_norm": 2.709026575088501, + "learning_rate": 3.248953178530644e-06, + "loss": 1.4474, + "step": 1228 + }, + { + "epoch": 1.2624550590652286, + "grad_norm": 2.628399610519409, + "learning_rate": 3.247049866768177e-06, + "loss": 1.4075, + "step": 1229 + }, + { + "epoch": 1.263482280431433, + "grad_norm": 2.681164503097534, + "learning_rate": 3.2451465550057103e-06, + "loss": 1.3683, + "step": 1230 + }, + { + "epoch": 1.263482280431433, + "eval_loss": 1.8543845415115356, + "eval_runtime": 37.1602, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1230 + }, + { + "epoch": 1.2645095017976373, + "grad_norm": 2.6538004875183105, + "learning_rate": 3.2432432432432437e-06, + "loss": 1.4214, + "step": 1231 + }, + { + "epoch": 1.2655367231638417, + "grad_norm": 2.714805841445923, + "learning_rate": 3.241339931480777e-06, + "loss": 1.4617, + "step": 1232 + }, + { + "epoch": 1.2665639445300463, + "grad_norm": 2.743016481399536, + "learning_rate": 3.2394366197183104e-06, + "loss": 1.4556, + "step": 1233 + }, + { + "epoch": 1.2675911658962506, + "grad_norm": 2.618269681930542, + "learning_rate": 3.2375333079558434e-06, + "loss": 1.419, + "step": 1234 + }, + { + "epoch": 1.268618387262455, + "grad_norm": 2.672124147415161, + "learning_rate": 3.2356299961933767e-06, + "loss": 1.4477, + "step": 1235 + }, + { + "epoch": 1.2696456086286596, + "grad_norm": 2.831037998199463, + "learning_rate": 3.23372668443091e-06, + "loss": 1.4552, + "step": 1236 + }, + { + "epoch": 1.270672829994864, + "grad_norm": 2.713193893432617, + "learning_rate": 3.2318233726684435e-06, + "loss": 1.4653, + "step": 1237 + }, + { + "epoch": 1.2717000513610683, + "grad_norm": 2.645097017288208, + "learning_rate": 3.229920060905977e-06, + "loss": 1.4087, + "step": 1238 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.741934061050415, + "learning_rate": 3.2280167491435098e-06, + "loss": 1.4155, + "step": 1239 + }, + { + "epoch": 1.273754494093477, + "grad_norm": 2.5912556648254395, + "learning_rate": 3.226113437381043e-06, + "loss": 1.4214, + "step": 1240 + }, + { + "epoch": 1.273754494093477, + "eval_loss": 1.8524503707885742, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1240 + }, + { + "epoch": 1.2747817154596817, + "grad_norm": 2.5774216651916504, + "learning_rate": 3.2242101256185765e-06, + "loss": 1.4309, + "step": 1241 + }, + { + "epoch": 1.275808936825886, + "grad_norm": 2.7143211364746094, + "learning_rate": 3.22230681385611e-06, + "loss": 1.4928, + "step": 1242 + }, + { + "epoch": 1.2768361581920904, + "grad_norm": 2.8101789951324463, + "learning_rate": 3.220403502093643e-06, + "loss": 1.4325, + "step": 1243 + }, + { + "epoch": 1.2778633795582948, + "grad_norm": 2.704641580581665, + "learning_rate": 3.218500190331176e-06, + "loss": 1.4729, + "step": 1244 + }, + { + "epoch": 1.2788906009244991, + "grad_norm": 2.649195671081543, + "learning_rate": 3.2165968785687095e-06, + "loss": 1.4186, + "step": 1245 + }, + { + "epoch": 1.2799178222907037, + "grad_norm": 2.7863168716430664, + "learning_rate": 3.214693566806243e-06, + "loss": 1.4474, + "step": 1246 + }, + { + "epoch": 1.280945043656908, + "grad_norm": 2.693589687347412, + "learning_rate": 3.2127902550437762e-06, + "loss": 1.4293, + "step": 1247 + }, + { + "epoch": 1.2819722650231125, + "grad_norm": 2.7287509441375732, + "learning_rate": 3.21088694328131e-06, + "loss": 1.4305, + "step": 1248 + }, + { + "epoch": 1.2829994863893168, + "grad_norm": 2.5521676540374756, + "learning_rate": 3.2089836315188434e-06, + "loss": 1.4203, + "step": 1249 + }, + { + "epoch": 1.2840267077555212, + "grad_norm": 2.745425224304199, + "learning_rate": 3.2070803197563767e-06, + "loss": 1.4234, + "step": 1250 + }, + { + "epoch": 1.2840267077555212, + "eval_loss": 1.8586679697036743, + "eval_runtime": 37.1173, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1250 + }, + { + "epoch": 1.2850539291217258, + "grad_norm": 2.687504768371582, + "learning_rate": 3.2051770079939097e-06, + "loss": 1.443, + "step": 1251 + }, + { + "epoch": 1.2860811504879301, + "grad_norm": 2.7841391563415527, + "learning_rate": 3.203273696231443e-06, + "loss": 1.4722, + "step": 1252 + }, + { + "epoch": 1.2871083718541345, + "grad_norm": 2.7715537548065186, + "learning_rate": 3.2013703844689764e-06, + "loss": 1.4063, + "step": 1253 + }, + { + "epoch": 1.288135593220339, + "grad_norm": 2.8183677196502686, + "learning_rate": 3.1994670727065098e-06, + "loss": 1.4534, + "step": 1254 + }, + { + "epoch": 1.2891628145865435, + "grad_norm": 2.795215606689453, + "learning_rate": 3.197563760944043e-06, + "loss": 1.4779, + "step": 1255 + }, + { + "epoch": 1.2901900359527478, + "grad_norm": 2.7606441974639893, + "learning_rate": 3.195660449181576e-06, + "loss": 1.4668, + "step": 1256 + }, + { + "epoch": 1.2912172573189522, + "grad_norm": 2.629718065261841, + "learning_rate": 3.1937571374191094e-06, + "loss": 1.4114, + "step": 1257 + }, + { + "epoch": 1.2922444786851566, + "grad_norm": 2.776845932006836, + "learning_rate": 3.191853825656643e-06, + "loss": 1.4348, + "step": 1258 + }, + { + "epoch": 1.2932717000513612, + "grad_norm": 2.672982931137085, + "learning_rate": 3.189950513894176e-06, + "loss": 1.4274, + "step": 1259 + }, + { + "epoch": 1.2942989214175655, + "grad_norm": 2.5834884643554688, + "learning_rate": 3.1880472021317095e-06, + "loss": 1.4282, + "step": 1260 + }, + { + "epoch": 1.2942989214175655, + "eval_loss": 1.8541877269744873, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1260 + }, + { + "epoch": 1.29532614278377, + "grad_norm": 2.752054452896118, + "learning_rate": 3.1861438903692425e-06, + "loss": 1.3797, + "step": 1261 + }, + { + "epoch": 1.2963533641499743, + "grad_norm": 2.723689556121826, + "learning_rate": 3.184240578606776e-06, + "loss": 1.4137, + "step": 1262 + }, + { + "epoch": 1.2973805855161786, + "grad_norm": 2.806439161300659, + "learning_rate": 3.182337266844309e-06, + "loss": 1.4396, + "step": 1263 + }, + { + "epoch": 1.2984078068823832, + "grad_norm": 2.8140552043914795, + "learning_rate": 3.1804339550818426e-06, + "loss": 1.4677, + "step": 1264 + }, + { + "epoch": 1.2994350282485876, + "grad_norm": 2.78153920173645, + "learning_rate": 3.178530643319376e-06, + "loss": 1.433, + "step": 1265 + }, + { + "epoch": 1.300462249614792, + "grad_norm": 2.6926140785217285, + "learning_rate": 3.176627331556909e-06, + "loss": 1.3901, + "step": 1266 + }, + { + "epoch": 1.3014894709809965, + "grad_norm": 2.636629581451416, + "learning_rate": 3.1747240197944422e-06, + "loss": 1.4331, + "step": 1267 + }, + { + "epoch": 1.3025166923472007, + "grad_norm": 2.625173330307007, + "learning_rate": 3.172820708031976e-06, + "loss": 1.3918, + "step": 1268 + }, + { + "epoch": 1.3035439137134053, + "grad_norm": 2.745666980743408, + "learning_rate": 3.1709173962695094e-06, + "loss": 1.4669, + "step": 1269 + }, + { + "epoch": 1.3045711350796096, + "grad_norm": 2.8781471252441406, + "learning_rate": 3.1690140845070427e-06, + "loss": 1.4586, + "step": 1270 + }, + { + "epoch": 1.3045711350796096, + "eval_loss": 1.8759416341781616, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1270 + }, + { + "epoch": 1.305598356445814, + "grad_norm": 2.832078456878662, + "learning_rate": 3.167110772744576e-06, + "loss": 1.4385, + "step": 1271 + }, + { + "epoch": 1.3066255778120186, + "grad_norm": 2.7153384685516357, + "learning_rate": 3.1652074609821095e-06, + "loss": 1.427, + "step": 1272 + }, + { + "epoch": 1.307652799178223, + "grad_norm": 2.6645848751068115, + "learning_rate": 3.1633041492196424e-06, + "loss": 1.4147, + "step": 1273 + }, + { + "epoch": 1.3086800205444273, + "grad_norm": 2.7231268882751465, + "learning_rate": 3.1614008374571758e-06, + "loss": 1.436, + "step": 1274 + }, + { + "epoch": 1.3097072419106317, + "grad_norm": 2.8131442070007324, + "learning_rate": 3.159497525694709e-06, + "loss": 1.4357, + "step": 1275 + }, + { + "epoch": 1.310734463276836, + "grad_norm": 2.6366660594940186, + "learning_rate": 3.1575942139322425e-06, + "loss": 1.4089, + "step": 1276 + }, + { + "epoch": 1.3117616846430407, + "grad_norm": 2.7849857807159424, + "learning_rate": 3.155690902169776e-06, + "loss": 1.4486, + "step": 1277 + }, + { + "epoch": 1.312788906009245, + "grad_norm": 2.771298408508301, + "learning_rate": 3.1537875904073088e-06, + "loss": 1.4085, + "step": 1278 + }, + { + "epoch": 1.3138161273754494, + "grad_norm": 2.7618181705474854, + "learning_rate": 3.151884278644842e-06, + "loss": 1.4272, + "step": 1279 + }, + { + "epoch": 1.3148433487416538, + "grad_norm": 2.6741080284118652, + "learning_rate": 3.1499809668823755e-06, + "loss": 1.4135, + "step": 1280 + }, + { + "epoch": 1.3148433487416538, + "eval_loss": 1.8567897081375122, + "eval_runtime": 37.1429, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1280 + }, + { + "epoch": 1.3158705701078581, + "grad_norm": 2.6671698093414307, + "learning_rate": 3.148077655119909e-06, + "loss": 1.4598, + "step": 1281 + }, + { + "epoch": 1.3168977914740627, + "grad_norm": 2.6641244888305664, + "learning_rate": 3.1461743433574422e-06, + "loss": 1.3998, + "step": 1282 + }, + { + "epoch": 1.317925012840267, + "grad_norm": 2.747417449951172, + "learning_rate": 3.144271031594975e-06, + "loss": 1.4602, + "step": 1283 + }, + { + "epoch": 1.3189522342064715, + "grad_norm": 2.708984136581421, + "learning_rate": 3.1423677198325085e-06, + "loss": 1.4562, + "step": 1284 + }, + { + "epoch": 1.319979455572676, + "grad_norm": 2.5973010063171387, + "learning_rate": 3.140464408070042e-06, + "loss": 1.4121, + "step": 1285 + }, + { + "epoch": 1.3210066769388804, + "grad_norm": 2.6334621906280518, + "learning_rate": 3.1385610963075753e-06, + "loss": 1.4049, + "step": 1286 + }, + { + "epoch": 1.3220338983050848, + "grad_norm": 2.772655963897705, + "learning_rate": 3.1366577845451086e-06, + "loss": 1.3953, + "step": 1287 + }, + { + "epoch": 1.3230611196712891, + "grad_norm": 2.5006916522979736, + "learning_rate": 3.134754472782642e-06, + "loss": 1.3842, + "step": 1288 + }, + { + "epoch": 1.3240883410374935, + "grad_norm": 2.6052703857421875, + "learning_rate": 3.1328511610201758e-06, + "loss": 1.3801, + "step": 1289 + }, + { + "epoch": 1.325115562403698, + "grad_norm": 2.703693151473999, + "learning_rate": 3.1309478492577087e-06, + "loss": 1.3672, + "step": 1290 + }, + { + "epoch": 1.325115562403698, + "eval_loss": 1.8470649719238281, + "eval_runtime": 37.2714, + "eval_samples_per_second": 26.83, + "eval_steps_per_second": 1.127, + "step": 1290 + }, + { + "epoch": 1.3261427837699025, + "grad_norm": 2.6864330768585205, + "learning_rate": 3.129044537495242e-06, + "loss": 1.4252, + "step": 1291 + }, + { + "epoch": 1.3271700051361068, + "grad_norm": 2.7768399715423584, + "learning_rate": 3.1271412257327754e-06, + "loss": 1.4483, + "step": 1292 + }, + { + "epoch": 1.3281972265023112, + "grad_norm": 2.6658599376678467, + "learning_rate": 3.125237913970309e-06, + "loss": 1.4059, + "step": 1293 + }, + { + "epoch": 1.3292244478685156, + "grad_norm": 2.754718065261841, + "learning_rate": 3.123334602207842e-06, + "loss": 1.3787, + "step": 1294 + }, + { + "epoch": 1.3302516692347202, + "grad_norm": 2.7994842529296875, + "learning_rate": 3.121431290445375e-06, + "loss": 1.4623, + "step": 1295 + }, + { + "epoch": 1.3312788906009245, + "grad_norm": 2.805941343307495, + "learning_rate": 3.1195279786829085e-06, + "loss": 1.5186, + "step": 1296 + }, + { + "epoch": 1.332306111967129, + "grad_norm": 2.6829605102539062, + "learning_rate": 3.117624666920442e-06, + "loss": 1.4052, + "step": 1297 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.7221593856811523, + "learning_rate": 3.115721355157975e-06, + "loss": 1.4031, + "step": 1298 + }, + { + "epoch": 1.3343605546995376, + "grad_norm": 2.579638719558716, + "learning_rate": 3.1138180433955085e-06, + "loss": 1.3627, + "step": 1299 + }, + { + "epoch": 1.3353877760657422, + "grad_norm": 2.861189842224121, + "learning_rate": 3.1119147316330415e-06, + "loss": 1.4405, + "step": 1300 + }, + { + "epoch": 1.3353877760657422, + "eval_loss": 1.8476718664169312, + "eval_runtime": 37.4298, + "eval_samples_per_second": 26.717, + "eval_steps_per_second": 1.122, + "step": 1300 + }, + { + "epoch": 1.3364149974319466, + "grad_norm": 2.8968658447265625, + "learning_rate": 3.110011419870575e-06, + "loss": 1.4774, + "step": 1301 + }, + { + "epoch": 1.337442218798151, + "grad_norm": 2.8708558082580566, + "learning_rate": 3.1081081081081082e-06, + "loss": 1.4085, + "step": 1302 + }, + { + "epoch": 1.3384694401643555, + "grad_norm": 2.802922487258911, + "learning_rate": 3.1062047963456416e-06, + "loss": 1.4486, + "step": 1303 + }, + { + "epoch": 1.33949666153056, + "grad_norm": 2.9320907592773438, + "learning_rate": 3.104301484583175e-06, + "loss": 1.4932, + "step": 1304 + }, + { + "epoch": 1.3405238828967643, + "grad_norm": 2.7708749771118164, + "learning_rate": 3.1023981728207083e-06, + "loss": 1.4601, + "step": 1305 + }, + { + "epoch": 1.3415511042629686, + "grad_norm": 2.7464756965637207, + "learning_rate": 3.1004948610582412e-06, + "loss": 1.4245, + "step": 1306 + }, + { + "epoch": 1.342578325629173, + "grad_norm": 2.767235517501831, + "learning_rate": 3.0985915492957746e-06, + "loss": 1.4199, + "step": 1307 + }, + { + "epoch": 1.3436055469953776, + "grad_norm": 2.6810076236724854, + "learning_rate": 3.096688237533308e-06, + "loss": 1.3851, + "step": 1308 + }, + { + "epoch": 1.344632768361582, + "grad_norm": 2.6390531063079834, + "learning_rate": 3.0947849257708418e-06, + "loss": 1.4688, + "step": 1309 + }, + { + "epoch": 1.3456599897277863, + "grad_norm": 2.729029893875122, + "learning_rate": 3.092881614008375e-06, + "loss": 1.4248, + "step": 1310 + }, + { + "epoch": 1.3456599897277863, + "eval_loss": 1.854402780532837, + "eval_runtime": 36.9357, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.3466872110939907, + "grad_norm": 2.8098297119140625, + "learning_rate": 3.0909783022459085e-06, + "loss": 1.4761, + "step": 1311 + }, + { + "epoch": 1.347714432460195, + "grad_norm": 2.8076772689819336, + "learning_rate": 3.0890749904834414e-06, + "loss": 1.4119, + "step": 1312 + }, + { + "epoch": 1.3487416538263997, + "grad_norm": 2.8214125633239746, + "learning_rate": 3.0871716787209748e-06, + "loss": 1.4693, + "step": 1313 + }, + { + "epoch": 1.349768875192604, + "grad_norm": 2.830083131790161, + "learning_rate": 3.085268366958508e-06, + "loss": 1.438, + "step": 1314 + }, + { + "epoch": 1.3507960965588084, + "grad_norm": 2.7671611309051514, + "learning_rate": 3.0833650551960415e-06, + "loss": 1.4417, + "step": 1315 + }, + { + "epoch": 1.3518233179250128, + "grad_norm": 2.6464383602142334, + "learning_rate": 3.081461743433575e-06, + "loss": 1.4096, + "step": 1316 + }, + { + "epoch": 1.3528505392912171, + "grad_norm": 2.651646137237549, + "learning_rate": 3.079558431671108e-06, + "loss": 1.4264, + "step": 1317 + }, + { + "epoch": 1.3538777606574217, + "grad_norm": 2.6222681999206543, + "learning_rate": 3.077655119908641e-06, + "loss": 1.3836, + "step": 1318 + }, + { + "epoch": 1.354904982023626, + "grad_norm": 2.5674147605895996, + "learning_rate": 3.0757518081461745e-06, + "loss": 1.4206, + "step": 1319 + }, + { + "epoch": 1.3559322033898304, + "grad_norm": 2.7767961025238037, + "learning_rate": 3.073848496383708e-06, + "loss": 1.4348, + "step": 1320 + }, + { + "epoch": 1.3559322033898304, + "eval_loss": 1.8607728481292725, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 1320 + }, + { + "epoch": 1.356959424756035, + "grad_norm": 2.7811837196350098, + "learning_rate": 3.0719451846212413e-06, + "loss": 1.4082, + "step": 1321 + }, + { + "epoch": 1.3579866461222394, + "grad_norm": 2.5633227825164795, + "learning_rate": 3.0700418728587746e-06, + "loss": 1.4352, + "step": 1322 + }, + { + "epoch": 1.3590138674884438, + "grad_norm": 2.7205374240875244, + "learning_rate": 3.0681385610963076e-06, + "loss": 1.405, + "step": 1323 + }, + { + "epoch": 1.3600410888546481, + "grad_norm": 2.82409930229187, + "learning_rate": 3.066235249333841e-06, + "loss": 1.448, + "step": 1324 + }, + { + "epoch": 1.3610683102208525, + "grad_norm": 2.7932820320129395, + "learning_rate": 3.0643319375713743e-06, + "loss": 1.4785, + "step": 1325 + }, + { + "epoch": 1.362095531587057, + "grad_norm": 2.68669056892395, + "learning_rate": 3.0624286258089076e-06, + "loss": 1.4232, + "step": 1326 + }, + { + "epoch": 1.3631227529532615, + "grad_norm": 2.6492018699645996, + "learning_rate": 3.060525314046441e-06, + "loss": 1.4052, + "step": 1327 + }, + { + "epoch": 1.3641499743194658, + "grad_norm": 2.7677114009857178, + "learning_rate": 3.058622002283974e-06, + "loss": 1.4426, + "step": 1328 + }, + { + "epoch": 1.3651771956856702, + "grad_norm": 2.7124569416046143, + "learning_rate": 3.0567186905215077e-06, + "loss": 1.4631, + "step": 1329 + }, + { + "epoch": 1.3662044170518746, + "grad_norm": 2.7084693908691406, + "learning_rate": 3.054815378759041e-06, + "loss": 1.4656, + "step": 1330 + }, + { + "epoch": 1.3662044170518746, + "eval_loss": 1.8457012176513672, + "eval_runtime": 37.0141, + "eval_samples_per_second": 27.017, + "eval_steps_per_second": 1.135, + "step": 1330 + }, + { + "epoch": 1.3672316384180792, + "grad_norm": 2.7612688541412354, + "learning_rate": 3.0529120669965745e-06, + "loss": 1.455, + "step": 1331 + }, + { + "epoch": 1.3682588597842835, + "grad_norm": 2.5172924995422363, + "learning_rate": 3.051008755234108e-06, + "loss": 1.3915, + "step": 1332 + }, + { + "epoch": 1.3692860811504879, + "grad_norm": 2.7225120067596436, + "learning_rate": 3.049105443471641e-06, + "loss": 1.4025, + "step": 1333 + }, + { + "epoch": 1.3703133025166925, + "grad_norm": 2.835470676422119, + "learning_rate": 3.0472021317091745e-06, + "loss": 1.5423, + "step": 1334 + }, + { + "epoch": 1.3713405238828968, + "grad_norm": 2.6247284412384033, + "learning_rate": 3.0452988199467075e-06, + "loss": 1.4562, + "step": 1335 + }, + { + "epoch": 1.3723677452491012, + "grad_norm": 2.6346960067749023, + "learning_rate": 3.043395508184241e-06, + "loss": 1.4566, + "step": 1336 + }, + { + "epoch": 1.3733949666153056, + "grad_norm": 2.6278724670410156, + "learning_rate": 3.041492196421774e-06, + "loss": 1.403, + "step": 1337 + }, + { + "epoch": 1.37442218798151, + "grad_norm": 2.8066110610961914, + "learning_rate": 3.0395888846593076e-06, + "loss": 1.4433, + "step": 1338 + }, + { + "epoch": 1.3754494093477145, + "grad_norm": 2.6565256118774414, + "learning_rate": 3.037685572896841e-06, + "loss": 1.4177, + "step": 1339 + }, + { + "epoch": 1.376476630713919, + "grad_norm": 2.6529884338378906, + "learning_rate": 3.035782261134374e-06, + "loss": 1.4378, + "step": 1340 + }, + { + "epoch": 1.376476630713919, + "eval_loss": 1.8618167638778687, + "eval_runtime": 37.025, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1340 + }, + { + "epoch": 1.3775038520801233, + "grad_norm": 2.820838689804077, + "learning_rate": 3.0338789493719072e-06, + "loss": 1.4471, + "step": 1341 + }, + { + "epoch": 1.3785310734463276, + "grad_norm": 2.6786739826202393, + "learning_rate": 3.0319756376094406e-06, + "loss": 1.4393, + "step": 1342 + }, + { + "epoch": 1.379558294812532, + "grad_norm": 2.7962610721588135, + "learning_rate": 3.030072325846974e-06, + "loss": 1.4407, + "step": 1343 + }, + { + "epoch": 1.3805855161787366, + "grad_norm": 2.739959716796875, + "learning_rate": 3.0281690140845073e-06, + "loss": 1.4059, + "step": 1344 + }, + { + "epoch": 1.381612737544941, + "grad_norm": 2.7812039852142334, + "learning_rate": 3.0262657023220403e-06, + "loss": 1.4449, + "step": 1345 + }, + { + "epoch": 1.3826399589111453, + "grad_norm": 2.749671459197998, + "learning_rate": 3.0243623905595736e-06, + "loss": 1.4459, + "step": 1346 + }, + { + "epoch": 1.3836671802773497, + "grad_norm": 2.7494630813598633, + "learning_rate": 3.022459078797107e-06, + "loss": 1.4204, + "step": 1347 + }, + { + "epoch": 1.384694401643554, + "grad_norm": 2.7188141345977783, + "learning_rate": 3.0205557670346404e-06, + "loss": 1.4249, + "step": 1348 + }, + { + "epoch": 1.3857216230097587, + "grad_norm": 2.712442636489868, + "learning_rate": 3.018652455272174e-06, + "loss": 1.3975, + "step": 1349 + }, + { + "epoch": 1.386748844375963, + "grad_norm": 2.9888060092926025, + "learning_rate": 3.0167491435097075e-06, + "loss": 1.5258, + "step": 1350 + }, + { + "epoch": 1.386748844375963, + "eval_loss": 1.8591758012771606, + "eval_runtime": 37.0078, + "eval_samples_per_second": 27.021, + "eval_steps_per_second": 1.135, + "step": 1350 + }, + { + "epoch": 1.3877760657421674, + "grad_norm": 2.789698600769043, + "learning_rate": 3.014845831747241e-06, + "loss": 1.4305, + "step": 1351 + }, + { + "epoch": 1.388803287108372, + "grad_norm": 2.747840166091919, + "learning_rate": 3.012942519984774e-06, + "loss": 1.4031, + "step": 1352 + }, + { + "epoch": 1.3898305084745763, + "grad_norm": 2.7078518867492676, + "learning_rate": 3.011039208222307e-06, + "loss": 1.4139, + "step": 1353 + }, + { + "epoch": 1.3908577298407807, + "grad_norm": 2.615677833557129, + "learning_rate": 3.0091358964598405e-06, + "loss": 1.3728, + "step": 1354 + }, + { + "epoch": 1.391884951206985, + "grad_norm": 2.7485580444335938, + "learning_rate": 3.007232584697374e-06, + "loss": 1.4272, + "step": 1355 + }, + { + "epoch": 1.3929121725731894, + "grad_norm": 2.7168128490448, + "learning_rate": 3.0053292729349073e-06, + "loss": 1.4353, + "step": 1356 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 2.7295777797698975, + "learning_rate": 3.00342596117244e-06, + "loss": 1.4324, + "step": 1357 + }, + { + "epoch": 1.3949666153055984, + "grad_norm": 2.8791208267211914, + "learning_rate": 3.0015226494099736e-06, + "loss": 1.4303, + "step": 1358 + }, + { + "epoch": 1.3959938366718028, + "grad_norm": 2.647028684616089, + "learning_rate": 2.999619337647507e-06, + "loss": 1.402, + "step": 1359 + }, + { + "epoch": 1.3970210580380071, + "grad_norm": 2.869901180267334, + "learning_rate": 2.9977160258850403e-06, + "loss": 1.404, + "step": 1360 + }, + { + "epoch": 1.3970210580380071, + "eval_loss": 1.8588491678237915, + "eval_runtime": 37.0159, + "eval_samples_per_second": 27.015, + "eval_steps_per_second": 1.135, + "step": 1360 + }, + { + "epoch": 1.3980482794042115, + "grad_norm": 2.7664403915405273, + "learning_rate": 2.9958127141225736e-06, + "loss": 1.3895, + "step": 1361 + }, + { + "epoch": 1.399075500770416, + "grad_norm": 2.8101043701171875, + "learning_rate": 2.9939094023601066e-06, + "loss": 1.4629, + "step": 1362 + }, + { + "epoch": 1.4001027221366205, + "grad_norm": 2.7114098072052, + "learning_rate": 2.99200609059764e-06, + "loss": 1.4365, + "step": 1363 + }, + { + "epoch": 1.4011299435028248, + "grad_norm": 2.6739253997802734, + "learning_rate": 2.9901027788351733e-06, + "loss": 1.382, + "step": 1364 + }, + { + "epoch": 1.4021571648690292, + "grad_norm": 2.8102970123291016, + "learning_rate": 2.9881994670727067e-06, + "loss": 1.392, + "step": 1365 + }, + { + "epoch": 1.4031843862352336, + "grad_norm": 2.645071506500244, + "learning_rate": 2.98629615531024e-06, + "loss": 1.4157, + "step": 1366 + }, + { + "epoch": 1.4042116076014381, + "grad_norm": 2.6110310554504395, + "learning_rate": 2.984392843547773e-06, + "loss": 1.3996, + "step": 1367 + }, + { + "epoch": 1.4052388289676425, + "grad_norm": 2.784090042114258, + "learning_rate": 2.9824895317853063e-06, + "loss": 1.4178, + "step": 1368 + }, + { + "epoch": 1.4062660503338469, + "grad_norm": 2.7293829917907715, + "learning_rate": 2.98058622002284e-06, + "loss": 1.4193, + "step": 1369 + }, + { + "epoch": 1.4072932717000515, + "grad_norm": 2.6924493312835693, + "learning_rate": 2.9786829082603735e-06, + "loss": 1.476, + "step": 1370 + }, + { + "epoch": 1.4072932717000515, + "eval_loss": 1.8642805814743042, + "eval_runtime": 37.0452, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1370 + }, + { + "epoch": 1.4083204930662558, + "grad_norm": 2.7714414596557617, + "learning_rate": 2.976779596497907e-06, + "loss": 1.4574, + "step": 1371 + }, + { + "epoch": 1.4093477144324602, + "grad_norm": 2.7013490200042725, + "learning_rate": 2.97487628473544e-06, + "loss": 1.4718, + "step": 1372 + }, + { + "epoch": 1.4103749357986646, + "grad_norm": 2.9120547771453857, + "learning_rate": 2.9729729729729736e-06, + "loss": 1.4298, + "step": 1373 + }, + { + "epoch": 1.411402157164869, + "grad_norm": 2.9127988815307617, + "learning_rate": 2.9710696612105065e-06, + "loss": 1.4703, + "step": 1374 + }, + { + "epoch": 1.4124293785310735, + "grad_norm": 2.696772575378418, + "learning_rate": 2.96916634944804e-06, + "loss": 1.412, + "step": 1375 + }, + { + "epoch": 1.413456599897278, + "grad_norm": 2.8330435752868652, + "learning_rate": 2.9672630376855732e-06, + "loss": 1.4731, + "step": 1376 + }, + { + "epoch": 1.4144838212634823, + "grad_norm": 2.7354564666748047, + "learning_rate": 2.9653597259231066e-06, + "loss": 1.4342, + "step": 1377 + }, + { + "epoch": 1.4155110426296866, + "grad_norm": 2.7386012077331543, + "learning_rate": 2.96345641416064e-06, + "loss": 1.3935, + "step": 1378 + }, + { + "epoch": 1.416538263995891, + "grad_norm": 2.5881052017211914, + "learning_rate": 2.961553102398173e-06, + "loss": 1.3282, + "step": 1379 + }, + { + "epoch": 1.4175654853620956, + "grad_norm": 2.805558443069458, + "learning_rate": 2.9596497906357063e-06, + "loss": 1.4691, + "step": 1380 + }, + { + "epoch": 1.4175654853620956, + "eval_loss": 1.84714937210083, + "eval_runtime": 37.0349, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1380 + }, + { + "epoch": 1.4185927067283, + "grad_norm": 2.8639137744903564, + "learning_rate": 2.9577464788732396e-06, + "loss": 1.4745, + "step": 1381 + }, + { + "epoch": 1.4196199280945043, + "grad_norm": 2.6688785552978516, + "learning_rate": 2.955843167110773e-06, + "loss": 1.4016, + "step": 1382 + }, + { + "epoch": 1.420647149460709, + "grad_norm": 2.6851978302001953, + "learning_rate": 2.9539398553483064e-06, + "loss": 1.4205, + "step": 1383 + }, + { + "epoch": 1.4216743708269133, + "grad_norm": 2.7381017208099365, + "learning_rate": 2.9520365435858393e-06, + "loss": 1.4315, + "step": 1384 + }, + { + "epoch": 1.4227015921931176, + "grad_norm": 2.9249424934387207, + "learning_rate": 2.9501332318233727e-06, + "loss": 1.49, + "step": 1385 + }, + { + "epoch": 1.423728813559322, + "grad_norm": 2.765490770339966, + "learning_rate": 2.948229920060906e-06, + "loss": 1.4359, + "step": 1386 + }, + { + "epoch": 1.4247560349255264, + "grad_norm": 2.578561544418335, + "learning_rate": 2.9463266082984394e-06, + "loss": 1.4032, + "step": 1387 + }, + { + "epoch": 1.425783256291731, + "grad_norm": 2.6673238277435303, + "learning_rate": 2.9444232965359727e-06, + "loss": 1.4144, + "step": 1388 + }, + { + "epoch": 1.4268104776579353, + "grad_norm": 2.737516164779663, + "learning_rate": 2.9425199847735057e-06, + "loss": 1.4375, + "step": 1389 + }, + { + "epoch": 1.4278376990241397, + "grad_norm": 2.656177282333374, + "learning_rate": 2.94061667301104e-06, + "loss": 1.4007, + "step": 1390 + }, + { + "epoch": 1.4278376990241397, + "eval_loss": 1.8427128791809082, + "eval_runtime": 37.1322, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1390 + }, + { + "epoch": 1.428864920390344, + "grad_norm": 2.6011996269226074, + "learning_rate": 2.938713361248573e-06, + "loss": 1.339, + "step": 1391 + }, + { + "epoch": 1.4298921417565484, + "grad_norm": 2.6374619007110596, + "learning_rate": 2.936810049486106e-06, + "loss": 1.4092, + "step": 1392 + }, + { + "epoch": 1.430919363122753, + "grad_norm": 2.6879520416259766, + "learning_rate": 2.9349067377236396e-06, + "loss": 1.4235, + "step": 1393 + }, + { + "epoch": 1.4319465844889574, + "grad_norm": 2.8400580883026123, + "learning_rate": 2.933003425961173e-06, + "loss": 1.381, + "step": 1394 + }, + { + "epoch": 1.4329738058551618, + "grad_norm": 2.742767810821533, + "learning_rate": 2.9311001141987063e-06, + "loss": 1.4055, + "step": 1395 + }, + { + "epoch": 1.4340010272213661, + "grad_norm": 2.734239101409912, + "learning_rate": 2.9291968024362392e-06, + "loss": 1.3997, + "step": 1396 + }, + { + "epoch": 1.4350282485875705, + "grad_norm": 2.6337532997131348, + "learning_rate": 2.9272934906737726e-06, + "loss": 1.3825, + "step": 1397 + }, + { + "epoch": 1.436055469953775, + "grad_norm": 2.668790578842163, + "learning_rate": 2.925390178911306e-06, + "loss": 1.3918, + "step": 1398 + }, + { + "epoch": 1.4370826913199795, + "grad_norm": 2.8307507038116455, + "learning_rate": 2.9234868671488393e-06, + "loss": 1.4183, + "step": 1399 + }, + { + "epoch": 1.4381099126861838, + "grad_norm": 2.6661787033081055, + "learning_rate": 2.9215835553863727e-06, + "loss": 1.4225, + "step": 1400 + }, + { + "epoch": 1.4381099126861838, + "eval_loss": 1.8518847227096558, + "eval_runtime": 37.3672, + "eval_samples_per_second": 26.761, + "eval_steps_per_second": 1.124, + "step": 1400 + }, + { + "epoch": 1.4391371340523884, + "grad_norm": 2.698025703430176, + "learning_rate": 2.9196802436239056e-06, + "loss": 1.4115, + "step": 1401 + }, + { + "epoch": 1.4401643554185928, + "grad_norm": 2.6792502403259277, + "learning_rate": 2.917776931861439e-06, + "loss": 1.3869, + "step": 1402 + }, + { + "epoch": 1.4411915767847971, + "grad_norm": 2.693514347076416, + "learning_rate": 2.9158736200989723e-06, + "loss": 1.3875, + "step": 1403 + }, + { + "epoch": 1.4422187981510015, + "grad_norm": 2.6383583545684814, + "learning_rate": 2.9139703083365057e-06, + "loss": 1.3942, + "step": 1404 + }, + { + "epoch": 1.4432460195172059, + "grad_norm": 2.6992499828338623, + "learning_rate": 2.912066996574039e-06, + "loss": 1.4243, + "step": 1405 + }, + { + "epoch": 1.4442732408834105, + "grad_norm": 2.7901880741119385, + "learning_rate": 2.910163684811572e-06, + "loss": 1.3559, + "step": 1406 + }, + { + "epoch": 1.4453004622496148, + "grad_norm": 2.9373207092285156, + "learning_rate": 2.9082603730491054e-06, + "loss": 1.4734, + "step": 1407 + }, + { + "epoch": 1.4463276836158192, + "grad_norm": 2.8224740028381348, + "learning_rate": 2.9063570612866387e-06, + "loss": 1.4474, + "step": 1408 + }, + { + "epoch": 1.4473549049820236, + "grad_norm": 2.9534835815429688, + "learning_rate": 2.904453749524172e-06, + "loss": 1.3691, + "step": 1409 + }, + { + "epoch": 1.448382126348228, + "grad_norm": 2.8153076171875, + "learning_rate": 2.902550437761706e-06, + "loss": 1.3954, + "step": 1410 + }, + { + "epoch": 1.448382126348228, + "eval_loss": 1.842891812324524, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1410 + }, + { + "epoch": 1.4494093477144325, + "grad_norm": 2.8490853309631348, + "learning_rate": 2.9006471259992392e-06, + "loss": 1.4114, + "step": 1411 + }, + { + "epoch": 1.450436569080637, + "grad_norm": 2.7475955486297607, + "learning_rate": 2.8987438142367726e-06, + "loss": 1.4068, + "step": 1412 + }, + { + "epoch": 1.4514637904468413, + "grad_norm": 2.7209911346435547, + "learning_rate": 2.8968405024743055e-06, + "loss": 1.4658, + "step": 1413 + }, + { + "epoch": 1.4524910118130456, + "grad_norm": 2.578934907913208, + "learning_rate": 2.894937190711839e-06, + "loss": 1.3686, + "step": 1414 + }, + { + "epoch": 1.45351823317925, + "grad_norm": 2.722381114959717, + "learning_rate": 2.8930338789493723e-06, + "loss": 1.4337, + "step": 1415 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.702178478240967, + "learning_rate": 2.8911305671869056e-06, + "loss": 1.4163, + "step": 1416 + }, + { + "epoch": 1.455572675911659, + "grad_norm": 2.6742405891418457, + "learning_rate": 2.889227255424439e-06, + "loss": 1.3585, + "step": 1417 + }, + { + "epoch": 1.4565998972778633, + "grad_norm": 2.6732680797576904, + "learning_rate": 2.887323943661972e-06, + "loss": 1.406, + "step": 1418 + }, + { + "epoch": 1.457627118644068, + "grad_norm": 2.698864459991455, + "learning_rate": 2.8854206318995053e-06, + "loss": 1.3872, + "step": 1419 + }, + { + "epoch": 1.4586543400102723, + "grad_norm": 2.7534351348876953, + "learning_rate": 2.8835173201370386e-06, + "loss": 1.4055, + "step": 1420 + }, + { + "epoch": 1.4586543400102723, + "eval_loss": 1.8564038276672363, + "eval_runtime": 37.0708, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.4596815613764766, + "grad_norm": 2.9538612365722656, + "learning_rate": 2.881614008374572e-06, + "loss": 1.4596, + "step": 1421 + }, + { + "epoch": 1.460708782742681, + "grad_norm": 2.829641103744507, + "learning_rate": 2.8797106966121054e-06, + "loss": 1.4265, + "step": 1422 + }, + { + "epoch": 1.4617360041088854, + "grad_norm": 2.6814422607421875, + "learning_rate": 2.8778073848496383e-06, + "loss": 1.4112, + "step": 1423 + }, + { + "epoch": 1.46276322547509, + "grad_norm": 2.7880561351776123, + "learning_rate": 2.8759040730871717e-06, + "loss": 1.4435, + "step": 1424 + }, + { + "epoch": 1.4637904468412943, + "grad_norm": 2.676196336746216, + "learning_rate": 2.874000761324705e-06, + "loss": 1.4104, + "step": 1425 + }, + { + "epoch": 1.4648176682074987, + "grad_norm": 2.7417874336242676, + "learning_rate": 2.8720974495622384e-06, + "loss": 1.4335, + "step": 1426 + }, + { + "epoch": 1.465844889573703, + "grad_norm": 2.7870116233825684, + "learning_rate": 2.8701941377997718e-06, + "loss": 1.4226, + "step": 1427 + }, + { + "epoch": 1.4668721109399074, + "grad_norm": 2.765194892883301, + "learning_rate": 2.8682908260373047e-06, + "loss": 1.4582, + "step": 1428 + }, + { + "epoch": 1.467899332306112, + "grad_norm": 2.608175277709961, + "learning_rate": 2.866387514274838e-06, + "loss": 1.4285, + "step": 1429 + }, + { + "epoch": 1.4689265536723164, + "grad_norm": 2.7573461532592773, + "learning_rate": 2.864484202512372e-06, + "loss": 1.4395, + "step": 1430 + }, + { + "epoch": 1.4689265536723164, + "eval_loss": 1.8600319623947144, + "eval_runtime": 37.1192, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1430 + }, + { + "epoch": 1.4699537750385208, + "grad_norm": 2.635175943374634, + "learning_rate": 2.8625808907499052e-06, + "loss": 1.402, + "step": 1431 + }, + { + "epoch": 1.4709809964047253, + "grad_norm": 2.7318811416625977, + "learning_rate": 2.8606775789874386e-06, + "loss": 1.4352, + "step": 1432 + }, + { + "epoch": 1.4720082177709297, + "grad_norm": 2.783669948577881, + "learning_rate": 2.858774267224972e-06, + "loss": 1.4052, + "step": 1433 + }, + { + "epoch": 1.473035439137134, + "grad_norm": 2.7327511310577393, + "learning_rate": 2.8568709554625053e-06, + "loss": 1.4209, + "step": 1434 + }, + { + "epoch": 1.4740626605033385, + "grad_norm": 2.76542329788208, + "learning_rate": 2.8549676437000382e-06, + "loss": 1.4753, + "step": 1435 + }, + { + "epoch": 1.4750898818695428, + "grad_norm": 2.8031232357025146, + "learning_rate": 2.8530643319375716e-06, + "loss": 1.4399, + "step": 1436 + }, + { + "epoch": 1.4761171032357474, + "grad_norm": 2.737053155899048, + "learning_rate": 2.851161020175105e-06, + "loss": 1.4085, + "step": 1437 + }, + { + "epoch": 1.4771443246019518, + "grad_norm": 2.609987735748291, + "learning_rate": 2.8492577084126383e-06, + "loss": 1.3809, + "step": 1438 + }, + { + "epoch": 1.4781715459681561, + "grad_norm": 2.7482123374938965, + "learning_rate": 2.8473543966501717e-06, + "loss": 1.4234, + "step": 1439 + }, + { + "epoch": 1.4791987673343605, + "grad_norm": 2.80330753326416, + "learning_rate": 2.8454510848877046e-06, + "loss": 1.4665, + "step": 1440 + }, + { + "epoch": 1.4791987673343605, + "eval_loss": 1.8637592792510986, + "eval_runtime": 37.1066, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 1440 + }, + { + "epoch": 1.4802259887005649, + "grad_norm": 2.698328733444214, + "learning_rate": 2.843547773125238e-06, + "loss": 1.4224, + "step": 1441 + }, + { + "epoch": 1.4812532100667695, + "grad_norm": 2.663471221923828, + "learning_rate": 2.8416444613627714e-06, + "loss": 1.4205, + "step": 1442 + }, + { + "epoch": 1.4822804314329738, + "grad_norm": 2.7955918312072754, + "learning_rate": 2.8397411496003047e-06, + "loss": 1.3694, + "step": 1443 + }, + { + "epoch": 1.4833076527991782, + "grad_norm": 2.7320404052734375, + "learning_rate": 2.837837837837838e-06, + "loss": 1.4208, + "step": 1444 + }, + { + "epoch": 1.4843348741653826, + "grad_norm": 2.7786989212036133, + "learning_rate": 2.835934526075371e-06, + "loss": 1.4392, + "step": 1445 + }, + { + "epoch": 1.485362095531587, + "grad_norm": 2.6282734870910645, + "learning_rate": 2.8340312143129044e-06, + "loss": 1.3866, + "step": 1446 + }, + { + "epoch": 1.4863893168977915, + "grad_norm": 2.615072011947632, + "learning_rate": 2.8321279025504377e-06, + "loss": 1.3695, + "step": 1447 + }, + { + "epoch": 1.487416538263996, + "grad_norm": 2.884423017501831, + "learning_rate": 2.830224590787971e-06, + "loss": 1.4867, + "step": 1448 + }, + { + "epoch": 1.4884437596302003, + "grad_norm": 2.7745864391326904, + "learning_rate": 2.8283212790255045e-06, + "loss": 1.42, + "step": 1449 + }, + { + "epoch": 1.4894709809964048, + "grad_norm": 2.718130111694336, + "learning_rate": 2.8264179672630383e-06, + "loss": 1.4665, + "step": 1450 + }, + { + "epoch": 1.4894709809964048, + "eval_loss": 1.8556917905807495, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1450 + }, + { + "epoch": 1.4904982023626092, + "grad_norm": 2.7502355575561523, + "learning_rate": 2.8245146555005716e-06, + "loss": 1.3946, + "step": 1451 + }, + { + "epoch": 1.4915254237288136, + "grad_norm": 2.635500192642212, + "learning_rate": 2.8226113437381046e-06, + "loss": 1.3612, + "step": 1452 + }, + { + "epoch": 1.492552645095018, + "grad_norm": 2.617408275604248, + "learning_rate": 2.820708031975638e-06, + "loss": 1.4146, + "step": 1453 + }, + { + "epoch": 1.4935798664612223, + "grad_norm": 2.7706692218780518, + "learning_rate": 2.8188047202131713e-06, + "loss": 1.4474, + "step": 1454 + }, + { + "epoch": 1.494607087827427, + "grad_norm": 2.7609055042266846, + "learning_rate": 2.8169014084507046e-06, + "loss": 1.4618, + "step": 1455 + }, + { + "epoch": 1.4956343091936313, + "grad_norm": 2.757258892059326, + "learning_rate": 2.814998096688238e-06, + "loss": 1.4131, + "step": 1456 + }, + { + "epoch": 1.4966615305598356, + "grad_norm": 2.5547049045562744, + "learning_rate": 2.813094784925771e-06, + "loss": 1.3692, + "step": 1457 + }, + { + "epoch": 1.49768875192604, + "grad_norm": 2.817617177963257, + "learning_rate": 2.8111914731633043e-06, + "loss": 1.4664, + "step": 1458 + }, + { + "epoch": 1.4987159732922444, + "grad_norm": 2.6999337673187256, + "learning_rate": 2.8092881614008377e-06, + "loss": 1.4126, + "step": 1459 + }, + { + "epoch": 1.499743194658449, + "grad_norm": 2.571727991104126, + "learning_rate": 2.807384849638371e-06, + "loss": 1.3828, + "step": 1460 + }, + { + "epoch": 1.499743194658449, + "eval_loss": 1.8413318395614624, + "eval_runtime": 37.1803, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1460 + }, + { + "epoch": 1.5007704160246533, + "grad_norm": 2.615567684173584, + "learning_rate": 2.8054815378759044e-06, + "loss": 1.4336, + "step": 1461 + }, + { + "epoch": 1.5017976373908577, + "grad_norm": 2.696854591369629, + "learning_rate": 2.8035782261134378e-06, + "loss": 1.4015, + "step": 1462 + }, + { + "epoch": 1.5028248587570623, + "grad_norm": 2.7395496368408203, + "learning_rate": 2.8016749143509707e-06, + "loss": 1.4009, + "step": 1463 + }, + { + "epoch": 1.5038520801232664, + "grad_norm": 2.8281755447387695, + "learning_rate": 2.799771602588504e-06, + "loss": 1.4069, + "step": 1464 + }, + { + "epoch": 1.504879301489471, + "grad_norm": 2.641793727874756, + "learning_rate": 2.7978682908260374e-06, + "loss": 1.4341, + "step": 1465 + }, + { + "epoch": 1.5059065228556754, + "grad_norm": 2.8969004154205322, + "learning_rate": 2.795964979063571e-06, + "loss": 1.4493, + "step": 1466 + }, + { + "epoch": 1.5069337442218798, + "grad_norm": 2.758255958557129, + "learning_rate": 2.794061667301104e-06, + "loss": 1.423, + "step": 1467 + }, + { + "epoch": 1.5079609655880843, + "grad_norm": 2.7737441062927246, + "learning_rate": 2.792158355538637e-06, + "loss": 1.43, + "step": 1468 + }, + { + "epoch": 1.5089881869542885, + "grad_norm": 2.798095226287842, + "learning_rate": 2.7902550437761705e-06, + "loss": 1.4276, + "step": 1469 + }, + { + "epoch": 1.510015408320493, + "grad_norm": 2.760437488555908, + "learning_rate": 2.7883517320137042e-06, + "loss": 1.3927, + "step": 1470 + }, + { + "epoch": 1.510015408320493, + "eval_loss": 1.8517959117889404, + "eval_runtime": 37.163, + "eval_samples_per_second": 26.908, + "eval_steps_per_second": 1.13, + "step": 1470 + }, + { + "epoch": 1.5110426296866974, + "grad_norm": 2.689023733139038, + "learning_rate": 2.7864484202512376e-06, + "loss": 1.387, + "step": 1471 + }, + { + "epoch": 1.5120698510529018, + "grad_norm": 2.876424551010132, + "learning_rate": 2.784545108488771e-06, + "loss": 1.4183, + "step": 1472 + }, + { + "epoch": 1.5130970724191064, + "grad_norm": 2.789368152618408, + "learning_rate": 2.7826417967263043e-06, + "loss": 1.4239, + "step": 1473 + }, + { + "epoch": 1.5141242937853108, + "grad_norm": 2.945103883743286, + "learning_rate": 2.7807384849638373e-06, + "loss": 1.458, + "step": 1474 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.694589614868164, + "learning_rate": 2.7788351732013706e-06, + "loss": 1.4145, + "step": 1475 + }, + { + "epoch": 1.5161787365177197, + "grad_norm": 2.7276177406311035, + "learning_rate": 2.776931861438904e-06, + "loss": 1.4063, + "step": 1476 + }, + { + "epoch": 1.5172059578839239, + "grad_norm": 2.694197177886963, + "learning_rate": 2.7750285496764374e-06, + "loss": 1.4125, + "step": 1477 + }, + { + "epoch": 1.5182331792501285, + "grad_norm": 2.832883834838867, + "learning_rate": 2.7731252379139707e-06, + "loss": 1.4176, + "step": 1478 + }, + { + "epoch": 1.5192604006163328, + "grad_norm": 2.67136549949646, + "learning_rate": 2.771221926151504e-06, + "loss": 1.3633, + "step": 1479 + }, + { + "epoch": 1.5202876219825372, + "grad_norm": 2.9618499279022217, + "learning_rate": 2.769318614389037e-06, + "loss": 1.4283, + "step": 1480 + }, + { + "epoch": 1.5202876219825372, + "eval_loss": 1.8486552238464355, + "eval_runtime": 37.1808, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1480 + }, + { + "epoch": 1.5213148433487418, + "grad_norm": 2.91403865814209, + "learning_rate": 2.7674153026265704e-06, + "loss": 1.4254, + "step": 1481 + }, + { + "epoch": 1.522342064714946, + "grad_norm": 2.9121804237365723, + "learning_rate": 2.7655119908641037e-06, + "loss": 1.4251, + "step": 1482 + }, + { + "epoch": 1.5233692860811505, + "grad_norm": 2.7484965324401855, + "learning_rate": 2.763608679101637e-06, + "loss": 1.4141, + "step": 1483 + }, + { + "epoch": 1.5243965074473549, + "grad_norm": 2.913203716278076, + "learning_rate": 2.7617053673391705e-06, + "loss": 1.4516, + "step": 1484 + }, + { + "epoch": 1.5254237288135593, + "grad_norm": 2.7844395637512207, + "learning_rate": 2.7598020555767034e-06, + "loss": 1.45, + "step": 1485 + }, + { + "epoch": 1.5264509501797638, + "grad_norm": 2.8250396251678467, + "learning_rate": 2.7578987438142368e-06, + "loss": 1.3746, + "step": 1486 + }, + { + "epoch": 1.5274781715459682, + "grad_norm": 2.749629497528076, + "learning_rate": 2.75599543205177e-06, + "loss": 1.433, + "step": 1487 + }, + { + "epoch": 1.5285053929121726, + "grad_norm": 2.7559521198272705, + "learning_rate": 2.7540921202893035e-06, + "loss": 1.452, + "step": 1488 + }, + { + "epoch": 1.529532614278377, + "grad_norm": 2.825148344039917, + "learning_rate": 2.752188808526837e-06, + "loss": 1.427, + "step": 1489 + }, + { + "epoch": 1.5305598356445813, + "grad_norm": 2.645327091217041, + "learning_rate": 2.75028549676437e-06, + "loss": 1.4089, + "step": 1490 + }, + { + "epoch": 1.5305598356445813, + "eval_loss": 1.8497828245162964, + "eval_runtime": 37.3756, + "eval_samples_per_second": 26.755, + "eval_steps_per_second": 1.124, + "step": 1490 + }, + { + "epoch": 1.531587057010786, + "grad_norm": 2.753156900405884, + "learning_rate": 2.7483821850019036e-06, + "loss": 1.4163, + "step": 1491 + }, + { + "epoch": 1.5326142783769903, + "grad_norm": 2.8842933177948, + "learning_rate": 2.746478873239437e-06, + "loss": 1.4041, + "step": 1492 + }, + { + "epoch": 1.5336414997431946, + "grad_norm": 2.7529265880584717, + "learning_rate": 2.7445755614769703e-06, + "loss": 1.41, + "step": 1493 + }, + { + "epoch": 1.5346687211093992, + "grad_norm": 2.6907176971435547, + "learning_rate": 2.7426722497145037e-06, + "loss": 1.3841, + "step": 1494 + }, + { + "epoch": 1.5356959424756034, + "grad_norm": 2.793073892593384, + "learning_rate": 2.740768937952037e-06, + "loss": 1.3885, + "step": 1495 + }, + { + "epoch": 1.536723163841808, + "grad_norm": 2.7895116806030273, + "learning_rate": 2.7388656261895704e-06, + "loss": 1.4065, + "step": 1496 + }, + { + "epoch": 1.5377503852080123, + "grad_norm": 2.903615713119507, + "learning_rate": 2.7369623144271033e-06, + "loss": 1.4812, + "step": 1497 + }, + { + "epoch": 1.5387776065742167, + "grad_norm": 2.7081339359283447, + "learning_rate": 2.7350590026646367e-06, + "loss": 1.403, + "step": 1498 + }, + { + "epoch": 1.5398048279404213, + "grad_norm": 2.75414776802063, + "learning_rate": 2.73315569090217e-06, + "loss": 1.3629, + "step": 1499 + }, + { + "epoch": 1.5408320493066254, + "grad_norm": 2.795982599258423, + "learning_rate": 2.7312523791397034e-06, + "loss": 1.4551, + "step": 1500 + }, + { + "epoch": 1.5408320493066254, + "eval_loss": 1.8510199785232544, + "eval_runtime": 37.3806, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 1500 + }, + { + "epoch": 1.54185927067283, + "grad_norm": 2.894488573074341, + "learning_rate": 2.7293490673772368e-06, + "loss": 1.4794, + "step": 1501 + }, + { + "epoch": 1.5428864920390344, + "grad_norm": 2.7982053756713867, + "learning_rate": 2.7274457556147697e-06, + "loss": 1.4365, + "step": 1502 + }, + { + "epoch": 1.5439137134052388, + "grad_norm": 2.913367509841919, + "learning_rate": 2.725542443852303e-06, + "loss": 1.4132, + "step": 1503 + }, + { + "epoch": 1.5449409347714433, + "grad_norm": 2.6658358573913574, + "learning_rate": 2.7236391320898365e-06, + "loss": 1.4288, + "step": 1504 + }, + { + "epoch": 1.5459681561376477, + "grad_norm": 2.710700035095215, + "learning_rate": 2.72173582032737e-06, + "loss": 1.3936, + "step": 1505 + }, + { + "epoch": 1.546995377503852, + "grad_norm": 2.7881481647491455, + "learning_rate": 2.719832508564903e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.5480225988700564, + "grad_norm": 2.7621042728424072, + "learning_rate": 2.717929196802436e-06, + "loss": 1.4218, + "step": 1507 + }, + { + "epoch": 1.5490498202362608, + "grad_norm": 2.7504117488861084, + "learning_rate": 2.7160258850399695e-06, + "loss": 1.3918, + "step": 1508 + }, + { + "epoch": 1.5500770416024654, + "grad_norm": 2.719702959060669, + "learning_rate": 2.714122573277503e-06, + "loss": 1.4154, + "step": 1509 + }, + { + "epoch": 1.5511042629686698, + "grad_norm": 2.6698062419891357, + "learning_rate": 2.712219261515036e-06, + "loss": 1.4084, + "step": 1510 + }, + { + "epoch": 1.5511042629686698, + "eval_loss": 1.8468451499938965, + "eval_runtime": 36.9179, + "eval_samples_per_second": 27.087, + "eval_steps_per_second": 1.138, + "step": 1510 + }, + { + "epoch": 1.5521314843348741, + "grad_norm": 2.816978693008423, + "learning_rate": 2.71031594975257e-06, + "loss": 1.4195, + "step": 1511 + }, + { + "epoch": 1.5531587057010787, + "grad_norm": 2.8047056198120117, + "learning_rate": 2.7084126379901034e-06, + "loss": 1.4395, + "step": 1512 + }, + { + "epoch": 1.5541859270672829, + "grad_norm": 2.8461363315582275, + "learning_rate": 2.7065093262276367e-06, + "loss": 1.4472, + "step": 1513 + }, + { + "epoch": 1.5552131484334875, + "grad_norm": 2.712890863418579, + "learning_rate": 2.7046060144651697e-06, + "loss": 1.4538, + "step": 1514 + }, + { + "epoch": 1.5562403697996918, + "grad_norm": 2.844165802001953, + "learning_rate": 2.702702702702703e-06, + "loss": 1.4719, + "step": 1515 + }, + { + "epoch": 1.5572675911658962, + "grad_norm": 2.629138231277466, + "learning_rate": 2.7007993909402364e-06, + "loss": 1.3441, + "step": 1516 + }, + { + "epoch": 1.5582948125321008, + "grad_norm": 2.7072019577026367, + "learning_rate": 2.6988960791777697e-06, + "loss": 1.4692, + "step": 1517 + }, + { + "epoch": 1.559322033898305, + "grad_norm": 2.739567279815674, + "learning_rate": 2.696992767415303e-06, + "loss": 1.4215, + "step": 1518 + }, + { + "epoch": 1.5603492552645095, + "grad_norm": 2.7846791744232178, + "learning_rate": 2.695089455652836e-06, + "loss": 1.4151, + "step": 1519 + }, + { + "epoch": 1.5613764766307139, + "grad_norm": 2.7417876720428467, + "learning_rate": 2.6931861438903694e-06, + "loss": 1.3797, + "step": 1520 + }, + { + "epoch": 1.5613764766307139, + "eval_loss": 1.847011923789978, + "eval_runtime": 37.0386, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1520 + }, + { + "epoch": 1.5624036979969183, + "grad_norm": 2.743271827697754, + "learning_rate": 2.6912828321279028e-06, + "loss": 1.4372, + "step": 1521 + }, + { + "epoch": 1.5634309193631228, + "grad_norm": 2.8119285106658936, + "learning_rate": 2.689379520365436e-06, + "loss": 1.4075, + "step": 1522 + }, + { + "epoch": 1.5644581407293272, + "grad_norm": 2.7443952560424805, + "learning_rate": 2.6874762086029695e-06, + "loss": 1.4442, + "step": 1523 + }, + { + "epoch": 1.5654853620955316, + "grad_norm": 2.7869346141815186, + "learning_rate": 2.6855728968405024e-06, + "loss": 1.3948, + "step": 1524 + }, + { + "epoch": 1.5665125834617362, + "grad_norm": 2.750185251235962, + "learning_rate": 2.683669585078036e-06, + "loss": 1.4313, + "step": 1525 + }, + { + "epoch": 1.5675398048279403, + "grad_norm": 2.787389039993286, + "learning_rate": 2.681766273315569e-06, + "loss": 1.3748, + "step": 1526 + }, + { + "epoch": 1.568567026194145, + "grad_norm": 2.6675543785095215, + "learning_rate": 2.6798629615531025e-06, + "loss": 1.4089, + "step": 1527 + }, + { + "epoch": 1.5695942475603493, + "grad_norm": 2.6468663215637207, + "learning_rate": 2.677959649790636e-06, + "loss": 1.4148, + "step": 1528 + }, + { + "epoch": 1.5706214689265536, + "grad_norm": 2.7241196632385254, + "learning_rate": 2.676056338028169e-06, + "loss": 1.4498, + "step": 1529 + }, + { + "epoch": 1.5716486902927582, + "grad_norm": 2.705486536026001, + "learning_rate": 2.674153026265702e-06, + "loss": 1.4324, + "step": 1530 + }, + { + "epoch": 1.5716486902927582, + "eval_loss": 1.8467438220977783, + "eval_runtime": 37.0576, + "eval_samples_per_second": 26.985, + "eval_steps_per_second": 1.133, + "step": 1530 + }, + { + "epoch": 1.5726759116589624, + "grad_norm": 2.805429458618164, + "learning_rate": 2.672249714503236e-06, + "loss": 1.4475, + "step": 1531 + }, + { + "epoch": 1.573703133025167, + "grad_norm": 2.7074034214019775, + "learning_rate": 2.6703464027407693e-06, + "loss": 1.3997, + "step": 1532 + }, + { + "epoch": 1.5747303543913713, + "grad_norm": 2.737571954727173, + "learning_rate": 2.6684430909783027e-06, + "loss": 1.4238, + "step": 1533 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.703036069869995, + "learning_rate": 2.666539779215836e-06, + "loss": 1.4107, + "step": 1534 + }, + { + "epoch": 1.5767847971237803, + "grad_norm": 2.7987451553344727, + "learning_rate": 2.6646364674533694e-06, + "loss": 1.3927, + "step": 1535 + }, + { + "epoch": 1.5778120184899846, + "grad_norm": 2.622497081756592, + "learning_rate": 2.6627331556909024e-06, + "loss": 1.3215, + "step": 1536 + }, + { + "epoch": 1.578839239856189, + "grad_norm": 2.5890607833862305, + "learning_rate": 2.6608298439284357e-06, + "loss": 1.3994, + "step": 1537 + }, + { + "epoch": 1.5798664612223934, + "grad_norm": 2.6438405513763428, + "learning_rate": 2.658926532165969e-06, + "loss": 1.4042, + "step": 1538 + }, + { + "epoch": 1.5808936825885977, + "grad_norm": 2.690793752670288, + "learning_rate": 2.6570232204035024e-06, + "loss": 1.3673, + "step": 1539 + }, + { + "epoch": 1.5819209039548023, + "grad_norm": 2.7485456466674805, + "learning_rate": 2.655119908641036e-06, + "loss": 1.3979, + "step": 1540 + }, + { + "epoch": 1.5819209039548023, + "eval_loss": 1.852581262588501, + "eval_runtime": 37.0688, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1540 + }, + { + "epoch": 1.5829481253210067, + "grad_norm": 2.8531415462493896, + "learning_rate": 2.6532165968785687e-06, + "loss": 1.4094, + "step": 1541 + }, + { + "epoch": 1.583975346687211, + "grad_norm": 2.638291358947754, + "learning_rate": 2.651313285116102e-06, + "loss": 1.3708, + "step": 1542 + }, + { + "epoch": 1.5850025680534157, + "grad_norm": 2.717550754547119, + "learning_rate": 2.6494099733536355e-06, + "loss": 1.3821, + "step": 1543 + }, + { + "epoch": 1.5860297894196198, + "grad_norm": 2.91959810256958, + "learning_rate": 2.647506661591169e-06, + "loss": 1.5532, + "step": 1544 + }, + { + "epoch": 1.5870570107858244, + "grad_norm": 2.7281882762908936, + "learning_rate": 2.645603349828702e-06, + "loss": 1.4361, + "step": 1545 + }, + { + "epoch": 1.5880842321520288, + "grad_norm": 2.732283592224121, + "learning_rate": 2.643700038066235e-06, + "loss": 1.3669, + "step": 1546 + }, + { + "epoch": 1.5891114535182331, + "grad_norm": 2.731055498123169, + "learning_rate": 2.6417967263037685e-06, + "loss": 1.4223, + "step": 1547 + }, + { + "epoch": 1.5901386748844377, + "grad_norm": 2.831334352493286, + "learning_rate": 2.639893414541302e-06, + "loss": 1.4454, + "step": 1548 + }, + { + "epoch": 1.5911658962506419, + "grad_norm": 2.920548677444458, + "learning_rate": 2.6379901027788352e-06, + "loss": 1.4686, + "step": 1549 + }, + { + "epoch": 1.5921931176168465, + "grad_norm": 2.6671719551086426, + "learning_rate": 2.6360867910163686e-06, + "loss": 1.398, + "step": 1550 + }, + { + "epoch": 1.5921931176168465, + "eval_loss": 1.857794165611267, + "eval_runtime": 37.0817, + "eval_samples_per_second": 26.967, + "eval_steps_per_second": 1.133, + "step": 1550 + }, + { + "epoch": 1.5932203389830508, + "grad_norm": 2.685175657272339, + "learning_rate": 2.6341834792539024e-06, + "loss": 1.4176, + "step": 1551 + }, + { + "epoch": 1.5942475603492552, + "grad_norm": 2.7412822246551514, + "learning_rate": 2.6322801674914357e-06, + "loss": 1.4195, + "step": 1552 + }, + { + "epoch": 1.5952747817154598, + "grad_norm": 2.7461724281311035, + "learning_rate": 2.6303768557289687e-06, + "loss": 1.4081, + "step": 1553 + }, + { + "epoch": 1.5963020030816641, + "grad_norm": 2.6494853496551514, + "learning_rate": 2.628473543966502e-06, + "loss": 1.3759, + "step": 1554 + }, + { + "epoch": 1.5973292244478685, + "grad_norm": 2.736752986907959, + "learning_rate": 2.6265702322040354e-06, + "loss": 1.4415, + "step": 1555 + }, + { + "epoch": 1.5983564458140729, + "grad_norm": 2.733452796936035, + "learning_rate": 2.6246669204415688e-06, + "loss": 1.3867, + "step": 1556 + }, + { + "epoch": 1.5993836671802772, + "grad_norm": 2.692986249923706, + "learning_rate": 2.622763608679102e-06, + "loss": 1.3811, + "step": 1557 + }, + { + "epoch": 1.6004108885464818, + "grad_norm": 2.580052614212036, + "learning_rate": 2.620860296916635e-06, + "loss": 1.399, + "step": 1558 + }, + { + "epoch": 1.6014381099126862, + "grad_norm": 2.651503086090088, + "learning_rate": 2.6189569851541684e-06, + "loss": 1.34, + "step": 1559 + }, + { + "epoch": 1.6024653312788906, + "grad_norm": 2.8157193660736084, + "learning_rate": 2.617053673391702e-06, + "loss": 1.4437, + "step": 1560 + }, + { + "epoch": 1.6024653312788906, + "eval_loss": 1.8588496446609497, + "eval_runtime": 37.0786, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 1560 + }, + { + "epoch": 1.6034925526450952, + "grad_norm": 2.893589973449707, + "learning_rate": 2.615150361629235e-06, + "loss": 1.4775, + "step": 1561 + }, + { + "epoch": 1.6045197740112993, + "grad_norm": 2.719688653945923, + "learning_rate": 2.6132470498667685e-06, + "loss": 1.432, + "step": 1562 + }, + { + "epoch": 1.605546995377504, + "grad_norm": 2.5626275539398193, + "learning_rate": 2.6113437381043015e-06, + "loss": 1.4118, + "step": 1563 + }, + { + "epoch": 1.6065742167437083, + "grad_norm": 2.6879632472991943, + "learning_rate": 2.609440426341835e-06, + "loss": 1.398, + "step": 1564 + }, + { + "epoch": 1.6076014381099126, + "grad_norm": 2.890155792236328, + "learning_rate": 2.607537114579368e-06, + "loss": 1.4336, + "step": 1565 + }, + { + "epoch": 1.6086286594761172, + "grad_norm": 2.8333349227905273, + "learning_rate": 2.6056338028169015e-06, + "loss": 1.4148, + "step": 1566 + }, + { + "epoch": 1.6096558808423214, + "grad_norm": 2.729215383529663, + "learning_rate": 2.603730491054435e-06, + "loss": 1.3831, + "step": 1567 + }, + { + "epoch": 1.610683102208526, + "grad_norm": 2.800276279449463, + "learning_rate": 2.601827179291968e-06, + "loss": 1.4315, + "step": 1568 + }, + { + "epoch": 1.6117103235747303, + "grad_norm": 2.7900125980377197, + "learning_rate": 2.599923867529501e-06, + "loss": 1.4257, + "step": 1569 + }, + { + "epoch": 1.6127375449409347, + "grad_norm": 2.7753336429595947, + "learning_rate": 2.5980205557670346e-06, + "loss": 1.4089, + "step": 1570 + }, + { + "epoch": 1.6127375449409347, + "eval_loss": 1.8533762693405151, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1570 + }, + { + "epoch": 1.6137647663071393, + "grad_norm": 2.9164156913757324, + "learning_rate": 2.5961172440045684e-06, + "loss": 1.4467, + "step": 1571 + }, + { + "epoch": 1.6147919876733436, + "grad_norm": 2.747676134109497, + "learning_rate": 2.5942139322421017e-06, + "loss": 1.4521, + "step": 1572 + }, + { + "epoch": 1.615819209039548, + "grad_norm": 2.6693432331085205, + "learning_rate": 2.592310620479635e-06, + "loss": 1.3982, + "step": 1573 + }, + { + "epoch": 1.6168464304057526, + "grad_norm": 2.5721216201782227, + "learning_rate": 2.5904073087171684e-06, + "loss": 1.3657, + "step": 1574 + }, + { + "epoch": 1.6178736517719567, + "grad_norm": 2.706143617630005, + "learning_rate": 2.5885039969547014e-06, + "loss": 1.4315, + "step": 1575 + }, + { + "epoch": 1.6189008731381613, + "grad_norm": 2.59006404876709, + "learning_rate": 2.5866006851922347e-06, + "loss": 1.372, + "step": 1576 + }, + { + "epoch": 1.6199280945043657, + "grad_norm": 2.7117512226104736, + "learning_rate": 2.584697373429768e-06, + "loss": 1.3834, + "step": 1577 + }, + { + "epoch": 1.62095531587057, + "grad_norm": 2.718029737472534, + "learning_rate": 2.5827940616673015e-06, + "loss": 1.3978, + "step": 1578 + }, + { + "epoch": 1.6219825372367747, + "grad_norm": 2.7660539150238037, + "learning_rate": 2.580890749904835e-06, + "loss": 1.4505, + "step": 1579 + }, + { + "epoch": 1.6230097586029788, + "grad_norm": 2.7671236991882324, + "learning_rate": 2.5789874381423678e-06, + "loss": 1.3951, + "step": 1580 + }, + { + "epoch": 1.6230097586029788, + "eval_loss": 1.8412485122680664, + "eval_runtime": 37.1162, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.6240369799691834, + "grad_norm": 2.6504807472229004, + "learning_rate": 2.577084126379901e-06, + "loss": 1.3957, + "step": 1581 + }, + { + "epoch": 1.6250642013353878, + "grad_norm": 2.688166618347168, + "learning_rate": 2.5751808146174345e-06, + "loss": 1.3686, + "step": 1582 + }, + { + "epoch": 1.6260914227015921, + "grad_norm": 2.7034637928009033, + "learning_rate": 2.573277502854968e-06, + "loss": 1.4201, + "step": 1583 + }, + { + "epoch": 1.6271186440677967, + "grad_norm": 2.6097683906555176, + "learning_rate": 2.5713741910925012e-06, + "loss": 1.3881, + "step": 1584 + }, + { + "epoch": 1.6281458654340009, + "grad_norm": 2.6197433471679688, + "learning_rate": 2.569470879330034e-06, + "loss": 1.3962, + "step": 1585 + }, + { + "epoch": 1.6291730868002054, + "grad_norm": 3.0436367988586426, + "learning_rate": 2.5675675675675675e-06, + "loss": 1.4714, + "step": 1586 + }, + { + "epoch": 1.6302003081664098, + "grad_norm": 2.708744764328003, + "learning_rate": 2.565664255805101e-06, + "loss": 1.4007, + "step": 1587 + }, + { + "epoch": 1.6312275295326142, + "grad_norm": 2.708279848098755, + "learning_rate": 2.5637609440426343e-06, + "loss": 1.4284, + "step": 1588 + }, + { + "epoch": 1.6322547508988188, + "grad_norm": 2.7955429553985596, + "learning_rate": 2.5618576322801676e-06, + "loss": 1.3736, + "step": 1589 + }, + { + "epoch": 1.6332819722650231, + "grad_norm": 2.890211820602417, + "learning_rate": 2.5599543205177006e-06, + "loss": 1.4671, + "step": 1590 + }, + { + "epoch": 1.6332819722650231, + "eval_loss": 1.8467649221420288, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 1590 + }, + { + "epoch": 1.6343091936312275, + "grad_norm": 2.698066234588623, + "learning_rate": 2.558051008755234e-06, + "loss": 1.4229, + "step": 1591 + }, + { + "epoch": 1.635336414997432, + "grad_norm": 2.8393352031707764, + "learning_rate": 2.5561476969927677e-06, + "loss": 1.469, + "step": 1592 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.589914321899414, + "learning_rate": 2.554244385230301e-06, + "loss": 1.3853, + "step": 1593 + }, + { + "epoch": 1.6373908577298408, + "grad_norm": 2.630443572998047, + "learning_rate": 2.5523410734678344e-06, + "loss": 1.4176, + "step": 1594 + }, + { + "epoch": 1.6384180790960452, + "grad_norm": 2.784212112426758, + "learning_rate": 2.550437761705368e-06, + "loss": 1.344, + "step": 1595 + }, + { + "epoch": 1.6394453004622496, + "grad_norm": 2.736009359359741, + "learning_rate": 2.548534449942901e-06, + "loss": 1.3867, + "step": 1596 + }, + { + "epoch": 1.6404725218284542, + "grad_norm": 2.9134042263031006, + "learning_rate": 2.546631138180434e-06, + "loss": 1.4214, + "step": 1597 + }, + { + "epoch": 1.6414997431946583, + "grad_norm": 2.995575428009033, + "learning_rate": 2.5447278264179675e-06, + "loss": 1.409, + "step": 1598 + }, + { + "epoch": 1.642526964560863, + "grad_norm": 2.744499683380127, + "learning_rate": 2.542824514655501e-06, + "loss": 1.3727, + "step": 1599 + }, + { + "epoch": 1.6435541859270673, + "grad_norm": 2.835494041442871, + "learning_rate": 2.540921202893034e-06, + "loss": 1.4636, + "step": 1600 + }, + { + "epoch": 1.6435541859270673, + "eval_loss": 1.8454105854034424, + "eval_runtime": 37.4381, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1600 + }, + { + "epoch": 1.6445814072932716, + "grad_norm": 2.817537307739258, + "learning_rate": 2.5390178911305675e-06, + "loss": 1.4416, + "step": 1601 + }, + { + "epoch": 1.6456086286594762, + "grad_norm": 2.868093252182007, + "learning_rate": 2.5371145793681005e-06, + "loss": 1.4187, + "step": 1602 + }, + { + "epoch": 1.6466358500256806, + "grad_norm": 2.861865758895874, + "learning_rate": 2.535211267605634e-06, + "loss": 1.4067, + "step": 1603 + }, + { + "epoch": 1.647663071391885, + "grad_norm": 2.7015087604522705, + "learning_rate": 2.533307955843167e-06, + "loss": 1.3528, + "step": 1604 + }, + { + "epoch": 1.6486902927580893, + "grad_norm": 2.7585439682006836, + "learning_rate": 2.5314046440807006e-06, + "loss": 1.4157, + "step": 1605 + }, + { + "epoch": 1.6497175141242937, + "grad_norm": 2.853614330291748, + "learning_rate": 2.529501332318234e-06, + "loss": 1.3932, + "step": 1606 + }, + { + "epoch": 1.6507447354904983, + "grad_norm": 2.9736239910125732, + "learning_rate": 2.5275980205557673e-06, + "loss": 1.4511, + "step": 1607 + }, + { + "epoch": 1.6517719568567026, + "grad_norm": 2.955389976501465, + "learning_rate": 2.5256947087933002e-06, + "loss": 1.4777, + "step": 1608 + }, + { + "epoch": 1.652799178222907, + "grad_norm": 2.9508774280548096, + "learning_rate": 2.5237913970308336e-06, + "loss": 1.4523, + "step": 1609 + }, + { + "epoch": 1.6538263995891116, + "grad_norm": 2.88675856590271, + "learning_rate": 2.521888085268367e-06, + "loss": 1.4064, + "step": 1610 + }, + { + "epoch": 1.6538263995891116, + "eval_loss": 1.8569127321243286, + "eval_runtime": 36.9819, + "eval_samples_per_second": 27.04, + "eval_steps_per_second": 1.136, + "step": 1610 + }, + { + "epoch": 1.6548536209553157, + "grad_norm": 2.8439440727233887, + "learning_rate": 2.5199847735059003e-06, + "loss": 1.4591, + "step": 1611 + }, + { + "epoch": 1.6558808423215203, + "grad_norm": 2.9314420223236084, + "learning_rate": 2.518081461743434e-06, + "loss": 1.39, + "step": 1612 + }, + { + "epoch": 1.6569080636877247, + "grad_norm": 2.6517245769500732, + "learning_rate": 2.5161781499809675e-06, + "loss": 1.4234, + "step": 1613 + }, + { + "epoch": 1.657935285053929, + "grad_norm": 2.5690908432006836, + "learning_rate": 2.5142748382185004e-06, + "loss": 1.3762, + "step": 1614 + }, + { + "epoch": 1.6589625064201337, + "grad_norm": 2.808892011642456, + "learning_rate": 2.5123715264560338e-06, + "loss": 1.3727, + "step": 1615 + }, + { + "epoch": 1.6599897277863378, + "grad_norm": 2.777846336364746, + "learning_rate": 2.510468214693567e-06, + "loss": 1.3941, + "step": 1616 + }, + { + "epoch": 1.6610169491525424, + "grad_norm": 2.8298423290252686, + "learning_rate": 2.5085649029311005e-06, + "loss": 1.3455, + "step": 1617 + }, + { + "epoch": 1.6620441705187468, + "grad_norm": 2.6464784145355225, + "learning_rate": 2.506661591168634e-06, + "loss": 1.3061, + "step": 1618 + }, + { + "epoch": 1.6630713918849511, + "grad_norm": 2.7149951457977295, + "learning_rate": 2.504758279406167e-06, + "loss": 1.3995, + "step": 1619 + }, + { + "epoch": 1.6640986132511557, + "grad_norm": 3.0328569412231445, + "learning_rate": 2.5028549676437e-06, + "loss": 1.4343, + "step": 1620 + }, + { + "epoch": 1.6640986132511557, + "eval_loss": 1.8629075288772583, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1620 + }, + { + "epoch": 1.66512583461736, + "grad_norm": 2.8838441371917725, + "learning_rate": 2.5009516558812335e-06, + "loss": 1.3817, + "step": 1621 + }, + { + "epoch": 1.6661530559835644, + "grad_norm": 2.843242645263672, + "learning_rate": 2.499048344118767e-06, + "loss": 1.4322, + "step": 1622 + }, + { + "epoch": 1.667180277349769, + "grad_norm": 2.642946720123291, + "learning_rate": 2.4971450323563002e-06, + "loss": 1.4702, + "step": 1623 + }, + { + "epoch": 1.6682074987159732, + "grad_norm": 2.666194200515747, + "learning_rate": 2.4952417205938336e-06, + "loss": 1.3808, + "step": 1624 + }, + { + "epoch": 1.6692347200821778, + "grad_norm": 2.7515366077423096, + "learning_rate": 2.4933384088313666e-06, + "loss": 1.3892, + "step": 1625 + }, + { + "epoch": 1.6702619414483821, + "grad_norm": 2.7051661014556885, + "learning_rate": 2.4914350970689e-06, + "loss": 1.3809, + "step": 1626 + }, + { + "epoch": 1.6712891628145865, + "grad_norm": 2.7311975955963135, + "learning_rate": 2.4895317853064337e-06, + "loss": 1.3952, + "step": 1627 + }, + { + "epoch": 1.672316384180791, + "grad_norm": 2.8365893363952637, + "learning_rate": 2.4876284735439666e-06, + "loss": 1.4505, + "step": 1628 + }, + { + "epoch": 1.6733436055469952, + "grad_norm": 2.644507884979248, + "learning_rate": 2.4857251617815e-06, + "loss": 1.4114, + "step": 1629 + }, + { + "epoch": 1.6743708269131998, + "grad_norm": 2.7952826023101807, + "learning_rate": 2.4838218500190334e-06, + "loss": 1.4223, + "step": 1630 + }, + { + "epoch": 1.6743708269131998, + "eval_loss": 1.8577097654342651, + "eval_runtime": 37.1645, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 1630 + }, + { + "epoch": 1.6753980482794042, + "grad_norm": 2.787752866744995, + "learning_rate": 2.4819185382565667e-06, + "loss": 1.4022, + "step": 1631 + }, + { + "epoch": 1.6764252696456086, + "grad_norm": 2.6963443756103516, + "learning_rate": 2.4800152264941e-06, + "loss": 1.3974, + "step": 1632 + }, + { + "epoch": 1.6774524910118132, + "grad_norm": 2.6753835678100586, + "learning_rate": 2.478111914731633e-06, + "loss": 1.3639, + "step": 1633 + }, + { + "epoch": 1.6784797123780173, + "grad_norm": 2.7086808681488037, + "learning_rate": 2.4762086029691664e-06, + "loss": 1.3987, + "step": 1634 + }, + { + "epoch": 1.6795069337442219, + "grad_norm": 3.2228872776031494, + "learning_rate": 2.4743052912066998e-06, + "loss": 1.5202, + "step": 1635 + }, + { + "epoch": 1.6805341551104263, + "grad_norm": 2.8254599571228027, + "learning_rate": 2.472401979444233e-06, + "loss": 1.3781, + "step": 1636 + }, + { + "epoch": 1.6815613764766306, + "grad_norm": 2.9406354427337646, + "learning_rate": 2.4704986676817665e-06, + "loss": 1.4686, + "step": 1637 + }, + { + "epoch": 1.6825885978428352, + "grad_norm": 2.8287105560302734, + "learning_rate": 2.4685953559193e-06, + "loss": 1.4026, + "step": 1638 + }, + { + "epoch": 1.6836158192090396, + "grad_norm": 2.923633337020874, + "learning_rate": 2.466692044156833e-06, + "loss": 1.4493, + "step": 1639 + }, + { + "epoch": 1.684643040575244, + "grad_norm": 2.9239513874053955, + "learning_rate": 2.4647887323943666e-06, + "loss": 1.3878, + "step": 1640 + }, + { + "epoch": 1.684643040575244, + "eval_loss": 1.853607416152954, + "eval_runtime": 37.1661, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1640 + }, + { + "epoch": 1.6856702619414485, + "grad_norm": 2.699998617172241, + "learning_rate": 2.4628854206319e-06, + "loss": 1.3637, + "step": 1641 + }, + { + "epoch": 1.6866974833076527, + "grad_norm": 2.901700258255005, + "learning_rate": 2.460982108869433e-06, + "loss": 1.3554, + "step": 1642 + }, + { + "epoch": 1.6877247046738573, + "grad_norm": 3.011157512664795, + "learning_rate": 2.4590787971069662e-06, + "loss": 1.4487, + "step": 1643 + }, + { + "epoch": 1.6887519260400616, + "grad_norm": 2.672830581665039, + "learning_rate": 2.4571754853444996e-06, + "loss": 1.3906, + "step": 1644 + }, + { + "epoch": 1.689779147406266, + "grad_norm": 2.7201497554779053, + "learning_rate": 2.455272173582033e-06, + "loss": 1.3747, + "step": 1645 + }, + { + "epoch": 1.6908063687724706, + "grad_norm": 2.979395627975464, + "learning_rate": 2.4533688618195663e-06, + "loss": 1.3895, + "step": 1646 + }, + { + "epoch": 1.6918335901386747, + "grad_norm": 2.8048512935638428, + "learning_rate": 2.4514655500570997e-06, + "loss": 1.3271, + "step": 1647 + }, + { + "epoch": 1.6928608115048793, + "grad_norm": 2.6457700729370117, + "learning_rate": 2.449562238294633e-06, + "loss": 1.3679, + "step": 1648 + }, + { + "epoch": 1.6938880328710837, + "grad_norm": 2.732454299926758, + "learning_rate": 2.4476589265321664e-06, + "loss": 1.3332, + "step": 1649 + }, + { + "epoch": 1.694915254237288, + "grad_norm": 3.190871238708496, + "learning_rate": 2.4457556147696993e-06, + "loss": 1.4437, + "step": 1650 + }, + { + "epoch": 1.694915254237288, + "eval_loss": 1.8642219305038452, + "eval_runtime": 37.1865, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 1650 + }, + { + "epoch": 1.6959424756034926, + "grad_norm": 2.85784912109375, + "learning_rate": 2.4438523030072327e-06, + "loss": 1.4121, + "step": 1651 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.766545534133911, + "learning_rate": 2.441948991244766e-06, + "loss": 1.4231, + "step": 1652 + }, + { + "epoch": 1.6979969183359014, + "grad_norm": 2.8962197303771973, + "learning_rate": 2.4400456794822994e-06, + "loss": 1.4146, + "step": 1653 + }, + { + "epoch": 1.6990241397021058, + "grad_norm": 2.907231569290161, + "learning_rate": 2.438142367719833e-06, + "loss": 1.4375, + "step": 1654 + }, + { + "epoch": 1.7000513610683101, + "grad_norm": 2.8746557235717773, + "learning_rate": 2.4362390559573657e-06, + "loss": 1.4002, + "step": 1655 + }, + { + "epoch": 1.7010785824345147, + "grad_norm": 2.741013765335083, + "learning_rate": 2.434335744194899e-06, + "loss": 1.3865, + "step": 1656 + }, + { + "epoch": 1.702105803800719, + "grad_norm": 3.118908643722534, + "learning_rate": 2.432432432432433e-06, + "loss": 1.4412, + "step": 1657 + }, + { + "epoch": 1.7031330251669234, + "grad_norm": 2.789179563522339, + "learning_rate": 2.4305291206699662e-06, + "loss": 1.3446, + "step": 1658 + }, + { + "epoch": 1.704160246533128, + "grad_norm": 2.9241371154785156, + "learning_rate": 2.428625808907499e-06, + "loss": 1.4263, + "step": 1659 + }, + { + "epoch": 1.7051874678993322, + "grad_norm": 2.8632888793945312, + "learning_rate": 2.4267224971450325e-06, + "loss": 1.4413, + "step": 1660 + }, + { + "epoch": 1.7051874678993322, + "eval_loss": 1.85648512840271, + "eval_runtime": 37.199, + "eval_samples_per_second": 26.882, + "eval_steps_per_second": 1.129, + "step": 1660 + }, + { + "epoch": 1.7062146892655368, + "grad_norm": 2.948183059692383, + "learning_rate": 2.424819185382566e-06, + "loss": 1.4058, + "step": 1661 + }, + { + "epoch": 1.7072419106317411, + "grad_norm": 3.04127836227417, + "learning_rate": 2.4229158736200993e-06, + "loss": 1.3582, + "step": 1662 + }, + { + "epoch": 1.7082691319979455, + "grad_norm": 2.6956393718719482, + "learning_rate": 2.4210125618576326e-06, + "loss": 1.3705, + "step": 1663 + }, + { + "epoch": 1.70929635336415, + "grad_norm": 2.7690250873565674, + "learning_rate": 2.4191092500951656e-06, + "loss": 1.3869, + "step": 1664 + }, + { + "epoch": 1.7103235747303542, + "grad_norm": 2.851231813430786, + "learning_rate": 2.417205938332699e-06, + "loss": 1.4198, + "step": 1665 + }, + { + "epoch": 1.7113507960965588, + "grad_norm": 2.9241318702697754, + "learning_rate": 2.4153026265702323e-06, + "loss": 1.3441, + "step": 1666 + }, + { + "epoch": 1.7123780174627632, + "grad_norm": 2.7888636589050293, + "learning_rate": 2.4133993148077657e-06, + "loss": 1.3765, + "step": 1667 + }, + { + "epoch": 1.7134052388289676, + "grad_norm": 2.891366481781006, + "learning_rate": 2.411496003045299e-06, + "loss": 1.4393, + "step": 1668 + }, + { + "epoch": 1.7144324601951721, + "grad_norm": 2.8162693977355957, + "learning_rate": 2.4095926912828324e-06, + "loss": 1.3811, + "step": 1669 + }, + { + "epoch": 1.7154596815613765, + "grad_norm": 2.6741931438446045, + "learning_rate": 2.4076893795203658e-06, + "loss": 1.404, + "step": 1670 + }, + { + "epoch": 1.7154596815613765, + "eval_loss": 1.8440946340560913, + "eval_runtime": 37.1273, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1670 + }, + { + "epoch": 1.7164869029275809, + "grad_norm": 2.834951400756836, + "learning_rate": 2.405786067757899e-06, + "loss": 1.377, + "step": 1671 + }, + { + "epoch": 1.7175141242937855, + "grad_norm": 2.846113681793213, + "learning_rate": 2.403882755995432e-06, + "loss": 1.4689, + "step": 1672 + }, + { + "epoch": 1.7185413456599896, + "grad_norm": 2.6862854957580566, + "learning_rate": 2.4019794442329654e-06, + "loss": 1.3979, + "step": 1673 + }, + { + "epoch": 1.7195685670261942, + "grad_norm": 2.7214972972869873, + "learning_rate": 2.4000761324704988e-06, + "loss": 1.3738, + "step": 1674 + }, + { + "epoch": 1.7205957883923986, + "grad_norm": 2.7037723064422607, + "learning_rate": 2.398172820708032e-06, + "loss": 1.4184, + "step": 1675 + }, + { + "epoch": 1.721623009758603, + "grad_norm": 2.7854912281036377, + "learning_rate": 2.3962695089455655e-06, + "loss": 1.3854, + "step": 1676 + }, + { + "epoch": 1.7226502311248075, + "grad_norm": 2.640059471130371, + "learning_rate": 2.3943661971830984e-06, + "loss": 1.3656, + "step": 1677 + }, + { + "epoch": 1.7236774524910117, + "grad_norm": 2.6955630779266357, + "learning_rate": 2.3924628854206322e-06, + "loss": 1.3764, + "step": 1678 + }, + { + "epoch": 1.7247046738572163, + "grad_norm": 3.018975257873535, + "learning_rate": 2.3905595736581656e-06, + "loss": 1.3922, + "step": 1679 + }, + { + "epoch": 1.7257318952234206, + "grad_norm": 2.7709803581237793, + "learning_rate": 2.388656261895699e-06, + "loss": 1.4288, + "step": 1680 + }, + { + "epoch": 1.7257318952234206, + "eval_loss": 1.8474006652832031, + "eval_runtime": 37.1233, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 1680 + }, + { + "epoch": 1.726759116589625, + "grad_norm": 2.8898143768310547, + "learning_rate": 2.386752950133232e-06, + "loss": 1.4143, + "step": 1681 + }, + { + "epoch": 1.7277863379558296, + "grad_norm": 2.9092485904693604, + "learning_rate": 2.3848496383707653e-06, + "loss": 1.4097, + "step": 1682 + }, + { + "epoch": 1.7288135593220337, + "grad_norm": 2.886662483215332, + "learning_rate": 2.3829463266082986e-06, + "loss": 1.3968, + "step": 1683 + }, + { + "epoch": 1.7298407806882383, + "grad_norm": 2.7398223876953125, + "learning_rate": 2.381043014845832e-06, + "loss": 1.4198, + "step": 1684 + }, + { + "epoch": 1.7308680020544427, + "grad_norm": 2.626732587814331, + "learning_rate": 2.3791397030833653e-06, + "loss": 1.4116, + "step": 1685 + }, + { + "epoch": 1.731895223420647, + "grad_norm": 2.6434433460235596, + "learning_rate": 2.3772363913208983e-06, + "loss": 1.3857, + "step": 1686 + }, + { + "epoch": 1.7329224447868516, + "grad_norm": 2.83876633644104, + "learning_rate": 2.3753330795584316e-06, + "loss": 1.3672, + "step": 1687 + }, + { + "epoch": 1.733949666153056, + "grad_norm": 2.925100564956665, + "learning_rate": 2.3734297677959654e-06, + "loss": 1.3957, + "step": 1688 + }, + { + "epoch": 1.7349768875192604, + "grad_norm": 2.833876371383667, + "learning_rate": 2.3715264560334984e-06, + "loss": 1.442, + "step": 1689 + }, + { + "epoch": 1.736004108885465, + "grad_norm": 2.8355112075805664, + "learning_rate": 2.3696231442710317e-06, + "loss": 1.4153, + "step": 1690 + }, + { + "epoch": 1.736004108885465, + "eval_loss": 1.8604445457458496, + "eval_runtime": 37.2018, + "eval_samples_per_second": 26.88, + "eval_steps_per_second": 1.129, + "step": 1690 + }, + { + "epoch": 1.7370313302516691, + "grad_norm": 2.7500364780426025, + "learning_rate": 2.367719832508565e-06, + "loss": 1.4165, + "step": 1691 + }, + { + "epoch": 1.7380585516178737, + "grad_norm": 2.87259578704834, + "learning_rate": 2.3658165207460985e-06, + "loss": 1.4431, + "step": 1692 + }, + { + "epoch": 1.739085772984078, + "grad_norm": 2.899538993835449, + "learning_rate": 2.363913208983632e-06, + "loss": 1.4134, + "step": 1693 + }, + { + "epoch": 1.7401129943502824, + "grad_norm": 2.799593687057495, + "learning_rate": 2.362009897221165e-06, + "loss": 1.4249, + "step": 1694 + }, + { + "epoch": 1.741140215716487, + "grad_norm": 2.8966968059539795, + "learning_rate": 2.360106585458698e-06, + "loss": 1.4184, + "step": 1695 + }, + { + "epoch": 1.7421674370826912, + "grad_norm": 2.6944682598114014, + "learning_rate": 2.3582032736962315e-06, + "loss": 1.3678, + "step": 1696 + }, + { + "epoch": 1.7431946584488958, + "grad_norm": 2.819845199584961, + "learning_rate": 2.356299961933765e-06, + "loss": 1.407, + "step": 1697 + }, + { + "epoch": 1.7442218798151001, + "grad_norm": 2.6873581409454346, + "learning_rate": 2.354396650171298e-06, + "loss": 1.3771, + "step": 1698 + }, + { + "epoch": 1.7452491011813045, + "grad_norm": 2.6941304206848145, + "learning_rate": 2.3524933384088316e-06, + "loss": 1.3743, + "step": 1699 + }, + { + "epoch": 1.746276322547509, + "grad_norm": 2.925926685333252, + "learning_rate": 2.350590026646365e-06, + "loss": 1.4549, + "step": 1700 + }, + { + "epoch": 1.746276322547509, + "eval_loss": 1.8619909286499023, + "eval_runtime": 37.3963, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 1700 + }, + { + "epoch": 1.7473035439137135, + "grad_norm": 2.7275006771087646, + "learning_rate": 2.3486867148838983e-06, + "loss": 1.385, + "step": 1701 + }, + { + "epoch": 1.7483307652799178, + "grad_norm": 2.7096338272094727, + "learning_rate": 2.3467834031214317e-06, + "loss": 1.3732, + "step": 1702 + }, + { + "epoch": 1.7493579866461222, + "grad_norm": 2.718862295150757, + "learning_rate": 2.3448800913589646e-06, + "loss": 1.3889, + "step": 1703 + }, + { + "epoch": 1.7503852080123266, + "grad_norm": 2.8386173248291016, + "learning_rate": 2.342976779596498e-06, + "loss": 1.4083, + "step": 1704 + }, + { + "epoch": 1.7514124293785311, + "grad_norm": 2.868790864944458, + "learning_rate": 2.3410734678340313e-06, + "loss": 1.3958, + "step": 1705 + }, + { + "epoch": 1.7524396507447355, + "grad_norm": 2.869119882583618, + "learning_rate": 2.3391701560715647e-06, + "loss": 1.4344, + "step": 1706 + }, + { + "epoch": 1.7534668721109399, + "grad_norm": 2.702211380004883, + "learning_rate": 2.337266844309098e-06, + "loss": 1.3928, + "step": 1707 + }, + { + "epoch": 1.7544940934771445, + "grad_norm": 2.6727139949798584, + "learning_rate": 2.3353635325466314e-06, + "loss": 1.3872, + "step": 1708 + }, + { + "epoch": 1.7555213148433486, + "grad_norm": 2.875605583190918, + "learning_rate": 2.3334602207841648e-06, + "loss": 1.4097, + "step": 1709 + }, + { + "epoch": 1.7565485362095532, + "grad_norm": 2.7349722385406494, + "learning_rate": 2.331556909021698e-06, + "loss": 1.4041, + "step": 1710 + }, + { + "epoch": 1.7565485362095532, + "eval_loss": 1.8533939123153687, + "eval_runtime": 36.9293, + "eval_samples_per_second": 27.079, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 2.7475039958953857, + "learning_rate": 2.3296535972592315e-06, + "loss": 1.4339, + "step": 1711 + }, + { + "epoch": 1.758602978941962, + "grad_norm": 2.705641746520996, + "learning_rate": 2.3277502854967644e-06, + "loss": 1.3763, + "step": 1712 + }, + { + "epoch": 1.7596302003081665, + "grad_norm": 2.77583646774292, + "learning_rate": 2.325846973734298e-06, + "loss": 1.4423, + "step": 1713 + }, + { + "epoch": 1.7606574216743707, + "grad_norm": 2.662750482559204, + "learning_rate": 2.323943661971831e-06, + "loss": 1.3864, + "step": 1714 + }, + { + "epoch": 1.7616846430405753, + "grad_norm": 2.769313335418701, + "learning_rate": 2.3220403502093645e-06, + "loss": 1.4272, + "step": 1715 + }, + { + "epoch": 1.7627118644067796, + "grad_norm": 2.7852976322174072, + "learning_rate": 2.320137038446898e-06, + "loss": 1.4398, + "step": 1716 + }, + { + "epoch": 1.763739085772984, + "grad_norm": 2.7452852725982666, + "learning_rate": 2.318233726684431e-06, + "loss": 1.4504, + "step": 1717 + }, + { + "epoch": 1.7647663071391886, + "grad_norm": 2.762258291244507, + "learning_rate": 2.3163304149219646e-06, + "loss": 1.4325, + "step": 1718 + }, + { + "epoch": 1.765793528505393, + "grad_norm": 2.7612709999084473, + "learning_rate": 2.314427103159498e-06, + "loss": 1.4027, + "step": 1719 + }, + { + "epoch": 1.7668207498715973, + "grad_norm": 2.7290871143341064, + "learning_rate": 2.312523791397031e-06, + "loss": 1.3919, + "step": 1720 + }, + { + "epoch": 1.7668207498715973, + "eval_loss": 1.8454608917236328, + "eval_runtime": 37.0509, + "eval_samples_per_second": 26.99, + "eval_steps_per_second": 1.134, + "step": 1720 + }, + { + "epoch": 1.767847971237802, + "grad_norm": 2.5960779190063477, + "learning_rate": 2.3106204796345643e-06, + "loss": 1.3632, + "step": 1721 + }, + { + "epoch": 1.768875192604006, + "grad_norm": 2.7832717895507812, + "learning_rate": 2.3087171678720976e-06, + "loss": 1.3899, + "step": 1722 + }, + { + "epoch": 1.7699024139702106, + "grad_norm": 2.8227648735046387, + "learning_rate": 2.306813856109631e-06, + "loss": 1.4517, + "step": 1723 + }, + { + "epoch": 1.770929635336415, + "grad_norm": 2.880059242248535, + "learning_rate": 2.3049105443471644e-06, + "loss": 1.4104, + "step": 1724 + }, + { + "epoch": 1.7719568567026194, + "grad_norm": 2.6653501987457275, + "learning_rate": 2.3030072325846973e-06, + "loss": 1.3908, + "step": 1725 + }, + { + "epoch": 1.772984078068824, + "grad_norm": 2.844914197921753, + "learning_rate": 2.3011039208222307e-06, + "loss": 1.3781, + "step": 1726 + }, + { + "epoch": 1.774011299435028, + "grad_norm": 2.8133163452148438, + "learning_rate": 2.299200609059764e-06, + "loss": 1.3775, + "step": 1727 + }, + { + "epoch": 1.7750385208012327, + "grad_norm": 2.7466955184936523, + "learning_rate": 2.297297297297298e-06, + "loss": 1.3926, + "step": 1728 + }, + { + "epoch": 1.776065742167437, + "grad_norm": 2.6311862468719482, + "learning_rate": 2.2953939855348308e-06, + "loss": 1.3486, + "step": 1729 + }, + { + "epoch": 1.7770929635336414, + "grad_norm": 2.7905211448669434, + "learning_rate": 2.293490673772364e-06, + "loss": 1.4369, + "step": 1730 + }, + { + "epoch": 1.7770929635336414, + "eval_loss": 1.8418751955032349, + "eval_runtime": 37.0448, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1730 + }, + { + "epoch": 1.778120184899846, + "grad_norm": 2.862295627593994, + "learning_rate": 2.2915873620098975e-06, + "loss": 1.4209, + "step": 1731 + }, + { + "epoch": 1.7791474062660502, + "grad_norm": 2.765862464904785, + "learning_rate": 2.289684050247431e-06, + "loss": 1.4135, + "step": 1732 + }, + { + "epoch": 1.7801746276322548, + "grad_norm": 2.7296290397644043, + "learning_rate": 2.287780738484964e-06, + "loss": 1.3814, + "step": 1733 + }, + { + "epoch": 1.7812018489984591, + "grad_norm": 2.7937533855438232, + "learning_rate": 2.285877426722497e-06, + "loss": 1.4203, + "step": 1734 + }, + { + "epoch": 1.7822290703646635, + "grad_norm": 2.7680230140686035, + "learning_rate": 2.2839741149600305e-06, + "loss": 1.4354, + "step": 1735 + }, + { + "epoch": 1.783256291730868, + "grad_norm": 2.8785109519958496, + "learning_rate": 2.282070803197564e-06, + "loss": 1.4338, + "step": 1736 + }, + { + "epoch": 1.7842835130970724, + "grad_norm": 2.747436046600342, + "learning_rate": 2.2801674914350972e-06, + "loss": 1.4087, + "step": 1737 + }, + { + "epoch": 1.7853107344632768, + "grad_norm": 2.6810362339019775, + "learning_rate": 2.2782641796726306e-06, + "loss": 1.3993, + "step": 1738 + }, + { + "epoch": 1.7863379558294814, + "grad_norm": 2.7365734577178955, + "learning_rate": 2.276360867910164e-06, + "loss": 1.4235, + "step": 1739 + }, + { + "epoch": 1.7873651771956856, + "grad_norm": 2.6946303844451904, + "learning_rate": 2.2744575561476973e-06, + "loss": 1.3626, + "step": 1740 + }, + { + "epoch": 1.7873651771956856, + "eval_loss": 1.8456190824508667, + "eval_runtime": 37.069, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1740 + }, + { + "epoch": 1.7883923985618901, + "grad_norm": 2.756920337677002, + "learning_rate": 2.2725542443852307e-06, + "loss": 1.396, + "step": 1741 + }, + { + "epoch": 1.7894196199280945, + "grad_norm": 2.765428066253662, + "learning_rate": 2.2706509326227636e-06, + "loss": 1.4374, + "step": 1742 + }, + { + "epoch": 1.7904468412942989, + "grad_norm": 2.8224971294403076, + "learning_rate": 2.268747620860297e-06, + "loss": 1.4113, + "step": 1743 + }, + { + "epoch": 1.7914740626605035, + "grad_norm": 2.90537428855896, + "learning_rate": 2.2668443090978303e-06, + "loss": 1.4124, + "step": 1744 + }, + { + "epoch": 1.7925012840267076, + "grad_norm": 2.712862491607666, + "learning_rate": 2.2649409973353637e-06, + "loss": 1.4256, + "step": 1745 + }, + { + "epoch": 1.7935285053929122, + "grad_norm": 2.688558340072632, + "learning_rate": 2.263037685572897e-06, + "loss": 1.3842, + "step": 1746 + }, + { + "epoch": 1.7945557267591166, + "grad_norm": 2.6504769325256348, + "learning_rate": 2.26113437381043e-06, + "loss": 1.3703, + "step": 1747 + }, + { + "epoch": 1.795582948125321, + "grad_norm": 3.0093724727630615, + "learning_rate": 2.259231062047964e-06, + "loss": 1.4106, + "step": 1748 + }, + { + "epoch": 1.7966101694915255, + "grad_norm": 2.727687120437622, + "learning_rate": 2.257327750285497e-06, + "loss": 1.4286, + "step": 1749 + }, + { + "epoch": 1.7976373908577299, + "grad_norm": 2.679896116256714, + "learning_rate": 2.2554244385230305e-06, + "loss": 1.3937, + "step": 1750 + }, + { + "epoch": 1.7976373908577299, + "eval_loss": 1.8520207405090332, + "eval_runtime": 37.062, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1750 + }, + { + "epoch": 1.7986646122239343, + "grad_norm": 2.8119945526123047, + "learning_rate": 2.2535211267605635e-06, + "loss": 1.3935, + "step": 1751 + }, + { + "epoch": 1.7996918335901386, + "grad_norm": 2.8470423221588135, + "learning_rate": 2.251617814998097e-06, + "loss": 1.4138, + "step": 1752 + }, + { + "epoch": 1.800719054956343, + "grad_norm": 2.6278114318847656, + "learning_rate": 2.24971450323563e-06, + "loss": 1.328, + "step": 1753 + }, + { + "epoch": 1.8017462763225476, + "grad_norm": 2.861860990524292, + "learning_rate": 2.2478111914731636e-06, + "loss": 1.416, + "step": 1754 + }, + { + "epoch": 1.802773497688752, + "grad_norm": 2.559736728668213, + "learning_rate": 2.245907879710697e-06, + "loss": 1.2891, + "step": 1755 + }, + { + "epoch": 1.8038007190549563, + "grad_norm": 2.8461058139801025, + "learning_rate": 2.24400456794823e-06, + "loss": 1.4194, + "step": 1756 + }, + { + "epoch": 1.804827940421161, + "grad_norm": 2.7069878578186035, + "learning_rate": 2.2421012561857632e-06, + "loss": 1.382, + "step": 1757 + }, + { + "epoch": 1.805855161787365, + "grad_norm": 2.727435827255249, + "learning_rate": 2.240197944423297e-06, + "loss": 1.3714, + "step": 1758 + }, + { + "epoch": 1.8068823831535696, + "grad_norm": 2.7298061847686768, + "learning_rate": 2.23829463266083e-06, + "loss": 1.3834, + "step": 1759 + }, + { + "epoch": 1.807909604519774, + "grad_norm": 2.7419843673706055, + "learning_rate": 2.2363913208983633e-06, + "loss": 1.3687, + "step": 1760 + }, + { + "epoch": 1.807909604519774, + "eval_loss": 1.8575356006622314, + "eval_runtime": 37.0563, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1760 + }, + { + "epoch": 1.8089368258859784, + "grad_norm": 2.8058817386627197, + "learning_rate": 2.2344880091358967e-06, + "loss": 1.415, + "step": 1761 + }, + { + "epoch": 1.809964047252183, + "grad_norm": 2.7287003993988037, + "learning_rate": 2.23258469737343e-06, + "loss": 1.3778, + "step": 1762 + }, + { + "epoch": 1.810991268618387, + "grad_norm": 2.7943124771118164, + "learning_rate": 2.2306813856109634e-06, + "loss": 1.433, + "step": 1763 + }, + { + "epoch": 1.8120184899845917, + "grad_norm": 2.7254626750946045, + "learning_rate": 2.2287780738484963e-06, + "loss": 1.3822, + "step": 1764 + }, + { + "epoch": 1.813045711350796, + "grad_norm": 2.799793004989624, + "learning_rate": 2.2268747620860297e-06, + "loss": 1.4595, + "step": 1765 + }, + { + "epoch": 1.8140729327170004, + "grad_norm": 2.79420804977417, + "learning_rate": 2.224971450323563e-06, + "loss": 1.3949, + "step": 1766 + }, + { + "epoch": 1.815100154083205, + "grad_norm": 2.861016035079956, + "learning_rate": 2.2230681385610964e-06, + "loss": 1.3301, + "step": 1767 + }, + { + "epoch": 1.8161273754494094, + "grad_norm": 2.9270451068878174, + "learning_rate": 2.2211648267986298e-06, + "loss": 1.3895, + "step": 1768 + }, + { + "epoch": 1.8171545968156138, + "grad_norm": 2.709031105041504, + "learning_rate": 2.219261515036163e-06, + "loss": 1.3757, + "step": 1769 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 2.7763030529022217, + "learning_rate": 2.2173582032736965e-06, + "loss": 1.4135, + "step": 1770 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.8521159887313843, + "eval_runtime": 37.1265, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 1.8192090395480225, + "grad_norm": 2.97048282623291, + "learning_rate": 2.21545489151123e-06, + "loss": 1.4345, + "step": 1771 + }, + { + "epoch": 1.820236260914227, + "grad_norm": 2.8238744735717773, + "learning_rate": 2.2135515797487632e-06, + "loss": 1.3948, + "step": 1772 + }, + { + "epoch": 1.8212634822804314, + "grad_norm": 2.6430611610412598, + "learning_rate": 2.211648267986296e-06, + "loss": 1.3878, + "step": 1773 + }, + { + "epoch": 1.8222907036466358, + "grad_norm": 2.602883815765381, + "learning_rate": 2.2097449562238295e-06, + "loss": 1.3978, + "step": 1774 + }, + { + "epoch": 1.8233179250128404, + "grad_norm": 2.707813262939453, + "learning_rate": 2.207841644461363e-06, + "loss": 1.3937, + "step": 1775 + }, + { + "epoch": 1.8243451463790445, + "grad_norm": 2.662684679031372, + "learning_rate": 2.2059383326988963e-06, + "loss": 1.4381, + "step": 1776 + }, + { + "epoch": 1.8253723677452491, + "grad_norm": 2.9407193660736084, + "learning_rate": 2.2040350209364296e-06, + "loss": 1.4059, + "step": 1777 + }, + { + "epoch": 1.8263995891114535, + "grad_norm": 2.6678171157836914, + "learning_rate": 2.2021317091739626e-06, + "loss": 1.3718, + "step": 1778 + }, + { + "epoch": 1.8274268104776579, + "grad_norm": 2.7536516189575195, + "learning_rate": 2.2002283974114963e-06, + "loss": 1.3998, + "step": 1779 + }, + { + "epoch": 1.8284540318438625, + "grad_norm": 2.7340128421783447, + "learning_rate": 2.1983250856490297e-06, + "loss": 1.3299, + "step": 1780 + }, + { + "epoch": 1.8284540318438625, + "eval_loss": 1.854264736175537, + "eval_runtime": 37.0874, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1780 + }, + { + "epoch": 1.8294812532100666, + "grad_norm": 2.844285726547241, + "learning_rate": 2.1964217738865626e-06, + "loss": 1.3746, + "step": 1781 + }, + { + "epoch": 1.8305084745762712, + "grad_norm": 2.9005022048950195, + "learning_rate": 2.194518462124096e-06, + "loss": 1.4087, + "step": 1782 + }, + { + "epoch": 1.8315356959424756, + "grad_norm": 2.8825247287750244, + "learning_rate": 2.1926151503616294e-06, + "loss": 1.4458, + "step": 1783 + }, + { + "epoch": 1.83256291730868, + "grad_norm": 2.712778091430664, + "learning_rate": 2.1907118385991627e-06, + "loss": 1.414, + "step": 1784 + }, + { + "epoch": 1.8335901386748845, + "grad_norm": 2.7036001682281494, + "learning_rate": 2.188808526836696e-06, + "loss": 1.3632, + "step": 1785 + }, + { + "epoch": 1.8346173600410889, + "grad_norm": 2.704606771469116, + "learning_rate": 2.1869052150742295e-06, + "loss": 1.3992, + "step": 1786 + }, + { + "epoch": 1.8356445814072933, + "grad_norm": 2.7564282417297363, + "learning_rate": 2.1850019033117624e-06, + "loss": 1.3331, + "step": 1787 + }, + { + "epoch": 1.8366718027734978, + "grad_norm": 2.898256540298462, + "learning_rate": 2.1830985915492958e-06, + "loss": 1.3881, + "step": 1788 + }, + { + "epoch": 1.837699024139702, + "grad_norm": 2.7897069454193115, + "learning_rate": 2.1811952797868295e-06, + "loss": 1.4081, + "step": 1789 + }, + { + "epoch": 1.8387262455059066, + "grad_norm": 2.7293193340301514, + "learning_rate": 2.1792919680243625e-06, + "loss": 1.3705, + "step": 1790 + }, + { + "epoch": 1.8387262455059066, + "eval_loss": 1.8483723402023315, + "eval_runtime": 37.4058, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 1790 + }, + { + "epoch": 1.839753466872111, + "grad_norm": 2.9257752895355225, + "learning_rate": 2.177388656261896e-06, + "loss": 1.4071, + "step": 1791 + }, + { + "epoch": 1.8407806882383153, + "grad_norm": 3.016033172607422, + "learning_rate": 2.1754853444994292e-06, + "loss": 1.4774, + "step": 1792 + }, + { + "epoch": 1.84180790960452, + "grad_norm": 2.74198842048645, + "learning_rate": 2.1735820327369626e-06, + "loss": 1.301, + "step": 1793 + }, + { + "epoch": 1.842835130970724, + "grad_norm": 2.809215545654297, + "learning_rate": 2.171678720974496e-06, + "loss": 1.4192, + "step": 1794 + }, + { + "epoch": 1.8438623523369286, + "grad_norm": 2.673997402191162, + "learning_rate": 2.169775409212029e-06, + "loss": 1.4182, + "step": 1795 + }, + { + "epoch": 1.844889573703133, + "grad_norm": 2.9519944190979004, + "learning_rate": 2.1678720974495622e-06, + "loss": 1.4253, + "step": 1796 + }, + { + "epoch": 1.8459167950693374, + "grad_norm": 2.7683424949645996, + "learning_rate": 2.1659687856870956e-06, + "loss": 1.4162, + "step": 1797 + }, + { + "epoch": 1.846944016435542, + "grad_norm": 2.861701726913452, + "learning_rate": 2.164065473924629e-06, + "loss": 1.4167, + "step": 1798 + }, + { + "epoch": 1.8479712378017463, + "grad_norm": 2.62581467628479, + "learning_rate": 2.1621621621621623e-06, + "loss": 1.3568, + "step": 1799 + }, + { + "epoch": 1.8489984591679507, + "grad_norm": 2.782197952270508, + "learning_rate": 2.1602588503996957e-06, + "loss": 1.4184, + "step": 1800 + }, + { + "epoch": 1.8489984591679507, + "eval_loss": 1.8527827262878418, + "eval_runtime": 37.4743, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 1.850025680534155, + "grad_norm": 2.7289304733276367, + "learning_rate": 2.158355538637229e-06, + "loss": 1.3979, + "step": 1801 + }, + { + "epoch": 1.8510529019003594, + "grad_norm": 2.7153542041778564, + "learning_rate": 2.1564522268747624e-06, + "loss": 1.3863, + "step": 1802 + }, + { + "epoch": 1.852080123266564, + "grad_norm": 2.884798765182495, + "learning_rate": 2.1545489151122958e-06, + "loss": 1.4656, + "step": 1803 + }, + { + "epoch": 1.8531073446327684, + "grad_norm": 2.6483845710754395, + "learning_rate": 2.1526456033498287e-06, + "loss": 1.4118, + "step": 1804 + }, + { + "epoch": 1.8541345659989728, + "grad_norm": 2.7074198722839355, + "learning_rate": 2.150742291587362e-06, + "loss": 1.41, + "step": 1805 + }, + { + "epoch": 1.8551617873651773, + "grad_norm": 2.734137535095215, + "learning_rate": 2.1488389798248954e-06, + "loss": 1.3968, + "step": 1806 + }, + { + "epoch": 1.8561890087313815, + "grad_norm": 2.7855992317199707, + "learning_rate": 2.146935668062429e-06, + "loss": 1.427, + "step": 1807 + }, + { + "epoch": 1.857216230097586, + "grad_norm": 2.7769339084625244, + "learning_rate": 2.145032356299962e-06, + "loss": 1.3476, + "step": 1808 + }, + { + "epoch": 1.8582434514637904, + "grad_norm": 2.6978964805603027, + "learning_rate": 2.1431290445374955e-06, + "loss": 1.4047, + "step": 1809 + }, + { + "epoch": 1.8592706728299948, + "grad_norm": 2.806438446044922, + "learning_rate": 2.141225732775029e-06, + "loss": 1.3693, + "step": 1810 + }, + { + "epoch": 1.8592706728299948, + "eval_loss": 1.8606462478637695, + "eval_runtime": 36.9431, + "eval_samples_per_second": 27.069, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 1.8602978941961994, + "grad_norm": 2.886394500732422, + "learning_rate": 2.1393224210125623e-06, + "loss": 1.4022, + "step": 1811 + }, + { + "epoch": 1.8613251155624035, + "grad_norm": 2.7849435806274414, + "learning_rate": 2.137419109250095e-06, + "loss": 1.3907, + "step": 1812 + }, + { + "epoch": 1.8623523369286081, + "grad_norm": 2.85810923576355, + "learning_rate": 2.1355157974876286e-06, + "loss": 1.4451, + "step": 1813 + }, + { + "epoch": 1.8633795582948125, + "grad_norm": 2.8717806339263916, + "learning_rate": 2.133612485725162e-06, + "loss": 1.4303, + "step": 1814 + }, + { + "epoch": 1.8644067796610169, + "grad_norm": 2.6955783367156982, + "learning_rate": 2.1317091739626953e-06, + "loss": 1.3454, + "step": 1815 + }, + { + "epoch": 1.8654340010272215, + "grad_norm": 2.835347890853882, + "learning_rate": 2.1298058622002286e-06, + "loss": 1.3671, + "step": 1816 + }, + { + "epoch": 1.8664612223934258, + "grad_norm": 2.804835796356201, + "learning_rate": 2.1279025504377616e-06, + "loss": 1.388, + "step": 1817 + }, + { + "epoch": 1.8674884437596302, + "grad_norm": 2.69130539894104, + "learning_rate": 2.125999238675295e-06, + "loss": 1.3815, + "step": 1818 + }, + { + "epoch": 1.8685156651258348, + "grad_norm": 2.856457233428955, + "learning_rate": 2.1240959269128287e-06, + "loss": 1.4323, + "step": 1819 + }, + { + "epoch": 1.869542886492039, + "grad_norm": 2.8100039958953857, + "learning_rate": 2.122192615150362e-06, + "loss": 1.3942, + "step": 1820 + }, + { + "epoch": 1.869542886492039, + "eval_loss": 1.860397219657898, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 1820 + }, + { + "epoch": 1.8705701078582435, + "grad_norm": 2.6977529525756836, + "learning_rate": 2.120289303387895e-06, + "loss": 1.3371, + "step": 1821 + }, + { + "epoch": 1.8715973292244479, + "grad_norm": 2.8403844833374023, + "learning_rate": 2.1183859916254284e-06, + "loss": 1.3776, + "step": 1822 + }, + { + "epoch": 1.8726245505906522, + "grad_norm": 2.829756498336792, + "learning_rate": 2.1164826798629618e-06, + "loss": 1.4229, + "step": 1823 + }, + { + "epoch": 1.8736517719568568, + "grad_norm": 2.6810450553894043, + "learning_rate": 2.114579368100495e-06, + "loss": 1.3757, + "step": 1824 + }, + { + "epoch": 1.874678993323061, + "grad_norm": 2.988982677459717, + "learning_rate": 2.1126760563380285e-06, + "loss": 1.3671, + "step": 1825 + }, + { + "epoch": 1.8757062146892656, + "grad_norm": 2.661953926086426, + "learning_rate": 2.1107727445755614e-06, + "loss": 1.3382, + "step": 1826 + }, + { + "epoch": 1.87673343605547, + "grad_norm": 2.7993199825286865, + "learning_rate": 2.1088694328130948e-06, + "loss": 1.4621, + "step": 1827 + }, + { + "epoch": 1.8777606574216743, + "grad_norm": 2.681609869003296, + "learning_rate": 2.106966121050628e-06, + "loss": 1.3596, + "step": 1828 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.988734722137451, + "learning_rate": 2.1050628092881615e-06, + "loss": 1.4269, + "step": 1829 + }, + { + "epoch": 1.879815100154083, + "grad_norm": 2.867772102355957, + "learning_rate": 2.103159497525695e-06, + "loss": 1.4618, + "step": 1830 + }, + { + "epoch": 1.879815100154083, + "eval_loss": 1.8595913648605347, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1830 + }, + { + "epoch": 1.8808423215202876, + "grad_norm": 2.7212493419647217, + "learning_rate": 2.1012561857632282e-06, + "loss": 1.3738, + "step": 1831 + }, + { + "epoch": 1.881869542886492, + "grad_norm": 2.7997097969055176, + "learning_rate": 2.0993528740007616e-06, + "loss": 1.4268, + "step": 1832 + }, + { + "epoch": 1.8828967642526964, + "grad_norm": 2.8695220947265625, + "learning_rate": 2.097449562238295e-06, + "loss": 1.4397, + "step": 1833 + }, + { + "epoch": 1.883923985618901, + "grad_norm": 2.681015729904175, + "learning_rate": 2.095546250475828e-06, + "loss": 1.3873, + "step": 1834 + }, + { + "epoch": 1.8849512069851053, + "grad_norm": 2.787360429763794, + "learning_rate": 2.0936429387133613e-06, + "loss": 1.3944, + "step": 1835 + }, + { + "epoch": 1.8859784283513097, + "grad_norm": 2.830164670944214, + "learning_rate": 2.0917396269508946e-06, + "loss": 1.4485, + "step": 1836 + }, + { + "epoch": 1.8870056497175143, + "grad_norm": 2.9803659915924072, + "learning_rate": 2.089836315188428e-06, + "loss": 1.4259, + "step": 1837 + }, + { + "epoch": 1.8880328710837184, + "grad_norm": 2.846390962600708, + "learning_rate": 2.0879330034259614e-06, + "loss": 1.4371, + "step": 1838 + }, + { + "epoch": 1.889060092449923, + "grad_norm": 2.636085033416748, + "learning_rate": 2.0860296916634947e-06, + "loss": 1.3679, + "step": 1839 + }, + { + "epoch": 1.8900873138161274, + "grad_norm": 2.806547164916992, + "learning_rate": 2.084126379901028e-06, + "loss": 1.4662, + "step": 1840 + }, + { + "epoch": 1.8900873138161274, + "eval_loss": 1.8502073287963867, + "eval_runtime": 37.1321, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 1.8911145351823317, + "grad_norm": 2.8658089637756348, + "learning_rate": 2.0822230681385614e-06, + "loss": 1.3593, + "step": 1841 + }, + { + "epoch": 1.8921417565485363, + "grad_norm": 2.87060284614563, + "learning_rate": 2.080319756376095e-06, + "loss": 1.4189, + "step": 1842 + }, + { + "epoch": 1.8931689779147405, + "grad_norm": 2.7965638637542725, + "learning_rate": 2.0784164446136277e-06, + "loss": 1.4208, + "step": 1843 + }, + { + "epoch": 1.894196199280945, + "grad_norm": 2.5999045372009277, + "learning_rate": 2.076513132851161e-06, + "loss": 1.375, + "step": 1844 + }, + { + "epoch": 1.8952234206471494, + "grad_norm": 2.833935260772705, + "learning_rate": 2.0746098210886945e-06, + "loss": 1.4137, + "step": 1845 + }, + { + "epoch": 1.8962506420133538, + "grad_norm": 2.9809396266937256, + "learning_rate": 2.072706509326228e-06, + "loss": 1.3874, + "step": 1846 + }, + { + "epoch": 1.8972778633795584, + "grad_norm": 2.9341282844543457, + "learning_rate": 2.070803197563761e-06, + "loss": 1.3953, + "step": 1847 + }, + { + "epoch": 1.8983050847457628, + "grad_norm": 2.989488124847412, + "learning_rate": 2.068899885801294e-06, + "loss": 1.3913, + "step": 1848 + }, + { + "epoch": 1.8993323061119671, + "grad_norm": 2.8224217891693115, + "learning_rate": 2.066996574038828e-06, + "loss": 1.4032, + "step": 1849 + }, + { + "epoch": 1.9003595274781715, + "grad_norm": 2.8850255012512207, + "learning_rate": 2.0650932622763613e-06, + "loss": 1.4402, + "step": 1850 + }, + { + "epoch": 1.9003595274781715, + "eval_loss": 1.8440051078796387, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1850 + }, + { + "epoch": 1.9013867488443759, + "grad_norm": 2.927288293838501, + "learning_rate": 2.0631899505138942e-06, + "loss": 1.3215, + "step": 1851 + }, + { + "epoch": 1.9024139702105805, + "grad_norm": 2.7680368423461914, + "learning_rate": 2.0612866387514276e-06, + "loss": 1.3636, + "step": 1852 + }, + { + "epoch": 1.9034411915767848, + "grad_norm": 2.8515689373016357, + "learning_rate": 2.059383326988961e-06, + "loss": 1.4049, + "step": 1853 + }, + { + "epoch": 1.9044684129429892, + "grad_norm": 2.7973034381866455, + "learning_rate": 2.0574800152264943e-06, + "loss": 1.3876, + "step": 1854 + }, + { + "epoch": 1.9054956343091938, + "grad_norm": 2.7860023975372314, + "learning_rate": 2.0555767034640277e-06, + "loss": 1.3687, + "step": 1855 + }, + { + "epoch": 1.906522855675398, + "grad_norm": 2.807511568069458, + "learning_rate": 2.053673391701561e-06, + "loss": 1.3382, + "step": 1856 + }, + { + "epoch": 1.9075500770416025, + "grad_norm": 2.9342591762542725, + "learning_rate": 2.051770079939094e-06, + "loss": 1.4025, + "step": 1857 + }, + { + "epoch": 1.9085772984078069, + "grad_norm": 2.85943865776062, + "learning_rate": 2.0498667681766273e-06, + "loss": 1.3626, + "step": 1858 + }, + { + "epoch": 1.9096045197740112, + "grad_norm": 2.8514609336853027, + "learning_rate": 2.047963456414161e-06, + "loss": 1.3675, + "step": 1859 + }, + { + "epoch": 1.9106317411402158, + "grad_norm": 2.8177332878112793, + "learning_rate": 2.046060144651694e-06, + "loss": 1.3968, + "step": 1860 + }, + { + "epoch": 1.9106317411402158, + "eval_loss": 1.8451731204986572, + "eval_runtime": 37.1047, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1860 + }, + { + "epoch": 1.91165896250642, + "grad_norm": 2.74876070022583, + "learning_rate": 2.0441568328892274e-06, + "loss": 1.4022, + "step": 1861 + }, + { + "epoch": 1.9126861838726246, + "grad_norm": 2.6633286476135254, + "learning_rate": 2.0422535211267608e-06, + "loss": 1.4004, + "step": 1862 + }, + { + "epoch": 1.913713405238829, + "grad_norm": 2.6457369327545166, + "learning_rate": 2.040350209364294e-06, + "loss": 1.3767, + "step": 1863 + }, + { + "epoch": 1.9147406266050333, + "grad_norm": 2.7011449337005615, + "learning_rate": 2.0384468976018275e-06, + "loss": 1.373, + "step": 1864 + }, + { + "epoch": 1.915767847971238, + "grad_norm": 2.676337480545044, + "learning_rate": 2.0365435858393604e-06, + "loss": 1.3713, + "step": 1865 + }, + { + "epoch": 1.9167950693374423, + "grad_norm": 2.7401342391967773, + "learning_rate": 2.034640274076894e-06, + "loss": 1.3644, + "step": 1866 + }, + { + "epoch": 1.9178222907036466, + "grad_norm": 2.7799527645111084, + "learning_rate": 2.032736962314427e-06, + "loss": 1.433, + "step": 1867 + }, + { + "epoch": 1.9188495120698512, + "grad_norm": 2.7372307777404785, + "learning_rate": 2.0308336505519605e-06, + "loss": 1.3781, + "step": 1868 + }, + { + "epoch": 1.9198767334360554, + "grad_norm": 2.986297845840454, + "learning_rate": 2.028930338789494e-06, + "loss": 1.4406, + "step": 1869 + }, + { + "epoch": 1.92090395480226, + "grad_norm": 2.815706491470337, + "learning_rate": 2.0270270270270273e-06, + "loss": 1.3555, + "step": 1870 + }, + { + "epoch": 1.92090395480226, + "eval_loss": 1.8498139381408691, + "eval_runtime": 37.1146, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1870 + }, + { + "epoch": 1.9219311761684643, + "grad_norm": 2.734189987182617, + "learning_rate": 2.0251237152645606e-06, + "loss": 1.373, + "step": 1871 + }, + { + "epoch": 1.9229583975346687, + "grad_norm": 2.7900359630584717, + "learning_rate": 2.023220403502094e-06, + "loss": 1.3623, + "step": 1872 + }, + { + "epoch": 1.9239856189008733, + "grad_norm": 2.8590710163116455, + "learning_rate": 2.0213170917396273e-06, + "loss": 1.3424, + "step": 1873 + }, + { + "epoch": 1.9250128402670774, + "grad_norm": 2.880814790725708, + "learning_rate": 2.0194137799771603e-06, + "loss": 1.3683, + "step": 1874 + }, + { + "epoch": 1.926040061633282, + "grad_norm": 2.7775630950927734, + "learning_rate": 2.0175104682146937e-06, + "loss": 1.3879, + "step": 1875 + }, + { + "epoch": 1.9270672829994864, + "grad_norm": 2.713557481765747, + "learning_rate": 2.015607156452227e-06, + "loss": 1.3216, + "step": 1876 + }, + { + "epoch": 1.9280945043656907, + "grad_norm": 2.8787734508514404, + "learning_rate": 2.0137038446897604e-06, + "loss": 1.3659, + "step": 1877 + }, + { + "epoch": 1.9291217257318953, + "grad_norm": 2.8224005699157715, + "learning_rate": 2.0118005329272937e-06, + "loss": 1.3572, + "step": 1878 + }, + { + "epoch": 1.9301489470980995, + "grad_norm": 2.865316390991211, + "learning_rate": 2.0098972211648267e-06, + "loss": 1.3817, + "step": 1879 + }, + { + "epoch": 1.931176168464304, + "grad_norm": 2.7488853931427, + "learning_rate": 2.0079939094023605e-06, + "loss": 1.3766, + "step": 1880 + }, + { + "epoch": 1.931176168464304, + "eval_loss": 1.8444081544876099, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1880 + }, + { + "epoch": 1.9322033898305084, + "grad_norm": 2.902949333190918, + "learning_rate": 2.006090597639894e-06, + "loss": 1.4183, + "step": 1881 + }, + { + "epoch": 1.9332306111967128, + "grad_norm": 2.8154122829437256, + "learning_rate": 2.0041872858774268e-06, + "loss": 1.4525, + "step": 1882 + }, + { + "epoch": 1.9342578325629174, + "grad_norm": 2.8343429565429688, + "learning_rate": 2.00228397411496e-06, + "loss": 1.4099, + "step": 1883 + }, + { + "epoch": 1.9352850539291218, + "grad_norm": 2.863067388534546, + "learning_rate": 2.0003806623524935e-06, + "loss": 1.3973, + "step": 1884 + }, + { + "epoch": 1.9363122752953261, + "grad_norm": 2.784085988998413, + "learning_rate": 1.998477350590027e-06, + "loss": 1.3117, + "step": 1885 + }, + { + "epoch": 1.9373394966615307, + "grad_norm": 2.7667958736419678, + "learning_rate": 1.9965740388275602e-06, + "loss": 1.4056, + "step": 1886 + }, + { + "epoch": 1.9383667180277349, + "grad_norm": 2.747432231903076, + "learning_rate": 1.994670727065093e-06, + "loss": 1.3927, + "step": 1887 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9162001609802246, + "learning_rate": 1.9927674153026265e-06, + "loss": 1.3453, + "step": 1888 + }, + { + "epoch": 1.9404211607601438, + "grad_norm": 3.0909414291381836, + "learning_rate": 1.99086410354016e-06, + "loss": 1.4148, + "step": 1889 + }, + { + "epoch": 1.9414483821263482, + "grad_norm": 2.774672746658325, + "learning_rate": 1.9889607917776937e-06, + "loss": 1.3762, + "step": 1890 + }, + { + "epoch": 1.9414483821263482, + "eval_loss": 1.848209023475647, + "eval_runtime": 37.2784, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 1.9424756034925528, + "grad_norm": 2.7985880374908447, + "learning_rate": 1.9870574800152266e-06, + "loss": 1.3908, + "step": 1891 + }, + { + "epoch": 1.943502824858757, + "grad_norm": 2.749664783477783, + "learning_rate": 1.98515416825276e-06, + "loss": 1.4088, + "step": 1892 + }, + { + "epoch": 1.9445300462249615, + "grad_norm": 2.770726442337036, + "learning_rate": 1.9832508564902933e-06, + "loss": 1.4054, + "step": 1893 + }, + { + "epoch": 1.9455572675911659, + "grad_norm": 2.634516477584839, + "learning_rate": 1.9813475447278267e-06, + "loss": 1.3247, + "step": 1894 + }, + { + "epoch": 1.9465844889573702, + "grad_norm": 2.886481285095215, + "learning_rate": 1.97944423296536e-06, + "loss": 1.4025, + "step": 1895 + }, + { + "epoch": 1.9476117103235748, + "grad_norm": 2.8149733543395996, + "learning_rate": 1.977540921202893e-06, + "loss": 1.3745, + "step": 1896 + }, + { + "epoch": 1.9486389316897792, + "grad_norm": 2.643970251083374, + "learning_rate": 1.9756376094404264e-06, + "loss": 1.3581, + "step": 1897 + }, + { + "epoch": 1.9496661530559836, + "grad_norm": 2.7485244274139404, + "learning_rate": 1.9737342976779597e-06, + "loss": 1.3426, + "step": 1898 + }, + { + "epoch": 1.950693374422188, + "grad_norm": 2.930513858795166, + "learning_rate": 1.971830985915493e-06, + "loss": 1.4194, + "step": 1899 + }, + { + "epoch": 1.9517205957883923, + "grad_norm": 2.7637267112731934, + "learning_rate": 1.9699276741530264e-06, + "loss": 1.3908, + "step": 1900 + }, + { + "epoch": 1.9517205957883923, + "eval_loss": 1.8510984182357788, + "eval_runtime": 37.4089, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1900 + }, + { + "epoch": 1.9527478171545969, + "grad_norm": 2.8946938514709473, + "learning_rate": 1.96802436239056e-06, + "loss": 1.3892, + "step": 1901 + }, + { + "epoch": 1.9537750385208013, + "grad_norm": 2.7064263820648193, + "learning_rate": 1.966121050628093e-06, + "loss": 1.3628, + "step": 1902 + }, + { + "epoch": 1.9548022598870056, + "grad_norm": 2.8588147163391113, + "learning_rate": 1.9642177388656265e-06, + "loss": 1.4097, + "step": 1903 + }, + { + "epoch": 1.9558294812532102, + "grad_norm": 2.7746825218200684, + "learning_rate": 1.9623144271031595e-06, + "loss": 1.4255, + "step": 1904 + }, + { + "epoch": 1.9568567026194144, + "grad_norm": 2.872587203979492, + "learning_rate": 1.960411115340693e-06, + "loss": 1.4308, + "step": 1905 + }, + { + "epoch": 1.957883923985619, + "grad_norm": 2.714769124984741, + "learning_rate": 1.958507803578226e-06, + "loss": 1.3847, + "step": 1906 + }, + { + "epoch": 1.9589111453518233, + "grad_norm": 2.7117342948913574, + "learning_rate": 1.9566044918157596e-06, + "loss": 1.406, + "step": 1907 + }, + { + "epoch": 1.9599383667180277, + "grad_norm": 2.754995346069336, + "learning_rate": 1.954701180053293e-06, + "loss": 1.4216, + "step": 1908 + }, + { + "epoch": 1.9609655880842323, + "grad_norm": 2.6597256660461426, + "learning_rate": 1.952797868290826e-06, + "loss": 1.3748, + "step": 1909 + }, + { + "epoch": 1.9619928094504364, + "grad_norm": 2.7966320514678955, + "learning_rate": 1.9508945565283596e-06, + "loss": 1.4237, + "step": 1910 + }, + { + "epoch": 1.9619928094504364, + "eval_loss": 1.8453261852264404, + "eval_runtime": 36.9326, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1910 + }, + { + "epoch": 1.963020030816641, + "grad_norm": 2.7504498958587646, + "learning_rate": 1.948991244765893e-06, + "loss": 1.3832, + "step": 1911 + }, + { + "epoch": 1.9640472521828454, + "grad_norm": 2.7729601860046387, + "learning_rate": 1.9470879330034264e-06, + "loss": 1.4176, + "step": 1912 + }, + { + "epoch": 1.9650744735490497, + "grad_norm": 2.6513381004333496, + "learning_rate": 1.9451846212409593e-06, + "loss": 1.3524, + "step": 1913 + }, + { + "epoch": 1.9661016949152543, + "grad_norm": 2.8147811889648438, + "learning_rate": 1.9432813094784927e-06, + "loss": 1.4119, + "step": 1914 + }, + { + "epoch": 1.9671289162814587, + "grad_norm": 2.7253763675689697, + "learning_rate": 1.941377997716026e-06, + "loss": 1.3458, + "step": 1915 + }, + { + "epoch": 1.968156137647663, + "grad_norm": 2.6741955280303955, + "learning_rate": 1.9394746859535594e-06, + "loss": 1.3764, + "step": 1916 + }, + { + "epoch": 1.9691833590138677, + "grad_norm": 2.785856008529663, + "learning_rate": 1.9375713741910928e-06, + "loss": 1.3484, + "step": 1917 + }, + { + "epoch": 1.9702105803800718, + "grad_norm": 2.743168592453003, + "learning_rate": 1.9356680624286257e-06, + "loss": 1.3779, + "step": 1918 + }, + { + "epoch": 1.9712378017462764, + "grad_norm": 2.827883243560791, + "learning_rate": 1.933764750666159e-06, + "loss": 1.3246, + "step": 1919 + }, + { + "epoch": 1.9722650231124808, + "grad_norm": 2.7379672527313232, + "learning_rate": 1.931861438903693e-06, + "loss": 1.383, + "step": 1920 + }, + { + "epoch": 1.9722650231124808, + "eval_loss": 1.8433737754821777, + "eval_runtime": 37.0421, + "eval_samples_per_second": 26.996, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 1.9732922444786851, + "grad_norm": 2.7514965534210205, + "learning_rate": 1.929958127141226e-06, + "loss": 1.3513, + "step": 1921 + }, + { + "epoch": 1.9743194658448897, + "grad_norm": 2.6764447689056396, + "learning_rate": 1.928054815378759e-06, + "loss": 1.3718, + "step": 1922 + }, + { + "epoch": 1.9753466872110939, + "grad_norm": 2.8592004776000977, + "learning_rate": 1.9261515036162925e-06, + "loss": 1.4479, + "step": 1923 + }, + { + "epoch": 1.9763739085772984, + "grad_norm": 2.8035528659820557, + "learning_rate": 1.924248191853826e-06, + "loss": 1.4222, + "step": 1924 + }, + { + "epoch": 1.9774011299435028, + "grad_norm": 2.7685303688049316, + "learning_rate": 1.9223448800913592e-06, + "loss": 1.3939, + "step": 1925 + }, + { + "epoch": 1.9784283513097072, + "grad_norm": 2.6679201126098633, + "learning_rate": 1.9204415683288926e-06, + "loss": 1.385, + "step": 1926 + }, + { + "epoch": 1.9794555726759118, + "grad_norm": 2.7546098232269287, + "learning_rate": 1.9185382565664255e-06, + "loss": 1.4134, + "step": 1927 + }, + { + "epoch": 1.980482794042116, + "grad_norm": 2.794991970062256, + "learning_rate": 1.916634944803959e-06, + "loss": 1.4212, + "step": 1928 + }, + { + "epoch": 1.9815100154083205, + "grad_norm": 2.767075538635254, + "learning_rate": 1.9147316330414923e-06, + "loss": 1.3979, + "step": 1929 + }, + { + "epoch": 1.9825372367745249, + "grad_norm": 2.7712795734405518, + "learning_rate": 1.9128283212790256e-06, + "loss": 1.3827, + "step": 1930 + }, + { + "epoch": 1.9825372367745249, + "eval_loss": 1.8437753915786743, + "eval_runtime": 37.0885, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1930 + }, + { + "epoch": 1.9835644581407292, + "grad_norm": 2.791165351867676, + "learning_rate": 1.910925009516559e-06, + "loss": 1.4328, + "step": 1931 + }, + { + "epoch": 1.9845916795069338, + "grad_norm": 2.7170767784118652, + "learning_rate": 1.9090216977540924e-06, + "loss": 1.3958, + "step": 1932 + }, + { + "epoch": 1.9856189008731382, + "grad_norm": 2.8529574871063232, + "learning_rate": 1.9071183859916257e-06, + "loss": 1.4379, + "step": 1933 + }, + { + "epoch": 1.9866461222393426, + "grad_norm": 2.8424859046936035, + "learning_rate": 1.9052150742291589e-06, + "loss": 1.4066, + "step": 1934 + }, + { + "epoch": 1.9876733436055471, + "grad_norm": 2.8055100440979004, + "learning_rate": 1.9033117624666922e-06, + "loss": 1.4, + "step": 1935 + }, + { + "epoch": 1.9887005649717513, + "grad_norm": 2.73415470123291, + "learning_rate": 1.9014084507042254e-06, + "loss": 1.3652, + "step": 1936 + }, + { + "epoch": 1.9897277863379559, + "grad_norm": 2.7907590866088867, + "learning_rate": 1.8995051389417587e-06, + "loss": 1.3475, + "step": 1937 + }, + { + "epoch": 1.9907550077041603, + "grad_norm": 2.6994218826293945, + "learning_rate": 1.8976018271792921e-06, + "loss": 1.3428, + "step": 1938 + }, + { + "epoch": 1.9917822290703646, + "grad_norm": 2.8237955570220947, + "learning_rate": 1.8956985154168253e-06, + "loss": 1.3918, + "step": 1939 + }, + { + "epoch": 1.9928094504365692, + "grad_norm": 2.779270648956299, + "learning_rate": 1.8937952036543588e-06, + "loss": 1.3726, + "step": 1940 + }, + { + "epoch": 1.9928094504365692, + "eval_loss": 1.8435344696044922, + "eval_runtime": 37.0797, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1940 + }, + { + "epoch": 1.9938366718027734, + "grad_norm": 2.7808098793029785, + "learning_rate": 1.8918918918918922e-06, + "loss": 1.3368, + "step": 1941 + }, + { + "epoch": 1.994863893168978, + "grad_norm": 2.7355751991271973, + "learning_rate": 1.8899885801294253e-06, + "loss": 1.39, + "step": 1942 + }, + { + "epoch": 1.9958911145351823, + "grad_norm": 2.907468318939209, + "learning_rate": 1.8880852683669587e-06, + "loss": 1.399, + "step": 1943 + }, + { + "epoch": 1.9969183359013867, + "grad_norm": 2.8597185611724854, + "learning_rate": 1.886181956604492e-06, + "loss": 1.3731, + "step": 1944 + }, + { + "epoch": 1.9979455572675913, + "grad_norm": 2.8638176918029785, + "learning_rate": 1.8842786448420252e-06, + "loss": 1.3696, + "step": 1945 + }, + { + "epoch": 1.9989727786337956, + "grad_norm": 2.880169153213501, + "learning_rate": 1.8823753330795586e-06, + "loss": 1.3971, + "step": 1946 + }, + { + "epoch": 2.0, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.8804720213170917e-06, + "loss": 1.3026, + "step": 1947 + }, + { + "epoch": 2.0010272213662046, + "grad_norm": 2.9506170749664307, + "learning_rate": 1.878568709554625e-06, + "loss": 1.3298, + "step": 1948 + }, + { + "epoch": 2.0020544427324087, + "grad_norm": 2.8743488788604736, + "learning_rate": 1.8766653977921585e-06, + "loss": 1.3808, + "step": 1949 + }, + { + "epoch": 2.0030816640986133, + "grad_norm": 2.8676979541778564, + "learning_rate": 1.874762086029692e-06, + "loss": 1.3394, + "step": 1950 + }, + { + "epoch": 2.0030816640986133, + "eval_loss": 1.8473349809646606, + "eval_runtime": 37.093, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1950 + }, + { + "epoch": 2.0041088854648175, + "grad_norm": 3.1344428062438965, + "learning_rate": 1.8728587742672252e-06, + "loss": 1.3774, + "step": 1951 + }, + { + "epoch": 2.005136106831022, + "grad_norm": 3.1567065715789795, + "learning_rate": 1.8709554625047585e-06, + "loss": 1.3863, + "step": 1952 + }, + { + "epoch": 2.0061633281972266, + "grad_norm": 3.109293222427368, + "learning_rate": 1.8690521507422917e-06, + "loss": 1.3196, + "step": 1953 + }, + { + "epoch": 2.007190549563431, + "grad_norm": 3.1035029888153076, + "learning_rate": 1.867148838979825e-06, + "loss": 1.3473, + "step": 1954 + }, + { + "epoch": 2.0082177709296354, + "grad_norm": 3.02809476852417, + "learning_rate": 1.8652455272173584e-06, + "loss": 1.3718, + "step": 1955 + }, + { + "epoch": 2.00924499229584, + "grad_norm": 2.918850898742676, + "learning_rate": 1.8633422154548916e-06, + "loss": 1.3794, + "step": 1956 + }, + { + "epoch": 2.010272213662044, + "grad_norm": 2.89725923538208, + "learning_rate": 1.861438903692425e-06, + "loss": 1.2861, + "step": 1957 + }, + { + "epoch": 2.0112994350282487, + "grad_norm": 3.051868200302124, + "learning_rate": 1.859535591929958e-06, + "loss": 1.3122, + "step": 1958 + }, + { + "epoch": 2.012326656394453, + "grad_norm": 3.1091904640197754, + "learning_rate": 1.8576322801674915e-06, + "loss": 1.3941, + "step": 1959 + }, + { + "epoch": 2.0133538777606574, + "grad_norm": 3.046663761138916, + "learning_rate": 1.855728968405025e-06, + "loss": 1.3293, + "step": 1960 + }, + { + "epoch": 2.0133538777606574, + "eval_loss": 1.851180911064148, + "eval_runtime": 37.0842, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1960 + }, + { + "epoch": 2.014381099126862, + "grad_norm": 3.2143752574920654, + "learning_rate": 1.8538256566425584e-06, + "loss": 1.3431, + "step": 1961 + }, + { + "epoch": 2.015408320493066, + "grad_norm": 3.162942409515381, + "learning_rate": 1.8519223448800915e-06, + "loss": 1.3883, + "step": 1962 + }, + { + "epoch": 2.0164355418592708, + "grad_norm": 3.1271183490753174, + "learning_rate": 1.850019033117625e-06, + "loss": 1.3295, + "step": 1963 + }, + { + "epoch": 2.017462763225475, + "grad_norm": 3.07658314704895, + "learning_rate": 1.848115721355158e-06, + "loss": 1.3323, + "step": 1964 + }, + { + "epoch": 2.0184899845916795, + "grad_norm": 3.075946569442749, + "learning_rate": 1.8462124095926914e-06, + "loss": 1.3524, + "step": 1965 + }, + { + "epoch": 2.019517205957884, + "grad_norm": 3.0359086990356445, + "learning_rate": 1.8443090978302248e-06, + "loss": 1.3578, + "step": 1966 + }, + { + "epoch": 2.0205444273240882, + "grad_norm": 3.1957743167877197, + "learning_rate": 1.842405786067758e-06, + "loss": 1.3909, + "step": 1967 + }, + { + "epoch": 2.021571648690293, + "grad_norm": 3.002462863922119, + "learning_rate": 1.8405024743052913e-06, + "loss": 1.3466, + "step": 1968 + }, + { + "epoch": 2.022598870056497, + "grad_norm": 3.0507266521453857, + "learning_rate": 1.8385991625428247e-06, + "loss": 1.3717, + "step": 1969 + }, + { + "epoch": 2.0236260914227016, + "grad_norm": 3.0719404220581055, + "learning_rate": 1.8366958507803578e-06, + "loss": 1.313, + "step": 1970 + }, + { + "epoch": 2.0236260914227016, + "eval_loss": 1.8496923446655273, + "eval_runtime": 37.1112, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 1970 + }, + { + "epoch": 2.024653312788906, + "grad_norm": 3.068819522857666, + "learning_rate": 1.8347925390178914e-06, + "loss": 1.3614, + "step": 1971 + }, + { + "epoch": 2.0256805341551103, + "grad_norm": 3.005855083465576, + "learning_rate": 1.8328892272554247e-06, + "loss": 1.3007, + "step": 1972 + }, + { + "epoch": 2.026707755521315, + "grad_norm": 3.068032741546631, + "learning_rate": 1.8309859154929579e-06, + "loss": 1.3644, + "step": 1973 + }, + { + "epoch": 2.0277349768875195, + "grad_norm": 3.066153049468994, + "learning_rate": 1.8290826037304913e-06, + "loss": 1.3518, + "step": 1974 + }, + { + "epoch": 2.0287621982537236, + "grad_norm": 3.019429922103882, + "learning_rate": 1.8271792919680246e-06, + "loss": 1.3208, + "step": 1975 + }, + { + "epoch": 2.029789419619928, + "grad_norm": 3.2104897499084473, + "learning_rate": 1.8252759802055578e-06, + "loss": 1.329, + "step": 1976 + }, + { + "epoch": 2.0308166409861323, + "grad_norm": 3.272581100463867, + "learning_rate": 1.8233726684430911e-06, + "loss": 1.3665, + "step": 1977 + }, + { + "epoch": 2.031843862352337, + "grad_norm": 3.2155191898345947, + "learning_rate": 1.8214693566806243e-06, + "loss": 1.3178, + "step": 1978 + }, + { + "epoch": 2.0328710837185415, + "grad_norm": 3.130934238433838, + "learning_rate": 1.8195660449181576e-06, + "loss": 1.3492, + "step": 1979 + }, + { + "epoch": 2.0338983050847457, + "grad_norm": 3.211571216583252, + "learning_rate": 1.817662733155691e-06, + "loss": 1.3477, + "step": 1980 + }, + { + "epoch": 2.0338983050847457, + "eval_loss": 1.8516111373901367, + "eval_runtime": 37.1099, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 1980 + }, + { + "epoch": 2.0349255264509503, + "grad_norm": 2.9799704551696777, + "learning_rate": 1.8157594213932244e-06, + "loss": 1.3469, + "step": 1981 + }, + { + "epoch": 2.0359527478171544, + "grad_norm": 3.0847010612487793, + "learning_rate": 1.8138561096307577e-06, + "loss": 1.3587, + "step": 1982 + }, + { + "epoch": 2.036979969183359, + "grad_norm": 3.0739681720733643, + "learning_rate": 1.811952797868291e-06, + "loss": 1.3053, + "step": 1983 + }, + { + "epoch": 2.0380071905495636, + "grad_norm": 3.0315258502960205, + "learning_rate": 1.8100494861058242e-06, + "loss": 1.3381, + "step": 1984 + }, + { + "epoch": 2.0390344119157677, + "grad_norm": 2.9987258911132812, + "learning_rate": 1.8081461743433576e-06, + "loss": 1.3545, + "step": 1985 + }, + { + "epoch": 2.0400616332819723, + "grad_norm": 2.9383034706115723, + "learning_rate": 1.806242862580891e-06, + "loss": 1.3085, + "step": 1986 + }, + { + "epoch": 2.0410888546481765, + "grad_norm": 3.064450740814209, + "learning_rate": 1.8043395508184241e-06, + "loss": 1.3572, + "step": 1987 + }, + { + "epoch": 2.042116076014381, + "grad_norm": 3.102689027786255, + "learning_rate": 1.8024362390559575e-06, + "loss": 1.3453, + "step": 1988 + }, + { + "epoch": 2.0431432973805856, + "grad_norm": 3.0369763374328613, + "learning_rate": 1.8005329272934906e-06, + "loss": 1.3123, + "step": 1989 + }, + { + "epoch": 2.04417051874679, + "grad_norm": 3.2758162021636963, + "learning_rate": 1.798629615531024e-06, + "loss": 1.4068, + "step": 1990 + }, + { + "epoch": 2.04417051874679, + "eval_loss": 1.8500858545303345, + "eval_runtime": 37.334, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 1990 + }, + { + "epoch": 2.0451977401129944, + "grad_norm": 3.1648876667022705, + "learning_rate": 1.7967263037685576e-06, + "loss": 1.3328, + "step": 1991 + }, + { + "epoch": 2.046224961479199, + "grad_norm": 3.104123830795288, + "learning_rate": 1.794822992006091e-06, + "loss": 1.3297, + "step": 1992 + }, + { + "epoch": 2.047252182845403, + "grad_norm": 3.198483943939209, + "learning_rate": 1.792919680243624e-06, + "loss": 1.3851, + "step": 1993 + }, + { + "epoch": 2.0482794042116077, + "grad_norm": 3.212456226348877, + "learning_rate": 1.7910163684811574e-06, + "loss": 1.3062, + "step": 1994 + }, + { + "epoch": 2.049306625577812, + "grad_norm": 3.058353900909424, + "learning_rate": 1.7891130567186906e-06, + "loss": 1.3347, + "step": 1995 + }, + { + "epoch": 2.0503338469440164, + "grad_norm": 2.963183879852295, + "learning_rate": 1.787209744956224e-06, + "loss": 1.3163, + "step": 1996 + }, + { + "epoch": 2.051361068310221, + "grad_norm": 3.1013290882110596, + "learning_rate": 1.7853064331937573e-06, + "loss": 1.3377, + "step": 1997 + }, + { + "epoch": 2.052388289676425, + "grad_norm": 3.068808078765869, + "learning_rate": 1.7834031214312905e-06, + "loss": 1.3772, + "step": 1998 + }, + { + "epoch": 2.0534155110426298, + "grad_norm": 3.0556447505950928, + "learning_rate": 1.7814998096688238e-06, + "loss": 1.3113, + "step": 1999 + }, + { + "epoch": 2.054442732408834, + "grad_norm": 3.115323543548584, + "learning_rate": 1.779596497906357e-06, + "loss": 1.3499, + "step": 2000 + }, + { + "epoch": 2.054442732408834, + "eval_loss": 1.8396267890930176, + "eval_runtime": 37.4474, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 2000 + }, + { + "epoch": 2.0554699537750385, + "grad_norm": 3.133096694946289, + "learning_rate": 1.7776931861438906e-06, + "loss": 1.3938, + "step": 2001 + }, + { + "epoch": 2.056497175141243, + "grad_norm": 3.259174108505249, + "learning_rate": 1.775789874381424e-06, + "loss": 1.382, + "step": 2002 + }, + { + "epoch": 2.0575243965074472, + "grad_norm": 3.17948317527771, + "learning_rate": 1.7738865626189573e-06, + "loss": 1.3643, + "step": 2003 + }, + { + "epoch": 2.058551617873652, + "grad_norm": 3.0918309688568115, + "learning_rate": 1.7719832508564904e-06, + "loss": 1.3067, + "step": 2004 + }, + { + "epoch": 2.0595788392398564, + "grad_norm": 2.9949889183044434, + "learning_rate": 1.7700799390940238e-06, + "loss": 1.3323, + "step": 2005 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.044935464859009, + "learning_rate": 1.768176627331557e-06, + "loss": 1.359, + "step": 2006 + }, + { + "epoch": 2.061633281972265, + "grad_norm": 2.9965357780456543, + "learning_rate": 1.7662733155690903e-06, + "loss": 1.3456, + "step": 2007 + }, + { + "epoch": 2.0626605033384693, + "grad_norm": 3.092984676361084, + "learning_rate": 1.7643700038066237e-06, + "loss": 1.3831, + "step": 2008 + }, + { + "epoch": 2.063687724704674, + "grad_norm": 3.004263162612915, + "learning_rate": 1.7624666920441568e-06, + "loss": 1.3087, + "step": 2009 + }, + { + "epoch": 2.0647149460708785, + "grad_norm": 3.0408692359924316, + "learning_rate": 1.7605633802816902e-06, + "loss": 1.3058, + "step": 2010 + }, + { + "epoch": 2.0647149460708785, + "eval_loss": 1.8413820266723633, + "eval_runtime": 36.9354, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.0657421674370826, + "grad_norm": 3.155033826828003, + "learning_rate": 1.7586600685192238e-06, + "loss": 1.394, + "step": 2011 + }, + { + "epoch": 2.066769388803287, + "grad_norm": 3.3466567993164062, + "learning_rate": 1.756756756756757e-06, + "loss": 1.401, + "step": 2012 + }, + { + "epoch": 2.0677966101694913, + "grad_norm": 3.219165802001953, + "learning_rate": 1.7548534449942903e-06, + "loss": 1.3564, + "step": 2013 + }, + { + "epoch": 2.068823831535696, + "grad_norm": 3.1733009815216064, + "learning_rate": 1.7529501332318236e-06, + "loss": 1.3446, + "step": 2014 + }, + { + "epoch": 2.0698510529019005, + "grad_norm": 3.1392745971679688, + "learning_rate": 1.7510468214693568e-06, + "loss": 1.3212, + "step": 2015 + }, + { + "epoch": 2.0708782742681047, + "grad_norm": 2.9845967292785645, + "learning_rate": 1.7491435097068902e-06, + "loss": 1.3117, + "step": 2016 + }, + { + "epoch": 2.0719054956343093, + "grad_norm": 3.2182602882385254, + "learning_rate": 1.7472401979444233e-06, + "loss": 1.3539, + "step": 2017 + }, + { + "epoch": 2.0729327170005134, + "grad_norm": 3.0809013843536377, + "learning_rate": 1.7453368861819567e-06, + "loss": 1.2891, + "step": 2018 + }, + { + "epoch": 2.073959938366718, + "grad_norm": 3.0690958499908447, + "learning_rate": 1.74343357441949e-06, + "loss": 1.3381, + "step": 2019 + }, + { + "epoch": 2.0749871597329226, + "grad_norm": 3.091454267501831, + "learning_rate": 1.7415302626570232e-06, + "loss": 1.3874, + "step": 2020 + }, + { + "epoch": 2.0749871597329226, + "eval_loss": 1.8516818284988403, + "eval_runtime": 37.0619, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.0760143810991267, + "grad_norm": 3.039327621459961, + "learning_rate": 1.7396269508945568e-06, + "loss": 1.3295, + "step": 2021 + }, + { + "epoch": 2.0770416024653313, + "grad_norm": 3.0967671871185303, + "learning_rate": 1.7377236391320901e-06, + "loss": 1.3122, + "step": 2022 + }, + { + "epoch": 2.078068823831536, + "grad_norm": 3.228410005569458, + "learning_rate": 1.7358203273696233e-06, + "loss": 1.3766, + "step": 2023 + }, + { + "epoch": 2.07909604519774, + "grad_norm": 3.1973841190338135, + "learning_rate": 1.7339170156071566e-06, + "loss": 1.3325, + "step": 2024 + }, + { + "epoch": 2.0801232665639446, + "grad_norm": 3.131504535675049, + "learning_rate": 1.73201370384469e-06, + "loss": 1.3355, + "step": 2025 + }, + { + "epoch": 2.081150487930149, + "grad_norm": 3.132685661315918, + "learning_rate": 1.7301103920822231e-06, + "loss": 1.3446, + "step": 2026 + }, + { + "epoch": 2.0821777092963534, + "grad_norm": 3.2344162464141846, + "learning_rate": 1.7282070803197565e-06, + "loss": 1.3525, + "step": 2027 + }, + { + "epoch": 2.083204930662558, + "grad_norm": 3.1193928718566895, + "learning_rate": 1.7263037685572897e-06, + "loss": 1.3415, + "step": 2028 + }, + { + "epoch": 2.084232152028762, + "grad_norm": 3.164365291595459, + "learning_rate": 1.724400456794823e-06, + "loss": 1.3652, + "step": 2029 + }, + { + "epoch": 2.0852593733949667, + "grad_norm": 3.0825226306915283, + "learning_rate": 1.7224971450323564e-06, + "loss": 1.3242, + "step": 2030 + }, + { + "epoch": 2.0852593733949667, + "eval_loss": 1.8433457612991333, + "eval_runtime": 37.0921, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.086286594761171, + "grad_norm": 3.1651787757873535, + "learning_rate": 1.72059383326989e-06, + "loss": 1.333, + "step": 2031 + }, + { + "epoch": 2.0873138161273754, + "grad_norm": 3.295436382293701, + "learning_rate": 1.7186905215074231e-06, + "loss": 1.4064, + "step": 2032 + }, + { + "epoch": 2.08834103749358, + "grad_norm": 3.177628517150879, + "learning_rate": 1.7167872097449565e-06, + "loss": 1.3173, + "step": 2033 + }, + { + "epoch": 2.089368258859784, + "grad_norm": 3.112971544265747, + "learning_rate": 1.7148838979824896e-06, + "loss": 1.3401, + "step": 2034 + }, + { + "epoch": 2.0903954802259888, + "grad_norm": 3.150045871734619, + "learning_rate": 1.712980586220023e-06, + "loss": 1.3371, + "step": 2035 + }, + { + "epoch": 2.091422701592193, + "grad_norm": 3.0645205974578857, + "learning_rate": 1.7110772744575563e-06, + "loss": 1.3213, + "step": 2036 + }, + { + "epoch": 2.0924499229583975, + "grad_norm": 3.230185031890869, + "learning_rate": 1.7091739626950895e-06, + "loss": 1.3317, + "step": 2037 + }, + { + "epoch": 2.093477144324602, + "grad_norm": 3.1509249210357666, + "learning_rate": 1.7072706509326229e-06, + "loss": 1.3706, + "step": 2038 + }, + { + "epoch": 2.0945043656908062, + "grad_norm": 3.185981273651123, + "learning_rate": 1.7053673391701562e-06, + "loss": 1.3398, + "step": 2039 + }, + { + "epoch": 2.095531587057011, + "grad_norm": 3.000025510787964, + "learning_rate": 1.7034640274076894e-06, + "loss": 1.3197, + "step": 2040 + }, + { + "epoch": 2.095531587057011, + "eval_loss": 1.8413926362991333, + "eval_runtime": 37.0971, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 2040 + }, + { + "epoch": 2.0965588084232154, + "grad_norm": 3.156095027923584, + "learning_rate": 1.701560715645223e-06, + "loss": 1.3864, + "step": 2041 + }, + { + "epoch": 2.0975860297894195, + "grad_norm": 2.970560073852539, + "learning_rate": 1.6996574038827563e-06, + "loss": 1.3016, + "step": 2042 + }, + { + "epoch": 2.098613251155624, + "grad_norm": 3.141657829284668, + "learning_rate": 1.6977540921202895e-06, + "loss": 1.3488, + "step": 2043 + }, + { + "epoch": 2.0996404725218283, + "grad_norm": 3.125731945037842, + "learning_rate": 1.6958507803578228e-06, + "loss": 1.3275, + "step": 2044 + }, + { + "epoch": 2.100667693888033, + "grad_norm": 3.2703487873077393, + "learning_rate": 1.693947468595356e-06, + "loss": 1.3577, + "step": 2045 + }, + { + "epoch": 2.1016949152542375, + "grad_norm": 3.1855714321136475, + "learning_rate": 1.6920441568328893e-06, + "loss": 1.3671, + "step": 2046 + }, + { + "epoch": 2.1027221366204416, + "grad_norm": 3.0699303150177, + "learning_rate": 1.6901408450704227e-06, + "loss": 1.3066, + "step": 2047 + }, + { + "epoch": 2.103749357986646, + "grad_norm": 3.2417423725128174, + "learning_rate": 1.6882375333079559e-06, + "loss": 1.3274, + "step": 2048 + }, + { + "epoch": 2.1047765793528503, + "grad_norm": 3.265108108520508, + "learning_rate": 1.6863342215454892e-06, + "loss": 1.3924, + "step": 2049 + }, + { + "epoch": 2.105803800719055, + "grad_norm": 3.2078635692596436, + "learning_rate": 1.6844309097830226e-06, + "loss": 1.3782, + "step": 2050 + }, + { + "epoch": 2.105803800719055, + "eval_loss": 1.8441476821899414, + "eval_runtime": 37.1042, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.1068310220852595, + "grad_norm": 3.180485248565674, + "learning_rate": 1.682527598020556e-06, + "loss": 1.3613, + "step": 2051 + }, + { + "epoch": 2.1078582434514637, + "grad_norm": 3.110530376434326, + "learning_rate": 1.6806242862580893e-06, + "loss": 1.376, + "step": 2052 + }, + { + "epoch": 2.1088854648176683, + "grad_norm": 2.997999429702759, + "learning_rate": 1.6787209744956227e-06, + "loss": 1.3172, + "step": 2053 + }, + { + "epoch": 2.109912686183873, + "grad_norm": 3.0154783725738525, + "learning_rate": 1.6768176627331558e-06, + "loss": 1.3015, + "step": 2054 + }, + { + "epoch": 2.110939907550077, + "grad_norm": 2.992138624191284, + "learning_rate": 1.6749143509706892e-06, + "loss": 1.3517, + "step": 2055 + }, + { + "epoch": 2.1119671289162816, + "grad_norm": 3.0572457313537598, + "learning_rate": 1.6730110392082225e-06, + "loss": 1.3301, + "step": 2056 + }, + { + "epoch": 2.1129943502824857, + "grad_norm": 2.9924540519714355, + "learning_rate": 1.6711077274457557e-06, + "loss": 1.3534, + "step": 2057 + }, + { + "epoch": 2.1140215716486903, + "grad_norm": 2.9569015502929688, + "learning_rate": 1.669204415683289e-06, + "loss": 1.2972, + "step": 2058 + }, + { + "epoch": 2.115048793014895, + "grad_norm": 3.064371109008789, + "learning_rate": 1.6673011039208222e-06, + "loss": 1.3553, + "step": 2059 + }, + { + "epoch": 2.116076014381099, + "grad_norm": 3.0954525470733643, + "learning_rate": 1.6653977921583556e-06, + "loss": 1.3421, + "step": 2060 + }, + { + "epoch": 2.116076014381099, + "eval_loss": 1.8407723903656006, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 2060 + }, + { + "epoch": 2.1171032357473036, + "grad_norm": 3.2661712169647217, + "learning_rate": 1.663494480395889e-06, + "loss": 1.3582, + "step": 2061 + }, + { + "epoch": 2.118130457113508, + "grad_norm": 3.1663613319396973, + "learning_rate": 1.6615911686334225e-06, + "loss": 1.3203, + "step": 2062 + }, + { + "epoch": 2.1191576784797124, + "grad_norm": 3.0864570140838623, + "learning_rate": 1.6596878568709557e-06, + "loss": 1.3431, + "step": 2063 + }, + { + "epoch": 2.120184899845917, + "grad_norm": 3.1531989574432373, + "learning_rate": 1.657784545108489e-06, + "loss": 1.3275, + "step": 2064 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.2258596420288086, + "learning_rate": 1.6558812333460222e-06, + "loss": 1.3711, + "step": 2065 + }, + { + "epoch": 2.1222393425783257, + "grad_norm": 3.2881453037261963, + "learning_rate": 1.6539779215835555e-06, + "loss": 1.3688, + "step": 2066 + }, + { + "epoch": 2.12326656394453, + "grad_norm": 3.0734169483184814, + "learning_rate": 1.652074609821089e-06, + "loss": 1.3277, + "step": 2067 + }, + { + "epoch": 2.1242937853107344, + "grad_norm": 2.9943618774414062, + "learning_rate": 1.650171298058622e-06, + "loss": 1.3268, + "step": 2068 + }, + { + "epoch": 2.125321006676939, + "grad_norm": 3.034625291824341, + "learning_rate": 1.6482679862961554e-06, + "loss": 1.3265, + "step": 2069 + }, + { + "epoch": 2.126348228043143, + "grad_norm": 3.0360612869262695, + "learning_rate": 1.6463646745336886e-06, + "loss": 1.3256, + "step": 2070 + }, + { + "epoch": 2.126348228043143, + "eval_loss": 1.8443471193313599, + "eval_runtime": 37.0917, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.1273754494093478, + "grad_norm": 3.0917110443115234, + "learning_rate": 1.644461362771222e-06, + "loss": 1.3099, + "step": 2071 + }, + { + "epoch": 2.128402670775552, + "grad_norm": 3.1142187118530273, + "learning_rate": 1.6425580510087555e-06, + "loss": 1.3421, + "step": 2072 + }, + { + "epoch": 2.1294298921417565, + "grad_norm": 3.2128524780273438, + "learning_rate": 1.6406547392462889e-06, + "loss": 1.3123, + "step": 2073 + }, + { + "epoch": 2.130457113507961, + "grad_norm": 3.060520887374878, + "learning_rate": 1.638751427483822e-06, + "loss": 1.3069, + "step": 2074 + }, + { + "epoch": 2.131484334874165, + "grad_norm": 3.092120885848999, + "learning_rate": 1.6368481157213554e-06, + "loss": 1.3537, + "step": 2075 + }, + { + "epoch": 2.13251155624037, + "grad_norm": 3.2490074634552, + "learning_rate": 1.6349448039588885e-06, + "loss": 1.3324, + "step": 2076 + }, + { + "epoch": 2.1335387776065744, + "grad_norm": 3.2517645359039307, + "learning_rate": 1.6330414921964219e-06, + "loss": 1.385, + "step": 2077 + }, + { + "epoch": 2.1345659989727785, + "grad_norm": 3.1669421195983887, + "learning_rate": 1.6311381804339553e-06, + "loss": 1.3054, + "step": 2078 + }, + { + "epoch": 2.135593220338983, + "grad_norm": 3.2323193550109863, + "learning_rate": 1.6292348686714884e-06, + "loss": 1.3488, + "step": 2079 + }, + { + "epoch": 2.1366204417051873, + "grad_norm": 3.1058294773101807, + "learning_rate": 1.6273315569090218e-06, + "loss": 1.3256, + "step": 2080 + }, + { + "epoch": 2.1366204417051873, + "eval_loss": 1.8446909189224243, + "eval_runtime": 37.1231, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.137647663071392, + "grad_norm": 3.1662075519561768, + "learning_rate": 1.625428245146555e-06, + "loss": 1.4073, + "step": 2081 + }, + { + "epoch": 2.1386748844375965, + "grad_norm": 3.083482503890991, + "learning_rate": 1.6235249333840885e-06, + "loss": 1.3897, + "step": 2082 + }, + { + "epoch": 2.1397021058038006, + "grad_norm": 3.040388345718384, + "learning_rate": 1.6216216216216219e-06, + "loss": 1.349, + "step": 2083 + }, + { + "epoch": 2.140729327170005, + "grad_norm": 3.124114513397217, + "learning_rate": 1.6197183098591552e-06, + "loss": 1.3732, + "step": 2084 + }, + { + "epoch": 2.1417565485362093, + "grad_norm": 3.2662036418914795, + "learning_rate": 1.6178149980966884e-06, + "loss": 1.3522, + "step": 2085 + }, + { + "epoch": 2.142783769902414, + "grad_norm": 3.149472951889038, + "learning_rate": 1.6159116863342217e-06, + "loss": 1.3179, + "step": 2086 + }, + { + "epoch": 2.1438109912686185, + "grad_norm": 3.256852865219116, + "learning_rate": 1.6140083745717549e-06, + "loss": 1.3897, + "step": 2087 + }, + { + "epoch": 2.1448382126348227, + "grad_norm": 3.1227493286132812, + "learning_rate": 1.6121050628092882e-06, + "loss": 1.3104, + "step": 2088 + }, + { + "epoch": 2.1458654340010272, + "grad_norm": 3.3213047981262207, + "learning_rate": 1.6102017510468216e-06, + "loss": 1.3829, + "step": 2089 + }, + { + "epoch": 2.146892655367232, + "grad_norm": 3.160428047180176, + "learning_rate": 1.6082984392843548e-06, + "loss": 1.2939, + "step": 2090 + }, + { + "epoch": 2.146892655367232, + "eval_loss": 1.8534585237503052, + "eval_runtime": 37.3333, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.147919876733436, + "grad_norm": 3.224741220474243, + "learning_rate": 1.6063951275218881e-06, + "loss": 1.3418, + "step": 2091 + }, + { + "epoch": 2.1489470980996406, + "grad_norm": 3.3884503841400146, + "learning_rate": 1.6044918157594217e-06, + "loss": 1.395, + "step": 2092 + }, + { + "epoch": 2.1499743194658447, + "grad_norm": 3.1761722564697266, + "learning_rate": 1.6025885039969548e-06, + "loss": 1.3208, + "step": 2093 + }, + { + "epoch": 2.1510015408320493, + "grad_norm": 3.0476465225219727, + "learning_rate": 1.6006851922344882e-06, + "loss": 1.3465, + "step": 2094 + }, + { + "epoch": 2.152028762198254, + "grad_norm": 3.1662650108337402, + "learning_rate": 1.5987818804720216e-06, + "loss": 1.3645, + "step": 2095 + }, + { + "epoch": 2.153055983564458, + "grad_norm": 3.1860709190368652, + "learning_rate": 1.5968785687095547e-06, + "loss": 1.361, + "step": 2096 + }, + { + "epoch": 2.1540832049306626, + "grad_norm": 3.104501962661743, + "learning_rate": 1.594975256947088e-06, + "loss": 1.3135, + "step": 2097 + }, + { + "epoch": 2.1551104262968668, + "grad_norm": 3.067155599594116, + "learning_rate": 1.5930719451846212e-06, + "loss": 1.3693, + "step": 2098 + }, + { + "epoch": 2.1561376476630714, + "grad_norm": 3.195204019546509, + "learning_rate": 1.5911686334221546e-06, + "loss": 1.3143, + "step": 2099 + }, + { + "epoch": 2.157164869029276, + "grad_norm": 3.159278154373169, + "learning_rate": 1.589265321659688e-06, + "loss": 1.3607, + "step": 2100 + }, + { + "epoch": 2.157164869029276, + "eval_loss": 1.8502479791641235, + "eval_runtime": 37.4583, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.15819209039548, + "grad_norm": 2.973191261291504, + "learning_rate": 1.5873620098972211e-06, + "loss": 1.2776, + "step": 2101 + }, + { + "epoch": 2.1592193117616847, + "grad_norm": 3.0078418254852295, + "learning_rate": 1.5854586981347547e-06, + "loss": 1.3173, + "step": 2102 + }, + { + "epoch": 2.1602465331278893, + "grad_norm": 3.2115561962127686, + "learning_rate": 1.583555386372288e-06, + "loss": 1.3621, + "step": 2103 + }, + { + "epoch": 2.1612737544940934, + "grad_norm": 3.234210968017578, + "learning_rate": 1.5816520746098212e-06, + "loss": 1.374, + "step": 2104 + }, + { + "epoch": 2.162300975860298, + "grad_norm": 3.195405960083008, + "learning_rate": 1.5797487628473546e-06, + "loss": 1.3402, + "step": 2105 + }, + { + "epoch": 2.163328197226502, + "grad_norm": 3.016078233718872, + "learning_rate": 1.577845451084888e-06, + "loss": 1.3036, + "step": 2106 + }, + { + "epoch": 2.1643554185927067, + "grad_norm": 3.2147345542907715, + "learning_rate": 1.575942139322421e-06, + "loss": 1.3511, + "step": 2107 + }, + { + "epoch": 2.1653826399589113, + "grad_norm": 2.980320692062378, + "learning_rate": 1.5740388275599544e-06, + "loss": 1.2635, + "step": 2108 + }, + { + "epoch": 2.1664098613251155, + "grad_norm": 3.0693414211273193, + "learning_rate": 1.5721355157974876e-06, + "loss": 1.3435, + "step": 2109 + }, + { + "epoch": 2.16743708269132, + "grad_norm": 3.1445682048797607, + "learning_rate": 1.570232204035021e-06, + "loss": 1.3294, + "step": 2110 + }, + { + "epoch": 2.16743708269132, + "eval_loss": 1.8479558229446411, + "eval_runtime": 36.9467, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.168464304057524, + "grad_norm": 3.1309633255004883, + "learning_rate": 1.5683288922725543e-06, + "loss": 1.3312, + "step": 2111 + }, + { + "epoch": 2.169491525423729, + "grad_norm": 3.0906994342803955, + "learning_rate": 1.5664255805100879e-06, + "loss": 1.342, + "step": 2112 + }, + { + "epoch": 2.1705187467899334, + "grad_norm": 3.1987171173095703, + "learning_rate": 1.564522268747621e-06, + "loss": 1.3358, + "step": 2113 + }, + { + "epoch": 2.1715459681561375, + "grad_norm": 3.221889019012451, + "learning_rate": 1.5626189569851544e-06, + "loss": 1.3484, + "step": 2114 + }, + { + "epoch": 2.172573189522342, + "grad_norm": 3.1508629322052, + "learning_rate": 1.5607156452226875e-06, + "loss": 1.2785, + "step": 2115 + }, + { + "epoch": 2.1736004108885463, + "grad_norm": 3.115274667739868, + "learning_rate": 1.558812333460221e-06, + "loss": 1.3463, + "step": 2116 + }, + { + "epoch": 2.174627632254751, + "grad_norm": 3.0454249382019043, + "learning_rate": 1.5569090216977543e-06, + "loss": 1.3534, + "step": 2117 + }, + { + "epoch": 2.1756548536209555, + "grad_norm": 3.20078182220459, + "learning_rate": 1.5550057099352874e-06, + "loss": 1.3469, + "step": 2118 + }, + { + "epoch": 2.1766820749871596, + "grad_norm": 3.091568946838379, + "learning_rate": 1.5531023981728208e-06, + "loss": 1.3547, + "step": 2119 + }, + { + "epoch": 2.177709296353364, + "grad_norm": 3.0242369174957275, + "learning_rate": 1.5511990864103542e-06, + "loss": 1.3198, + "step": 2120 + }, + { + "epoch": 2.177709296353364, + "eval_loss": 1.8469057083129883, + "eval_runtime": 37.0881, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.1787365177195683, + "grad_norm": 3.1775739192962646, + "learning_rate": 1.5492957746478873e-06, + "loss": 1.3491, + "step": 2121 + }, + { + "epoch": 2.179763739085773, + "grad_norm": 3.239668846130371, + "learning_rate": 1.5473924628854209e-06, + "loss": 1.3641, + "step": 2122 + }, + { + "epoch": 2.1807909604519775, + "grad_norm": 3.194678544998169, + "learning_rate": 1.5454891511229542e-06, + "loss": 1.3317, + "step": 2123 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.129948377609253, + "learning_rate": 1.5435858393604874e-06, + "loss": 1.3405, + "step": 2124 + }, + { + "epoch": 2.1828454031843862, + "grad_norm": 3.2040436267852783, + "learning_rate": 1.5416825275980208e-06, + "loss": 1.3596, + "step": 2125 + }, + { + "epoch": 2.183872624550591, + "grad_norm": 3.2529456615448, + "learning_rate": 1.539779215835554e-06, + "loss": 1.3171, + "step": 2126 + }, + { + "epoch": 2.184899845916795, + "grad_norm": 3.2071499824523926, + "learning_rate": 1.5378759040730873e-06, + "loss": 1.318, + "step": 2127 + }, + { + "epoch": 2.1859270672829996, + "grad_norm": 3.2709555625915527, + "learning_rate": 1.5359725923106206e-06, + "loss": 1.3811, + "step": 2128 + }, + { + "epoch": 2.1869542886492037, + "grad_norm": 3.244680643081665, + "learning_rate": 1.5340692805481538e-06, + "loss": 1.3539, + "step": 2129 + }, + { + "epoch": 2.1879815100154083, + "grad_norm": 3.068683624267578, + "learning_rate": 1.5321659687856871e-06, + "loss": 1.3207, + "step": 2130 + }, + { + "epoch": 2.1879815100154083, + "eval_loss": 1.8478527069091797, + "eval_runtime": 37.138, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.189008731381613, + "grad_norm": 3.1829938888549805, + "learning_rate": 1.5302626570232205e-06, + "loss": 1.3127, + "step": 2131 + }, + { + "epoch": 2.190035952747817, + "grad_norm": 3.060347080230713, + "learning_rate": 1.5283593452607539e-06, + "loss": 1.287, + "step": 2132 + }, + { + "epoch": 2.1910631741140216, + "grad_norm": 3.3016905784606934, + "learning_rate": 1.5264560334982872e-06, + "loss": 1.3838, + "step": 2133 + }, + { + "epoch": 2.1920903954802258, + "grad_norm": 3.1945807933807373, + "learning_rate": 1.5245527217358206e-06, + "loss": 1.3311, + "step": 2134 + }, + { + "epoch": 2.1931176168464304, + "grad_norm": 3.182347059249878, + "learning_rate": 1.5226494099733537e-06, + "loss": 1.3544, + "step": 2135 + }, + { + "epoch": 2.194144838212635, + "grad_norm": 3.069206476211548, + "learning_rate": 1.520746098210887e-06, + "loss": 1.3148, + "step": 2136 + }, + { + "epoch": 2.195172059578839, + "grad_norm": 3.1632168292999268, + "learning_rate": 1.5188427864484205e-06, + "loss": 1.3857, + "step": 2137 + }, + { + "epoch": 2.1961992809450437, + "grad_norm": 3.055793523788452, + "learning_rate": 1.5169394746859536e-06, + "loss": 1.3116, + "step": 2138 + }, + { + "epoch": 2.1972265023112483, + "grad_norm": 3.1330678462982178, + "learning_rate": 1.515036162923487e-06, + "loss": 1.3214, + "step": 2139 + }, + { + "epoch": 2.1982537236774524, + "grad_norm": 3.118450880050659, + "learning_rate": 1.5131328511610201e-06, + "loss": 1.3479, + "step": 2140 + }, + { + "epoch": 2.1982537236774524, + "eval_loss": 1.8538312911987305, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.199280945043657, + "grad_norm": 3.249545097351074, + "learning_rate": 1.5112295393985535e-06, + "loss": 1.379, + "step": 2141 + }, + { + "epoch": 2.200308166409861, + "grad_norm": 3.1331286430358887, + "learning_rate": 1.509326227636087e-06, + "loss": 1.3621, + "step": 2142 + }, + { + "epoch": 2.2013353877760657, + "grad_norm": 3.150987148284912, + "learning_rate": 1.5074229158736204e-06, + "loss": 1.291, + "step": 2143 + }, + { + "epoch": 2.2023626091422703, + "grad_norm": 3.1020572185516357, + "learning_rate": 1.5055196041111536e-06, + "loss": 1.3381, + "step": 2144 + }, + { + "epoch": 2.2033898305084745, + "grad_norm": 3.129390239715576, + "learning_rate": 1.503616292348687e-06, + "loss": 1.3405, + "step": 2145 + }, + { + "epoch": 2.204417051874679, + "grad_norm": 3.2193870544433594, + "learning_rate": 1.50171298058622e-06, + "loss": 1.3744, + "step": 2146 + }, + { + "epoch": 2.205444273240883, + "grad_norm": 3.043828248977661, + "learning_rate": 1.4998096688237535e-06, + "loss": 1.2782, + "step": 2147 + }, + { + "epoch": 2.206471494607088, + "grad_norm": 3.2130300998687744, + "learning_rate": 1.4979063570612868e-06, + "loss": 1.3625, + "step": 2148 + }, + { + "epoch": 2.2074987159732924, + "grad_norm": 3.250370979309082, + "learning_rate": 1.49600304529882e-06, + "loss": 1.3396, + "step": 2149 + }, + { + "epoch": 2.2085259373394965, + "grad_norm": 3.0954599380493164, + "learning_rate": 1.4940997335363533e-06, + "loss": 1.3187, + "step": 2150 + }, + { + "epoch": 2.2085259373394965, + "eval_loss": 1.8469802141189575, + "eval_runtime": 37.2109, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 2150 + }, + { + "epoch": 2.209553158705701, + "grad_norm": 3.153545379638672, + "learning_rate": 1.4921964217738865e-06, + "loss": 1.3488, + "step": 2151 + }, + { + "epoch": 2.2105803800719057, + "grad_norm": 3.1206071376800537, + "learning_rate": 1.49029311001142e-06, + "loss": 1.336, + "step": 2152 + }, + { + "epoch": 2.21160760143811, + "grad_norm": 3.151784896850586, + "learning_rate": 1.4883897982489534e-06, + "loss": 1.3613, + "step": 2153 + }, + { + "epoch": 2.2126348228043144, + "grad_norm": 3.18965744972229, + "learning_rate": 1.4864864864864868e-06, + "loss": 1.3698, + "step": 2154 + }, + { + "epoch": 2.2136620441705186, + "grad_norm": 3.108281135559082, + "learning_rate": 1.48458317472402e-06, + "loss": 1.3287, + "step": 2155 + }, + { + "epoch": 2.214689265536723, + "grad_norm": 3.2309465408325195, + "learning_rate": 1.4826798629615533e-06, + "loss": 1.3425, + "step": 2156 + }, + { + "epoch": 2.2157164869029278, + "grad_norm": 3.0829577445983887, + "learning_rate": 1.4807765511990864e-06, + "loss": 1.2917, + "step": 2157 + }, + { + "epoch": 2.216743708269132, + "grad_norm": 3.1417033672332764, + "learning_rate": 1.4788732394366198e-06, + "loss": 1.3113, + "step": 2158 + }, + { + "epoch": 2.2177709296353365, + "grad_norm": 3.1996822357177734, + "learning_rate": 1.4769699276741532e-06, + "loss": 1.3248, + "step": 2159 + }, + { + "epoch": 2.2187981510015407, + "grad_norm": 3.262814998626709, + "learning_rate": 1.4750666159116863e-06, + "loss": 1.3186, + "step": 2160 + }, + { + "epoch": 2.2187981510015407, + "eval_loss": 1.844687819480896, + "eval_runtime": 37.2211, + "eval_samples_per_second": 26.866, + "eval_steps_per_second": 1.128, + "step": 2160 + }, + { + "epoch": 2.2198253723677452, + "grad_norm": 3.0836942195892334, + "learning_rate": 1.4731633041492197e-06, + "loss": 1.2935, + "step": 2161 + }, + { + "epoch": 2.22085259373395, + "grad_norm": 3.1720023155212402, + "learning_rate": 1.4712599923867528e-06, + "loss": 1.3622, + "step": 2162 + }, + { + "epoch": 2.221879815100154, + "grad_norm": 3.2829599380493164, + "learning_rate": 1.4693566806242864e-06, + "loss": 1.3794, + "step": 2163 + }, + { + "epoch": 2.2229070364663586, + "grad_norm": 3.045949935913086, + "learning_rate": 1.4674533688618198e-06, + "loss": 1.3054, + "step": 2164 + }, + { + "epoch": 2.2239342578325627, + "grad_norm": 3.1561572551727295, + "learning_rate": 1.4655500570993531e-06, + "loss": 1.3289, + "step": 2165 + }, + { + "epoch": 2.2249614791987673, + "grad_norm": 3.0877439975738525, + "learning_rate": 1.4636467453368863e-06, + "loss": 1.329, + "step": 2166 + }, + { + "epoch": 2.225988700564972, + "grad_norm": 3.1330180168151855, + "learning_rate": 1.4617434335744197e-06, + "loss": 1.3853, + "step": 2167 + }, + { + "epoch": 2.227015921931176, + "grad_norm": 3.231693744659424, + "learning_rate": 1.4598401218119528e-06, + "loss": 1.378, + "step": 2168 + }, + { + "epoch": 2.2280431432973806, + "grad_norm": 3.202930212020874, + "learning_rate": 1.4579368100494862e-06, + "loss": 1.3549, + "step": 2169 + }, + { + "epoch": 2.2290703646635848, + "grad_norm": 3.164461135864258, + "learning_rate": 1.4560334982870195e-06, + "loss": 1.3365, + "step": 2170 + }, + { + "epoch": 2.2290703646635848, + "eval_loss": 1.8483937978744507, + "eval_runtime": 37.1839, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.2300975860297894, + "grad_norm": 3.1866867542266846, + "learning_rate": 1.4541301865245527e-06, + "loss": 1.3503, + "step": 2171 + }, + { + "epoch": 2.231124807395994, + "grad_norm": 3.2202508449554443, + "learning_rate": 1.452226874762086e-06, + "loss": 1.354, + "step": 2172 + }, + { + "epoch": 2.232152028762198, + "grad_norm": 3.190744161605835, + "learning_rate": 1.4503235629996196e-06, + "loss": 1.3073, + "step": 2173 + }, + { + "epoch": 2.2331792501284027, + "grad_norm": 3.2158405780792236, + "learning_rate": 1.4484202512371528e-06, + "loss": 1.3731, + "step": 2174 + }, + { + "epoch": 2.2342064714946073, + "grad_norm": 3.0294954776763916, + "learning_rate": 1.4465169394746861e-06, + "loss": 1.2772, + "step": 2175 + }, + { + "epoch": 2.2352336928608114, + "grad_norm": 3.2871217727661133, + "learning_rate": 1.4446136277122195e-06, + "loss": 1.3888, + "step": 2176 + }, + { + "epoch": 2.236260914227016, + "grad_norm": 3.0494141578674316, + "learning_rate": 1.4427103159497526e-06, + "loss": 1.2991, + "step": 2177 + }, + { + "epoch": 2.23728813559322, + "grad_norm": 3.122699022293091, + "learning_rate": 1.440807004187286e-06, + "loss": 1.3186, + "step": 2178 + }, + { + "epoch": 2.2383153569594247, + "grad_norm": 3.119854211807251, + "learning_rate": 1.4389036924248192e-06, + "loss": 1.3421, + "step": 2179 + }, + { + "epoch": 2.2393425783256293, + "grad_norm": 3.2588119506835938, + "learning_rate": 1.4370003806623525e-06, + "loss": 1.3657, + "step": 2180 + }, + { + "epoch": 2.2393425783256293, + "eval_loss": 1.8473975658416748, + "eval_runtime": 37.1561, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.2403697996918335, + "grad_norm": 3.138901948928833, + "learning_rate": 1.4350970688998859e-06, + "loss": 1.3273, + "step": 2181 + }, + { + "epoch": 2.241397021058038, + "grad_norm": 3.2792508602142334, + "learning_rate": 1.433193757137419e-06, + "loss": 1.3763, + "step": 2182 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.065627098083496, + "learning_rate": 1.4312904453749526e-06, + "loss": 1.3292, + "step": 2183 + }, + { + "epoch": 2.243451463790447, + "grad_norm": 3.101377010345459, + "learning_rate": 1.429387133612486e-06, + "loss": 1.2957, + "step": 2184 + }, + { + "epoch": 2.2444786851566514, + "grad_norm": 3.140158176422119, + "learning_rate": 1.4274838218500191e-06, + "loss": 1.3406, + "step": 2185 + }, + { + "epoch": 2.2455059065228555, + "grad_norm": 3.2498116493225098, + "learning_rate": 1.4255805100875525e-06, + "loss": 1.3499, + "step": 2186 + }, + { + "epoch": 2.24653312788906, + "grad_norm": 3.2681965827941895, + "learning_rate": 1.4236771983250858e-06, + "loss": 1.407, + "step": 2187 + }, + { + "epoch": 2.2475603492552647, + "grad_norm": 3.2457778453826904, + "learning_rate": 1.421773886562619e-06, + "loss": 1.3278, + "step": 2188 + }, + { + "epoch": 2.248587570621469, + "grad_norm": 3.099942207336426, + "learning_rate": 1.4198705748001524e-06, + "loss": 1.3114, + "step": 2189 + }, + { + "epoch": 2.2496147919876734, + "grad_norm": 3.090271234512329, + "learning_rate": 1.4179672630376855e-06, + "loss": 1.3285, + "step": 2190 + }, + { + "epoch": 2.2496147919876734, + "eval_loss": 1.8479118347167969, + "eval_runtime": 37.2331, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.2506420133538776, + "grad_norm": 3.0344860553741455, + "learning_rate": 1.4160639512752189e-06, + "loss": 1.3104, + "step": 2191 + }, + { + "epoch": 2.251669234720082, + "grad_norm": 3.087860584259033, + "learning_rate": 1.4141606395127522e-06, + "loss": 1.3167, + "step": 2192 + }, + { + "epoch": 2.2526964560862868, + "grad_norm": 3.0974831581115723, + "learning_rate": 1.4122573277502858e-06, + "loss": 1.2926, + "step": 2193 + }, + { + "epoch": 2.253723677452491, + "grad_norm": 3.228492498397827, + "learning_rate": 1.410354015987819e-06, + "loss": 1.3541, + "step": 2194 + }, + { + "epoch": 2.2547508988186955, + "grad_norm": 3.1761956214904785, + "learning_rate": 1.4084507042253523e-06, + "loss": 1.3342, + "step": 2195 + }, + { + "epoch": 2.2557781201848996, + "grad_norm": 3.1890695095062256, + "learning_rate": 1.4065473924628855e-06, + "loss": 1.362, + "step": 2196 + }, + { + "epoch": 2.2568053415511042, + "grad_norm": 3.1543149948120117, + "learning_rate": 1.4046440807004188e-06, + "loss": 1.329, + "step": 2197 + }, + { + "epoch": 2.257832562917309, + "grad_norm": 3.2313342094421387, + "learning_rate": 1.4027407689379522e-06, + "loss": 1.3256, + "step": 2198 + }, + { + "epoch": 2.258859784283513, + "grad_norm": 3.130805253982544, + "learning_rate": 1.4008374571754854e-06, + "loss": 1.3047, + "step": 2199 + }, + { + "epoch": 2.2598870056497176, + "grad_norm": 3.0488216876983643, + "learning_rate": 1.3989341454130187e-06, + "loss": 1.3461, + "step": 2200 + }, + { + "epoch": 2.2598870056497176, + "eval_loss": 1.8517944812774658, + "eval_runtime": 37.4528, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 2200 + }, + { + "epoch": 2.260914227015922, + "grad_norm": 3.0915918350219727, + "learning_rate": 1.397030833650552e-06, + "loss": 1.3391, + "step": 2201 + }, + { + "epoch": 2.2619414483821263, + "grad_norm": 3.156053066253662, + "learning_rate": 1.3951275218880852e-06, + "loss": 1.3525, + "step": 2202 + }, + { + "epoch": 2.262968669748331, + "grad_norm": 3.4035868644714355, + "learning_rate": 1.3932242101256188e-06, + "loss": 1.4122, + "step": 2203 + }, + { + "epoch": 2.263995891114535, + "grad_norm": 2.9787867069244385, + "learning_rate": 1.3913208983631522e-06, + "loss": 1.3096, + "step": 2204 + }, + { + "epoch": 2.2650231124807396, + "grad_norm": 3.154271364212036, + "learning_rate": 1.3894175866006853e-06, + "loss": 1.3421, + "step": 2205 + }, + { + "epoch": 2.266050333846944, + "grad_norm": 3.1899831295013428, + "learning_rate": 1.3875142748382187e-06, + "loss": 1.2949, + "step": 2206 + }, + { + "epoch": 2.2670775552131484, + "grad_norm": 3.126694679260254, + "learning_rate": 1.385610963075752e-06, + "loss": 1.2898, + "step": 2207 + }, + { + "epoch": 2.268104776579353, + "grad_norm": 3.149989128112793, + "learning_rate": 1.3837076513132852e-06, + "loss": 1.3369, + "step": 2208 + }, + { + "epoch": 2.269131997945557, + "grad_norm": 3.250481128692627, + "learning_rate": 1.3818043395508186e-06, + "loss": 1.3605, + "step": 2209 + }, + { + "epoch": 2.2701592193117617, + "grad_norm": 3.2549924850463867, + "learning_rate": 1.3799010277883517e-06, + "loss": 1.3428, + "step": 2210 + }, + { + "epoch": 2.2701592193117617, + "eval_loss": 1.8562498092651367, + "eval_runtime": 37.0713, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 2210 + }, + { + "epoch": 2.2711864406779663, + "grad_norm": 3.1369433403015137, + "learning_rate": 1.377997716025885e-06, + "loss": 1.333, + "step": 2211 + }, + { + "epoch": 2.2722136620441704, + "grad_norm": 3.2522025108337402, + "learning_rate": 1.3760944042634184e-06, + "loss": 1.3496, + "step": 2212 + }, + { + "epoch": 2.273240883410375, + "grad_norm": 3.225821018218994, + "learning_rate": 1.3741910925009518e-06, + "loss": 1.3716, + "step": 2213 + }, + { + "epoch": 2.2742681047765796, + "grad_norm": 3.1488454341888428, + "learning_rate": 1.3722877807384852e-06, + "loss": 1.3508, + "step": 2214 + }, + { + "epoch": 2.2752953261427837, + "grad_norm": 3.258491039276123, + "learning_rate": 1.3703844689760185e-06, + "loss": 1.3565, + "step": 2215 + }, + { + "epoch": 2.2763225475089883, + "grad_norm": 3.1634750366210938, + "learning_rate": 1.3684811572135517e-06, + "loss": 1.3269, + "step": 2216 + }, + { + "epoch": 2.2773497688751925, + "grad_norm": 3.1604928970336914, + "learning_rate": 1.366577845451085e-06, + "loss": 1.2832, + "step": 2217 + }, + { + "epoch": 2.278376990241397, + "grad_norm": 3.2551920413970947, + "learning_rate": 1.3646745336886184e-06, + "loss": 1.3567, + "step": 2218 + }, + { + "epoch": 2.279404211607601, + "grad_norm": 3.392820119857788, + "learning_rate": 1.3627712219261515e-06, + "loss": 1.3601, + "step": 2219 + }, + { + "epoch": 2.280431432973806, + "grad_norm": 3.057724952697754, + "learning_rate": 1.360867910163685e-06, + "loss": 1.3138, + "step": 2220 + }, + { + "epoch": 2.280431432973806, + "eval_loss": 1.848382830619812, + "eval_runtime": 37.2255, + "eval_samples_per_second": 26.863, + "eval_steps_per_second": 1.128, + "step": 2220 + }, + { + "epoch": 2.2814586543400104, + "grad_norm": 3.2199649810791016, + "learning_rate": 1.358964598401218e-06, + "loss": 1.3338, + "step": 2221 + }, + { + "epoch": 2.2824858757062145, + "grad_norm": 3.252066135406494, + "learning_rate": 1.3570612866387514e-06, + "loss": 1.3413, + "step": 2222 + }, + { + "epoch": 2.283513097072419, + "grad_norm": 3.038052558898926, + "learning_rate": 1.355157974876285e-06, + "loss": 1.3268, + "step": 2223 + }, + { + "epoch": 2.2845403184386237, + "grad_norm": 3.2332992553710938, + "learning_rate": 1.3532546631138184e-06, + "loss": 1.3866, + "step": 2224 + }, + { + "epoch": 2.285567539804828, + "grad_norm": 3.14321231842041, + "learning_rate": 1.3513513513513515e-06, + "loss": 1.3466, + "step": 2225 + }, + { + "epoch": 2.2865947611710324, + "grad_norm": 3.0714094638824463, + "learning_rate": 1.3494480395888849e-06, + "loss": 1.331, + "step": 2226 + }, + { + "epoch": 2.2876219825372366, + "grad_norm": 3.1167569160461426, + "learning_rate": 1.347544727826418e-06, + "loss": 1.3242, + "step": 2227 + }, + { + "epoch": 2.288649203903441, + "grad_norm": 3.072951316833496, + "learning_rate": 1.3456414160639514e-06, + "loss": 1.3139, + "step": 2228 + }, + { + "epoch": 2.2896764252696458, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.3524, + "step": 2229 + }, + { + "epoch": 2.29070364663585, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.334, + "step": 2230 + }, + { + "epoch": 2.29070364663585, + "eval_loss": 1.848712682723999, + "eval_runtime": 37.2687, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2230 + }, + { + "epoch": 2.2917308680020545, + "grad_norm": 3.056426525115967, + "learning_rate": 1.341834792539018e-06, + "loss": 1.3344, + "step": 2231 + }, + { + "epoch": 2.2927580893682586, + "grad_norm": 3.2412500381469727, + "learning_rate": 1.3399314807765513e-06, + "loss": 1.3416, + "step": 2232 + }, + { + "epoch": 2.2937853107344632, + "grad_norm": 3.178572416305542, + "learning_rate": 1.3380281690140844e-06, + "loss": 1.3472, + "step": 2233 + }, + { + "epoch": 2.294812532100668, + "grad_norm": 3.2567203044891357, + "learning_rate": 1.336124857251618e-06, + "loss": 1.311, + "step": 2234 + }, + { + "epoch": 2.295839753466872, + "grad_norm": 3.2018561363220215, + "learning_rate": 1.3342215454891513e-06, + "loss": 1.301, + "step": 2235 + }, + { + "epoch": 2.2968669748330766, + "grad_norm": 3.0722200870513916, + "learning_rate": 1.3323182337266847e-06, + "loss": 1.3325, + "step": 2236 + }, + { + "epoch": 2.297894196199281, + "grad_norm": 3.2170779705047607, + "learning_rate": 1.3304149219642179e-06, + "loss": 1.3752, + "step": 2237 + }, + { + "epoch": 2.2989214175654853, + "grad_norm": 3.1309115886688232, + "learning_rate": 1.3285116102017512e-06, + "loss": 1.3261, + "step": 2238 + }, + { + "epoch": 2.29994863893169, + "grad_norm": 3.1463022232055664, + "learning_rate": 1.3266082984392844e-06, + "loss": 1.3316, + "step": 2239 + }, + { + "epoch": 2.300975860297894, + "grad_norm": 3.174701690673828, + "learning_rate": 1.3247049866768177e-06, + "loss": 1.3148, + "step": 2240 + }, + { + "epoch": 2.300975860297894, + "eval_loss": 1.8436999320983887, + "eval_runtime": 37.2577, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 2240 + }, + { + "epoch": 2.3020030816640986, + "grad_norm": 3.18149471282959, + "learning_rate": 1.322801674914351e-06, + "loss": 1.312, + "step": 2241 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.213361978530884, + "learning_rate": 1.3208983631518843e-06, + "loss": 1.3406, + "step": 2242 + }, + { + "epoch": 2.3040575243965074, + "grad_norm": 3.128354787826538, + "learning_rate": 1.3189950513894176e-06, + "loss": 1.3286, + "step": 2243 + }, + { + "epoch": 2.305084745762712, + "grad_norm": 3.210330009460449, + "learning_rate": 1.3170917396269512e-06, + "loss": 1.3406, + "step": 2244 + }, + { + "epoch": 2.306111967128916, + "grad_norm": 3.187161445617676, + "learning_rate": 1.3151884278644843e-06, + "loss": 1.3621, + "step": 2245 + }, + { + "epoch": 2.3071391884951207, + "grad_norm": 3.108806848526001, + "learning_rate": 1.3132851161020177e-06, + "loss": 1.3094, + "step": 2246 + }, + { + "epoch": 2.3081664098613253, + "grad_norm": 3.336510181427002, + "learning_rate": 1.311381804339551e-06, + "loss": 1.3998, + "step": 2247 + }, + { + "epoch": 2.3091936312275294, + "grad_norm": 3.159184455871582, + "learning_rate": 1.3094784925770842e-06, + "loss": 1.3082, + "step": 2248 + }, + { + "epoch": 2.310220852593734, + "grad_norm": 3.128321409225464, + "learning_rate": 1.3075751808146176e-06, + "loss": 1.3494, + "step": 2249 + }, + { + "epoch": 2.3112480739599386, + "grad_norm": 3.122690439224243, + "learning_rate": 1.3056718690521507e-06, + "loss": 1.3064, + "step": 2250 + }, + { + "epoch": 2.3112480739599386, + "eval_loss": 1.848455786705017, + "eval_runtime": 37.2647, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 2250 + }, + { + "epoch": 2.3122752953261427, + "grad_norm": 3.1642186641693115, + "learning_rate": 1.303768557289684e-06, + "loss": 1.3134, + "step": 2251 + }, + { + "epoch": 2.3133025166923473, + "grad_norm": 3.1710383892059326, + "learning_rate": 1.3018652455272175e-06, + "loss": 1.3126, + "step": 2252 + }, + { + "epoch": 2.3143297380585515, + "grad_norm": 3.1513214111328125, + "learning_rate": 1.2999619337647506e-06, + "loss": 1.3218, + "step": 2253 + }, + { + "epoch": 2.315356959424756, + "grad_norm": 3.1538784503936768, + "learning_rate": 1.2980586220022842e-06, + "loss": 1.3473, + "step": 2254 + }, + { + "epoch": 2.3163841807909606, + "grad_norm": 3.2409846782684326, + "learning_rate": 1.2961553102398175e-06, + "loss": 1.371, + "step": 2255 + }, + { + "epoch": 2.317411402157165, + "grad_norm": 3.3224987983703613, + "learning_rate": 1.2942519984773507e-06, + "loss": 1.3506, + "step": 2256 + }, + { + "epoch": 2.3184386235233694, + "grad_norm": 3.1684653759002686, + "learning_rate": 1.292348686714884e-06, + "loss": 1.3127, + "step": 2257 + }, + { + "epoch": 2.3194658448895735, + "grad_norm": 3.1784262657165527, + "learning_rate": 1.2904453749524174e-06, + "loss": 1.3541, + "step": 2258 + }, + { + "epoch": 2.320493066255778, + "grad_norm": 3.1121561527252197, + "learning_rate": 1.2885420631899506e-06, + "loss": 1.2843, + "step": 2259 + }, + { + "epoch": 2.3215202876219827, + "grad_norm": 3.200490951538086, + "learning_rate": 1.286638751427484e-06, + "loss": 1.3191, + "step": 2260 + }, + { + "epoch": 2.3215202876219827, + "eval_loss": 1.8469436168670654, + "eval_runtime": 37.2695, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2260 + }, + { + "epoch": 2.322547508988187, + "grad_norm": 3.143329620361328, + "learning_rate": 1.284735439665017e-06, + "loss": 1.3454, + "step": 2261 + }, + { + "epoch": 2.3235747303543914, + "grad_norm": 3.1955461502075195, + "learning_rate": 1.2828321279025504e-06, + "loss": 1.3505, + "step": 2262 + }, + { + "epoch": 2.324601951720596, + "grad_norm": 3.3297348022460938, + "learning_rate": 1.2809288161400838e-06, + "loss": 1.3374, + "step": 2263 + }, + { + "epoch": 2.3256291730868, + "grad_norm": 3.144385814666748, + "learning_rate": 1.279025504377617e-06, + "loss": 1.3198, + "step": 2264 + }, + { + "epoch": 2.3266563944530048, + "grad_norm": 3.2952210903167725, + "learning_rate": 1.2771221926151505e-06, + "loss": 1.3744, + "step": 2265 + }, + { + "epoch": 2.327683615819209, + "grad_norm": 3.212287187576294, + "learning_rate": 1.275218880852684e-06, + "loss": 1.3173, + "step": 2266 + }, + { + "epoch": 2.3287108371854135, + "grad_norm": 3.235272169113159, + "learning_rate": 1.273315569090217e-06, + "loss": 1.3231, + "step": 2267 + }, + { + "epoch": 2.3297380585516176, + "grad_norm": 3.2961113452911377, + "learning_rate": 1.2714122573277504e-06, + "loss": 1.3909, + "step": 2268 + }, + { + "epoch": 2.3307652799178222, + "grad_norm": 3.13678240776062, + "learning_rate": 1.2695089455652838e-06, + "loss": 1.3911, + "step": 2269 + }, + { + "epoch": 2.331792501284027, + "grad_norm": 3.166139841079712, + "learning_rate": 1.267605633802817e-06, + "loss": 1.3496, + "step": 2270 + }, + { + "epoch": 2.331792501284027, + "eval_loss": 1.8424044847488403, + "eval_runtime": 37.128, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 2270 + }, + { + "epoch": 2.332819722650231, + "grad_norm": 3.20859432220459, + "learning_rate": 1.2657023220403503e-06, + "loss": 1.3402, + "step": 2271 + }, + { + "epoch": 2.3338469440164356, + "grad_norm": 3.18941593170166, + "learning_rate": 1.2637990102778836e-06, + "loss": 1.3272, + "step": 2272 + }, + { + "epoch": 2.33487416538264, + "grad_norm": 3.2382333278656006, + "learning_rate": 1.2618956985154168e-06, + "loss": 1.3242, + "step": 2273 + }, + { + "epoch": 2.3359013867488443, + "grad_norm": 3.2496516704559326, + "learning_rate": 1.2599923867529502e-06, + "loss": 1.3675, + "step": 2274 + }, + { + "epoch": 2.336928608115049, + "grad_norm": 3.1282081604003906, + "learning_rate": 1.2580890749904837e-06, + "loss": 1.3343, + "step": 2275 + }, + { + "epoch": 2.337955829481253, + "grad_norm": 3.34047794342041, + "learning_rate": 1.2561857632280169e-06, + "loss": 1.3651, + "step": 2276 + }, + { + "epoch": 2.3389830508474576, + "grad_norm": 3.3369057178497314, + "learning_rate": 1.2542824514655502e-06, + "loss": 1.3944, + "step": 2277 + }, + { + "epoch": 2.340010272213662, + "grad_norm": 3.282226085662842, + "learning_rate": 1.2523791397030834e-06, + "loss": 1.3595, + "step": 2278 + }, + { + "epoch": 2.3410374935798663, + "grad_norm": 3.111595630645752, + "learning_rate": 1.2504758279406168e-06, + "loss": 1.3325, + "step": 2279 + }, + { + "epoch": 2.342064714946071, + "grad_norm": 3.306988477706909, + "learning_rate": 1.2485725161781501e-06, + "loss": 1.3595, + "step": 2280 + }, + { + "epoch": 2.342064714946071, + "eval_loss": 1.8465285301208496, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 2280 + }, + { + "epoch": 2.343091936312275, + "grad_norm": 3.4089088439941406, + "learning_rate": 1.2466692044156833e-06, + "loss": 1.4337, + "step": 2281 + }, + { + "epoch": 2.3441191576784797, + "grad_norm": 3.1354665756225586, + "learning_rate": 1.2447658926532168e-06, + "loss": 1.2996, + "step": 2282 + }, + { + "epoch": 2.3451463790446843, + "grad_norm": 3.067383289337158, + "learning_rate": 1.24286258089075e-06, + "loss": 1.3446, + "step": 2283 + }, + { + "epoch": 2.3461736004108884, + "grad_norm": 3.162349224090576, + "learning_rate": 1.2409592691282834e-06, + "loss": 1.3068, + "step": 2284 + }, + { + "epoch": 2.347200821777093, + "grad_norm": 3.2846035957336426, + "learning_rate": 1.2390559573658165e-06, + "loss": 1.3496, + "step": 2285 + }, + { + "epoch": 2.3482280431432976, + "grad_norm": 3.290997266769409, + "learning_rate": 1.2371526456033499e-06, + "loss": 1.3471, + "step": 2286 + }, + { + "epoch": 2.3492552645095017, + "grad_norm": 3.1327056884765625, + "learning_rate": 1.2352493338408832e-06, + "loss": 1.3108, + "step": 2287 + }, + { + "epoch": 2.3502824858757063, + "grad_norm": 3.295562744140625, + "learning_rate": 1.2333460220784166e-06, + "loss": 1.4226, + "step": 2288 + }, + { + "epoch": 2.3513097072419105, + "grad_norm": 3.3461313247680664, + "learning_rate": 1.23144271031595e-06, + "loss": 1.3494, + "step": 2289 + }, + { + "epoch": 2.352336928608115, + "grad_norm": 3.4178872108459473, + "learning_rate": 1.2295393985534831e-06, + "loss": 1.3876, + "step": 2290 + }, + { + "epoch": 2.352336928608115, + "eval_loss": 1.855657935142517, + "eval_runtime": 37.2052, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 2290 + }, + { + "epoch": 2.3533641499743196, + "grad_norm": 3.0909383296966553, + "learning_rate": 1.2276360867910165e-06, + "loss": 1.2625, + "step": 2291 + }, + { + "epoch": 2.354391371340524, + "grad_norm": 3.2940239906311035, + "learning_rate": 1.2257327750285498e-06, + "loss": 1.3836, + "step": 2292 + }, + { + "epoch": 2.3554185927067284, + "grad_norm": 3.166198253631592, + "learning_rate": 1.2238294632660832e-06, + "loss": 1.3014, + "step": 2293 + }, + { + "epoch": 2.3564458140729325, + "grad_norm": 3.1372265815734863, + "learning_rate": 1.2219261515036164e-06, + "loss": 1.3133, + "step": 2294 + }, + { + "epoch": 2.357473035439137, + "grad_norm": 3.2904369831085205, + "learning_rate": 1.2200228397411497e-06, + "loss": 1.3578, + "step": 2295 + }, + { + "epoch": 2.3585002568053417, + "grad_norm": 3.1268763542175293, + "learning_rate": 1.2181195279786829e-06, + "loss": 1.391, + "step": 2296 + }, + { + "epoch": 2.359527478171546, + "grad_norm": 3.1357498168945312, + "learning_rate": 1.2162162162162164e-06, + "loss": 1.327, + "step": 2297 + }, + { + "epoch": 2.3605546995377504, + "grad_norm": 3.1766469478607178, + "learning_rate": 1.2143129044537496e-06, + "loss": 1.3146, + "step": 2298 + }, + { + "epoch": 2.361581920903955, + "grad_norm": 3.1096513271331787, + "learning_rate": 1.212409592691283e-06, + "loss": 1.3235, + "step": 2299 + }, + { + "epoch": 2.362609142270159, + "grad_norm": 3.161182403564453, + "learning_rate": 1.2105062809288163e-06, + "loss": 1.3411, + "step": 2300 + }, + { + "epoch": 2.362609142270159, + "eval_loss": 1.852128267288208, + "eval_runtime": 37.4484, + "eval_samples_per_second": 26.703, + "eval_steps_per_second": 1.122, + "step": 2300 + }, + { + "epoch": 2.3636363636363638, + "grad_norm": 3.2012717723846436, + "learning_rate": 1.2086029691663495e-06, + "loss": 1.3425, + "step": 2301 + }, + { + "epoch": 2.364663585002568, + "grad_norm": 3.0783276557922363, + "learning_rate": 1.2066996574038828e-06, + "loss": 1.2886, + "step": 2302 + }, + { + "epoch": 2.3656908063687725, + "grad_norm": 3.2996575832366943, + "learning_rate": 1.2047963456414162e-06, + "loss": 1.36, + "step": 2303 + }, + { + "epoch": 2.366718027734977, + "grad_norm": 3.2035586833953857, + "learning_rate": 1.2028930338789496e-06, + "loss": 1.3454, + "step": 2304 + }, + { + "epoch": 2.3677452491011812, + "grad_norm": 3.0552821159362793, + "learning_rate": 1.2009897221164827e-06, + "loss": 1.2788, + "step": 2305 + }, + { + "epoch": 2.368772470467386, + "grad_norm": 3.2013018131256104, + "learning_rate": 1.199086410354016e-06, + "loss": 1.3467, + "step": 2306 + }, + { + "epoch": 2.36979969183359, + "grad_norm": 3.226489305496216, + "learning_rate": 1.1971830985915492e-06, + "loss": 1.3177, + "step": 2307 + }, + { + "epoch": 2.3708269131997946, + "grad_norm": 3.320812940597534, + "learning_rate": 1.1952797868290828e-06, + "loss": 1.3118, + "step": 2308 + }, + { + "epoch": 2.371854134565999, + "grad_norm": 3.223085641860962, + "learning_rate": 1.193376475066616e-06, + "loss": 1.3363, + "step": 2309 + }, + { + "epoch": 2.3728813559322033, + "grad_norm": 3.3175759315490723, + "learning_rate": 1.1914731633041493e-06, + "loss": 1.3476, + "step": 2310 + }, + { + "epoch": 2.3728813559322033, + "eval_loss": 1.855944037437439, + "eval_runtime": 36.9815, + "eval_samples_per_second": 27.041, + "eval_steps_per_second": 1.136, + "step": 2310 + }, + { + "epoch": 2.373908577298408, + "grad_norm": 3.1612298488616943, + "learning_rate": 1.1895698515416827e-06, + "loss": 1.3083, + "step": 2311 + }, + { + "epoch": 2.3749357986646125, + "grad_norm": 3.18595814704895, + "learning_rate": 1.1876665397792158e-06, + "loss": 1.3048, + "step": 2312 + }, + { + "epoch": 2.3759630200308166, + "grad_norm": 3.3328146934509277, + "learning_rate": 1.1857632280167492e-06, + "loss": 1.3496, + "step": 2313 + }, + { + "epoch": 2.376990241397021, + "grad_norm": 3.1661181449890137, + "learning_rate": 1.1838599162542825e-06, + "loss": 1.284, + "step": 2314 + }, + { + "epoch": 2.3780174627632253, + "grad_norm": 3.298358201980591, + "learning_rate": 1.181956604491816e-06, + "loss": 1.359, + "step": 2315 + }, + { + "epoch": 2.37904468412943, + "grad_norm": 3.24114727973938, + "learning_rate": 1.180053292729349e-06, + "loss": 1.3264, + "step": 2316 + }, + { + "epoch": 2.380071905495634, + "grad_norm": 3.4791877269744873, + "learning_rate": 1.1781499809668824e-06, + "loss": 1.4022, + "step": 2317 + }, + { + "epoch": 2.3810991268618387, + "grad_norm": 3.366711139678955, + "learning_rate": 1.1762466692044158e-06, + "loss": 1.3341, + "step": 2318 + }, + { + "epoch": 2.3821263482280433, + "grad_norm": 3.3023927211761475, + "learning_rate": 1.1743433574419491e-06, + "loss": 1.3693, + "step": 2319 + }, + { + "epoch": 2.3831535695942474, + "grad_norm": 3.175189256668091, + "learning_rate": 1.1724400456794823e-06, + "loss": 1.3068, + "step": 2320 + }, + { + "epoch": 2.3831535695942474, + "eval_loss": 1.8537958860397339, + "eval_runtime": 37.079, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 2320 + }, + { + "epoch": 2.384180790960452, + "grad_norm": 3.326911211013794, + "learning_rate": 1.1705367339170157e-06, + "loss": 1.3565, + "step": 2321 + }, + { + "epoch": 2.3852080123266566, + "grad_norm": 3.4324638843536377, + "learning_rate": 1.168633422154549e-06, + "loss": 1.3412, + "step": 2322 + }, + { + "epoch": 2.3862352336928607, + "grad_norm": 3.170397996902466, + "learning_rate": 1.1667301103920824e-06, + "loss": 1.3, + "step": 2323 + }, + { + "epoch": 2.3872624550590653, + "grad_norm": 3.1137685775756836, + "learning_rate": 1.1648267986296157e-06, + "loss": 1.2722, + "step": 2324 + }, + { + "epoch": 2.3882896764252695, + "grad_norm": 3.2298166751861572, + "learning_rate": 1.162923486867149e-06, + "loss": 1.326, + "step": 2325 + }, + { + "epoch": 2.389316897791474, + "grad_norm": 3.2462759017944336, + "learning_rate": 1.1610201751046823e-06, + "loss": 1.344, + "step": 2326 + }, + { + "epoch": 2.3903441191576786, + "grad_norm": 3.3631975650787354, + "learning_rate": 1.1591168633422154e-06, + "loss": 1.3909, + "step": 2327 + }, + { + "epoch": 2.391371340523883, + "grad_norm": 3.279407024383545, + "learning_rate": 1.157213551579749e-06, + "loss": 1.3544, + "step": 2328 + }, + { + "epoch": 2.3923985618900874, + "grad_norm": 3.1387593746185303, + "learning_rate": 1.1553102398172821e-06, + "loss": 1.33, + "step": 2329 + }, + { + "epoch": 2.3934257832562915, + "grad_norm": 3.218921661376953, + "learning_rate": 1.1534069280548155e-06, + "loss": 1.3117, + "step": 2330 + }, + { + "epoch": 2.3934257832562915, + "eval_loss": 1.8487199544906616, + "eval_runtime": 37.1303, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 2330 + }, + { + "epoch": 2.394453004622496, + "grad_norm": 3.221266746520996, + "learning_rate": 1.1515036162923487e-06, + "loss": 1.3521, + "step": 2331 + }, + { + "epoch": 2.3954802259887007, + "grad_norm": 3.112572431564331, + "learning_rate": 1.149600304529882e-06, + "loss": 1.3161, + "step": 2332 + }, + { + "epoch": 2.396507447354905, + "grad_norm": 3.2089130878448486, + "learning_rate": 1.1476969927674154e-06, + "loss": 1.3236, + "step": 2333 + }, + { + "epoch": 2.3975346687211094, + "grad_norm": 2.985661029815674, + "learning_rate": 1.1457936810049487e-06, + "loss": 1.3295, + "step": 2334 + }, + { + "epoch": 2.398561890087314, + "grad_norm": 3.189842939376831, + "learning_rate": 1.143890369242482e-06, + "loss": 1.3685, + "step": 2335 + }, + { + "epoch": 2.399589111453518, + "grad_norm": 3.124861478805542, + "learning_rate": 1.1419870574800153e-06, + "loss": 1.293, + "step": 2336 + }, + { + "epoch": 2.4006163328197228, + "grad_norm": 3.064199209213257, + "learning_rate": 1.1400837457175486e-06, + "loss": 1.3229, + "step": 2337 + }, + { + "epoch": 2.401643554185927, + "grad_norm": 3.1462392807006836, + "learning_rate": 1.138180433955082e-06, + "loss": 1.2931, + "step": 2338 + }, + { + "epoch": 2.4026707755521315, + "grad_norm": 3.192121982574463, + "learning_rate": 1.1362771221926153e-06, + "loss": 1.3569, + "step": 2339 + }, + { + "epoch": 2.403697996918336, + "grad_norm": 3.2683119773864746, + "learning_rate": 1.1343738104301485e-06, + "loss": 1.3437, + "step": 2340 + }, + { + "epoch": 2.403697996918336, + "eval_loss": 1.8514460325241089, + "eval_runtime": 37.185, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.129, + "step": 2340 + }, + { + "epoch": 2.4047252182845402, + "grad_norm": 3.168116331100464, + "learning_rate": 1.1324704986676819e-06, + "loss": 1.3177, + "step": 2341 + }, + { + "epoch": 2.405752439650745, + "grad_norm": 3.291511058807373, + "learning_rate": 1.130567186905215e-06, + "loss": 1.3731, + "step": 2342 + }, + { + "epoch": 2.406779661016949, + "grad_norm": 3.0953369140625, + "learning_rate": 1.1286638751427486e-06, + "loss": 1.3231, + "step": 2343 + }, + { + "epoch": 2.4078068823831535, + "grad_norm": 3.1431541442871094, + "learning_rate": 1.1267605633802817e-06, + "loss": 1.3227, + "step": 2344 + }, + { + "epoch": 2.408834103749358, + "grad_norm": 3.2409775257110596, + "learning_rate": 1.124857251617815e-06, + "loss": 1.3603, + "step": 2345 + }, + { + "epoch": 2.4098613251155623, + "grad_norm": 3.240900993347168, + "learning_rate": 1.1229539398553485e-06, + "loss": 1.3912, + "step": 2346 + }, + { + "epoch": 2.410888546481767, + "grad_norm": 3.1512467861175537, + "learning_rate": 1.1210506280928816e-06, + "loss": 1.3211, + "step": 2347 + }, + { + "epoch": 2.4119157678479715, + "grad_norm": 3.2743258476257324, + "learning_rate": 1.119147316330415e-06, + "loss": 1.3704, + "step": 2348 + }, + { + "epoch": 2.4129429892141756, + "grad_norm": 3.381438732147217, + "learning_rate": 1.1172440045679483e-06, + "loss": 1.335, + "step": 2349 + }, + { + "epoch": 2.41397021058038, + "grad_norm": 3.19822096824646, + "learning_rate": 1.1153406928054817e-06, + "loss": 1.3355, + "step": 2350 + }, + { + "epoch": 2.41397021058038, + "eval_loss": 1.8520244359970093, + "eval_runtime": 37.1581, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2350 + }, + { + "epoch": 2.4149974319465843, + "grad_norm": 3.3650894165039062, + "learning_rate": 1.1134373810430148e-06, + "loss": 1.3576, + "step": 2351 + }, + { + "epoch": 2.416024653312789, + "grad_norm": 3.3263492584228516, + "learning_rate": 1.1115340692805482e-06, + "loss": 1.3715, + "step": 2352 + }, + { + "epoch": 2.4170518746789935, + "grad_norm": 3.280782699584961, + "learning_rate": 1.1096307575180816e-06, + "loss": 1.3282, + "step": 2353 + }, + { + "epoch": 2.4180790960451977, + "grad_norm": 3.1976306438446045, + "learning_rate": 1.107727445755615e-06, + "loss": 1.3013, + "step": 2354 + }, + { + "epoch": 2.4191063174114023, + "grad_norm": 3.1435351371765137, + "learning_rate": 1.105824133993148e-06, + "loss": 1.3163, + "step": 2355 + }, + { + "epoch": 2.4201335387776064, + "grad_norm": 3.2856905460357666, + "learning_rate": 1.1039208222306814e-06, + "loss": 1.3378, + "step": 2356 + }, + { + "epoch": 2.421160760143811, + "grad_norm": 3.241748809814453, + "learning_rate": 1.1020175104682148e-06, + "loss": 1.3285, + "step": 2357 + }, + { + "epoch": 2.4221879815100156, + "grad_norm": 3.1425986289978027, + "learning_rate": 1.1001141987057482e-06, + "loss": 1.3155, + "step": 2358 + }, + { + "epoch": 2.4232152028762197, + "grad_norm": 3.311344623565674, + "learning_rate": 1.0982108869432813e-06, + "loss": 1.3291, + "step": 2359 + }, + { + "epoch": 2.4242424242424243, + "grad_norm": 3.2412517070770264, + "learning_rate": 1.0963075751808147e-06, + "loss": 1.3655, + "step": 2360 + }, + { + "epoch": 2.4242424242424243, + "eval_loss": 1.853851318359375, + "eval_runtime": 37.1577, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2360 + }, + { + "epoch": 2.425269645608629, + "grad_norm": 3.166773796081543, + "learning_rate": 1.094404263418348e-06, + "loss": 1.3316, + "step": 2361 + }, + { + "epoch": 2.426296866974833, + "grad_norm": 3.1892364025115967, + "learning_rate": 1.0925009516558812e-06, + "loss": 1.3304, + "step": 2362 + }, + { + "epoch": 2.4273240883410376, + "grad_norm": 3.129615306854248, + "learning_rate": 1.0905976398934148e-06, + "loss": 1.3211, + "step": 2363 + }, + { + "epoch": 2.428351309707242, + "grad_norm": 3.2534453868865967, + "learning_rate": 1.088694328130948e-06, + "loss": 1.3387, + "step": 2364 + }, + { + "epoch": 2.4293785310734464, + "grad_norm": 3.071871757507324, + "learning_rate": 1.0867910163684813e-06, + "loss": 1.2949, + "step": 2365 + }, + { + "epoch": 2.4304057524396505, + "grad_norm": 3.2920479774475098, + "learning_rate": 1.0848877046060144e-06, + "loss": 1.3345, + "step": 2366 + }, + { + "epoch": 2.431432973805855, + "grad_norm": 3.2313623428344727, + "learning_rate": 1.0829843928435478e-06, + "loss": 1.3308, + "step": 2367 + }, + { + "epoch": 2.4324601951720597, + "grad_norm": 3.1129872798919678, + "learning_rate": 1.0810810810810812e-06, + "loss": 1.3333, + "step": 2368 + }, + { + "epoch": 2.433487416538264, + "grad_norm": 3.278503894805908, + "learning_rate": 1.0791777693186145e-06, + "loss": 1.3458, + "step": 2369 + }, + { + "epoch": 2.4345146379044684, + "grad_norm": 3.390359401702881, + "learning_rate": 1.0772744575561479e-06, + "loss": 1.3262, + "step": 2370 + }, + { + "epoch": 2.4345146379044684, + "eval_loss": 1.8534605503082275, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 2370 + }, + { + "epoch": 2.435541859270673, + "grad_norm": 3.2505109310150146, + "learning_rate": 1.075371145793681e-06, + "loss": 1.3144, + "step": 2371 + }, + { + "epoch": 2.436569080636877, + "grad_norm": 3.1556410789489746, + "learning_rate": 1.0734678340312144e-06, + "loss": 1.3213, + "step": 2372 + }, + { + "epoch": 2.4375963020030817, + "grad_norm": 3.1231777667999268, + "learning_rate": 1.0715645222687478e-06, + "loss": 1.3124, + "step": 2373 + }, + { + "epoch": 2.438623523369286, + "grad_norm": 3.1159915924072266, + "learning_rate": 1.0696612105062811e-06, + "loss": 1.3191, + "step": 2374 + }, + { + "epoch": 2.4396507447354905, + "grad_norm": 3.2663285732269287, + "learning_rate": 1.0677578987438143e-06, + "loss": 1.3491, + "step": 2375 + }, + { + "epoch": 2.440677966101695, + "grad_norm": 3.0827524662017822, + "learning_rate": 1.0658545869813476e-06, + "loss": 1.3057, + "step": 2376 + }, + { + "epoch": 2.441705187467899, + "grad_norm": 3.1982173919677734, + "learning_rate": 1.0639512752188808e-06, + "loss": 1.312, + "step": 2377 + }, + { + "epoch": 2.442732408834104, + "grad_norm": 3.078274965286255, + "learning_rate": 1.0620479634564144e-06, + "loss": 1.3261, + "step": 2378 + }, + { + "epoch": 2.443759630200308, + "grad_norm": 3.0313515663146973, + "learning_rate": 1.0601446516939475e-06, + "loss": 1.2953, + "step": 2379 + }, + { + "epoch": 2.4447868515665125, + "grad_norm": 3.1637635231018066, + "learning_rate": 1.0582413399314809e-06, + "loss": 1.2973, + "step": 2380 + }, + { + "epoch": 2.4447868515665125, + "eval_loss": 1.8524926900863647, + "eval_runtime": 37.1576, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2380 + }, + { + "epoch": 2.445814072932717, + "grad_norm": 3.2800862789154053, + "learning_rate": 1.0563380281690142e-06, + "loss": 1.3048, + "step": 2381 + }, + { + "epoch": 2.4468412942989213, + "grad_norm": 3.113825559616089, + "learning_rate": 1.0544347164065474e-06, + "loss": 1.3081, + "step": 2382 + }, + { + "epoch": 2.447868515665126, + "grad_norm": 3.201627254486084, + "learning_rate": 1.0525314046440808e-06, + "loss": 1.3362, + "step": 2383 + }, + { + "epoch": 2.4488957370313305, + "grad_norm": 3.1768431663513184, + "learning_rate": 1.0506280928816141e-06, + "loss": 1.3135, + "step": 2384 + }, + { + "epoch": 2.4499229583975346, + "grad_norm": 3.2254347801208496, + "learning_rate": 1.0487247811191475e-06, + "loss": 1.3168, + "step": 2385 + }, + { + "epoch": 2.450950179763739, + "grad_norm": 3.2468769550323486, + "learning_rate": 1.0468214693566806e-06, + "loss": 1.3301, + "step": 2386 + }, + { + "epoch": 2.4519774011299433, + "grad_norm": 3.214177370071411, + "learning_rate": 1.044918157594214e-06, + "loss": 1.3481, + "step": 2387 + }, + { + "epoch": 2.453004622496148, + "grad_norm": 3.1789238452911377, + "learning_rate": 1.0430148458317474e-06, + "loss": 1.3065, + "step": 2388 + }, + { + "epoch": 2.4540318438623525, + "grad_norm": 3.082336187362671, + "learning_rate": 1.0411115340692807e-06, + "loss": 1.2923, + "step": 2389 + }, + { + "epoch": 2.4550590652285567, + "grad_norm": 3.305126905441284, + "learning_rate": 1.0392082223068139e-06, + "loss": 1.3777, + "step": 2390 + }, + { + "epoch": 2.4550590652285567, + "eval_loss": 1.8477189540863037, + "eval_runtime": 37.3455, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 2390 + }, + { + "epoch": 2.4560862865947612, + "grad_norm": 3.322448492050171, + "learning_rate": 1.0373049105443472e-06, + "loss": 1.3309, + "step": 2391 + }, + { + "epoch": 2.4571135079609654, + "grad_norm": 3.2111802101135254, + "learning_rate": 1.0354015987818806e-06, + "loss": 1.3376, + "step": 2392 + }, + { + "epoch": 2.45814072932717, + "grad_norm": 3.0796451568603516, + "learning_rate": 1.033498287019414e-06, + "loss": 1.3093, + "step": 2393 + }, + { + "epoch": 2.4591679506933746, + "grad_norm": 3.265315055847168, + "learning_rate": 1.0315949752569471e-06, + "loss": 1.3652, + "step": 2394 + }, + { + "epoch": 2.4601951720595787, + "grad_norm": 3.1291234493255615, + "learning_rate": 1.0296916634944805e-06, + "loss": 1.2947, + "step": 2395 + }, + { + "epoch": 2.4612223934257833, + "grad_norm": 3.314732789993286, + "learning_rate": 1.0277883517320138e-06, + "loss": 1.3356, + "step": 2396 + }, + { + "epoch": 2.462249614791988, + "grad_norm": 3.2572879791259766, + "learning_rate": 1.025885039969547e-06, + "loss": 1.3665, + "step": 2397 + }, + { + "epoch": 2.463276836158192, + "grad_norm": 3.065584897994995, + "learning_rate": 1.0239817282070806e-06, + "loss": 1.2614, + "step": 2398 + }, + { + "epoch": 2.4643040575243966, + "grad_norm": 3.2398078441619873, + "learning_rate": 1.0220784164446137e-06, + "loss": 1.2986, + "step": 2399 + }, + { + "epoch": 2.4653312788906008, + "grad_norm": 3.1384856700897217, + "learning_rate": 1.020175104682147e-06, + "loss": 1.3157, + "step": 2400 + }, + { + "epoch": 2.4653312788906008, + "eval_loss": 1.8481135368347168, + "eval_runtime": 37.4463, + "eval_samples_per_second": 26.705, + "eval_steps_per_second": 1.122, + "step": 2400 + }, + { + "epoch": 2.4663585002568054, + "grad_norm": 3.2159786224365234, + "learning_rate": 1.0182717929196802e-06, + "loss": 1.3289, + "step": 2401 + }, + { + "epoch": 2.4673857216230095, + "grad_norm": 3.353900909423828, + "learning_rate": 1.0163684811572136e-06, + "loss": 1.3565, + "step": 2402 + }, + { + "epoch": 2.468412942989214, + "grad_norm": 3.1520562171936035, + "learning_rate": 1.014465169394747e-06, + "loss": 1.2855, + "step": 2403 + }, + { + "epoch": 2.4694401643554187, + "grad_norm": 3.2022621631622314, + "learning_rate": 1.0125618576322803e-06, + "loss": 1.3257, + "step": 2404 + }, + { + "epoch": 2.470467385721623, + "grad_norm": 3.1945600509643555, + "learning_rate": 1.0106585458698137e-06, + "loss": 1.3101, + "step": 2405 + }, + { + "epoch": 2.4714946070878274, + "grad_norm": 3.234997510910034, + "learning_rate": 1.0087552341073468e-06, + "loss": 1.3037, + "step": 2406 + }, + { + "epoch": 2.472521828454032, + "grad_norm": 3.3020741939544678, + "learning_rate": 1.0068519223448802e-06, + "loss": 1.3819, + "step": 2407 + }, + { + "epoch": 2.473549049820236, + "grad_norm": 3.4102628231048584, + "learning_rate": 1.0049486105824133e-06, + "loss": 1.3646, + "step": 2408 + }, + { + "epoch": 2.4745762711864407, + "grad_norm": 3.191983222961426, + "learning_rate": 1.003045298819947e-06, + "loss": 1.3036, + "step": 2409 + }, + { + "epoch": 2.4756034925526453, + "grad_norm": 3.3082430362701416, + "learning_rate": 1.00114198705748e-06, + "loss": 1.2973, + "step": 2410 + }, + { + "epoch": 2.4756034925526453, + "eval_loss": 1.8559728860855103, + "eval_runtime": 36.9665, + "eval_samples_per_second": 27.052, + "eval_steps_per_second": 1.136, + "step": 2410 + }, + { + "epoch": 2.4766307139188495, + "grad_norm": 3.3207387924194336, + "learning_rate": 9.992386752950134e-07, + "loss": 1.302, + "step": 2411 + }, + { + "epoch": 2.477657935285054, + "grad_norm": 3.2640819549560547, + "learning_rate": 9.973353635325466e-07, + "loss": 1.2963, + "step": 2412 + }, + { + "epoch": 2.478685156651258, + "grad_norm": 3.3404572010040283, + "learning_rate": 9.9543205177008e-07, + "loss": 1.2979, + "step": 2413 + }, + { + "epoch": 2.479712378017463, + "grad_norm": 3.2177815437316895, + "learning_rate": 9.935287400076133e-07, + "loss": 1.3022, + "step": 2414 + }, + { + "epoch": 2.480739599383667, + "grad_norm": 3.1873295307159424, + "learning_rate": 9.916254282451467e-07, + "loss": 1.2808, + "step": 2415 + }, + { + "epoch": 2.4817668207498715, + "grad_norm": 3.3681764602661133, + "learning_rate": 9.8972211648268e-07, + "loss": 1.3237, + "step": 2416 + }, + { + "epoch": 2.482794042116076, + "grad_norm": 3.2793691158294678, + "learning_rate": 9.878188047202132e-07, + "loss": 1.3437, + "step": 2417 + }, + { + "epoch": 2.4838212634822803, + "grad_norm": 3.1983675956726074, + "learning_rate": 9.859154929577465e-07, + "loss": 1.3211, + "step": 2418 + }, + { + "epoch": 2.484848484848485, + "grad_norm": 3.121095895767212, + "learning_rate": 9.8401218119528e-07, + "loss": 1.2957, + "step": 2419 + }, + { + "epoch": 2.4858757062146895, + "grad_norm": 3.3415417671203613, + "learning_rate": 9.821088694328133e-07, + "loss": 1.3116, + "step": 2420 + }, + { + "epoch": 2.4858757062146895, + "eval_loss": 1.8509575128555298, + "eval_runtime": 37.0683, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 2420 + }, + { + "epoch": 2.4869029275808936, + "grad_norm": 3.196071147918701, + "learning_rate": 9.802055576703464e-07, + "loss": 1.3574, + "step": 2421 + }, + { + "epoch": 2.487930148947098, + "grad_norm": 3.260653018951416, + "learning_rate": 9.783022459078798e-07, + "loss": 1.3297, + "step": 2422 + }, + { + "epoch": 2.4889573703133023, + "grad_norm": 3.1694345474243164, + "learning_rate": 9.76398934145413e-07, + "loss": 1.3266, + "step": 2423 + }, + { + "epoch": 2.489984591679507, + "grad_norm": 3.227717161178589, + "learning_rate": 9.744956223829465e-07, + "loss": 1.3408, + "step": 2424 + }, + { + "epoch": 2.4910118130457115, + "grad_norm": 3.2542760372161865, + "learning_rate": 9.725923106204797e-07, + "loss": 1.3524, + "step": 2425 + }, + { + "epoch": 2.4920390344119157, + "grad_norm": 3.308953285217285, + "learning_rate": 9.70688998858013e-07, + "loss": 1.3758, + "step": 2426 + }, + { + "epoch": 2.4930662557781202, + "grad_norm": 3.467383623123169, + "learning_rate": 9.687856870955464e-07, + "loss": 1.3549, + "step": 2427 + }, + { + "epoch": 2.4940934771443244, + "grad_norm": 3.2643022537231445, + "learning_rate": 9.668823753330795e-07, + "loss": 1.3438, + "step": 2428 + }, + { + "epoch": 2.495120698510529, + "grad_norm": 3.142022132873535, + "learning_rate": 9.64979063570613e-07, + "loss": 1.2852, + "step": 2429 + }, + { + "epoch": 2.4961479198767336, + "grad_norm": 3.220409393310547, + "learning_rate": 9.630757518081463e-07, + "loss": 1.3315, + "step": 2430 + }, + { + "epoch": 2.4961479198767336, + "eval_loss": 1.8497542142868042, + "eval_runtime": 37.1119, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 2430 + }, + { + "epoch": 2.4971751412429377, + "grad_norm": 3.296198844909668, + "learning_rate": 9.611724400456796e-07, + "loss": 1.3241, + "step": 2431 + }, + { + "epoch": 2.4982023626091423, + "grad_norm": 3.197559356689453, + "learning_rate": 9.592691282832128e-07, + "loss": 1.2976, + "step": 2432 + }, + { + "epoch": 2.499229583975347, + "grad_norm": 3.261249542236328, + "learning_rate": 9.573658165207461e-07, + "loss": 1.3264, + "step": 2433 + }, + { + "epoch": 2.500256805341551, + "grad_norm": 3.235084295272827, + "learning_rate": 9.554625047582795e-07, + "loss": 1.3465, + "step": 2434 + }, + { + "epoch": 2.5012840267077556, + "grad_norm": 3.1297430992126465, + "learning_rate": 9.535591929958129e-07, + "loss": 1.299, + "step": 2435 + }, + { + "epoch": 2.50231124807396, + "grad_norm": 3.2298195362091064, + "learning_rate": 9.516558812333461e-07, + "loss": 1.3374, + "step": 2436 + }, + { + "epoch": 2.5033384694401644, + "grad_norm": 3.2262182235717773, + "learning_rate": 9.497525694708794e-07, + "loss": 1.3432, + "step": 2437 + }, + { + "epoch": 2.5043656908063685, + "grad_norm": 3.227480888366699, + "learning_rate": 9.478492577084126e-07, + "loss": 1.3076, + "step": 2438 + }, + { + "epoch": 2.505392912172573, + "grad_norm": 3.0988104343414307, + "learning_rate": 9.459459459459461e-07, + "loss": 1.3167, + "step": 2439 + }, + { + "epoch": 2.5064201335387777, + "grad_norm": 3.2276222705841064, + "learning_rate": 9.440426341834794e-07, + "loss": 1.3069, + "step": 2440 + }, + { + "epoch": 2.5064201335387777, + "eval_loss": 1.8424607515335083, + "eval_runtime": 37.1264, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 2440 + }, + { + "epoch": 2.507447354904982, + "grad_norm": 3.1694467067718506, + "learning_rate": 9.421393224210126e-07, + "loss": 1.3194, + "step": 2441 + }, + { + "epoch": 2.5084745762711864, + "grad_norm": 3.2202742099761963, + "learning_rate": 9.402360106585459e-07, + "loss": 1.3288, + "step": 2442 + }, + { + "epoch": 2.509501797637391, + "grad_norm": 3.156813621520996, + "learning_rate": 9.383326988960792e-07, + "loss": 1.2853, + "step": 2443 + }, + { + "epoch": 2.510529019003595, + "grad_norm": 3.199050188064575, + "learning_rate": 9.364293871336126e-07, + "loss": 1.2818, + "step": 2444 + }, + { + "epoch": 2.5115562403697997, + "grad_norm": 3.1687068939208984, + "learning_rate": 9.345260753711458e-07, + "loss": 1.3115, + "step": 2445 + }, + { + "epoch": 2.5125834617360043, + "grad_norm": 3.224169969558716, + "learning_rate": 9.326227636086792e-07, + "loss": 1.3357, + "step": 2446 + }, + { + "epoch": 2.5136106831022085, + "grad_norm": 3.1867403984069824, + "learning_rate": 9.307194518462125e-07, + "loss": 1.277, + "step": 2447 + }, + { + "epoch": 2.514637904468413, + "grad_norm": 3.4736502170562744, + "learning_rate": 9.288161400837457e-07, + "loss": 1.3972, + "step": 2448 + }, + { + "epoch": 2.515665125834617, + "grad_norm": 3.147759437561035, + "learning_rate": 9.269128283212792e-07, + "loss": 1.287, + "step": 2449 + }, + { + "epoch": 2.516692347200822, + "grad_norm": 3.1714866161346436, + "learning_rate": 9.250095165588125e-07, + "loss": 1.3214, + "step": 2450 + }, + { + "epoch": 2.516692347200822, + "eval_loss": 1.8418748378753662, + "eval_runtime": 37.1781, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 2450 + }, + { + "epoch": 2.517719568567026, + "grad_norm": 3.308077573776245, + "learning_rate": 9.231062047963457e-07, + "loss": 1.3745, + "step": 2451 + }, + { + "epoch": 2.5187467899332305, + "grad_norm": 3.318955421447754, + "learning_rate": 9.21202893033879e-07, + "loss": 1.3767, + "step": 2452 + }, + { + "epoch": 2.519774011299435, + "grad_norm": 3.236506462097168, + "learning_rate": 9.192995812714123e-07, + "loss": 1.3311, + "step": 2453 + }, + { + "epoch": 2.5208012326656393, + "grad_norm": 3.3041205406188965, + "learning_rate": 9.173962695089457e-07, + "loss": 1.3243, + "step": 2454 + }, + { + "epoch": 2.521828454031844, + "grad_norm": 3.1637344360351562, + "learning_rate": 9.154929577464789e-07, + "loss": 1.3216, + "step": 2455 + }, + { + "epoch": 2.5228556753980484, + "grad_norm": 3.151911735534668, + "learning_rate": 9.135896459840123e-07, + "loss": 1.3494, + "step": 2456 + }, + { + "epoch": 2.5238828967642526, + "grad_norm": 3.335695743560791, + "learning_rate": 9.116863342215456e-07, + "loss": 1.3225, + "step": 2457 + }, + { + "epoch": 2.524910118130457, + "grad_norm": 3.2160415649414062, + "learning_rate": 9.097830224590788e-07, + "loss": 1.3693, + "step": 2458 + }, + { + "epoch": 2.5259373394966618, + "grad_norm": 3.0803115367889404, + "learning_rate": 9.078797106966122e-07, + "loss": 1.2844, + "step": 2459 + }, + { + "epoch": 2.526964560862866, + "grad_norm": 3.2147631645202637, + "learning_rate": 9.059763989341455e-07, + "loss": 1.3602, + "step": 2460 + }, + { + "epoch": 2.526964560862866, + "eval_loss": 1.8384426832199097, + "eval_runtime": 37.1832, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 2460 + }, + { + "epoch": 2.5279917822290705, + "grad_norm": 3.198880672454834, + "learning_rate": 9.040730871716788e-07, + "loss": 1.2999, + "step": 2461 + }, + { + "epoch": 2.5290190035952747, + "grad_norm": 3.095944404602051, + "learning_rate": 9.021697754092121e-07, + "loss": 1.324, + "step": 2462 + }, + { + "epoch": 2.5300462249614792, + "grad_norm": 3.150750160217285, + "learning_rate": 9.002664636467453e-07, + "loss": 1.3259, + "step": 2463 + }, + { + "epoch": 2.5310734463276834, + "grad_norm": 3.1914174556732178, + "learning_rate": 8.983631518842788e-07, + "loss": 1.3401, + "step": 2464 + }, + { + "epoch": 2.532100667693888, + "grad_norm": 3.2229790687561035, + "learning_rate": 8.96459840121812e-07, + "loss": 1.3624, + "step": 2465 + }, + { + "epoch": 2.5331278890600926, + "grad_norm": 3.161130666732788, + "learning_rate": 8.945565283593453e-07, + "loss": 1.3183, + "step": 2466 + }, + { + "epoch": 2.5341551104262967, + "grad_norm": 3.0963833332061768, + "learning_rate": 8.926532165968787e-07, + "loss": 1.3446, + "step": 2467 + }, + { + "epoch": 2.5351823317925013, + "grad_norm": 3.324404001235962, + "learning_rate": 8.907499048344119e-07, + "loss": 1.3394, + "step": 2468 + }, + { + "epoch": 2.536209553158706, + "grad_norm": 3.122046709060669, + "learning_rate": 8.888465930719453e-07, + "loss": 1.2868, + "step": 2469 + }, + { + "epoch": 2.53723677452491, + "grad_norm": 3.1900618076324463, + "learning_rate": 8.869432813094786e-07, + "loss": 1.2872, + "step": 2470 + }, + { + "epoch": 2.53723677452491, + "eval_loss": 1.8466377258300781, + "eval_runtime": 37.1472, + "eval_samples_per_second": 26.92, + "eval_steps_per_second": 1.131, + "step": 2470 + }, + { + "epoch": 2.5382639958911146, + "grad_norm": 3.353994369506836, + "learning_rate": 8.850399695470119e-07, + "loss": 1.3304, + "step": 2471 + }, + { + "epoch": 2.539291217257319, + "grad_norm": 3.3058688640594482, + "learning_rate": 8.831366577845452e-07, + "loss": 1.344, + "step": 2472 + }, + { + "epoch": 2.5403184386235234, + "grad_norm": 3.3035926818847656, + "learning_rate": 8.812333460220784e-07, + "loss": 1.3029, + "step": 2473 + }, + { + "epoch": 2.541345659989728, + "grad_norm": 3.173894166946411, + "learning_rate": 8.793300342596119e-07, + "loss": 1.2742, + "step": 2474 + }, + { + "epoch": 2.542372881355932, + "grad_norm": 3.1931140422821045, + "learning_rate": 8.774267224971451e-07, + "loss": 1.3369, + "step": 2475 + }, + { + "epoch": 2.5434001027221367, + "grad_norm": 3.1834166049957275, + "learning_rate": 8.755234107346784e-07, + "loss": 1.3605, + "step": 2476 + }, + { + "epoch": 2.544427324088341, + "grad_norm": 3.244100332260132, + "learning_rate": 8.736200989722117e-07, + "loss": 1.2971, + "step": 2477 + }, + { + "epoch": 2.5454545454545454, + "grad_norm": 3.223937749862671, + "learning_rate": 8.71716787209745e-07, + "loss": 1.3174, + "step": 2478 + }, + { + "epoch": 2.54648176682075, + "grad_norm": 3.1803457736968994, + "learning_rate": 8.698134754472784e-07, + "loss": 1.3486, + "step": 2479 + }, + { + "epoch": 2.547508988186954, + "grad_norm": 2.991408109664917, + "learning_rate": 8.679101636848116e-07, + "loss": 1.2632, + "step": 2480 + }, + { + "epoch": 2.547508988186954, + "eval_loss": 1.8453017473220825, + "eval_runtime": 37.1608, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 2480 + }, + { + "epoch": 2.5485362095531587, + "grad_norm": 3.2398955821990967, + "learning_rate": 8.66006851922345e-07, + "loss": 1.3418, + "step": 2481 + }, + { + "epoch": 2.5495634309193633, + "grad_norm": 3.0708181858062744, + "learning_rate": 8.641035401598783e-07, + "loss": 1.324, + "step": 2482 + }, + { + "epoch": 2.5505906522855675, + "grad_norm": 3.213524103164673, + "learning_rate": 8.622002283974115e-07, + "loss": 1.3071, + "step": 2483 + }, + { + "epoch": 2.551617873651772, + "grad_norm": 3.27577805519104, + "learning_rate": 8.60296916634945e-07, + "loss": 1.3458, + "step": 2484 + }, + { + "epoch": 2.5526450950179767, + "grad_norm": 3.110621213912964, + "learning_rate": 8.583936048724782e-07, + "loss": 1.3027, + "step": 2485 + }, + { + "epoch": 2.553672316384181, + "grad_norm": 3.2169458866119385, + "learning_rate": 8.564902931100115e-07, + "loss": 1.307, + "step": 2486 + }, + { + "epoch": 2.554699537750385, + "grad_norm": 3.26141357421875, + "learning_rate": 8.545869813475448e-07, + "loss": 1.3232, + "step": 2487 + }, + { + "epoch": 2.5557267591165895, + "grad_norm": 3.294462203979492, + "learning_rate": 8.526836695850781e-07, + "loss": 1.3328, + "step": 2488 + }, + { + "epoch": 2.556753980482794, + "grad_norm": 3.301452159881592, + "learning_rate": 8.507803578226115e-07, + "loss": 1.3408, + "step": 2489 + }, + { + "epoch": 2.5577812018489983, + "grad_norm": 3.342216968536377, + "learning_rate": 8.488770460601447e-07, + "loss": 1.3315, + "step": 2490 + }, + { + "epoch": 2.5577812018489983, + "eval_loss": 1.8501962423324585, + "eval_runtime": 37.2524, + "eval_samples_per_second": 26.844, + "eval_steps_per_second": 1.127, + "step": 2490 + }, + { + "epoch": 2.558808423215203, + "grad_norm": 3.3099851608276367, + "learning_rate": 8.46973734297678e-07, + "loss": 1.2939, + "step": 2491 + }, + { + "epoch": 2.5598356445814074, + "grad_norm": 3.141737461090088, + "learning_rate": 8.450704225352114e-07, + "loss": 1.3091, + "step": 2492 + }, + { + "epoch": 2.5608628659476116, + "grad_norm": 3.344872236251831, + "learning_rate": 8.431671107727446e-07, + "loss": 1.3001, + "step": 2493 + }, + { + "epoch": 2.561890087313816, + "grad_norm": 3.3284311294555664, + "learning_rate": 8.41263799010278e-07, + "loss": 1.3393, + "step": 2494 + }, + { + "epoch": 2.5629173086800208, + "grad_norm": 3.349339485168457, + "learning_rate": 8.393604872478113e-07, + "loss": 1.3197, + "step": 2495 + }, + { + "epoch": 2.563944530046225, + "grad_norm": 3.331051826477051, + "learning_rate": 8.374571754853446e-07, + "loss": 1.3074, + "step": 2496 + }, + { + "epoch": 2.5649717514124295, + "grad_norm": 3.2611076831817627, + "learning_rate": 8.355538637228778e-07, + "loss": 1.3172, + "step": 2497 + }, + { + "epoch": 2.5659989727786336, + "grad_norm": 3.284789800643921, + "learning_rate": 8.336505519604111e-07, + "loss": 1.3147, + "step": 2498 + }, + { + "epoch": 2.5670261941448382, + "grad_norm": 3.1650099754333496, + "learning_rate": 8.317472401979445e-07, + "loss": 1.3439, + "step": 2499 + }, + { + "epoch": 2.5680534155110424, + "grad_norm": 3.436192512512207, + "learning_rate": 8.298439284354778e-07, + "loss": 1.3644, + "step": 2500 + }, + { + "epoch": 2.5680534155110424, + "eval_loss": 1.8527119159698486, + "eval_runtime": 37.4654, + "eval_samples_per_second": 26.691, + "eval_steps_per_second": 1.121, + "step": 2500 + } + ], + "logging_steps": 1, + "max_steps": 2919, + "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.7932735234048e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2500/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/latest b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/latest new file mode 100644 index 0000000000000000000000000000000000000000..cfe584fa09082a54d48545d05e97aee6fd00673d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/latest @@ -0,0 +1 @@ +global_step2600 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..0b782d4602b361f7577abb52b710a8c0e088588d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/tokenizer_config.json @@ -0,0 +1,2078 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..4d43704b192c397a9e22a3dd94b9c56211cdff39 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/trainer_state.json @@ -0,0 +1,20313 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 2.6707755521314844, + "eval_steps": 10, + "global_step": 2600, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.001027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7522, + "step": 1 + }, + { + "epoch": 0.002054442732408834, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7541, + "step": 2 + }, + { + "epoch": 0.0030816640986132513, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8031, + "step": 3 + }, + { + "epoch": 0.004108885464817668, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8062, + "step": 4 + }, + { + "epoch": 0.005136106831022085, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7733, + "step": 5 + }, + { + "epoch": 0.0061633281972265025, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7949, + "step": 6 + }, + { + "epoch": 0.007190549563430919, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7657, + "step": 7 + }, + { + "epoch": 0.008217770929635337, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.803, + "step": 8 + }, + { + "epoch": 0.009244992295839754, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7793, + "step": 9 + }, + { + "epoch": 0.01027221366204417, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7935, + "step": 10 + }, + { + "epoch": 0.01027221366204417, + "eval_loss": 2.8114004135131836, + "eval_runtime": 37.8463, + "eval_samples_per_second": 26.423, + "eval_steps_per_second": 1.11, + "step": 10 + }, + { + "epoch": 0.011299435028248588, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7546, + "step": 11 + }, + { + "epoch": 0.012326656394453005, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7423, + "step": 12 + }, + { + "epoch": 0.01335387776065742, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.8057, + "step": 13 + }, + { + "epoch": 0.014381099126861838, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.7663, + "step": 14 + }, + { + "epoch": 0.015408320493066256, + "grad_norm": 4.694424152374268, + "learning_rate": 1.7123287671232876e-08, + "loss": 2.7715, + "step": 15 + }, + { + "epoch": 0.016435541859270673, + "grad_norm": 4.905329704284668, + "learning_rate": 3.424657534246575e-08, + "loss": 2.751, + "step": 16 + }, + { + "epoch": 0.01746276322547509, + "grad_norm": 4.705279350280762, + "learning_rate": 5.136986301369863e-08, + "loss": 2.7186, + "step": 17 + }, + { + "epoch": 0.01848998459167951, + "grad_norm": 5.154810905456543, + "learning_rate": 6.84931506849315e-08, + "loss": 2.8073, + "step": 18 + }, + { + "epoch": 0.019517205957883924, + "grad_norm": 5.070403099060059, + "learning_rate": 8.561643835616439e-08, + "loss": 2.7237, + "step": 19 + }, + { + "epoch": 0.02054442732408834, + "grad_norm": 4.901423454284668, + "learning_rate": 1.0273972602739726e-07, + "loss": 2.7882, + "step": 20 + }, + { + "epoch": 0.02054442732408834, + "eval_loss": 2.810307502746582, + "eval_runtime": 37.65, + "eval_samples_per_second": 26.56, + "eval_steps_per_second": 1.116, + "step": 20 + }, + { + "epoch": 0.02157164869029276, + "grad_norm": 4.914156436920166, + "learning_rate": 1.1986301369863014e-07, + "loss": 2.8005, + "step": 21 + }, + { + "epoch": 0.022598870056497175, + "grad_norm": 4.937129974365234, + "learning_rate": 1.36986301369863e-07, + "loss": 2.7902, + "step": 22 + }, + { + "epoch": 0.02362609142270159, + "grad_norm": 5.0690202713012695, + "learning_rate": 1.541095890410959e-07, + "loss": 2.7474, + "step": 23 + }, + { + "epoch": 0.02465331278890601, + "grad_norm": 4.980844974517822, + "learning_rate": 1.7123287671232878e-07, + "loss": 2.7695, + "step": 24 + }, + { + "epoch": 0.025680534155110426, + "grad_norm": 4.8092451095581055, + "learning_rate": 1.8835616438356165e-07, + "loss": 2.7611, + "step": 25 + }, + { + "epoch": 0.02670775552131484, + "grad_norm": 4.765528202056885, + "learning_rate": 2.0547945205479452e-07, + "loss": 2.7823, + "step": 26 + }, + { + "epoch": 0.02773497688751926, + "grad_norm": 3.9759938716888428, + "learning_rate": 2.226027397260274e-07, + "loss": 2.7445, + "step": 27 + }, + { + "epoch": 0.028762198253723677, + "grad_norm": 4.023419380187988, + "learning_rate": 2.397260273972603e-07, + "loss": 2.7679, + "step": 28 + }, + { + "epoch": 0.029789419619928096, + "grad_norm": 4.161378383636475, + "learning_rate": 2.568493150684932e-07, + "loss": 2.7704, + "step": 29 + }, + { + "epoch": 0.030816640986132512, + "grad_norm": 3.8363842964172363, + "learning_rate": 2.73972602739726e-07, + "loss": 2.7598, + "step": 30 + }, + { + "epoch": 0.030816640986132512, + "eval_loss": 2.772540330886841, + "eval_runtime": 37.2624, + "eval_samples_per_second": 26.837, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03184386235233693, + "grad_norm": 3.718747615814209, + "learning_rate": 2.910958904109589e-07, + "loss": 2.7241, + "step": 31 + }, + { + "epoch": 0.03287108371854135, + "grad_norm": 3.555985927581787, + "learning_rate": 3.082191780821918e-07, + "loss": 2.7751, + "step": 32 + }, + { + "epoch": 0.03389830508474576, + "grad_norm": 3.6041972637176514, + "learning_rate": 3.2534246575342466e-07, + "loss": 2.7322, + "step": 33 + }, + { + "epoch": 0.03492552645095018, + "grad_norm": 3.4408438205718994, + "learning_rate": 3.4246575342465755e-07, + "loss": 2.727, + "step": 34 + }, + { + "epoch": 0.035952747817154594, + "grad_norm": 3.4169280529022217, + "learning_rate": 3.595890410958904e-07, + "loss": 2.7062, + "step": 35 + }, + { + "epoch": 0.03697996918335902, + "grad_norm": 3.3882925510406494, + "learning_rate": 3.767123287671233e-07, + "loss": 2.6271, + "step": 36 + }, + { + "epoch": 0.03800719054956343, + "grad_norm": 3.0491855144500732, + "learning_rate": 3.938356164383562e-07, + "loss": 2.7309, + "step": 37 + }, + { + "epoch": 0.03903441191576785, + "grad_norm": 2.8444902896881104, + "learning_rate": 4.1095890410958903e-07, + "loss": 2.647, + "step": 38 + }, + { + "epoch": 0.040061633281972264, + "grad_norm": 2.7438602447509766, + "learning_rate": 4.2808219178082193e-07, + "loss": 2.6478, + "step": 39 + }, + { + "epoch": 0.04108885464817668, + "grad_norm": 2.8886020183563232, + "learning_rate": 4.452054794520548e-07, + "loss": 2.7065, + "step": 40 + }, + { + "epoch": 0.04108885464817668, + "eval_loss": 2.7021424770355225, + "eval_runtime": 37.3227, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.042116076014381096, + "grad_norm": 2.8825268745422363, + "learning_rate": 4.6232876712328767e-07, + "loss": 2.6666, + "step": 41 + }, + { + "epoch": 0.04314329738058552, + "grad_norm": 2.8848748207092285, + "learning_rate": 4.794520547945206e-07, + "loss": 2.597, + "step": 42 + }, + { + "epoch": 0.044170518746789934, + "grad_norm": 2.9176979064941406, + "learning_rate": 4.965753424657534e-07, + "loss": 2.6638, + "step": 43 + }, + { + "epoch": 0.04519774011299435, + "grad_norm": 2.6903059482574463, + "learning_rate": 5.136986301369864e-07, + "loss": 2.6202, + "step": 44 + }, + { + "epoch": 0.046224961479198766, + "grad_norm": 2.5349855422973633, + "learning_rate": 5.308219178082192e-07, + "loss": 2.6466, + "step": 45 + }, + { + "epoch": 0.04725218284540318, + "grad_norm": 2.405383825302124, + "learning_rate": 5.47945205479452e-07, + "loss": 2.6291, + "step": 46 + }, + { + "epoch": 0.048279404211607604, + "grad_norm": 2.526106595993042, + "learning_rate": 5.65068493150685e-07, + "loss": 2.6718, + "step": 47 + }, + { + "epoch": 0.04930662557781202, + "grad_norm": 2.3981597423553467, + "learning_rate": 5.821917808219178e-07, + "loss": 2.592, + "step": 48 + }, + { + "epoch": 0.050333846944016436, + "grad_norm": 2.61842679977417, + "learning_rate": 5.993150684931507e-07, + "loss": 2.6039, + "step": 49 + }, + { + "epoch": 0.05136106831022085, + "grad_norm": 2.3001420497894287, + "learning_rate": 6.164383561643836e-07, + "loss": 2.5481, + "step": 50 + }, + { + "epoch": 0.05136106831022085, + "eval_loss": 2.6244444847106934, + "eval_runtime": 37.3248, + "eval_samples_per_second": 26.792, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05238828967642527, + "grad_norm": 2.12744140625, + "learning_rate": 6.335616438356165e-07, + "loss": 2.5336, + "step": 51 + }, + { + "epoch": 0.05341551104262968, + "grad_norm": 1.974413275718689, + "learning_rate": 6.506849315068493e-07, + "loss": 2.5882, + "step": 52 + }, + { + "epoch": 0.054442732408834106, + "grad_norm": 2.098527669906616, + "learning_rate": 6.678082191780823e-07, + "loss": 2.5073, + "step": 53 + }, + { + "epoch": 0.05546995377503852, + "grad_norm": 2.056288719177246, + "learning_rate": 6.849315068493151e-07, + "loss": 2.5612, + "step": 54 + }, + { + "epoch": 0.05649717514124294, + "grad_norm": 2.2291629314422607, + "learning_rate": 7.020547945205481e-07, + "loss": 2.527, + "step": 55 + }, + { + "epoch": 0.05752439650744735, + "grad_norm": 2.140164375305176, + "learning_rate": 7.191780821917808e-07, + "loss": 2.4972, + "step": 56 + }, + { + "epoch": 0.05855161787365177, + "grad_norm": 1.8682485818862915, + "learning_rate": 7.363013698630137e-07, + "loss": 2.5231, + "step": 57 + }, + { + "epoch": 0.05957883923985619, + "grad_norm": 1.9373308420181274, + "learning_rate": 7.534246575342466e-07, + "loss": 2.4509, + "step": 58 + }, + { + "epoch": 0.06060606060606061, + "grad_norm": 1.8535712957382202, + "learning_rate": 7.705479452054795e-07, + "loss": 2.5164, + "step": 59 + }, + { + "epoch": 0.061633281972265024, + "grad_norm": 1.7497329711914062, + "learning_rate": 7.876712328767124e-07, + "loss": 2.5177, + "step": 60 + }, + { + "epoch": 0.061633281972265024, + "eval_loss": 2.5608646869659424, + "eval_runtime": 37.3426, + "eval_samples_per_second": 26.779, + "eval_steps_per_second": 1.125, + "step": 60 + }, + { + "epoch": 0.06266050333846944, + "grad_norm": 1.8476629257202148, + "learning_rate": 8.047945205479453e-07, + "loss": 2.4838, + "step": 61 + }, + { + "epoch": 0.06368772470467386, + "grad_norm": 1.7130696773529053, + "learning_rate": 8.219178082191781e-07, + "loss": 2.473, + "step": 62 + }, + { + "epoch": 0.06471494607087827, + "grad_norm": 1.751081109046936, + "learning_rate": 8.39041095890411e-07, + "loss": 2.5152, + "step": 63 + }, + { + "epoch": 0.0657421674370827, + "grad_norm": 1.7758578062057495, + "learning_rate": 8.561643835616439e-07, + "loss": 2.5031, + "step": 64 + }, + { + "epoch": 0.0667693888032871, + "grad_norm": 1.9316363334655762, + "learning_rate": 8.732876712328768e-07, + "loss": 2.3695, + "step": 65 + }, + { + "epoch": 0.06779661016949153, + "grad_norm": 1.7765878438949585, + "learning_rate": 8.904109589041097e-07, + "loss": 2.4669, + "step": 66 + }, + { + "epoch": 0.06882383153569595, + "grad_norm": 1.7339810132980347, + "learning_rate": 9.075342465753426e-07, + "loss": 2.4112, + "step": 67 + }, + { + "epoch": 0.06985105290190036, + "grad_norm": 1.717781662940979, + "learning_rate": 9.246575342465753e-07, + "loss": 2.4677, + "step": 68 + }, + { + "epoch": 0.07087827426810478, + "grad_norm": 1.7711046934127808, + "learning_rate": 9.417808219178083e-07, + "loss": 2.4131, + "step": 69 + }, + { + "epoch": 0.07190549563430919, + "grad_norm": 1.708680510520935, + "learning_rate": 9.589041095890411e-07, + "loss": 2.5191, + "step": 70 + }, + { + "epoch": 0.07190549563430919, + "eval_loss": 2.4920735359191895, + "eval_runtime": 37.3501, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 70 + }, + { + "epoch": 0.07293271700051361, + "grad_norm": 1.9429680109024048, + "learning_rate": 9.76027397260274e-07, + "loss": 2.4099, + "step": 71 + }, + { + "epoch": 0.07395993836671803, + "grad_norm": 1.7411247491836548, + "learning_rate": 9.931506849315068e-07, + "loss": 2.3632, + "step": 72 + }, + { + "epoch": 0.07498715973292244, + "grad_norm": 1.7577983140945435, + "learning_rate": 1.0102739726027399e-06, + "loss": 2.4214, + "step": 73 + }, + { + "epoch": 0.07601438109912687, + "grad_norm": 2.025926351547241, + "learning_rate": 1.0273972602739727e-06, + "loss": 2.394, + "step": 74 + }, + { + "epoch": 0.07704160246533127, + "grad_norm": 2.4366860389709473, + "learning_rate": 1.0445205479452056e-06, + "loss": 2.3943, + "step": 75 + }, + { + "epoch": 0.0780688238315357, + "grad_norm": 2.437108278274536, + "learning_rate": 1.0616438356164384e-06, + "loss": 2.3844, + "step": 76 + }, + { + "epoch": 0.07909604519774012, + "grad_norm": 2.1352434158325195, + "learning_rate": 1.0787671232876712e-06, + "loss": 2.3871, + "step": 77 + }, + { + "epoch": 0.08012326656394453, + "grad_norm": 1.8895397186279297, + "learning_rate": 1.095890410958904e-06, + "loss": 2.3927, + "step": 78 + }, + { + "epoch": 0.08115048793014895, + "grad_norm": 2.2960093021392822, + "learning_rate": 1.1130136986301371e-06, + "loss": 2.3343, + "step": 79 + }, + { + "epoch": 0.08217770929635336, + "grad_norm": 2.0891849994659424, + "learning_rate": 1.13013698630137e-06, + "loss": 2.3732, + "step": 80 + }, + { + "epoch": 0.08217770929635336, + "eval_loss": 2.4189229011535645, + "eval_runtime": 37.3645, + "eval_samples_per_second": 26.763, + "eval_steps_per_second": 1.124, + "step": 80 + }, + { + "epoch": 0.08320493066255778, + "grad_norm": 2.450587034225464, + "learning_rate": 1.1472602739726028e-06, + "loss": 2.3163, + "step": 81 + }, + { + "epoch": 0.08423215202876219, + "grad_norm": 2.079949378967285, + "learning_rate": 1.1643835616438357e-06, + "loss": 2.3381, + "step": 82 + }, + { + "epoch": 0.08525937339496661, + "grad_norm": 2.160737991333008, + "learning_rate": 1.1815068493150685e-06, + "loss": 2.3376, + "step": 83 + }, + { + "epoch": 0.08628659476117104, + "grad_norm": 2.463890552520752, + "learning_rate": 1.1986301369863014e-06, + "loss": 2.308, + "step": 84 + }, + { + "epoch": 0.08731381612737545, + "grad_norm": 2.3923065662384033, + "learning_rate": 1.2157534246575344e-06, + "loss": 2.2682, + "step": 85 + }, + { + "epoch": 0.08834103749357987, + "grad_norm": 2.2096004486083984, + "learning_rate": 1.2328767123287673e-06, + "loss": 2.3256, + "step": 86 + }, + { + "epoch": 0.08936825885978428, + "grad_norm": 2.1402556896209717, + "learning_rate": 1.25e-06, + "loss": 2.2767, + "step": 87 + }, + { + "epoch": 0.0903954802259887, + "grad_norm": 2.500629186630249, + "learning_rate": 1.267123287671233e-06, + "loss": 2.2576, + "step": 88 + }, + { + "epoch": 0.09142270159219312, + "grad_norm": 2.0257675647735596, + "learning_rate": 1.284246575342466e-06, + "loss": 2.2722, + "step": 89 + }, + { + "epoch": 0.09244992295839753, + "grad_norm": 2.148428440093994, + "learning_rate": 1.3013698630136986e-06, + "loss": 2.2309, + "step": 90 + }, + { + "epoch": 0.09244992295839753, + "eval_loss": 2.3439993858337402, + "eval_runtime": 37.3803, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 90 + }, + { + "epoch": 0.09347714432460195, + "grad_norm": 2.3541946411132812, + "learning_rate": 1.3184931506849317e-06, + "loss": 2.2347, + "step": 91 + }, + { + "epoch": 0.09450436569080636, + "grad_norm": 2.6239383220672607, + "learning_rate": 1.3356164383561645e-06, + "loss": 2.2589, + "step": 92 + }, + { + "epoch": 0.09553158705701079, + "grad_norm": 2.1981804370880127, + "learning_rate": 1.3527397260273976e-06, + "loss": 2.2372, + "step": 93 + }, + { + "epoch": 0.09655880842321521, + "grad_norm": 2.3652358055114746, + "learning_rate": 1.3698630136986302e-06, + "loss": 2.2169, + "step": 94 + }, + { + "epoch": 0.09758602978941962, + "grad_norm": 2.2078158855438232, + "learning_rate": 1.386986301369863e-06, + "loss": 2.2761, + "step": 95 + }, + { + "epoch": 0.09861325115562404, + "grad_norm": 2.494685649871826, + "learning_rate": 1.4041095890410961e-06, + "loss": 2.1691, + "step": 96 + }, + { + "epoch": 0.09964047252182845, + "grad_norm": 2.3293051719665527, + "learning_rate": 1.421232876712329e-06, + "loss": 2.2246, + "step": 97 + }, + { + "epoch": 0.10066769388803287, + "grad_norm": 2.410687208175659, + "learning_rate": 1.4383561643835616e-06, + "loss": 2.1166, + "step": 98 + }, + { + "epoch": 0.1016949152542373, + "grad_norm": 2.5168960094451904, + "learning_rate": 1.4554794520547946e-06, + "loss": 2.1417, + "step": 99 + }, + { + "epoch": 0.1027221366204417, + "grad_norm": 2.178412914276123, + "learning_rate": 1.4726027397260275e-06, + "loss": 2.1638, + "step": 100 + }, + { + "epoch": 0.1027221366204417, + "eval_loss": 2.273622751235962, + "eval_runtime": 37.3892, + "eval_samples_per_second": 26.746, + "eval_steps_per_second": 1.123, + "step": 100 + }, + { + "epoch": 0.10374935798664613, + "grad_norm": 2.8951706886291504, + "learning_rate": 1.4897260273972605e-06, + "loss": 2.1631, + "step": 101 + }, + { + "epoch": 0.10477657935285054, + "grad_norm": 2.103206157684326, + "learning_rate": 1.5068493150684932e-06, + "loss": 2.1419, + "step": 102 + }, + { + "epoch": 0.10580380071905496, + "grad_norm": 2.495969772338867, + "learning_rate": 1.5239726027397262e-06, + "loss": 2.1029, + "step": 103 + }, + { + "epoch": 0.10683102208525937, + "grad_norm": 2.5686826705932617, + "learning_rate": 1.541095890410959e-06, + "loss": 2.1235, + "step": 104 + }, + { + "epoch": 0.10785824345146379, + "grad_norm": 3.4710471630096436, + "learning_rate": 1.5582191780821921e-06, + "loss": 2.1832, + "step": 105 + }, + { + "epoch": 0.10888546481766821, + "grad_norm": 2.2563316822052, + "learning_rate": 1.5753424657534248e-06, + "loss": 2.1441, + "step": 106 + }, + { + "epoch": 0.10991268618387262, + "grad_norm": 3.860506534576416, + "learning_rate": 1.5924657534246576e-06, + "loss": 2.1764, + "step": 107 + }, + { + "epoch": 0.11093990755007704, + "grad_norm": 2.550135850906372, + "learning_rate": 1.6095890410958907e-06, + "loss": 2.1455, + "step": 108 + }, + { + "epoch": 0.11196712891628145, + "grad_norm": 3.168283700942993, + "learning_rate": 1.6267123287671235e-06, + "loss": 2.0246, + "step": 109 + }, + { + "epoch": 0.11299435028248588, + "grad_norm": 2.577069044113159, + "learning_rate": 1.6438356164383561e-06, + "loss": 2.0921, + "step": 110 + }, + { + "epoch": 0.11299435028248588, + "eval_loss": 2.2218997478485107, + "eval_runtime": 37.246, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 110 + }, + { + "epoch": 0.1140215716486903, + "grad_norm": 2.3515374660491943, + "learning_rate": 1.6609589041095892e-06, + "loss": 2.0209, + "step": 111 + }, + { + "epoch": 0.1150487930148947, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.678082191780822e-06, + "loss": 2.1165, + "step": 112 + }, + { + "epoch": 0.11607601438109913, + "grad_norm": 2.567336082458496, + "learning_rate": 1.695205479452055e-06, + "loss": 2.0529, + "step": 113 + }, + { + "epoch": 0.11710323574730354, + "grad_norm": 2.4126152992248535, + "learning_rate": 1.7123287671232877e-06, + "loss": 2.1028, + "step": 114 + }, + { + "epoch": 0.11813045711350796, + "grad_norm": 3.577528476715088, + "learning_rate": 1.7294520547945206e-06, + "loss": 2.0317, + "step": 115 + }, + { + "epoch": 0.11915767847971238, + "grad_norm": 2.9624195098876953, + "learning_rate": 1.7465753424657536e-06, + "loss": 2.1318, + "step": 116 + }, + { + "epoch": 0.12018489984591679, + "grad_norm": 2.789854049682617, + "learning_rate": 1.7636986301369865e-06, + "loss": 2.0822, + "step": 117 + }, + { + "epoch": 0.12121212121212122, + "grad_norm": 2.8168365955352783, + "learning_rate": 1.7808219178082193e-06, + "loss": 2.08, + "step": 118 + }, + { + "epoch": 0.12223934257832562, + "grad_norm": 2.7616257667541504, + "learning_rate": 1.7979452054794521e-06, + "loss": 2.0668, + "step": 119 + }, + { + "epoch": 0.12326656394453005, + "grad_norm": 2.5610315799713135, + "learning_rate": 1.8150684931506852e-06, + "loss": 2.1518, + "step": 120 + }, + { + "epoch": 0.12326656394453005, + "eval_loss": 2.184033155441284, + "eval_runtime": 37.3784, + "eval_samples_per_second": 26.753, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.12429378531073447, + "grad_norm": 2.817063093185425, + "learning_rate": 1.832191780821918e-06, + "loss": 2.0813, + "step": 121 + }, + { + "epoch": 0.12532100667693888, + "grad_norm": 2.987205982208252, + "learning_rate": 1.8493150684931507e-06, + "loss": 2.0421, + "step": 122 + }, + { + "epoch": 0.1263482280431433, + "grad_norm": 2.8480496406555176, + "learning_rate": 1.8664383561643837e-06, + "loss": 2.0018, + "step": 123 + }, + { + "epoch": 0.12737544940934772, + "grad_norm": 2.994274616241455, + "learning_rate": 1.8835616438356166e-06, + "loss": 2.0266, + "step": 124 + }, + { + "epoch": 0.12840267077555212, + "grad_norm": 2.64136004447937, + "learning_rate": 1.9006849315068496e-06, + "loss": 1.9873, + "step": 125 + }, + { + "epoch": 0.12942989214175654, + "grad_norm": 2.9426915645599365, + "learning_rate": 1.9178082191780823e-06, + "loss": 2.0566, + "step": 126 + }, + { + "epoch": 0.13045711350796096, + "grad_norm": 2.9018619060516357, + "learning_rate": 1.9349315068493153e-06, + "loss": 1.9674, + "step": 127 + }, + { + "epoch": 0.1314843348741654, + "grad_norm": 3.6921045780181885, + "learning_rate": 1.952054794520548e-06, + "loss": 2.0295, + "step": 128 + }, + { + "epoch": 0.1325115562403698, + "grad_norm": 2.4194657802581787, + "learning_rate": 1.969178082191781e-06, + "loss": 2.0639, + "step": 129 + }, + { + "epoch": 0.1335387776065742, + "grad_norm": 3.252349615097046, + "learning_rate": 1.9863013698630136e-06, + "loss": 2.0613, + "step": 130 + }, + { + "epoch": 0.1335387776065742, + "eval_loss": 2.153157949447632, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 130 + }, + { + "epoch": 0.13456599897277863, + "grad_norm": 2.9361023902893066, + "learning_rate": 2.0034246575342467e-06, + "loss": 1.9817, + "step": 131 + }, + { + "epoch": 0.13559322033898305, + "grad_norm": 2.834803581237793, + "learning_rate": 2.0205479452054797e-06, + "loss": 2.0539, + "step": 132 + }, + { + "epoch": 0.13662044170518747, + "grad_norm": 2.8814785480499268, + "learning_rate": 2.037671232876713e-06, + "loss": 2.0296, + "step": 133 + }, + { + "epoch": 0.1376476630713919, + "grad_norm": 2.647125482559204, + "learning_rate": 2.0547945205479454e-06, + "loss": 1.9739, + "step": 134 + }, + { + "epoch": 0.1386748844375963, + "grad_norm": 2.9608750343322754, + "learning_rate": 2.071917808219178e-06, + "loss": 2.0083, + "step": 135 + }, + { + "epoch": 0.1397021058038007, + "grad_norm": 2.8570032119750977, + "learning_rate": 2.089041095890411e-06, + "loss": 2.0216, + "step": 136 + }, + { + "epoch": 0.14072932717000514, + "grad_norm": 2.806358575820923, + "learning_rate": 2.106164383561644e-06, + "loss": 1.9457, + "step": 137 + }, + { + "epoch": 0.14175654853620956, + "grad_norm": 3.1003353595733643, + "learning_rate": 2.123287671232877e-06, + "loss": 2.0324, + "step": 138 + }, + { + "epoch": 0.14278376990241398, + "grad_norm": 2.679588556289673, + "learning_rate": 2.14041095890411e-06, + "loss": 1.9699, + "step": 139 + }, + { + "epoch": 0.14381099126861838, + "grad_norm": 2.9436194896698, + "learning_rate": 2.1575342465753425e-06, + "loss": 1.9824, + "step": 140 + }, + { + "epoch": 0.14381099126861838, + "eval_loss": 2.123953104019165, + "eval_runtime": 37.4214, + "eval_samples_per_second": 26.723, + "eval_steps_per_second": 1.122, + "step": 140 + }, + { + "epoch": 0.1448382126348228, + "grad_norm": 3.088301420211792, + "learning_rate": 2.1746575342465755e-06, + "loss": 1.9861, + "step": 141 + }, + { + "epoch": 0.14586543400102722, + "grad_norm": 3.011859178543091, + "learning_rate": 2.191780821917808e-06, + "loss": 1.9383, + "step": 142 + }, + { + "epoch": 0.14689265536723164, + "grad_norm": 2.930264472961426, + "learning_rate": 2.2089041095890412e-06, + "loss": 2.0674, + "step": 143 + }, + { + "epoch": 0.14791987673343607, + "grad_norm": 3.361328601837158, + "learning_rate": 2.2260273972602743e-06, + "loss": 1.9495, + "step": 144 + }, + { + "epoch": 0.14894709809964046, + "grad_norm": 2.7825112342834473, + "learning_rate": 2.243150684931507e-06, + "loss": 1.9335, + "step": 145 + }, + { + "epoch": 0.14997431946584489, + "grad_norm": 3.7215847969055176, + "learning_rate": 2.26027397260274e-06, + "loss": 1.9403, + "step": 146 + }, + { + "epoch": 0.1510015408320493, + "grad_norm": 2.5125460624694824, + "learning_rate": 2.2773972602739726e-06, + "loss": 1.9437, + "step": 147 + }, + { + "epoch": 0.15202876219825373, + "grad_norm": 3.0963492393493652, + "learning_rate": 2.2945205479452057e-06, + "loss": 1.9651, + "step": 148 + }, + { + "epoch": 0.15305598356445815, + "grad_norm": 2.9147121906280518, + "learning_rate": 2.3116438356164387e-06, + "loss": 2.0567, + "step": 149 + }, + { + "epoch": 0.15408320493066255, + "grad_norm": 3.5275802612304688, + "learning_rate": 2.3287671232876713e-06, + "loss": 1.9773, + "step": 150 + }, + { + "epoch": 0.15408320493066255, + "eval_loss": 2.112344264984131, + "eval_runtime": 37.4142, + "eval_samples_per_second": 26.728, + "eval_steps_per_second": 1.123, + "step": 150 + }, + { + "epoch": 0.15511042629686697, + "grad_norm": 3.013786554336548, + "learning_rate": 2.3458904109589044e-06, + "loss": 1.8753, + "step": 151 + }, + { + "epoch": 0.1561376476630714, + "grad_norm": 3.433615207672119, + "learning_rate": 2.363013698630137e-06, + "loss": 1.9256, + "step": 152 + }, + { + "epoch": 0.15716486902927582, + "grad_norm": 3.1526033878326416, + "learning_rate": 2.38013698630137e-06, + "loss": 1.9815, + "step": 153 + }, + { + "epoch": 0.15819209039548024, + "grad_norm": 3.2099974155426025, + "learning_rate": 2.3972602739726027e-06, + "loss": 1.9628, + "step": 154 + }, + { + "epoch": 0.15921931176168463, + "grad_norm": 3.2713584899902344, + "learning_rate": 2.4143835616438358e-06, + "loss": 1.9128, + "step": 155 + }, + { + "epoch": 0.16024653312788906, + "grad_norm": 3.287649393081665, + "learning_rate": 2.431506849315069e-06, + "loss": 1.8764, + "step": 156 + }, + { + "epoch": 0.16127375449409348, + "grad_norm": 3.1439101696014404, + "learning_rate": 2.4486301369863015e-06, + "loss": 1.9428, + "step": 157 + }, + { + "epoch": 0.1623009758602979, + "grad_norm": 3.006457805633545, + "learning_rate": 2.4657534246575345e-06, + "loss": 1.9516, + "step": 158 + }, + { + "epoch": 0.1633281972265023, + "grad_norm": 2.9567787647247314, + "learning_rate": 2.482876712328767e-06, + "loss": 1.915, + "step": 159 + }, + { + "epoch": 0.16435541859270672, + "grad_norm": 3.2475216388702393, + "learning_rate": 2.5e-06, + "loss": 1.9451, + "step": 160 + }, + { + "epoch": 0.16435541859270672, + "eval_loss": 2.086299419403076, + "eval_runtime": 37.3939, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 160 + }, + { + "epoch": 0.16538263995891114, + "grad_norm": 2.9586498737335205, + "learning_rate": 2.5171232876712333e-06, + "loss": 1.9298, + "step": 161 + }, + { + "epoch": 0.16640986132511557, + "grad_norm": 3.066910982131958, + "learning_rate": 2.534246575342466e-06, + "loss": 1.9103, + "step": 162 + }, + { + "epoch": 0.16743708269132, + "grad_norm": 2.814741611480713, + "learning_rate": 2.551369863013699e-06, + "loss": 1.8678, + "step": 163 + }, + { + "epoch": 0.16846430405752438, + "grad_norm": 3.7239339351654053, + "learning_rate": 2.568493150684932e-06, + "loss": 1.9321, + "step": 164 + }, + { + "epoch": 0.1694915254237288, + "grad_norm": 2.829899787902832, + "learning_rate": 2.585616438356164e-06, + "loss": 1.9587, + "step": 165 + }, + { + "epoch": 0.17051874678993323, + "grad_norm": 3.826289176940918, + "learning_rate": 2.6027397260273973e-06, + "loss": 1.9288, + "step": 166 + }, + { + "epoch": 0.17154596815613765, + "grad_norm": 3.1028499603271484, + "learning_rate": 2.6198630136986303e-06, + "loss": 1.8685, + "step": 167 + }, + { + "epoch": 0.17257318952234207, + "grad_norm": 3.6923508644104004, + "learning_rate": 2.6369863013698634e-06, + "loss": 1.9276, + "step": 168 + }, + { + "epoch": 0.17360041088854647, + "grad_norm": 3.0577642917633057, + "learning_rate": 2.654109589041096e-06, + "loss": 1.9653, + "step": 169 + }, + { + "epoch": 0.1746276322547509, + "grad_norm": 3.363762617111206, + "learning_rate": 2.671232876712329e-06, + "loss": 1.9465, + "step": 170 + }, + { + "epoch": 0.1746276322547509, + "eval_loss": 2.0753531455993652, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 170 + }, + { + "epoch": 0.17565485362095531, + "grad_norm": 2.9063165187835693, + "learning_rate": 2.688356164383562e-06, + "loss": 1.8827, + "step": 171 + }, + { + "epoch": 0.17668207498715974, + "grad_norm": 3.4709181785583496, + "learning_rate": 2.705479452054795e-06, + "loss": 1.882, + "step": 172 + }, + { + "epoch": 0.17770929635336416, + "grad_norm": 3.395397663116455, + "learning_rate": 2.7226027397260274e-06, + "loss": 1.8711, + "step": 173 + }, + { + "epoch": 0.17873651771956856, + "grad_norm": 3.9973363876342773, + "learning_rate": 2.7397260273972604e-06, + "loss": 1.8223, + "step": 174 + }, + { + "epoch": 0.17976373908577298, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.947, + "step": 175 + }, + { + "epoch": 0.1807909604519774, + "grad_norm": 3.1841156482696533, + "learning_rate": 2.7568493150684935e-06, + "loss": 1.9192, + "step": 176 + }, + { + "epoch": 0.18181818181818182, + "grad_norm": 3.84712290763855, + "learning_rate": 2.773972602739726e-06, + "loss": 1.8344, + "step": 177 + }, + { + "epoch": 0.18284540318438625, + "grad_norm": 3.487525463104248, + "learning_rate": 2.791095890410959e-06, + "loss": 1.9242, + "step": 178 + }, + { + "epoch": 0.18387262455059064, + "grad_norm": 3.1980319023132324, + "learning_rate": 2.8082191780821922e-06, + "loss": 1.7959, + "step": 179 + }, + { + "epoch": 0.18489984591679506, + "grad_norm": 3.56462025642395, + "learning_rate": 2.8253424657534253e-06, + "loss": 1.9668, + "step": 180 + }, + { + "epoch": 0.18489984591679506, + "eval_loss": 2.0606908798217773, + "eval_runtime": 37.3994, + "eval_samples_per_second": 26.738, + "eval_steps_per_second": 1.123, + "step": 180 + }, + { + "epoch": 0.1859270672829995, + "grad_norm": 3.613842248916626, + "learning_rate": 2.842465753424658e-06, + "loss": 1.925, + "step": 181 + }, + { + "epoch": 0.1869542886492039, + "grad_norm": 2.9755642414093018, + "learning_rate": 2.8595890410958905e-06, + "loss": 1.9251, + "step": 182 + }, + { + "epoch": 0.18798151001540833, + "grad_norm": 2.9678146839141846, + "learning_rate": 2.876712328767123e-06, + "loss": 1.9406, + "step": 183 + }, + { + "epoch": 0.18900873138161273, + "grad_norm": 3.3536393642425537, + "learning_rate": 2.8938356164383562e-06, + "loss": 1.8041, + "step": 184 + }, + { + "epoch": 0.19003595274781715, + "grad_norm": 2.714902877807617, + "learning_rate": 2.9109589041095893e-06, + "loss": 1.8213, + "step": 185 + }, + { + "epoch": 0.19106317411402157, + "grad_norm": 3.0686237812042236, + "learning_rate": 2.9280821917808223e-06, + "loss": 1.817, + "step": 186 + }, + { + "epoch": 0.192090395480226, + "grad_norm": 3.510448932647705, + "learning_rate": 2.945205479452055e-06, + "loss": 1.8317, + "step": 187 + }, + { + "epoch": 0.19311761684643042, + "grad_norm": 3.2060787677764893, + "learning_rate": 2.962328767123288e-06, + "loss": 1.894, + "step": 188 + }, + { + "epoch": 0.1941448382126348, + "grad_norm": 3.1756906509399414, + "learning_rate": 2.979452054794521e-06, + "loss": 1.8788, + "step": 189 + }, + { + "epoch": 0.19517205957883924, + "grad_norm": 3.059863805770874, + "learning_rate": 2.9965753424657533e-06, + "loss": 1.8198, + "step": 190 + }, + { + "epoch": 0.19517205957883924, + "eval_loss": 2.0506513118743896, + "eval_runtime": 37.4014, + "eval_samples_per_second": 26.737, + "eval_steps_per_second": 1.123, + "step": 190 + }, + { + "epoch": 0.19619928094504366, + "grad_norm": 3.193685531616211, + "learning_rate": 3.0136986301369864e-06, + "loss": 1.8818, + "step": 191 + }, + { + "epoch": 0.19722650231124808, + "grad_norm": 2.905402183532715, + "learning_rate": 3.0308219178082194e-06, + "loss": 1.8546, + "step": 192 + }, + { + "epoch": 0.1982537236774525, + "grad_norm": 3.423109531402588, + "learning_rate": 3.0479452054794525e-06, + "loss": 1.8142, + "step": 193 + }, + { + "epoch": 0.1992809450436569, + "grad_norm": 3.136094331741333, + "learning_rate": 3.065068493150685e-06, + "loss": 1.8553, + "step": 194 + }, + { + "epoch": 0.20030816640986132, + "grad_norm": 3.0746538639068604, + "learning_rate": 3.082191780821918e-06, + "loss": 1.8153, + "step": 195 + }, + { + "epoch": 0.20133538777606574, + "grad_norm": 2.9354681968688965, + "learning_rate": 3.099315068493151e-06, + "loss": 1.8301, + "step": 196 + }, + { + "epoch": 0.20236260914227017, + "grad_norm": 3.00426983833313, + "learning_rate": 3.1164383561643843e-06, + "loss": 1.8233, + "step": 197 + }, + { + "epoch": 0.2033898305084746, + "grad_norm": 3.2754054069519043, + "learning_rate": 3.1335616438356165e-06, + "loss": 1.787, + "step": 198 + }, + { + "epoch": 0.20441705187467898, + "grad_norm": 2.9854509830474854, + "learning_rate": 3.1506849315068495e-06, + "loss": 1.9341, + "step": 199 + }, + { + "epoch": 0.2054442732408834, + "grad_norm": 3.5655088424682617, + "learning_rate": 3.167808219178082e-06, + "loss": 1.7964, + "step": 200 + }, + { + "epoch": 0.2054442732408834, + "eval_loss": 2.0440263748168945, + "eval_runtime": 37.4054, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 200 + }, + { + "epoch": 0.20647149460708783, + "grad_norm": 3.132373571395874, + "learning_rate": 3.184931506849315e-06, + "loss": 1.8019, + "step": 201 + }, + { + "epoch": 0.20749871597329225, + "grad_norm": 3.3017780780792236, + "learning_rate": 3.2020547945205483e-06, + "loss": 1.7786, + "step": 202 + }, + { + "epoch": 0.20852593733949668, + "grad_norm": 3.009382486343384, + "learning_rate": 3.2191780821917813e-06, + "loss": 1.7639, + "step": 203 + }, + { + "epoch": 0.20955315870570107, + "grad_norm": 3.5784854888916016, + "learning_rate": 3.236301369863014e-06, + "loss": 1.76, + "step": 204 + }, + { + "epoch": 0.2105803800719055, + "grad_norm": 3.236858367919922, + "learning_rate": 3.253424657534247e-06, + "loss": 1.7295, + "step": 205 + }, + { + "epoch": 0.21160760143810992, + "grad_norm": 3.6157479286193848, + "learning_rate": 3.2705479452054796e-06, + "loss": 1.8287, + "step": 206 + }, + { + "epoch": 0.21263482280431434, + "grad_norm": 3.51790452003479, + "learning_rate": 3.2876712328767123e-06, + "loss": 1.8004, + "step": 207 + }, + { + "epoch": 0.21366204417051873, + "grad_norm": 3.0700764656066895, + "learning_rate": 3.3047945205479453e-06, + "loss": 1.7581, + "step": 208 + }, + { + "epoch": 0.21468926553672316, + "grad_norm": 3.2651591300964355, + "learning_rate": 3.3219178082191784e-06, + "loss": 1.7724, + "step": 209 + }, + { + "epoch": 0.21571648690292758, + "grad_norm": 2.9267404079437256, + "learning_rate": 3.3390410958904114e-06, + "loss": 1.8154, + "step": 210 + }, + { + "epoch": 0.21571648690292758, + "eval_loss": 2.0323731899261475, + "eval_runtime": 37.2156, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 210 + }, + { + "epoch": 0.216743708269132, + "grad_norm": 3.463411808013916, + "learning_rate": 3.356164383561644e-06, + "loss": 1.8309, + "step": 211 + }, + { + "epoch": 0.21777092963533642, + "grad_norm": 3.123573064804077, + "learning_rate": 3.373287671232877e-06, + "loss": 1.8793, + "step": 212 + }, + { + "epoch": 0.21879815100154082, + "grad_norm": 3.427051305770874, + "learning_rate": 3.39041095890411e-06, + "loss": 1.8259, + "step": 213 + }, + { + "epoch": 0.21982537236774524, + "grad_norm": 3.306366443634033, + "learning_rate": 3.4075342465753424e-06, + "loss": 1.8565, + "step": 214 + }, + { + "epoch": 0.22085259373394966, + "grad_norm": 3.557340383529663, + "learning_rate": 3.4246575342465754e-06, + "loss": 1.7531, + "step": 215 + }, + { + "epoch": 0.2218798151001541, + "grad_norm": 3.20525860786438, + "learning_rate": 3.4417808219178085e-06, + "loss": 1.8032, + "step": 216 + }, + { + "epoch": 0.2229070364663585, + "grad_norm": 3.2657511234283447, + "learning_rate": 3.458904109589041e-06, + "loss": 1.8026, + "step": 217 + }, + { + "epoch": 0.2239342578325629, + "grad_norm": 3.1869699954986572, + "learning_rate": 3.476027397260274e-06, + "loss": 1.814, + "step": 218 + }, + { + "epoch": 0.22496147919876733, + "grad_norm": 2.661536931991577, + "learning_rate": 3.4931506849315072e-06, + "loss": 1.8005, + "step": 219 + }, + { + "epoch": 0.22598870056497175, + "grad_norm": 3.1108739376068115, + "learning_rate": 3.5102739726027403e-06, + "loss": 1.8095, + "step": 220 + }, + { + "epoch": 0.22598870056497175, + "eval_loss": 2.015268325805664, + "eval_runtime": 37.2867, + "eval_samples_per_second": 26.819, + "eval_steps_per_second": 1.126, + "step": 220 + }, + { + "epoch": 0.22701592193117617, + "grad_norm": 2.8078837394714355, + "learning_rate": 3.527397260273973e-06, + "loss": 1.8016, + "step": 221 + }, + { + "epoch": 0.2280431432973806, + "grad_norm": 3.131399631500244, + "learning_rate": 3.5445205479452056e-06, + "loss": 1.7787, + "step": 222 + }, + { + "epoch": 0.229070364663585, + "grad_norm": 3.114422559738159, + "learning_rate": 3.5616438356164386e-06, + "loss": 1.7899, + "step": 223 + }, + { + "epoch": 0.2300975860297894, + "grad_norm": 3.040816068649292, + "learning_rate": 3.5787671232876712e-06, + "loss": 1.8601, + "step": 224 + }, + { + "epoch": 0.23112480739599384, + "grad_norm": 3.043100357055664, + "learning_rate": 3.5958904109589043e-06, + "loss": 1.8244, + "step": 225 + }, + { + "epoch": 0.23215202876219826, + "grad_norm": 3.4232981204986572, + "learning_rate": 3.6130136986301373e-06, + "loss": 1.7634, + "step": 226 + }, + { + "epoch": 0.23317925012840268, + "grad_norm": 2.8170571327209473, + "learning_rate": 3.6301369863013704e-06, + "loss": 1.7793, + "step": 227 + }, + { + "epoch": 0.23420647149460708, + "grad_norm": 3.9133572578430176, + "learning_rate": 3.647260273972603e-06, + "loss": 1.8239, + "step": 228 + }, + { + "epoch": 0.2352336928608115, + "grad_norm": 2.7053382396698, + "learning_rate": 3.664383561643836e-06, + "loss": 1.8209, + "step": 229 + }, + { + "epoch": 0.23626091422701592, + "grad_norm": 3.4366891384124756, + "learning_rate": 3.6815068493150687e-06, + "loss": 1.7539, + "step": 230 + }, + { + "epoch": 0.23626091422701592, + "eval_loss": 2.0176751613616943, + "eval_runtime": 37.3064, + "eval_samples_per_second": 26.805, + "eval_steps_per_second": 1.126, + "step": 230 + }, + { + "epoch": 0.23728813559322035, + "grad_norm": 2.592280149459839, + "learning_rate": 3.6986301369863014e-06, + "loss": 1.7858, + "step": 231 + }, + { + "epoch": 0.23831535695942477, + "grad_norm": 2.8370869159698486, + "learning_rate": 3.7157534246575344e-06, + "loss": 1.7414, + "step": 232 + }, + { + "epoch": 0.23934257832562916, + "grad_norm": 3.2120535373687744, + "learning_rate": 3.7328767123287675e-06, + "loss": 1.8467, + "step": 233 + }, + { + "epoch": 0.24036979969183359, + "grad_norm": 2.9314255714416504, + "learning_rate": 3.7500000000000005e-06, + "loss": 1.7602, + "step": 234 + }, + { + "epoch": 0.241397021058038, + "grad_norm": 3.056293249130249, + "learning_rate": 3.767123287671233e-06, + "loss": 1.7972, + "step": 235 + }, + { + "epoch": 0.24242424242424243, + "grad_norm": 2.5918917655944824, + "learning_rate": 3.784246575342466e-06, + "loss": 1.8274, + "step": 236 + }, + { + "epoch": 0.24345146379044685, + "grad_norm": 2.7911906242370605, + "learning_rate": 3.8013698630136993e-06, + "loss": 1.753, + "step": 237 + }, + { + "epoch": 0.24447868515665125, + "grad_norm": 2.810612916946411, + "learning_rate": 3.818493150684932e-06, + "loss": 1.7822, + "step": 238 + }, + { + "epoch": 0.24550590652285567, + "grad_norm": 2.8863744735717773, + "learning_rate": 3.8356164383561645e-06, + "loss": 1.7258, + "step": 239 + }, + { + "epoch": 0.2465331278890601, + "grad_norm": 3.002087116241455, + "learning_rate": 3.852739726027397e-06, + "loss": 1.8423, + "step": 240 + }, + { + "epoch": 0.2465331278890601, + "eval_loss": 2.020223617553711, + "eval_runtime": 37.3197, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 240 + }, + { + "epoch": 0.24756034925526452, + "grad_norm": 2.892317056655884, + "learning_rate": 3.869863013698631e-06, + "loss": 1.8181, + "step": 241 + }, + { + "epoch": 0.24858757062146894, + "grad_norm": 2.895620584487915, + "learning_rate": 3.886986301369863e-06, + "loss": 1.7538, + "step": 242 + }, + { + "epoch": 0.24961479198767333, + "grad_norm": 3.015660524368286, + "learning_rate": 3.904109589041096e-06, + "loss": 1.766, + "step": 243 + }, + { + "epoch": 0.25064201335387776, + "grad_norm": 2.7516112327575684, + "learning_rate": 3.921232876712329e-06, + "loss": 1.7324, + "step": 244 + }, + { + "epoch": 0.2516692347200822, + "grad_norm": 2.937777519226074, + "learning_rate": 3.938356164383562e-06, + "loss": 1.7468, + "step": 245 + }, + { + "epoch": 0.2526964560862866, + "grad_norm": 3.1706550121307373, + "learning_rate": 3.9554794520547955e-06, + "loss": 1.7035, + "step": 246 + }, + { + "epoch": 0.253723677452491, + "grad_norm": 2.7795372009277344, + "learning_rate": 3.972602739726027e-06, + "loss": 1.7912, + "step": 247 + }, + { + "epoch": 0.25475089881869545, + "grad_norm": 3.0375940799713135, + "learning_rate": 3.989726027397261e-06, + "loss": 1.7208, + "step": 248 + }, + { + "epoch": 0.25577812018489987, + "grad_norm": 2.863913059234619, + "learning_rate": 4.006849315068493e-06, + "loss": 1.7891, + "step": 249 + }, + { + "epoch": 0.25680534155110424, + "grad_norm": 3.02720046043396, + "learning_rate": 4.023972602739726e-06, + "loss": 1.7864, + "step": 250 + }, + { + "epoch": 0.25680534155110424, + "eval_loss": 1.9981943368911743, + "eval_runtime": 37.3019, + "eval_samples_per_second": 26.808, + "eval_steps_per_second": 1.126, + "step": 250 + }, + { + "epoch": 0.25783256291730866, + "grad_norm": 2.90144681930542, + "learning_rate": 4.0410958904109595e-06, + "loss": 1.7508, + "step": 251 + }, + { + "epoch": 0.2588597842835131, + "grad_norm": 2.7670342922210693, + "learning_rate": 4.058219178082192e-06, + "loss": 1.7668, + "step": 252 + }, + { + "epoch": 0.2598870056497175, + "grad_norm": 3.0309600830078125, + "learning_rate": 4.075342465753426e-06, + "loss": 1.7853, + "step": 253 + }, + { + "epoch": 0.26091422701592193, + "grad_norm": 2.995190143585205, + "learning_rate": 4.092465753424658e-06, + "loss": 1.7409, + "step": 254 + }, + { + "epoch": 0.26194144838212635, + "grad_norm": 3.127767324447632, + "learning_rate": 4.109589041095891e-06, + "loss": 1.743, + "step": 255 + }, + { + "epoch": 0.2629686697483308, + "grad_norm": 3.7104549407958984, + "learning_rate": 4.1267123287671235e-06, + "loss": 1.8149, + "step": 256 + }, + { + "epoch": 0.2639958911145352, + "grad_norm": 3.042569160461426, + "learning_rate": 4.143835616438356e-06, + "loss": 1.7544, + "step": 257 + }, + { + "epoch": 0.2650231124807396, + "grad_norm": 2.9337198734283447, + "learning_rate": 4.16095890410959e-06, + "loss": 1.823, + "step": 258 + }, + { + "epoch": 0.26605033384694404, + "grad_norm": 3.1853089332580566, + "learning_rate": 4.178082191780822e-06, + "loss": 1.8002, + "step": 259 + }, + { + "epoch": 0.2670775552131484, + "grad_norm": 2.9620680809020996, + "learning_rate": 4.195205479452055e-06, + "loss": 1.7401, + "step": 260 + }, + { + "epoch": 0.2670775552131484, + "eval_loss": 1.9844669103622437, + "eval_runtime": 37.3162, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 260 + }, + { + "epoch": 0.26810477657935283, + "grad_norm": 3.477985382080078, + "learning_rate": 4.212328767123288e-06, + "loss": 1.7456, + "step": 261 + }, + { + "epoch": 0.26913199794555726, + "grad_norm": 2.881277084350586, + "learning_rate": 4.229452054794521e-06, + "loss": 1.7427, + "step": 262 + }, + { + "epoch": 0.2701592193117617, + "grad_norm": 3.3313355445861816, + "learning_rate": 4.246575342465754e-06, + "loss": 1.7537, + "step": 263 + }, + { + "epoch": 0.2711864406779661, + "grad_norm": 2.947119951248169, + "learning_rate": 4.263698630136986e-06, + "loss": 1.7349, + "step": 264 + }, + { + "epoch": 0.2722136620441705, + "grad_norm": 3.0780889987945557, + "learning_rate": 4.28082191780822e-06, + "loss": 1.694, + "step": 265 + }, + { + "epoch": 0.27324088341037495, + "grad_norm": 3.3371143341064453, + "learning_rate": 4.297945205479452e-06, + "loss": 1.8489, + "step": 266 + }, + { + "epoch": 0.27426810477657937, + "grad_norm": 3.01694917678833, + "learning_rate": 4.315068493150685e-06, + "loss": 1.7478, + "step": 267 + }, + { + "epoch": 0.2752953261427838, + "grad_norm": 3.001850128173828, + "learning_rate": 4.3321917808219185e-06, + "loss": 1.7298, + "step": 268 + }, + { + "epoch": 0.2763225475089882, + "grad_norm": 2.9096696376800537, + "learning_rate": 4.349315068493151e-06, + "loss": 1.7542, + "step": 269 + }, + { + "epoch": 0.2773497688751926, + "grad_norm": 3.209601402282715, + "learning_rate": 4.3664383561643846e-06, + "loss": 1.7248, + "step": 270 + }, + { + "epoch": 0.2773497688751926, + "eval_loss": 1.9732507467269897, + "eval_runtime": 37.3151, + "eval_samples_per_second": 26.799, + "eval_steps_per_second": 1.126, + "step": 270 + }, + { + "epoch": 0.278376990241397, + "grad_norm": 2.9635558128356934, + "learning_rate": 4.383561643835616e-06, + "loss": 1.727, + "step": 271 + }, + { + "epoch": 0.2794042116076014, + "grad_norm": 3.143821954727173, + "learning_rate": 4.40068493150685e-06, + "loss": 1.8045, + "step": 272 + }, + { + "epoch": 0.28043143297380585, + "grad_norm": 3.4248714447021484, + "learning_rate": 4.4178082191780825e-06, + "loss": 1.7649, + "step": 273 + }, + { + "epoch": 0.2814586543400103, + "grad_norm": 2.902162790298462, + "learning_rate": 4.434931506849315e-06, + "loss": 1.7324, + "step": 274 + }, + { + "epoch": 0.2824858757062147, + "grad_norm": 2.8899712562561035, + "learning_rate": 4.4520547945205486e-06, + "loss": 1.6886, + "step": 275 + }, + { + "epoch": 0.2835130970724191, + "grad_norm": 3.2267258167266846, + "learning_rate": 4.469178082191781e-06, + "loss": 1.7442, + "step": 276 + }, + { + "epoch": 0.28454031843862354, + "grad_norm": 3.275735378265381, + "learning_rate": 4.486301369863014e-06, + "loss": 1.7334, + "step": 277 + }, + { + "epoch": 0.28556753980482796, + "grad_norm": 2.991509199142456, + "learning_rate": 4.503424657534247e-06, + "loss": 1.7853, + "step": 278 + }, + { + "epoch": 0.28659476117103233, + "grad_norm": 2.8380391597747803, + "learning_rate": 4.52054794520548e-06, + "loss": 1.6885, + "step": 279 + }, + { + "epoch": 0.28762198253723675, + "grad_norm": 2.909980297088623, + "learning_rate": 4.537671232876713e-06, + "loss": 1.7146, + "step": 280 + }, + { + "epoch": 0.28762198253723675, + "eval_loss": 1.9755980968475342, + "eval_runtime": 37.3524, + "eval_samples_per_second": 26.772, + "eval_steps_per_second": 1.124, + "step": 280 + }, + { + "epoch": 0.2886492039034412, + "grad_norm": 3.2689175605773926, + "learning_rate": 4.554794520547945e-06, + "loss": 1.7747, + "step": 281 + }, + { + "epoch": 0.2896764252696456, + "grad_norm": 2.8242290019989014, + "learning_rate": 4.571917808219179e-06, + "loss": 1.7657, + "step": 282 + }, + { + "epoch": 0.29070364663585, + "grad_norm": 3.03828501701355, + "learning_rate": 4.589041095890411e-06, + "loss": 1.7555, + "step": 283 + }, + { + "epoch": 0.29173086800205444, + "grad_norm": 2.714351177215576, + "learning_rate": 4.606164383561644e-06, + "loss": 1.7055, + "step": 284 + }, + { + "epoch": 0.29275808936825887, + "grad_norm": 2.9540348052978516, + "learning_rate": 4.6232876712328774e-06, + "loss": 1.7258, + "step": 285 + }, + { + "epoch": 0.2937853107344633, + "grad_norm": 3.107060670852661, + "learning_rate": 4.64041095890411e-06, + "loss": 1.7376, + "step": 286 + }, + { + "epoch": 0.2948125321006677, + "grad_norm": 3.3266639709472656, + "learning_rate": 4.657534246575343e-06, + "loss": 1.7063, + "step": 287 + }, + { + "epoch": 0.29583975346687214, + "grad_norm": 3.3113174438476562, + "learning_rate": 4.674657534246575e-06, + "loss": 1.7514, + "step": 288 + }, + { + "epoch": 0.2968669748330765, + "grad_norm": 3.313321352005005, + "learning_rate": 4.691780821917809e-06, + "loss": 1.73, + "step": 289 + }, + { + "epoch": 0.2978941961992809, + "grad_norm": 2.901522159576416, + "learning_rate": 4.7089041095890414e-06, + "loss": 1.7709, + "step": 290 + }, + { + "epoch": 0.2978941961992809, + "eval_loss": 1.9608837366104126, + "eval_runtime": 37.3687, + "eval_samples_per_second": 26.76, + "eval_steps_per_second": 1.124, + "step": 290 + }, + { + "epoch": 0.29892141756548535, + "grad_norm": 3.4208264350891113, + "learning_rate": 4.726027397260274e-06, + "loss": 1.73, + "step": 291 + }, + { + "epoch": 0.29994863893168977, + "grad_norm": 2.8982441425323486, + "learning_rate": 4.7431506849315075e-06, + "loss": 1.7416, + "step": 292 + }, + { + "epoch": 0.3009758602978942, + "grad_norm": 3.2743732929229736, + "learning_rate": 4.76027397260274e-06, + "loss": 1.7443, + "step": 293 + }, + { + "epoch": 0.3020030816640986, + "grad_norm": 3.028764486312866, + "learning_rate": 4.777397260273973e-06, + "loss": 1.6879, + "step": 294 + }, + { + "epoch": 0.30303030303030304, + "grad_norm": 3.3531785011291504, + "learning_rate": 4.7945205479452054e-06, + "loss": 1.7096, + "step": 295 + }, + { + "epoch": 0.30405752439650746, + "grad_norm": 2.5602540969848633, + "learning_rate": 4.811643835616439e-06, + "loss": 1.7125, + "step": 296 + }, + { + "epoch": 0.3050847457627119, + "grad_norm": 3.1823129653930664, + "learning_rate": 4.8287671232876716e-06, + "loss": 1.6531, + "step": 297 + }, + { + "epoch": 0.3061119671289163, + "grad_norm": 2.6700282096862793, + "learning_rate": 4.845890410958904e-06, + "loss": 1.6983, + "step": 298 + }, + { + "epoch": 0.3071391884951207, + "grad_norm": 2.7118732929229736, + "learning_rate": 4.863013698630138e-06, + "loss": 1.7056, + "step": 299 + }, + { + "epoch": 0.3081664098613251, + "grad_norm": 3.3493311405181885, + "learning_rate": 4.88013698630137e-06, + "loss": 1.6944, + "step": 300 + }, + { + "epoch": 0.3081664098613251, + "eval_loss": 1.9489911794662476, + "eval_runtime": 37.35, + "eval_samples_per_second": 26.774, + "eval_steps_per_second": 1.124, + "step": 300 + }, + { + "epoch": 0.3091936312275295, + "grad_norm": 2.9436538219451904, + "learning_rate": 4.897260273972603e-06, + "loss": 1.6717, + "step": 301 + }, + { + "epoch": 0.31022085259373394, + "grad_norm": 2.894737482070923, + "learning_rate": 4.914383561643836e-06, + "loss": 1.7895, + "step": 302 + }, + { + "epoch": 0.31124807395993837, + "grad_norm": 2.8645665645599365, + "learning_rate": 4.931506849315069e-06, + "loss": 1.721, + "step": 303 + }, + { + "epoch": 0.3122752953261428, + "grad_norm": 3.050203323364258, + "learning_rate": 4.948630136986302e-06, + "loss": 1.7564, + "step": 304 + }, + { + "epoch": 0.3133025166923472, + "grad_norm": 3.146395206451416, + "learning_rate": 4.965753424657534e-06, + "loss": 1.7938, + "step": 305 + }, + { + "epoch": 0.31432973805855163, + "grad_norm": 3.1021251678466797, + "learning_rate": 4.982876712328768e-06, + "loss": 1.6682, + "step": 306 + }, + { + "epoch": 0.31535695942475606, + "grad_norm": 2.7301125526428223, + "learning_rate": 5e-06, + "loss": 1.6974, + "step": 307 + }, + { + "epoch": 0.3163841807909605, + "grad_norm": 2.7602250576019287, + "learning_rate": 4.998096688237534e-06, + "loss": 1.6523, + "step": 308 + }, + { + "epoch": 0.31741140215716485, + "grad_norm": 2.8149876594543457, + "learning_rate": 4.996193376475067e-06, + "loss": 1.7535, + "step": 309 + }, + { + "epoch": 0.31843862352336927, + "grad_norm": 2.54823637008667, + "learning_rate": 4.9942900647126005e-06, + "loss": 1.63, + "step": 310 + }, + { + "epoch": 0.31843862352336927, + "eval_loss": 1.928893804550171, + "eval_runtime": 37.2236, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3194658448895737, + "grad_norm": 2.7819766998291016, + "learning_rate": 4.992386752950134e-06, + "loss": 1.6461, + "step": 311 + }, + { + "epoch": 0.3204930662557781, + "grad_norm": 2.999209403991699, + "learning_rate": 4.990483441187667e-06, + "loss": 1.6861, + "step": 312 + }, + { + "epoch": 0.32152028762198254, + "grad_norm": 2.7288851737976074, + "learning_rate": 4.9885801294252e-06, + "loss": 1.7076, + "step": 313 + }, + { + "epoch": 0.32254750898818696, + "grad_norm": 2.8722457885742188, + "learning_rate": 4.986676817662733e-06, + "loss": 1.7079, + "step": 314 + }, + { + "epoch": 0.3235747303543914, + "grad_norm": 2.853344678878784, + "learning_rate": 4.9847735059002665e-06, + "loss": 1.6699, + "step": 315 + }, + { + "epoch": 0.3246019517205958, + "grad_norm": 2.7404720783233643, + "learning_rate": 4.9828701941378e-06, + "loss": 1.7423, + "step": 316 + }, + { + "epoch": 0.3256291730868002, + "grad_norm": 2.6062633991241455, + "learning_rate": 4.980966882375333e-06, + "loss": 1.6811, + "step": 317 + }, + { + "epoch": 0.3266563944530046, + "grad_norm": 2.834162712097168, + "learning_rate": 4.979063570612867e-06, + "loss": 1.6575, + "step": 318 + }, + { + "epoch": 0.327683615819209, + "grad_norm": 2.7741048336029053, + "learning_rate": 4.9771602588504e-06, + "loss": 1.6158, + "step": 319 + }, + { + "epoch": 0.32871083718541344, + "grad_norm": 2.6216888427734375, + "learning_rate": 4.975256947087933e-06, + "loss": 1.7295, + "step": 320 + }, + { + "epoch": 0.32871083718541344, + "eval_loss": 1.9309543371200562, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 320 + }, + { + "epoch": 0.32973805855161786, + "grad_norm": 2.6282548904418945, + "learning_rate": 4.973353635325467e-06, + "loss": 1.6266, + "step": 321 + }, + { + "epoch": 0.3307652799178223, + "grad_norm": 2.632124662399292, + "learning_rate": 4.971450323563e-06, + "loss": 1.6887, + "step": 322 + }, + { + "epoch": 0.3317925012840267, + "grad_norm": 2.5844857692718506, + "learning_rate": 4.969547011800533e-06, + "loss": 1.6287, + "step": 323 + }, + { + "epoch": 0.33281972265023113, + "grad_norm": 2.597620964050293, + "learning_rate": 4.967643700038067e-06, + "loss": 1.6876, + "step": 324 + }, + { + "epoch": 0.33384694401643555, + "grad_norm": 2.6529343128204346, + "learning_rate": 4.9657403882756e-06, + "loss": 1.6883, + "step": 325 + }, + { + "epoch": 0.33487416538264, + "grad_norm": 2.592874526977539, + "learning_rate": 4.9638370765131335e-06, + "loss": 1.653, + "step": 326 + }, + { + "epoch": 0.3359013867488444, + "grad_norm": 2.523489236831665, + "learning_rate": 4.961933764750667e-06, + "loss": 1.6501, + "step": 327 + }, + { + "epoch": 0.33692860811504877, + "grad_norm": 2.6721699237823486, + "learning_rate": 4.9600304529882e-06, + "loss": 1.7126, + "step": 328 + }, + { + "epoch": 0.3379558294812532, + "grad_norm": 2.513122797012329, + "learning_rate": 4.9581271412257335e-06, + "loss": 1.7268, + "step": 329 + }, + { + "epoch": 0.3389830508474576, + "grad_norm": 2.571118116378784, + "learning_rate": 4.956223829463266e-06, + "loss": 1.6862, + "step": 330 + }, + { + "epoch": 0.3389830508474576, + "eval_loss": 1.9501315355300903, + "eval_runtime": 37.3287, + "eval_samples_per_second": 26.789, + "eval_steps_per_second": 1.125, + "step": 330 + }, + { + "epoch": 0.34001027221366203, + "grad_norm": 2.5783371925354004, + "learning_rate": 4.954320517700799e-06, + "loss": 1.6041, + "step": 331 + }, + { + "epoch": 0.34103749357986646, + "grad_norm": 2.6829233169555664, + "learning_rate": 4.952417205938333e-06, + "loss": 1.7111, + "step": 332 + }, + { + "epoch": 0.3420647149460709, + "grad_norm": 2.6643247604370117, + "learning_rate": 4.950513894175866e-06, + "loss": 1.6797, + "step": 333 + }, + { + "epoch": 0.3430919363122753, + "grad_norm": 2.776308298110962, + "learning_rate": 4.9486105824133995e-06, + "loss": 1.6766, + "step": 334 + }, + { + "epoch": 0.3441191576784797, + "grad_norm": 2.722667932510376, + "learning_rate": 4.946707270650933e-06, + "loss": 1.6865, + "step": 335 + }, + { + "epoch": 0.34514637904468415, + "grad_norm": 2.7050845623016357, + "learning_rate": 4.944803958888466e-06, + "loss": 1.6967, + "step": 336 + }, + { + "epoch": 0.34617360041088857, + "grad_norm": 3.0009098052978516, + "learning_rate": 4.942900647126e-06, + "loss": 1.714, + "step": 337 + }, + { + "epoch": 0.34720082177709294, + "grad_norm": 2.793621301651001, + "learning_rate": 4.940997335363533e-06, + "loss": 1.6591, + "step": 338 + }, + { + "epoch": 0.34822804314329736, + "grad_norm": 2.844383478164673, + "learning_rate": 4.939094023601066e-06, + "loss": 1.611, + "step": 339 + }, + { + "epoch": 0.3492552645095018, + "grad_norm": 2.829378366470337, + "learning_rate": 4.9371907118386e-06, + "loss": 1.666, + "step": 340 + }, + { + "epoch": 0.3492552645095018, + "eval_loss": 1.96309494972229, + "eval_runtime": 37.3346, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 340 + }, + { + "epoch": 0.3502824858757062, + "grad_norm": 2.9702041149139404, + "learning_rate": 4.935287400076133e-06, + "loss": 1.6703, + "step": 341 + }, + { + "epoch": 0.35130970724191063, + "grad_norm": 2.7555935382843018, + "learning_rate": 4.933384088313666e-06, + "loss": 1.6879, + "step": 342 + }, + { + "epoch": 0.35233692860811505, + "grad_norm": 2.8968169689178467, + "learning_rate": 4.9314807765512e-06, + "loss": 1.6342, + "step": 343 + }, + { + "epoch": 0.3533641499743195, + "grad_norm": 3.010115385055542, + "learning_rate": 4.929577464788733e-06, + "loss": 1.5883, + "step": 344 + }, + { + "epoch": 0.3543913713405239, + "grad_norm": 2.6449785232543945, + "learning_rate": 4.9276741530262665e-06, + "loss": 1.654, + "step": 345 + }, + { + "epoch": 0.3554185927067283, + "grad_norm": 2.976109266281128, + "learning_rate": 4.9257708412638e-06, + "loss": 1.6483, + "step": 346 + }, + { + "epoch": 0.35644581407293274, + "grad_norm": 2.856144666671753, + "learning_rate": 4.923867529501332e-06, + "loss": 1.6536, + "step": 347 + }, + { + "epoch": 0.3574730354391371, + "grad_norm": 2.8142166137695312, + "learning_rate": 4.921964217738866e-06, + "loss": 1.6415, + "step": 348 + }, + { + "epoch": 0.35850025680534153, + "grad_norm": 2.70807147026062, + "learning_rate": 4.920060905976399e-06, + "loss": 1.7099, + "step": 349 + }, + { + "epoch": 0.35952747817154596, + "grad_norm": 2.7715506553649902, + "learning_rate": 4.9181575942139325e-06, + "loss": 1.6947, + "step": 350 + }, + { + "epoch": 0.35952747817154596, + "eval_loss": 1.9507063627243042, + "eval_runtime": 37.3094, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 350 + }, + { + "epoch": 0.3605546995377504, + "grad_norm": 2.8816769123077393, + "learning_rate": 4.916254282451466e-06, + "loss": 1.6906, + "step": 351 + }, + { + "epoch": 0.3615819209039548, + "grad_norm": 3.08797025680542, + "learning_rate": 4.914350970688999e-06, + "loss": 1.7387, + "step": 352 + }, + { + "epoch": 0.3626091422701592, + "grad_norm": 2.6018118858337402, + "learning_rate": 4.9124476589265325e-06, + "loss": 1.6319, + "step": 353 + }, + { + "epoch": 0.36363636363636365, + "grad_norm": 2.7790255546569824, + "learning_rate": 4.910544347164066e-06, + "loss": 1.6357, + "step": 354 + }, + { + "epoch": 0.36466358500256807, + "grad_norm": 2.7360289096832275, + "learning_rate": 4.908641035401599e-06, + "loss": 1.7127, + "step": 355 + }, + { + "epoch": 0.3656908063687725, + "grad_norm": 2.6693854331970215, + "learning_rate": 4.906737723639133e-06, + "loss": 1.7747, + "step": 356 + }, + { + "epoch": 0.3667180277349769, + "grad_norm": 2.642868995666504, + "learning_rate": 4.904834411876665e-06, + "loss": 1.6401, + "step": 357 + }, + { + "epoch": 0.3677452491011813, + "grad_norm": 2.7230374813079834, + "learning_rate": 4.902931100114199e-06, + "loss": 1.6833, + "step": 358 + }, + { + "epoch": 0.3687724704673857, + "grad_norm": 2.499457359313965, + "learning_rate": 4.901027788351733e-06, + "loss": 1.6007, + "step": 359 + }, + { + "epoch": 0.3697996918335901, + "grad_norm": 2.7695276737213135, + "learning_rate": 4.899124476589266e-06, + "loss": 1.6486, + "step": 360 + }, + { + "epoch": 0.3697996918335901, + "eval_loss": 1.9434194564819336, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.37082691319979455, + "grad_norm": 2.5715794563293457, + "learning_rate": 4.8972211648267994e-06, + "loss": 1.6183, + "step": 361 + }, + { + "epoch": 0.371854134565999, + "grad_norm": 2.637859582901001, + "learning_rate": 4.895317853064333e-06, + "loss": 1.6648, + "step": 362 + }, + { + "epoch": 0.3728813559322034, + "grad_norm": 2.5601117610931396, + "learning_rate": 4.893414541301866e-06, + "loss": 1.6475, + "step": 363 + }, + { + "epoch": 0.3739085772984078, + "grad_norm": 2.429337739944458, + "learning_rate": 4.891511229539399e-06, + "loss": 1.6435, + "step": 364 + }, + { + "epoch": 0.37493579866461224, + "grad_norm": 2.6890885829925537, + "learning_rate": 4.889607917776932e-06, + "loss": 1.6577, + "step": 365 + }, + { + "epoch": 0.37596302003081666, + "grad_norm": 2.55492901802063, + "learning_rate": 4.887704606014465e-06, + "loss": 1.6523, + "step": 366 + }, + { + "epoch": 0.37699024139702103, + "grad_norm": 2.841362476348877, + "learning_rate": 4.885801294251999e-06, + "loss": 1.6742, + "step": 367 + }, + { + "epoch": 0.37801746276322545, + "grad_norm": 2.8041367530822754, + "learning_rate": 4.883897982489532e-06, + "loss": 1.5946, + "step": 368 + }, + { + "epoch": 0.3790446841294299, + "grad_norm": 2.7135822772979736, + "learning_rate": 4.8819946707270655e-06, + "loss": 1.6909, + "step": 369 + }, + { + "epoch": 0.3800719054956343, + "grad_norm": 2.9020211696624756, + "learning_rate": 4.880091358964599e-06, + "loss": 1.6804, + "step": 370 + }, + { + "epoch": 0.3800719054956343, + "eval_loss": 1.9353317022323608, + "eval_runtime": 37.3188, + "eval_samples_per_second": 26.796, + "eval_steps_per_second": 1.125, + "step": 370 + }, + { + "epoch": 0.3810991268618387, + "grad_norm": 2.567310333251953, + "learning_rate": 4.878188047202132e-06, + "loss": 1.5885, + "step": 371 + }, + { + "epoch": 0.38212634822804314, + "grad_norm": 2.910834789276123, + "learning_rate": 4.876284735439666e-06, + "loss": 1.7681, + "step": 372 + }, + { + "epoch": 0.38315356959424757, + "grad_norm": 2.7507543563842773, + "learning_rate": 4.874381423677199e-06, + "loss": 1.7228, + "step": 373 + }, + { + "epoch": 0.384180790960452, + "grad_norm": 2.9230854511260986, + "learning_rate": 4.8724781119147315e-06, + "loss": 1.6491, + "step": 374 + }, + { + "epoch": 0.3852080123266564, + "grad_norm": 2.608980178833008, + "learning_rate": 4.870574800152265e-06, + "loss": 1.6757, + "step": 375 + }, + { + "epoch": 0.38623523369286084, + "grad_norm": 2.9557063579559326, + "learning_rate": 4.868671488389798e-06, + "loss": 1.607, + "step": 376 + }, + { + "epoch": 0.3872624550590652, + "grad_norm": 2.982243537902832, + "learning_rate": 4.8667681766273316e-06, + "loss": 1.6169, + "step": 377 + }, + { + "epoch": 0.3882896764252696, + "grad_norm": 2.775012254714966, + "learning_rate": 4.864864864864866e-06, + "loss": 1.5835, + "step": 378 + }, + { + "epoch": 0.38931689779147405, + "grad_norm": 2.9500741958618164, + "learning_rate": 4.862961553102399e-06, + "loss": 1.6504, + "step": 379 + }, + { + "epoch": 0.39034411915767847, + "grad_norm": 2.889702796936035, + "learning_rate": 4.8610582413399325e-06, + "loss": 1.6814, + "step": 380 + }, + { + "epoch": 0.39034411915767847, + "eval_loss": 1.9426418542861938, + "eval_runtime": 37.3276, + "eval_samples_per_second": 26.79, + "eval_steps_per_second": 1.125, + "step": 380 + }, + { + "epoch": 0.3913713405238829, + "grad_norm": 2.5075154304504395, + "learning_rate": 4.859154929577465e-06, + "loss": 1.6368, + "step": 381 + }, + { + "epoch": 0.3923985618900873, + "grad_norm": 2.6461148262023926, + "learning_rate": 4.857251617814998e-06, + "loss": 1.6526, + "step": 382 + }, + { + "epoch": 0.39342578325629174, + "grad_norm": 2.681501626968384, + "learning_rate": 4.855348306052532e-06, + "loss": 1.6846, + "step": 383 + }, + { + "epoch": 0.39445300462249616, + "grad_norm": 2.557311773300171, + "learning_rate": 4.853444994290065e-06, + "loss": 1.6645, + "step": 384 + }, + { + "epoch": 0.3954802259887006, + "grad_norm": 2.6819186210632324, + "learning_rate": 4.8515416825275985e-06, + "loss": 1.6323, + "step": 385 + }, + { + "epoch": 0.396507447354905, + "grad_norm": 2.8803794384002686, + "learning_rate": 4.849638370765132e-06, + "loss": 1.6991, + "step": 386 + }, + { + "epoch": 0.3975346687211094, + "grad_norm": 2.7181358337402344, + "learning_rate": 4.847735059002665e-06, + "loss": 1.6256, + "step": 387 + }, + { + "epoch": 0.3985618900873138, + "grad_norm": 2.5668489933013916, + "learning_rate": 4.8458317472401985e-06, + "loss": 1.58, + "step": 388 + }, + { + "epoch": 0.3995891114535182, + "grad_norm": 2.708804130554199, + "learning_rate": 4.843928435477732e-06, + "loss": 1.67, + "step": 389 + }, + { + "epoch": 0.40061633281972264, + "grad_norm": 2.6565728187561035, + "learning_rate": 4.842025123715265e-06, + "loss": 1.622, + "step": 390 + }, + { + "epoch": 0.40061633281972264, + "eval_loss": 1.9529728889465332, + "eval_runtime": 37.3489, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 390 + }, + { + "epoch": 0.40164355418592707, + "grad_norm": 2.8743324279785156, + "learning_rate": 4.840121811952798e-06, + "loss": 1.6618, + "step": 391 + }, + { + "epoch": 0.4026707755521315, + "grad_norm": 2.86134672164917, + "learning_rate": 4.838218500190331e-06, + "loss": 1.6544, + "step": 392 + }, + { + "epoch": 0.4036979969183359, + "grad_norm": 3.3492071628570557, + "learning_rate": 4.8363151884278645e-06, + "loss": 1.6683, + "step": 393 + }, + { + "epoch": 0.40472521828454033, + "grad_norm": 2.6630825996398926, + "learning_rate": 4.834411876665398e-06, + "loss": 1.6447, + "step": 394 + }, + { + "epoch": 0.40575243965074476, + "grad_norm": 2.820096492767334, + "learning_rate": 4.832508564902931e-06, + "loss": 1.6579, + "step": 395 + }, + { + "epoch": 0.4067796610169492, + "grad_norm": 3.076234817504883, + "learning_rate": 4.830605253140465e-06, + "loss": 1.6039, + "step": 396 + }, + { + "epoch": 0.40780688238315355, + "grad_norm": 2.667719841003418, + "learning_rate": 4.828701941377998e-06, + "loss": 1.6364, + "step": 397 + }, + { + "epoch": 0.40883410374935797, + "grad_norm": 3.2132771015167236, + "learning_rate": 4.826798629615531e-06, + "loss": 1.5906, + "step": 398 + }, + { + "epoch": 0.4098613251155624, + "grad_norm": 2.661484479904175, + "learning_rate": 4.824895317853065e-06, + "loss": 1.613, + "step": 399 + }, + { + "epoch": 0.4108885464817668, + "grad_norm": 3.002636671066284, + "learning_rate": 4.822992006090598e-06, + "loss": 1.6305, + "step": 400 + }, + { + "epoch": 0.4108885464817668, + "eval_loss": 1.9345510005950928, + "eval_runtime": 37.3369, + "eval_samples_per_second": 26.783, + "eval_steps_per_second": 1.125, + "step": 400 + }, + { + "epoch": 0.41191576784797124, + "grad_norm": 2.6405227184295654, + "learning_rate": 4.821088694328131e-06, + "loss": 1.5881, + "step": 401 + }, + { + "epoch": 0.41294298921417566, + "grad_norm": 3.006169080734253, + "learning_rate": 4.819185382565665e-06, + "loss": 1.6126, + "step": 402 + }, + { + "epoch": 0.4139702105803801, + "grad_norm": 2.7614858150482178, + "learning_rate": 4.817282070803198e-06, + "loss": 1.6695, + "step": 403 + }, + { + "epoch": 0.4149974319465845, + "grad_norm": 2.797260046005249, + "learning_rate": 4.8153787590407315e-06, + "loss": 1.6539, + "step": 404 + }, + { + "epoch": 0.41602465331278893, + "grad_norm": 2.8156301975250244, + "learning_rate": 4.813475447278265e-06, + "loss": 1.5845, + "step": 405 + }, + { + "epoch": 0.41705187467899335, + "grad_norm": 2.74100399017334, + "learning_rate": 4.811572135515798e-06, + "loss": 1.6699, + "step": 406 + }, + { + "epoch": 0.4180790960451977, + "grad_norm": 2.7785775661468506, + "learning_rate": 4.809668823753332e-06, + "loss": 1.5987, + "step": 407 + }, + { + "epoch": 0.41910631741140214, + "grad_norm": 2.663504123687744, + "learning_rate": 4.807765511990864e-06, + "loss": 1.5987, + "step": 408 + }, + { + "epoch": 0.42013353877760656, + "grad_norm": 2.903374433517456, + "learning_rate": 4.8058622002283975e-06, + "loss": 1.6302, + "step": 409 + }, + { + "epoch": 0.421160760143811, + "grad_norm": 2.696049690246582, + "learning_rate": 4.803958888465931e-06, + "loss": 1.671, + "step": 410 + }, + { + "epoch": 0.421160760143811, + "eval_loss": 1.9333878755569458, + "eval_runtime": 37.2472, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 410 + }, + { + "epoch": 0.4221879815100154, + "grad_norm": 2.486461877822876, + "learning_rate": 4.802055576703464e-06, + "loss": 1.6282, + "step": 411 + }, + { + "epoch": 0.42321520287621983, + "grad_norm": 2.6812057495117188, + "learning_rate": 4.8001522649409976e-06, + "loss": 1.7144, + "step": 412 + }, + { + "epoch": 0.42424242424242425, + "grad_norm": 2.8370249271392822, + "learning_rate": 4.798248953178531e-06, + "loss": 1.5979, + "step": 413 + }, + { + "epoch": 0.4252696456086287, + "grad_norm": 2.6073036193847656, + "learning_rate": 4.796345641416064e-06, + "loss": 1.5703, + "step": 414 + }, + { + "epoch": 0.4262968669748331, + "grad_norm": 2.5891220569610596, + "learning_rate": 4.794442329653598e-06, + "loss": 1.6212, + "step": 415 + }, + { + "epoch": 0.42732408834103747, + "grad_norm": 2.6296865940093994, + "learning_rate": 4.792539017891131e-06, + "loss": 1.6711, + "step": 416 + }, + { + "epoch": 0.4283513097072419, + "grad_norm": 2.573472499847412, + "learning_rate": 4.790635706128664e-06, + "loss": 1.5994, + "step": 417 + }, + { + "epoch": 0.4293785310734463, + "grad_norm": 2.734464168548584, + "learning_rate": 4.788732394366197e-06, + "loss": 1.6383, + "step": 418 + }, + { + "epoch": 0.43040575243965074, + "grad_norm": 2.6412851810455322, + "learning_rate": 4.786829082603731e-06, + "loss": 1.5646, + "step": 419 + }, + { + "epoch": 0.43143297380585516, + "grad_norm": 2.5065555572509766, + "learning_rate": 4.7849257708412645e-06, + "loss": 1.5636, + "step": 420 + }, + { + "epoch": 0.43143297380585516, + "eval_loss": 1.9357348680496216, + "eval_runtime": 37.372, + "eval_samples_per_second": 26.758, + "eval_steps_per_second": 1.124, + "step": 420 + }, + { + "epoch": 0.4324601951720596, + "grad_norm": 2.4889843463897705, + "learning_rate": 4.783022459078798e-06, + "loss": 1.6457, + "step": 421 + }, + { + "epoch": 0.433487416538264, + "grad_norm": 2.544801950454712, + "learning_rate": 4.781119147316331e-06, + "loss": 1.5897, + "step": 422 + }, + { + "epoch": 0.4345146379044684, + "grad_norm": 2.7834064960479736, + "learning_rate": 4.7792158355538645e-06, + "loss": 1.6174, + "step": 423 + }, + { + "epoch": 0.43554185927067285, + "grad_norm": 2.602952480316162, + "learning_rate": 4.777312523791398e-06, + "loss": 1.6152, + "step": 424 + }, + { + "epoch": 0.43656908063687727, + "grad_norm": 2.629746198654175, + "learning_rate": 4.77540921202893e-06, + "loss": 1.5991, + "step": 425 + }, + { + "epoch": 0.43759630200308164, + "grad_norm": 2.4987270832061768, + "learning_rate": 4.773505900266464e-06, + "loss": 1.6342, + "step": 426 + }, + { + "epoch": 0.43862352336928606, + "grad_norm": 2.607611894607544, + "learning_rate": 4.771602588503997e-06, + "loss": 1.6145, + "step": 427 + }, + { + "epoch": 0.4396507447354905, + "grad_norm": 2.5654876232147217, + "learning_rate": 4.7696992767415305e-06, + "loss": 1.6162, + "step": 428 + }, + { + "epoch": 0.4406779661016949, + "grad_norm": 2.5701870918273926, + "learning_rate": 4.767795964979064e-06, + "loss": 1.6516, + "step": 429 + }, + { + "epoch": 0.44170518746789933, + "grad_norm": 2.8381104469299316, + "learning_rate": 4.765892653216597e-06, + "loss": 1.659, + "step": 430 + }, + { + "epoch": 0.44170518746789933, + "eval_loss": 1.9259562492370605, + "eval_runtime": 37.3906, + "eval_samples_per_second": 26.745, + "eval_steps_per_second": 1.123, + "step": 430 + }, + { + "epoch": 0.44273240883410375, + "grad_norm": 2.5777862071990967, + "learning_rate": 4.763989341454131e-06, + "loss": 1.6571, + "step": 431 + }, + { + "epoch": 0.4437596302003082, + "grad_norm": 2.6656861305236816, + "learning_rate": 4.762086029691664e-06, + "loss": 1.5786, + "step": 432 + }, + { + "epoch": 0.4447868515665126, + "grad_norm": 2.724851131439209, + "learning_rate": 4.760182717929197e-06, + "loss": 1.5916, + "step": 433 + }, + { + "epoch": 0.445814072932717, + "grad_norm": 2.532191514968872, + "learning_rate": 4.758279406166731e-06, + "loss": 1.608, + "step": 434 + }, + { + "epoch": 0.44684129429892144, + "grad_norm": 2.6252050399780273, + "learning_rate": 4.756376094404263e-06, + "loss": 1.5297, + "step": 435 + }, + { + "epoch": 0.4478685156651258, + "grad_norm": 2.674530029296875, + "learning_rate": 4.7544727826417966e-06, + "loss": 1.615, + "step": 436 + }, + { + "epoch": 0.44889573703133023, + "grad_norm": 2.9209136962890625, + "learning_rate": 4.75256947087933e-06, + "loss": 1.6749, + "step": 437 + }, + { + "epoch": 0.44992295839753466, + "grad_norm": 2.4959394931793213, + "learning_rate": 4.750666159116863e-06, + "loss": 1.6186, + "step": 438 + }, + { + "epoch": 0.4509501797637391, + "grad_norm": 2.6317503452301025, + "learning_rate": 4.7487628473543975e-06, + "loss": 1.5873, + "step": 439 + }, + { + "epoch": 0.4519774011299435, + "grad_norm": 2.6485610008239746, + "learning_rate": 4.746859535591931e-06, + "loss": 1.5451, + "step": 440 + }, + { + "epoch": 0.4519774011299435, + "eval_loss": 1.9235283136367798, + "eval_runtime": 37.367, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 440 + }, + { + "epoch": 0.4530046224961479, + "grad_norm": 2.491852045059204, + "learning_rate": 4.744956223829464e-06, + "loss": 1.6016, + "step": 441 + }, + { + "epoch": 0.45403184386235235, + "grad_norm": 2.5354514122009277, + "learning_rate": 4.743052912066997e-06, + "loss": 1.5795, + "step": 442 + }, + { + "epoch": 0.45505906522855677, + "grad_norm": 2.7756357192993164, + "learning_rate": 4.74114960030453e-06, + "loss": 1.6299, + "step": 443 + }, + { + "epoch": 0.4560862865947612, + "grad_norm": 2.638906955718994, + "learning_rate": 4.7392462885420635e-06, + "loss": 1.6429, + "step": 444 + }, + { + "epoch": 0.4571135079609656, + "grad_norm": 2.5858829021453857, + "learning_rate": 4.737342976779597e-06, + "loss": 1.5316, + "step": 445 + }, + { + "epoch": 0.45814072932717, + "grad_norm": 2.798567295074463, + "learning_rate": 4.73543966501713e-06, + "loss": 1.6166, + "step": 446 + }, + { + "epoch": 0.4591679506933744, + "grad_norm": 2.512143135070801, + "learning_rate": 4.7335363532546636e-06, + "loss": 1.5952, + "step": 447 + }, + { + "epoch": 0.4601951720595788, + "grad_norm": 2.7063655853271484, + "learning_rate": 4.731633041492197e-06, + "loss": 1.6051, + "step": 448 + }, + { + "epoch": 0.46122239342578325, + "grad_norm": 2.6358017921447754, + "learning_rate": 4.72972972972973e-06, + "loss": 1.6008, + "step": 449 + }, + { + "epoch": 0.4622496147919877, + "grad_norm": 2.75992751121521, + "learning_rate": 4.727826417967264e-06, + "loss": 1.644, + "step": 450 + }, + { + "epoch": 0.4622496147919877, + "eval_loss": 1.927345633506775, + "eval_runtime": 37.3707, + "eval_samples_per_second": 26.759, + "eval_steps_per_second": 1.124, + "step": 450 + }, + { + "epoch": 0.4632768361581921, + "grad_norm": 2.9103775024414062, + "learning_rate": 4.725923106204797e-06, + "loss": 1.6176, + "step": 451 + }, + { + "epoch": 0.4643040575243965, + "grad_norm": 2.627481460571289, + "learning_rate": 4.72401979444233e-06, + "loss": 1.586, + "step": 452 + }, + { + "epoch": 0.46533127889060094, + "grad_norm": 2.8432116508483887, + "learning_rate": 4.722116482679863e-06, + "loss": 1.6303, + "step": 453 + }, + { + "epoch": 0.46635850025680536, + "grad_norm": 2.8088958263397217, + "learning_rate": 4.720213170917396e-06, + "loss": 1.6163, + "step": 454 + }, + { + "epoch": 0.46738572162300973, + "grad_norm": 2.6040408611297607, + "learning_rate": 4.71830985915493e-06, + "loss": 1.6346, + "step": 455 + }, + { + "epoch": 0.46841294298921415, + "grad_norm": 2.8951191902160645, + "learning_rate": 4.716406547392463e-06, + "loss": 1.5859, + "step": 456 + }, + { + "epoch": 0.4694401643554186, + "grad_norm": 2.792473793029785, + "learning_rate": 4.714503235629996e-06, + "loss": 1.6441, + "step": 457 + }, + { + "epoch": 0.470467385721623, + "grad_norm": 2.6793878078460693, + "learning_rate": 4.71259992386753e-06, + "loss": 1.5699, + "step": 458 + }, + { + "epoch": 0.4714946070878274, + "grad_norm": 2.7223684787750244, + "learning_rate": 4.710696612105063e-06, + "loss": 1.6038, + "step": 459 + }, + { + "epoch": 0.47252182845403184, + "grad_norm": 2.8043856620788574, + "learning_rate": 4.708793300342596e-06, + "loss": 1.6018, + "step": 460 + }, + { + "epoch": 0.47252182845403184, + "eval_loss": 1.9213919639587402, + "eval_runtime": 37.3451, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 460 + }, + { + "epoch": 0.47354904982023627, + "grad_norm": 2.7657470703125, + "learning_rate": 4.70688998858013e-06, + "loss": 1.5896, + "step": 461 + }, + { + "epoch": 0.4745762711864407, + "grad_norm": 2.8417739868164062, + "learning_rate": 4.704986676817663e-06, + "loss": 1.5892, + "step": 462 + }, + { + "epoch": 0.4756034925526451, + "grad_norm": 2.5660197734832764, + "learning_rate": 4.7030833650551965e-06, + "loss": 1.615, + "step": 463 + }, + { + "epoch": 0.47663071391884954, + "grad_norm": 2.762880325317383, + "learning_rate": 4.70118005329273e-06, + "loss": 1.5747, + "step": 464 + }, + { + "epoch": 0.4776579352850539, + "grad_norm": 2.7191319465637207, + "learning_rate": 4.699276741530263e-06, + "loss": 1.6529, + "step": 465 + }, + { + "epoch": 0.4786851566512583, + "grad_norm": 2.4995453357696533, + "learning_rate": 4.697373429767797e-06, + "loss": 1.6285, + "step": 466 + }, + { + "epoch": 0.47971237801746275, + "grad_norm": 2.8300375938415527, + "learning_rate": 4.69547011800533e-06, + "loss": 1.6171, + "step": 467 + }, + { + "epoch": 0.48073959938366717, + "grad_norm": 2.6028811931610107, + "learning_rate": 4.693566806242863e-06, + "loss": 1.6014, + "step": 468 + }, + { + "epoch": 0.4817668207498716, + "grad_norm": 3.080156087875366, + "learning_rate": 4.691663494480397e-06, + "loss": 1.6055, + "step": 469 + }, + { + "epoch": 0.482794042116076, + "grad_norm": 2.6707916259765625, + "learning_rate": 4.689760182717929e-06, + "loss": 1.627, + "step": 470 + }, + { + "epoch": 0.482794042116076, + "eval_loss": 1.9015462398529053, + "eval_runtime": 37.3476, + "eval_samples_per_second": 26.776, + "eval_steps_per_second": 1.125, + "step": 470 + }, + { + "epoch": 0.48382126348228044, + "grad_norm": 2.6820614337921143, + "learning_rate": 4.6878568709554626e-06, + "loss": 1.5261, + "step": 471 + }, + { + "epoch": 0.48484848484848486, + "grad_norm": 2.841031789779663, + "learning_rate": 4.685953559192996e-06, + "loss": 1.5903, + "step": 472 + }, + { + "epoch": 0.4858757062146893, + "grad_norm": 2.470867156982422, + "learning_rate": 4.684050247430529e-06, + "loss": 1.6091, + "step": 473 + }, + { + "epoch": 0.4869029275808937, + "grad_norm": 2.4955153465270996, + "learning_rate": 4.682146935668063e-06, + "loss": 1.544, + "step": 474 + }, + { + "epoch": 0.4879301489470981, + "grad_norm": 2.9290695190429688, + "learning_rate": 4.680243623905596e-06, + "loss": 1.5893, + "step": 475 + }, + { + "epoch": 0.4889573703133025, + "grad_norm": 2.5902767181396484, + "learning_rate": 4.678340312143129e-06, + "loss": 1.6117, + "step": 476 + }, + { + "epoch": 0.4899845916795069, + "grad_norm": 2.6067540645599365, + "learning_rate": 4.676437000380663e-06, + "loss": 1.5703, + "step": 477 + }, + { + "epoch": 0.49101181304571134, + "grad_norm": 2.490105152130127, + "learning_rate": 4.674533688618196e-06, + "loss": 1.6931, + "step": 478 + }, + { + "epoch": 0.49203903441191577, + "grad_norm": 2.7736282348632812, + "learning_rate": 4.6726303768557295e-06, + "loss": 1.6204, + "step": 479 + }, + { + "epoch": 0.4930662557781202, + "grad_norm": 2.741196870803833, + "learning_rate": 4.670727065093263e-06, + "loss": 1.5331, + "step": 480 + }, + { + "epoch": 0.4930662557781202, + "eval_loss": 1.9095442295074463, + "eval_runtime": 37.3351, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 480 + }, + { + "epoch": 0.4940934771443246, + "grad_norm": 2.7624332904815674, + "learning_rate": 4.668823753330796e-06, + "loss": 1.5591, + "step": 481 + }, + { + "epoch": 0.49512069851052903, + "grad_norm": 2.86407470703125, + "learning_rate": 4.6669204415683295e-06, + "loss": 1.5844, + "step": 482 + }, + { + "epoch": 0.49614791987673346, + "grad_norm": 2.770030975341797, + "learning_rate": 4.665017129805863e-06, + "loss": 1.6515, + "step": 483 + }, + { + "epoch": 0.4971751412429379, + "grad_norm": 2.569862127304077, + "learning_rate": 4.663113818043396e-06, + "loss": 1.6228, + "step": 484 + }, + { + "epoch": 0.49820236260914225, + "grad_norm": 2.7176733016967773, + "learning_rate": 4.66121050628093e-06, + "loss": 1.5938, + "step": 485 + }, + { + "epoch": 0.49922958397534667, + "grad_norm": 2.4514758586883545, + "learning_rate": 4.659307194518463e-06, + "loss": 1.6307, + "step": 486 + }, + { + "epoch": 0.5002568053415511, + "grad_norm": 2.5439529418945312, + "learning_rate": 4.6574038827559955e-06, + "loss": 1.5855, + "step": 487 + }, + { + "epoch": 0.5012840267077555, + "grad_norm": 2.6395790576934814, + "learning_rate": 4.655500570993529e-06, + "loss": 1.6124, + "step": 488 + }, + { + "epoch": 0.50231124807396, + "grad_norm": 2.69413685798645, + "learning_rate": 4.653597259231062e-06, + "loss": 1.6358, + "step": 489 + }, + { + "epoch": 0.5033384694401644, + "grad_norm": 2.693129777908325, + "learning_rate": 4.651693947468596e-06, + "loss": 1.5218, + "step": 490 + }, + { + "epoch": 0.5033384694401644, + "eval_loss": 1.9201244115829468, + "eval_runtime": 37.3324, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 490 + }, + { + "epoch": 0.5043656908063687, + "grad_norm": 2.630596876144409, + "learning_rate": 4.649790635706129e-06, + "loss": 1.6719, + "step": 491 + }, + { + "epoch": 0.5053929121725732, + "grad_norm": 2.7230262756347656, + "learning_rate": 4.647887323943662e-06, + "loss": 1.5705, + "step": 492 + }, + { + "epoch": 0.5064201335387776, + "grad_norm": 2.529944896697998, + "learning_rate": 4.645984012181196e-06, + "loss": 1.5767, + "step": 493 + }, + { + "epoch": 0.507447354904982, + "grad_norm": 2.546849489212036, + "learning_rate": 4.644080700418729e-06, + "loss": 1.5039, + "step": 494 + }, + { + "epoch": 0.5084745762711864, + "grad_norm": 2.6698246002197266, + "learning_rate": 4.642177388656262e-06, + "loss": 1.6229, + "step": 495 + }, + { + "epoch": 0.5095017976373909, + "grad_norm": 2.7093148231506348, + "learning_rate": 4.640274076893796e-06, + "loss": 1.5797, + "step": 496 + }, + { + "epoch": 0.5105290190035953, + "grad_norm": 2.48629093170166, + "learning_rate": 4.638370765131328e-06, + "loss": 1.6366, + "step": 497 + }, + { + "epoch": 0.5115562403697997, + "grad_norm": 2.5818612575531006, + "learning_rate": 4.636467453368862e-06, + "loss": 1.5608, + "step": 498 + }, + { + "epoch": 0.5125834617360041, + "grad_norm": 2.6263651847839355, + "learning_rate": 4.634564141606396e-06, + "loss": 1.5518, + "step": 499 + }, + { + "epoch": 0.5136106831022085, + "grad_norm": 2.5724668502807617, + "learning_rate": 4.632660829843929e-06, + "loss": 1.6195, + "step": 500 + }, + { + "epoch": 0.5136106831022085, + "eval_loss": 1.9094111919403076, + "eval_runtime": 37.3483, + "eval_samples_per_second": 26.775, + "eval_steps_per_second": 1.125, + "step": 500 + }, + { + "epoch": 0.514637904468413, + "grad_norm": 2.5028088092803955, + "learning_rate": 4.630757518081463e-06, + "loss": 1.5454, + "step": 501 + }, + { + "epoch": 0.5156651258346173, + "grad_norm": 2.588115692138672, + "learning_rate": 4.628854206318996e-06, + "loss": 1.6676, + "step": 502 + }, + { + "epoch": 0.5166923472008218, + "grad_norm": 2.9701285362243652, + "learning_rate": 4.626950894556529e-06, + "loss": 1.5822, + "step": 503 + }, + { + "epoch": 0.5177195685670262, + "grad_norm": 2.593867301940918, + "learning_rate": 4.625047582794062e-06, + "loss": 1.6091, + "step": 504 + }, + { + "epoch": 0.5187467899332306, + "grad_norm": 2.559642791748047, + "learning_rate": 4.623144271031595e-06, + "loss": 1.5359, + "step": 505 + }, + { + "epoch": 0.519774011299435, + "grad_norm": 2.4158153533935547, + "learning_rate": 4.6212409592691286e-06, + "loss": 1.5476, + "step": 506 + }, + { + "epoch": 0.5208012326656395, + "grad_norm": 2.666099786758423, + "learning_rate": 4.619337647506662e-06, + "loss": 1.6297, + "step": 507 + }, + { + "epoch": 0.5218284540318439, + "grad_norm": 2.7026472091674805, + "learning_rate": 4.617434335744195e-06, + "loss": 1.5891, + "step": 508 + }, + { + "epoch": 0.5228556753980482, + "grad_norm": 2.4591612815856934, + "learning_rate": 4.615531023981729e-06, + "loss": 1.5546, + "step": 509 + }, + { + "epoch": 0.5238828967642527, + "grad_norm": 2.870255708694458, + "learning_rate": 4.613627712219262e-06, + "loss": 1.5817, + "step": 510 + }, + { + "epoch": 0.5238828967642527, + "eval_loss": 1.910629153251648, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 510 + }, + { + "epoch": 0.5249101181304571, + "grad_norm": 2.653801679611206, + "learning_rate": 4.611724400456795e-06, + "loss": 1.586, + "step": 511 + }, + { + "epoch": 0.5259373394966615, + "grad_norm": 2.4235377311706543, + "learning_rate": 4.609821088694329e-06, + "loss": 1.5886, + "step": 512 + }, + { + "epoch": 0.5269645608628659, + "grad_norm": 2.8782477378845215, + "learning_rate": 4.607917776931862e-06, + "loss": 1.5634, + "step": 513 + }, + { + "epoch": 0.5279917822290704, + "grad_norm": 2.792876958847046, + "learning_rate": 4.606014465169395e-06, + "loss": 1.6073, + "step": 514 + }, + { + "epoch": 0.5290190035952748, + "grad_norm": 2.5579030513763428, + "learning_rate": 4.604111153406928e-06, + "loss": 1.6161, + "step": 515 + }, + { + "epoch": 0.5300462249614792, + "grad_norm": 2.6486189365386963, + "learning_rate": 4.602207841644461e-06, + "loss": 1.5297, + "step": 516 + }, + { + "epoch": 0.5310734463276836, + "grad_norm": 2.9047019481658936, + "learning_rate": 4.600304529881995e-06, + "loss": 1.626, + "step": 517 + }, + { + "epoch": 0.5321006676938881, + "grad_norm": 2.9164223670959473, + "learning_rate": 4.598401218119528e-06, + "loss": 1.6051, + "step": 518 + }, + { + "epoch": 0.5331278890600925, + "grad_norm": 2.544311046600342, + "learning_rate": 4.5964979063570614e-06, + "loss": 1.6247, + "step": 519 + }, + { + "epoch": 0.5341551104262968, + "grad_norm": 2.8632311820983887, + "learning_rate": 4.594594594594596e-06, + "loss": 1.5615, + "step": 520 + }, + { + "epoch": 0.5341551104262968, + "eval_loss": 1.9171669483184814, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 520 + }, + { + "epoch": 0.5351823317925013, + "grad_norm": 2.507697105407715, + "learning_rate": 4.592691282832128e-06, + "loss": 1.5929, + "step": 521 + }, + { + "epoch": 0.5362095531587057, + "grad_norm": 2.657752752304077, + "learning_rate": 4.5907879710696615e-06, + "loss": 1.6159, + "step": 522 + }, + { + "epoch": 0.5372367745249101, + "grad_norm": 2.32254695892334, + "learning_rate": 4.588884659307195e-06, + "loss": 1.5318, + "step": 523 + }, + { + "epoch": 0.5382639958911145, + "grad_norm": 2.5979015827178955, + "learning_rate": 4.586981347544728e-06, + "loss": 1.5999, + "step": 524 + }, + { + "epoch": 0.539291217257319, + "grad_norm": 2.839956521987915, + "learning_rate": 4.585078035782262e-06, + "loss": 1.547, + "step": 525 + }, + { + "epoch": 0.5403184386235234, + "grad_norm": 2.700131893157959, + "learning_rate": 4.583174724019795e-06, + "loss": 1.5831, + "step": 526 + }, + { + "epoch": 0.5413456599897278, + "grad_norm": 2.365183115005493, + "learning_rate": 4.581271412257328e-06, + "loss": 1.5357, + "step": 527 + }, + { + "epoch": 0.5423728813559322, + "grad_norm": 2.675567865371704, + "learning_rate": 4.579368100494862e-06, + "loss": 1.5212, + "step": 528 + }, + { + "epoch": 0.5434001027221366, + "grad_norm": 2.9096713066101074, + "learning_rate": 4.577464788732395e-06, + "loss": 1.6196, + "step": 529 + }, + { + "epoch": 0.544427324088341, + "grad_norm": 2.6186270713806152, + "learning_rate": 4.575561476969928e-06, + "loss": 1.5895, + "step": 530 + }, + { + "epoch": 0.544427324088341, + "eval_loss": 1.926998496055603, + "eval_runtime": 37.2248, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 530 + }, + { + "epoch": 0.5454545454545454, + "grad_norm": 2.7291345596313477, + "learning_rate": 4.573658165207461e-06, + "loss": 1.6335, + "step": 531 + }, + { + "epoch": 0.5464817668207499, + "grad_norm": 2.7084522247314453, + "learning_rate": 4.571754853444994e-06, + "loss": 1.5968, + "step": 532 + }, + { + "epoch": 0.5475089881869543, + "grad_norm": 2.699721336364746, + "learning_rate": 4.569851541682528e-06, + "loss": 1.5775, + "step": 533 + }, + { + "epoch": 0.5485362095531587, + "grad_norm": 2.602526903152466, + "learning_rate": 4.567948229920061e-06, + "loss": 1.5497, + "step": 534 + }, + { + "epoch": 0.5495634309193631, + "grad_norm": 2.597362518310547, + "learning_rate": 4.566044918157594e-06, + "loss": 1.5218, + "step": 535 + }, + { + "epoch": 0.5505906522855676, + "grad_norm": 2.6624414920806885, + "learning_rate": 4.564141606395128e-06, + "loss": 1.5553, + "step": 536 + }, + { + "epoch": 0.551617873651772, + "grad_norm": 2.750807285308838, + "learning_rate": 4.562238294632661e-06, + "loss": 1.6078, + "step": 537 + }, + { + "epoch": 0.5526450950179764, + "grad_norm": 2.7480952739715576, + "learning_rate": 4.5603349828701945e-06, + "loss": 1.5707, + "step": 538 + }, + { + "epoch": 0.5536723163841808, + "grad_norm": 2.6287965774536133, + "learning_rate": 4.558431671107728e-06, + "loss": 1.5961, + "step": 539 + }, + { + "epoch": 0.5546995377503852, + "grad_norm": 2.721311569213867, + "learning_rate": 4.556528359345261e-06, + "loss": 1.6335, + "step": 540 + }, + { + "epoch": 0.5546995377503852, + "eval_loss": 1.9290324449539185, + "eval_runtime": 37.2243, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5557267591165896, + "grad_norm": 2.470325469970703, + "learning_rate": 4.5546250475827946e-06, + "loss": 1.5645, + "step": 541 + }, + { + "epoch": 0.556753980482794, + "grad_norm": 2.6089508533477783, + "learning_rate": 4.552721735820328e-06, + "loss": 1.5256, + "step": 542 + }, + { + "epoch": 0.5577812018489985, + "grad_norm": 2.541785478591919, + "learning_rate": 4.550818424057861e-06, + "loss": 1.5641, + "step": 543 + }, + { + "epoch": 0.5588084232152029, + "grad_norm": 2.6722488403320312, + "learning_rate": 4.548915112295395e-06, + "loss": 1.5708, + "step": 544 + }, + { + "epoch": 0.5598356445814073, + "grad_norm": 2.583695650100708, + "learning_rate": 4.547011800532928e-06, + "loss": 1.5799, + "step": 545 + }, + { + "epoch": 0.5608628659476117, + "grad_norm": 2.956885576248169, + "learning_rate": 4.545108488770461e-06, + "loss": 1.5345, + "step": 546 + }, + { + "epoch": 0.5618900873138162, + "grad_norm": 2.5534746646881104, + "learning_rate": 4.543205177007995e-06, + "loss": 1.636, + "step": 547 + }, + { + "epoch": 0.5629173086800205, + "grad_norm": 2.8133325576782227, + "learning_rate": 4.541301865245527e-06, + "loss": 1.5678, + "step": 548 + }, + { + "epoch": 0.5639445300462249, + "grad_norm": 2.5969228744506836, + "learning_rate": 4.539398553483061e-06, + "loss": 1.5975, + "step": 549 + }, + { + "epoch": 0.5649717514124294, + "grad_norm": 2.7594687938690186, + "learning_rate": 4.537495241720594e-06, + "loss": 1.5622, + "step": 550 + }, + { + "epoch": 0.5649717514124294, + "eval_loss": 1.8870681524276733, + "eval_runtime": 37.2094, + "eval_samples_per_second": 26.875, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.5659989727786338, + "grad_norm": 2.7413411140441895, + "learning_rate": 4.535591929958127e-06, + "loss": 1.6031, + "step": 551 + }, + { + "epoch": 0.5670261941448382, + "grad_norm": 2.6182661056518555, + "learning_rate": 4.533688618195661e-06, + "loss": 1.5828, + "step": 552 + }, + { + "epoch": 0.5680534155110426, + "grad_norm": 2.5016422271728516, + "learning_rate": 4.531785306433194e-06, + "loss": 1.5712, + "step": 553 + }, + { + "epoch": 0.5690806368772471, + "grad_norm": 2.7788732051849365, + "learning_rate": 4.529881994670727e-06, + "loss": 1.5472, + "step": 554 + }, + { + "epoch": 0.5701078582434514, + "grad_norm": 2.7602436542510986, + "learning_rate": 4.527978682908261e-06, + "loss": 1.5526, + "step": 555 + }, + { + "epoch": 0.5711350796096559, + "grad_norm": 2.5898544788360596, + "learning_rate": 4.526075371145794e-06, + "loss": 1.5644, + "step": 556 + }, + { + "epoch": 0.5721623009758603, + "grad_norm": 2.439344882965088, + "learning_rate": 4.5241720593833275e-06, + "loss": 1.5892, + "step": 557 + }, + { + "epoch": 0.5731895223420647, + "grad_norm": 2.6324563026428223, + "learning_rate": 4.52226874762086e-06, + "loss": 1.5729, + "step": 558 + }, + { + "epoch": 0.5742167437082691, + "grad_norm": 2.674042224884033, + "learning_rate": 4.520365435858393e-06, + "loss": 1.6016, + "step": 559 + }, + { + "epoch": 0.5752439650744735, + "grad_norm": 2.692619800567627, + "learning_rate": 4.518462124095928e-06, + "loss": 1.6229, + "step": 560 + }, + { + "epoch": 0.5752439650744735, + "eval_loss": 1.8689097166061401, + "eval_runtime": 37.2177, + "eval_samples_per_second": 26.869, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.576271186440678, + "grad_norm": 2.636924982070923, + "learning_rate": 4.516558812333461e-06, + "loss": 1.5748, + "step": 561 + }, + { + "epoch": 0.5772984078068824, + "grad_norm": 2.549285888671875, + "learning_rate": 4.514655500570994e-06, + "loss": 1.5656, + "step": 562 + }, + { + "epoch": 0.5783256291730868, + "grad_norm": 2.5873067378997803, + "learning_rate": 4.512752188808528e-06, + "loss": 1.631, + "step": 563 + }, + { + "epoch": 0.5793528505392912, + "grad_norm": 2.552121162414551, + "learning_rate": 4.510848877046061e-06, + "loss": 1.6334, + "step": 564 + }, + { + "epoch": 0.5803800719054957, + "grad_norm": 2.697615146636963, + "learning_rate": 4.5089455652835936e-06, + "loss": 1.5951, + "step": 565 + }, + { + "epoch": 0.5814072932717, + "grad_norm": 2.64585280418396, + "learning_rate": 4.507042253521127e-06, + "loss": 1.556, + "step": 566 + }, + { + "epoch": 0.5824345146379045, + "grad_norm": 2.6383707523345947, + "learning_rate": 4.50513894175866e-06, + "loss": 1.5184, + "step": 567 + }, + { + "epoch": 0.5834617360041089, + "grad_norm": 2.7571511268615723, + "learning_rate": 4.503235629996194e-06, + "loss": 1.5984, + "step": 568 + }, + { + "epoch": 0.5844889573703133, + "grad_norm": 2.6637237071990967, + "learning_rate": 4.501332318233727e-06, + "loss": 1.5989, + "step": 569 + }, + { + "epoch": 0.5855161787365177, + "grad_norm": 2.725754976272583, + "learning_rate": 4.49942900647126e-06, + "loss": 1.5594, + "step": 570 + }, + { + "epoch": 0.5855161787365177, + "eval_loss": 1.8722314834594727, + "eval_runtime": 37.1859, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 570 + }, + { + "epoch": 0.5865434001027221, + "grad_norm": 2.6657378673553467, + "learning_rate": 4.497525694708794e-06, + "loss": 1.5954, + "step": 571 + }, + { + "epoch": 0.5875706214689266, + "grad_norm": 2.78940486907959, + "learning_rate": 4.495622382946327e-06, + "loss": 1.592, + "step": 572 + }, + { + "epoch": 0.588597842835131, + "grad_norm": 2.580956220626831, + "learning_rate": 4.4937190711838605e-06, + "loss": 1.5444, + "step": 573 + }, + { + "epoch": 0.5896250642013354, + "grad_norm": 2.7783710956573486, + "learning_rate": 4.491815759421394e-06, + "loss": 1.6257, + "step": 574 + }, + { + "epoch": 0.5906522855675398, + "grad_norm": 2.9381537437438965, + "learning_rate": 4.489912447658926e-06, + "loss": 1.5489, + "step": 575 + }, + { + "epoch": 0.5916795069337443, + "grad_norm": 2.7915165424346924, + "learning_rate": 4.48800913589646e-06, + "loss": 1.5471, + "step": 576 + }, + { + "epoch": 0.5927067282999486, + "grad_norm": 2.5928280353546143, + "learning_rate": 4.486105824133993e-06, + "loss": 1.5864, + "step": 577 + }, + { + "epoch": 0.593733949666153, + "grad_norm": 2.844512462615967, + "learning_rate": 4.4842025123715264e-06, + "loss": 1.5841, + "step": 578 + }, + { + "epoch": 0.5947611710323575, + "grad_norm": 2.562718391418457, + "learning_rate": 4.48229920060906e-06, + "loss": 1.5476, + "step": 579 + }, + { + "epoch": 0.5957883923985619, + "grad_norm": 2.535759925842285, + "learning_rate": 4.480395888846594e-06, + "loss": 1.5274, + "step": 580 + }, + { + "epoch": 0.5957883923985619, + "eval_loss": 1.8881579637527466, + "eval_runtime": 37.2533, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.5968156137647663, + "grad_norm": 2.6940653324127197, + "learning_rate": 4.478492577084127e-06, + "loss": 1.6377, + "step": 581 + }, + { + "epoch": 0.5978428351309707, + "grad_norm": 2.4839298725128174, + "learning_rate": 4.47658926532166e-06, + "loss": 1.534, + "step": 582 + }, + { + "epoch": 0.5988700564971752, + "grad_norm": 2.5035688877105713, + "learning_rate": 4.474685953559193e-06, + "loss": 1.5764, + "step": 583 + }, + { + "epoch": 0.5998972778633795, + "grad_norm": 2.791907548904419, + "learning_rate": 4.472782641796727e-06, + "loss": 1.704, + "step": 584 + }, + { + "epoch": 0.600924499229584, + "grad_norm": 2.5384254455566406, + "learning_rate": 4.47087933003426e-06, + "loss": 1.5184, + "step": 585 + }, + { + "epoch": 0.6019517205957884, + "grad_norm": 2.4990930557250977, + "learning_rate": 4.468976018271793e-06, + "loss": 1.5335, + "step": 586 + }, + { + "epoch": 0.6029789419619929, + "grad_norm": 2.470872163772583, + "learning_rate": 4.467072706509327e-06, + "loss": 1.5464, + "step": 587 + }, + { + "epoch": 0.6040061633281972, + "grad_norm": 2.539036989212036, + "learning_rate": 4.46516939474686e-06, + "loss": 1.5793, + "step": 588 + }, + { + "epoch": 0.6050333846944016, + "grad_norm": 2.495309829711914, + "learning_rate": 4.463266082984393e-06, + "loss": 1.5312, + "step": 589 + }, + { + "epoch": 0.6060606060606061, + "grad_norm": 2.5937912464141846, + "learning_rate": 4.461362771221927e-06, + "loss": 1.6508, + "step": 590 + }, + { + "epoch": 0.6060606060606061, + "eval_loss": 1.8893816471099854, + "eval_runtime": 37.2539, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 590 + }, + { + "epoch": 0.6070878274268104, + "grad_norm": 2.53300404548645, + "learning_rate": 4.45945945945946e-06, + "loss": 1.5408, + "step": 591 + }, + { + "epoch": 0.6081150487930149, + "grad_norm": 2.5760276317596436, + "learning_rate": 4.457556147696993e-06, + "loss": 1.5575, + "step": 592 + }, + { + "epoch": 0.6091422701592193, + "grad_norm": 2.638353109359741, + "learning_rate": 4.455652835934526e-06, + "loss": 1.6286, + "step": 593 + }, + { + "epoch": 0.6101694915254238, + "grad_norm": 2.916550397872925, + "learning_rate": 4.453749524172059e-06, + "loss": 1.6334, + "step": 594 + }, + { + "epoch": 0.6111967128916281, + "grad_norm": 2.560878038406372, + "learning_rate": 4.451846212409593e-06, + "loss": 1.5806, + "step": 595 + }, + { + "epoch": 0.6122239342578326, + "grad_norm": 2.637882947921753, + "learning_rate": 4.449942900647126e-06, + "loss": 1.5791, + "step": 596 + }, + { + "epoch": 0.613251155624037, + "grad_norm": 2.752870798110962, + "learning_rate": 4.4480395888846595e-06, + "loss": 1.5117, + "step": 597 + }, + { + "epoch": 0.6142783769902413, + "grad_norm": 2.702087640762329, + "learning_rate": 4.446136277122193e-06, + "loss": 1.5783, + "step": 598 + }, + { + "epoch": 0.6153055983564458, + "grad_norm": 2.504819631576538, + "learning_rate": 4.444232965359726e-06, + "loss": 1.5136, + "step": 599 + }, + { + "epoch": 0.6163328197226502, + "grad_norm": 2.521670341491699, + "learning_rate": 4.4423296535972596e-06, + "loss": 1.5741, + "step": 600 + }, + { + "epoch": 0.6163328197226502, + "eval_loss": 1.8973714113235474, + "eval_runtime": 37.3167, + "eval_samples_per_second": 26.798, + "eval_steps_per_second": 1.126, + "step": 600 + }, + { + "epoch": 0.6173600410888547, + "grad_norm": 2.927755355834961, + "learning_rate": 4.440426341834793e-06, + "loss": 1.5262, + "step": 601 + }, + { + "epoch": 0.618387262455059, + "grad_norm": 2.477053642272949, + "learning_rate": 4.438523030072326e-06, + "loss": 1.572, + "step": 602 + }, + { + "epoch": 0.6194144838212635, + "grad_norm": 2.6500253677368164, + "learning_rate": 4.43661971830986e-06, + "loss": 1.596, + "step": 603 + }, + { + "epoch": 0.6204417051874679, + "grad_norm": 2.5134825706481934, + "learning_rate": 4.434716406547393e-06, + "loss": 1.5432, + "step": 604 + }, + { + "epoch": 0.6214689265536724, + "grad_norm": 2.4142072200775146, + "learning_rate": 4.432813094784926e-06, + "loss": 1.5391, + "step": 605 + }, + { + "epoch": 0.6224961479198767, + "grad_norm": 2.752556800842285, + "learning_rate": 4.43090978302246e-06, + "loss": 1.6055, + "step": 606 + }, + { + "epoch": 0.6235233692860811, + "grad_norm": 2.5731008052825928, + "learning_rate": 4.429006471259993e-06, + "loss": 1.5648, + "step": 607 + }, + { + "epoch": 0.6245505906522856, + "grad_norm": 2.531656503677368, + "learning_rate": 4.4271031594975265e-06, + "loss": 1.5492, + "step": 608 + }, + { + "epoch": 0.6255778120184899, + "grad_norm": 2.6665475368499756, + "learning_rate": 4.425199847735059e-06, + "loss": 1.5194, + "step": 609 + }, + { + "epoch": 0.6266050333846944, + "grad_norm": 2.437051773071289, + "learning_rate": 4.423296535972592e-06, + "loss": 1.5271, + "step": 610 + }, + { + "epoch": 0.6266050333846944, + "eval_loss": 1.88953697681427, + "eval_runtime": 36.8703, + "eval_samples_per_second": 27.122, + "eval_steps_per_second": 1.139, + "step": 610 + }, + { + "epoch": 0.6276322547508988, + "grad_norm": 2.5932440757751465, + "learning_rate": 4.421393224210126e-06, + "loss": 1.5563, + "step": 611 + }, + { + "epoch": 0.6286594761171033, + "grad_norm": 2.589447021484375, + "learning_rate": 4.419489912447659e-06, + "loss": 1.5207, + "step": 612 + }, + { + "epoch": 0.6296866974833076, + "grad_norm": 2.570096731185913, + "learning_rate": 4.4175866006851924e-06, + "loss": 1.5995, + "step": 613 + }, + { + "epoch": 0.6307139188495121, + "grad_norm": 2.5356791019439697, + "learning_rate": 4.415683288922726e-06, + "loss": 1.5703, + "step": 614 + }, + { + "epoch": 0.6317411402157165, + "grad_norm": 2.705193281173706, + "learning_rate": 4.413779977160259e-06, + "loss": 1.5405, + "step": 615 + }, + { + "epoch": 0.632768361581921, + "grad_norm": 2.555168390274048, + "learning_rate": 4.4118766653977925e-06, + "loss": 1.5487, + "step": 616 + }, + { + "epoch": 0.6337955829481253, + "grad_norm": 2.804696559906006, + "learning_rate": 4.409973353635326e-06, + "loss": 1.6094, + "step": 617 + }, + { + "epoch": 0.6348228043143297, + "grad_norm": 2.6889376640319824, + "learning_rate": 4.408070041872859e-06, + "loss": 1.5656, + "step": 618 + }, + { + "epoch": 0.6358500256805342, + "grad_norm": 2.576746702194214, + "learning_rate": 4.406166730110393e-06, + "loss": 1.5244, + "step": 619 + }, + { + "epoch": 0.6368772470467385, + "grad_norm": 2.811156749725342, + "learning_rate": 4.404263418347925e-06, + "loss": 1.5153, + "step": 620 + }, + { + "epoch": 0.6368772470467385, + "eval_loss": 1.8949775695800781, + "eval_runtime": 36.978, + "eval_samples_per_second": 27.043, + "eval_steps_per_second": 1.136, + "step": 620 + }, + { + "epoch": 0.637904468412943, + "grad_norm": 2.6560428142547607, + "learning_rate": 4.402360106585459e-06, + "loss": 1.5347, + "step": 621 + }, + { + "epoch": 0.6389316897791474, + "grad_norm": 2.668100118637085, + "learning_rate": 4.400456794822993e-06, + "loss": 1.5236, + "step": 622 + }, + { + "epoch": 0.6399589111453519, + "grad_norm": 2.925978660583496, + "learning_rate": 4.398553483060526e-06, + "loss": 1.5494, + "step": 623 + }, + { + "epoch": 0.6409861325115562, + "grad_norm": 2.5964415073394775, + "learning_rate": 4.396650171298059e-06, + "loss": 1.522, + "step": 624 + }, + { + "epoch": 0.6420133538777607, + "grad_norm": 2.643888235092163, + "learning_rate": 4.394746859535593e-06, + "loss": 1.5332, + "step": 625 + }, + { + "epoch": 0.6430405752439651, + "grad_norm": 2.805192708969116, + "learning_rate": 4.392843547773125e-06, + "loss": 1.5031, + "step": 626 + }, + { + "epoch": 0.6440677966101694, + "grad_norm": 2.6862149238586426, + "learning_rate": 4.390940236010659e-06, + "loss": 1.5335, + "step": 627 + }, + { + "epoch": 0.6450950179763739, + "grad_norm": 2.5579185485839844, + "learning_rate": 4.389036924248192e-06, + "loss": 1.5466, + "step": 628 + }, + { + "epoch": 0.6461222393425783, + "grad_norm": 2.4550797939300537, + "learning_rate": 4.387133612485725e-06, + "loss": 1.5776, + "step": 629 + }, + { + "epoch": 0.6471494607087828, + "grad_norm": 2.6051297187805176, + "learning_rate": 4.385230300723259e-06, + "loss": 1.5435, + "step": 630 + }, + { + "epoch": 0.6471494607087828, + "eval_loss": 1.8860505819320679, + "eval_runtime": 36.989, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 630 + }, + { + "epoch": 0.6481766820749871, + "grad_norm": 2.3723294734954834, + "learning_rate": 4.383326988960792e-06, + "loss": 1.5413, + "step": 631 + }, + { + "epoch": 0.6492039034411916, + "grad_norm": 2.44370174407959, + "learning_rate": 4.3814236771983255e-06, + "loss": 1.5152, + "step": 632 + }, + { + "epoch": 0.650231124807396, + "grad_norm": 2.474081039428711, + "learning_rate": 4.379520365435859e-06, + "loss": 1.579, + "step": 633 + }, + { + "epoch": 0.6512583461736005, + "grad_norm": 2.837843179702759, + "learning_rate": 4.377617053673392e-06, + "loss": 1.559, + "step": 634 + }, + { + "epoch": 0.6522855675398048, + "grad_norm": 2.593322515487671, + "learning_rate": 4.3757137419109256e-06, + "loss": 1.552, + "step": 635 + }, + { + "epoch": 0.6533127889060092, + "grad_norm": 3.018051862716675, + "learning_rate": 4.373810430148459e-06, + "loss": 1.6231, + "step": 636 + }, + { + "epoch": 0.6543400102722137, + "grad_norm": 2.6338863372802734, + "learning_rate": 4.3719071183859914e-06, + "loss": 1.517, + "step": 637 + }, + { + "epoch": 0.655367231638418, + "grad_norm": 2.544140577316284, + "learning_rate": 4.370003806623525e-06, + "loss": 1.5763, + "step": 638 + }, + { + "epoch": 0.6563944530046225, + "grad_norm": 2.6375792026519775, + "learning_rate": 4.368100494861058e-06, + "loss": 1.5098, + "step": 639 + }, + { + "epoch": 0.6574216743708269, + "grad_norm": 2.5452051162719727, + "learning_rate": 4.3661971830985915e-06, + "loss": 1.5671, + "step": 640 + }, + { + "epoch": 0.6574216743708269, + "eval_loss": 1.881069540977478, + "eval_runtime": 36.9961, + "eval_samples_per_second": 27.03, + "eval_steps_per_second": 1.135, + "step": 640 + }, + { + "epoch": 0.6584488957370314, + "grad_norm": 2.623411178588867, + "learning_rate": 4.364293871336126e-06, + "loss": 1.5215, + "step": 641 + }, + { + "epoch": 0.6594761171032357, + "grad_norm": 2.6149892807006836, + "learning_rate": 4.362390559573659e-06, + "loss": 1.5507, + "step": 642 + }, + { + "epoch": 0.6605033384694402, + "grad_norm": 2.5570266246795654, + "learning_rate": 4.3604872478111925e-06, + "loss": 1.5143, + "step": 643 + }, + { + "epoch": 0.6615305598356446, + "grad_norm": 2.4813005924224854, + "learning_rate": 4.358583936048725e-06, + "loss": 1.4518, + "step": 644 + }, + { + "epoch": 0.662557781201849, + "grad_norm": 3.0318472385406494, + "learning_rate": 4.356680624286258e-06, + "loss": 1.5812, + "step": 645 + }, + { + "epoch": 0.6635850025680534, + "grad_norm": 2.46936297416687, + "learning_rate": 4.354777312523792e-06, + "loss": 1.5396, + "step": 646 + }, + { + "epoch": 0.6646122239342578, + "grad_norm": 2.5295984745025635, + "learning_rate": 4.352874000761325e-06, + "loss": 1.586, + "step": 647 + }, + { + "epoch": 0.6656394453004623, + "grad_norm": 2.7826883792877197, + "learning_rate": 4.3509706889988584e-06, + "loss": 1.5339, + "step": 648 + }, + { + "epoch": 0.6666666666666666, + "grad_norm": 2.4273526668548584, + "learning_rate": 4.349067377236392e-06, + "loss": 1.598, + "step": 649 + }, + { + "epoch": 0.6676938880328711, + "grad_norm": 2.644308090209961, + "learning_rate": 4.347164065473925e-06, + "loss": 1.5889, + "step": 650 + }, + { + "epoch": 0.6676938880328711, + "eval_loss": 1.8861087560653687, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 650 + }, + { + "epoch": 0.6687211093990755, + "grad_norm": 2.827376127243042, + "learning_rate": 4.3452607537114585e-06, + "loss": 1.606, + "step": 651 + }, + { + "epoch": 0.66974833076528, + "grad_norm": 2.7463080883026123, + "learning_rate": 4.343357441948992e-06, + "loss": 1.5208, + "step": 652 + }, + { + "epoch": 0.6707755521314843, + "grad_norm": 2.6082730293273926, + "learning_rate": 4.341454130186525e-06, + "loss": 1.5597, + "step": 653 + }, + { + "epoch": 0.6718027734976888, + "grad_norm": 2.7131078243255615, + "learning_rate": 4.339550818424058e-06, + "loss": 1.526, + "step": 654 + }, + { + "epoch": 0.6728299948638932, + "grad_norm": 2.6635708808898926, + "learning_rate": 4.337647506661591e-06, + "loss": 1.5052, + "step": 655 + }, + { + "epoch": 0.6738572162300975, + "grad_norm": 2.5686581134796143, + "learning_rate": 4.3357441948991245e-06, + "loss": 1.5374, + "step": 656 + }, + { + "epoch": 0.674884437596302, + "grad_norm": 3.0532894134521484, + "learning_rate": 4.333840883136658e-06, + "loss": 1.5787, + "step": 657 + }, + { + "epoch": 0.6759116589625064, + "grad_norm": 2.6494507789611816, + "learning_rate": 4.331937571374191e-06, + "loss": 1.5121, + "step": 658 + }, + { + "epoch": 0.6769388803287109, + "grad_norm": 2.748142719268799, + "learning_rate": 4.3300342596117246e-06, + "loss": 1.5294, + "step": 659 + }, + { + "epoch": 0.6779661016949152, + "grad_norm": 2.837799310684204, + "learning_rate": 4.328130947849258e-06, + "loss": 1.5424, + "step": 660 + }, + { + "epoch": 0.6779661016949152, + "eval_loss": 1.8644919395446777, + "eval_runtime": 37.0096, + "eval_samples_per_second": 27.02, + "eval_steps_per_second": 1.135, + "step": 660 + }, + { + "epoch": 0.6789933230611197, + "grad_norm": 2.7405757904052734, + "learning_rate": 4.326227636086791e-06, + "loss": 1.4949, + "step": 661 + }, + { + "epoch": 0.6800205444273241, + "grad_norm": 2.545466661453247, + "learning_rate": 4.324324324324325e-06, + "loss": 1.5445, + "step": 662 + }, + { + "epoch": 0.6810477657935285, + "grad_norm": 2.5715537071228027, + "learning_rate": 4.322421012561858e-06, + "loss": 1.5468, + "step": 663 + }, + { + "epoch": 0.6820749871597329, + "grad_norm": 2.7073662281036377, + "learning_rate": 4.320517700799391e-06, + "loss": 1.5293, + "step": 664 + }, + { + "epoch": 0.6831022085259374, + "grad_norm": 2.62264347076416, + "learning_rate": 4.318614389036925e-06, + "loss": 1.6258, + "step": 665 + }, + { + "epoch": 0.6841294298921418, + "grad_norm": 2.4506373405456543, + "learning_rate": 4.316711077274458e-06, + "loss": 1.5845, + "step": 666 + }, + { + "epoch": 0.6851566512583461, + "grad_norm": 2.849473237991333, + "learning_rate": 4.3148077655119915e-06, + "loss": 1.4959, + "step": 667 + }, + { + "epoch": 0.6861838726245506, + "grad_norm": 2.5865421295166016, + "learning_rate": 4.312904453749525e-06, + "loss": 1.599, + "step": 668 + }, + { + "epoch": 0.687211093990755, + "grad_norm": 2.4636387825012207, + "learning_rate": 4.311001141987058e-06, + "loss": 1.5495, + "step": 669 + }, + { + "epoch": 0.6882383153569595, + "grad_norm": 2.638047218322754, + "learning_rate": 4.3090978302245916e-06, + "loss": 1.5507, + "step": 670 + }, + { + "epoch": 0.6882383153569595, + "eval_loss": 1.8834826946258545, + "eval_runtime": 37.0335, + "eval_samples_per_second": 27.003, + "eval_steps_per_second": 1.134, + "step": 670 + }, + { + "epoch": 0.6892655367231638, + "grad_norm": 2.669201612472534, + "learning_rate": 4.307194518462124e-06, + "loss": 1.5494, + "step": 671 + }, + { + "epoch": 0.6902927580893683, + "grad_norm": 2.634634494781494, + "learning_rate": 4.3052912066996574e-06, + "loss": 1.6177, + "step": 672 + }, + { + "epoch": 0.6913199794555727, + "grad_norm": 2.5489139556884766, + "learning_rate": 4.303387894937191e-06, + "loss": 1.5876, + "step": 673 + }, + { + "epoch": 0.6923472008217771, + "grad_norm": 2.580073833465576, + "learning_rate": 4.301484583174724e-06, + "loss": 1.5106, + "step": 674 + }, + { + "epoch": 0.6933744221879815, + "grad_norm": 2.5126636028289795, + "learning_rate": 4.2995812714122575e-06, + "loss": 1.4988, + "step": 675 + }, + { + "epoch": 0.6944016435541859, + "grad_norm": 2.4630191326141357, + "learning_rate": 4.297677959649791e-06, + "loss": 1.5218, + "step": 676 + }, + { + "epoch": 0.6954288649203904, + "grad_norm": 2.8822433948516846, + "learning_rate": 4.295774647887324e-06, + "loss": 1.5592, + "step": 677 + }, + { + "epoch": 0.6964560862865947, + "grad_norm": 2.5498058795928955, + "learning_rate": 4.293871336124858e-06, + "loss": 1.5713, + "step": 678 + }, + { + "epoch": 0.6974833076527992, + "grad_norm": 2.6661157608032227, + "learning_rate": 4.291968024362391e-06, + "loss": 1.5431, + "step": 679 + }, + { + "epoch": 0.6985105290190036, + "grad_norm": 2.617971897125244, + "learning_rate": 4.290064712599924e-06, + "loss": 1.5792, + "step": 680 + }, + { + "epoch": 0.6985105290190036, + "eval_loss": 1.8702285289764404, + "eval_runtime": 37.0376, + "eval_samples_per_second": 27.0, + "eval_steps_per_second": 1.134, + "step": 680 + }, + { + "epoch": 0.699537750385208, + "grad_norm": 2.5892348289489746, + "learning_rate": 4.288161400837458e-06, + "loss": 1.5626, + "step": 681 + }, + { + "epoch": 0.7005649717514124, + "grad_norm": 2.5460269451141357, + "learning_rate": 4.286258089074991e-06, + "loss": 1.4876, + "step": 682 + }, + { + "epoch": 0.7015921931176169, + "grad_norm": 2.601632595062256, + "learning_rate": 4.284354777312524e-06, + "loss": 1.5547, + "step": 683 + }, + { + "epoch": 0.7026194144838213, + "grad_norm": 2.6032373905181885, + "learning_rate": 4.282451465550058e-06, + "loss": 1.5133, + "step": 684 + }, + { + "epoch": 0.7036466358500256, + "grad_norm": 2.6283886432647705, + "learning_rate": 4.280548153787591e-06, + "loss": 1.5375, + "step": 685 + }, + { + "epoch": 0.7046738572162301, + "grad_norm": 2.5892159938812256, + "learning_rate": 4.2786448420251245e-06, + "loss": 1.5338, + "step": 686 + }, + { + "epoch": 0.7057010785824345, + "grad_norm": 2.518629789352417, + "learning_rate": 4.276741530262658e-06, + "loss": 1.5003, + "step": 687 + }, + { + "epoch": 0.706728299948639, + "grad_norm": 2.8705344200134277, + "learning_rate": 4.27483821850019e-06, + "loss": 1.5178, + "step": 688 + }, + { + "epoch": 0.7077555213148433, + "grad_norm": 2.7153639793395996, + "learning_rate": 4.272934906737724e-06, + "loss": 1.5914, + "step": 689 + }, + { + "epoch": 0.7087827426810478, + "grad_norm": 2.4585697650909424, + "learning_rate": 4.271031594975257e-06, + "loss": 1.5726, + "step": 690 + }, + { + "epoch": 0.7087827426810478, + "eval_loss": 1.8637797832489014, + "eval_runtime": 37.0465, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 690 + }, + { + "epoch": 0.7098099640472522, + "grad_norm": 2.6370503902435303, + "learning_rate": 4.2691282832127905e-06, + "loss": 1.5706, + "step": 691 + }, + { + "epoch": 0.7108371854134566, + "grad_norm": 2.70202898979187, + "learning_rate": 4.267224971450324e-06, + "loss": 1.5298, + "step": 692 + }, + { + "epoch": 0.711864406779661, + "grad_norm": 2.6149585247039795, + "learning_rate": 4.265321659687857e-06, + "loss": 1.477, + "step": 693 + }, + { + "epoch": 0.7128916281458655, + "grad_norm": 2.9561493396759033, + "learning_rate": 4.2634183479253906e-06, + "loss": 1.5189, + "step": 694 + }, + { + "epoch": 0.7139188495120699, + "grad_norm": 2.697680711746216, + "learning_rate": 4.261515036162924e-06, + "loss": 1.5001, + "step": 695 + }, + { + "epoch": 0.7149460708782742, + "grad_norm": 2.55727219581604, + "learning_rate": 4.259611724400457e-06, + "loss": 1.533, + "step": 696 + }, + { + "epoch": 0.7159732922444787, + "grad_norm": 2.854290723800659, + "learning_rate": 4.257708412637991e-06, + "loss": 1.5057, + "step": 697 + }, + { + "epoch": 0.7170005136106831, + "grad_norm": 2.73824405670166, + "learning_rate": 4.255805100875523e-06, + "loss": 1.4567, + "step": 698 + }, + { + "epoch": 0.7180277349768875, + "grad_norm": 2.730050802230835, + "learning_rate": 4.2539017891130565e-06, + "loss": 1.4873, + "step": 699 + }, + { + "epoch": 0.7190549563430919, + "grad_norm": 2.966660737991333, + "learning_rate": 4.25199847735059e-06, + "loss": 1.574, + "step": 700 + }, + { + "epoch": 0.7190549563430919, + "eval_loss": 1.8623589277267456, + "eval_runtime": 37.0761, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 700 + }, + { + "epoch": 0.7200821777092964, + "grad_norm": 2.7526276111602783, + "learning_rate": 4.250095165588123e-06, + "loss": 1.4831, + "step": 701 + }, + { + "epoch": 0.7211093990755008, + "grad_norm": 2.9722564220428467, + "learning_rate": 4.2481918538256575e-06, + "loss": 1.548, + "step": 702 + }, + { + "epoch": 0.7221366204417052, + "grad_norm": 2.679373025894165, + "learning_rate": 4.246288542063191e-06, + "loss": 1.5145, + "step": 703 + }, + { + "epoch": 0.7231638418079096, + "grad_norm": 2.733139753341675, + "learning_rate": 4.244385230300724e-06, + "loss": 1.5542, + "step": 704 + }, + { + "epoch": 0.724191063174114, + "grad_norm": 2.6549174785614014, + "learning_rate": 4.242481918538257e-06, + "loss": 1.4958, + "step": 705 + }, + { + "epoch": 0.7252182845403184, + "grad_norm": 2.562958002090454, + "learning_rate": 4.24057860677579e-06, + "loss": 1.524, + "step": 706 + }, + { + "epoch": 0.7262455059065228, + "grad_norm": 2.641123056411743, + "learning_rate": 4.2386752950133234e-06, + "loss": 1.5723, + "step": 707 + }, + { + "epoch": 0.7272727272727273, + "grad_norm": 2.8128373622894287, + "learning_rate": 4.236771983250857e-06, + "loss": 1.5151, + "step": 708 + }, + { + "epoch": 0.7282999486389317, + "grad_norm": 2.676765203475952, + "learning_rate": 4.23486867148839e-06, + "loss": 1.5024, + "step": 709 + }, + { + "epoch": 0.7293271700051361, + "grad_norm": 2.917178153991699, + "learning_rate": 4.2329653597259235e-06, + "loss": 1.5326, + "step": 710 + }, + { + "epoch": 0.7293271700051361, + "eval_loss": 1.8676133155822754, + "eval_runtime": 36.8304, + "eval_samples_per_second": 27.152, + "eval_steps_per_second": 1.14, + "step": 710 + }, + { + "epoch": 0.7303543913713405, + "grad_norm": 2.8313796520233154, + "learning_rate": 4.231062047963457e-06, + "loss": 1.5623, + "step": 711 + }, + { + "epoch": 0.731381612737545, + "grad_norm": 2.61542010307312, + "learning_rate": 4.22915873620099e-06, + "loss": 1.5725, + "step": 712 + }, + { + "epoch": 0.7324088341037494, + "grad_norm": 2.5788609981536865, + "learning_rate": 4.227255424438524e-06, + "loss": 1.5381, + "step": 713 + }, + { + "epoch": 0.7334360554699538, + "grad_norm": 2.720306158065796, + "learning_rate": 4.225352112676057e-06, + "loss": 1.572, + "step": 714 + }, + { + "epoch": 0.7344632768361582, + "grad_norm": 2.780654191970825, + "learning_rate": 4.2234488009135895e-06, + "loss": 1.5084, + "step": 715 + }, + { + "epoch": 0.7354904982023626, + "grad_norm": 2.4448273181915283, + "learning_rate": 4.221545489151123e-06, + "loss": 1.4614, + "step": 716 + }, + { + "epoch": 0.736517719568567, + "grad_norm": 2.6579184532165527, + "learning_rate": 4.219642177388656e-06, + "loss": 1.5022, + "step": 717 + }, + { + "epoch": 0.7375449409347714, + "grad_norm": 2.7092721462249756, + "learning_rate": 4.2177388656261896e-06, + "loss": 1.5307, + "step": 718 + }, + { + "epoch": 0.7385721623009759, + "grad_norm": 2.65014910697937, + "learning_rate": 4.215835553863723e-06, + "loss": 1.4936, + "step": 719 + }, + { + "epoch": 0.7395993836671803, + "grad_norm": 2.7403628826141357, + "learning_rate": 4.213932242101256e-06, + "loss": 1.4458, + "step": 720 + }, + { + "epoch": 0.7395993836671803, + "eval_loss": 1.8778676986694336, + "eval_runtime": 36.9378, + "eval_samples_per_second": 27.073, + "eval_steps_per_second": 1.137, + "step": 720 + }, + { + "epoch": 0.7406266050333847, + "grad_norm": 2.3937759399414062, + "learning_rate": 4.21202893033879e-06, + "loss": 1.5086, + "step": 721 + }, + { + "epoch": 0.7416538263995891, + "grad_norm": 2.5786867141723633, + "learning_rate": 4.210125618576323e-06, + "loss": 1.5321, + "step": 722 + }, + { + "epoch": 0.7426810477657936, + "grad_norm": 2.607236862182617, + "learning_rate": 4.208222306813856e-06, + "loss": 1.5244, + "step": 723 + }, + { + "epoch": 0.743708269131998, + "grad_norm": 2.5461602210998535, + "learning_rate": 4.20631899505139e-06, + "loss": 1.5841, + "step": 724 + }, + { + "epoch": 0.7447354904982023, + "grad_norm": 2.5051612854003906, + "learning_rate": 4.204415683288923e-06, + "loss": 1.43, + "step": 725 + }, + { + "epoch": 0.7457627118644068, + "grad_norm": 2.692816734313965, + "learning_rate": 4.2025123715264565e-06, + "loss": 1.5439, + "step": 726 + }, + { + "epoch": 0.7467899332306112, + "grad_norm": 2.4827022552490234, + "learning_rate": 4.20060905976399e-06, + "loss": 1.4987, + "step": 727 + }, + { + "epoch": 0.7478171545968156, + "grad_norm": 2.7261135578155518, + "learning_rate": 4.198705748001523e-06, + "loss": 1.4573, + "step": 728 + }, + { + "epoch": 0.74884437596302, + "grad_norm": 2.5710837841033936, + "learning_rate": 4.1968024362390566e-06, + "loss": 1.564, + "step": 729 + }, + { + "epoch": 0.7498715973292245, + "grad_norm": 2.61181640625, + "learning_rate": 4.19489912447659e-06, + "loss": 1.5211, + "step": 730 + }, + { + "epoch": 0.7498715973292245, + "eval_loss": 1.8794434070587158, + "eval_runtime": 36.9676, + "eval_samples_per_second": 27.051, + "eval_steps_per_second": 1.136, + "step": 730 + }, + { + "epoch": 0.7508988186954288, + "grad_norm": 2.6946470737457275, + "learning_rate": 4.192995812714123e-06, + "loss": 1.5837, + "step": 731 + }, + { + "epoch": 0.7519260400616333, + "grad_norm": 2.5337390899658203, + "learning_rate": 4.191092500951656e-06, + "loss": 1.5251, + "step": 732 + }, + { + "epoch": 0.7529532614278377, + "grad_norm": 2.541532278060913, + "learning_rate": 4.189189189189189e-06, + "loss": 1.503, + "step": 733 + }, + { + "epoch": 0.7539804827940421, + "grad_norm": 2.639124631881714, + "learning_rate": 4.1872858774267225e-06, + "loss": 1.4963, + "step": 734 + }, + { + "epoch": 0.7550077041602465, + "grad_norm": 2.872861862182617, + "learning_rate": 4.185382565664256e-06, + "loss": 1.5632, + "step": 735 + }, + { + "epoch": 0.7560349255264509, + "grad_norm": 2.588980197906494, + "learning_rate": 4.183479253901789e-06, + "loss": 1.4998, + "step": 736 + }, + { + "epoch": 0.7570621468926554, + "grad_norm": 2.9199588298797607, + "learning_rate": 4.181575942139323e-06, + "loss": 1.466, + "step": 737 + }, + { + "epoch": 0.7580893682588598, + "grad_norm": 2.5677788257598877, + "learning_rate": 4.179672630376856e-06, + "loss": 1.5247, + "step": 738 + }, + { + "epoch": 0.7591165896250642, + "grad_norm": 2.481142520904541, + "learning_rate": 4.177769318614389e-06, + "loss": 1.4915, + "step": 739 + }, + { + "epoch": 0.7601438109912686, + "grad_norm": 2.7072012424468994, + "learning_rate": 4.175866006851923e-06, + "loss": 1.5783, + "step": 740 + }, + { + "epoch": 0.7601438109912686, + "eval_loss": 1.8817057609558105, + "eval_runtime": 36.9635, + "eval_samples_per_second": 27.054, + "eval_steps_per_second": 1.136, + "step": 740 + }, + { + "epoch": 0.7611710323574731, + "grad_norm": 2.5433032512664795, + "learning_rate": 4.173962695089456e-06, + "loss": 1.5457, + "step": 741 + }, + { + "epoch": 0.7621982537236774, + "grad_norm": 2.702077627182007, + "learning_rate": 4.1720593833269894e-06, + "loss": 1.5285, + "step": 742 + }, + { + "epoch": 0.7632254750898819, + "grad_norm": 2.6678309440612793, + "learning_rate": 4.170156071564523e-06, + "loss": 1.5381, + "step": 743 + }, + { + "epoch": 0.7642526964560863, + "grad_norm": 2.6097185611724854, + "learning_rate": 4.168252759802056e-06, + "loss": 1.6076, + "step": 744 + }, + { + "epoch": 0.7652799178222907, + "grad_norm": 2.5975704193115234, + "learning_rate": 4.1663494480395895e-06, + "loss": 1.515, + "step": 745 + }, + { + "epoch": 0.7663071391884951, + "grad_norm": 2.6858558654785156, + "learning_rate": 4.164446136277123e-06, + "loss": 1.5499, + "step": 746 + }, + { + "epoch": 0.7673343605546995, + "grad_norm": 2.565023899078369, + "learning_rate": 4.162542824514656e-06, + "loss": 1.5502, + "step": 747 + }, + { + "epoch": 0.768361581920904, + "grad_norm": 2.61715030670166, + "learning_rate": 4.16063951275219e-06, + "loss": 1.5271, + "step": 748 + }, + { + "epoch": 0.7693888032871083, + "grad_norm": 2.564382314682007, + "learning_rate": 4.158736200989722e-06, + "loss": 1.4756, + "step": 749 + }, + { + "epoch": 0.7704160246533128, + "grad_norm": 2.64457368850708, + "learning_rate": 4.1568328892272555e-06, + "loss": 1.5246, + "step": 750 + }, + { + "epoch": 0.7704160246533128, + "eval_loss": 1.8777766227722168, + "eval_runtime": 36.9737, + "eval_samples_per_second": 27.046, + "eval_steps_per_second": 1.136, + "step": 750 + }, + { + "epoch": 0.7714432460195172, + "grad_norm": 2.773602247238159, + "learning_rate": 4.154929577464789e-06, + "loss": 1.5337, + "step": 751 + }, + { + "epoch": 0.7724704673857217, + "grad_norm": 2.5466997623443604, + "learning_rate": 4.153026265702322e-06, + "loss": 1.5269, + "step": 752 + }, + { + "epoch": 0.773497688751926, + "grad_norm": 2.702888250350952, + "learning_rate": 4.1511229539398556e-06, + "loss": 1.4636, + "step": 753 + }, + { + "epoch": 0.7745249101181304, + "grad_norm": 2.629251718521118, + "learning_rate": 4.149219642177389e-06, + "loss": 1.513, + "step": 754 + }, + { + "epoch": 0.7755521314843349, + "grad_norm": 2.52544903755188, + "learning_rate": 4.147316330414922e-06, + "loss": 1.5225, + "step": 755 + }, + { + "epoch": 0.7765793528505393, + "grad_norm": 2.4267778396606445, + "learning_rate": 4.145413018652456e-06, + "loss": 1.4377, + "step": 756 + }, + { + "epoch": 0.7776065742167437, + "grad_norm": 2.6061577796936035, + "learning_rate": 4.143509706889989e-06, + "loss": 1.5402, + "step": 757 + }, + { + "epoch": 0.7786337955829481, + "grad_norm": 2.6571085453033447, + "learning_rate": 4.141606395127522e-06, + "loss": 1.527, + "step": 758 + }, + { + "epoch": 0.7796610169491526, + "grad_norm": 2.48740291595459, + "learning_rate": 4.139703083365056e-06, + "loss": 1.517, + "step": 759 + }, + { + "epoch": 0.7806882383153569, + "grad_norm": 2.5044426918029785, + "learning_rate": 4.137799771602588e-06, + "loss": 1.5579, + "step": 760 + }, + { + "epoch": 0.7806882383153569, + "eval_loss": 1.8778951168060303, + "eval_runtime": 37.0045, + "eval_samples_per_second": 27.024, + "eval_steps_per_second": 1.135, + "step": 760 + }, + { + "epoch": 0.7817154596815614, + "grad_norm": 2.4728074073791504, + "learning_rate": 4.135896459840122e-06, + "loss": 1.5321, + "step": 761 + }, + { + "epoch": 0.7827426810477658, + "grad_norm": 2.6541659832000732, + "learning_rate": 4.133993148077656e-06, + "loss": 1.4866, + "step": 762 + }, + { + "epoch": 0.7837699024139703, + "grad_norm": 2.526716947555542, + "learning_rate": 4.132089836315189e-06, + "loss": 1.5035, + "step": 763 + }, + { + "epoch": 0.7847971237801746, + "grad_norm": 2.683306932449341, + "learning_rate": 4.1301865245527226e-06, + "loss": 1.5221, + "step": 764 + }, + { + "epoch": 0.785824345146379, + "grad_norm": 2.4980056285858154, + "learning_rate": 4.128283212790256e-06, + "loss": 1.4643, + "step": 765 + }, + { + "epoch": 0.7868515665125835, + "grad_norm": 2.3985488414764404, + "learning_rate": 4.1263799010277884e-06, + "loss": 1.4348, + "step": 766 + }, + { + "epoch": 0.7878787878787878, + "grad_norm": 2.600689649581909, + "learning_rate": 4.124476589265322e-06, + "loss": 1.5647, + "step": 767 + }, + { + "epoch": 0.7889060092449923, + "grad_norm": 2.5934395790100098, + "learning_rate": 4.122573277502855e-06, + "loss": 1.4887, + "step": 768 + }, + { + "epoch": 0.7899332306111967, + "grad_norm": 2.5209388732910156, + "learning_rate": 4.1206699657403885e-06, + "loss": 1.4893, + "step": 769 + }, + { + "epoch": 0.7909604519774012, + "grad_norm": 2.6656298637390137, + "learning_rate": 4.118766653977922e-06, + "loss": 1.5239, + "step": 770 + }, + { + "epoch": 0.7909604519774012, + "eval_loss": 1.8737632036209106, + "eval_runtime": 37.056, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 770 + }, + { + "epoch": 0.7919876733436055, + "grad_norm": 2.5147693157196045, + "learning_rate": 4.116863342215455e-06, + "loss": 1.5, + "step": 771 + }, + { + "epoch": 0.79301489470981, + "grad_norm": 2.4653773307800293, + "learning_rate": 4.114960030452989e-06, + "loss": 1.4802, + "step": 772 + }, + { + "epoch": 0.7940421160760144, + "grad_norm": 2.643160820007324, + "learning_rate": 4.113056718690522e-06, + "loss": 1.5086, + "step": 773 + }, + { + "epoch": 0.7950693374422187, + "grad_norm": 2.4873125553131104, + "learning_rate": 4.111153406928055e-06, + "loss": 1.4558, + "step": 774 + }, + { + "epoch": 0.7960965588084232, + "grad_norm": 2.541308879852295, + "learning_rate": 4.109250095165589e-06, + "loss": 1.5483, + "step": 775 + }, + { + "epoch": 0.7971237801746276, + "grad_norm": 2.701874256134033, + "learning_rate": 4.107346783403122e-06, + "loss": 1.5551, + "step": 776 + }, + { + "epoch": 0.7981510015408321, + "grad_norm": 2.457371234893799, + "learning_rate": 4.105443471640655e-06, + "loss": 1.4968, + "step": 777 + }, + { + "epoch": 0.7991782229070364, + "grad_norm": 2.5294973850250244, + "learning_rate": 4.103540159878188e-06, + "loss": 1.4756, + "step": 778 + }, + { + "epoch": 0.8002054442732409, + "grad_norm": 2.464932918548584, + "learning_rate": 4.101636848115721e-06, + "loss": 1.5215, + "step": 779 + }, + { + "epoch": 0.8012326656394453, + "grad_norm": 2.5919454097747803, + "learning_rate": 4.099733536353255e-06, + "loss": 1.4807, + "step": 780 + }, + { + "epoch": 0.8012326656394453, + "eval_loss": 1.875250220298767, + "eval_runtime": 37.035, + "eval_samples_per_second": 27.001, + "eval_steps_per_second": 1.134, + "step": 780 + }, + { + "epoch": 0.8022598870056498, + "grad_norm": 2.505749464035034, + "learning_rate": 4.097830224590788e-06, + "loss": 1.5224, + "step": 781 + }, + { + "epoch": 0.8032871083718541, + "grad_norm": 2.747262954711914, + "learning_rate": 4.095926912828322e-06, + "loss": 1.6051, + "step": 782 + }, + { + "epoch": 0.8043143297380585, + "grad_norm": 2.6147146224975586, + "learning_rate": 4.094023601065855e-06, + "loss": 1.565, + "step": 783 + }, + { + "epoch": 0.805341551104263, + "grad_norm": 2.5381031036376953, + "learning_rate": 4.092120289303388e-06, + "loss": 1.511, + "step": 784 + }, + { + "epoch": 0.8063687724704673, + "grad_norm": 2.6476190090179443, + "learning_rate": 4.0902169775409215e-06, + "loss": 1.5147, + "step": 785 + }, + { + "epoch": 0.8073959938366718, + "grad_norm": 2.486919641494751, + "learning_rate": 4.088313665778455e-06, + "loss": 1.4697, + "step": 786 + }, + { + "epoch": 0.8084232152028762, + "grad_norm": 2.7006654739379883, + "learning_rate": 4.086410354015988e-06, + "loss": 1.4965, + "step": 787 + }, + { + "epoch": 0.8094504365690807, + "grad_norm": 2.7587921619415283, + "learning_rate": 4.0845070422535216e-06, + "loss": 1.4781, + "step": 788 + }, + { + "epoch": 0.810477657935285, + "grad_norm": 2.64239501953125, + "learning_rate": 4.082603730491055e-06, + "loss": 1.4945, + "step": 789 + }, + { + "epoch": 0.8115048793014895, + "grad_norm": 2.659919261932373, + "learning_rate": 4.080700418728588e-06, + "loss": 1.5378, + "step": 790 + }, + { + "epoch": 0.8115048793014895, + "eval_loss": 1.881503939628601, + "eval_runtime": 37.2277, + "eval_samples_per_second": 26.862, + "eval_steps_per_second": 1.128, + "step": 790 + }, + { + "epoch": 0.8125321006676939, + "grad_norm": 2.820996046066284, + "learning_rate": 4.078797106966122e-06, + "loss": 1.5134, + "step": 791 + }, + { + "epoch": 0.8135593220338984, + "grad_norm": 2.4667489528656006, + "learning_rate": 4.076893795203655e-06, + "loss": 1.4165, + "step": 792 + }, + { + "epoch": 0.8145865434001027, + "grad_norm": 2.5528056621551514, + "learning_rate": 4.074990483441188e-06, + "loss": 1.481, + "step": 793 + }, + { + "epoch": 0.8156137647663071, + "grad_norm": 2.632922887802124, + "learning_rate": 4.073087171678721e-06, + "loss": 1.5502, + "step": 794 + }, + { + "epoch": 0.8166409861325116, + "grad_norm": 2.662536144256592, + "learning_rate": 4.071183859916254e-06, + "loss": 1.5302, + "step": 795 + }, + { + "epoch": 0.8176682074987159, + "grad_norm": 2.662454605102539, + "learning_rate": 4.069280548153788e-06, + "loss": 1.5131, + "step": 796 + }, + { + "epoch": 0.8186954288649204, + "grad_norm": 2.6387386322021484, + "learning_rate": 4.067377236391321e-06, + "loss": 1.4677, + "step": 797 + }, + { + "epoch": 0.8197226502311248, + "grad_norm": 2.8879177570343018, + "learning_rate": 4.065473924628854e-06, + "loss": 1.5106, + "step": 798 + }, + { + "epoch": 0.8207498715973293, + "grad_norm": 2.680610418319702, + "learning_rate": 4.063570612866388e-06, + "loss": 1.5389, + "step": 799 + }, + { + "epoch": 0.8217770929635336, + "grad_norm": 2.659590244293213, + "learning_rate": 4.061667301103921e-06, + "loss": 1.5254, + "step": 800 + }, + { + "epoch": 0.8217770929635336, + "eval_loss": 1.881851077079773, + "eval_runtime": 37.3385, + "eval_samples_per_second": 26.782, + "eval_steps_per_second": 1.125, + "step": 800 + }, + { + "epoch": 0.8228043143297381, + "grad_norm": 2.711195945739746, + "learning_rate": 4.0597639893414544e-06, + "loss": 1.5026, + "step": 801 + }, + { + "epoch": 0.8238315356959425, + "grad_norm": 2.6387939453125, + "learning_rate": 4.057860677578988e-06, + "loss": 1.5101, + "step": 802 + }, + { + "epoch": 0.8248587570621468, + "grad_norm": 2.507387638092041, + "learning_rate": 4.055957365816521e-06, + "loss": 1.5277, + "step": 803 + }, + { + "epoch": 0.8258859784283513, + "grad_norm": 2.610370397567749, + "learning_rate": 4.0540540540540545e-06, + "loss": 1.4695, + "step": 804 + }, + { + "epoch": 0.8269131997945557, + "grad_norm": 2.8345787525177, + "learning_rate": 4.052150742291588e-06, + "loss": 1.467, + "step": 805 + }, + { + "epoch": 0.8279404211607602, + "grad_norm": 2.5142805576324463, + "learning_rate": 4.050247430529121e-06, + "loss": 1.4637, + "step": 806 + }, + { + "epoch": 0.8289676425269645, + "grad_norm": 2.7722535133361816, + "learning_rate": 4.048344118766655e-06, + "loss": 1.5427, + "step": 807 + }, + { + "epoch": 0.829994863893169, + "grad_norm": 2.816789150238037, + "learning_rate": 4.046440807004188e-06, + "loss": 1.5422, + "step": 808 + }, + { + "epoch": 0.8310220852593734, + "grad_norm": 2.6485376358032227, + "learning_rate": 4.044537495241721e-06, + "loss": 1.5078, + "step": 809 + }, + { + "epoch": 0.8320493066255779, + "grad_norm": 2.4760921001434326, + "learning_rate": 4.042634183479255e-06, + "loss": 1.4827, + "step": 810 + }, + { + "epoch": 0.8320493066255779, + "eval_loss": 1.8709155321121216, + "eval_runtime": 36.9015, + "eval_samples_per_second": 27.099, + "eval_steps_per_second": 1.138, + "step": 810 + }, + { + "epoch": 0.8330765279917822, + "grad_norm": 2.658062219619751, + "learning_rate": 4.040730871716787e-06, + "loss": 1.5434, + "step": 811 + }, + { + "epoch": 0.8341037493579867, + "grad_norm": 2.5443618297576904, + "learning_rate": 4.038827559954321e-06, + "loss": 1.5165, + "step": 812 + }, + { + "epoch": 0.8351309707241911, + "grad_norm": 2.698432207107544, + "learning_rate": 4.036924248191854e-06, + "loss": 1.5553, + "step": 813 + }, + { + "epoch": 0.8361581920903954, + "grad_norm": 2.5598180294036865, + "learning_rate": 4.035020936429387e-06, + "loss": 1.5289, + "step": 814 + }, + { + "epoch": 0.8371854134565999, + "grad_norm": 2.7002663612365723, + "learning_rate": 4.033117624666921e-06, + "loss": 1.509, + "step": 815 + }, + { + "epoch": 0.8382126348228043, + "grad_norm": 2.7882776260375977, + "learning_rate": 4.031214312904454e-06, + "loss": 1.4787, + "step": 816 + }, + { + "epoch": 0.8392398561890088, + "grad_norm": 2.5583086013793945, + "learning_rate": 4.029311001141987e-06, + "loss": 1.4642, + "step": 817 + }, + { + "epoch": 0.8402670775552131, + "grad_norm": 2.584573268890381, + "learning_rate": 4.027407689379521e-06, + "loss": 1.4343, + "step": 818 + }, + { + "epoch": 0.8412942989214176, + "grad_norm": 2.5062575340270996, + "learning_rate": 4.025504377617054e-06, + "loss": 1.4975, + "step": 819 + }, + { + "epoch": 0.842321520287622, + "grad_norm": 2.3546457290649414, + "learning_rate": 4.0236010658545875e-06, + "loss": 1.4795, + "step": 820 + }, + { + "epoch": 0.842321520287622, + "eval_loss": 1.8716342449188232, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 820 + }, + { + "epoch": 0.8433487416538265, + "grad_norm": 2.513228178024292, + "learning_rate": 4.02169775409212e-06, + "loss": 1.5191, + "step": 821 + }, + { + "epoch": 0.8443759630200308, + "grad_norm": 2.6360220909118652, + "learning_rate": 4.019794442329653e-06, + "loss": 1.5739, + "step": 822 + }, + { + "epoch": 0.8454031843862352, + "grad_norm": 2.5351850986480713, + "learning_rate": 4.0178911305671876e-06, + "loss": 1.4487, + "step": 823 + }, + { + "epoch": 0.8464304057524397, + "grad_norm": 2.6693851947784424, + "learning_rate": 4.015987818804721e-06, + "loss": 1.5319, + "step": 824 + }, + { + "epoch": 0.847457627118644, + "grad_norm": 2.5210046768188477, + "learning_rate": 4.014084507042254e-06, + "loss": 1.4723, + "step": 825 + }, + { + "epoch": 0.8484848484848485, + "grad_norm": 2.5799994468688965, + "learning_rate": 4.012181195279788e-06, + "loss": 1.4409, + "step": 826 + }, + { + "epoch": 0.8495120698510529, + "grad_norm": 2.5206313133239746, + "learning_rate": 4.010277883517321e-06, + "loss": 1.5449, + "step": 827 + }, + { + "epoch": 0.8505392912172574, + "grad_norm": 2.480584144592285, + "learning_rate": 4.0083745717548535e-06, + "loss": 1.5306, + "step": 828 + }, + { + "epoch": 0.8515665125834617, + "grad_norm": 2.5229053497314453, + "learning_rate": 4.006471259992387e-06, + "loss": 1.5001, + "step": 829 + }, + { + "epoch": 0.8525937339496662, + "grad_norm": 2.53357195854187, + "learning_rate": 4.00456794822992e-06, + "loss": 1.5188, + "step": 830 + }, + { + "epoch": 0.8525937339496662, + "eval_loss": 1.8724256753921509, + "eval_runtime": 37.1114, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 830 + }, + { + "epoch": 0.8536209553158706, + "grad_norm": 2.6796159744262695, + "learning_rate": 4.002664636467454e-06, + "loss": 1.5577, + "step": 831 + }, + { + "epoch": 0.8546481766820749, + "grad_norm": 2.4717817306518555, + "learning_rate": 4.000761324704987e-06, + "loss": 1.5124, + "step": 832 + }, + { + "epoch": 0.8556753980482794, + "grad_norm": 2.649057626724243, + "learning_rate": 3.99885801294252e-06, + "loss": 1.4516, + "step": 833 + }, + { + "epoch": 0.8567026194144838, + "grad_norm": 2.592406749725342, + "learning_rate": 3.996954701180054e-06, + "loss": 1.5215, + "step": 834 + }, + { + "epoch": 0.8577298407806883, + "grad_norm": 2.46956205368042, + "learning_rate": 3.995051389417587e-06, + "loss": 1.4324, + "step": 835 + }, + { + "epoch": 0.8587570621468926, + "grad_norm": 2.8484537601470947, + "learning_rate": 3.9931480776551204e-06, + "loss": 1.4574, + "step": 836 + }, + { + "epoch": 0.8597842835130971, + "grad_norm": 2.6475489139556885, + "learning_rate": 3.991244765892654e-06, + "loss": 1.5581, + "step": 837 + }, + { + "epoch": 0.8608115048793015, + "grad_norm": 2.527153491973877, + "learning_rate": 3.989341454130186e-06, + "loss": 1.4713, + "step": 838 + }, + { + "epoch": 0.861838726245506, + "grad_norm": 2.6767020225524902, + "learning_rate": 3.98743814236772e-06, + "loss": 1.4459, + "step": 839 + }, + { + "epoch": 0.8628659476117103, + "grad_norm": 2.6406850814819336, + "learning_rate": 3.985534830605253e-06, + "loss": 1.4964, + "step": 840 + }, + { + "epoch": 0.8628659476117103, + "eval_loss": 1.886762261390686, + "eval_runtime": 37.1378, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 840 + }, + { + "epoch": 0.8638931689779148, + "grad_norm": 2.466372489929199, + "learning_rate": 3.983631518842786e-06, + "loss": 1.5136, + "step": 841 + }, + { + "epoch": 0.8649203903441192, + "grad_norm": 2.830249071121216, + "learning_rate": 3.98172820708032e-06, + "loss": 1.5372, + "step": 842 + }, + { + "epoch": 0.8659476117103235, + "grad_norm": 2.675889015197754, + "learning_rate": 3.979824895317854e-06, + "loss": 1.4881, + "step": 843 + }, + { + "epoch": 0.866974833076528, + "grad_norm": 2.7418212890625, + "learning_rate": 3.977921583555387e-06, + "loss": 1.4541, + "step": 844 + }, + { + "epoch": 0.8680020544427324, + "grad_norm": 2.572862386703491, + "learning_rate": 3.97601827179292e-06, + "loss": 1.5191, + "step": 845 + }, + { + "epoch": 0.8690292758089369, + "grad_norm": 2.6524274349212646, + "learning_rate": 3.974114960030453e-06, + "loss": 1.4647, + "step": 846 + }, + { + "epoch": 0.8700564971751412, + "grad_norm": 2.794415235519409, + "learning_rate": 3.9722116482679866e-06, + "loss": 1.504, + "step": 847 + }, + { + "epoch": 0.8710837185413457, + "grad_norm": 2.5947723388671875, + "learning_rate": 3.97030833650552e-06, + "loss": 1.492, + "step": 848 + }, + { + "epoch": 0.8721109399075501, + "grad_norm": 2.8539984226226807, + "learning_rate": 3.968405024743053e-06, + "loss": 1.5049, + "step": 849 + }, + { + "epoch": 0.8731381612737545, + "grad_norm": 2.522786855697632, + "learning_rate": 3.966501712980587e-06, + "loss": 1.4223, + "step": 850 + }, + { + "epoch": 0.8731381612737545, + "eval_loss": 1.8691684007644653, + "eval_runtime": 37.1287, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 850 + }, + { + "epoch": 0.8741653826399589, + "grad_norm": 2.6483359336853027, + "learning_rate": 3.96459840121812e-06, + "loss": 1.4815, + "step": 851 + }, + { + "epoch": 0.8751926040061633, + "grad_norm": 2.581429958343506, + "learning_rate": 3.962695089455653e-06, + "loss": 1.4837, + "step": 852 + }, + { + "epoch": 0.8762198253723678, + "grad_norm": 2.5967583656311035, + "learning_rate": 3.960791777693187e-06, + "loss": 1.5289, + "step": 853 + }, + { + "epoch": 0.8772470467385721, + "grad_norm": 2.7169225215911865, + "learning_rate": 3.95888846593072e-06, + "loss": 1.5173, + "step": 854 + }, + { + "epoch": 0.8782742681047766, + "grad_norm": 2.644656181335449, + "learning_rate": 3.956985154168253e-06, + "loss": 1.5095, + "step": 855 + }, + { + "epoch": 0.879301489470981, + "grad_norm": 2.6397511959075928, + "learning_rate": 3.955081842405786e-06, + "loss": 1.5422, + "step": 856 + }, + { + "epoch": 0.8803287108371854, + "grad_norm": 2.691540479660034, + "learning_rate": 3.953178530643319e-06, + "loss": 1.5703, + "step": 857 + }, + { + "epoch": 0.8813559322033898, + "grad_norm": 2.50526762008667, + "learning_rate": 3.951275218880853e-06, + "loss": 1.5885, + "step": 858 + }, + { + "epoch": 0.8823831535695943, + "grad_norm": 2.3846609592437744, + "learning_rate": 3.949371907118386e-06, + "loss": 1.4248, + "step": 859 + }, + { + "epoch": 0.8834103749357987, + "grad_norm": 2.7107760906219482, + "learning_rate": 3.9474685953559194e-06, + "loss": 1.5358, + "step": 860 + }, + { + "epoch": 0.8834103749357987, + "eval_loss": 1.876391887664795, + "eval_runtime": 37.1284, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.884437596302003, + "grad_norm": 2.6206252574920654, + "learning_rate": 3.945565283593453e-06, + "loss": 1.4907, + "step": 861 + }, + { + "epoch": 0.8854648176682075, + "grad_norm": 2.6641664505004883, + "learning_rate": 3.943661971830986e-06, + "loss": 1.4888, + "step": 862 + }, + { + "epoch": 0.8864920390344119, + "grad_norm": 2.7342002391815186, + "learning_rate": 3.9417586600685195e-06, + "loss": 1.5421, + "step": 863 + }, + { + "epoch": 0.8875192604006163, + "grad_norm": 2.7120890617370605, + "learning_rate": 3.939855348306053e-06, + "loss": 1.5154, + "step": 864 + }, + { + "epoch": 0.8885464817668207, + "grad_norm": 2.6350717544555664, + "learning_rate": 3.937952036543586e-06, + "loss": 1.5135, + "step": 865 + }, + { + "epoch": 0.8895737031330252, + "grad_norm": 2.570857524871826, + "learning_rate": 3.93604872478112e-06, + "loss": 1.5072, + "step": 866 + }, + { + "epoch": 0.8906009244992296, + "grad_norm": 2.855071544647217, + "learning_rate": 3.934145413018653e-06, + "loss": 1.5171, + "step": 867 + }, + { + "epoch": 0.891628145865434, + "grad_norm": 2.5592195987701416, + "learning_rate": 3.932242101256186e-06, + "loss": 1.4695, + "step": 868 + }, + { + "epoch": 0.8926553672316384, + "grad_norm": 2.6110830307006836, + "learning_rate": 3.93033878949372e-06, + "loss": 1.4802, + "step": 869 + }, + { + "epoch": 0.8936825885978429, + "grad_norm": 2.7121548652648926, + "learning_rate": 3.928435477731253e-06, + "loss": 1.4422, + "step": 870 + }, + { + "epoch": 0.8936825885978429, + "eval_loss": 1.8629941940307617, + "eval_runtime": 37.1091, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.8947098099640473, + "grad_norm": 2.7763099670410156, + "learning_rate": 3.9265321659687864e-06, + "loss": 1.5275, + "step": 871 + }, + { + "epoch": 0.8957370313302516, + "grad_norm": 2.596220016479492, + "learning_rate": 3.924628854206319e-06, + "loss": 1.5374, + "step": 872 + }, + { + "epoch": 0.8967642526964561, + "grad_norm": 2.575990915298462, + "learning_rate": 3.922725542443852e-06, + "loss": 1.5071, + "step": 873 + }, + { + "epoch": 0.8977914740626605, + "grad_norm": 2.7052223682403564, + "learning_rate": 3.920822230681386e-06, + "loss": 1.4656, + "step": 874 + }, + { + "epoch": 0.8988186954288649, + "grad_norm": 2.550267219543457, + "learning_rate": 3.918918918918919e-06, + "loss": 1.4837, + "step": 875 + }, + { + "epoch": 0.8998459167950693, + "grad_norm": 2.6948060989379883, + "learning_rate": 3.917015607156452e-06, + "loss": 1.5156, + "step": 876 + }, + { + "epoch": 0.9008731381612738, + "grad_norm": 2.575596809387207, + "learning_rate": 3.915112295393986e-06, + "loss": 1.4881, + "step": 877 + }, + { + "epoch": 0.9019003595274782, + "grad_norm": 2.9835500717163086, + "learning_rate": 3.913208983631519e-06, + "loss": 1.5729, + "step": 878 + }, + { + "epoch": 0.9029275808936826, + "grad_norm": 2.5961520671844482, + "learning_rate": 3.9113056718690525e-06, + "loss": 1.5469, + "step": 879 + }, + { + "epoch": 0.903954802259887, + "grad_norm": 2.518260955810547, + "learning_rate": 3.909402360106586e-06, + "loss": 1.4704, + "step": 880 + }, + { + "epoch": 0.903954802259887, + "eval_loss": 1.8570330142974854, + "eval_runtime": 37.1293, + "eval_samples_per_second": 26.933, + "eval_steps_per_second": 1.131, + "step": 880 + }, + { + "epoch": 0.9049820236260914, + "grad_norm": 2.740879535675049, + "learning_rate": 3.907499048344119e-06, + "loss": 1.5092, + "step": 881 + }, + { + "epoch": 0.9060092449922958, + "grad_norm": 2.5227973461151123, + "learning_rate": 3.905595736581652e-06, + "loss": 1.5252, + "step": 882 + }, + { + "epoch": 0.9070364663585002, + "grad_norm": 2.483125925064087, + "learning_rate": 3.903692424819186e-06, + "loss": 1.5216, + "step": 883 + }, + { + "epoch": 0.9080636877247047, + "grad_norm": 2.4979538917541504, + "learning_rate": 3.901789113056719e-06, + "loss": 1.4734, + "step": 884 + }, + { + "epoch": 0.9090909090909091, + "grad_norm": 2.707669496536255, + "learning_rate": 3.899885801294253e-06, + "loss": 1.5347, + "step": 885 + }, + { + "epoch": 0.9101181304571135, + "grad_norm": 2.6548409461975098, + "learning_rate": 3.897982489531786e-06, + "loss": 1.5079, + "step": 886 + }, + { + "epoch": 0.9111453518233179, + "grad_norm": 2.6678848266601562, + "learning_rate": 3.896079177769319e-06, + "loss": 1.5659, + "step": 887 + }, + { + "epoch": 0.9121725731895224, + "grad_norm": 2.570976734161377, + "learning_rate": 3.894175866006853e-06, + "loss": 1.5281, + "step": 888 + }, + { + "epoch": 0.9131997945557268, + "grad_norm": 2.5939226150512695, + "learning_rate": 3.892272554244385e-06, + "loss": 1.56, + "step": 889 + }, + { + "epoch": 0.9142270159219312, + "grad_norm": 2.4577109813690186, + "learning_rate": 3.890369242481919e-06, + "loss": 1.4702, + "step": 890 + }, + { + "epoch": 0.9142270159219312, + "eval_loss": 1.8534926176071167, + "eval_runtime": 37.3353, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 890 + }, + { + "epoch": 0.9152542372881356, + "grad_norm": 2.5012643337249756, + "learning_rate": 3.888465930719452e-06, + "loss": 1.5411, + "step": 891 + }, + { + "epoch": 0.91628145865434, + "grad_norm": 2.470905065536499, + "learning_rate": 3.886562618956985e-06, + "loss": 1.4766, + "step": 892 + }, + { + "epoch": 0.9173086800205444, + "grad_norm": 2.4610393047332764, + "learning_rate": 3.884659307194519e-06, + "loss": 1.4925, + "step": 893 + }, + { + "epoch": 0.9183359013867488, + "grad_norm": 2.573151111602783, + "learning_rate": 3.882755995432052e-06, + "loss": 1.4389, + "step": 894 + }, + { + "epoch": 0.9193631227529533, + "grad_norm": 2.628934144973755, + "learning_rate": 3.8808526836695854e-06, + "loss": 1.5099, + "step": 895 + }, + { + "epoch": 0.9203903441191577, + "grad_norm": 2.639920949935913, + "learning_rate": 3.878949371907119e-06, + "loss": 1.4724, + "step": 896 + }, + { + "epoch": 0.9214175654853621, + "grad_norm": 2.4775230884552, + "learning_rate": 3.877046060144652e-06, + "loss": 1.4657, + "step": 897 + }, + { + "epoch": 0.9224447868515665, + "grad_norm": 2.5242490768432617, + "learning_rate": 3.8751427483821855e-06, + "loss": 1.459, + "step": 898 + }, + { + "epoch": 0.923472008217771, + "grad_norm": 2.4444870948791504, + "learning_rate": 3.873239436619718e-06, + "loss": 1.494, + "step": 899 + }, + { + "epoch": 0.9244992295839753, + "grad_norm": 2.4326930046081543, + "learning_rate": 3.871336124857251e-06, + "loss": 1.4449, + "step": 900 + }, + { + "epoch": 0.9244992295839753, + "eval_loss": 1.8844366073608398, + "eval_runtime": 37.432, + "eval_samples_per_second": 26.715, + "eval_steps_per_second": 1.122, + "step": 900 + }, + { + "epoch": 0.9255264509501797, + "grad_norm": 2.621652603149414, + "learning_rate": 3.869432813094785e-06, + "loss": 1.5055, + "step": 901 + }, + { + "epoch": 0.9265536723163842, + "grad_norm": 2.5909814834594727, + "learning_rate": 3.867529501332318e-06, + "loss": 1.5176, + "step": 902 + }, + { + "epoch": 0.9275808936825886, + "grad_norm": 2.512760877609253, + "learning_rate": 3.8656261895698515e-06, + "loss": 1.5027, + "step": 903 + }, + { + "epoch": 0.928608115048793, + "grad_norm": 2.6954784393310547, + "learning_rate": 3.863722877807386e-06, + "loss": 1.4912, + "step": 904 + }, + { + "epoch": 0.9296353364149974, + "grad_norm": 2.481905221939087, + "learning_rate": 3.861819566044919e-06, + "loss": 1.529, + "step": 905 + }, + { + "epoch": 0.9306625577812019, + "grad_norm": 2.612758159637451, + "learning_rate": 3.859916254282452e-06, + "loss": 1.5658, + "step": 906 + }, + { + "epoch": 0.9316897791474062, + "grad_norm": 2.523171901702881, + "learning_rate": 3.858012942519985e-06, + "loss": 1.4804, + "step": 907 + }, + { + "epoch": 0.9327170005136107, + "grad_norm": 2.5053422451019287, + "learning_rate": 3.856109630757518e-06, + "loss": 1.5007, + "step": 908 + }, + { + "epoch": 0.9337442218798151, + "grad_norm": 2.6110904216766357, + "learning_rate": 3.854206318995052e-06, + "loss": 1.4758, + "step": 909 + }, + { + "epoch": 0.9347714432460195, + "grad_norm": 2.7004992961883545, + "learning_rate": 3.852303007232585e-06, + "loss": 1.5255, + "step": 910 + }, + { + "epoch": 0.9347714432460195, + "eval_loss": 1.892087697982788, + "eval_runtime": 36.9356, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 910 + }, + { + "epoch": 0.9357986646122239, + "grad_norm": 2.617180109024048, + "learning_rate": 3.850399695470118e-06, + "loss": 1.4563, + "step": 911 + }, + { + "epoch": 0.9368258859784283, + "grad_norm": 2.5294559001922607, + "learning_rate": 3.848496383707652e-06, + "loss": 1.4767, + "step": 912 + }, + { + "epoch": 0.9378531073446328, + "grad_norm": 2.594238042831421, + "learning_rate": 3.846593071945185e-06, + "loss": 1.4617, + "step": 913 + }, + { + "epoch": 0.9388803287108372, + "grad_norm": 2.6076996326446533, + "learning_rate": 3.8446897601827185e-06, + "loss": 1.446, + "step": 914 + }, + { + "epoch": 0.9399075500770416, + "grad_norm": 2.6047134399414062, + "learning_rate": 3.842786448420252e-06, + "loss": 1.5698, + "step": 915 + }, + { + "epoch": 0.940934771443246, + "grad_norm": 2.5670995712280273, + "learning_rate": 3.840883136657785e-06, + "loss": 1.5803, + "step": 916 + }, + { + "epoch": 0.9419619928094505, + "grad_norm": 2.6153712272644043, + "learning_rate": 3.838979824895318e-06, + "loss": 1.5565, + "step": 917 + }, + { + "epoch": 0.9429892141756548, + "grad_norm": 2.6707098484039307, + "learning_rate": 3.837076513132851e-06, + "loss": 1.4417, + "step": 918 + }, + { + "epoch": 0.9440164355418593, + "grad_norm": 2.591336488723755, + "learning_rate": 3.8351732013703844e-06, + "loss": 1.5159, + "step": 919 + }, + { + "epoch": 0.9450436569080637, + "grad_norm": 2.4689993858337402, + "learning_rate": 3.833269889607918e-06, + "loss": 1.4735, + "step": 920 + }, + { + "epoch": 0.9450436569080637, + "eval_loss": 1.875109076499939, + "eval_runtime": 37.0384, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 920 + }, + { + "epoch": 0.9460708782742681, + "grad_norm": 2.739758253097534, + "learning_rate": 3.831366577845451e-06, + "loss": 1.4566, + "step": 921 + }, + { + "epoch": 0.9470980996404725, + "grad_norm": 2.65278697013855, + "learning_rate": 3.8294632660829845e-06, + "loss": 1.4545, + "step": 922 + }, + { + "epoch": 0.9481253210066769, + "grad_norm": 2.678321123123169, + "learning_rate": 3.827559954320518e-06, + "loss": 1.4864, + "step": 923 + }, + { + "epoch": 0.9491525423728814, + "grad_norm": 2.7516839504241943, + "learning_rate": 3.825656642558051e-06, + "loss": 1.5031, + "step": 924 + }, + { + "epoch": 0.9501797637390857, + "grad_norm": 2.6125247478485107, + "learning_rate": 3.823753330795585e-06, + "loss": 1.521, + "step": 925 + }, + { + "epoch": 0.9512069851052902, + "grad_norm": 2.5134871006011963, + "learning_rate": 3.821850019033118e-06, + "loss": 1.5019, + "step": 926 + }, + { + "epoch": 0.9522342064714946, + "grad_norm": 2.5274763107299805, + "learning_rate": 3.819946707270651e-06, + "loss": 1.4358, + "step": 927 + }, + { + "epoch": 0.9532614278376991, + "grad_norm": 2.480628728866577, + "learning_rate": 3.818043395508185e-06, + "loss": 1.483, + "step": 928 + }, + { + "epoch": 0.9542886492039034, + "grad_norm": 2.6450936794281006, + "learning_rate": 3.816140083745718e-06, + "loss": 1.4892, + "step": 929 + }, + { + "epoch": 0.9553158705701078, + "grad_norm": 2.6758296489715576, + "learning_rate": 3.8142367719832514e-06, + "loss": 1.5701, + "step": 930 + }, + { + "epoch": 0.9553158705701078, + "eval_loss": 1.8651176691055298, + "eval_runtime": 37.0553, + "eval_samples_per_second": 26.987, + "eval_steps_per_second": 1.133, + "step": 930 + }, + { + "epoch": 0.9563430919363123, + "grad_norm": 2.6887295246124268, + "learning_rate": 3.8123334602207844e-06, + "loss": 1.4596, + "step": 931 + }, + { + "epoch": 0.9573703133025167, + "grad_norm": 2.577507734298706, + "learning_rate": 3.8104301484583177e-06, + "loss": 1.4664, + "step": 932 + }, + { + "epoch": 0.9583975346687211, + "grad_norm": 2.6431517601013184, + "learning_rate": 3.808526836695851e-06, + "loss": 1.4715, + "step": 933 + }, + { + "epoch": 0.9594247560349255, + "grad_norm": 2.8451743125915527, + "learning_rate": 3.8066235249333845e-06, + "loss": 1.5096, + "step": 934 + }, + { + "epoch": 0.96045197740113, + "grad_norm": 2.5525193214416504, + "learning_rate": 3.804720213170918e-06, + "loss": 1.4975, + "step": 935 + }, + { + "epoch": 0.9614791987673343, + "grad_norm": 2.6353704929351807, + "learning_rate": 3.8028169014084508e-06, + "loss": 1.4979, + "step": 936 + }, + { + "epoch": 0.9625064201335388, + "grad_norm": 2.923781156539917, + "learning_rate": 3.800913589645984e-06, + "loss": 1.5118, + "step": 937 + }, + { + "epoch": 0.9635336414997432, + "grad_norm": 2.7363650798797607, + "learning_rate": 3.7990102778835175e-06, + "loss": 1.5016, + "step": 938 + }, + { + "epoch": 0.9645608628659477, + "grad_norm": 2.7126598358154297, + "learning_rate": 3.797106966121051e-06, + "loss": 1.5022, + "step": 939 + }, + { + "epoch": 0.965588084232152, + "grad_norm": 3.0102460384368896, + "learning_rate": 3.7952036543585842e-06, + "loss": 1.5638, + "step": 940 + }, + { + "epoch": 0.965588084232152, + "eval_loss": 1.8475635051727295, + "eval_runtime": 37.0871, + "eval_samples_per_second": 26.964, + "eval_steps_per_second": 1.132, + "step": 940 + }, + { + "epoch": 0.9666153055983564, + "grad_norm": 2.5455925464630127, + "learning_rate": 3.793300342596117e-06, + "loss": 1.5218, + "step": 941 + }, + { + "epoch": 0.9676425269645609, + "grad_norm": 2.5489683151245117, + "learning_rate": 3.7913970308336505e-06, + "loss": 1.5375, + "step": 942 + }, + { + "epoch": 0.9686697483307652, + "grad_norm": 2.672626256942749, + "learning_rate": 3.789493719071184e-06, + "loss": 1.5217, + "step": 943 + }, + { + "epoch": 0.9696969696969697, + "grad_norm": 2.498378038406372, + "learning_rate": 3.7875904073087177e-06, + "loss": 1.475, + "step": 944 + }, + { + "epoch": 0.9707241910631741, + "grad_norm": 2.515190362930298, + "learning_rate": 3.785687095546251e-06, + "loss": 1.4341, + "step": 945 + }, + { + "epoch": 0.9717514124293786, + "grad_norm": 2.7561733722686768, + "learning_rate": 3.7837837837837844e-06, + "loss": 1.4838, + "step": 946 + }, + { + "epoch": 0.9727786337955829, + "grad_norm": 2.544776201248169, + "learning_rate": 3.7818804720213178e-06, + "loss": 1.4539, + "step": 947 + }, + { + "epoch": 0.9738058551617874, + "grad_norm": 2.744417428970337, + "learning_rate": 3.7799771602588507e-06, + "loss": 1.4296, + "step": 948 + }, + { + "epoch": 0.9748330765279918, + "grad_norm": 2.7759811878204346, + "learning_rate": 3.778073848496384e-06, + "loss": 1.4314, + "step": 949 + }, + { + "epoch": 0.9758602978941961, + "grad_norm": 2.5204875469207764, + "learning_rate": 3.7761705367339174e-06, + "loss": 1.495, + "step": 950 + }, + { + "epoch": 0.9758602978941961, + "eval_loss": 1.8491986989974976, + "eval_runtime": 37.1266, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 0.9768875192604006, + "grad_norm": 2.5156631469726562, + "learning_rate": 3.7742672249714508e-06, + "loss": 1.5228, + "step": 951 + }, + { + "epoch": 0.977914740626605, + "grad_norm": 2.469834089279175, + "learning_rate": 3.772363913208984e-06, + "loss": 1.5188, + "step": 952 + }, + { + "epoch": 0.9789419619928095, + "grad_norm": 2.4933393001556396, + "learning_rate": 3.770460601446517e-06, + "loss": 1.4818, + "step": 953 + }, + { + "epoch": 0.9799691833590138, + "grad_norm": 2.5382790565490723, + "learning_rate": 3.7685572896840504e-06, + "loss": 1.5107, + "step": 954 + }, + { + "epoch": 0.9809964047252183, + "grad_norm": 2.6962392330169678, + "learning_rate": 3.766653977921584e-06, + "loss": 1.4534, + "step": 955 + }, + { + "epoch": 0.9820236260914227, + "grad_norm": 2.5411603450775146, + "learning_rate": 3.764750666159117e-06, + "loss": 1.3942, + "step": 956 + }, + { + "epoch": 0.9830508474576272, + "grad_norm": 2.7101845741271973, + "learning_rate": 3.7628473543966505e-06, + "loss": 1.5003, + "step": 957 + }, + { + "epoch": 0.9840780688238315, + "grad_norm": 2.5780858993530273, + "learning_rate": 3.7609440426341835e-06, + "loss": 1.4508, + "step": 958 + }, + { + "epoch": 0.9851052901900359, + "grad_norm": 2.814004421234131, + "learning_rate": 3.759040730871717e-06, + "loss": 1.5435, + "step": 959 + }, + { + "epoch": 0.9861325115562404, + "grad_norm": 2.5701088905334473, + "learning_rate": 3.75713741910925e-06, + "loss": 1.4386, + "step": 960 + }, + { + "epoch": 0.9861325115562404, + "eval_loss": 1.8556313514709473, + "eval_runtime": 37.1375, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 0.9871597329224447, + "grad_norm": 2.407271146774292, + "learning_rate": 3.7552341073467836e-06, + "loss": 1.4338, + "step": 961 + }, + { + "epoch": 0.9881869542886492, + "grad_norm": 2.433256149291992, + "learning_rate": 3.753330795584317e-06, + "loss": 1.47, + "step": 962 + }, + { + "epoch": 0.9892141756548536, + "grad_norm": 2.679339647293091, + "learning_rate": 3.75142748382185e-06, + "loss": 1.5459, + "step": 963 + }, + { + "epoch": 0.9902413970210581, + "grad_norm": 2.325840473175049, + "learning_rate": 3.749524172059384e-06, + "loss": 1.4385, + "step": 964 + }, + { + "epoch": 0.9912686183872624, + "grad_norm": 2.7060739994049072, + "learning_rate": 3.747620860296917e-06, + "loss": 1.4938, + "step": 965 + }, + { + "epoch": 0.9922958397534669, + "grad_norm": 2.589808464050293, + "learning_rate": 3.7457175485344504e-06, + "loss": 1.5258, + "step": 966 + }, + { + "epoch": 0.9933230611196713, + "grad_norm": 2.4876766204833984, + "learning_rate": 3.7438142367719837e-06, + "loss": 1.5361, + "step": 967 + }, + { + "epoch": 0.9943502824858758, + "grad_norm": 2.7370550632476807, + "learning_rate": 3.741910925009517e-06, + "loss": 1.4999, + "step": 968 + }, + { + "epoch": 0.9953775038520801, + "grad_norm": 2.6552398204803467, + "learning_rate": 3.7400076132470505e-06, + "loss": 1.4612, + "step": 969 + }, + { + "epoch": 0.9964047252182845, + "grad_norm": 2.5229146480560303, + "learning_rate": 3.7381043014845834e-06, + "loss": 1.4794, + "step": 970 + }, + { + "epoch": 0.9964047252182845, + "eval_loss": 1.8614000082015991, + "eval_runtime": 37.1311, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 970 + }, + { + "epoch": 0.997431946584489, + "grad_norm": 2.5604071617126465, + "learning_rate": 3.7362009897221168e-06, + "loss": 1.5078, + "step": 971 + }, + { + "epoch": 0.9984591679506933, + "grad_norm": 2.614192485809326, + "learning_rate": 3.73429767795965e-06, + "loss": 1.4849, + "step": 972 + }, + { + "epoch": 0.9994863893168978, + "grad_norm": 2.5245072841644287, + "learning_rate": 3.7323943661971835e-06, + "loss": 1.4993, + "step": 973 + }, + { + "epoch": 1.0005136106831023, + "grad_norm": 2.5631163120269775, + "learning_rate": 3.730491054434717e-06, + "loss": 1.4136, + "step": 974 + }, + { + "epoch": 1.0015408320493067, + "grad_norm": 2.625396728515625, + "learning_rate": 3.72858774267225e-06, + "loss": 1.3623, + "step": 975 + }, + { + "epoch": 1.002568053415511, + "grad_norm": 2.7067251205444336, + "learning_rate": 3.726684430909783e-06, + "loss": 1.4527, + "step": 976 + }, + { + "epoch": 1.0035952747817154, + "grad_norm": 2.5352942943573, + "learning_rate": 3.7247811191473165e-06, + "loss": 1.474, + "step": 977 + }, + { + "epoch": 1.00462249614792, + "grad_norm": 2.818143367767334, + "learning_rate": 3.72287780738485e-06, + "loss": 1.4852, + "step": 978 + }, + { + "epoch": 1.0056497175141244, + "grad_norm": 2.8269577026367188, + "learning_rate": 3.7209744956223832e-06, + "loss": 1.4667, + "step": 979 + }, + { + "epoch": 1.0066769388803287, + "grad_norm": 2.837454319000244, + "learning_rate": 3.719071183859916e-06, + "loss": 1.4728, + "step": 980 + }, + { + "epoch": 1.0066769388803287, + "eval_loss": 1.8701857328414917, + "eval_runtime": 37.2859, + "eval_samples_per_second": 26.82, + "eval_steps_per_second": 1.126, + "step": 980 + }, + { + "epoch": 1.007704160246533, + "grad_norm": 2.6601309776306152, + "learning_rate": 3.7171678720974495e-06, + "loss": 1.3917, + "step": 981 + }, + { + "epoch": 1.0087313816127375, + "grad_norm": 2.761413097381592, + "learning_rate": 3.715264560334983e-06, + "loss": 1.5049, + "step": 982 + }, + { + "epoch": 1.009758602978942, + "grad_norm": 2.681429147720337, + "learning_rate": 3.7133612485725163e-06, + "loss": 1.4579, + "step": 983 + }, + { + "epoch": 1.0107858243451464, + "grad_norm": 2.693094253540039, + "learning_rate": 3.71145793681005e-06, + "loss": 1.4766, + "step": 984 + }, + { + "epoch": 1.0118130457113508, + "grad_norm": 2.589165210723877, + "learning_rate": 3.7095546250475834e-06, + "loss": 1.4011, + "step": 985 + }, + { + "epoch": 1.0128402670775551, + "grad_norm": 2.738140821456909, + "learning_rate": 3.7076513132851168e-06, + "loss": 1.5109, + "step": 986 + }, + { + "epoch": 1.0138674884437597, + "grad_norm": 2.7863996028900146, + "learning_rate": 3.7057480015226497e-06, + "loss": 1.4344, + "step": 987 + }, + { + "epoch": 1.014894709809964, + "grad_norm": 2.6011335849761963, + "learning_rate": 3.703844689760183e-06, + "loss": 1.3981, + "step": 988 + }, + { + "epoch": 1.0159219311761685, + "grad_norm": 2.5715689659118652, + "learning_rate": 3.7019413779977164e-06, + "loss": 1.4876, + "step": 989 + }, + { + "epoch": 1.0169491525423728, + "grad_norm": 2.6804113388061523, + "learning_rate": 3.70003806623525e-06, + "loss": 1.4576, + "step": 990 + }, + { + "epoch": 1.0169491525423728, + "eval_loss": 1.8802876472473145, + "eval_runtime": 37.3955, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 990 + }, + { + "epoch": 1.0179763739085772, + "grad_norm": 2.737604856491089, + "learning_rate": 3.698134754472783e-06, + "loss": 1.4399, + "step": 991 + }, + { + "epoch": 1.0190035952747818, + "grad_norm": 2.658388137817383, + "learning_rate": 3.696231442710316e-06, + "loss": 1.5035, + "step": 992 + }, + { + "epoch": 1.0200308166409862, + "grad_norm": 2.7263925075531006, + "learning_rate": 3.6943281309478495e-06, + "loss": 1.4469, + "step": 993 + }, + { + "epoch": 1.0210580380071905, + "grad_norm": 2.7885773181915283, + "learning_rate": 3.692424819185383e-06, + "loss": 1.4641, + "step": 994 + }, + { + "epoch": 1.022085259373395, + "grad_norm": 2.8929779529571533, + "learning_rate": 3.690521507422916e-06, + "loss": 1.4727, + "step": 995 + }, + { + "epoch": 1.0231124807395995, + "grad_norm": 2.6605303287506104, + "learning_rate": 3.6886181956604496e-06, + "loss": 1.4153, + "step": 996 + }, + { + "epoch": 1.0241397021058039, + "grad_norm": 2.8678219318389893, + "learning_rate": 3.686714883897983e-06, + "loss": 1.4199, + "step": 997 + }, + { + "epoch": 1.0251669234720082, + "grad_norm": 2.884453058242798, + "learning_rate": 3.684811572135516e-06, + "loss": 1.4909, + "step": 998 + }, + { + "epoch": 1.0261941448382126, + "grad_norm": 2.6681578159332275, + "learning_rate": 3.6829082603730492e-06, + "loss": 1.4903, + "step": 999 + }, + { + "epoch": 1.027221366204417, + "grad_norm": 2.8013756275177, + "learning_rate": 3.6810049486105826e-06, + "loss": 1.449, + "step": 1000 + }, + { + "epoch": 1.027221366204417, + "eval_loss": 1.8735578060150146, + "eval_runtime": 37.4416, + "eval_samples_per_second": 26.708, + "eval_steps_per_second": 1.122, + "step": 1000 + }, + { + "epoch": 1.0282485875706215, + "grad_norm": 2.787750244140625, + "learning_rate": 3.679101636848116e-06, + "loss": 1.4808, + "step": 1001 + }, + { + "epoch": 1.029275808936826, + "grad_norm": 2.666874647140503, + "learning_rate": 3.6771983250856493e-06, + "loss": 1.4739, + "step": 1002 + }, + { + "epoch": 1.0303030303030303, + "grad_norm": 2.720586061477661, + "learning_rate": 3.6752950133231822e-06, + "loss": 1.4439, + "step": 1003 + }, + { + "epoch": 1.0313302516692346, + "grad_norm": 2.707771062850952, + "learning_rate": 3.6733917015607156e-06, + "loss": 1.4765, + "step": 1004 + }, + { + "epoch": 1.0323574730354392, + "grad_norm": 2.658092737197876, + "learning_rate": 3.6714883897982494e-06, + "loss": 1.4822, + "step": 1005 + }, + { + "epoch": 1.0333846944016436, + "grad_norm": 2.66788911819458, + "learning_rate": 3.6695850780357828e-06, + "loss": 1.4237, + "step": 1006 + }, + { + "epoch": 1.034411915767848, + "grad_norm": 2.638956308364868, + "learning_rate": 3.667681766273316e-06, + "loss": 1.4168, + "step": 1007 + }, + { + "epoch": 1.0354391371340523, + "grad_norm": 2.739384412765503, + "learning_rate": 3.6657784545108495e-06, + "loss": 1.4338, + "step": 1008 + }, + { + "epoch": 1.0364663585002567, + "grad_norm": 2.599773406982422, + "learning_rate": 3.6638751427483824e-06, + "loss": 1.5188, + "step": 1009 + }, + { + "epoch": 1.0374935798664613, + "grad_norm": 2.705876588821411, + "learning_rate": 3.6619718309859158e-06, + "loss": 1.468, + "step": 1010 + }, + { + "epoch": 1.0374935798664613, + "eval_loss": 1.8751709461212158, + "eval_runtime": 36.9681, + "eval_samples_per_second": 27.05, + "eval_steps_per_second": 1.136, + "step": 1010 + }, + { + "epoch": 1.0385208012326657, + "grad_norm": 2.712103843688965, + "learning_rate": 3.660068519223449e-06, + "loss": 1.508, + "step": 1011 + }, + { + "epoch": 1.03954802259887, + "grad_norm": 2.6621742248535156, + "learning_rate": 3.6581652074609825e-06, + "loss": 1.464, + "step": 1012 + }, + { + "epoch": 1.0405752439650744, + "grad_norm": 2.6141257286071777, + "learning_rate": 3.656261895698516e-06, + "loss": 1.4429, + "step": 1013 + }, + { + "epoch": 1.041602465331279, + "grad_norm": 2.5651750564575195, + "learning_rate": 3.6543585839360492e-06, + "loss": 1.4254, + "step": 1014 + }, + { + "epoch": 1.0426296866974833, + "grad_norm": 2.598686933517456, + "learning_rate": 3.652455272173582e-06, + "loss": 1.436, + "step": 1015 + }, + { + "epoch": 1.0436569080636877, + "grad_norm": 2.559375524520874, + "learning_rate": 3.6505519604111155e-06, + "loss": 1.4359, + "step": 1016 + }, + { + "epoch": 1.044684129429892, + "grad_norm": 2.57920503616333, + "learning_rate": 3.648648648648649e-06, + "loss": 1.4524, + "step": 1017 + }, + { + "epoch": 1.0457113507960965, + "grad_norm": 2.6629042625427246, + "learning_rate": 3.6467453368861823e-06, + "loss": 1.4901, + "step": 1018 + }, + { + "epoch": 1.046738572162301, + "grad_norm": 2.6846625804901123, + "learning_rate": 3.6448420251237156e-06, + "loss": 1.4345, + "step": 1019 + }, + { + "epoch": 1.0477657935285054, + "grad_norm": 2.7288694381713867, + "learning_rate": 3.6429387133612486e-06, + "loss": 1.4632, + "step": 1020 + }, + { + "epoch": 1.0477657935285054, + "eval_loss": 1.878115177154541, + "eval_runtime": 37.0627, + "eval_samples_per_second": 26.981, + "eval_steps_per_second": 1.133, + "step": 1020 + }, + { + "epoch": 1.0487930148947098, + "grad_norm": 2.6689510345458984, + "learning_rate": 3.641035401598782e-06, + "loss": 1.4934, + "step": 1021 + }, + { + "epoch": 1.0498202362609141, + "grad_norm": 2.6762588024139404, + "learning_rate": 3.6391320898363153e-06, + "loss": 1.4284, + "step": 1022 + }, + { + "epoch": 1.0508474576271187, + "grad_norm": 2.7875330448150635, + "learning_rate": 3.6372287780738487e-06, + "loss": 1.4717, + "step": 1023 + }, + { + "epoch": 1.051874678993323, + "grad_norm": 2.9239425659179688, + "learning_rate": 3.635325466311382e-06, + "loss": 1.4504, + "step": 1024 + }, + { + "epoch": 1.0529019003595275, + "grad_norm": 2.700913429260254, + "learning_rate": 3.633422154548916e-06, + "loss": 1.4701, + "step": 1025 + }, + { + "epoch": 1.0539291217257318, + "grad_norm": 2.693148612976074, + "learning_rate": 3.6315188427864487e-06, + "loss": 1.4889, + "step": 1026 + }, + { + "epoch": 1.0549563430919364, + "grad_norm": 2.6215498447418213, + "learning_rate": 3.629615531023982e-06, + "loss": 1.4586, + "step": 1027 + }, + { + "epoch": 1.0559835644581408, + "grad_norm": 2.704425573348999, + "learning_rate": 3.6277122192615155e-06, + "loss": 1.4182, + "step": 1028 + }, + { + "epoch": 1.0570107858243452, + "grad_norm": 2.691352128982544, + "learning_rate": 3.625808907499049e-06, + "loss": 1.468, + "step": 1029 + }, + { + "epoch": 1.0580380071905495, + "grad_norm": 2.769744634628296, + "learning_rate": 3.623905595736582e-06, + "loss": 1.4375, + "step": 1030 + }, + { + "epoch": 1.0580380071905495, + "eval_loss": 1.8686742782592773, + "eval_runtime": 37.1278, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1030 + }, + { + "epoch": 1.059065228556754, + "grad_norm": 2.6012840270996094, + "learning_rate": 3.6220022839741156e-06, + "loss": 1.4334, + "step": 1031 + }, + { + "epoch": 1.0600924499229585, + "grad_norm": 2.641416072845459, + "learning_rate": 3.6200989722116485e-06, + "loss": 1.4186, + "step": 1032 + }, + { + "epoch": 1.0611196712891628, + "grad_norm": 2.821176052093506, + "learning_rate": 3.618195660449182e-06, + "loss": 1.4438, + "step": 1033 + }, + { + "epoch": 1.0621468926553672, + "grad_norm": 2.719303607940674, + "learning_rate": 3.6162923486867152e-06, + "loss": 1.4519, + "step": 1034 + }, + { + "epoch": 1.0631741140215716, + "grad_norm": 2.76882004737854, + "learning_rate": 3.6143890369242486e-06, + "loss": 1.3941, + "step": 1035 + }, + { + "epoch": 1.064201335387776, + "grad_norm": 2.601137399673462, + "learning_rate": 3.612485725161782e-06, + "loss": 1.4167, + "step": 1036 + }, + { + "epoch": 1.0652285567539805, + "grad_norm": 2.691140651702881, + "learning_rate": 3.610582413399315e-06, + "loss": 1.4361, + "step": 1037 + }, + { + "epoch": 1.066255778120185, + "grad_norm": 2.5779640674591064, + "learning_rate": 3.6086791016368482e-06, + "loss": 1.4237, + "step": 1038 + }, + { + "epoch": 1.0672829994863893, + "grad_norm": 2.6535801887512207, + "learning_rate": 3.6067757898743816e-06, + "loss": 1.436, + "step": 1039 + }, + { + "epoch": 1.0683102208525936, + "grad_norm": 2.8365395069122314, + "learning_rate": 3.604872478111915e-06, + "loss": 1.5067, + "step": 1040 + }, + { + "epoch": 1.0683102208525936, + "eval_loss": 1.8686083555221558, + "eval_runtime": 37.1495, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 1040 + }, + { + "epoch": 1.0693374422187982, + "grad_norm": 2.704638957977295, + "learning_rate": 3.6029691663494483e-06, + "loss": 1.5105, + "step": 1041 + }, + { + "epoch": 1.0703646635850026, + "grad_norm": 2.6670243740081787, + "learning_rate": 3.6010658545869813e-06, + "loss": 1.451, + "step": 1042 + }, + { + "epoch": 1.071391884951207, + "grad_norm": 2.8808884620666504, + "learning_rate": 3.5991625428245146e-06, + "loss": 1.4795, + "step": 1043 + }, + { + "epoch": 1.0724191063174113, + "grad_norm": 2.800075054168701, + "learning_rate": 3.597259231062048e-06, + "loss": 1.4758, + "step": 1044 + }, + { + "epoch": 1.073446327683616, + "grad_norm": 2.4989686012268066, + "learning_rate": 3.5953559192995818e-06, + "loss": 1.42, + "step": 1045 + }, + { + "epoch": 1.0744735490498203, + "grad_norm": 2.7234370708465576, + "learning_rate": 3.593452607537115e-06, + "loss": 1.4549, + "step": 1046 + }, + { + "epoch": 1.0755007704160247, + "grad_norm": 2.7219443321228027, + "learning_rate": 3.5915492957746485e-06, + "loss": 1.5245, + "step": 1047 + }, + { + "epoch": 1.076527991782229, + "grad_norm": 2.6963443756103516, + "learning_rate": 3.589645984012182e-06, + "loss": 1.4472, + "step": 1048 + }, + { + "epoch": 1.0775552131484334, + "grad_norm": 2.7101387977600098, + "learning_rate": 3.587742672249715e-06, + "loss": 1.4659, + "step": 1049 + }, + { + "epoch": 1.078582434514638, + "grad_norm": 2.7673099040985107, + "learning_rate": 3.585839360487248e-06, + "loss": 1.4284, + "step": 1050 + }, + { + "epoch": 1.078582434514638, + "eval_loss": 1.879286289215088, + "eval_runtime": 37.1517, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.13, + "step": 1050 + }, + { + "epoch": 1.0796096558808423, + "grad_norm": 2.81614351272583, + "learning_rate": 3.5839360487247815e-06, + "loss": 1.4251, + "step": 1051 + }, + { + "epoch": 1.0806368772470467, + "grad_norm": 2.770350933074951, + "learning_rate": 3.582032736962315e-06, + "loss": 1.4372, + "step": 1052 + }, + { + "epoch": 1.081664098613251, + "grad_norm": 2.8099257946014404, + "learning_rate": 3.5801294251998483e-06, + "loss": 1.495, + "step": 1053 + }, + { + "epoch": 1.0826913199794557, + "grad_norm": 2.582378387451172, + "learning_rate": 3.578226113437381e-06, + "loss": 1.4005, + "step": 1054 + }, + { + "epoch": 1.08371854134566, + "grad_norm": 2.6215474605560303, + "learning_rate": 3.5763228016749146e-06, + "loss": 1.4428, + "step": 1055 + }, + { + "epoch": 1.0847457627118644, + "grad_norm": 2.7204151153564453, + "learning_rate": 3.574419489912448e-06, + "loss": 1.5197, + "step": 1056 + }, + { + "epoch": 1.0857729840780688, + "grad_norm": 2.6563117504119873, + "learning_rate": 3.5725161781499813e-06, + "loss": 1.457, + "step": 1057 + }, + { + "epoch": 1.0868002054442731, + "grad_norm": 2.559816837310791, + "learning_rate": 3.5706128663875147e-06, + "loss": 1.4376, + "step": 1058 + }, + { + "epoch": 1.0878274268104777, + "grad_norm": 2.6742372512817383, + "learning_rate": 3.5687095546250476e-06, + "loss": 1.4626, + "step": 1059 + }, + { + "epoch": 1.088854648176682, + "grad_norm": 2.8052852153778076, + "learning_rate": 3.566806242862581e-06, + "loss": 1.4785, + "step": 1060 + }, + { + "epoch": 1.088854648176682, + "eval_loss": 1.8775676488876343, + "eval_runtime": 37.1523, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1060 + }, + { + "epoch": 1.0898818695428865, + "grad_norm": 2.687411308288574, + "learning_rate": 3.5649029311001143e-06, + "loss": 1.4491, + "step": 1061 + }, + { + "epoch": 1.0909090909090908, + "grad_norm": 2.662604808807373, + "learning_rate": 3.5629996193376477e-06, + "loss": 1.4274, + "step": 1062 + }, + { + "epoch": 1.0919363122752954, + "grad_norm": 2.619046211242676, + "learning_rate": 3.561096307575181e-06, + "loss": 1.4449, + "step": 1063 + }, + { + "epoch": 1.0929635336414998, + "grad_norm": 2.672553539276123, + "learning_rate": 3.559192995812714e-06, + "loss": 1.4719, + "step": 1064 + }, + { + "epoch": 1.0939907550077042, + "grad_norm": 2.868760824203491, + "learning_rate": 3.557289684050248e-06, + "loss": 1.4955, + "step": 1065 + }, + { + "epoch": 1.0950179763739085, + "grad_norm": 2.7635276317596436, + "learning_rate": 3.555386372287781e-06, + "loss": 1.4765, + "step": 1066 + }, + { + "epoch": 1.0960451977401129, + "grad_norm": 2.883657455444336, + "learning_rate": 3.5534830605253145e-06, + "loss": 1.4923, + "step": 1067 + }, + { + "epoch": 1.0970724191063175, + "grad_norm": 2.7595503330230713, + "learning_rate": 3.551579748762848e-06, + "loss": 1.4527, + "step": 1068 + }, + { + "epoch": 1.0980996404725218, + "grad_norm": 2.8310179710388184, + "learning_rate": 3.5496764370003812e-06, + "loss": 1.4485, + "step": 1069 + }, + { + "epoch": 1.0991268618387262, + "grad_norm": 2.699432134628296, + "learning_rate": 3.5477731252379146e-06, + "loss": 1.4322, + "step": 1070 + }, + { + "epoch": 1.0991268618387262, + "eval_loss": 1.8706557750701904, + "eval_runtime": 37.1509, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1070 + }, + { + "epoch": 1.1001540832049306, + "grad_norm": 2.6249091625213623, + "learning_rate": 3.5458698134754475e-06, + "loss": 1.4031, + "step": 1071 + }, + { + "epoch": 1.1011813045711352, + "grad_norm": 2.6882944107055664, + "learning_rate": 3.543966501712981e-06, + "loss": 1.4075, + "step": 1072 + }, + { + "epoch": 1.1022085259373395, + "grad_norm": 2.6578996181488037, + "learning_rate": 3.5420631899505142e-06, + "loss": 1.4256, + "step": 1073 + }, + { + "epoch": 1.103235747303544, + "grad_norm": 2.8124637603759766, + "learning_rate": 3.5401598781880476e-06, + "loss": 1.5263, + "step": 1074 + }, + { + "epoch": 1.1042629686697483, + "grad_norm": 2.6022191047668457, + "learning_rate": 3.538256566425581e-06, + "loss": 1.3482, + "step": 1075 + }, + { + "epoch": 1.1052901900359529, + "grad_norm": 2.753009080886841, + "learning_rate": 3.536353254663114e-06, + "loss": 1.483, + "step": 1076 + }, + { + "epoch": 1.1063174114021572, + "grad_norm": 2.575666904449463, + "learning_rate": 3.5344499429006473e-06, + "loss": 1.4325, + "step": 1077 + }, + { + "epoch": 1.1073446327683616, + "grad_norm": 2.594136953353882, + "learning_rate": 3.5325466311381806e-06, + "loss": 1.4187, + "step": 1078 + }, + { + "epoch": 1.108371854134566, + "grad_norm": 2.807074785232544, + "learning_rate": 3.530643319375714e-06, + "loss": 1.4763, + "step": 1079 + }, + { + "epoch": 1.1093990755007703, + "grad_norm": 2.5865330696105957, + "learning_rate": 3.5287400076132474e-06, + "loss": 1.3896, + "step": 1080 + }, + { + "epoch": 1.1093990755007703, + "eval_loss": 1.8865792751312256, + "eval_runtime": 37.1514, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 1080 + }, + { + "epoch": 1.110426296866975, + "grad_norm": 2.715419054031372, + "learning_rate": 3.5268366958507803e-06, + "loss": 1.4259, + "step": 1081 + }, + { + "epoch": 1.1114535182331793, + "grad_norm": 2.6837334632873535, + "learning_rate": 3.5249333840883137e-06, + "loss": 1.4581, + "step": 1082 + }, + { + "epoch": 1.1124807395993837, + "grad_norm": 2.722128391265869, + "learning_rate": 3.523030072325847e-06, + "loss": 1.4486, + "step": 1083 + }, + { + "epoch": 1.113507960965588, + "grad_norm": 2.7390329837799072, + "learning_rate": 3.5211267605633804e-06, + "loss": 1.4225, + "step": 1084 + }, + { + "epoch": 1.1145351823317924, + "grad_norm": 2.740271806716919, + "learning_rate": 3.519223448800914e-06, + "loss": 1.4274, + "step": 1085 + }, + { + "epoch": 1.115562403697997, + "grad_norm": 2.8321313858032227, + "learning_rate": 3.5173201370384475e-06, + "loss": 1.4337, + "step": 1086 + }, + { + "epoch": 1.1165896250642013, + "grad_norm": 2.740826368331909, + "learning_rate": 3.515416825275981e-06, + "loss": 1.4331, + "step": 1087 + }, + { + "epoch": 1.1176168464304057, + "grad_norm": 2.7382850646972656, + "learning_rate": 3.513513513513514e-06, + "loss": 1.4495, + "step": 1088 + }, + { + "epoch": 1.11864406779661, + "grad_norm": 2.7677972316741943, + "learning_rate": 3.511610201751047e-06, + "loss": 1.4628, + "step": 1089 + }, + { + "epoch": 1.1196712891628147, + "grad_norm": 2.8850181102752686, + "learning_rate": 3.5097068899885806e-06, + "loss": 1.3739, + "step": 1090 + }, + { + "epoch": 1.1196712891628147, + "eval_loss": 1.8794182538986206, + "eval_runtime": 37.4388, + "eval_samples_per_second": 26.71, + "eval_steps_per_second": 1.122, + "step": 1090 + }, + { + "epoch": 1.120698510529019, + "grad_norm": 2.73730731010437, + "learning_rate": 3.507803578226114e-06, + "loss": 1.4444, + "step": 1091 + }, + { + "epoch": 1.1217257318952234, + "grad_norm": 3.027599811553955, + "learning_rate": 3.5059002664636473e-06, + "loss": 1.5123, + "step": 1092 + }, + { + "epoch": 1.1227529532614278, + "grad_norm": 2.613658905029297, + "learning_rate": 3.5039969547011802e-06, + "loss": 1.3798, + "step": 1093 + }, + { + "epoch": 1.1237801746276324, + "grad_norm": 2.6396987438201904, + "learning_rate": 3.5020936429387136e-06, + "loss": 1.4357, + "step": 1094 + }, + { + "epoch": 1.1248073959938367, + "grad_norm": 2.625540018081665, + "learning_rate": 3.500190331176247e-06, + "loss": 1.4354, + "step": 1095 + }, + { + "epoch": 1.125834617360041, + "grad_norm": 2.8413279056549072, + "learning_rate": 3.4982870194137803e-06, + "loss": 1.4846, + "step": 1096 + }, + { + "epoch": 1.1268618387262455, + "grad_norm": 2.7898473739624023, + "learning_rate": 3.4963837076513137e-06, + "loss": 1.421, + "step": 1097 + }, + { + "epoch": 1.1278890600924498, + "grad_norm": 2.6385879516601562, + "learning_rate": 3.4944803958888466e-06, + "loss": 1.4241, + "step": 1098 + }, + { + "epoch": 1.1289162814586544, + "grad_norm": 2.677098512649536, + "learning_rate": 3.49257708412638e-06, + "loss": 1.3671, + "step": 1099 + }, + { + "epoch": 1.1299435028248588, + "grad_norm": 2.8582913875579834, + "learning_rate": 3.4906737723639133e-06, + "loss": 1.4886, + "step": 1100 + }, + { + "epoch": 1.1299435028248588, + "eval_loss": 1.8696506023406982, + "eval_runtime": 37.462, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 1100 + }, + { + "epoch": 1.1309707241910631, + "grad_norm": 2.7157933712005615, + "learning_rate": 3.4887704606014467e-06, + "loss": 1.4964, + "step": 1101 + }, + { + "epoch": 1.1319979455572675, + "grad_norm": 2.6703319549560547, + "learning_rate": 3.48686714883898e-06, + "loss": 1.4509, + "step": 1102 + }, + { + "epoch": 1.133025166923472, + "grad_norm": 2.6307899951934814, + "learning_rate": 3.484963837076513e-06, + "loss": 1.485, + "step": 1103 + }, + { + "epoch": 1.1340523882896765, + "grad_norm": 2.8333141803741455, + "learning_rate": 3.4830605253140464e-06, + "loss": 1.432, + "step": 1104 + }, + { + "epoch": 1.1350796096558808, + "grad_norm": 2.545804023742676, + "learning_rate": 3.4811572135515797e-06, + "loss": 1.3799, + "step": 1105 + }, + { + "epoch": 1.1361068310220852, + "grad_norm": 2.7070441246032715, + "learning_rate": 3.4792539017891135e-06, + "loss": 1.4193, + "step": 1106 + }, + { + "epoch": 1.1371340523882898, + "grad_norm": 2.697140693664551, + "learning_rate": 3.477350590026647e-06, + "loss": 1.398, + "step": 1107 + }, + { + "epoch": 1.1381612737544942, + "grad_norm": 2.784795045852661, + "learning_rate": 3.4754472782641802e-06, + "loss": 1.4708, + "step": 1108 + }, + { + "epoch": 1.1391884951206985, + "grad_norm": 2.8518831729888916, + "learning_rate": 3.4735439665017136e-06, + "loss": 1.4707, + "step": 1109 + }, + { + "epoch": 1.140215716486903, + "grad_norm": 3.0481607913970947, + "learning_rate": 3.4716406547392465e-06, + "loss": 1.4385, + "step": 1110 + }, + { + "epoch": 1.140215716486903, + "eval_loss": 1.8367345333099365, + "eval_runtime": 36.9539, + "eval_samples_per_second": 27.061, + "eval_steps_per_second": 1.137, + "step": 1110 + }, + { + "epoch": 1.1412429378531073, + "grad_norm": 2.803053140640259, + "learning_rate": 3.46973734297678e-06, + "loss": 1.4361, + "step": 1111 + }, + { + "epoch": 1.1422701592193119, + "grad_norm": 2.762638568878174, + "learning_rate": 3.4678340312143133e-06, + "loss": 1.4522, + "step": 1112 + }, + { + "epoch": 1.1432973805855162, + "grad_norm": 2.9458558559417725, + "learning_rate": 3.4659307194518466e-06, + "loss": 1.4914, + "step": 1113 + }, + { + "epoch": 1.1443246019517206, + "grad_norm": 2.774721622467041, + "learning_rate": 3.46402740768938e-06, + "loss": 1.4723, + "step": 1114 + }, + { + "epoch": 1.145351823317925, + "grad_norm": 2.6556880474090576, + "learning_rate": 3.462124095926913e-06, + "loss": 1.4073, + "step": 1115 + }, + { + "epoch": 1.1463790446841293, + "grad_norm": 2.7482454776763916, + "learning_rate": 3.4602207841644463e-06, + "loss": 1.4022, + "step": 1116 + }, + { + "epoch": 1.147406266050334, + "grad_norm": 2.627567768096924, + "learning_rate": 3.4583174724019797e-06, + "loss": 1.4428, + "step": 1117 + }, + { + "epoch": 1.1484334874165383, + "grad_norm": 2.750666856765747, + "learning_rate": 3.456414160639513e-06, + "loss": 1.4972, + "step": 1118 + }, + { + "epoch": 1.1494607087827426, + "grad_norm": 2.842184543609619, + "learning_rate": 3.4545108488770464e-06, + "loss": 1.4753, + "step": 1119 + }, + { + "epoch": 1.150487930148947, + "grad_norm": 2.829437732696533, + "learning_rate": 3.4526075371145793e-06, + "loss": 1.4525, + "step": 1120 + }, + { + "epoch": 1.150487930148947, + "eval_loss": 1.8498426675796509, + "eval_runtime": 37.0234, + "eval_samples_per_second": 27.01, + "eval_steps_per_second": 1.134, + "step": 1120 + }, + { + "epoch": 1.1515151515151516, + "grad_norm": 2.7674920558929443, + "learning_rate": 3.4507042253521127e-06, + "loss": 1.4516, + "step": 1121 + }, + { + "epoch": 1.152542372881356, + "grad_norm": 2.726806879043579, + "learning_rate": 3.448800913589646e-06, + "loss": 1.5113, + "step": 1122 + }, + { + "epoch": 1.1535695942475603, + "grad_norm": 2.7698283195495605, + "learning_rate": 3.4468976018271794e-06, + "loss": 1.4295, + "step": 1123 + }, + { + "epoch": 1.1545968156137647, + "grad_norm": 2.6224706172943115, + "learning_rate": 3.4449942900647128e-06, + "loss": 1.5179, + "step": 1124 + }, + { + "epoch": 1.1556240369799693, + "grad_norm": 2.7266266345977783, + "learning_rate": 3.443090978302246e-06, + "loss": 1.4784, + "step": 1125 + }, + { + "epoch": 1.1566512583461737, + "grad_norm": 2.631751298904419, + "learning_rate": 3.44118766653978e-06, + "loss": 1.4659, + "step": 1126 + }, + { + "epoch": 1.157678479712378, + "grad_norm": 2.7077064514160156, + "learning_rate": 3.439284354777313e-06, + "loss": 1.4654, + "step": 1127 + }, + { + "epoch": 1.1587057010785824, + "grad_norm": 2.735102415084839, + "learning_rate": 3.4373810430148462e-06, + "loss": 1.4604, + "step": 1128 + }, + { + "epoch": 1.1597329224447868, + "grad_norm": 2.9314727783203125, + "learning_rate": 3.4354777312523796e-06, + "loss": 1.3964, + "step": 1129 + }, + { + "epoch": 1.1607601438109914, + "grad_norm": 2.658930778503418, + "learning_rate": 3.433574419489913e-06, + "loss": 1.4331, + "step": 1130 + }, + { + "epoch": 1.1607601438109914, + "eval_loss": 1.8545702695846558, + "eval_runtime": 37.0493, + "eval_samples_per_second": 26.991, + "eval_steps_per_second": 1.134, + "step": 1130 + }, + { + "epoch": 1.1617873651771957, + "grad_norm": 2.8146262168884277, + "learning_rate": 3.4316711077274463e-06, + "loss": 1.4711, + "step": 1131 + }, + { + "epoch": 1.1628145865434, + "grad_norm": 2.6918272972106934, + "learning_rate": 3.4297677959649792e-06, + "loss": 1.4327, + "step": 1132 + }, + { + "epoch": 1.1638418079096045, + "grad_norm": 2.680551528930664, + "learning_rate": 3.4278644842025126e-06, + "loss": 1.4886, + "step": 1133 + }, + { + "epoch": 1.1648690292758088, + "grad_norm": 2.617286443710327, + "learning_rate": 3.425961172440046e-06, + "loss": 1.4167, + "step": 1134 + }, + { + "epoch": 1.1658962506420134, + "grad_norm": 2.6772518157958984, + "learning_rate": 3.4240578606775793e-06, + "loss": 1.5274, + "step": 1135 + }, + { + "epoch": 1.1669234720082178, + "grad_norm": 2.640723705291748, + "learning_rate": 3.4221545489151127e-06, + "loss": 1.4398, + "step": 1136 + }, + { + "epoch": 1.1679506933744221, + "grad_norm": 2.7104315757751465, + "learning_rate": 3.4202512371526456e-06, + "loss": 1.4367, + "step": 1137 + }, + { + "epoch": 1.1689779147406265, + "grad_norm": 2.831645965576172, + "learning_rate": 3.418347925390179e-06, + "loss": 1.5329, + "step": 1138 + }, + { + "epoch": 1.170005136106831, + "grad_norm": 2.7174274921417236, + "learning_rate": 3.4164446136277124e-06, + "loss": 1.481, + "step": 1139 + }, + { + "epoch": 1.1710323574730355, + "grad_norm": 2.6496286392211914, + "learning_rate": 3.4145413018652457e-06, + "loss": 1.4489, + "step": 1140 + }, + { + "epoch": 1.1710323574730355, + "eval_loss": 1.8524384498596191, + "eval_runtime": 37.1347, + "eval_samples_per_second": 26.929, + "eval_steps_per_second": 1.131, + "step": 1140 + }, + { + "epoch": 1.1720595788392398, + "grad_norm": 2.610058069229126, + "learning_rate": 3.412637990102779e-06, + "loss": 1.4755, + "step": 1141 + }, + { + "epoch": 1.1730868002054442, + "grad_norm": 2.642070770263672, + "learning_rate": 3.4107346783403125e-06, + "loss": 1.4981, + "step": 1142 + }, + { + "epoch": 1.1741140215716488, + "grad_norm": 2.57493257522583, + "learning_rate": 3.4088313665778454e-06, + "loss": 1.4039, + "step": 1143 + }, + { + "epoch": 1.1751412429378532, + "grad_norm": 2.5919628143310547, + "learning_rate": 3.4069280548153788e-06, + "loss": 1.4391, + "step": 1144 + }, + { + "epoch": 1.1761684643040575, + "grad_norm": 2.698862314224243, + "learning_rate": 3.405024743052912e-06, + "loss": 1.4198, + "step": 1145 + }, + { + "epoch": 1.177195685670262, + "grad_norm": 2.6387715339660645, + "learning_rate": 3.403121431290446e-06, + "loss": 1.4373, + "step": 1146 + }, + { + "epoch": 1.1782229070364663, + "grad_norm": 2.764643430709839, + "learning_rate": 3.4012181195279793e-06, + "loss": 1.4348, + "step": 1147 + }, + { + "epoch": 1.1792501284026708, + "grad_norm": 2.721038579940796, + "learning_rate": 3.3993148077655126e-06, + "loss": 1.4232, + "step": 1148 + }, + { + "epoch": 1.1802773497688752, + "grad_norm": 2.8942360877990723, + "learning_rate": 3.3974114960030456e-06, + "loss": 1.4405, + "step": 1149 + }, + { + "epoch": 1.1813045711350796, + "grad_norm": 2.687194585800171, + "learning_rate": 3.395508184240579e-06, + "loss": 1.3854, + "step": 1150 + }, + { + "epoch": 1.1813045711350796, + "eval_loss": 1.8701989650726318, + "eval_runtime": 37.132, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1150 + }, + { + "epoch": 1.182331792501284, + "grad_norm": 2.7137982845306396, + "learning_rate": 3.3936048724781123e-06, + "loss": 1.4174, + "step": 1151 + }, + { + "epoch": 1.1833590138674885, + "grad_norm": 2.8916432857513428, + "learning_rate": 3.3917015607156457e-06, + "loss": 1.5043, + "step": 1152 + }, + { + "epoch": 1.184386235233693, + "grad_norm": 2.6625723838806152, + "learning_rate": 3.389798248953179e-06, + "loss": 1.3832, + "step": 1153 + }, + { + "epoch": 1.1854134565998973, + "grad_norm": 2.8155555725097656, + "learning_rate": 3.387894937190712e-06, + "loss": 1.4098, + "step": 1154 + }, + { + "epoch": 1.1864406779661016, + "grad_norm": 2.634751796722412, + "learning_rate": 3.3859916254282453e-06, + "loss": 1.4522, + "step": 1155 + }, + { + "epoch": 1.1874678993323062, + "grad_norm": 2.7085063457489014, + "learning_rate": 3.3840883136657787e-06, + "loss": 1.4539, + "step": 1156 + }, + { + "epoch": 1.1884951206985106, + "grad_norm": 2.65588116645813, + "learning_rate": 3.382185001903312e-06, + "loss": 1.5084, + "step": 1157 + }, + { + "epoch": 1.189522342064715, + "grad_norm": 2.582489252090454, + "learning_rate": 3.3802816901408454e-06, + "loss": 1.4218, + "step": 1158 + }, + { + "epoch": 1.1905495634309193, + "grad_norm": 2.6364378929138184, + "learning_rate": 3.3783783783783788e-06, + "loss": 1.4522, + "step": 1159 + }, + { + "epoch": 1.1915767847971237, + "grad_norm": 2.6329915523529053, + "learning_rate": 3.3764750666159117e-06, + "loss": 1.442, + "step": 1160 + }, + { + "epoch": 1.1915767847971237, + "eval_loss": 1.8593298196792603, + "eval_runtime": 37.1143, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1160 + }, + { + "epoch": 1.1926040061633283, + "grad_norm": 2.610050916671753, + "learning_rate": 3.374571754853445e-06, + "loss": 1.4397, + "step": 1161 + }, + { + "epoch": 1.1936312275295327, + "grad_norm": 2.5503945350646973, + "learning_rate": 3.3726684430909784e-06, + "loss": 1.4211, + "step": 1162 + }, + { + "epoch": 1.194658448895737, + "grad_norm": 2.6648080348968506, + "learning_rate": 3.370765131328512e-06, + "loss": 1.3938, + "step": 1163 + }, + { + "epoch": 1.1956856702619414, + "grad_norm": 2.6705355644226074, + "learning_rate": 3.368861819566045e-06, + "loss": 1.4107, + "step": 1164 + }, + { + "epoch": 1.1967128916281458, + "grad_norm": 2.7582640647888184, + "learning_rate": 3.366958507803578e-06, + "loss": 1.4327, + "step": 1165 + }, + { + "epoch": 1.1977401129943503, + "grad_norm": 2.7768008708953857, + "learning_rate": 3.365055196041112e-06, + "loss": 1.454, + "step": 1166 + }, + { + "epoch": 1.1987673343605547, + "grad_norm": 2.628997325897217, + "learning_rate": 3.3631518842786452e-06, + "loss": 1.3988, + "step": 1167 + }, + { + "epoch": 1.199794555726759, + "grad_norm": 2.637505531311035, + "learning_rate": 3.3612485725161786e-06, + "loss": 1.4766, + "step": 1168 + }, + { + "epoch": 1.2008217770929634, + "grad_norm": 2.6935489177703857, + "learning_rate": 3.359345260753712e-06, + "loss": 1.4063, + "step": 1169 + }, + { + "epoch": 1.201848998459168, + "grad_norm": 2.7827870845794678, + "learning_rate": 3.3574419489912453e-06, + "loss": 1.4882, + "step": 1170 + }, + { + "epoch": 1.201848998459168, + "eval_loss": 1.8575458526611328, + "eval_runtime": 37.1207, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1170 + }, + { + "epoch": 1.2028762198253724, + "grad_norm": 2.789053201675415, + "learning_rate": 3.3555386372287783e-06, + "loss": 1.4699, + "step": 1171 + }, + { + "epoch": 1.2039034411915768, + "grad_norm": 3.053184747695923, + "learning_rate": 3.3536353254663116e-06, + "loss": 1.4993, + "step": 1172 + }, + { + "epoch": 1.2049306625577811, + "grad_norm": 2.6963112354278564, + "learning_rate": 3.351732013703845e-06, + "loss": 1.3985, + "step": 1173 + }, + { + "epoch": 1.2059578839239857, + "grad_norm": 2.689101219177246, + "learning_rate": 3.3498287019413784e-06, + "loss": 1.4464, + "step": 1174 + }, + { + "epoch": 1.20698510529019, + "grad_norm": 2.7748899459838867, + "learning_rate": 3.3479253901789117e-06, + "loss": 1.4554, + "step": 1175 + }, + { + "epoch": 1.2080123266563945, + "grad_norm": 2.8673830032348633, + "learning_rate": 3.346022078416445e-06, + "loss": 1.4796, + "step": 1176 + }, + { + "epoch": 1.2090395480225988, + "grad_norm": 2.7719533443450928, + "learning_rate": 3.344118766653978e-06, + "loss": 1.3716, + "step": 1177 + }, + { + "epoch": 1.2100667693888032, + "grad_norm": 2.71771240234375, + "learning_rate": 3.3422154548915114e-06, + "loss": 1.4554, + "step": 1178 + }, + { + "epoch": 1.2110939907550078, + "grad_norm": 2.7095048427581787, + "learning_rate": 3.3403121431290448e-06, + "loss": 1.4233, + "step": 1179 + }, + { + "epoch": 1.2121212121212122, + "grad_norm": 2.631638526916504, + "learning_rate": 3.338408831366578e-06, + "loss": 1.4581, + "step": 1180 + }, + { + "epoch": 1.2121212121212122, + "eval_loss": 1.855039119720459, + "eval_runtime": 37.1083, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1180 + }, + { + "epoch": 1.2131484334874165, + "grad_norm": 2.657257318496704, + "learning_rate": 3.3365055196041115e-06, + "loss": 1.3864, + "step": 1181 + }, + { + "epoch": 1.214175654853621, + "grad_norm": 2.894169569015503, + "learning_rate": 3.3346022078416444e-06, + "loss": 1.4673, + "step": 1182 + }, + { + "epoch": 1.2152028762198253, + "grad_norm": 2.852644920349121, + "learning_rate": 3.3326988960791778e-06, + "loss": 1.5026, + "step": 1183 + }, + { + "epoch": 1.2162300975860298, + "grad_norm": 2.720878839492798, + "learning_rate": 3.330795584316711e-06, + "loss": 1.4281, + "step": 1184 + }, + { + "epoch": 1.2172573189522342, + "grad_norm": 2.6779472827911377, + "learning_rate": 3.3288922725542445e-06, + "loss": 1.3559, + "step": 1185 + }, + { + "epoch": 1.2182845403184386, + "grad_norm": 2.836782217025757, + "learning_rate": 3.326988960791778e-06, + "loss": 1.4185, + "step": 1186 + }, + { + "epoch": 1.219311761684643, + "grad_norm": 2.9181265830993652, + "learning_rate": 3.3250856490293117e-06, + "loss": 1.4379, + "step": 1187 + }, + { + "epoch": 1.2203389830508475, + "grad_norm": 2.7315778732299805, + "learning_rate": 3.323182337266845e-06, + "loss": 1.4824, + "step": 1188 + }, + { + "epoch": 1.221366204417052, + "grad_norm": 2.8483448028564453, + "learning_rate": 3.321279025504378e-06, + "loss": 1.4766, + "step": 1189 + }, + { + "epoch": 1.2223934257832563, + "grad_norm": 2.736323118209839, + "learning_rate": 3.3193757137419113e-06, + "loss": 1.4647, + "step": 1190 + }, + { + "epoch": 1.2223934257832563, + "eval_loss": 1.8581864833831787, + "eval_runtime": 37.3401, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 1190 + }, + { + "epoch": 1.2234206471494606, + "grad_norm": 2.8863027095794678, + "learning_rate": 3.3174724019794447e-06, + "loss": 1.4533, + "step": 1191 + }, + { + "epoch": 1.2244478685156652, + "grad_norm": 2.951228380203247, + "learning_rate": 3.315569090216978e-06, + "loss": 1.495, + "step": 1192 + }, + { + "epoch": 1.2254750898818696, + "grad_norm": 2.7486696243286133, + "learning_rate": 3.3136657784545114e-06, + "loss": 1.4082, + "step": 1193 + }, + { + "epoch": 1.226502311248074, + "grad_norm": 2.9105849266052246, + "learning_rate": 3.3117624666920443e-06, + "loss": 1.4988, + "step": 1194 + }, + { + "epoch": 1.2275295326142783, + "grad_norm": 2.8165998458862305, + "learning_rate": 3.3098591549295777e-06, + "loss": 1.4162, + "step": 1195 + }, + { + "epoch": 1.2285567539804827, + "grad_norm": 2.8999876976013184, + "learning_rate": 3.307955843167111e-06, + "loss": 1.3909, + "step": 1196 + }, + { + "epoch": 1.2295839753466873, + "grad_norm": 2.8325014114379883, + "learning_rate": 3.3060525314046444e-06, + "loss": 1.4625, + "step": 1197 + }, + { + "epoch": 1.2306111967128917, + "grad_norm": 2.6553618907928467, + "learning_rate": 3.304149219642178e-06, + "loss": 1.4453, + "step": 1198 + }, + { + "epoch": 1.231638418079096, + "grad_norm": 2.584757089614868, + "learning_rate": 3.3022459078797107e-06, + "loss": 1.4412, + "step": 1199 + }, + { + "epoch": 1.2326656394453004, + "grad_norm": 2.68215274810791, + "learning_rate": 3.300342596117244e-06, + "loss": 1.4511, + "step": 1200 + }, + { + "epoch": 1.2326656394453004, + "eval_loss": 1.849502682685852, + "eval_runtime": 37.394, + "eval_samples_per_second": 26.742, + "eval_steps_per_second": 1.123, + "step": 1200 + }, + { + "epoch": 1.2336928608115048, + "grad_norm": 2.735494375228882, + "learning_rate": 3.2984392843547775e-06, + "loss": 1.4433, + "step": 1201 + }, + { + "epoch": 1.2347200821777093, + "grad_norm": 2.6691815853118896, + "learning_rate": 3.296535972592311e-06, + "loss": 1.4033, + "step": 1202 + }, + { + "epoch": 1.2357473035439137, + "grad_norm": 2.6378355026245117, + "learning_rate": 3.294632660829844e-06, + "loss": 1.378, + "step": 1203 + }, + { + "epoch": 1.236774524910118, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4227, + "step": 1204 + }, + { + "epoch": 1.2378017462763227, + "grad_norm": 2.8933019638061523, + "learning_rate": 3.292729349067377e-06, + "loss": 1.4204, + "step": 1205 + }, + { + "epoch": 1.238828967642527, + "grad_norm": 2.786475896835327, + "learning_rate": 3.2908260373049105e-06, + "loss": 1.4165, + "step": 1206 + }, + { + "epoch": 1.2398561890087314, + "grad_norm": 2.936825752258301, + "learning_rate": 3.288922725542444e-06, + "loss": 1.4712, + "step": 1207 + }, + { + "epoch": 1.2408834103749358, + "grad_norm": 2.7255940437316895, + "learning_rate": 3.2870194137799776e-06, + "loss": 1.4177, + "step": 1208 + }, + { + "epoch": 1.2419106317411401, + "grad_norm": 2.6526482105255127, + "learning_rate": 3.285116102017511e-06, + "loss": 1.3975, + "step": 1209 + }, + { + "epoch": 1.2429378531073447, + "grad_norm": 2.7539143562316895, + "learning_rate": 3.2832127902550444e-06, + "loss": 1.4717, + "step": 1210 + }, + { + "epoch": 1.2429378531073447, + "eval_loss": 1.8445062637329102, + "eval_runtime": 36.9889, + "eval_samples_per_second": 27.035, + "eval_steps_per_second": 1.135, + "step": 1210 + }, + { + "epoch": 1.243965074473549, + "grad_norm": 2.907252073287964, + "learning_rate": 3.2813094784925777e-06, + "loss": 1.4484, + "step": 1211 + }, + { + "epoch": 1.2449922958397535, + "grad_norm": 2.7312281131744385, + "learning_rate": 3.2794061667301107e-06, + "loss": 1.3863, + "step": 1212 + }, + { + "epoch": 1.2460195172059578, + "grad_norm": 2.739184617996216, + "learning_rate": 3.277502854967644e-06, + "loss": 1.4302, + "step": 1213 + }, + { + "epoch": 1.2470467385721622, + "grad_norm": 2.798630475997925, + "learning_rate": 3.2755995432051774e-06, + "loss": 1.4134, + "step": 1214 + }, + { + "epoch": 1.2480739599383668, + "grad_norm": 2.835686683654785, + "learning_rate": 3.2736962314427107e-06, + "loss": 1.4258, + "step": 1215 + }, + { + "epoch": 1.2491011813045712, + "grad_norm": 2.7404251098632812, + "learning_rate": 3.271792919680244e-06, + "loss": 1.4347, + "step": 1216 + }, + { + "epoch": 1.2501284026707755, + "grad_norm": 2.8362114429473877, + "learning_rate": 3.269889607917777e-06, + "loss": 1.4909, + "step": 1217 + }, + { + "epoch": 1.25115562403698, + "grad_norm": 2.842283248901367, + "learning_rate": 3.2679862961553104e-06, + "loss": 1.4065, + "step": 1218 + }, + { + "epoch": 1.2521828454031843, + "grad_norm": 2.717536449432373, + "learning_rate": 3.2660829843928438e-06, + "loss": 1.4446, + "step": 1219 + }, + { + "epoch": 1.2532100667693888, + "grad_norm": 2.687009572982788, + "learning_rate": 3.264179672630377e-06, + "loss": 1.4826, + "step": 1220 + }, + { + "epoch": 1.2532100667693888, + "eval_loss": 1.8576964139938354, + "eval_runtime": 37.0759, + "eval_samples_per_second": 26.972, + "eval_steps_per_second": 1.133, + "step": 1220 + }, + { + "epoch": 1.2542372881355932, + "grad_norm": 2.7234137058258057, + "learning_rate": 3.2622763608679105e-06, + "loss": 1.4503, + "step": 1221 + }, + { + "epoch": 1.2552645095017976, + "grad_norm": 2.733876943588257, + "learning_rate": 3.2603730491054434e-06, + "loss": 1.4076, + "step": 1222 + }, + { + "epoch": 1.2562917308680022, + "grad_norm": 2.6480257511138916, + "learning_rate": 3.258469737342977e-06, + "loss": 1.422, + "step": 1223 + }, + { + "epoch": 1.2573189522342065, + "grad_norm": 2.7006945610046387, + "learning_rate": 3.25656642558051e-06, + "loss": 1.3975, + "step": 1224 + }, + { + "epoch": 1.258346173600411, + "grad_norm": 2.6176297664642334, + "learning_rate": 3.2546631138180435e-06, + "loss": 1.4138, + "step": 1225 + }, + { + "epoch": 1.2593733949666153, + "grad_norm": 2.717742443084717, + "learning_rate": 3.252759802055577e-06, + "loss": 1.4092, + "step": 1226 + }, + { + "epoch": 1.2604006163328196, + "grad_norm": 2.693643093109131, + "learning_rate": 3.25085649029311e-06, + "loss": 1.4221, + "step": 1227 + }, + { + "epoch": 1.2614278376990242, + "grad_norm": 2.709026575088501, + "learning_rate": 3.248953178530644e-06, + "loss": 1.4474, + "step": 1228 + }, + { + "epoch": 1.2624550590652286, + "grad_norm": 2.628399610519409, + "learning_rate": 3.247049866768177e-06, + "loss": 1.4075, + "step": 1229 + }, + { + "epoch": 1.263482280431433, + "grad_norm": 2.681164503097534, + "learning_rate": 3.2451465550057103e-06, + "loss": 1.3683, + "step": 1230 + }, + { + "epoch": 1.263482280431433, + "eval_loss": 1.8543845415115356, + "eval_runtime": 37.1602, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 1230 + }, + { + "epoch": 1.2645095017976373, + "grad_norm": 2.6538004875183105, + "learning_rate": 3.2432432432432437e-06, + "loss": 1.4214, + "step": 1231 + }, + { + "epoch": 1.2655367231638417, + "grad_norm": 2.714805841445923, + "learning_rate": 3.241339931480777e-06, + "loss": 1.4617, + "step": 1232 + }, + { + "epoch": 1.2665639445300463, + "grad_norm": 2.743016481399536, + "learning_rate": 3.2394366197183104e-06, + "loss": 1.4556, + "step": 1233 + }, + { + "epoch": 1.2675911658962506, + "grad_norm": 2.618269681930542, + "learning_rate": 3.2375333079558434e-06, + "loss": 1.419, + "step": 1234 + }, + { + "epoch": 1.268618387262455, + "grad_norm": 2.672124147415161, + "learning_rate": 3.2356299961933767e-06, + "loss": 1.4477, + "step": 1235 + }, + { + "epoch": 1.2696456086286596, + "grad_norm": 2.831037998199463, + "learning_rate": 3.23372668443091e-06, + "loss": 1.4552, + "step": 1236 + }, + { + "epoch": 1.270672829994864, + "grad_norm": 2.713193893432617, + "learning_rate": 3.2318233726684435e-06, + "loss": 1.4653, + "step": 1237 + }, + { + "epoch": 1.2717000513610683, + "grad_norm": 2.645097017288208, + "learning_rate": 3.229920060905977e-06, + "loss": 1.4087, + "step": 1238 + }, + { + "epoch": 1.2727272727272727, + "grad_norm": 2.741934061050415, + "learning_rate": 3.2280167491435098e-06, + "loss": 1.4155, + "step": 1239 + }, + { + "epoch": 1.273754494093477, + "grad_norm": 2.5912556648254395, + "learning_rate": 3.226113437381043e-06, + "loss": 1.4214, + "step": 1240 + }, + { + "epoch": 1.273754494093477, + "eval_loss": 1.8524503707885742, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1240 + }, + { + "epoch": 1.2747817154596817, + "grad_norm": 2.5774216651916504, + "learning_rate": 3.2242101256185765e-06, + "loss": 1.4309, + "step": 1241 + }, + { + "epoch": 1.275808936825886, + "grad_norm": 2.7143211364746094, + "learning_rate": 3.22230681385611e-06, + "loss": 1.4928, + "step": 1242 + }, + { + "epoch": 1.2768361581920904, + "grad_norm": 2.8101789951324463, + "learning_rate": 3.220403502093643e-06, + "loss": 1.4325, + "step": 1243 + }, + { + "epoch": 1.2778633795582948, + "grad_norm": 2.704641580581665, + "learning_rate": 3.218500190331176e-06, + "loss": 1.4729, + "step": 1244 + }, + { + "epoch": 1.2788906009244991, + "grad_norm": 2.649195671081543, + "learning_rate": 3.2165968785687095e-06, + "loss": 1.4186, + "step": 1245 + }, + { + "epoch": 1.2799178222907037, + "grad_norm": 2.7863168716430664, + "learning_rate": 3.214693566806243e-06, + "loss": 1.4474, + "step": 1246 + }, + { + "epoch": 1.280945043656908, + "grad_norm": 2.693589687347412, + "learning_rate": 3.2127902550437762e-06, + "loss": 1.4293, + "step": 1247 + }, + { + "epoch": 1.2819722650231125, + "grad_norm": 2.7287509441375732, + "learning_rate": 3.21088694328131e-06, + "loss": 1.4305, + "step": 1248 + }, + { + "epoch": 1.2829994863893168, + "grad_norm": 2.5521676540374756, + "learning_rate": 3.2089836315188434e-06, + "loss": 1.4203, + "step": 1249 + }, + { + "epoch": 1.2840267077555212, + "grad_norm": 2.745425224304199, + "learning_rate": 3.2070803197563767e-06, + "loss": 1.4234, + "step": 1250 + }, + { + "epoch": 1.2840267077555212, + "eval_loss": 1.8586679697036743, + "eval_runtime": 37.1173, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1250 + }, + { + "epoch": 1.2850539291217258, + "grad_norm": 2.687504768371582, + "learning_rate": 3.2051770079939097e-06, + "loss": 1.443, + "step": 1251 + }, + { + "epoch": 1.2860811504879301, + "grad_norm": 2.7841391563415527, + "learning_rate": 3.203273696231443e-06, + "loss": 1.4722, + "step": 1252 + }, + { + "epoch": 1.2871083718541345, + "grad_norm": 2.7715537548065186, + "learning_rate": 3.2013703844689764e-06, + "loss": 1.4063, + "step": 1253 + }, + { + "epoch": 1.288135593220339, + "grad_norm": 2.8183677196502686, + "learning_rate": 3.1994670727065098e-06, + "loss": 1.4534, + "step": 1254 + }, + { + "epoch": 1.2891628145865435, + "grad_norm": 2.795215606689453, + "learning_rate": 3.197563760944043e-06, + "loss": 1.4779, + "step": 1255 + }, + { + "epoch": 1.2901900359527478, + "grad_norm": 2.7606441974639893, + "learning_rate": 3.195660449181576e-06, + "loss": 1.4668, + "step": 1256 + }, + { + "epoch": 1.2912172573189522, + "grad_norm": 2.629718065261841, + "learning_rate": 3.1937571374191094e-06, + "loss": 1.4114, + "step": 1257 + }, + { + "epoch": 1.2922444786851566, + "grad_norm": 2.776845932006836, + "learning_rate": 3.191853825656643e-06, + "loss": 1.4348, + "step": 1258 + }, + { + "epoch": 1.2932717000513612, + "grad_norm": 2.672982931137085, + "learning_rate": 3.189950513894176e-06, + "loss": 1.4274, + "step": 1259 + }, + { + "epoch": 1.2942989214175655, + "grad_norm": 2.5834884643554688, + "learning_rate": 3.1880472021317095e-06, + "loss": 1.4282, + "step": 1260 + }, + { + "epoch": 1.2942989214175655, + "eval_loss": 1.8541877269744873, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1260 + }, + { + "epoch": 1.29532614278377, + "grad_norm": 2.752054452896118, + "learning_rate": 3.1861438903692425e-06, + "loss": 1.3797, + "step": 1261 + }, + { + "epoch": 1.2963533641499743, + "grad_norm": 2.723689556121826, + "learning_rate": 3.184240578606776e-06, + "loss": 1.4137, + "step": 1262 + }, + { + "epoch": 1.2973805855161786, + "grad_norm": 2.806439161300659, + "learning_rate": 3.182337266844309e-06, + "loss": 1.4396, + "step": 1263 + }, + { + "epoch": 1.2984078068823832, + "grad_norm": 2.8140552043914795, + "learning_rate": 3.1804339550818426e-06, + "loss": 1.4677, + "step": 1264 + }, + { + "epoch": 1.2994350282485876, + "grad_norm": 2.78153920173645, + "learning_rate": 3.178530643319376e-06, + "loss": 1.433, + "step": 1265 + }, + { + "epoch": 1.300462249614792, + "grad_norm": 2.6926140785217285, + "learning_rate": 3.176627331556909e-06, + "loss": 1.3901, + "step": 1266 + }, + { + "epoch": 1.3014894709809965, + "grad_norm": 2.636629581451416, + "learning_rate": 3.1747240197944422e-06, + "loss": 1.4331, + "step": 1267 + }, + { + "epoch": 1.3025166923472007, + "grad_norm": 2.625173330307007, + "learning_rate": 3.172820708031976e-06, + "loss": 1.3918, + "step": 1268 + }, + { + "epoch": 1.3035439137134053, + "grad_norm": 2.745666980743408, + "learning_rate": 3.1709173962695094e-06, + "loss": 1.4669, + "step": 1269 + }, + { + "epoch": 1.3045711350796096, + "grad_norm": 2.8781471252441406, + "learning_rate": 3.1690140845070427e-06, + "loss": 1.4586, + "step": 1270 + }, + { + "epoch": 1.3045711350796096, + "eval_loss": 1.8759416341781616, + "eval_runtime": 37.1372, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 1270 + }, + { + "epoch": 1.305598356445814, + "grad_norm": 2.832078456878662, + "learning_rate": 3.167110772744576e-06, + "loss": 1.4385, + "step": 1271 + }, + { + "epoch": 1.3066255778120186, + "grad_norm": 2.7153384685516357, + "learning_rate": 3.1652074609821095e-06, + "loss": 1.427, + "step": 1272 + }, + { + "epoch": 1.307652799178223, + "grad_norm": 2.6645848751068115, + "learning_rate": 3.1633041492196424e-06, + "loss": 1.4147, + "step": 1273 + }, + { + "epoch": 1.3086800205444273, + "grad_norm": 2.7231268882751465, + "learning_rate": 3.1614008374571758e-06, + "loss": 1.436, + "step": 1274 + }, + { + "epoch": 1.3097072419106317, + "grad_norm": 2.8131442070007324, + "learning_rate": 3.159497525694709e-06, + "loss": 1.4357, + "step": 1275 + }, + { + "epoch": 1.310734463276836, + "grad_norm": 2.6366660594940186, + "learning_rate": 3.1575942139322425e-06, + "loss": 1.4089, + "step": 1276 + }, + { + "epoch": 1.3117616846430407, + "grad_norm": 2.7849857807159424, + "learning_rate": 3.155690902169776e-06, + "loss": 1.4486, + "step": 1277 + }, + { + "epoch": 1.312788906009245, + "grad_norm": 2.771298408508301, + "learning_rate": 3.1537875904073088e-06, + "loss": 1.4085, + "step": 1278 + }, + { + "epoch": 1.3138161273754494, + "grad_norm": 2.7618181705474854, + "learning_rate": 3.151884278644842e-06, + "loss": 1.4272, + "step": 1279 + }, + { + "epoch": 1.3148433487416538, + "grad_norm": 2.6741080284118652, + "learning_rate": 3.1499809668823755e-06, + "loss": 1.4135, + "step": 1280 + }, + { + "epoch": 1.3148433487416538, + "eval_loss": 1.8567897081375122, + "eval_runtime": 37.1429, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 1280 + }, + { + "epoch": 1.3158705701078581, + "grad_norm": 2.6671698093414307, + "learning_rate": 3.148077655119909e-06, + "loss": 1.4598, + "step": 1281 + }, + { + "epoch": 1.3168977914740627, + "grad_norm": 2.6641244888305664, + "learning_rate": 3.1461743433574422e-06, + "loss": 1.3998, + "step": 1282 + }, + { + "epoch": 1.317925012840267, + "grad_norm": 2.747417449951172, + "learning_rate": 3.144271031594975e-06, + "loss": 1.4602, + "step": 1283 + }, + { + "epoch": 1.3189522342064715, + "grad_norm": 2.708984136581421, + "learning_rate": 3.1423677198325085e-06, + "loss": 1.4562, + "step": 1284 + }, + { + "epoch": 1.319979455572676, + "grad_norm": 2.5973010063171387, + "learning_rate": 3.140464408070042e-06, + "loss": 1.4121, + "step": 1285 + }, + { + "epoch": 1.3210066769388804, + "grad_norm": 2.6334621906280518, + "learning_rate": 3.1385610963075753e-06, + "loss": 1.4049, + "step": 1286 + }, + { + "epoch": 1.3220338983050848, + "grad_norm": 2.772655963897705, + "learning_rate": 3.1366577845451086e-06, + "loss": 1.3953, + "step": 1287 + }, + { + "epoch": 1.3230611196712891, + "grad_norm": 2.5006916522979736, + "learning_rate": 3.134754472782642e-06, + "loss": 1.3842, + "step": 1288 + }, + { + "epoch": 1.3240883410374935, + "grad_norm": 2.6052703857421875, + "learning_rate": 3.1328511610201758e-06, + "loss": 1.3801, + "step": 1289 + }, + { + "epoch": 1.325115562403698, + "grad_norm": 2.703693151473999, + "learning_rate": 3.1309478492577087e-06, + "loss": 1.3672, + "step": 1290 + }, + { + "epoch": 1.325115562403698, + "eval_loss": 1.8470649719238281, + "eval_runtime": 37.2714, + "eval_samples_per_second": 26.83, + "eval_steps_per_second": 1.127, + "step": 1290 + }, + { + "epoch": 1.3261427837699025, + "grad_norm": 2.6864330768585205, + "learning_rate": 3.129044537495242e-06, + "loss": 1.4252, + "step": 1291 + }, + { + "epoch": 1.3271700051361068, + "grad_norm": 2.7768399715423584, + "learning_rate": 3.1271412257327754e-06, + "loss": 1.4483, + "step": 1292 + }, + { + "epoch": 1.3281972265023112, + "grad_norm": 2.6658599376678467, + "learning_rate": 3.125237913970309e-06, + "loss": 1.4059, + "step": 1293 + }, + { + "epoch": 1.3292244478685156, + "grad_norm": 2.754718065261841, + "learning_rate": 3.123334602207842e-06, + "loss": 1.3787, + "step": 1294 + }, + { + "epoch": 1.3302516692347202, + "grad_norm": 2.7994842529296875, + "learning_rate": 3.121431290445375e-06, + "loss": 1.4623, + "step": 1295 + }, + { + "epoch": 1.3312788906009245, + "grad_norm": 2.805941343307495, + "learning_rate": 3.1195279786829085e-06, + "loss": 1.5186, + "step": 1296 + }, + { + "epoch": 1.332306111967129, + "grad_norm": 2.6829605102539062, + "learning_rate": 3.117624666920442e-06, + "loss": 1.4052, + "step": 1297 + }, + { + "epoch": 1.3333333333333333, + "grad_norm": 2.7221593856811523, + "learning_rate": 3.115721355157975e-06, + "loss": 1.4031, + "step": 1298 + }, + { + "epoch": 1.3343605546995376, + "grad_norm": 2.579638719558716, + "learning_rate": 3.1138180433955085e-06, + "loss": 1.3627, + "step": 1299 + }, + { + "epoch": 1.3353877760657422, + "grad_norm": 2.861189842224121, + "learning_rate": 3.1119147316330415e-06, + "loss": 1.4405, + "step": 1300 + }, + { + "epoch": 1.3353877760657422, + "eval_loss": 1.8476718664169312, + "eval_runtime": 37.4298, + "eval_samples_per_second": 26.717, + "eval_steps_per_second": 1.122, + "step": 1300 + }, + { + "epoch": 1.3364149974319466, + "grad_norm": 2.8968658447265625, + "learning_rate": 3.110011419870575e-06, + "loss": 1.4774, + "step": 1301 + }, + { + "epoch": 1.337442218798151, + "grad_norm": 2.8708558082580566, + "learning_rate": 3.1081081081081082e-06, + "loss": 1.4085, + "step": 1302 + }, + { + "epoch": 1.3384694401643555, + "grad_norm": 2.802922487258911, + "learning_rate": 3.1062047963456416e-06, + "loss": 1.4486, + "step": 1303 + }, + { + "epoch": 1.33949666153056, + "grad_norm": 2.9320907592773438, + "learning_rate": 3.104301484583175e-06, + "loss": 1.4932, + "step": 1304 + }, + { + "epoch": 1.3405238828967643, + "grad_norm": 2.7708749771118164, + "learning_rate": 3.1023981728207083e-06, + "loss": 1.4601, + "step": 1305 + }, + { + "epoch": 1.3415511042629686, + "grad_norm": 2.7464756965637207, + "learning_rate": 3.1004948610582412e-06, + "loss": 1.4245, + "step": 1306 + }, + { + "epoch": 1.342578325629173, + "grad_norm": 2.767235517501831, + "learning_rate": 3.0985915492957746e-06, + "loss": 1.4199, + "step": 1307 + }, + { + "epoch": 1.3436055469953776, + "grad_norm": 2.6810076236724854, + "learning_rate": 3.096688237533308e-06, + "loss": 1.3851, + "step": 1308 + }, + { + "epoch": 1.344632768361582, + "grad_norm": 2.6390531063079834, + "learning_rate": 3.0947849257708418e-06, + "loss": 1.4688, + "step": 1309 + }, + { + "epoch": 1.3456599897277863, + "grad_norm": 2.729029893875122, + "learning_rate": 3.092881614008375e-06, + "loss": 1.4248, + "step": 1310 + }, + { + "epoch": 1.3456599897277863, + "eval_loss": 1.854402780532837, + "eval_runtime": 36.9357, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1310 + }, + { + "epoch": 1.3466872110939907, + "grad_norm": 2.8098297119140625, + "learning_rate": 3.0909783022459085e-06, + "loss": 1.4761, + "step": 1311 + }, + { + "epoch": 1.347714432460195, + "grad_norm": 2.8076772689819336, + "learning_rate": 3.0890749904834414e-06, + "loss": 1.4119, + "step": 1312 + }, + { + "epoch": 1.3487416538263997, + "grad_norm": 2.8214125633239746, + "learning_rate": 3.0871716787209748e-06, + "loss": 1.4693, + "step": 1313 + }, + { + "epoch": 1.349768875192604, + "grad_norm": 2.830083131790161, + "learning_rate": 3.085268366958508e-06, + "loss": 1.438, + "step": 1314 + }, + { + "epoch": 1.3507960965588084, + "grad_norm": 2.7671611309051514, + "learning_rate": 3.0833650551960415e-06, + "loss": 1.4417, + "step": 1315 + }, + { + "epoch": 1.3518233179250128, + "grad_norm": 2.6464383602142334, + "learning_rate": 3.081461743433575e-06, + "loss": 1.4096, + "step": 1316 + }, + { + "epoch": 1.3528505392912171, + "grad_norm": 2.651646137237549, + "learning_rate": 3.079558431671108e-06, + "loss": 1.4264, + "step": 1317 + }, + { + "epoch": 1.3538777606574217, + "grad_norm": 2.6222681999206543, + "learning_rate": 3.077655119908641e-06, + "loss": 1.3836, + "step": 1318 + }, + { + "epoch": 1.354904982023626, + "grad_norm": 2.5674147605895996, + "learning_rate": 3.0757518081461745e-06, + "loss": 1.4206, + "step": 1319 + }, + { + "epoch": 1.3559322033898304, + "grad_norm": 2.7767961025238037, + "learning_rate": 3.073848496383708e-06, + "loss": 1.4348, + "step": 1320 + }, + { + "epoch": 1.3559322033898304, + "eval_loss": 1.8607728481292725, + "eval_runtime": 36.99, + "eval_samples_per_second": 27.034, + "eval_steps_per_second": 1.135, + "step": 1320 + }, + { + "epoch": 1.356959424756035, + "grad_norm": 2.7811837196350098, + "learning_rate": 3.0719451846212413e-06, + "loss": 1.4082, + "step": 1321 + }, + { + "epoch": 1.3579866461222394, + "grad_norm": 2.5633227825164795, + "learning_rate": 3.0700418728587746e-06, + "loss": 1.4352, + "step": 1322 + }, + { + "epoch": 1.3590138674884438, + "grad_norm": 2.7205374240875244, + "learning_rate": 3.0681385610963076e-06, + "loss": 1.405, + "step": 1323 + }, + { + "epoch": 1.3600410888546481, + "grad_norm": 2.82409930229187, + "learning_rate": 3.066235249333841e-06, + "loss": 1.448, + "step": 1324 + }, + { + "epoch": 1.3610683102208525, + "grad_norm": 2.7932820320129395, + "learning_rate": 3.0643319375713743e-06, + "loss": 1.4785, + "step": 1325 + }, + { + "epoch": 1.362095531587057, + "grad_norm": 2.68669056892395, + "learning_rate": 3.0624286258089076e-06, + "loss": 1.4232, + "step": 1326 + }, + { + "epoch": 1.3631227529532615, + "grad_norm": 2.6492018699645996, + "learning_rate": 3.060525314046441e-06, + "loss": 1.4052, + "step": 1327 + }, + { + "epoch": 1.3641499743194658, + "grad_norm": 2.7677114009857178, + "learning_rate": 3.058622002283974e-06, + "loss": 1.4426, + "step": 1328 + }, + { + "epoch": 1.3651771956856702, + "grad_norm": 2.7124569416046143, + "learning_rate": 3.0567186905215077e-06, + "loss": 1.4631, + "step": 1329 + }, + { + "epoch": 1.3662044170518746, + "grad_norm": 2.7084693908691406, + "learning_rate": 3.054815378759041e-06, + "loss": 1.4656, + "step": 1330 + }, + { + "epoch": 1.3662044170518746, + "eval_loss": 1.8457012176513672, + "eval_runtime": 37.0141, + "eval_samples_per_second": 27.017, + "eval_steps_per_second": 1.135, + "step": 1330 + }, + { + "epoch": 1.3672316384180792, + "grad_norm": 2.7612688541412354, + "learning_rate": 3.0529120669965745e-06, + "loss": 1.455, + "step": 1331 + }, + { + "epoch": 1.3682588597842835, + "grad_norm": 2.5172924995422363, + "learning_rate": 3.051008755234108e-06, + "loss": 1.3915, + "step": 1332 + }, + { + "epoch": 1.3692860811504879, + "grad_norm": 2.7225120067596436, + "learning_rate": 3.049105443471641e-06, + "loss": 1.4025, + "step": 1333 + }, + { + "epoch": 1.3703133025166925, + "grad_norm": 2.835470676422119, + "learning_rate": 3.0472021317091745e-06, + "loss": 1.5423, + "step": 1334 + }, + { + "epoch": 1.3713405238828968, + "grad_norm": 2.6247284412384033, + "learning_rate": 3.0452988199467075e-06, + "loss": 1.4562, + "step": 1335 + }, + { + "epoch": 1.3723677452491012, + "grad_norm": 2.6346960067749023, + "learning_rate": 3.043395508184241e-06, + "loss": 1.4566, + "step": 1336 + }, + { + "epoch": 1.3733949666153056, + "grad_norm": 2.6278724670410156, + "learning_rate": 3.041492196421774e-06, + "loss": 1.403, + "step": 1337 + }, + { + "epoch": 1.37442218798151, + "grad_norm": 2.8066110610961914, + "learning_rate": 3.0395888846593076e-06, + "loss": 1.4433, + "step": 1338 + }, + { + "epoch": 1.3754494093477145, + "grad_norm": 2.6565256118774414, + "learning_rate": 3.037685572896841e-06, + "loss": 1.4177, + "step": 1339 + }, + { + "epoch": 1.376476630713919, + "grad_norm": 2.6529884338378906, + "learning_rate": 3.035782261134374e-06, + "loss": 1.4378, + "step": 1340 + }, + { + "epoch": 1.376476630713919, + "eval_loss": 1.8618167638778687, + "eval_runtime": 37.025, + "eval_samples_per_second": 27.009, + "eval_steps_per_second": 1.134, + "step": 1340 + }, + { + "epoch": 1.3775038520801233, + "grad_norm": 2.820838689804077, + "learning_rate": 3.0338789493719072e-06, + "loss": 1.4471, + "step": 1341 + }, + { + "epoch": 1.3785310734463276, + "grad_norm": 2.6786739826202393, + "learning_rate": 3.0319756376094406e-06, + "loss": 1.4393, + "step": 1342 + }, + { + "epoch": 1.379558294812532, + "grad_norm": 2.7962610721588135, + "learning_rate": 3.030072325846974e-06, + "loss": 1.4407, + "step": 1343 + }, + { + "epoch": 1.3805855161787366, + "grad_norm": 2.739959716796875, + "learning_rate": 3.0281690140845073e-06, + "loss": 1.4059, + "step": 1344 + }, + { + "epoch": 1.381612737544941, + "grad_norm": 2.7812039852142334, + "learning_rate": 3.0262657023220403e-06, + "loss": 1.4449, + "step": 1345 + }, + { + "epoch": 1.3826399589111453, + "grad_norm": 2.749671459197998, + "learning_rate": 3.0243623905595736e-06, + "loss": 1.4459, + "step": 1346 + }, + { + "epoch": 1.3836671802773497, + "grad_norm": 2.7494630813598633, + "learning_rate": 3.022459078797107e-06, + "loss": 1.4204, + "step": 1347 + }, + { + "epoch": 1.384694401643554, + "grad_norm": 2.7188141345977783, + "learning_rate": 3.0205557670346404e-06, + "loss": 1.4249, + "step": 1348 + }, + { + "epoch": 1.3857216230097587, + "grad_norm": 2.712442636489868, + "learning_rate": 3.018652455272174e-06, + "loss": 1.3975, + "step": 1349 + }, + { + "epoch": 1.386748844375963, + "grad_norm": 2.9888060092926025, + "learning_rate": 3.0167491435097075e-06, + "loss": 1.5258, + "step": 1350 + }, + { + "epoch": 1.386748844375963, + "eval_loss": 1.8591758012771606, + "eval_runtime": 37.0078, + "eval_samples_per_second": 27.021, + "eval_steps_per_second": 1.135, + "step": 1350 + }, + { + "epoch": 1.3877760657421674, + "grad_norm": 2.789698600769043, + "learning_rate": 3.014845831747241e-06, + "loss": 1.4305, + "step": 1351 + }, + { + "epoch": 1.388803287108372, + "grad_norm": 2.747840166091919, + "learning_rate": 3.012942519984774e-06, + "loss": 1.4031, + "step": 1352 + }, + { + "epoch": 1.3898305084745763, + "grad_norm": 2.7078518867492676, + "learning_rate": 3.011039208222307e-06, + "loss": 1.4139, + "step": 1353 + }, + { + "epoch": 1.3908577298407807, + "grad_norm": 2.615677833557129, + "learning_rate": 3.0091358964598405e-06, + "loss": 1.3728, + "step": 1354 + }, + { + "epoch": 1.391884951206985, + "grad_norm": 2.7485580444335938, + "learning_rate": 3.007232584697374e-06, + "loss": 1.4272, + "step": 1355 + }, + { + "epoch": 1.3929121725731894, + "grad_norm": 2.7168128490448, + "learning_rate": 3.0053292729349073e-06, + "loss": 1.4353, + "step": 1356 + }, + { + "epoch": 1.393939393939394, + "grad_norm": 2.7295777797698975, + "learning_rate": 3.00342596117244e-06, + "loss": 1.4324, + "step": 1357 + }, + { + "epoch": 1.3949666153055984, + "grad_norm": 2.8791208267211914, + "learning_rate": 3.0015226494099736e-06, + "loss": 1.4303, + "step": 1358 + }, + { + "epoch": 1.3959938366718028, + "grad_norm": 2.647028684616089, + "learning_rate": 2.999619337647507e-06, + "loss": 1.402, + "step": 1359 + }, + { + "epoch": 1.3970210580380071, + "grad_norm": 2.869901180267334, + "learning_rate": 2.9977160258850403e-06, + "loss": 1.404, + "step": 1360 + }, + { + "epoch": 1.3970210580380071, + "eval_loss": 1.8588491678237915, + "eval_runtime": 37.0159, + "eval_samples_per_second": 27.015, + "eval_steps_per_second": 1.135, + "step": 1360 + }, + { + "epoch": 1.3980482794042115, + "grad_norm": 2.7664403915405273, + "learning_rate": 2.9958127141225736e-06, + "loss": 1.3895, + "step": 1361 + }, + { + "epoch": 1.399075500770416, + "grad_norm": 2.8101043701171875, + "learning_rate": 2.9939094023601066e-06, + "loss": 1.4629, + "step": 1362 + }, + { + "epoch": 1.4001027221366205, + "grad_norm": 2.7114098072052, + "learning_rate": 2.99200609059764e-06, + "loss": 1.4365, + "step": 1363 + }, + { + "epoch": 1.4011299435028248, + "grad_norm": 2.6739253997802734, + "learning_rate": 2.9901027788351733e-06, + "loss": 1.382, + "step": 1364 + }, + { + "epoch": 1.4021571648690292, + "grad_norm": 2.8102970123291016, + "learning_rate": 2.9881994670727067e-06, + "loss": 1.392, + "step": 1365 + }, + { + "epoch": 1.4031843862352336, + "grad_norm": 2.645071506500244, + "learning_rate": 2.98629615531024e-06, + "loss": 1.4157, + "step": 1366 + }, + { + "epoch": 1.4042116076014381, + "grad_norm": 2.6110310554504395, + "learning_rate": 2.984392843547773e-06, + "loss": 1.3996, + "step": 1367 + }, + { + "epoch": 1.4052388289676425, + "grad_norm": 2.784090042114258, + "learning_rate": 2.9824895317853063e-06, + "loss": 1.4178, + "step": 1368 + }, + { + "epoch": 1.4062660503338469, + "grad_norm": 2.7293829917907715, + "learning_rate": 2.98058622002284e-06, + "loss": 1.4193, + "step": 1369 + }, + { + "epoch": 1.4072932717000515, + "grad_norm": 2.6924493312835693, + "learning_rate": 2.9786829082603735e-06, + "loss": 1.476, + "step": 1370 + }, + { + "epoch": 1.4072932717000515, + "eval_loss": 1.8642805814743042, + "eval_runtime": 37.0452, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1370 + }, + { + "epoch": 1.4083204930662558, + "grad_norm": 2.7714414596557617, + "learning_rate": 2.976779596497907e-06, + "loss": 1.4574, + "step": 1371 + }, + { + "epoch": 1.4093477144324602, + "grad_norm": 2.7013490200042725, + "learning_rate": 2.97487628473544e-06, + "loss": 1.4718, + "step": 1372 + }, + { + "epoch": 1.4103749357986646, + "grad_norm": 2.9120547771453857, + "learning_rate": 2.9729729729729736e-06, + "loss": 1.4298, + "step": 1373 + }, + { + "epoch": 1.411402157164869, + "grad_norm": 2.9127988815307617, + "learning_rate": 2.9710696612105065e-06, + "loss": 1.4703, + "step": 1374 + }, + { + "epoch": 1.4124293785310735, + "grad_norm": 2.696772575378418, + "learning_rate": 2.96916634944804e-06, + "loss": 1.412, + "step": 1375 + }, + { + "epoch": 1.413456599897278, + "grad_norm": 2.8330435752868652, + "learning_rate": 2.9672630376855732e-06, + "loss": 1.4731, + "step": 1376 + }, + { + "epoch": 1.4144838212634823, + "grad_norm": 2.7354564666748047, + "learning_rate": 2.9653597259231066e-06, + "loss": 1.4342, + "step": 1377 + }, + { + "epoch": 1.4155110426296866, + "grad_norm": 2.7386012077331543, + "learning_rate": 2.96345641416064e-06, + "loss": 1.3935, + "step": 1378 + }, + { + "epoch": 1.416538263995891, + "grad_norm": 2.5881052017211914, + "learning_rate": 2.961553102398173e-06, + "loss": 1.3282, + "step": 1379 + }, + { + "epoch": 1.4175654853620956, + "grad_norm": 2.805558443069458, + "learning_rate": 2.9596497906357063e-06, + "loss": 1.4691, + "step": 1380 + }, + { + "epoch": 1.4175654853620956, + "eval_loss": 1.84714937210083, + "eval_runtime": 37.0349, + "eval_samples_per_second": 27.002, + "eval_steps_per_second": 1.134, + "step": 1380 + }, + { + "epoch": 1.4185927067283, + "grad_norm": 2.8639137744903564, + "learning_rate": 2.9577464788732396e-06, + "loss": 1.4745, + "step": 1381 + }, + { + "epoch": 1.4196199280945043, + "grad_norm": 2.6688785552978516, + "learning_rate": 2.955843167110773e-06, + "loss": 1.4016, + "step": 1382 + }, + { + "epoch": 1.420647149460709, + "grad_norm": 2.6851978302001953, + "learning_rate": 2.9539398553483064e-06, + "loss": 1.4205, + "step": 1383 + }, + { + "epoch": 1.4216743708269133, + "grad_norm": 2.7381017208099365, + "learning_rate": 2.9520365435858393e-06, + "loss": 1.4315, + "step": 1384 + }, + { + "epoch": 1.4227015921931176, + "grad_norm": 2.9249424934387207, + "learning_rate": 2.9501332318233727e-06, + "loss": 1.49, + "step": 1385 + }, + { + "epoch": 1.423728813559322, + "grad_norm": 2.765490770339966, + "learning_rate": 2.948229920060906e-06, + "loss": 1.4359, + "step": 1386 + }, + { + "epoch": 1.4247560349255264, + "grad_norm": 2.578561544418335, + "learning_rate": 2.9463266082984394e-06, + "loss": 1.4032, + "step": 1387 + }, + { + "epoch": 1.425783256291731, + "grad_norm": 2.6673238277435303, + "learning_rate": 2.9444232965359727e-06, + "loss": 1.4144, + "step": 1388 + }, + { + "epoch": 1.4268104776579353, + "grad_norm": 2.737516164779663, + "learning_rate": 2.9425199847735057e-06, + "loss": 1.4375, + "step": 1389 + }, + { + "epoch": 1.4278376990241397, + "grad_norm": 2.656177282333374, + "learning_rate": 2.94061667301104e-06, + "loss": 1.4007, + "step": 1390 + }, + { + "epoch": 1.4278376990241397, + "eval_loss": 1.8427128791809082, + "eval_runtime": 37.1322, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1390 + }, + { + "epoch": 1.428864920390344, + "grad_norm": 2.6011996269226074, + "learning_rate": 2.938713361248573e-06, + "loss": 1.339, + "step": 1391 + }, + { + "epoch": 1.4298921417565484, + "grad_norm": 2.6374619007110596, + "learning_rate": 2.936810049486106e-06, + "loss": 1.4092, + "step": 1392 + }, + { + "epoch": 1.430919363122753, + "grad_norm": 2.6879520416259766, + "learning_rate": 2.9349067377236396e-06, + "loss": 1.4235, + "step": 1393 + }, + { + "epoch": 1.4319465844889574, + "grad_norm": 2.8400580883026123, + "learning_rate": 2.933003425961173e-06, + "loss": 1.381, + "step": 1394 + }, + { + "epoch": 1.4329738058551618, + "grad_norm": 2.742767810821533, + "learning_rate": 2.9311001141987063e-06, + "loss": 1.4055, + "step": 1395 + }, + { + "epoch": 1.4340010272213661, + "grad_norm": 2.734239101409912, + "learning_rate": 2.9291968024362392e-06, + "loss": 1.3997, + "step": 1396 + }, + { + "epoch": 1.4350282485875705, + "grad_norm": 2.6337532997131348, + "learning_rate": 2.9272934906737726e-06, + "loss": 1.3825, + "step": 1397 + }, + { + "epoch": 1.436055469953775, + "grad_norm": 2.668790578842163, + "learning_rate": 2.925390178911306e-06, + "loss": 1.3918, + "step": 1398 + }, + { + "epoch": 1.4370826913199795, + "grad_norm": 2.8307507038116455, + "learning_rate": 2.9234868671488393e-06, + "loss": 1.4183, + "step": 1399 + }, + { + "epoch": 1.4381099126861838, + "grad_norm": 2.6661787033081055, + "learning_rate": 2.9215835553863727e-06, + "loss": 1.4225, + "step": 1400 + }, + { + "epoch": 1.4381099126861838, + "eval_loss": 1.8518847227096558, + "eval_runtime": 37.3672, + "eval_samples_per_second": 26.761, + "eval_steps_per_second": 1.124, + "step": 1400 + }, + { + "epoch": 1.4391371340523884, + "grad_norm": 2.698025703430176, + "learning_rate": 2.9196802436239056e-06, + "loss": 1.4115, + "step": 1401 + }, + { + "epoch": 1.4401643554185928, + "grad_norm": 2.6792502403259277, + "learning_rate": 2.917776931861439e-06, + "loss": 1.3869, + "step": 1402 + }, + { + "epoch": 1.4411915767847971, + "grad_norm": 2.693514347076416, + "learning_rate": 2.9158736200989723e-06, + "loss": 1.3875, + "step": 1403 + }, + { + "epoch": 1.4422187981510015, + "grad_norm": 2.6383583545684814, + "learning_rate": 2.9139703083365057e-06, + "loss": 1.3942, + "step": 1404 + }, + { + "epoch": 1.4432460195172059, + "grad_norm": 2.6992499828338623, + "learning_rate": 2.912066996574039e-06, + "loss": 1.4243, + "step": 1405 + }, + { + "epoch": 1.4442732408834105, + "grad_norm": 2.7901880741119385, + "learning_rate": 2.910163684811572e-06, + "loss": 1.3559, + "step": 1406 + }, + { + "epoch": 1.4453004622496148, + "grad_norm": 2.9373207092285156, + "learning_rate": 2.9082603730491054e-06, + "loss": 1.4734, + "step": 1407 + }, + { + "epoch": 1.4463276836158192, + "grad_norm": 2.8224740028381348, + "learning_rate": 2.9063570612866387e-06, + "loss": 1.4474, + "step": 1408 + }, + { + "epoch": 1.4473549049820236, + "grad_norm": 2.9534835815429688, + "learning_rate": 2.904453749524172e-06, + "loss": 1.3691, + "step": 1409 + }, + { + "epoch": 1.448382126348228, + "grad_norm": 2.8153076171875, + "learning_rate": 2.902550437761706e-06, + "loss": 1.3954, + "step": 1410 + }, + { + "epoch": 1.448382126348228, + "eval_loss": 1.842891812324524, + "eval_runtime": 36.9359, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 1410 + }, + { + "epoch": 1.4494093477144325, + "grad_norm": 2.8490853309631348, + "learning_rate": 2.9006471259992392e-06, + "loss": 1.4114, + "step": 1411 + }, + { + "epoch": 1.450436569080637, + "grad_norm": 2.7475955486297607, + "learning_rate": 2.8987438142367726e-06, + "loss": 1.4068, + "step": 1412 + }, + { + "epoch": 1.4514637904468413, + "grad_norm": 2.7209911346435547, + "learning_rate": 2.8968405024743055e-06, + "loss": 1.4658, + "step": 1413 + }, + { + "epoch": 1.4524910118130456, + "grad_norm": 2.578934907913208, + "learning_rate": 2.894937190711839e-06, + "loss": 1.3686, + "step": 1414 + }, + { + "epoch": 1.45351823317925, + "grad_norm": 2.722381114959717, + "learning_rate": 2.8930338789493723e-06, + "loss": 1.4337, + "step": 1415 + }, + { + "epoch": 1.4545454545454546, + "grad_norm": 2.702178478240967, + "learning_rate": 2.8911305671869056e-06, + "loss": 1.4163, + "step": 1416 + }, + { + "epoch": 1.455572675911659, + "grad_norm": 2.6742405891418457, + "learning_rate": 2.889227255424439e-06, + "loss": 1.3585, + "step": 1417 + }, + { + "epoch": 1.4565998972778633, + "grad_norm": 2.6732680797576904, + "learning_rate": 2.887323943661972e-06, + "loss": 1.406, + "step": 1418 + }, + { + "epoch": 1.457627118644068, + "grad_norm": 2.698864459991455, + "learning_rate": 2.8854206318995053e-06, + "loss": 1.3872, + "step": 1419 + }, + { + "epoch": 1.4586543400102723, + "grad_norm": 2.7534351348876953, + "learning_rate": 2.8835173201370386e-06, + "loss": 1.4055, + "step": 1420 + }, + { + "epoch": 1.4586543400102723, + "eval_loss": 1.8564038276672363, + "eval_runtime": 37.0708, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 1420 + }, + { + "epoch": 1.4596815613764766, + "grad_norm": 2.9538612365722656, + "learning_rate": 2.881614008374572e-06, + "loss": 1.4596, + "step": 1421 + }, + { + "epoch": 1.460708782742681, + "grad_norm": 2.829641103744507, + "learning_rate": 2.8797106966121054e-06, + "loss": 1.4265, + "step": 1422 + }, + { + "epoch": 1.4617360041088854, + "grad_norm": 2.6814422607421875, + "learning_rate": 2.8778073848496383e-06, + "loss": 1.4112, + "step": 1423 + }, + { + "epoch": 1.46276322547509, + "grad_norm": 2.7880561351776123, + "learning_rate": 2.8759040730871717e-06, + "loss": 1.4435, + "step": 1424 + }, + { + "epoch": 1.4637904468412943, + "grad_norm": 2.676196336746216, + "learning_rate": 2.874000761324705e-06, + "loss": 1.4104, + "step": 1425 + }, + { + "epoch": 1.4648176682074987, + "grad_norm": 2.7417874336242676, + "learning_rate": 2.8720974495622384e-06, + "loss": 1.4335, + "step": 1426 + }, + { + "epoch": 1.465844889573703, + "grad_norm": 2.7870116233825684, + "learning_rate": 2.8701941377997718e-06, + "loss": 1.4226, + "step": 1427 + }, + { + "epoch": 1.4668721109399074, + "grad_norm": 2.765194892883301, + "learning_rate": 2.8682908260373047e-06, + "loss": 1.4582, + "step": 1428 + }, + { + "epoch": 1.467899332306112, + "grad_norm": 2.608175277709961, + "learning_rate": 2.866387514274838e-06, + "loss": 1.4285, + "step": 1429 + }, + { + "epoch": 1.4689265536723164, + "grad_norm": 2.7573461532592773, + "learning_rate": 2.864484202512372e-06, + "loss": 1.4395, + "step": 1430 + }, + { + "epoch": 1.4689265536723164, + "eval_loss": 1.8600319623947144, + "eval_runtime": 37.1192, + "eval_samples_per_second": 26.94, + "eval_steps_per_second": 1.131, + "step": 1430 + }, + { + "epoch": 1.4699537750385208, + "grad_norm": 2.635175943374634, + "learning_rate": 2.8625808907499052e-06, + "loss": 1.402, + "step": 1431 + }, + { + "epoch": 1.4709809964047253, + "grad_norm": 2.7318811416625977, + "learning_rate": 2.8606775789874386e-06, + "loss": 1.4352, + "step": 1432 + }, + { + "epoch": 1.4720082177709297, + "grad_norm": 2.783669948577881, + "learning_rate": 2.858774267224972e-06, + "loss": 1.4052, + "step": 1433 + }, + { + "epoch": 1.473035439137134, + "grad_norm": 2.7327511310577393, + "learning_rate": 2.8568709554625053e-06, + "loss": 1.4209, + "step": 1434 + }, + { + "epoch": 1.4740626605033385, + "grad_norm": 2.76542329788208, + "learning_rate": 2.8549676437000382e-06, + "loss": 1.4753, + "step": 1435 + }, + { + "epoch": 1.4750898818695428, + "grad_norm": 2.8031232357025146, + "learning_rate": 2.8530643319375716e-06, + "loss": 1.4399, + "step": 1436 + }, + { + "epoch": 1.4761171032357474, + "grad_norm": 2.737053155899048, + "learning_rate": 2.851161020175105e-06, + "loss": 1.4085, + "step": 1437 + }, + { + "epoch": 1.4771443246019518, + "grad_norm": 2.609987735748291, + "learning_rate": 2.8492577084126383e-06, + "loss": 1.3809, + "step": 1438 + }, + { + "epoch": 1.4781715459681561, + "grad_norm": 2.7482123374938965, + "learning_rate": 2.8473543966501717e-06, + "loss": 1.4234, + "step": 1439 + }, + { + "epoch": 1.4791987673343605, + "grad_norm": 2.80330753326416, + "learning_rate": 2.8454510848877046e-06, + "loss": 1.4665, + "step": 1440 + }, + { + "epoch": 1.4791987673343605, + "eval_loss": 1.8637592792510986, + "eval_runtime": 37.1066, + "eval_samples_per_second": 26.949, + "eval_steps_per_second": 1.132, + "step": 1440 + }, + { + "epoch": 1.4802259887005649, + "grad_norm": 2.698328733444214, + "learning_rate": 2.843547773125238e-06, + "loss": 1.4224, + "step": 1441 + }, + { + "epoch": 1.4812532100667695, + "grad_norm": 2.663471221923828, + "learning_rate": 2.8416444613627714e-06, + "loss": 1.4205, + "step": 1442 + }, + { + "epoch": 1.4822804314329738, + "grad_norm": 2.7955918312072754, + "learning_rate": 2.8397411496003047e-06, + "loss": 1.3694, + "step": 1443 + }, + { + "epoch": 1.4833076527991782, + "grad_norm": 2.7320404052734375, + "learning_rate": 2.837837837837838e-06, + "loss": 1.4208, + "step": 1444 + }, + { + "epoch": 1.4843348741653826, + "grad_norm": 2.7786989212036133, + "learning_rate": 2.835934526075371e-06, + "loss": 1.4392, + "step": 1445 + }, + { + "epoch": 1.485362095531587, + "grad_norm": 2.6282734870910645, + "learning_rate": 2.8340312143129044e-06, + "loss": 1.3866, + "step": 1446 + }, + { + "epoch": 1.4863893168977915, + "grad_norm": 2.615072011947632, + "learning_rate": 2.8321279025504377e-06, + "loss": 1.3695, + "step": 1447 + }, + { + "epoch": 1.487416538263996, + "grad_norm": 2.884423017501831, + "learning_rate": 2.830224590787971e-06, + "loss": 1.4867, + "step": 1448 + }, + { + "epoch": 1.4884437596302003, + "grad_norm": 2.7745864391326904, + "learning_rate": 2.8283212790255045e-06, + "loss": 1.42, + "step": 1449 + }, + { + "epoch": 1.4894709809964048, + "grad_norm": 2.718130111694336, + "learning_rate": 2.8264179672630383e-06, + "loss": 1.4665, + "step": 1450 + }, + { + "epoch": 1.4894709809964048, + "eval_loss": 1.8556917905807495, + "eval_runtime": 37.123, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 1450 + }, + { + "epoch": 1.4904982023626092, + "grad_norm": 2.7502355575561523, + "learning_rate": 2.8245146555005716e-06, + "loss": 1.3946, + "step": 1451 + }, + { + "epoch": 1.4915254237288136, + "grad_norm": 2.635500192642212, + "learning_rate": 2.8226113437381046e-06, + "loss": 1.3612, + "step": 1452 + }, + { + "epoch": 1.492552645095018, + "grad_norm": 2.617408275604248, + "learning_rate": 2.820708031975638e-06, + "loss": 1.4146, + "step": 1453 + }, + { + "epoch": 1.4935798664612223, + "grad_norm": 2.7706692218780518, + "learning_rate": 2.8188047202131713e-06, + "loss": 1.4474, + "step": 1454 + }, + { + "epoch": 1.494607087827427, + "grad_norm": 2.7609055042266846, + "learning_rate": 2.8169014084507046e-06, + "loss": 1.4618, + "step": 1455 + }, + { + "epoch": 1.4956343091936313, + "grad_norm": 2.757258892059326, + "learning_rate": 2.814998096688238e-06, + "loss": 1.4131, + "step": 1456 + }, + { + "epoch": 1.4966615305598356, + "grad_norm": 2.5547049045562744, + "learning_rate": 2.813094784925771e-06, + "loss": 1.3692, + "step": 1457 + }, + { + "epoch": 1.49768875192604, + "grad_norm": 2.817617177963257, + "learning_rate": 2.8111914731633043e-06, + "loss": 1.4664, + "step": 1458 + }, + { + "epoch": 1.4987159732922444, + "grad_norm": 2.6999337673187256, + "learning_rate": 2.8092881614008377e-06, + "loss": 1.4126, + "step": 1459 + }, + { + "epoch": 1.499743194658449, + "grad_norm": 2.571727991104126, + "learning_rate": 2.807384849638371e-06, + "loss": 1.3828, + "step": 1460 + }, + { + "epoch": 1.499743194658449, + "eval_loss": 1.8413318395614624, + "eval_runtime": 37.1803, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1460 + }, + { + "epoch": 1.5007704160246533, + "grad_norm": 2.615567684173584, + "learning_rate": 2.8054815378759044e-06, + "loss": 1.4336, + "step": 1461 + }, + { + "epoch": 1.5017976373908577, + "grad_norm": 2.696854591369629, + "learning_rate": 2.8035782261134378e-06, + "loss": 1.4015, + "step": 1462 + }, + { + "epoch": 1.5028248587570623, + "grad_norm": 2.7395496368408203, + "learning_rate": 2.8016749143509707e-06, + "loss": 1.4009, + "step": 1463 + }, + { + "epoch": 1.5038520801232664, + "grad_norm": 2.8281755447387695, + "learning_rate": 2.799771602588504e-06, + "loss": 1.4069, + "step": 1464 + }, + { + "epoch": 1.504879301489471, + "grad_norm": 2.641793727874756, + "learning_rate": 2.7978682908260374e-06, + "loss": 1.4341, + "step": 1465 + }, + { + "epoch": 1.5059065228556754, + "grad_norm": 2.8969004154205322, + "learning_rate": 2.795964979063571e-06, + "loss": 1.4493, + "step": 1466 + }, + { + "epoch": 1.5069337442218798, + "grad_norm": 2.758255958557129, + "learning_rate": 2.794061667301104e-06, + "loss": 1.423, + "step": 1467 + }, + { + "epoch": 1.5079609655880843, + "grad_norm": 2.7737441062927246, + "learning_rate": 2.792158355538637e-06, + "loss": 1.43, + "step": 1468 + }, + { + "epoch": 1.5089881869542885, + "grad_norm": 2.798095226287842, + "learning_rate": 2.7902550437761705e-06, + "loss": 1.4276, + "step": 1469 + }, + { + "epoch": 1.510015408320493, + "grad_norm": 2.760437488555908, + "learning_rate": 2.7883517320137042e-06, + "loss": 1.3927, + "step": 1470 + }, + { + "epoch": 1.510015408320493, + "eval_loss": 1.8517959117889404, + "eval_runtime": 37.163, + "eval_samples_per_second": 26.908, + "eval_steps_per_second": 1.13, + "step": 1470 + }, + { + "epoch": 1.5110426296866974, + "grad_norm": 2.689023733139038, + "learning_rate": 2.7864484202512376e-06, + "loss": 1.387, + "step": 1471 + }, + { + "epoch": 1.5120698510529018, + "grad_norm": 2.876424551010132, + "learning_rate": 2.784545108488771e-06, + "loss": 1.4183, + "step": 1472 + }, + { + "epoch": 1.5130970724191064, + "grad_norm": 2.789368152618408, + "learning_rate": 2.7826417967263043e-06, + "loss": 1.4239, + "step": 1473 + }, + { + "epoch": 1.5141242937853108, + "grad_norm": 2.945103883743286, + "learning_rate": 2.7807384849638373e-06, + "loss": 1.458, + "step": 1474 + }, + { + "epoch": 1.5151515151515151, + "grad_norm": 2.694589614868164, + "learning_rate": 2.7788351732013706e-06, + "loss": 1.4145, + "step": 1475 + }, + { + "epoch": 1.5161787365177197, + "grad_norm": 2.7276177406311035, + "learning_rate": 2.776931861438904e-06, + "loss": 1.4063, + "step": 1476 + }, + { + "epoch": 1.5172059578839239, + "grad_norm": 2.694197177886963, + "learning_rate": 2.7750285496764374e-06, + "loss": 1.4125, + "step": 1477 + }, + { + "epoch": 1.5182331792501285, + "grad_norm": 2.832883834838867, + "learning_rate": 2.7731252379139707e-06, + "loss": 1.4176, + "step": 1478 + }, + { + "epoch": 1.5192604006163328, + "grad_norm": 2.67136549949646, + "learning_rate": 2.771221926151504e-06, + "loss": 1.3633, + "step": 1479 + }, + { + "epoch": 1.5202876219825372, + "grad_norm": 2.9618499279022217, + "learning_rate": 2.769318614389037e-06, + "loss": 1.4283, + "step": 1480 + }, + { + "epoch": 1.5202876219825372, + "eval_loss": 1.8486552238464355, + "eval_runtime": 37.1808, + "eval_samples_per_second": 26.896, + "eval_steps_per_second": 1.13, + "step": 1480 + }, + { + "epoch": 1.5213148433487418, + "grad_norm": 2.91403865814209, + "learning_rate": 2.7674153026265704e-06, + "loss": 1.4254, + "step": 1481 + }, + { + "epoch": 1.522342064714946, + "grad_norm": 2.9121804237365723, + "learning_rate": 2.7655119908641037e-06, + "loss": 1.4251, + "step": 1482 + }, + { + "epoch": 1.5233692860811505, + "grad_norm": 2.7484965324401855, + "learning_rate": 2.763608679101637e-06, + "loss": 1.4141, + "step": 1483 + }, + { + "epoch": 1.5243965074473549, + "grad_norm": 2.913203716278076, + "learning_rate": 2.7617053673391705e-06, + "loss": 1.4516, + "step": 1484 + }, + { + "epoch": 1.5254237288135593, + "grad_norm": 2.7844395637512207, + "learning_rate": 2.7598020555767034e-06, + "loss": 1.45, + "step": 1485 + }, + { + "epoch": 1.5264509501797638, + "grad_norm": 2.8250396251678467, + "learning_rate": 2.7578987438142368e-06, + "loss": 1.3746, + "step": 1486 + }, + { + "epoch": 1.5274781715459682, + "grad_norm": 2.749629497528076, + "learning_rate": 2.75599543205177e-06, + "loss": 1.433, + "step": 1487 + }, + { + "epoch": 1.5285053929121726, + "grad_norm": 2.7559521198272705, + "learning_rate": 2.7540921202893035e-06, + "loss": 1.452, + "step": 1488 + }, + { + "epoch": 1.529532614278377, + "grad_norm": 2.825148344039917, + "learning_rate": 2.752188808526837e-06, + "loss": 1.427, + "step": 1489 + }, + { + "epoch": 1.5305598356445813, + "grad_norm": 2.645327091217041, + "learning_rate": 2.75028549676437e-06, + "loss": 1.4089, + "step": 1490 + }, + { + "epoch": 1.5305598356445813, + "eval_loss": 1.8497828245162964, + "eval_runtime": 37.3756, + "eval_samples_per_second": 26.755, + "eval_steps_per_second": 1.124, + "step": 1490 + }, + { + "epoch": 1.531587057010786, + "grad_norm": 2.753156900405884, + "learning_rate": 2.7483821850019036e-06, + "loss": 1.4163, + "step": 1491 + }, + { + "epoch": 1.5326142783769903, + "grad_norm": 2.8842933177948, + "learning_rate": 2.746478873239437e-06, + "loss": 1.4041, + "step": 1492 + }, + { + "epoch": 1.5336414997431946, + "grad_norm": 2.7529265880584717, + "learning_rate": 2.7445755614769703e-06, + "loss": 1.41, + "step": 1493 + }, + { + "epoch": 1.5346687211093992, + "grad_norm": 2.6907176971435547, + "learning_rate": 2.7426722497145037e-06, + "loss": 1.3841, + "step": 1494 + }, + { + "epoch": 1.5356959424756034, + "grad_norm": 2.793073892593384, + "learning_rate": 2.740768937952037e-06, + "loss": 1.3885, + "step": 1495 + }, + { + "epoch": 1.536723163841808, + "grad_norm": 2.7895116806030273, + "learning_rate": 2.7388656261895704e-06, + "loss": 1.4065, + "step": 1496 + }, + { + "epoch": 1.5377503852080123, + "grad_norm": 2.903615713119507, + "learning_rate": 2.7369623144271033e-06, + "loss": 1.4812, + "step": 1497 + }, + { + "epoch": 1.5387776065742167, + "grad_norm": 2.7081339359283447, + "learning_rate": 2.7350590026646367e-06, + "loss": 1.403, + "step": 1498 + }, + { + "epoch": 1.5398048279404213, + "grad_norm": 2.75414776802063, + "learning_rate": 2.73315569090217e-06, + "loss": 1.3629, + "step": 1499 + }, + { + "epoch": 1.5408320493066254, + "grad_norm": 2.795982599258423, + "learning_rate": 2.7312523791397034e-06, + "loss": 1.4551, + "step": 1500 + }, + { + "epoch": 1.5408320493066254, + "eval_loss": 1.8510199785232544, + "eval_runtime": 37.3806, + "eval_samples_per_second": 26.752, + "eval_steps_per_second": 1.124, + "step": 1500 + }, + { + "epoch": 1.54185927067283, + "grad_norm": 2.894488573074341, + "learning_rate": 2.7293490673772368e-06, + "loss": 1.4794, + "step": 1501 + }, + { + "epoch": 1.5428864920390344, + "grad_norm": 2.7982053756713867, + "learning_rate": 2.7274457556147697e-06, + "loss": 1.4365, + "step": 1502 + }, + { + "epoch": 1.5439137134052388, + "grad_norm": 2.913367509841919, + "learning_rate": 2.725542443852303e-06, + "loss": 1.4132, + "step": 1503 + }, + { + "epoch": 1.5449409347714433, + "grad_norm": 2.6658358573913574, + "learning_rate": 2.7236391320898365e-06, + "loss": 1.4288, + "step": 1504 + }, + { + "epoch": 1.5459681561376477, + "grad_norm": 2.710700035095215, + "learning_rate": 2.72173582032737e-06, + "loss": 1.3936, + "step": 1505 + }, + { + "epoch": 1.546995377503852, + "grad_norm": 2.7881481647491455, + "learning_rate": 2.719832508564903e-06, + "loss": 1.423, + "step": 1506 + }, + { + "epoch": 1.5480225988700564, + "grad_norm": 2.7621042728424072, + "learning_rate": 2.717929196802436e-06, + "loss": 1.4218, + "step": 1507 + }, + { + "epoch": 1.5490498202362608, + "grad_norm": 2.7504117488861084, + "learning_rate": 2.7160258850399695e-06, + "loss": 1.3918, + "step": 1508 + }, + { + "epoch": 1.5500770416024654, + "grad_norm": 2.719702959060669, + "learning_rate": 2.714122573277503e-06, + "loss": 1.4154, + "step": 1509 + }, + { + "epoch": 1.5511042629686698, + "grad_norm": 2.6698062419891357, + "learning_rate": 2.712219261515036e-06, + "loss": 1.4084, + "step": 1510 + }, + { + "epoch": 1.5511042629686698, + "eval_loss": 1.8468451499938965, + "eval_runtime": 36.9179, + "eval_samples_per_second": 27.087, + "eval_steps_per_second": 1.138, + "step": 1510 + }, + { + "epoch": 1.5521314843348741, + "grad_norm": 2.816978693008423, + "learning_rate": 2.71031594975257e-06, + "loss": 1.4195, + "step": 1511 + }, + { + "epoch": 1.5531587057010787, + "grad_norm": 2.8047056198120117, + "learning_rate": 2.7084126379901034e-06, + "loss": 1.4395, + "step": 1512 + }, + { + "epoch": 1.5541859270672829, + "grad_norm": 2.8461363315582275, + "learning_rate": 2.7065093262276367e-06, + "loss": 1.4472, + "step": 1513 + }, + { + "epoch": 1.5552131484334875, + "grad_norm": 2.712890863418579, + "learning_rate": 2.7046060144651697e-06, + "loss": 1.4538, + "step": 1514 + }, + { + "epoch": 1.5562403697996918, + "grad_norm": 2.844165802001953, + "learning_rate": 2.702702702702703e-06, + "loss": 1.4719, + "step": 1515 + }, + { + "epoch": 1.5572675911658962, + "grad_norm": 2.629138231277466, + "learning_rate": 2.7007993909402364e-06, + "loss": 1.3441, + "step": 1516 + }, + { + "epoch": 1.5582948125321008, + "grad_norm": 2.7072019577026367, + "learning_rate": 2.6988960791777697e-06, + "loss": 1.4692, + "step": 1517 + }, + { + "epoch": 1.559322033898305, + "grad_norm": 2.739567279815674, + "learning_rate": 2.696992767415303e-06, + "loss": 1.4215, + "step": 1518 + }, + { + "epoch": 1.5603492552645095, + "grad_norm": 2.7846791744232178, + "learning_rate": 2.695089455652836e-06, + "loss": 1.4151, + "step": 1519 + }, + { + "epoch": 1.5613764766307139, + "grad_norm": 2.7417876720428467, + "learning_rate": 2.6931861438903694e-06, + "loss": 1.3797, + "step": 1520 + }, + { + "epoch": 1.5613764766307139, + "eval_loss": 1.847011923789978, + "eval_runtime": 37.0386, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 1520 + }, + { + "epoch": 1.5624036979969183, + "grad_norm": 2.743271827697754, + "learning_rate": 2.6912828321279028e-06, + "loss": 1.4372, + "step": 1521 + }, + { + "epoch": 1.5634309193631228, + "grad_norm": 2.8119285106658936, + "learning_rate": 2.689379520365436e-06, + "loss": 1.4075, + "step": 1522 + }, + { + "epoch": 1.5644581407293272, + "grad_norm": 2.7443952560424805, + "learning_rate": 2.6874762086029695e-06, + "loss": 1.4442, + "step": 1523 + }, + { + "epoch": 1.5654853620955316, + "grad_norm": 2.7869346141815186, + "learning_rate": 2.6855728968405024e-06, + "loss": 1.3948, + "step": 1524 + }, + { + "epoch": 1.5665125834617362, + "grad_norm": 2.750185251235962, + "learning_rate": 2.683669585078036e-06, + "loss": 1.4313, + "step": 1525 + }, + { + "epoch": 1.5675398048279403, + "grad_norm": 2.787389039993286, + "learning_rate": 2.681766273315569e-06, + "loss": 1.3748, + "step": 1526 + }, + { + "epoch": 1.568567026194145, + "grad_norm": 2.6675543785095215, + "learning_rate": 2.6798629615531025e-06, + "loss": 1.4089, + "step": 1527 + }, + { + "epoch": 1.5695942475603493, + "grad_norm": 2.6468663215637207, + "learning_rate": 2.677959649790636e-06, + "loss": 1.4148, + "step": 1528 + }, + { + "epoch": 1.5706214689265536, + "grad_norm": 2.7241196632385254, + "learning_rate": 2.676056338028169e-06, + "loss": 1.4498, + "step": 1529 + }, + { + "epoch": 1.5716486902927582, + "grad_norm": 2.705486536026001, + "learning_rate": 2.674153026265702e-06, + "loss": 1.4324, + "step": 1530 + }, + { + "epoch": 1.5716486902927582, + "eval_loss": 1.8467438220977783, + "eval_runtime": 37.0576, + "eval_samples_per_second": 26.985, + "eval_steps_per_second": 1.133, + "step": 1530 + }, + { + "epoch": 1.5726759116589624, + "grad_norm": 2.805429458618164, + "learning_rate": 2.672249714503236e-06, + "loss": 1.4475, + "step": 1531 + }, + { + "epoch": 1.573703133025167, + "grad_norm": 2.7074034214019775, + "learning_rate": 2.6703464027407693e-06, + "loss": 1.3997, + "step": 1532 + }, + { + "epoch": 1.5747303543913713, + "grad_norm": 2.737571954727173, + "learning_rate": 2.6684430909783027e-06, + "loss": 1.4238, + "step": 1533 + }, + { + "epoch": 1.5757575757575757, + "grad_norm": 2.703036069869995, + "learning_rate": 2.666539779215836e-06, + "loss": 1.4107, + "step": 1534 + }, + { + "epoch": 1.5767847971237803, + "grad_norm": 2.7987451553344727, + "learning_rate": 2.6646364674533694e-06, + "loss": 1.3927, + "step": 1535 + }, + { + "epoch": 1.5778120184899846, + "grad_norm": 2.622497081756592, + "learning_rate": 2.6627331556909024e-06, + "loss": 1.3215, + "step": 1536 + }, + { + "epoch": 1.578839239856189, + "grad_norm": 2.5890607833862305, + "learning_rate": 2.6608298439284357e-06, + "loss": 1.3994, + "step": 1537 + }, + { + "epoch": 1.5798664612223934, + "grad_norm": 2.6438405513763428, + "learning_rate": 2.658926532165969e-06, + "loss": 1.4042, + "step": 1538 + }, + { + "epoch": 1.5808936825885977, + "grad_norm": 2.690793752670288, + "learning_rate": 2.6570232204035024e-06, + "loss": 1.3673, + "step": 1539 + }, + { + "epoch": 1.5819209039548023, + "grad_norm": 2.7485456466674805, + "learning_rate": 2.655119908641036e-06, + "loss": 1.3979, + "step": 1540 + }, + { + "epoch": 1.5819209039548023, + "eval_loss": 1.852581262588501, + "eval_runtime": 37.0688, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1540 + }, + { + "epoch": 1.5829481253210067, + "grad_norm": 2.8531415462493896, + "learning_rate": 2.6532165968785687e-06, + "loss": 1.4094, + "step": 1541 + }, + { + "epoch": 1.583975346687211, + "grad_norm": 2.638291358947754, + "learning_rate": 2.651313285116102e-06, + "loss": 1.3708, + "step": 1542 + }, + { + "epoch": 1.5850025680534157, + "grad_norm": 2.717550754547119, + "learning_rate": 2.6494099733536355e-06, + "loss": 1.3821, + "step": 1543 + }, + { + "epoch": 1.5860297894196198, + "grad_norm": 2.91959810256958, + "learning_rate": 2.647506661591169e-06, + "loss": 1.5532, + "step": 1544 + }, + { + "epoch": 1.5870570107858244, + "grad_norm": 2.7281882762908936, + "learning_rate": 2.645603349828702e-06, + "loss": 1.4361, + "step": 1545 + }, + { + "epoch": 1.5880842321520288, + "grad_norm": 2.732283592224121, + "learning_rate": 2.643700038066235e-06, + "loss": 1.3669, + "step": 1546 + }, + { + "epoch": 1.5891114535182331, + "grad_norm": 2.731055498123169, + "learning_rate": 2.6417967263037685e-06, + "loss": 1.4223, + "step": 1547 + }, + { + "epoch": 1.5901386748844377, + "grad_norm": 2.831334352493286, + "learning_rate": 2.639893414541302e-06, + "loss": 1.4454, + "step": 1548 + }, + { + "epoch": 1.5911658962506419, + "grad_norm": 2.920548677444458, + "learning_rate": 2.6379901027788352e-06, + "loss": 1.4686, + "step": 1549 + }, + { + "epoch": 1.5921931176168465, + "grad_norm": 2.6671719551086426, + "learning_rate": 2.6360867910163686e-06, + "loss": 1.398, + "step": 1550 + }, + { + "epoch": 1.5921931176168465, + "eval_loss": 1.857794165611267, + "eval_runtime": 37.0817, + "eval_samples_per_second": 26.967, + "eval_steps_per_second": 1.133, + "step": 1550 + }, + { + "epoch": 1.5932203389830508, + "grad_norm": 2.685175657272339, + "learning_rate": 2.6341834792539024e-06, + "loss": 1.4176, + "step": 1551 + }, + { + "epoch": 1.5942475603492552, + "grad_norm": 2.7412822246551514, + "learning_rate": 2.6322801674914357e-06, + "loss": 1.4195, + "step": 1552 + }, + { + "epoch": 1.5952747817154598, + "grad_norm": 2.7461724281311035, + "learning_rate": 2.6303768557289687e-06, + "loss": 1.4081, + "step": 1553 + }, + { + "epoch": 1.5963020030816641, + "grad_norm": 2.6494853496551514, + "learning_rate": 2.628473543966502e-06, + "loss": 1.3759, + "step": 1554 + }, + { + "epoch": 1.5973292244478685, + "grad_norm": 2.736752986907959, + "learning_rate": 2.6265702322040354e-06, + "loss": 1.4415, + "step": 1555 + }, + { + "epoch": 1.5983564458140729, + "grad_norm": 2.733452796936035, + "learning_rate": 2.6246669204415688e-06, + "loss": 1.3867, + "step": 1556 + }, + { + "epoch": 1.5993836671802772, + "grad_norm": 2.692986249923706, + "learning_rate": 2.622763608679102e-06, + "loss": 1.3811, + "step": 1557 + }, + { + "epoch": 1.6004108885464818, + "grad_norm": 2.580052614212036, + "learning_rate": 2.620860296916635e-06, + "loss": 1.399, + "step": 1558 + }, + { + "epoch": 1.6014381099126862, + "grad_norm": 2.651503086090088, + "learning_rate": 2.6189569851541684e-06, + "loss": 1.34, + "step": 1559 + }, + { + "epoch": 1.6024653312788906, + "grad_norm": 2.8157193660736084, + "learning_rate": 2.617053673391702e-06, + "loss": 1.4437, + "step": 1560 + }, + { + "epoch": 1.6024653312788906, + "eval_loss": 1.8588496446609497, + "eval_runtime": 37.0786, + "eval_samples_per_second": 26.97, + "eval_steps_per_second": 1.133, + "step": 1560 + }, + { + "epoch": 1.6034925526450952, + "grad_norm": 2.893589973449707, + "learning_rate": 2.615150361629235e-06, + "loss": 1.4775, + "step": 1561 + }, + { + "epoch": 1.6045197740112993, + "grad_norm": 2.719688653945923, + "learning_rate": 2.6132470498667685e-06, + "loss": 1.432, + "step": 1562 + }, + { + "epoch": 1.605546995377504, + "grad_norm": 2.5626275539398193, + "learning_rate": 2.6113437381043015e-06, + "loss": 1.4118, + "step": 1563 + }, + { + "epoch": 1.6065742167437083, + "grad_norm": 2.6879632472991943, + "learning_rate": 2.609440426341835e-06, + "loss": 1.398, + "step": 1564 + }, + { + "epoch": 1.6076014381099126, + "grad_norm": 2.890155792236328, + "learning_rate": 2.607537114579368e-06, + "loss": 1.4336, + "step": 1565 + }, + { + "epoch": 1.6086286594761172, + "grad_norm": 2.8333349227905273, + "learning_rate": 2.6056338028169015e-06, + "loss": 1.4148, + "step": 1566 + }, + { + "epoch": 1.6096558808423214, + "grad_norm": 2.729215383529663, + "learning_rate": 2.603730491054435e-06, + "loss": 1.3831, + "step": 1567 + }, + { + "epoch": 1.610683102208526, + "grad_norm": 2.800276279449463, + "learning_rate": 2.601827179291968e-06, + "loss": 1.4315, + "step": 1568 + }, + { + "epoch": 1.6117103235747303, + "grad_norm": 2.7900125980377197, + "learning_rate": 2.599923867529501e-06, + "loss": 1.4257, + "step": 1569 + }, + { + "epoch": 1.6127375449409347, + "grad_norm": 2.7753336429595947, + "learning_rate": 2.5980205557670346e-06, + "loss": 1.4089, + "step": 1570 + }, + { + "epoch": 1.6127375449409347, + "eval_loss": 1.8533762693405151, + "eval_runtime": 37.0837, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1570 + }, + { + "epoch": 1.6137647663071393, + "grad_norm": 2.9164156913757324, + "learning_rate": 2.5961172440045684e-06, + "loss": 1.4467, + "step": 1571 + }, + { + "epoch": 1.6147919876733436, + "grad_norm": 2.747676134109497, + "learning_rate": 2.5942139322421017e-06, + "loss": 1.4521, + "step": 1572 + }, + { + "epoch": 1.615819209039548, + "grad_norm": 2.6693432331085205, + "learning_rate": 2.592310620479635e-06, + "loss": 1.3982, + "step": 1573 + }, + { + "epoch": 1.6168464304057526, + "grad_norm": 2.5721216201782227, + "learning_rate": 2.5904073087171684e-06, + "loss": 1.3657, + "step": 1574 + }, + { + "epoch": 1.6178736517719567, + "grad_norm": 2.706143617630005, + "learning_rate": 2.5885039969547014e-06, + "loss": 1.4315, + "step": 1575 + }, + { + "epoch": 1.6189008731381613, + "grad_norm": 2.59006404876709, + "learning_rate": 2.5866006851922347e-06, + "loss": 1.372, + "step": 1576 + }, + { + "epoch": 1.6199280945043657, + "grad_norm": 2.7117512226104736, + "learning_rate": 2.584697373429768e-06, + "loss": 1.3834, + "step": 1577 + }, + { + "epoch": 1.62095531587057, + "grad_norm": 2.718029737472534, + "learning_rate": 2.5827940616673015e-06, + "loss": 1.3978, + "step": 1578 + }, + { + "epoch": 1.6219825372367747, + "grad_norm": 2.7660539150238037, + "learning_rate": 2.580890749904835e-06, + "loss": 1.4505, + "step": 1579 + }, + { + "epoch": 1.6230097586029788, + "grad_norm": 2.7671236991882324, + "learning_rate": 2.5789874381423678e-06, + "loss": 1.3951, + "step": 1580 + }, + { + "epoch": 1.6230097586029788, + "eval_loss": 1.8412485122680664, + "eval_runtime": 37.1162, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 1580 + }, + { + "epoch": 1.6240369799691834, + "grad_norm": 2.6504807472229004, + "learning_rate": 2.577084126379901e-06, + "loss": 1.3957, + "step": 1581 + }, + { + "epoch": 1.6250642013353878, + "grad_norm": 2.688166618347168, + "learning_rate": 2.5751808146174345e-06, + "loss": 1.3686, + "step": 1582 + }, + { + "epoch": 1.6260914227015921, + "grad_norm": 2.7034637928009033, + "learning_rate": 2.573277502854968e-06, + "loss": 1.4201, + "step": 1583 + }, + { + "epoch": 1.6271186440677967, + "grad_norm": 2.6097683906555176, + "learning_rate": 2.5713741910925012e-06, + "loss": 1.3881, + "step": 1584 + }, + { + "epoch": 1.6281458654340009, + "grad_norm": 2.6197433471679688, + "learning_rate": 2.569470879330034e-06, + "loss": 1.3962, + "step": 1585 + }, + { + "epoch": 1.6291730868002054, + "grad_norm": 3.0436367988586426, + "learning_rate": 2.5675675675675675e-06, + "loss": 1.4714, + "step": 1586 + }, + { + "epoch": 1.6302003081664098, + "grad_norm": 2.708744764328003, + "learning_rate": 2.565664255805101e-06, + "loss": 1.4007, + "step": 1587 + }, + { + "epoch": 1.6312275295326142, + "grad_norm": 2.708279848098755, + "learning_rate": 2.5637609440426343e-06, + "loss": 1.4284, + "step": 1588 + }, + { + "epoch": 1.6322547508988188, + "grad_norm": 2.7955429553985596, + "learning_rate": 2.5618576322801676e-06, + "loss": 1.3736, + "step": 1589 + }, + { + "epoch": 1.6332819722650231, + "grad_norm": 2.890211820602417, + "learning_rate": 2.5599543205177006e-06, + "loss": 1.4671, + "step": 1590 + }, + { + "epoch": 1.6332819722650231, + "eval_loss": 1.8467649221420288, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 1590 + }, + { + "epoch": 1.6343091936312275, + "grad_norm": 2.698066234588623, + "learning_rate": 2.558051008755234e-06, + "loss": 1.4229, + "step": 1591 + }, + { + "epoch": 1.635336414997432, + "grad_norm": 2.8393352031707764, + "learning_rate": 2.5561476969927677e-06, + "loss": 1.469, + "step": 1592 + }, + { + "epoch": 1.6363636363636362, + "grad_norm": 2.589914321899414, + "learning_rate": 2.554244385230301e-06, + "loss": 1.3853, + "step": 1593 + }, + { + "epoch": 1.6373908577298408, + "grad_norm": 2.630443572998047, + "learning_rate": 2.5523410734678344e-06, + "loss": 1.4176, + "step": 1594 + }, + { + "epoch": 1.6384180790960452, + "grad_norm": 2.784212112426758, + "learning_rate": 2.550437761705368e-06, + "loss": 1.344, + "step": 1595 + }, + { + "epoch": 1.6394453004622496, + "grad_norm": 2.736009359359741, + "learning_rate": 2.548534449942901e-06, + "loss": 1.3867, + "step": 1596 + }, + { + "epoch": 1.6404725218284542, + "grad_norm": 2.9134042263031006, + "learning_rate": 2.546631138180434e-06, + "loss": 1.4214, + "step": 1597 + }, + { + "epoch": 1.6414997431946583, + "grad_norm": 2.995575428009033, + "learning_rate": 2.5447278264179675e-06, + "loss": 1.409, + "step": 1598 + }, + { + "epoch": 1.642526964560863, + "grad_norm": 2.744499683380127, + "learning_rate": 2.542824514655501e-06, + "loss": 1.3727, + "step": 1599 + }, + { + "epoch": 1.6435541859270673, + "grad_norm": 2.835494041442871, + "learning_rate": 2.540921202893034e-06, + "loss": 1.4636, + "step": 1600 + }, + { + "epoch": 1.6435541859270673, + "eval_loss": 1.8454105854034424, + "eval_runtime": 37.4381, + "eval_samples_per_second": 26.711, + "eval_steps_per_second": 1.122, + "step": 1600 + }, + { + "epoch": 1.6445814072932716, + "grad_norm": 2.817537307739258, + "learning_rate": 2.5390178911305675e-06, + "loss": 1.4416, + "step": 1601 + }, + { + "epoch": 1.6456086286594762, + "grad_norm": 2.868093252182007, + "learning_rate": 2.5371145793681005e-06, + "loss": 1.4187, + "step": 1602 + }, + { + "epoch": 1.6466358500256806, + "grad_norm": 2.861865758895874, + "learning_rate": 2.535211267605634e-06, + "loss": 1.4067, + "step": 1603 + }, + { + "epoch": 1.647663071391885, + "grad_norm": 2.7015087604522705, + "learning_rate": 2.533307955843167e-06, + "loss": 1.3528, + "step": 1604 + }, + { + "epoch": 1.6486902927580893, + "grad_norm": 2.7585439682006836, + "learning_rate": 2.5314046440807006e-06, + "loss": 1.4157, + "step": 1605 + }, + { + "epoch": 1.6497175141242937, + "grad_norm": 2.853614330291748, + "learning_rate": 2.529501332318234e-06, + "loss": 1.3932, + "step": 1606 + }, + { + "epoch": 1.6507447354904983, + "grad_norm": 2.9736239910125732, + "learning_rate": 2.5275980205557673e-06, + "loss": 1.4511, + "step": 1607 + }, + { + "epoch": 1.6517719568567026, + "grad_norm": 2.955389976501465, + "learning_rate": 2.5256947087933002e-06, + "loss": 1.4777, + "step": 1608 + }, + { + "epoch": 1.652799178222907, + "grad_norm": 2.9508774280548096, + "learning_rate": 2.5237913970308336e-06, + "loss": 1.4523, + "step": 1609 + }, + { + "epoch": 1.6538263995891116, + "grad_norm": 2.88675856590271, + "learning_rate": 2.521888085268367e-06, + "loss": 1.4064, + "step": 1610 + }, + { + "epoch": 1.6538263995891116, + "eval_loss": 1.8569127321243286, + "eval_runtime": 36.9819, + "eval_samples_per_second": 27.04, + "eval_steps_per_second": 1.136, + "step": 1610 + }, + { + "epoch": 1.6548536209553157, + "grad_norm": 2.8439440727233887, + "learning_rate": 2.5199847735059003e-06, + "loss": 1.4591, + "step": 1611 + }, + { + "epoch": 1.6558808423215203, + "grad_norm": 2.9314420223236084, + "learning_rate": 2.518081461743434e-06, + "loss": 1.39, + "step": 1612 + }, + { + "epoch": 1.6569080636877247, + "grad_norm": 2.6517245769500732, + "learning_rate": 2.5161781499809675e-06, + "loss": 1.4234, + "step": 1613 + }, + { + "epoch": 1.657935285053929, + "grad_norm": 2.5690908432006836, + "learning_rate": 2.5142748382185004e-06, + "loss": 1.3762, + "step": 1614 + }, + { + "epoch": 1.6589625064201337, + "grad_norm": 2.808892011642456, + "learning_rate": 2.5123715264560338e-06, + "loss": 1.3727, + "step": 1615 + }, + { + "epoch": 1.6599897277863378, + "grad_norm": 2.777846336364746, + "learning_rate": 2.510468214693567e-06, + "loss": 1.3941, + "step": 1616 + }, + { + "epoch": 1.6610169491525424, + "grad_norm": 2.8298423290252686, + "learning_rate": 2.5085649029311005e-06, + "loss": 1.3455, + "step": 1617 + }, + { + "epoch": 1.6620441705187468, + "grad_norm": 2.6464784145355225, + "learning_rate": 2.506661591168634e-06, + "loss": 1.3061, + "step": 1618 + }, + { + "epoch": 1.6630713918849511, + "grad_norm": 2.7149951457977295, + "learning_rate": 2.504758279406167e-06, + "loss": 1.3995, + "step": 1619 + }, + { + "epoch": 1.6640986132511557, + "grad_norm": 3.0328569412231445, + "learning_rate": 2.5028549676437e-06, + "loss": 1.4343, + "step": 1620 + }, + { + "epoch": 1.6640986132511557, + "eval_loss": 1.8629075288772583, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 1620 + }, + { + "epoch": 1.66512583461736, + "grad_norm": 2.8838441371917725, + "learning_rate": 2.5009516558812335e-06, + "loss": 1.3817, + "step": 1621 + }, + { + "epoch": 1.6661530559835644, + "grad_norm": 2.843242645263672, + "learning_rate": 2.499048344118767e-06, + "loss": 1.4322, + "step": 1622 + }, + { + "epoch": 1.667180277349769, + "grad_norm": 2.642946720123291, + "learning_rate": 2.4971450323563002e-06, + "loss": 1.4702, + "step": 1623 + }, + { + "epoch": 1.6682074987159732, + "grad_norm": 2.666194200515747, + "learning_rate": 2.4952417205938336e-06, + "loss": 1.3808, + "step": 1624 + }, + { + "epoch": 1.6692347200821778, + "grad_norm": 2.7515366077423096, + "learning_rate": 2.4933384088313666e-06, + "loss": 1.3892, + "step": 1625 + }, + { + "epoch": 1.6702619414483821, + "grad_norm": 2.7051661014556885, + "learning_rate": 2.4914350970689e-06, + "loss": 1.3809, + "step": 1626 + }, + { + "epoch": 1.6712891628145865, + "grad_norm": 2.7311975955963135, + "learning_rate": 2.4895317853064337e-06, + "loss": 1.3952, + "step": 1627 + }, + { + "epoch": 1.672316384180791, + "grad_norm": 2.8365893363952637, + "learning_rate": 2.4876284735439666e-06, + "loss": 1.4505, + "step": 1628 + }, + { + "epoch": 1.6733436055469952, + "grad_norm": 2.644507884979248, + "learning_rate": 2.4857251617815e-06, + "loss": 1.4114, + "step": 1629 + }, + { + "epoch": 1.6743708269131998, + "grad_norm": 2.7952826023101807, + "learning_rate": 2.4838218500190334e-06, + "loss": 1.4223, + "step": 1630 + }, + { + "epoch": 1.6743708269131998, + "eval_loss": 1.8577097654342651, + "eval_runtime": 37.1645, + "eval_samples_per_second": 26.907, + "eval_steps_per_second": 1.13, + "step": 1630 + }, + { + "epoch": 1.6753980482794042, + "grad_norm": 2.787752866744995, + "learning_rate": 2.4819185382565667e-06, + "loss": 1.4022, + "step": 1631 + }, + { + "epoch": 1.6764252696456086, + "grad_norm": 2.6963443756103516, + "learning_rate": 2.4800152264941e-06, + "loss": 1.3974, + "step": 1632 + }, + { + "epoch": 1.6774524910118132, + "grad_norm": 2.6753835678100586, + "learning_rate": 2.478111914731633e-06, + "loss": 1.3639, + "step": 1633 + }, + { + "epoch": 1.6784797123780173, + "grad_norm": 2.7086808681488037, + "learning_rate": 2.4762086029691664e-06, + "loss": 1.3987, + "step": 1634 + }, + { + "epoch": 1.6795069337442219, + "grad_norm": 3.2228872776031494, + "learning_rate": 2.4743052912066998e-06, + "loss": 1.5202, + "step": 1635 + }, + { + "epoch": 1.6805341551104263, + "grad_norm": 2.8254599571228027, + "learning_rate": 2.472401979444233e-06, + "loss": 1.3781, + "step": 1636 + }, + { + "epoch": 1.6815613764766306, + "grad_norm": 2.9406354427337646, + "learning_rate": 2.4704986676817665e-06, + "loss": 1.4686, + "step": 1637 + }, + { + "epoch": 1.6825885978428352, + "grad_norm": 2.8287105560302734, + "learning_rate": 2.4685953559193e-06, + "loss": 1.4026, + "step": 1638 + }, + { + "epoch": 1.6836158192090396, + "grad_norm": 2.923633337020874, + "learning_rate": 2.466692044156833e-06, + "loss": 1.4493, + "step": 1639 + }, + { + "epoch": 1.684643040575244, + "grad_norm": 2.9239513874053955, + "learning_rate": 2.4647887323943666e-06, + "loss": 1.3878, + "step": 1640 + }, + { + "epoch": 1.684643040575244, + "eval_loss": 1.853607416152954, + "eval_runtime": 37.1661, + "eval_samples_per_second": 26.906, + "eval_steps_per_second": 1.13, + "step": 1640 + }, + { + "epoch": 1.6856702619414485, + "grad_norm": 2.699998617172241, + "learning_rate": 2.4628854206319e-06, + "loss": 1.3637, + "step": 1641 + }, + { + "epoch": 1.6866974833076527, + "grad_norm": 2.901700258255005, + "learning_rate": 2.460982108869433e-06, + "loss": 1.3554, + "step": 1642 + }, + { + "epoch": 1.6877247046738573, + "grad_norm": 3.011157512664795, + "learning_rate": 2.4590787971069662e-06, + "loss": 1.4487, + "step": 1643 + }, + { + "epoch": 1.6887519260400616, + "grad_norm": 2.672830581665039, + "learning_rate": 2.4571754853444996e-06, + "loss": 1.3906, + "step": 1644 + }, + { + "epoch": 1.689779147406266, + "grad_norm": 2.7201497554779053, + "learning_rate": 2.455272173582033e-06, + "loss": 1.3747, + "step": 1645 + }, + { + "epoch": 1.6908063687724706, + "grad_norm": 2.979395627975464, + "learning_rate": 2.4533688618195663e-06, + "loss": 1.3895, + "step": 1646 + }, + { + "epoch": 1.6918335901386747, + "grad_norm": 2.8048512935638428, + "learning_rate": 2.4514655500570997e-06, + "loss": 1.3271, + "step": 1647 + }, + { + "epoch": 1.6928608115048793, + "grad_norm": 2.6457700729370117, + "learning_rate": 2.449562238294633e-06, + "loss": 1.3679, + "step": 1648 + }, + { + "epoch": 1.6938880328710837, + "grad_norm": 2.732454299926758, + "learning_rate": 2.4476589265321664e-06, + "loss": 1.3332, + "step": 1649 + }, + { + "epoch": 1.694915254237288, + "grad_norm": 3.190871238708496, + "learning_rate": 2.4457556147696993e-06, + "loss": 1.4437, + "step": 1650 + }, + { + "epoch": 1.694915254237288, + "eval_loss": 1.8642219305038452, + "eval_runtime": 37.1865, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 1650 + }, + { + "epoch": 1.6959424756034926, + "grad_norm": 2.85784912109375, + "learning_rate": 2.4438523030072327e-06, + "loss": 1.4121, + "step": 1651 + }, + { + "epoch": 1.696969696969697, + "grad_norm": 2.766545534133911, + "learning_rate": 2.441948991244766e-06, + "loss": 1.4231, + "step": 1652 + }, + { + "epoch": 1.6979969183359014, + "grad_norm": 2.8962197303771973, + "learning_rate": 2.4400456794822994e-06, + "loss": 1.4146, + "step": 1653 + }, + { + "epoch": 1.6990241397021058, + "grad_norm": 2.907231569290161, + "learning_rate": 2.438142367719833e-06, + "loss": 1.4375, + "step": 1654 + }, + { + "epoch": 1.7000513610683101, + "grad_norm": 2.8746557235717773, + "learning_rate": 2.4362390559573657e-06, + "loss": 1.4002, + "step": 1655 + }, + { + "epoch": 1.7010785824345147, + "grad_norm": 2.741013765335083, + "learning_rate": 2.434335744194899e-06, + "loss": 1.3865, + "step": 1656 + }, + { + "epoch": 1.702105803800719, + "grad_norm": 3.118908643722534, + "learning_rate": 2.432432432432433e-06, + "loss": 1.4412, + "step": 1657 + }, + { + "epoch": 1.7031330251669234, + "grad_norm": 2.789179563522339, + "learning_rate": 2.4305291206699662e-06, + "loss": 1.3446, + "step": 1658 + }, + { + "epoch": 1.704160246533128, + "grad_norm": 2.9241371154785156, + "learning_rate": 2.428625808907499e-06, + "loss": 1.4263, + "step": 1659 + }, + { + "epoch": 1.7051874678993322, + "grad_norm": 2.8632888793945312, + "learning_rate": 2.4267224971450325e-06, + "loss": 1.4413, + "step": 1660 + }, + { + "epoch": 1.7051874678993322, + "eval_loss": 1.85648512840271, + "eval_runtime": 37.199, + "eval_samples_per_second": 26.882, + "eval_steps_per_second": 1.129, + "step": 1660 + }, + { + "epoch": 1.7062146892655368, + "grad_norm": 2.948183059692383, + "learning_rate": 2.424819185382566e-06, + "loss": 1.4058, + "step": 1661 + }, + { + "epoch": 1.7072419106317411, + "grad_norm": 3.04127836227417, + "learning_rate": 2.4229158736200993e-06, + "loss": 1.3582, + "step": 1662 + }, + { + "epoch": 1.7082691319979455, + "grad_norm": 2.6956393718719482, + "learning_rate": 2.4210125618576326e-06, + "loss": 1.3705, + "step": 1663 + }, + { + "epoch": 1.70929635336415, + "grad_norm": 2.7690250873565674, + "learning_rate": 2.4191092500951656e-06, + "loss": 1.3869, + "step": 1664 + }, + { + "epoch": 1.7103235747303542, + "grad_norm": 2.851231813430786, + "learning_rate": 2.417205938332699e-06, + "loss": 1.4198, + "step": 1665 + }, + { + "epoch": 1.7113507960965588, + "grad_norm": 2.9241318702697754, + "learning_rate": 2.4153026265702323e-06, + "loss": 1.3441, + "step": 1666 + }, + { + "epoch": 1.7123780174627632, + "grad_norm": 2.7888636589050293, + "learning_rate": 2.4133993148077657e-06, + "loss": 1.3765, + "step": 1667 + }, + { + "epoch": 1.7134052388289676, + "grad_norm": 2.891366481781006, + "learning_rate": 2.411496003045299e-06, + "loss": 1.4393, + "step": 1668 + }, + { + "epoch": 1.7144324601951721, + "grad_norm": 2.8162693977355957, + "learning_rate": 2.4095926912828324e-06, + "loss": 1.3811, + "step": 1669 + }, + { + "epoch": 1.7154596815613765, + "grad_norm": 2.6741931438446045, + "learning_rate": 2.4076893795203658e-06, + "loss": 1.404, + "step": 1670 + }, + { + "epoch": 1.7154596815613765, + "eval_loss": 1.8440946340560913, + "eval_runtime": 37.1273, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1670 + }, + { + "epoch": 1.7164869029275809, + "grad_norm": 2.834951400756836, + "learning_rate": 2.405786067757899e-06, + "loss": 1.377, + "step": 1671 + }, + { + "epoch": 1.7175141242937855, + "grad_norm": 2.846113681793213, + "learning_rate": 2.403882755995432e-06, + "loss": 1.4689, + "step": 1672 + }, + { + "epoch": 1.7185413456599896, + "grad_norm": 2.6862854957580566, + "learning_rate": 2.4019794442329654e-06, + "loss": 1.3979, + "step": 1673 + }, + { + "epoch": 1.7195685670261942, + "grad_norm": 2.7214972972869873, + "learning_rate": 2.4000761324704988e-06, + "loss": 1.3738, + "step": 1674 + }, + { + "epoch": 1.7205957883923986, + "grad_norm": 2.7037723064422607, + "learning_rate": 2.398172820708032e-06, + "loss": 1.4184, + "step": 1675 + }, + { + "epoch": 1.721623009758603, + "grad_norm": 2.7854912281036377, + "learning_rate": 2.3962695089455655e-06, + "loss": 1.3854, + "step": 1676 + }, + { + "epoch": 1.7226502311248075, + "grad_norm": 2.640059471130371, + "learning_rate": 2.3943661971830984e-06, + "loss": 1.3656, + "step": 1677 + }, + { + "epoch": 1.7236774524910117, + "grad_norm": 2.6955630779266357, + "learning_rate": 2.3924628854206322e-06, + "loss": 1.3764, + "step": 1678 + }, + { + "epoch": 1.7247046738572163, + "grad_norm": 3.018975257873535, + "learning_rate": 2.3905595736581656e-06, + "loss": 1.3922, + "step": 1679 + }, + { + "epoch": 1.7257318952234206, + "grad_norm": 2.7709803581237793, + "learning_rate": 2.388656261895699e-06, + "loss": 1.4288, + "step": 1680 + }, + { + "epoch": 1.7257318952234206, + "eval_loss": 1.8474006652832031, + "eval_runtime": 37.1233, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 1680 + }, + { + "epoch": 1.726759116589625, + "grad_norm": 2.8898143768310547, + "learning_rate": 2.386752950133232e-06, + "loss": 1.4143, + "step": 1681 + }, + { + "epoch": 1.7277863379558296, + "grad_norm": 2.9092485904693604, + "learning_rate": 2.3848496383707653e-06, + "loss": 1.4097, + "step": 1682 + }, + { + "epoch": 1.7288135593220337, + "grad_norm": 2.886662483215332, + "learning_rate": 2.3829463266082986e-06, + "loss": 1.3968, + "step": 1683 + }, + { + "epoch": 1.7298407806882383, + "grad_norm": 2.7398223876953125, + "learning_rate": 2.381043014845832e-06, + "loss": 1.4198, + "step": 1684 + }, + { + "epoch": 1.7308680020544427, + "grad_norm": 2.626732587814331, + "learning_rate": 2.3791397030833653e-06, + "loss": 1.4116, + "step": 1685 + }, + { + "epoch": 1.731895223420647, + "grad_norm": 2.6434433460235596, + "learning_rate": 2.3772363913208983e-06, + "loss": 1.3857, + "step": 1686 + }, + { + "epoch": 1.7329224447868516, + "grad_norm": 2.83876633644104, + "learning_rate": 2.3753330795584316e-06, + "loss": 1.3672, + "step": 1687 + }, + { + "epoch": 1.733949666153056, + "grad_norm": 2.925100564956665, + "learning_rate": 2.3734297677959654e-06, + "loss": 1.3957, + "step": 1688 + }, + { + "epoch": 1.7349768875192604, + "grad_norm": 2.833876371383667, + "learning_rate": 2.3715264560334984e-06, + "loss": 1.442, + "step": 1689 + }, + { + "epoch": 1.736004108885465, + "grad_norm": 2.8355112075805664, + "learning_rate": 2.3696231442710317e-06, + "loss": 1.4153, + "step": 1690 + }, + { + "epoch": 1.736004108885465, + "eval_loss": 1.8604445457458496, + "eval_runtime": 37.2018, + "eval_samples_per_second": 26.88, + "eval_steps_per_second": 1.129, + "step": 1690 + }, + { + "epoch": 1.7370313302516691, + "grad_norm": 2.7500364780426025, + "learning_rate": 2.367719832508565e-06, + "loss": 1.4165, + "step": 1691 + }, + { + "epoch": 1.7380585516178737, + "grad_norm": 2.87259578704834, + "learning_rate": 2.3658165207460985e-06, + "loss": 1.4431, + "step": 1692 + }, + { + "epoch": 1.739085772984078, + "grad_norm": 2.899538993835449, + "learning_rate": 2.363913208983632e-06, + "loss": 1.4134, + "step": 1693 + }, + { + "epoch": 1.7401129943502824, + "grad_norm": 2.799593687057495, + "learning_rate": 2.362009897221165e-06, + "loss": 1.4249, + "step": 1694 + }, + { + "epoch": 1.741140215716487, + "grad_norm": 2.8966968059539795, + "learning_rate": 2.360106585458698e-06, + "loss": 1.4184, + "step": 1695 + }, + { + "epoch": 1.7421674370826912, + "grad_norm": 2.6944682598114014, + "learning_rate": 2.3582032736962315e-06, + "loss": 1.3678, + "step": 1696 + }, + { + "epoch": 1.7431946584488958, + "grad_norm": 2.819845199584961, + "learning_rate": 2.356299961933765e-06, + "loss": 1.407, + "step": 1697 + }, + { + "epoch": 1.7442218798151001, + "grad_norm": 2.6873581409454346, + "learning_rate": 2.354396650171298e-06, + "loss": 1.3771, + "step": 1698 + }, + { + "epoch": 1.7452491011813045, + "grad_norm": 2.6941304206848145, + "learning_rate": 2.3524933384088316e-06, + "loss": 1.3743, + "step": 1699 + }, + { + "epoch": 1.746276322547509, + "grad_norm": 2.925926685333252, + "learning_rate": 2.350590026646365e-06, + "loss": 1.4549, + "step": 1700 + }, + { + "epoch": 1.746276322547509, + "eval_loss": 1.8619909286499023, + "eval_runtime": 37.3963, + "eval_samples_per_second": 26.741, + "eval_steps_per_second": 1.123, + "step": 1700 + }, + { + "epoch": 1.7473035439137135, + "grad_norm": 2.7275006771087646, + "learning_rate": 2.3486867148838983e-06, + "loss": 1.385, + "step": 1701 + }, + { + "epoch": 1.7483307652799178, + "grad_norm": 2.7096338272094727, + "learning_rate": 2.3467834031214317e-06, + "loss": 1.3732, + "step": 1702 + }, + { + "epoch": 1.7493579866461222, + "grad_norm": 2.718862295150757, + "learning_rate": 2.3448800913589646e-06, + "loss": 1.3889, + "step": 1703 + }, + { + "epoch": 1.7503852080123266, + "grad_norm": 2.8386173248291016, + "learning_rate": 2.342976779596498e-06, + "loss": 1.4083, + "step": 1704 + }, + { + "epoch": 1.7514124293785311, + "grad_norm": 2.868790864944458, + "learning_rate": 2.3410734678340313e-06, + "loss": 1.3958, + "step": 1705 + }, + { + "epoch": 1.7524396507447355, + "grad_norm": 2.869119882583618, + "learning_rate": 2.3391701560715647e-06, + "loss": 1.4344, + "step": 1706 + }, + { + "epoch": 1.7534668721109399, + "grad_norm": 2.702211380004883, + "learning_rate": 2.337266844309098e-06, + "loss": 1.3928, + "step": 1707 + }, + { + "epoch": 1.7544940934771445, + "grad_norm": 2.6727139949798584, + "learning_rate": 2.3353635325466314e-06, + "loss": 1.3872, + "step": 1708 + }, + { + "epoch": 1.7555213148433486, + "grad_norm": 2.875605583190918, + "learning_rate": 2.3334602207841648e-06, + "loss": 1.4097, + "step": 1709 + }, + { + "epoch": 1.7565485362095532, + "grad_norm": 2.7349722385406494, + "learning_rate": 2.331556909021698e-06, + "loss": 1.4041, + "step": 1710 + }, + { + "epoch": 1.7565485362095532, + "eval_loss": 1.8533939123153687, + "eval_runtime": 36.9293, + "eval_samples_per_second": 27.079, + "eval_steps_per_second": 1.137, + "step": 1710 + }, + { + "epoch": 1.7575757575757576, + "grad_norm": 2.7475039958953857, + "learning_rate": 2.3296535972592315e-06, + "loss": 1.4339, + "step": 1711 + }, + { + "epoch": 1.758602978941962, + "grad_norm": 2.705641746520996, + "learning_rate": 2.3277502854967644e-06, + "loss": 1.3763, + "step": 1712 + }, + { + "epoch": 1.7596302003081665, + "grad_norm": 2.77583646774292, + "learning_rate": 2.325846973734298e-06, + "loss": 1.4423, + "step": 1713 + }, + { + "epoch": 1.7606574216743707, + "grad_norm": 2.662750482559204, + "learning_rate": 2.323943661971831e-06, + "loss": 1.3864, + "step": 1714 + }, + { + "epoch": 1.7616846430405753, + "grad_norm": 2.769313335418701, + "learning_rate": 2.3220403502093645e-06, + "loss": 1.4272, + "step": 1715 + }, + { + "epoch": 1.7627118644067796, + "grad_norm": 2.7852976322174072, + "learning_rate": 2.320137038446898e-06, + "loss": 1.4398, + "step": 1716 + }, + { + "epoch": 1.763739085772984, + "grad_norm": 2.7452852725982666, + "learning_rate": 2.318233726684431e-06, + "loss": 1.4504, + "step": 1717 + }, + { + "epoch": 1.7647663071391886, + "grad_norm": 2.762258291244507, + "learning_rate": 2.3163304149219646e-06, + "loss": 1.4325, + "step": 1718 + }, + { + "epoch": 1.765793528505393, + "grad_norm": 2.7612709999084473, + "learning_rate": 2.314427103159498e-06, + "loss": 1.4027, + "step": 1719 + }, + { + "epoch": 1.7668207498715973, + "grad_norm": 2.7290871143341064, + "learning_rate": 2.312523791397031e-06, + "loss": 1.3919, + "step": 1720 + }, + { + "epoch": 1.7668207498715973, + "eval_loss": 1.8454608917236328, + "eval_runtime": 37.0509, + "eval_samples_per_second": 26.99, + "eval_steps_per_second": 1.134, + "step": 1720 + }, + { + "epoch": 1.767847971237802, + "grad_norm": 2.5960779190063477, + "learning_rate": 2.3106204796345643e-06, + "loss": 1.3632, + "step": 1721 + }, + { + "epoch": 1.768875192604006, + "grad_norm": 2.7832717895507812, + "learning_rate": 2.3087171678720976e-06, + "loss": 1.3899, + "step": 1722 + }, + { + "epoch": 1.7699024139702106, + "grad_norm": 2.8227648735046387, + "learning_rate": 2.306813856109631e-06, + "loss": 1.4517, + "step": 1723 + }, + { + "epoch": 1.770929635336415, + "grad_norm": 2.880059242248535, + "learning_rate": 2.3049105443471644e-06, + "loss": 1.4104, + "step": 1724 + }, + { + "epoch": 1.7719568567026194, + "grad_norm": 2.6653501987457275, + "learning_rate": 2.3030072325846973e-06, + "loss": 1.3908, + "step": 1725 + }, + { + "epoch": 1.772984078068824, + "grad_norm": 2.844914197921753, + "learning_rate": 2.3011039208222307e-06, + "loss": 1.3781, + "step": 1726 + }, + { + "epoch": 1.774011299435028, + "grad_norm": 2.8133163452148438, + "learning_rate": 2.299200609059764e-06, + "loss": 1.3775, + "step": 1727 + }, + { + "epoch": 1.7750385208012327, + "grad_norm": 2.7466955184936523, + "learning_rate": 2.297297297297298e-06, + "loss": 1.3926, + "step": 1728 + }, + { + "epoch": 1.776065742167437, + "grad_norm": 2.6311862468719482, + "learning_rate": 2.2953939855348308e-06, + "loss": 1.3486, + "step": 1729 + }, + { + "epoch": 1.7770929635336414, + "grad_norm": 2.7905211448669434, + "learning_rate": 2.293490673772364e-06, + "loss": 1.4369, + "step": 1730 + }, + { + "epoch": 1.7770929635336414, + "eval_loss": 1.8418751955032349, + "eval_runtime": 37.0448, + "eval_samples_per_second": 26.994, + "eval_steps_per_second": 1.134, + "step": 1730 + }, + { + "epoch": 1.778120184899846, + "grad_norm": 2.862295627593994, + "learning_rate": 2.2915873620098975e-06, + "loss": 1.4209, + "step": 1731 + }, + { + "epoch": 1.7791474062660502, + "grad_norm": 2.765862464904785, + "learning_rate": 2.289684050247431e-06, + "loss": 1.4135, + "step": 1732 + }, + { + "epoch": 1.7801746276322548, + "grad_norm": 2.7296290397644043, + "learning_rate": 2.287780738484964e-06, + "loss": 1.3814, + "step": 1733 + }, + { + "epoch": 1.7812018489984591, + "grad_norm": 2.7937533855438232, + "learning_rate": 2.285877426722497e-06, + "loss": 1.4203, + "step": 1734 + }, + { + "epoch": 1.7822290703646635, + "grad_norm": 2.7680230140686035, + "learning_rate": 2.2839741149600305e-06, + "loss": 1.4354, + "step": 1735 + }, + { + "epoch": 1.783256291730868, + "grad_norm": 2.8785109519958496, + "learning_rate": 2.282070803197564e-06, + "loss": 1.4338, + "step": 1736 + }, + { + "epoch": 1.7842835130970724, + "grad_norm": 2.747436046600342, + "learning_rate": 2.2801674914350972e-06, + "loss": 1.4087, + "step": 1737 + }, + { + "epoch": 1.7853107344632768, + "grad_norm": 2.6810362339019775, + "learning_rate": 2.2782641796726306e-06, + "loss": 1.3993, + "step": 1738 + }, + { + "epoch": 1.7863379558294814, + "grad_norm": 2.7365734577178955, + "learning_rate": 2.276360867910164e-06, + "loss": 1.4235, + "step": 1739 + }, + { + "epoch": 1.7873651771956856, + "grad_norm": 2.6946303844451904, + "learning_rate": 2.2744575561476973e-06, + "loss": 1.3626, + "step": 1740 + }, + { + "epoch": 1.7873651771956856, + "eval_loss": 1.8456190824508667, + "eval_runtime": 37.069, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 1740 + }, + { + "epoch": 1.7883923985618901, + "grad_norm": 2.756920337677002, + "learning_rate": 2.2725542443852307e-06, + "loss": 1.396, + "step": 1741 + }, + { + "epoch": 1.7894196199280945, + "grad_norm": 2.765428066253662, + "learning_rate": 2.2706509326227636e-06, + "loss": 1.4374, + "step": 1742 + }, + { + "epoch": 1.7904468412942989, + "grad_norm": 2.8224971294403076, + "learning_rate": 2.268747620860297e-06, + "loss": 1.4113, + "step": 1743 + }, + { + "epoch": 1.7914740626605035, + "grad_norm": 2.90537428855896, + "learning_rate": 2.2668443090978303e-06, + "loss": 1.4124, + "step": 1744 + }, + { + "epoch": 1.7925012840267076, + "grad_norm": 2.712862491607666, + "learning_rate": 2.2649409973353637e-06, + "loss": 1.4256, + "step": 1745 + }, + { + "epoch": 1.7935285053929122, + "grad_norm": 2.688558340072632, + "learning_rate": 2.263037685572897e-06, + "loss": 1.3842, + "step": 1746 + }, + { + "epoch": 1.7945557267591166, + "grad_norm": 2.6504769325256348, + "learning_rate": 2.26113437381043e-06, + "loss": 1.3703, + "step": 1747 + }, + { + "epoch": 1.795582948125321, + "grad_norm": 3.0093724727630615, + "learning_rate": 2.259231062047964e-06, + "loss": 1.4106, + "step": 1748 + }, + { + "epoch": 1.7966101694915255, + "grad_norm": 2.727687120437622, + "learning_rate": 2.257327750285497e-06, + "loss": 1.4286, + "step": 1749 + }, + { + "epoch": 1.7976373908577299, + "grad_norm": 2.679896116256714, + "learning_rate": 2.2554244385230305e-06, + "loss": 1.3937, + "step": 1750 + }, + { + "epoch": 1.7976373908577299, + "eval_loss": 1.8520207405090332, + "eval_runtime": 37.062, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 1750 + }, + { + "epoch": 1.7986646122239343, + "grad_norm": 2.8119945526123047, + "learning_rate": 2.2535211267605635e-06, + "loss": 1.3935, + "step": 1751 + }, + { + "epoch": 1.7996918335901386, + "grad_norm": 2.8470423221588135, + "learning_rate": 2.251617814998097e-06, + "loss": 1.4138, + "step": 1752 + }, + { + "epoch": 1.800719054956343, + "grad_norm": 2.6278114318847656, + "learning_rate": 2.24971450323563e-06, + "loss": 1.328, + "step": 1753 + }, + { + "epoch": 1.8017462763225476, + "grad_norm": 2.861860990524292, + "learning_rate": 2.2478111914731636e-06, + "loss": 1.416, + "step": 1754 + }, + { + "epoch": 1.802773497688752, + "grad_norm": 2.559736728668213, + "learning_rate": 2.245907879710697e-06, + "loss": 1.2891, + "step": 1755 + }, + { + "epoch": 1.8038007190549563, + "grad_norm": 2.8461058139801025, + "learning_rate": 2.24400456794823e-06, + "loss": 1.4194, + "step": 1756 + }, + { + "epoch": 1.804827940421161, + "grad_norm": 2.7069878578186035, + "learning_rate": 2.2421012561857632e-06, + "loss": 1.382, + "step": 1757 + }, + { + "epoch": 1.805855161787365, + "grad_norm": 2.727435827255249, + "learning_rate": 2.240197944423297e-06, + "loss": 1.3714, + "step": 1758 + }, + { + "epoch": 1.8068823831535696, + "grad_norm": 2.7298061847686768, + "learning_rate": 2.23829463266083e-06, + "loss": 1.3834, + "step": 1759 + }, + { + "epoch": 1.807909604519774, + "grad_norm": 2.7419843673706055, + "learning_rate": 2.2363913208983633e-06, + "loss": 1.3687, + "step": 1760 + }, + { + "epoch": 1.807909604519774, + "eval_loss": 1.8575356006622314, + "eval_runtime": 37.0563, + "eval_samples_per_second": 26.986, + "eval_steps_per_second": 1.133, + "step": 1760 + }, + { + "epoch": 1.8089368258859784, + "grad_norm": 2.8058817386627197, + "learning_rate": 2.2344880091358967e-06, + "loss": 1.415, + "step": 1761 + }, + { + "epoch": 1.809964047252183, + "grad_norm": 2.7287003993988037, + "learning_rate": 2.23258469737343e-06, + "loss": 1.3778, + "step": 1762 + }, + { + "epoch": 1.810991268618387, + "grad_norm": 2.7943124771118164, + "learning_rate": 2.2306813856109634e-06, + "loss": 1.433, + "step": 1763 + }, + { + "epoch": 1.8120184899845917, + "grad_norm": 2.7254626750946045, + "learning_rate": 2.2287780738484963e-06, + "loss": 1.3822, + "step": 1764 + }, + { + "epoch": 1.813045711350796, + "grad_norm": 2.799793004989624, + "learning_rate": 2.2268747620860297e-06, + "loss": 1.4595, + "step": 1765 + }, + { + "epoch": 1.8140729327170004, + "grad_norm": 2.79420804977417, + "learning_rate": 2.224971450323563e-06, + "loss": 1.3949, + "step": 1766 + }, + { + "epoch": 1.815100154083205, + "grad_norm": 2.861016035079956, + "learning_rate": 2.2230681385610964e-06, + "loss": 1.3301, + "step": 1767 + }, + { + "epoch": 1.8161273754494094, + "grad_norm": 2.9270451068878174, + "learning_rate": 2.2211648267986298e-06, + "loss": 1.3895, + "step": 1768 + }, + { + "epoch": 1.8171545968156138, + "grad_norm": 2.709031105041504, + "learning_rate": 2.219261515036163e-06, + "loss": 1.3757, + "step": 1769 + }, + { + "epoch": 1.8181818181818183, + "grad_norm": 2.7763030529022217, + "learning_rate": 2.2173582032736965e-06, + "loss": 1.4135, + "step": 1770 + }, + { + "epoch": 1.8181818181818183, + "eval_loss": 1.8521159887313843, + "eval_runtime": 37.1265, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 1770 + }, + { + "epoch": 1.8192090395480225, + "grad_norm": 2.97048282623291, + "learning_rate": 2.21545489151123e-06, + "loss": 1.4345, + "step": 1771 + }, + { + "epoch": 1.820236260914227, + "grad_norm": 2.8238744735717773, + "learning_rate": 2.2135515797487632e-06, + "loss": 1.3948, + "step": 1772 + }, + { + "epoch": 1.8212634822804314, + "grad_norm": 2.6430611610412598, + "learning_rate": 2.211648267986296e-06, + "loss": 1.3878, + "step": 1773 + }, + { + "epoch": 1.8222907036466358, + "grad_norm": 2.602883815765381, + "learning_rate": 2.2097449562238295e-06, + "loss": 1.3978, + "step": 1774 + }, + { + "epoch": 1.8233179250128404, + "grad_norm": 2.707813262939453, + "learning_rate": 2.207841644461363e-06, + "loss": 1.3937, + "step": 1775 + }, + { + "epoch": 1.8243451463790445, + "grad_norm": 2.662684679031372, + "learning_rate": 2.2059383326988963e-06, + "loss": 1.4381, + "step": 1776 + }, + { + "epoch": 1.8253723677452491, + "grad_norm": 2.9407193660736084, + "learning_rate": 2.2040350209364296e-06, + "loss": 1.4059, + "step": 1777 + }, + { + "epoch": 1.8263995891114535, + "grad_norm": 2.6678171157836914, + "learning_rate": 2.2021317091739626e-06, + "loss": 1.3718, + "step": 1778 + }, + { + "epoch": 1.8274268104776579, + "grad_norm": 2.7536516189575195, + "learning_rate": 2.2002283974114963e-06, + "loss": 1.3998, + "step": 1779 + }, + { + "epoch": 1.8284540318438625, + "grad_norm": 2.7340128421783447, + "learning_rate": 2.1983250856490297e-06, + "loss": 1.3299, + "step": 1780 + }, + { + "epoch": 1.8284540318438625, + "eval_loss": 1.854264736175537, + "eval_runtime": 37.0874, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1780 + }, + { + "epoch": 1.8294812532100666, + "grad_norm": 2.844285726547241, + "learning_rate": 2.1964217738865626e-06, + "loss": 1.3746, + "step": 1781 + }, + { + "epoch": 1.8305084745762712, + "grad_norm": 2.9005022048950195, + "learning_rate": 2.194518462124096e-06, + "loss": 1.4087, + "step": 1782 + }, + { + "epoch": 1.8315356959424756, + "grad_norm": 2.8825247287750244, + "learning_rate": 2.1926151503616294e-06, + "loss": 1.4458, + "step": 1783 + }, + { + "epoch": 1.83256291730868, + "grad_norm": 2.712778091430664, + "learning_rate": 2.1907118385991627e-06, + "loss": 1.414, + "step": 1784 + }, + { + "epoch": 1.8335901386748845, + "grad_norm": 2.7036001682281494, + "learning_rate": 2.188808526836696e-06, + "loss": 1.3632, + "step": 1785 + }, + { + "epoch": 1.8346173600410889, + "grad_norm": 2.704606771469116, + "learning_rate": 2.1869052150742295e-06, + "loss": 1.3992, + "step": 1786 + }, + { + "epoch": 1.8356445814072933, + "grad_norm": 2.7564282417297363, + "learning_rate": 2.1850019033117624e-06, + "loss": 1.3331, + "step": 1787 + }, + { + "epoch": 1.8366718027734978, + "grad_norm": 2.898256540298462, + "learning_rate": 2.1830985915492958e-06, + "loss": 1.3881, + "step": 1788 + }, + { + "epoch": 1.837699024139702, + "grad_norm": 2.7897069454193115, + "learning_rate": 2.1811952797868295e-06, + "loss": 1.4081, + "step": 1789 + }, + { + "epoch": 1.8387262455059066, + "grad_norm": 2.7293193340301514, + "learning_rate": 2.1792919680243625e-06, + "loss": 1.3705, + "step": 1790 + }, + { + "epoch": 1.8387262455059066, + "eval_loss": 1.8483723402023315, + "eval_runtime": 37.4058, + "eval_samples_per_second": 26.734, + "eval_steps_per_second": 1.123, + "step": 1790 + }, + { + "epoch": 1.839753466872111, + "grad_norm": 2.9257752895355225, + "learning_rate": 2.177388656261896e-06, + "loss": 1.4071, + "step": 1791 + }, + { + "epoch": 1.8407806882383153, + "grad_norm": 3.016033172607422, + "learning_rate": 2.1754853444994292e-06, + "loss": 1.4774, + "step": 1792 + }, + { + "epoch": 1.84180790960452, + "grad_norm": 2.74198842048645, + "learning_rate": 2.1735820327369626e-06, + "loss": 1.301, + "step": 1793 + }, + { + "epoch": 1.842835130970724, + "grad_norm": 2.809215545654297, + "learning_rate": 2.171678720974496e-06, + "loss": 1.4192, + "step": 1794 + }, + { + "epoch": 1.8438623523369286, + "grad_norm": 2.673997402191162, + "learning_rate": 2.169775409212029e-06, + "loss": 1.4182, + "step": 1795 + }, + { + "epoch": 1.844889573703133, + "grad_norm": 2.9519944190979004, + "learning_rate": 2.1678720974495622e-06, + "loss": 1.4253, + "step": 1796 + }, + { + "epoch": 1.8459167950693374, + "grad_norm": 2.7683424949645996, + "learning_rate": 2.1659687856870956e-06, + "loss": 1.4162, + "step": 1797 + }, + { + "epoch": 1.846944016435542, + "grad_norm": 2.861701726913452, + "learning_rate": 2.164065473924629e-06, + "loss": 1.4167, + "step": 1798 + }, + { + "epoch": 1.8479712378017463, + "grad_norm": 2.62581467628479, + "learning_rate": 2.1621621621621623e-06, + "loss": 1.3568, + "step": 1799 + }, + { + "epoch": 1.8489984591679507, + "grad_norm": 2.782197952270508, + "learning_rate": 2.1602588503996957e-06, + "loss": 1.4184, + "step": 1800 + }, + { + "epoch": 1.8489984591679507, + "eval_loss": 1.8527827262878418, + "eval_runtime": 37.4743, + "eval_samples_per_second": 26.685, + "eval_steps_per_second": 1.121, + "step": 1800 + }, + { + "epoch": 1.850025680534155, + "grad_norm": 2.7289304733276367, + "learning_rate": 2.158355538637229e-06, + "loss": 1.3979, + "step": 1801 + }, + { + "epoch": 1.8510529019003594, + "grad_norm": 2.7153542041778564, + "learning_rate": 2.1564522268747624e-06, + "loss": 1.3863, + "step": 1802 + }, + { + "epoch": 1.852080123266564, + "grad_norm": 2.884798765182495, + "learning_rate": 2.1545489151122958e-06, + "loss": 1.4656, + "step": 1803 + }, + { + "epoch": 1.8531073446327684, + "grad_norm": 2.6483845710754395, + "learning_rate": 2.1526456033498287e-06, + "loss": 1.4118, + "step": 1804 + }, + { + "epoch": 1.8541345659989728, + "grad_norm": 2.7074198722839355, + "learning_rate": 2.150742291587362e-06, + "loss": 1.41, + "step": 1805 + }, + { + "epoch": 1.8551617873651773, + "grad_norm": 2.734137535095215, + "learning_rate": 2.1488389798248954e-06, + "loss": 1.3968, + "step": 1806 + }, + { + "epoch": 1.8561890087313815, + "grad_norm": 2.7855992317199707, + "learning_rate": 2.146935668062429e-06, + "loss": 1.427, + "step": 1807 + }, + { + "epoch": 1.857216230097586, + "grad_norm": 2.7769339084625244, + "learning_rate": 2.145032356299962e-06, + "loss": 1.3476, + "step": 1808 + }, + { + "epoch": 1.8582434514637904, + "grad_norm": 2.6978964805603027, + "learning_rate": 2.1431290445374955e-06, + "loss": 1.4047, + "step": 1809 + }, + { + "epoch": 1.8592706728299948, + "grad_norm": 2.806438446044922, + "learning_rate": 2.141225732775029e-06, + "loss": 1.3693, + "step": 1810 + }, + { + "epoch": 1.8592706728299948, + "eval_loss": 1.8606462478637695, + "eval_runtime": 36.9431, + "eval_samples_per_second": 27.069, + "eval_steps_per_second": 1.137, + "step": 1810 + }, + { + "epoch": 1.8602978941961994, + "grad_norm": 2.886394500732422, + "learning_rate": 2.1393224210125623e-06, + "loss": 1.4022, + "step": 1811 + }, + { + "epoch": 1.8613251155624035, + "grad_norm": 2.7849435806274414, + "learning_rate": 2.137419109250095e-06, + "loss": 1.3907, + "step": 1812 + }, + { + "epoch": 1.8623523369286081, + "grad_norm": 2.85810923576355, + "learning_rate": 2.1355157974876286e-06, + "loss": 1.4451, + "step": 1813 + }, + { + "epoch": 1.8633795582948125, + "grad_norm": 2.8717806339263916, + "learning_rate": 2.133612485725162e-06, + "loss": 1.4303, + "step": 1814 + }, + { + "epoch": 1.8644067796610169, + "grad_norm": 2.6955783367156982, + "learning_rate": 2.1317091739626953e-06, + "loss": 1.3454, + "step": 1815 + }, + { + "epoch": 1.8654340010272215, + "grad_norm": 2.835347890853882, + "learning_rate": 2.1298058622002286e-06, + "loss": 1.3671, + "step": 1816 + }, + { + "epoch": 1.8664612223934258, + "grad_norm": 2.804835796356201, + "learning_rate": 2.1279025504377616e-06, + "loss": 1.388, + "step": 1817 + }, + { + "epoch": 1.8674884437596302, + "grad_norm": 2.69130539894104, + "learning_rate": 2.125999238675295e-06, + "loss": 1.3815, + "step": 1818 + }, + { + "epoch": 1.8685156651258348, + "grad_norm": 2.856457233428955, + "learning_rate": 2.1240959269128287e-06, + "loss": 1.4323, + "step": 1819 + }, + { + "epoch": 1.869542886492039, + "grad_norm": 2.8100039958953857, + "learning_rate": 2.122192615150362e-06, + "loss": 1.3942, + "step": 1820 + }, + { + "epoch": 1.869542886492039, + "eval_loss": 1.860397219657898, + "eval_runtime": 37.0473, + "eval_samples_per_second": 26.993, + "eval_steps_per_second": 1.134, + "step": 1820 + }, + { + "epoch": 1.8705701078582435, + "grad_norm": 2.6977529525756836, + "learning_rate": 2.120289303387895e-06, + "loss": 1.3371, + "step": 1821 + }, + { + "epoch": 1.8715973292244479, + "grad_norm": 2.8403844833374023, + "learning_rate": 2.1183859916254284e-06, + "loss": 1.3776, + "step": 1822 + }, + { + "epoch": 1.8726245505906522, + "grad_norm": 2.829756498336792, + "learning_rate": 2.1164826798629618e-06, + "loss": 1.4229, + "step": 1823 + }, + { + "epoch": 1.8736517719568568, + "grad_norm": 2.6810450553894043, + "learning_rate": 2.114579368100495e-06, + "loss": 1.3757, + "step": 1824 + }, + { + "epoch": 1.874678993323061, + "grad_norm": 2.988982677459717, + "learning_rate": 2.1126760563380285e-06, + "loss": 1.3671, + "step": 1825 + }, + { + "epoch": 1.8757062146892656, + "grad_norm": 2.661953926086426, + "learning_rate": 2.1107727445755614e-06, + "loss": 1.3382, + "step": 1826 + }, + { + "epoch": 1.87673343605547, + "grad_norm": 2.7993199825286865, + "learning_rate": 2.1088694328130948e-06, + "loss": 1.4621, + "step": 1827 + }, + { + "epoch": 1.8777606574216743, + "grad_norm": 2.681609869003296, + "learning_rate": 2.106966121050628e-06, + "loss": 1.3596, + "step": 1828 + }, + { + "epoch": 1.878787878787879, + "grad_norm": 2.988734722137451, + "learning_rate": 2.1050628092881615e-06, + "loss": 1.4269, + "step": 1829 + }, + { + "epoch": 1.879815100154083, + "grad_norm": 2.867772102355957, + "learning_rate": 2.103159497525695e-06, + "loss": 1.4618, + "step": 1830 + }, + { + "epoch": 1.879815100154083, + "eval_loss": 1.8595913648605347, + "eval_runtime": 37.1186, + "eval_samples_per_second": 26.941, + "eval_steps_per_second": 1.132, + "step": 1830 + }, + { + "epoch": 1.8808423215202876, + "grad_norm": 2.7212493419647217, + "learning_rate": 2.1012561857632282e-06, + "loss": 1.3738, + "step": 1831 + }, + { + "epoch": 1.881869542886492, + "grad_norm": 2.7997097969055176, + "learning_rate": 2.0993528740007616e-06, + "loss": 1.4268, + "step": 1832 + }, + { + "epoch": 1.8828967642526964, + "grad_norm": 2.8695220947265625, + "learning_rate": 2.097449562238295e-06, + "loss": 1.4397, + "step": 1833 + }, + { + "epoch": 1.883923985618901, + "grad_norm": 2.681015729904175, + "learning_rate": 2.095546250475828e-06, + "loss": 1.3873, + "step": 1834 + }, + { + "epoch": 1.8849512069851053, + "grad_norm": 2.787360429763794, + "learning_rate": 2.0936429387133613e-06, + "loss": 1.3944, + "step": 1835 + }, + { + "epoch": 1.8859784283513097, + "grad_norm": 2.830164670944214, + "learning_rate": 2.0917396269508946e-06, + "loss": 1.4485, + "step": 1836 + }, + { + "epoch": 1.8870056497175143, + "grad_norm": 2.9803659915924072, + "learning_rate": 2.089836315188428e-06, + "loss": 1.4259, + "step": 1837 + }, + { + "epoch": 1.8880328710837184, + "grad_norm": 2.846390962600708, + "learning_rate": 2.0879330034259614e-06, + "loss": 1.4371, + "step": 1838 + }, + { + "epoch": 1.889060092449923, + "grad_norm": 2.636085033416748, + "learning_rate": 2.0860296916634947e-06, + "loss": 1.3679, + "step": 1839 + }, + { + "epoch": 1.8900873138161274, + "grad_norm": 2.806547164916992, + "learning_rate": 2.084126379901028e-06, + "loss": 1.4662, + "step": 1840 + }, + { + "epoch": 1.8900873138161274, + "eval_loss": 1.8502073287963867, + "eval_runtime": 37.1321, + "eval_samples_per_second": 26.931, + "eval_steps_per_second": 1.131, + "step": 1840 + }, + { + "epoch": 1.8911145351823317, + "grad_norm": 2.8658089637756348, + "learning_rate": 2.0822230681385614e-06, + "loss": 1.3593, + "step": 1841 + }, + { + "epoch": 1.8921417565485363, + "grad_norm": 2.87060284614563, + "learning_rate": 2.080319756376095e-06, + "loss": 1.4189, + "step": 1842 + }, + { + "epoch": 1.8931689779147405, + "grad_norm": 2.7965638637542725, + "learning_rate": 2.0784164446136277e-06, + "loss": 1.4208, + "step": 1843 + }, + { + "epoch": 1.894196199280945, + "grad_norm": 2.5999045372009277, + "learning_rate": 2.076513132851161e-06, + "loss": 1.375, + "step": 1844 + }, + { + "epoch": 1.8952234206471494, + "grad_norm": 2.833935260772705, + "learning_rate": 2.0746098210886945e-06, + "loss": 1.4137, + "step": 1845 + }, + { + "epoch": 1.8962506420133538, + "grad_norm": 2.9809396266937256, + "learning_rate": 2.072706509326228e-06, + "loss": 1.3874, + "step": 1846 + }, + { + "epoch": 1.8972778633795584, + "grad_norm": 2.9341282844543457, + "learning_rate": 2.070803197563761e-06, + "loss": 1.3953, + "step": 1847 + }, + { + "epoch": 1.8983050847457628, + "grad_norm": 2.989488124847412, + "learning_rate": 2.068899885801294e-06, + "loss": 1.3913, + "step": 1848 + }, + { + "epoch": 1.8993323061119671, + "grad_norm": 2.8224217891693115, + "learning_rate": 2.066996574038828e-06, + "loss": 1.4032, + "step": 1849 + }, + { + "epoch": 1.9003595274781715, + "grad_norm": 2.8850255012512207, + "learning_rate": 2.0650932622763613e-06, + "loss": 1.4402, + "step": 1850 + }, + { + "epoch": 1.9003595274781715, + "eval_loss": 1.8440051078796387, + "eval_runtime": 37.1532, + "eval_samples_per_second": 26.916, + "eval_steps_per_second": 1.13, + "step": 1850 + }, + { + "epoch": 1.9013867488443759, + "grad_norm": 2.927288293838501, + "learning_rate": 2.0631899505138942e-06, + "loss": 1.3215, + "step": 1851 + }, + { + "epoch": 1.9024139702105805, + "grad_norm": 2.7680368423461914, + "learning_rate": 2.0612866387514276e-06, + "loss": 1.3636, + "step": 1852 + }, + { + "epoch": 1.9034411915767848, + "grad_norm": 2.8515689373016357, + "learning_rate": 2.059383326988961e-06, + "loss": 1.4049, + "step": 1853 + }, + { + "epoch": 1.9044684129429892, + "grad_norm": 2.7973034381866455, + "learning_rate": 2.0574800152264943e-06, + "loss": 1.3876, + "step": 1854 + }, + { + "epoch": 1.9054956343091938, + "grad_norm": 2.7860023975372314, + "learning_rate": 2.0555767034640277e-06, + "loss": 1.3687, + "step": 1855 + }, + { + "epoch": 1.906522855675398, + "grad_norm": 2.807511568069458, + "learning_rate": 2.053673391701561e-06, + "loss": 1.3382, + "step": 1856 + }, + { + "epoch": 1.9075500770416025, + "grad_norm": 2.9342591762542725, + "learning_rate": 2.051770079939094e-06, + "loss": 1.4025, + "step": 1857 + }, + { + "epoch": 1.9085772984078069, + "grad_norm": 2.85943865776062, + "learning_rate": 2.0498667681766273e-06, + "loss": 1.3626, + "step": 1858 + }, + { + "epoch": 1.9096045197740112, + "grad_norm": 2.8514609336853027, + "learning_rate": 2.047963456414161e-06, + "loss": 1.3675, + "step": 1859 + }, + { + "epoch": 1.9106317411402158, + "grad_norm": 2.8177332878112793, + "learning_rate": 2.046060144651694e-06, + "loss": 1.3968, + "step": 1860 + }, + { + "epoch": 1.9106317411402158, + "eval_loss": 1.8451731204986572, + "eval_runtime": 37.1047, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 1860 + }, + { + "epoch": 1.91165896250642, + "grad_norm": 2.74876070022583, + "learning_rate": 2.0441568328892274e-06, + "loss": 1.4022, + "step": 1861 + }, + { + "epoch": 1.9126861838726246, + "grad_norm": 2.6633286476135254, + "learning_rate": 2.0422535211267608e-06, + "loss": 1.4004, + "step": 1862 + }, + { + "epoch": 1.913713405238829, + "grad_norm": 2.6457369327545166, + "learning_rate": 2.040350209364294e-06, + "loss": 1.3767, + "step": 1863 + }, + { + "epoch": 1.9147406266050333, + "grad_norm": 2.7011449337005615, + "learning_rate": 2.0384468976018275e-06, + "loss": 1.373, + "step": 1864 + }, + { + "epoch": 1.915767847971238, + "grad_norm": 2.676337480545044, + "learning_rate": 2.0365435858393604e-06, + "loss": 1.3713, + "step": 1865 + }, + { + "epoch": 1.9167950693374423, + "grad_norm": 2.7401342391967773, + "learning_rate": 2.034640274076894e-06, + "loss": 1.3644, + "step": 1866 + }, + { + "epoch": 1.9178222907036466, + "grad_norm": 2.7799527645111084, + "learning_rate": 2.032736962314427e-06, + "loss": 1.433, + "step": 1867 + }, + { + "epoch": 1.9188495120698512, + "grad_norm": 2.7372307777404785, + "learning_rate": 2.0308336505519605e-06, + "loss": 1.3781, + "step": 1868 + }, + { + "epoch": 1.9198767334360554, + "grad_norm": 2.986297845840454, + "learning_rate": 2.028930338789494e-06, + "loss": 1.4406, + "step": 1869 + }, + { + "epoch": 1.92090395480226, + "grad_norm": 2.815706491470337, + "learning_rate": 2.0270270270270273e-06, + "loss": 1.3555, + "step": 1870 + }, + { + "epoch": 1.92090395480226, + "eval_loss": 1.8498139381408691, + "eval_runtime": 37.1146, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 1870 + }, + { + "epoch": 1.9219311761684643, + "grad_norm": 2.734189987182617, + "learning_rate": 2.0251237152645606e-06, + "loss": 1.373, + "step": 1871 + }, + { + "epoch": 1.9229583975346687, + "grad_norm": 2.7900359630584717, + "learning_rate": 2.023220403502094e-06, + "loss": 1.3623, + "step": 1872 + }, + { + "epoch": 1.9239856189008733, + "grad_norm": 2.8590710163116455, + "learning_rate": 2.0213170917396273e-06, + "loss": 1.3424, + "step": 1873 + }, + { + "epoch": 1.9250128402670774, + "grad_norm": 2.880814790725708, + "learning_rate": 2.0194137799771603e-06, + "loss": 1.3683, + "step": 1874 + }, + { + "epoch": 1.926040061633282, + "grad_norm": 2.7775630950927734, + "learning_rate": 2.0175104682146937e-06, + "loss": 1.3879, + "step": 1875 + }, + { + "epoch": 1.9270672829994864, + "grad_norm": 2.713557481765747, + "learning_rate": 2.015607156452227e-06, + "loss": 1.3216, + "step": 1876 + }, + { + "epoch": 1.9280945043656907, + "grad_norm": 2.8787734508514404, + "learning_rate": 2.0137038446897604e-06, + "loss": 1.3659, + "step": 1877 + }, + { + "epoch": 1.9291217257318953, + "grad_norm": 2.8224005699157715, + "learning_rate": 2.0118005329272937e-06, + "loss": 1.3572, + "step": 1878 + }, + { + "epoch": 1.9301489470980995, + "grad_norm": 2.865316390991211, + "learning_rate": 2.0098972211648267e-06, + "loss": 1.3817, + "step": 1879 + }, + { + "epoch": 1.931176168464304, + "grad_norm": 2.7488853931427, + "learning_rate": 2.0079939094023605e-06, + "loss": 1.3766, + "step": 1880 + }, + { + "epoch": 1.931176168464304, + "eval_loss": 1.8444081544876099, + "eval_runtime": 37.1079, + "eval_samples_per_second": 26.948, + "eval_steps_per_second": 1.132, + "step": 1880 + }, + { + "epoch": 1.9322033898305084, + "grad_norm": 2.902949333190918, + "learning_rate": 2.006090597639894e-06, + "loss": 1.4183, + "step": 1881 + }, + { + "epoch": 1.9332306111967128, + "grad_norm": 2.8154122829437256, + "learning_rate": 2.0041872858774268e-06, + "loss": 1.4525, + "step": 1882 + }, + { + "epoch": 1.9342578325629174, + "grad_norm": 2.8343429565429688, + "learning_rate": 2.00228397411496e-06, + "loss": 1.4099, + "step": 1883 + }, + { + "epoch": 1.9352850539291218, + "grad_norm": 2.863067388534546, + "learning_rate": 2.0003806623524935e-06, + "loss": 1.3973, + "step": 1884 + }, + { + "epoch": 1.9363122752953261, + "grad_norm": 2.784085988998413, + "learning_rate": 1.998477350590027e-06, + "loss": 1.3117, + "step": 1885 + }, + { + "epoch": 1.9373394966615307, + "grad_norm": 2.7667958736419678, + "learning_rate": 1.9965740388275602e-06, + "loss": 1.4056, + "step": 1886 + }, + { + "epoch": 1.9383667180277349, + "grad_norm": 2.747432231903076, + "learning_rate": 1.994670727065093e-06, + "loss": 1.3927, + "step": 1887 + }, + { + "epoch": 1.9393939393939394, + "grad_norm": 2.9162001609802246, + "learning_rate": 1.9927674153026265e-06, + "loss": 1.3453, + "step": 1888 + }, + { + "epoch": 1.9404211607601438, + "grad_norm": 3.0909414291381836, + "learning_rate": 1.99086410354016e-06, + "loss": 1.4148, + "step": 1889 + }, + { + "epoch": 1.9414483821263482, + "grad_norm": 2.774672746658325, + "learning_rate": 1.9889607917776937e-06, + "loss": 1.3762, + "step": 1890 + }, + { + "epoch": 1.9414483821263482, + "eval_loss": 1.848209023475647, + "eval_runtime": 37.2784, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 1890 + }, + { + "epoch": 1.9424756034925528, + "grad_norm": 2.7985880374908447, + "learning_rate": 1.9870574800152266e-06, + "loss": 1.3908, + "step": 1891 + }, + { + "epoch": 1.943502824858757, + "grad_norm": 2.749664783477783, + "learning_rate": 1.98515416825276e-06, + "loss": 1.4088, + "step": 1892 + }, + { + "epoch": 1.9445300462249615, + "grad_norm": 2.770726442337036, + "learning_rate": 1.9832508564902933e-06, + "loss": 1.4054, + "step": 1893 + }, + { + "epoch": 1.9455572675911659, + "grad_norm": 2.634516477584839, + "learning_rate": 1.9813475447278267e-06, + "loss": 1.3247, + "step": 1894 + }, + { + "epoch": 1.9465844889573702, + "grad_norm": 2.886481285095215, + "learning_rate": 1.97944423296536e-06, + "loss": 1.4025, + "step": 1895 + }, + { + "epoch": 1.9476117103235748, + "grad_norm": 2.8149733543395996, + "learning_rate": 1.977540921202893e-06, + "loss": 1.3745, + "step": 1896 + }, + { + "epoch": 1.9486389316897792, + "grad_norm": 2.643970251083374, + "learning_rate": 1.9756376094404264e-06, + "loss": 1.3581, + "step": 1897 + }, + { + "epoch": 1.9496661530559836, + "grad_norm": 2.7485244274139404, + "learning_rate": 1.9737342976779597e-06, + "loss": 1.3426, + "step": 1898 + }, + { + "epoch": 1.950693374422188, + "grad_norm": 2.930513858795166, + "learning_rate": 1.971830985915493e-06, + "loss": 1.4194, + "step": 1899 + }, + { + "epoch": 1.9517205957883923, + "grad_norm": 2.7637267112731934, + "learning_rate": 1.9699276741530264e-06, + "loss": 1.3908, + "step": 1900 + }, + { + "epoch": 1.9517205957883923, + "eval_loss": 1.8510984182357788, + "eval_runtime": 37.4089, + "eval_samples_per_second": 26.732, + "eval_steps_per_second": 1.123, + "step": 1900 + }, + { + "epoch": 1.9527478171545969, + "grad_norm": 2.8946938514709473, + "learning_rate": 1.96802436239056e-06, + "loss": 1.3892, + "step": 1901 + }, + { + "epoch": 1.9537750385208013, + "grad_norm": 2.7064263820648193, + "learning_rate": 1.966121050628093e-06, + "loss": 1.3628, + "step": 1902 + }, + { + "epoch": 1.9548022598870056, + "grad_norm": 2.8588147163391113, + "learning_rate": 1.9642177388656265e-06, + "loss": 1.4097, + "step": 1903 + }, + { + "epoch": 1.9558294812532102, + "grad_norm": 2.7746825218200684, + "learning_rate": 1.9623144271031595e-06, + "loss": 1.4255, + "step": 1904 + }, + { + "epoch": 1.9568567026194144, + "grad_norm": 2.872587203979492, + "learning_rate": 1.960411115340693e-06, + "loss": 1.4308, + "step": 1905 + }, + { + "epoch": 1.957883923985619, + "grad_norm": 2.714769124984741, + "learning_rate": 1.958507803578226e-06, + "loss": 1.3847, + "step": 1906 + }, + { + "epoch": 1.9589111453518233, + "grad_norm": 2.7117342948913574, + "learning_rate": 1.9566044918157596e-06, + "loss": 1.406, + "step": 1907 + }, + { + "epoch": 1.9599383667180277, + "grad_norm": 2.754995346069336, + "learning_rate": 1.954701180053293e-06, + "loss": 1.4216, + "step": 1908 + }, + { + "epoch": 1.9609655880842323, + "grad_norm": 2.6597256660461426, + "learning_rate": 1.952797868290826e-06, + "loss": 1.3748, + "step": 1909 + }, + { + "epoch": 1.9619928094504364, + "grad_norm": 2.7966320514678955, + "learning_rate": 1.9508945565283596e-06, + "loss": 1.4237, + "step": 1910 + }, + { + "epoch": 1.9619928094504364, + "eval_loss": 1.8453261852264404, + "eval_runtime": 36.9326, + "eval_samples_per_second": 27.076, + "eval_steps_per_second": 1.137, + "step": 1910 + }, + { + "epoch": 1.963020030816641, + "grad_norm": 2.7504498958587646, + "learning_rate": 1.948991244765893e-06, + "loss": 1.3832, + "step": 1911 + }, + { + "epoch": 1.9640472521828454, + "grad_norm": 2.7729601860046387, + "learning_rate": 1.9470879330034264e-06, + "loss": 1.4176, + "step": 1912 + }, + { + "epoch": 1.9650744735490497, + "grad_norm": 2.6513381004333496, + "learning_rate": 1.9451846212409593e-06, + "loss": 1.3524, + "step": 1913 + }, + { + "epoch": 1.9661016949152543, + "grad_norm": 2.8147811889648438, + "learning_rate": 1.9432813094784927e-06, + "loss": 1.4119, + "step": 1914 + }, + { + "epoch": 1.9671289162814587, + "grad_norm": 2.7253763675689697, + "learning_rate": 1.941377997716026e-06, + "loss": 1.3458, + "step": 1915 + }, + { + "epoch": 1.968156137647663, + "grad_norm": 2.6741955280303955, + "learning_rate": 1.9394746859535594e-06, + "loss": 1.3764, + "step": 1916 + }, + { + "epoch": 1.9691833590138677, + "grad_norm": 2.785856008529663, + "learning_rate": 1.9375713741910928e-06, + "loss": 1.3484, + "step": 1917 + }, + { + "epoch": 1.9702105803800718, + "grad_norm": 2.743168592453003, + "learning_rate": 1.9356680624286257e-06, + "loss": 1.3779, + "step": 1918 + }, + { + "epoch": 1.9712378017462764, + "grad_norm": 2.827883243560791, + "learning_rate": 1.933764750666159e-06, + "loss": 1.3246, + "step": 1919 + }, + { + "epoch": 1.9722650231124808, + "grad_norm": 2.7379672527313232, + "learning_rate": 1.931861438903693e-06, + "loss": 1.383, + "step": 1920 + }, + { + "epoch": 1.9722650231124808, + "eval_loss": 1.8433737754821777, + "eval_runtime": 37.0421, + "eval_samples_per_second": 26.996, + "eval_steps_per_second": 1.134, + "step": 1920 + }, + { + "epoch": 1.9732922444786851, + "grad_norm": 2.7514965534210205, + "learning_rate": 1.929958127141226e-06, + "loss": 1.3513, + "step": 1921 + }, + { + "epoch": 1.9743194658448897, + "grad_norm": 2.6764447689056396, + "learning_rate": 1.928054815378759e-06, + "loss": 1.3718, + "step": 1922 + }, + { + "epoch": 1.9753466872110939, + "grad_norm": 2.8592004776000977, + "learning_rate": 1.9261515036162925e-06, + "loss": 1.4479, + "step": 1923 + }, + { + "epoch": 1.9763739085772984, + "grad_norm": 2.8035528659820557, + "learning_rate": 1.924248191853826e-06, + "loss": 1.4222, + "step": 1924 + }, + { + "epoch": 1.9774011299435028, + "grad_norm": 2.7685303688049316, + "learning_rate": 1.9223448800913592e-06, + "loss": 1.3939, + "step": 1925 + }, + { + "epoch": 1.9784283513097072, + "grad_norm": 2.6679201126098633, + "learning_rate": 1.9204415683288926e-06, + "loss": 1.385, + "step": 1926 + }, + { + "epoch": 1.9794555726759118, + "grad_norm": 2.7546098232269287, + "learning_rate": 1.9185382565664255e-06, + "loss": 1.4134, + "step": 1927 + }, + { + "epoch": 1.980482794042116, + "grad_norm": 2.794991970062256, + "learning_rate": 1.916634944803959e-06, + "loss": 1.4212, + "step": 1928 + }, + { + "epoch": 1.9815100154083205, + "grad_norm": 2.767075538635254, + "learning_rate": 1.9147316330414923e-06, + "loss": 1.3979, + "step": 1929 + }, + { + "epoch": 1.9825372367745249, + "grad_norm": 2.7712795734405518, + "learning_rate": 1.9128283212790256e-06, + "loss": 1.3827, + "step": 1930 + }, + { + "epoch": 1.9825372367745249, + "eval_loss": 1.8437753915786743, + "eval_runtime": 37.0885, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 1930 + }, + { + "epoch": 1.9835644581407292, + "grad_norm": 2.791165351867676, + "learning_rate": 1.910925009516559e-06, + "loss": 1.4328, + "step": 1931 + }, + { + "epoch": 1.9845916795069338, + "grad_norm": 2.7170767784118652, + "learning_rate": 1.9090216977540924e-06, + "loss": 1.3958, + "step": 1932 + }, + { + "epoch": 1.9856189008731382, + "grad_norm": 2.8529574871063232, + "learning_rate": 1.9071183859916257e-06, + "loss": 1.4379, + "step": 1933 + }, + { + "epoch": 1.9866461222393426, + "grad_norm": 2.8424859046936035, + "learning_rate": 1.9052150742291589e-06, + "loss": 1.4066, + "step": 1934 + }, + { + "epoch": 1.9876733436055471, + "grad_norm": 2.8055100440979004, + "learning_rate": 1.9033117624666922e-06, + "loss": 1.4, + "step": 1935 + }, + { + "epoch": 1.9887005649717513, + "grad_norm": 2.73415470123291, + "learning_rate": 1.9014084507042254e-06, + "loss": 1.3652, + "step": 1936 + }, + { + "epoch": 1.9897277863379559, + "grad_norm": 2.7907590866088867, + "learning_rate": 1.8995051389417587e-06, + "loss": 1.3475, + "step": 1937 + }, + { + "epoch": 1.9907550077041603, + "grad_norm": 2.6994218826293945, + "learning_rate": 1.8976018271792921e-06, + "loss": 1.3428, + "step": 1938 + }, + { + "epoch": 1.9917822290703646, + "grad_norm": 2.8237955570220947, + "learning_rate": 1.8956985154168253e-06, + "loss": 1.3918, + "step": 1939 + }, + { + "epoch": 1.9928094504365692, + "grad_norm": 2.779270648956299, + "learning_rate": 1.8937952036543588e-06, + "loss": 1.3726, + "step": 1940 + }, + { + "epoch": 1.9928094504365692, + "eval_loss": 1.8435344696044922, + "eval_runtime": 37.0797, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 1940 + }, + { + "epoch": 1.9938366718027734, + "grad_norm": 2.7808098793029785, + "learning_rate": 1.8918918918918922e-06, + "loss": 1.3368, + "step": 1941 + }, + { + "epoch": 1.994863893168978, + "grad_norm": 2.7355751991271973, + "learning_rate": 1.8899885801294253e-06, + "loss": 1.39, + "step": 1942 + }, + { + "epoch": 1.9958911145351823, + "grad_norm": 2.907468318939209, + "learning_rate": 1.8880852683669587e-06, + "loss": 1.399, + "step": 1943 + }, + { + "epoch": 1.9969183359013867, + "grad_norm": 2.8597185611724854, + "learning_rate": 1.886181956604492e-06, + "loss": 1.3731, + "step": 1944 + }, + { + "epoch": 1.9979455572675913, + "grad_norm": 2.8638176918029785, + "learning_rate": 1.8842786448420252e-06, + "loss": 1.3696, + "step": 1945 + }, + { + "epoch": 1.9989727786337956, + "grad_norm": 2.880169153213501, + "learning_rate": 1.8823753330795586e-06, + "loss": 1.3971, + "step": 1946 + }, + { + "epoch": 2.0, + "grad_norm": 2.7660279273986816, + "learning_rate": 1.8804720213170917e-06, + "loss": 1.3026, + "step": 1947 + }, + { + "epoch": 2.0010272213662046, + "grad_norm": 2.9506170749664307, + "learning_rate": 1.878568709554625e-06, + "loss": 1.3298, + "step": 1948 + }, + { + "epoch": 2.0020544427324087, + "grad_norm": 2.8743488788604736, + "learning_rate": 1.8766653977921585e-06, + "loss": 1.3808, + "step": 1949 + }, + { + "epoch": 2.0030816640986133, + "grad_norm": 2.8676979541778564, + "learning_rate": 1.874762086029692e-06, + "loss": 1.3394, + "step": 1950 + }, + { + "epoch": 2.0030816640986133, + "eval_loss": 1.8473349809646606, + "eval_runtime": 37.093, + "eval_samples_per_second": 26.959, + "eval_steps_per_second": 1.132, + "step": 1950 + }, + { + "epoch": 2.0041088854648175, + "grad_norm": 3.1344428062438965, + "learning_rate": 1.8728587742672252e-06, + "loss": 1.3774, + "step": 1951 + }, + { + "epoch": 2.005136106831022, + "grad_norm": 3.1567065715789795, + "learning_rate": 1.8709554625047585e-06, + "loss": 1.3863, + "step": 1952 + }, + { + "epoch": 2.0061633281972266, + "grad_norm": 3.109293222427368, + "learning_rate": 1.8690521507422917e-06, + "loss": 1.3196, + "step": 1953 + }, + { + "epoch": 2.007190549563431, + "grad_norm": 3.1035029888153076, + "learning_rate": 1.867148838979825e-06, + "loss": 1.3473, + "step": 1954 + }, + { + "epoch": 2.0082177709296354, + "grad_norm": 3.02809476852417, + "learning_rate": 1.8652455272173584e-06, + "loss": 1.3718, + "step": 1955 + }, + { + "epoch": 2.00924499229584, + "grad_norm": 2.918850898742676, + "learning_rate": 1.8633422154548916e-06, + "loss": 1.3794, + "step": 1956 + }, + { + "epoch": 2.010272213662044, + "grad_norm": 2.89725923538208, + "learning_rate": 1.861438903692425e-06, + "loss": 1.2861, + "step": 1957 + }, + { + "epoch": 2.0112994350282487, + "grad_norm": 3.051868200302124, + "learning_rate": 1.859535591929958e-06, + "loss": 1.3122, + "step": 1958 + }, + { + "epoch": 2.012326656394453, + "grad_norm": 3.1091904640197754, + "learning_rate": 1.8576322801674915e-06, + "loss": 1.3941, + "step": 1959 + }, + { + "epoch": 2.0133538777606574, + "grad_norm": 3.046663761138916, + "learning_rate": 1.855728968405025e-06, + "loss": 1.3293, + "step": 1960 + }, + { + "epoch": 2.0133538777606574, + "eval_loss": 1.851180911064148, + "eval_runtime": 37.0842, + "eval_samples_per_second": 26.966, + "eval_steps_per_second": 1.133, + "step": 1960 + }, + { + "epoch": 2.014381099126862, + "grad_norm": 3.2143752574920654, + "learning_rate": 1.8538256566425584e-06, + "loss": 1.3431, + "step": 1961 + }, + { + "epoch": 2.015408320493066, + "grad_norm": 3.162942409515381, + "learning_rate": 1.8519223448800915e-06, + "loss": 1.3883, + "step": 1962 + }, + { + "epoch": 2.0164355418592708, + "grad_norm": 3.1271183490753174, + "learning_rate": 1.850019033117625e-06, + "loss": 1.3295, + "step": 1963 + }, + { + "epoch": 2.017462763225475, + "grad_norm": 3.07658314704895, + "learning_rate": 1.848115721355158e-06, + "loss": 1.3323, + "step": 1964 + }, + { + "epoch": 2.0184899845916795, + "grad_norm": 3.075946569442749, + "learning_rate": 1.8462124095926914e-06, + "loss": 1.3524, + "step": 1965 + }, + { + "epoch": 2.019517205957884, + "grad_norm": 3.0359086990356445, + "learning_rate": 1.8443090978302248e-06, + "loss": 1.3578, + "step": 1966 + }, + { + "epoch": 2.0205444273240882, + "grad_norm": 3.1957743167877197, + "learning_rate": 1.842405786067758e-06, + "loss": 1.3909, + "step": 1967 + }, + { + "epoch": 2.021571648690293, + "grad_norm": 3.002462863922119, + "learning_rate": 1.8405024743052913e-06, + "loss": 1.3466, + "step": 1968 + }, + { + "epoch": 2.022598870056497, + "grad_norm": 3.0507266521453857, + "learning_rate": 1.8385991625428247e-06, + "loss": 1.3717, + "step": 1969 + }, + { + "epoch": 2.0236260914227016, + "grad_norm": 3.0719404220581055, + "learning_rate": 1.8366958507803578e-06, + "loss": 1.313, + "step": 1970 + }, + { + "epoch": 2.0236260914227016, + "eval_loss": 1.8496923446655273, + "eval_runtime": 37.1112, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 1970 + }, + { + "epoch": 2.024653312788906, + "grad_norm": 3.068819522857666, + "learning_rate": 1.8347925390178914e-06, + "loss": 1.3614, + "step": 1971 + }, + { + "epoch": 2.0256805341551103, + "grad_norm": 3.005855083465576, + "learning_rate": 1.8328892272554247e-06, + "loss": 1.3007, + "step": 1972 + }, + { + "epoch": 2.026707755521315, + "grad_norm": 3.068032741546631, + "learning_rate": 1.8309859154929579e-06, + "loss": 1.3644, + "step": 1973 + }, + { + "epoch": 2.0277349768875195, + "grad_norm": 3.066153049468994, + "learning_rate": 1.8290826037304913e-06, + "loss": 1.3518, + "step": 1974 + }, + { + "epoch": 2.0287621982537236, + "grad_norm": 3.019429922103882, + "learning_rate": 1.8271792919680246e-06, + "loss": 1.3208, + "step": 1975 + }, + { + "epoch": 2.029789419619928, + "grad_norm": 3.2104897499084473, + "learning_rate": 1.8252759802055578e-06, + "loss": 1.329, + "step": 1976 + }, + { + "epoch": 2.0308166409861323, + "grad_norm": 3.272581100463867, + "learning_rate": 1.8233726684430911e-06, + "loss": 1.3665, + "step": 1977 + }, + { + "epoch": 2.031843862352337, + "grad_norm": 3.2155191898345947, + "learning_rate": 1.8214693566806243e-06, + "loss": 1.3178, + "step": 1978 + }, + { + "epoch": 2.0328710837185415, + "grad_norm": 3.130934238433838, + "learning_rate": 1.8195660449181576e-06, + "loss": 1.3492, + "step": 1979 + }, + { + "epoch": 2.0338983050847457, + "grad_norm": 3.211571216583252, + "learning_rate": 1.817662733155691e-06, + "loss": 1.3477, + "step": 1980 + }, + { + "epoch": 2.0338983050847457, + "eval_loss": 1.8516111373901367, + "eval_runtime": 37.1099, + "eval_samples_per_second": 26.947, + "eval_steps_per_second": 1.132, + "step": 1980 + }, + { + "epoch": 2.0349255264509503, + "grad_norm": 2.9799704551696777, + "learning_rate": 1.8157594213932244e-06, + "loss": 1.3469, + "step": 1981 + }, + { + "epoch": 2.0359527478171544, + "grad_norm": 3.0847010612487793, + "learning_rate": 1.8138561096307577e-06, + "loss": 1.3587, + "step": 1982 + }, + { + "epoch": 2.036979969183359, + "grad_norm": 3.0739681720733643, + "learning_rate": 1.811952797868291e-06, + "loss": 1.3053, + "step": 1983 + }, + { + "epoch": 2.0380071905495636, + "grad_norm": 3.0315258502960205, + "learning_rate": 1.8100494861058242e-06, + "loss": 1.3381, + "step": 1984 + }, + { + "epoch": 2.0390344119157677, + "grad_norm": 2.9987258911132812, + "learning_rate": 1.8081461743433576e-06, + "loss": 1.3545, + "step": 1985 + }, + { + "epoch": 2.0400616332819723, + "grad_norm": 2.9383034706115723, + "learning_rate": 1.806242862580891e-06, + "loss": 1.3085, + "step": 1986 + }, + { + "epoch": 2.0410888546481765, + "grad_norm": 3.064450740814209, + "learning_rate": 1.8043395508184241e-06, + "loss": 1.3572, + "step": 1987 + }, + { + "epoch": 2.042116076014381, + "grad_norm": 3.102689027786255, + "learning_rate": 1.8024362390559575e-06, + "loss": 1.3453, + "step": 1988 + }, + { + "epoch": 2.0431432973805856, + "grad_norm": 3.0369763374328613, + "learning_rate": 1.8005329272934906e-06, + "loss": 1.3123, + "step": 1989 + }, + { + "epoch": 2.04417051874679, + "grad_norm": 3.2758162021636963, + "learning_rate": 1.798629615531024e-06, + "loss": 1.4068, + "step": 1990 + }, + { + "epoch": 2.04417051874679, + "eval_loss": 1.8500858545303345, + "eval_runtime": 37.334, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 1990 + }, + { + "epoch": 2.0451977401129944, + "grad_norm": 3.1648876667022705, + "learning_rate": 1.7967263037685576e-06, + "loss": 1.3328, + "step": 1991 + }, + { + "epoch": 2.046224961479199, + "grad_norm": 3.104123830795288, + "learning_rate": 1.794822992006091e-06, + "loss": 1.3297, + "step": 1992 + }, + { + "epoch": 2.047252182845403, + "grad_norm": 3.198483943939209, + "learning_rate": 1.792919680243624e-06, + "loss": 1.3851, + "step": 1993 + }, + { + "epoch": 2.0482794042116077, + "grad_norm": 3.212456226348877, + "learning_rate": 1.7910163684811574e-06, + "loss": 1.3062, + "step": 1994 + }, + { + "epoch": 2.049306625577812, + "grad_norm": 3.058353900909424, + "learning_rate": 1.7891130567186906e-06, + "loss": 1.3347, + "step": 1995 + }, + { + "epoch": 2.0503338469440164, + "grad_norm": 2.963183879852295, + "learning_rate": 1.787209744956224e-06, + "loss": 1.3163, + "step": 1996 + }, + { + "epoch": 2.051361068310221, + "grad_norm": 3.1013290882110596, + "learning_rate": 1.7853064331937573e-06, + "loss": 1.3377, + "step": 1997 + }, + { + "epoch": 2.052388289676425, + "grad_norm": 3.068808078765869, + "learning_rate": 1.7834031214312905e-06, + "loss": 1.3772, + "step": 1998 + }, + { + "epoch": 2.0534155110426298, + "grad_norm": 3.0556447505950928, + "learning_rate": 1.7814998096688238e-06, + "loss": 1.3113, + "step": 1999 + }, + { + "epoch": 2.054442732408834, + "grad_norm": 3.115323543548584, + "learning_rate": 1.779596497906357e-06, + "loss": 1.3499, + "step": 2000 + }, + { + "epoch": 2.054442732408834, + "eval_loss": 1.8396267890930176, + "eval_runtime": 37.4474, + "eval_samples_per_second": 26.704, + "eval_steps_per_second": 1.122, + "step": 2000 + }, + { + "epoch": 2.0554699537750385, + "grad_norm": 3.133096694946289, + "learning_rate": 1.7776931861438906e-06, + "loss": 1.3938, + "step": 2001 + }, + { + "epoch": 2.056497175141243, + "grad_norm": 3.259174108505249, + "learning_rate": 1.775789874381424e-06, + "loss": 1.382, + "step": 2002 + }, + { + "epoch": 2.0575243965074472, + "grad_norm": 3.17948317527771, + "learning_rate": 1.7738865626189573e-06, + "loss": 1.3643, + "step": 2003 + }, + { + "epoch": 2.058551617873652, + "grad_norm": 3.0918309688568115, + "learning_rate": 1.7719832508564904e-06, + "loss": 1.3067, + "step": 2004 + }, + { + "epoch": 2.0595788392398564, + "grad_norm": 2.9949889183044434, + "learning_rate": 1.7700799390940238e-06, + "loss": 1.3323, + "step": 2005 + }, + { + "epoch": 2.0606060606060606, + "grad_norm": 3.044935464859009, + "learning_rate": 1.768176627331557e-06, + "loss": 1.359, + "step": 2006 + }, + { + "epoch": 2.061633281972265, + "grad_norm": 2.9965357780456543, + "learning_rate": 1.7662733155690903e-06, + "loss": 1.3456, + "step": 2007 + }, + { + "epoch": 2.0626605033384693, + "grad_norm": 3.092984676361084, + "learning_rate": 1.7643700038066237e-06, + "loss": 1.3831, + "step": 2008 + }, + { + "epoch": 2.063687724704674, + "grad_norm": 3.004263162612915, + "learning_rate": 1.7624666920441568e-06, + "loss": 1.3087, + "step": 2009 + }, + { + "epoch": 2.0647149460708785, + "grad_norm": 3.0408692359924316, + "learning_rate": 1.7605633802816902e-06, + "loss": 1.3058, + "step": 2010 + }, + { + "epoch": 2.0647149460708785, + "eval_loss": 1.8413820266723633, + "eval_runtime": 36.9354, + "eval_samples_per_second": 27.074, + "eval_steps_per_second": 1.137, + "step": 2010 + }, + { + "epoch": 2.0657421674370826, + "grad_norm": 3.155033826828003, + "learning_rate": 1.7586600685192238e-06, + "loss": 1.394, + "step": 2011 + }, + { + "epoch": 2.066769388803287, + "grad_norm": 3.3466567993164062, + "learning_rate": 1.756756756756757e-06, + "loss": 1.401, + "step": 2012 + }, + { + "epoch": 2.0677966101694913, + "grad_norm": 3.219165802001953, + "learning_rate": 1.7548534449942903e-06, + "loss": 1.3564, + "step": 2013 + }, + { + "epoch": 2.068823831535696, + "grad_norm": 3.1733009815216064, + "learning_rate": 1.7529501332318236e-06, + "loss": 1.3446, + "step": 2014 + }, + { + "epoch": 2.0698510529019005, + "grad_norm": 3.1392745971679688, + "learning_rate": 1.7510468214693568e-06, + "loss": 1.3212, + "step": 2015 + }, + { + "epoch": 2.0708782742681047, + "grad_norm": 2.9845967292785645, + "learning_rate": 1.7491435097068902e-06, + "loss": 1.3117, + "step": 2016 + }, + { + "epoch": 2.0719054956343093, + "grad_norm": 3.2182602882385254, + "learning_rate": 1.7472401979444233e-06, + "loss": 1.3539, + "step": 2017 + }, + { + "epoch": 2.0729327170005134, + "grad_norm": 3.0809013843536377, + "learning_rate": 1.7453368861819567e-06, + "loss": 1.2891, + "step": 2018 + }, + { + "epoch": 2.073959938366718, + "grad_norm": 3.0690958499908447, + "learning_rate": 1.74343357441949e-06, + "loss": 1.3381, + "step": 2019 + }, + { + "epoch": 2.0749871597329226, + "grad_norm": 3.091454267501831, + "learning_rate": 1.7415302626570232e-06, + "loss": 1.3874, + "step": 2020 + }, + { + "epoch": 2.0749871597329226, + "eval_loss": 1.8516818284988403, + "eval_runtime": 37.0619, + "eval_samples_per_second": 26.982, + "eval_steps_per_second": 1.133, + "step": 2020 + }, + { + "epoch": 2.0760143810991267, + "grad_norm": 3.039327621459961, + "learning_rate": 1.7396269508945568e-06, + "loss": 1.3295, + "step": 2021 + }, + { + "epoch": 2.0770416024653313, + "grad_norm": 3.0967671871185303, + "learning_rate": 1.7377236391320901e-06, + "loss": 1.3122, + "step": 2022 + }, + { + "epoch": 2.078068823831536, + "grad_norm": 3.228410005569458, + "learning_rate": 1.7358203273696233e-06, + "loss": 1.3766, + "step": 2023 + }, + { + "epoch": 2.07909604519774, + "grad_norm": 3.1973841190338135, + "learning_rate": 1.7339170156071566e-06, + "loss": 1.3325, + "step": 2024 + }, + { + "epoch": 2.0801232665639446, + "grad_norm": 3.131504535675049, + "learning_rate": 1.73201370384469e-06, + "loss": 1.3355, + "step": 2025 + }, + { + "epoch": 2.081150487930149, + "grad_norm": 3.132685661315918, + "learning_rate": 1.7301103920822231e-06, + "loss": 1.3446, + "step": 2026 + }, + { + "epoch": 2.0821777092963534, + "grad_norm": 3.2344162464141846, + "learning_rate": 1.7282070803197565e-06, + "loss": 1.3525, + "step": 2027 + }, + { + "epoch": 2.083204930662558, + "grad_norm": 3.1193928718566895, + "learning_rate": 1.7263037685572897e-06, + "loss": 1.3415, + "step": 2028 + }, + { + "epoch": 2.084232152028762, + "grad_norm": 3.164365291595459, + "learning_rate": 1.724400456794823e-06, + "loss": 1.3652, + "step": 2029 + }, + { + "epoch": 2.0852593733949667, + "grad_norm": 3.0825226306915283, + "learning_rate": 1.7224971450323564e-06, + "loss": 1.3242, + "step": 2030 + }, + { + "epoch": 2.0852593733949667, + "eval_loss": 1.8433457612991333, + "eval_runtime": 37.0921, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2030 + }, + { + "epoch": 2.086286594761171, + "grad_norm": 3.1651787757873535, + "learning_rate": 1.72059383326989e-06, + "loss": 1.333, + "step": 2031 + }, + { + "epoch": 2.0873138161273754, + "grad_norm": 3.295436382293701, + "learning_rate": 1.7186905215074231e-06, + "loss": 1.4064, + "step": 2032 + }, + { + "epoch": 2.08834103749358, + "grad_norm": 3.177628517150879, + "learning_rate": 1.7167872097449565e-06, + "loss": 1.3173, + "step": 2033 + }, + { + "epoch": 2.089368258859784, + "grad_norm": 3.112971544265747, + "learning_rate": 1.7148838979824896e-06, + "loss": 1.3401, + "step": 2034 + }, + { + "epoch": 2.0903954802259888, + "grad_norm": 3.150045871734619, + "learning_rate": 1.712980586220023e-06, + "loss": 1.3371, + "step": 2035 + }, + { + "epoch": 2.091422701592193, + "grad_norm": 3.0645205974578857, + "learning_rate": 1.7110772744575563e-06, + "loss": 1.3213, + "step": 2036 + }, + { + "epoch": 2.0924499229583975, + "grad_norm": 3.230185031890869, + "learning_rate": 1.7091739626950895e-06, + "loss": 1.3317, + "step": 2037 + }, + { + "epoch": 2.093477144324602, + "grad_norm": 3.1509249210357666, + "learning_rate": 1.7072706509326229e-06, + "loss": 1.3706, + "step": 2038 + }, + { + "epoch": 2.0945043656908062, + "grad_norm": 3.185981273651123, + "learning_rate": 1.7053673391701562e-06, + "loss": 1.3398, + "step": 2039 + }, + { + "epoch": 2.095531587057011, + "grad_norm": 3.000025510787964, + "learning_rate": 1.7034640274076894e-06, + "loss": 1.3197, + "step": 2040 + }, + { + "epoch": 2.095531587057011, + "eval_loss": 1.8413926362991333, + "eval_runtime": 37.0971, + "eval_samples_per_second": 26.956, + "eval_steps_per_second": 1.132, + "step": 2040 + }, + { + "epoch": 2.0965588084232154, + "grad_norm": 3.156095027923584, + "learning_rate": 1.701560715645223e-06, + "loss": 1.3864, + "step": 2041 + }, + { + "epoch": 2.0975860297894195, + "grad_norm": 2.970560073852539, + "learning_rate": 1.6996574038827563e-06, + "loss": 1.3016, + "step": 2042 + }, + { + "epoch": 2.098613251155624, + "grad_norm": 3.141657829284668, + "learning_rate": 1.6977540921202895e-06, + "loss": 1.3488, + "step": 2043 + }, + { + "epoch": 2.0996404725218283, + "grad_norm": 3.125731945037842, + "learning_rate": 1.6958507803578228e-06, + "loss": 1.3275, + "step": 2044 + }, + { + "epoch": 2.100667693888033, + "grad_norm": 3.2703487873077393, + "learning_rate": 1.693947468595356e-06, + "loss": 1.3577, + "step": 2045 + }, + { + "epoch": 2.1016949152542375, + "grad_norm": 3.1855714321136475, + "learning_rate": 1.6920441568328893e-06, + "loss": 1.3671, + "step": 2046 + }, + { + "epoch": 2.1027221366204416, + "grad_norm": 3.0699303150177, + "learning_rate": 1.6901408450704227e-06, + "loss": 1.3066, + "step": 2047 + }, + { + "epoch": 2.103749357986646, + "grad_norm": 3.2417423725128174, + "learning_rate": 1.6882375333079559e-06, + "loss": 1.3274, + "step": 2048 + }, + { + "epoch": 2.1047765793528503, + "grad_norm": 3.265108108520508, + "learning_rate": 1.6863342215454892e-06, + "loss": 1.3924, + "step": 2049 + }, + { + "epoch": 2.105803800719055, + "grad_norm": 3.2078635692596436, + "learning_rate": 1.6844309097830226e-06, + "loss": 1.3782, + "step": 2050 + }, + { + "epoch": 2.105803800719055, + "eval_loss": 1.8441476821899414, + "eval_runtime": 37.1042, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 2050 + }, + { + "epoch": 2.1068310220852595, + "grad_norm": 3.180485248565674, + "learning_rate": 1.682527598020556e-06, + "loss": 1.3613, + "step": 2051 + }, + { + "epoch": 2.1078582434514637, + "grad_norm": 3.110530376434326, + "learning_rate": 1.6806242862580893e-06, + "loss": 1.376, + "step": 2052 + }, + { + "epoch": 2.1088854648176683, + "grad_norm": 2.997999429702759, + "learning_rate": 1.6787209744956227e-06, + "loss": 1.3172, + "step": 2053 + }, + { + "epoch": 2.109912686183873, + "grad_norm": 3.0154783725738525, + "learning_rate": 1.6768176627331558e-06, + "loss": 1.3015, + "step": 2054 + }, + { + "epoch": 2.110939907550077, + "grad_norm": 2.992138624191284, + "learning_rate": 1.6749143509706892e-06, + "loss": 1.3517, + "step": 2055 + }, + { + "epoch": 2.1119671289162816, + "grad_norm": 3.0572457313537598, + "learning_rate": 1.6730110392082225e-06, + "loss": 1.3301, + "step": 2056 + }, + { + "epoch": 2.1129943502824857, + "grad_norm": 2.9924540519714355, + "learning_rate": 1.6711077274457557e-06, + "loss": 1.3534, + "step": 2057 + }, + { + "epoch": 2.1140215716486903, + "grad_norm": 2.9569015502929688, + "learning_rate": 1.669204415683289e-06, + "loss": 1.2972, + "step": 2058 + }, + { + "epoch": 2.115048793014895, + "grad_norm": 3.064371109008789, + "learning_rate": 1.6673011039208222e-06, + "loss": 1.3553, + "step": 2059 + }, + { + "epoch": 2.116076014381099, + "grad_norm": 3.0954525470733643, + "learning_rate": 1.6653977921583556e-06, + "loss": 1.3421, + "step": 2060 + }, + { + "epoch": 2.116076014381099, + "eval_loss": 1.8407723903656006, + "eval_runtime": 37.1122, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 2060 + }, + { + "epoch": 2.1171032357473036, + "grad_norm": 3.2661712169647217, + "learning_rate": 1.663494480395889e-06, + "loss": 1.3582, + "step": 2061 + }, + { + "epoch": 2.118130457113508, + "grad_norm": 3.1663613319396973, + "learning_rate": 1.6615911686334225e-06, + "loss": 1.3203, + "step": 2062 + }, + { + "epoch": 2.1191576784797124, + "grad_norm": 3.0864570140838623, + "learning_rate": 1.6596878568709557e-06, + "loss": 1.3431, + "step": 2063 + }, + { + "epoch": 2.120184899845917, + "grad_norm": 3.1531989574432373, + "learning_rate": 1.657784545108489e-06, + "loss": 1.3275, + "step": 2064 + }, + { + "epoch": 2.121212121212121, + "grad_norm": 3.2258596420288086, + "learning_rate": 1.6558812333460222e-06, + "loss": 1.3711, + "step": 2065 + }, + { + "epoch": 2.1222393425783257, + "grad_norm": 3.2881453037261963, + "learning_rate": 1.6539779215835555e-06, + "loss": 1.3688, + "step": 2066 + }, + { + "epoch": 2.12326656394453, + "grad_norm": 3.0734169483184814, + "learning_rate": 1.652074609821089e-06, + "loss": 1.3277, + "step": 2067 + }, + { + "epoch": 2.1242937853107344, + "grad_norm": 2.9943618774414062, + "learning_rate": 1.650171298058622e-06, + "loss": 1.3268, + "step": 2068 + }, + { + "epoch": 2.125321006676939, + "grad_norm": 3.034625291824341, + "learning_rate": 1.6482679862961554e-06, + "loss": 1.3265, + "step": 2069 + }, + { + "epoch": 2.126348228043143, + "grad_norm": 3.0360612869262695, + "learning_rate": 1.6463646745336886e-06, + "loss": 1.3256, + "step": 2070 + }, + { + "epoch": 2.126348228043143, + "eval_loss": 1.8443471193313599, + "eval_runtime": 37.0917, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2070 + }, + { + "epoch": 2.1273754494093478, + "grad_norm": 3.0917110443115234, + "learning_rate": 1.644461362771222e-06, + "loss": 1.3099, + "step": 2071 + }, + { + "epoch": 2.128402670775552, + "grad_norm": 3.1142187118530273, + "learning_rate": 1.6425580510087555e-06, + "loss": 1.3421, + "step": 2072 + }, + { + "epoch": 2.1294298921417565, + "grad_norm": 3.2128524780273438, + "learning_rate": 1.6406547392462889e-06, + "loss": 1.3123, + "step": 2073 + }, + { + "epoch": 2.130457113507961, + "grad_norm": 3.060520887374878, + "learning_rate": 1.638751427483822e-06, + "loss": 1.3069, + "step": 2074 + }, + { + "epoch": 2.131484334874165, + "grad_norm": 3.092120885848999, + "learning_rate": 1.6368481157213554e-06, + "loss": 1.3537, + "step": 2075 + }, + { + "epoch": 2.13251155624037, + "grad_norm": 3.2490074634552, + "learning_rate": 1.6349448039588885e-06, + "loss": 1.3324, + "step": 2076 + }, + { + "epoch": 2.1335387776065744, + "grad_norm": 3.2517645359039307, + "learning_rate": 1.6330414921964219e-06, + "loss": 1.385, + "step": 2077 + }, + { + "epoch": 2.1345659989727785, + "grad_norm": 3.1669421195983887, + "learning_rate": 1.6311381804339553e-06, + "loss": 1.3054, + "step": 2078 + }, + { + "epoch": 2.135593220338983, + "grad_norm": 3.2323193550109863, + "learning_rate": 1.6292348686714884e-06, + "loss": 1.3488, + "step": 2079 + }, + { + "epoch": 2.1366204417051873, + "grad_norm": 3.1058294773101807, + "learning_rate": 1.6273315569090218e-06, + "loss": 1.3256, + "step": 2080 + }, + { + "epoch": 2.1366204417051873, + "eval_loss": 1.8446909189224243, + "eval_runtime": 37.1231, + "eval_samples_per_second": 26.937, + "eval_steps_per_second": 1.131, + "step": 2080 + }, + { + "epoch": 2.137647663071392, + "grad_norm": 3.1662075519561768, + "learning_rate": 1.625428245146555e-06, + "loss": 1.4073, + "step": 2081 + }, + { + "epoch": 2.1386748844375965, + "grad_norm": 3.083482503890991, + "learning_rate": 1.6235249333840885e-06, + "loss": 1.3897, + "step": 2082 + }, + { + "epoch": 2.1397021058038006, + "grad_norm": 3.040388345718384, + "learning_rate": 1.6216216216216219e-06, + "loss": 1.349, + "step": 2083 + }, + { + "epoch": 2.140729327170005, + "grad_norm": 3.124114513397217, + "learning_rate": 1.6197183098591552e-06, + "loss": 1.3732, + "step": 2084 + }, + { + "epoch": 2.1417565485362093, + "grad_norm": 3.2662036418914795, + "learning_rate": 1.6178149980966884e-06, + "loss": 1.3522, + "step": 2085 + }, + { + "epoch": 2.142783769902414, + "grad_norm": 3.149472951889038, + "learning_rate": 1.6159116863342217e-06, + "loss": 1.3179, + "step": 2086 + }, + { + "epoch": 2.1438109912686185, + "grad_norm": 3.256852865219116, + "learning_rate": 1.6140083745717549e-06, + "loss": 1.3897, + "step": 2087 + }, + { + "epoch": 2.1448382126348227, + "grad_norm": 3.1227493286132812, + "learning_rate": 1.6121050628092882e-06, + "loss": 1.3104, + "step": 2088 + }, + { + "epoch": 2.1458654340010272, + "grad_norm": 3.3213047981262207, + "learning_rate": 1.6102017510468216e-06, + "loss": 1.3829, + "step": 2089 + }, + { + "epoch": 2.146892655367232, + "grad_norm": 3.160428047180176, + "learning_rate": 1.6082984392843548e-06, + "loss": 1.2939, + "step": 2090 + }, + { + "epoch": 2.146892655367232, + "eval_loss": 1.8534585237503052, + "eval_runtime": 37.3333, + "eval_samples_per_second": 26.786, + "eval_steps_per_second": 1.125, + "step": 2090 + }, + { + "epoch": 2.147919876733436, + "grad_norm": 3.224741220474243, + "learning_rate": 1.6063951275218881e-06, + "loss": 1.3418, + "step": 2091 + }, + { + "epoch": 2.1489470980996406, + "grad_norm": 3.3884503841400146, + "learning_rate": 1.6044918157594217e-06, + "loss": 1.395, + "step": 2092 + }, + { + "epoch": 2.1499743194658447, + "grad_norm": 3.1761722564697266, + "learning_rate": 1.6025885039969548e-06, + "loss": 1.3208, + "step": 2093 + }, + { + "epoch": 2.1510015408320493, + "grad_norm": 3.0476465225219727, + "learning_rate": 1.6006851922344882e-06, + "loss": 1.3465, + "step": 2094 + }, + { + "epoch": 2.152028762198254, + "grad_norm": 3.1662650108337402, + "learning_rate": 1.5987818804720216e-06, + "loss": 1.3645, + "step": 2095 + }, + { + "epoch": 2.153055983564458, + "grad_norm": 3.1860709190368652, + "learning_rate": 1.5968785687095547e-06, + "loss": 1.361, + "step": 2096 + }, + { + "epoch": 2.1540832049306626, + "grad_norm": 3.104501962661743, + "learning_rate": 1.594975256947088e-06, + "loss": 1.3135, + "step": 2097 + }, + { + "epoch": 2.1551104262968668, + "grad_norm": 3.067155599594116, + "learning_rate": 1.5930719451846212e-06, + "loss": 1.3693, + "step": 2098 + }, + { + "epoch": 2.1561376476630714, + "grad_norm": 3.195204019546509, + "learning_rate": 1.5911686334221546e-06, + "loss": 1.3143, + "step": 2099 + }, + { + "epoch": 2.157164869029276, + "grad_norm": 3.159278154373169, + "learning_rate": 1.589265321659688e-06, + "loss": 1.3607, + "step": 2100 + }, + { + "epoch": 2.157164869029276, + "eval_loss": 1.8502479791641235, + "eval_runtime": 37.4583, + "eval_samples_per_second": 26.696, + "eval_steps_per_second": 1.121, + "step": 2100 + }, + { + "epoch": 2.15819209039548, + "grad_norm": 2.973191261291504, + "learning_rate": 1.5873620098972211e-06, + "loss": 1.2776, + "step": 2101 + }, + { + "epoch": 2.1592193117616847, + "grad_norm": 3.0078418254852295, + "learning_rate": 1.5854586981347547e-06, + "loss": 1.3173, + "step": 2102 + }, + { + "epoch": 2.1602465331278893, + "grad_norm": 3.2115561962127686, + "learning_rate": 1.583555386372288e-06, + "loss": 1.3621, + "step": 2103 + }, + { + "epoch": 2.1612737544940934, + "grad_norm": 3.234210968017578, + "learning_rate": 1.5816520746098212e-06, + "loss": 1.374, + "step": 2104 + }, + { + "epoch": 2.162300975860298, + "grad_norm": 3.195405960083008, + "learning_rate": 1.5797487628473546e-06, + "loss": 1.3402, + "step": 2105 + }, + { + "epoch": 2.163328197226502, + "grad_norm": 3.016078233718872, + "learning_rate": 1.577845451084888e-06, + "loss": 1.3036, + "step": 2106 + }, + { + "epoch": 2.1643554185927067, + "grad_norm": 3.2147345542907715, + "learning_rate": 1.575942139322421e-06, + "loss": 1.3511, + "step": 2107 + }, + { + "epoch": 2.1653826399589113, + "grad_norm": 2.980320692062378, + "learning_rate": 1.5740388275599544e-06, + "loss": 1.2635, + "step": 2108 + }, + { + "epoch": 2.1664098613251155, + "grad_norm": 3.0693414211273193, + "learning_rate": 1.5721355157974876e-06, + "loss": 1.3435, + "step": 2109 + }, + { + "epoch": 2.16743708269132, + "grad_norm": 3.1445682048797607, + "learning_rate": 1.570232204035021e-06, + "loss": 1.3294, + "step": 2110 + }, + { + "epoch": 2.16743708269132, + "eval_loss": 1.8479558229446411, + "eval_runtime": 36.9467, + "eval_samples_per_second": 27.066, + "eval_steps_per_second": 1.137, + "step": 2110 + }, + { + "epoch": 2.168464304057524, + "grad_norm": 3.1309633255004883, + "learning_rate": 1.5683288922725543e-06, + "loss": 1.3312, + "step": 2111 + }, + { + "epoch": 2.169491525423729, + "grad_norm": 3.0906994342803955, + "learning_rate": 1.5664255805100879e-06, + "loss": 1.342, + "step": 2112 + }, + { + "epoch": 2.1705187467899334, + "grad_norm": 3.1987171173095703, + "learning_rate": 1.564522268747621e-06, + "loss": 1.3358, + "step": 2113 + }, + { + "epoch": 2.1715459681561375, + "grad_norm": 3.221889019012451, + "learning_rate": 1.5626189569851544e-06, + "loss": 1.3484, + "step": 2114 + }, + { + "epoch": 2.172573189522342, + "grad_norm": 3.1508629322052, + "learning_rate": 1.5607156452226875e-06, + "loss": 1.2785, + "step": 2115 + }, + { + "epoch": 2.1736004108885463, + "grad_norm": 3.115274667739868, + "learning_rate": 1.558812333460221e-06, + "loss": 1.3463, + "step": 2116 + }, + { + "epoch": 2.174627632254751, + "grad_norm": 3.0454249382019043, + "learning_rate": 1.5569090216977543e-06, + "loss": 1.3534, + "step": 2117 + }, + { + "epoch": 2.1756548536209555, + "grad_norm": 3.20078182220459, + "learning_rate": 1.5550057099352874e-06, + "loss": 1.3469, + "step": 2118 + }, + { + "epoch": 2.1766820749871596, + "grad_norm": 3.091568946838379, + "learning_rate": 1.5531023981728208e-06, + "loss": 1.3547, + "step": 2119 + }, + { + "epoch": 2.177709296353364, + "grad_norm": 3.0242369174957275, + "learning_rate": 1.5511990864103542e-06, + "loss": 1.3198, + "step": 2120 + }, + { + "epoch": 2.177709296353364, + "eval_loss": 1.8469057083129883, + "eval_runtime": 37.0881, + "eval_samples_per_second": 26.963, + "eval_steps_per_second": 1.132, + "step": 2120 + }, + { + "epoch": 2.1787365177195683, + "grad_norm": 3.1775739192962646, + "learning_rate": 1.5492957746478873e-06, + "loss": 1.3491, + "step": 2121 + }, + { + "epoch": 2.179763739085773, + "grad_norm": 3.239668846130371, + "learning_rate": 1.5473924628854209e-06, + "loss": 1.3641, + "step": 2122 + }, + { + "epoch": 2.1807909604519775, + "grad_norm": 3.194678544998169, + "learning_rate": 1.5454891511229542e-06, + "loss": 1.3317, + "step": 2123 + }, + { + "epoch": 2.1818181818181817, + "grad_norm": 3.129948377609253, + "learning_rate": 1.5435858393604874e-06, + "loss": 1.3405, + "step": 2124 + }, + { + "epoch": 2.1828454031843862, + "grad_norm": 3.2040436267852783, + "learning_rate": 1.5416825275980208e-06, + "loss": 1.3596, + "step": 2125 + }, + { + "epoch": 2.183872624550591, + "grad_norm": 3.2529456615448, + "learning_rate": 1.539779215835554e-06, + "loss": 1.3171, + "step": 2126 + }, + { + "epoch": 2.184899845916795, + "grad_norm": 3.2071499824523926, + "learning_rate": 1.5378759040730873e-06, + "loss": 1.318, + "step": 2127 + }, + { + "epoch": 2.1859270672829996, + "grad_norm": 3.2709555625915527, + "learning_rate": 1.5359725923106206e-06, + "loss": 1.3811, + "step": 2128 + }, + { + "epoch": 2.1869542886492037, + "grad_norm": 3.244680643081665, + "learning_rate": 1.5340692805481538e-06, + "loss": 1.3539, + "step": 2129 + }, + { + "epoch": 2.1879815100154083, + "grad_norm": 3.068683624267578, + "learning_rate": 1.5321659687856871e-06, + "loss": 1.3207, + "step": 2130 + }, + { + "epoch": 2.1879815100154083, + "eval_loss": 1.8478527069091797, + "eval_runtime": 37.138, + "eval_samples_per_second": 26.927, + "eval_steps_per_second": 1.131, + "step": 2130 + }, + { + "epoch": 2.189008731381613, + "grad_norm": 3.1829938888549805, + "learning_rate": 1.5302626570232205e-06, + "loss": 1.3127, + "step": 2131 + }, + { + "epoch": 2.190035952747817, + "grad_norm": 3.060347080230713, + "learning_rate": 1.5283593452607539e-06, + "loss": 1.287, + "step": 2132 + }, + { + "epoch": 2.1910631741140216, + "grad_norm": 3.3016905784606934, + "learning_rate": 1.5264560334982872e-06, + "loss": 1.3838, + "step": 2133 + }, + { + "epoch": 2.1920903954802258, + "grad_norm": 3.1945807933807373, + "learning_rate": 1.5245527217358206e-06, + "loss": 1.3311, + "step": 2134 + }, + { + "epoch": 2.1931176168464304, + "grad_norm": 3.182347059249878, + "learning_rate": 1.5226494099733537e-06, + "loss": 1.3544, + "step": 2135 + }, + { + "epoch": 2.194144838212635, + "grad_norm": 3.069206476211548, + "learning_rate": 1.520746098210887e-06, + "loss": 1.3148, + "step": 2136 + }, + { + "epoch": 2.195172059578839, + "grad_norm": 3.1632168292999268, + "learning_rate": 1.5188427864484205e-06, + "loss": 1.3857, + "step": 2137 + }, + { + "epoch": 2.1961992809450437, + "grad_norm": 3.055793523788452, + "learning_rate": 1.5169394746859536e-06, + "loss": 1.3116, + "step": 2138 + }, + { + "epoch": 2.1972265023112483, + "grad_norm": 3.1330678462982178, + "learning_rate": 1.515036162923487e-06, + "loss": 1.3214, + "step": 2139 + }, + { + "epoch": 2.1982537236774524, + "grad_norm": 3.118450880050659, + "learning_rate": 1.5131328511610201e-06, + "loss": 1.3479, + "step": 2140 + }, + { + "epoch": 2.1982537236774524, + "eval_loss": 1.8538312911987305, + "eval_runtime": 37.122, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 2140 + }, + { + "epoch": 2.199280945043657, + "grad_norm": 3.249545097351074, + "learning_rate": 1.5112295393985535e-06, + "loss": 1.379, + "step": 2141 + }, + { + "epoch": 2.200308166409861, + "grad_norm": 3.1331286430358887, + "learning_rate": 1.509326227636087e-06, + "loss": 1.3621, + "step": 2142 + }, + { + "epoch": 2.2013353877760657, + "grad_norm": 3.150987148284912, + "learning_rate": 1.5074229158736204e-06, + "loss": 1.291, + "step": 2143 + }, + { + "epoch": 2.2023626091422703, + "grad_norm": 3.1020572185516357, + "learning_rate": 1.5055196041111536e-06, + "loss": 1.3381, + "step": 2144 + }, + { + "epoch": 2.2033898305084745, + "grad_norm": 3.129390239715576, + "learning_rate": 1.503616292348687e-06, + "loss": 1.3405, + "step": 2145 + }, + { + "epoch": 2.204417051874679, + "grad_norm": 3.2193870544433594, + "learning_rate": 1.50171298058622e-06, + "loss": 1.3744, + "step": 2146 + }, + { + "epoch": 2.205444273240883, + "grad_norm": 3.043828248977661, + "learning_rate": 1.4998096688237535e-06, + "loss": 1.2782, + "step": 2147 + }, + { + "epoch": 2.206471494607088, + "grad_norm": 3.2130300998687744, + "learning_rate": 1.4979063570612868e-06, + "loss": 1.3625, + "step": 2148 + }, + { + "epoch": 2.2074987159732924, + "grad_norm": 3.250370979309082, + "learning_rate": 1.49600304529882e-06, + "loss": 1.3396, + "step": 2149 + }, + { + "epoch": 2.2085259373394965, + "grad_norm": 3.0954599380493164, + "learning_rate": 1.4940997335363533e-06, + "loss": 1.3187, + "step": 2150 + }, + { + "epoch": 2.2085259373394965, + "eval_loss": 1.8469802141189575, + "eval_runtime": 37.2109, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 2150 + }, + { + "epoch": 2.209553158705701, + "grad_norm": 3.153545379638672, + "learning_rate": 1.4921964217738865e-06, + "loss": 1.3488, + "step": 2151 + }, + { + "epoch": 2.2105803800719057, + "grad_norm": 3.1206071376800537, + "learning_rate": 1.49029311001142e-06, + "loss": 1.336, + "step": 2152 + }, + { + "epoch": 2.21160760143811, + "grad_norm": 3.151784896850586, + "learning_rate": 1.4883897982489534e-06, + "loss": 1.3613, + "step": 2153 + }, + { + "epoch": 2.2126348228043144, + "grad_norm": 3.18965744972229, + "learning_rate": 1.4864864864864868e-06, + "loss": 1.3698, + "step": 2154 + }, + { + "epoch": 2.2136620441705186, + "grad_norm": 3.108281135559082, + "learning_rate": 1.48458317472402e-06, + "loss": 1.3287, + "step": 2155 + }, + { + "epoch": 2.214689265536723, + "grad_norm": 3.2309465408325195, + "learning_rate": 1.4826798629615533e-06, + "loss": 1.3425, + "step": 2156 + }, + { + "epoch": 2.2157164869029278, + "grad_norm": 3.0829577445983887, + "learning_rate": 1.4807765511990864e-06, + "loss": 1.2917, + "step": 2157 + }, + { + "epoch": 2.216743708269132, + "grad_norm": 3.1417033672332764, + "learning_rate": 1.4788732394366198e-06, + "loss": 1.3113, + "step": 2158 + }, + { + "epoch": 2.2177709296353365, + "grad_norm": 3.1996822357177734, + "learning_rate": 1.4769699276741532e-06, + "loss": 1.3248, + "step": 2159 + }, + { + "epoch": 2.2187981510015407, + "grad_norm": 3.262814998626709, + "learning_rate": 1.4750666159116863e-06, + "loss": 1.3186, + "step": 2160 + }, + { + "epoch": 2.2187981510015407, + "eval_loss": 1.844687819480896, + "eval_runtime": 37.2211, + "eval_samples_per_second": 26.866, + "eval_steps_per_second": 1.128, + "step": 2160 + }, + { + "epoch": 2.2198253723677452, + "grad_norm": 3.0836942195892334, + "learning_rate": 1.4731633041492197e-06, + "loss": 1.2935, + "step": 2161 + }, + { + "epoch": 2.22085259373395, + "grad_norm": 3.1720023155212402, + "learning_rate": 1.4712599923867528e-06, + "loss": 1.3622, + "step": 2162 + }, + { + "epoch": 2.221879815100154, + "grad_norm": 3.2829599380493164, + "learning_rate": 1.4693566806242864e-06, + "loss": 1.3794, + "step": 2163 + }, + { + "epoch": 2.2229070364663586, + "grad_norm": 3.045949935913086, + "learning_rate": 1.4674533688618198e-06, + "loss": 1.3054, + "step": 2164 + }, + { + "epoch": 2.2239342578325627, + "grad_norm": 3.1561572551727295, + "learning_rate": 1.4655500570993531e-06, + "loss": 1.3289, + "step": 2165 + }, + { + "epoch": 2.2249614791987673, + "grad_norm": 3.0877439975738525, + "learning_rate": 1.4636467453368863e-06, + "loss": 1.329, + "step": 2166 + }, + { + "epoch": 2.225988700564972, + "grad_norm": 3.1330180168151855, + "learning_rate": 1.4617434335744197e-06, + "loss": 1.3853, + "step": 2167 + }, + { + "epoch": 2.227015921931176, + "grad_norm": 3.231693744659424, + "learning_rate": 1.4598401218119528e-06, + "loss": 1.378, + "step": 2168 + }, + { + "epoch": 2.2280431432973806, + "grad_norm": 3.202930212020874, + "learning_rate": 1.4579368100494862e-06, + "loss": 1.3549, + "step": 2169 + }, + { + "epoch": 2.2290703646635848, + "grad_norm": 3.164461135864258, + "learning_rate": 1.4560334982870195e-06, + "loss": 1.3365, + "step": 2170 + }, + { + "epoch": 2.2290703646635848, + "eval_loss": 1.8483937978744507, + "eval_runtime": 37.1839, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.13, + "step": 2170 + }, + { + "epoch": 2.2300975860297894, + "grad_norm": 3.1866867542266846, + "learning_rate": 1.4541301865245527e-06, + "loss": 1.3503, + "step": 2171 + }, + { + "epoch": 2.231124807395994, + "grad_norm": 3.2202508449554443, + "learning_rate": 1.452226874762086e-06, + "loss": 1.354, + "step": 2172 + }, + { + "epoch": 2.232152028762198, + "grad_norm": 3.190744161605835, + "learning_rate": 1.4503235629996196e-06, + "loss": 1.3073, + "step": 2173 + }, + { + "epoch": 2.2331792501284027, + "grad_norm": 3.2158405780792236, + "learning_rate": 1.4484202512371528e-06, + "loss": 1.3731, + "step": 2174 + }, + { + "epoch": 2.2342064714946073, + "grad_norm": 3.0294954776763916, + "learning_rate": 1.4465169394746861e-06, + "loss": 1.2772, + "step": 2175 + }, + { + "epoch": 2.2352336928608114, + "grad_norm": 3.2871217727661133, + "learning_rate": 1.4446136277122195e-06, + "loss": 1.3888, + "step": 2176 + }, + { + "epoch": 2.236260914227016, + "grad_norm": 3.0494141578674316, + "learning_rate": 1.4427103159497526e-06, + "loss": 1.2991, + "step": 2177 + }, + { + "epoch": 2.23728813559322, + "grad_norm": 3.122699022293091, + "learning_rate": 1.440807004187286e-06, + "loss": 1.3186, + "step": 2178 + }, + { + "epoch": 2.2383153569594247, + "grad_norm": 3.119854211807251, + "learning_rate": 1.4389036924248192e-06, + "loss": 1.3421, + "step": 2179 + }, + { + "epoch": 2.2393425783256293, + "grad_norm": 3.2588119506835938, + "learning_rate": 1.4370003806623525e-06, + "loss": 1.3657, + "step": 2180 + }, + { + "epoch": 2.2393425783256293, + "eval_loss": 1.8473975658416748, + "eval_runtime": 37.1561, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 2180 + }, + { + "epoch": 2.2403697996918335, + "grad_norm": 3.138901948928833, + "learning_rate": 1.4350970688998859e-06, + "loss": 1.3273, + "step": 2181 + }, + { + "epoch": 2.241397021058038, + "grad_norm": 3.2792508602142334, + "learning_rate": 1.433193757137419e-06, + "loss": 1.3763, + "step": 2182 + }, + { + "epoch": 2.242424242424242, + "grad_norm": 3.065627098083496, + "learning_rate": 1.4312904453749526e-06, + "loss": 1.3292, + "step": 2183 + }, + { + "epoch": 2.243451463790447, + "grad_norm": 3.101377010345459, + "learning_rate": 1.429387133612486e-06, + "loss": 1.2957, + "step": 2184 + }, + { + "epoch": 2.2444786851566514, + "grad_norm": 3.140158176422119, + "learning_rate": 1.4274838218500191e-06, + "loss": 1.3406, + "step": 2185 + }, + { + "epoch": 2.2455059065228555, + "grad_norm": 3.2498116493225098, + "learning_rate": 1.4255805100875525e-06, + "loss": 1.3499, + "step": 2186 + }, + { + "epoch": 2.24653312788906, + "grad_norm": 3.2681965827941895, + "learning_rate": 1.4236771983250858e-06, + "loss": 1.407, + "step": 2187 + }, + { + "epoch": 2.2475603492552647, + "grad_norm": 3.2457778453826904, + "learning_rate": 1.421773886562619e-06, + "loss": 1.3278, + "step": 2188 + }, + { + "epoch": 2.248587570621469, + "grad_norm": 3.099942207336426, + "learning_rate": 1.4198705748001524e-06, + "loss": 1.3114, + "step": 2189 + }, + { + "epoch": 2.2496147919876734, + "grad_norm": 3.090271234512329, + "learning_rate": 1.4179672630376855e-06, + "loss": 1.3285, + "step": 2190 + }, + { + "epoch": 2.2496147919876734, + "eval_loss": 1.8479118347167969, + "eval_runtime": 37.2331, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 2190 + }, + { + "epoch": 2.2506420133538776, + "grad_norm": 3.0344860553741455, + "learning_rate": 1.4160639512752189e-06, + "loss": 1.3104, + "step": 2191 + }, + { + "epoch": 2.251669234720082, + "grad_norm": 3.087860584259033, + "learning_rate": 1.4141606395127522e-06, + "loss": 1.3167, + "step": 2192 + }, + { + "epoch": 2.2526964560862868, + "grad_norm": 3.0974831581115723, + "learning_rate": 1.4122573277502858e-06, + "loss": 1.2926, + "step": 2193 + }, + { + "epoch": 2.253723677452491, + "grad_norm": 3.228492498397827, + "learning_rate": 1.410354015987819e-06, + "loss": 1.3541, + "step": 2194 + }, + { + "epoch": 2.2547508988186955, + "grad_norm": 3.1761956214904785, + "learning_rate": 1.4084507042253523e-06, + "loss": 1.3342, + "step": 2195 + }, + { + "epoch": 2.2557781201848996, + "grad_norm": 3.1890695095062256, + "learning_rate": 1.4065473924628855e-06, + "loss": 1.362, + "step": 2196 + }, + { + "epoch": 2.2568053415511042, + "grad_norm": 3.1543149948120117, + "learning_rate": 1.4046440807004188e-06, + "loss": 1.329, + "step": 2197 + }, + { + "epoch": 2.257832562917309, + "grad_norm": 3.2313342094421387, + "learning_rate": 1.4027407689379522e-06, + "loss": 1.3256, + "step": 2198 + }, + { + "epoch": 2.258859784283513, + "grad_norm": 3.130805253982544, + "learning_rate": 1.4008374571754854e-06, + "loss": 1.3047, + "step": 2199 + }, + { + "epoch": 2.2598870056497176, + "grad_norm": 3.0488216876983643, + "learning_rate": 1.3989341454130187e-06, + "loss": 1.3461, + "step": 2200 + }, + { + "epoch": 2.2598870056497176, + "eval_loss": 1.8517944812774658, + "eval_runtime": 37.4528, + "eval_samples_per_second": 26.7, + "eval_steps_per_second": 1.121, + "step": 2200 + }, + { + "epoch": 2.260914227015922, + "grad_norm": 3.0915918350219727, + "learning_rate": 1.397030833650552e-06, + "loss": 1.3391, + "step": 2201 + }, + { + "epoch": 2.2619414483821263, + "grad_norm": 3.156053066253662, + "learning_rate": 1.3951275218880852e-06, + "loss": 1.3525, + "step": 2202 + }, + { + "epoch": 2.262968669748331, + "grad_norm": 3.4035868644714355, + "learning_rate": 1.3932242101256188e-06, + "loss": 1.4122, + "step": 2203 + }, + { + "epoch": 2.263995891114535, + "grad_norm": 2.9787867069244385, + "learning_rate": 1.3913208983631522e-06, + "loss": 1.3096, + "step": 2204 + }, + { + "epoch": 2.2650231124807396, + "grad_norm": 3.154271364212036, + "learning_rate": 1.3894175866006853e-06, + "loss": 1.3421, + "step": 2205 + }, + { + "epoch": 2.266050333846944, + "grad_norm": 3.1899831295013428, + "learning_rate": 1.3875142748382187e-06, + "loss": 1.2949, + "step": 2206 + }, + { + "epoch": 2.2670775552131484, + "grad_norm": 3.126694679260254, + "learning_rate": 1.385610963075752e-06, + "loss": 1.2898, + "step": 2207 + }, + { + "epoch": 2.268104776579353, + "grad_norm": 3.149989128112793, + "learning_rate": 1.3837076513132852e-06, + "loss": 1.3369, + "step": 2208 + }, + { + "epoch": 2.269131997945557, + "grad_norm": 3.250481128692627, + "learning_rate": 1.3818043395508186e-06, + "loss": 1.3605, + "step": 2209 + }, + { + "epoch": 2.2701592193117617, + "grad_norm": 3.2549924850463867, + "learning_rate": 1.3799010277883517e-06, + "loss": 1.3428, + "step": 2210 + }, + { + "epoch": 2.2701592193117617, + "eval_loss": 1.8562498092651367, + "eval_runtime": 37.0713, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 2210 + }, + { + "epoch": 2.2711864406779663, + "grad_norm": 3.1369433403015137, + "learning_rate": 1.377997716025885e-06, + "loss": 1.333, + "step": 2211 + }, + { + "epoch": 2.2722136620441704, + "grad_norm": 3.2522025108337402, + "learning_rate": 1.3760944042634184e-06, + "loss": 1.3496, + "step": 2212 + }, + { + "epoch": 2.273240883410375, + "grad_norm": 3.225821018218994, + "learning_rate": 1.3741910925009518e-06, + "loss": 1.3716, + "step": 2213 + }, + { + "epoch": 2.2742681047765796, + "grad_norm": 3.1488454341888428, + "learning_rate": 1.3722877807384852e-06, + "loss": 1.3508, + "step": 2214 + }, + { + "epoch": 2.2752953261427837, + "grad_norm": 3.258491039276123, + "learning_rate": 1.3703844689760185e-06, + "loss": 1.3565, + "step": 2215 + }, + { + "epoch": 2.2763225475089883, + "grad_norm": 3.1634750366210938, + "learning_rate": 1.3684811572135517e-06, + "loss": 1.3269, + "step": 2216 + }, + { + "epoch": 2.2773497688751925, + "grad_norm": 3.1604928970336914, + "learning_rate": 1.366577845451085e-06, + "loss": 1.2832, + "step": 2217 + }, + { + "epoch": 2.278376990241397, + "grad_norm": 3.2551920413970947, + "learning_rate": 1.3646745336886184e-06, + "loss": 1.3567, + "step": 2218 + }, + { + "epoch": 2.279404211607601, + "grad_norm": 3.392820119857788, + "learning_rate": 1.3627712219261515e-06, + "loss": 1.3601, + "step": 2219 + }, + { + "epoch": 2.280431432973806, + "grad_norm": 3.057724952697754, + "learning_rate": 1.360867910163685e-06, + "loss": 1.3138, + "step": 2220 + }, + { + "epoch": 2.280431432973806, + "eval_loss": 1.848382830619812, + "eval_runtime": 37.2255, + "eval_samples_per_second": 26.863, + "eval_steps_per_second": 1.128, + "step": 2220 + }, + { + "epoch": 2.2814586543400104, + "grad_norm": 3.2199649810791016, + "learning_rate": 1.358964598401218e-06, + "loss": 1.3338, + "step": 2221 + }, + { + "epoch": 2.2824858757062145, + "grad_norm": 3.252066135406494, + "learning_rate": 1.3570612866387514e-06, + "loss": 1.3413, + "step": 2222 + }, + { + "epoch": 2.283513097072419, + "grad_norm": 3.038052558898926, + "learning_rate": 1.355157974876285e-06, + "loss": 1.3268, + "step": 2223 + }, + { + "epoch": 2.2845403184386237, + "grad_norm": 3.2332992553710938, + "learning_rate": 1.3532546631138184e-06, + "loss": 1.3866, + "step": 2224 + }, + { + "epoch": 2.285567539804828, + "grad_norm": 3.14321231842041, + "learning_rate": 1.3513513513513515e-06, + "loss": 1.3466, + "step": 2225 + }, + { + "epoch": 2.2865947611710324, + "grad_norm": 3.0714094638824463, + "learning_rate": 1.3494480395888849e-06, + "loss": 1.331, + "step": 2226 + }, + { + "epoch": 2.2876219825372366, + "grad_norm": 3.1167569160461426, + "learning_rate": 1.347544727826418e-06, + "loss": 1.3242, + "step": 2227 + }, + { + "epoch": 2.288649203903441, + "grad_norm": 3.072951316833496, + "learning_rate": 1.3456414160639514e-06, + "loss": 1.3139, + "step": 2228 + }, + { + "epoch": 2.2896764252696458, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.3524, + "step": 2229 + }, + { + "epoch": 2.29070364663585, + "grad_norm": 3.104212999343872, + "learning_rate": 1.3437381043014847e-06, + "loss": 1.334, + "step": 2230 + }, + { + "epoch": 2.29070364663585, + "eval_loss": 1.848712682723999, + "eval_runtime": 37.2687, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2230 + }, + { + "epoch": 2.2917308680020545, + "grad_norm": 3.056426525115967, + "learning_rate": 1.341834792539018e-06, + "loss": 1.3344, + "step": 2231 + }, + { + "epoch": 2.2927580893682586, + "grad_norm": 3.2412500381469727, + "learning_rate": 1.3399314807765513e-06, + "loss": 1.3416, + "step": 2232 + }, + { + "epoch": 2.2937853107344632, + "grad_norm": 3.178572416305542, + "learning_rate": 1.3380281690140844e-06, + "loss": 1.3472, + "step": 2233 + }, + { + "epoch": 2.294812532100668, + "grad_norm": 3.2567203044891357, + "learning_rate": 1.336124857251618e-06, + "loss": 1.311, + "step": 2234 + }, + { + "epoch": 2.295839753466872, + "grad_norm": 3.2018561363220215, + "learning_rate": 1.3342215454891513e-06, + "loss": 1.301, + "step": 2235 + }, + { + "epoch": 2.2968669748330766, + "grad_norm": 3.0722200870513916, + "learning_rate": 1.3323182337266847e-06, + "loss": 1.3325, + "step": 2236 + }, + { + "epoch": 2.297894196199281, + "grad_norm": 3.2170779705047607, + "learning_rate": 1.3304149219642179e-06, + "loss": 1.3752, + "step": 2237 + }, + { + "epoch": 2.2989214175654853, + "grad_norm": 3.1309115886688232, + "learning_rate": 1.3285116102017512e-06, + "loss": 1.3261, + "step": 2238 + }, + { + "epoch": 2.29994863893169, + "grad_norm": 3.1463022232055664, + "learning_rate": 1.3266082984392844e-06, + "loss": 1.3316, + "step": 2239 + }, + { + "epoch": 2.300975860297894, + "grad_norm": 3.174701690673828, + "learning_rate": 1.3247049866768177e-06, + "loss": 1.3148, + "step": 2240 + }, + { + "epoch": 2.300975860297894, + "eval_loss": 1.8436999320983887, + "eval_runtime": 37.2577, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 2240 + }, + { + "epoch": 2.3020030816640986, + "grad_norm": 3.18149471282959, + "learning_rate": 1.322801674914351e-06, + "loss": 1.312, + "step": 2241 + }, + { + "epoch": 2.303030303030303, + "grad_norm": 3.213361978530884, + "learning_rate": 1.3208983631518843e-06, + "loss": 1.3406, + "step": 2242 + }, + { + "epoch": 2.3040575243965074, + "grad_norm": 3.128354787826538, + "learning_rate": 1.3189950513894176e-06, + "loss": 1.3286, + "step": 2243 + }, + { + "epoch": 2.305084745762712, + "grad_norm": 3.210330009460449, + "learning_rate": 1.3170917396269512e-06, + "loss": 1.3406, + "step": 2244 + }, + { + "epoch": 2.306111967128916, + "grad_norm": 3.187161445617676, + "learning_rate": 1.3151884278644843e-06, + "loss": 1.3621, + "step": 2245 + }, + { + "epoch": 2.3071391884951207, + "grad_norm": 3.108806848526001, + "learning_rate": 1.3132851161020177e-06, + "loss": 1.3094, + "step": 2246 + }, + { + "epoch": 2.3081664098613253, + "grad_norm": 3.336510181427002, + "learning_rate": 1.311381804339551e-06, + "loss": 1.3998, + "step": 2247 + }, + { + "epoch": 2.3091936312275294, + "grad_norm": 3.159184455871582, + "learning_rate": 1.3094784925770842e-06, + "loss": 1.3082, + "step": 2248 + }, + { + "epoch": 2.310220852593734, + "grad_norm": 3.128321409225464, + "learning_rate": 1.3075751808146176e-06, + "loss": 1.3494, + "step": 2249 + }, + { + "epoch": 2.3112480739599386, + "grad_norm": 3.122690439224243, + "learning_rate": 1.3056718690521507e-06, + "loss": 1.3064, + "step": 2250 + }, + { + "epoch": 2.3112480739599386, + "eval_loss": 1.848455786705017, + "eval_runtime": 37.2647, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 2250 + }, + { + "epoch": 2.3122752953261427, + "grad_norm": 3.1642186641693115, + "learning_rate": 1.303768557289684e-06, + "loss": 1.3134, + "step": 2251 + }, + { + "epoch": 2.3133025166923473, + "grad_norm": 3.1710383892059326, + "learning_rate": 1.3018652455272175e-06, + "loss": 1.3126, + "step": 2252 + }, + { + "epoch": 2.3143297380585515, + "grad_norm": 3.1513214111328125, + "learning_rate": 1.2999619337647506e-06, + "loss": 1.3218, + "step": 2253 + }, + { + "epoch": 2.315356959424756, + "grad_norm": 3.1538784503936768, + "learning_rate": 1.2980586220022842e-06, + "loss": 1.3473, + "step": 2254 + }, + { + "epoch": 2.3163841807909606, + "grad_norm": 3.2409846782684326, + "learning_rate": 1.2961553102398175e-06, + "loss": 1.371, + "step": 2255 + }, + { + "epoch": 2.317411402157165, + "grad_norm": 3.3224987983703613, + "learning_rate": 1.2942519984773507e-06, + "loss": 1.3506, + "step": 2256 + }, + { + "epoch": 2.3184386235233694, + "grad_norm": 3.1684653759002686, + "learning_rate": 1.292348686714884e-06, + "loss": 1.3127, + "step": 2257 + }, + { + "epoch": 2.3194658448895735, + "grad_norm": 3.1784262657165527, + "learning_rate": 1.2904453749524174e-06, + "loss": 1.3541, + "step": 2258 + }, + { + "epoch": 2.320493066255778, + "grad_norm": 3.1121561527252197, + "learning_rate": 1.2885420631899506e-06, + "loss": 1.2843, + "step": 2259 + }, + { + "epoch": 2.3215202876219827, + "grad_norm": 3.200490951538086, + "learning_rate": 1.286638751427484e-06, + "loss": 1.3191, + "step": 2260 + }, + { + "epoch": 2.3215202876219827, + "eval_loss": 1.8469436168670654, + "eval_runtime": 37.2695, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 2260 + }, + { + "epoch": 2.322547508988187, + "grad_norm": 3.143329620361328, + "learning_rate": 1.284735439665017e-06, + "loss": 1.3454, + "step": 2261 + }, + { + "epoch": 2.3235747303543914, + "grad_norm": 3.1955461502075195, + "learning_rate": 1.2828321279025504e-06, + "loss": 1.3505, + "step": 2262 + }, + { + "epoch": 2.324601951720596, + "grad_norm": 3.3297348022460938, + "learning_rate": 1.2809288161400838e-06, + "loss": 1.3374, + "step": 2263 + }, + { + "epoch": 2.3256291730868, + "grad_norm": 3.144385814666748, + "learning_rate": 1.279025504377617e-06, + "loss": 1.3198, + "step": 2264 + }, + { + "epoch": 2.3266563944530048, + "grad_norm": 3.2952210903167725, + "learning_rate": 1.2771221926151505e-06, + "loss": 1.3744, + "step": 2265 + }, + { + "epoch": 2.327683615819209, + "grad_norm": 3.212287187576294, + "learning_rate": 1.275218880852684e-06, + "loss": 1.3173, + "step": 2266 + }, + { + "epoch": 2.3287108371854135, + "grad_norm": 3.235272169113159, + "learning_rate": 1.273315569090217e-06, + "loss": 1.3231, + "step": 2267 + }, + { + "epoch": 2.3297380585516176, + "grad_norm": 3.2961113452911377, + "learning_rate": 1.2714122573277504e-06, + "loss": 1.3909, + "step": 2268 + }, + { + "epoch": 2.3307652799178222, + "grad_norm": 3.13678240776062, + "learning_rate": 1.2695089455652838e-06, + "loss": 1.3911, + "step": 2269 + }, + { + "epoch": 2.331792501284027, + "grad_norm": 3.166139841079712, + "learning_rate": 1.267605633802817e-06, + "loss": 1.3496, + "step": 2270 + }, + { + "epoch": 2.331792501284027, + "eval_loss": 1.8424044847488403, + "eval_runtime": 37.128, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 2270 + }, + { + "epoch": 2.332819722650231, + "grad_norm": 3.20859432220459, + "learning_rate": 1.2657023220403503e-06, + "loss": 1.3402, + "step": 2271 + }, + { + "epoch": 2.3338469440164356, + "grad_norm": 3.18941593170166, + "learning_rate": 1.2637990102778836e-06, + "loss": 1.3272, + "step": 2272 + }, + { + "epoch": 2.33487416538264, + "grad_norm": 3.2382333278656006, + "learning_rate": 1.2618956985154168e-06, + "loss": 1.3242, + "step": 2273 + }, + { + "epoch": 2.3359013867488443, + "grad_norm": 3.2496516704559326, + "learning_rate": 1.2599923867529502e-06, + "loss": 1.3675, + "step": 2274 + }, + { + "epoch": 2.336928608115049, + "grad_norm": 3.1282081604003906, + "learning_rate": 1.2580890749904837e-06, + "loss": 1.3343, + "step": 2275 + }, + { + "epoch": 2.337955829481253, + "grad_norm": 3.34047794342041, + "learning_rate": 1.2561857632280169e-06, + "loss": 1.3651, + "step": 2276 + }, + { + "epoch": 2.3389830508474576, + "grad_norm": 3.3369057178497314, + "learning_rate": 1.2542824514655502e-06, + "loss": 1.3944, + "step": 2277 + }, + { + "epoch": 2.340010272213662, + "grad_norm": 3.282226085662842, + "learning_rate": 1.2523791397030834e-06, + "loss": 1.3595, + "step": 2278 + }, + { + "epoch": 2.3410374935798663, + "grad_norm": 3.111595630645752, + "learning_rate": 1.2504758279406168e-06, + "loss": 1.3325, + "step": 2279 + }, + { + "epoch": 2.342064714946071, + "grad_norm": 3.306988477706909, + "learning_rate": 1.2485725161781501e-06, + "loss": 1.3595, + "step": 2280 + }, + { + "epoch": 2.342064714946071, + "eval_loss": 1.8465285301208496, + "eval_runtime": 37.1211, + "eval_samples_per_second": 26.939, + "eval_steps_per_second": 1.131, + "step": 2280 + }, + { + "epoch": 2.343091936312275, + "grad_norm": 3.4089088439941406, + "learning_rate": 1.2466692044156833e-06, + "loss": 1.4337, + "step": 2281 + }, + { + "epoch": 2.3441191576784797, + "grad_norm": 3.1354665756225586, + "learning_rate": 1.2447658926532168e-06, + "loss": 1.2996, + "step": 2282 + }, + { + "epoch": 2.3451463790446843, + "grad_norm": 3.067383289337158, + "learning_rate": 1.24286258089075e-06, + "loss": 1.3446, + "step": 2283 + }, + { + "epoch": 2.3461736004108884, + "grad_norm": 3.162349224090576, + "learning_rate": 1.2409592691282834e-06, + "loss": 1.3068, + "step": 2284 + }, + { + "epoch": 2.347200821777093, + "grad_norm": 3.2846035957336426, + "learning_rate": 1.2390559573658165e-06, + "loss": 1.3496, + "step": 2285 + }, + { + "epoch": 2.3482280431432976, + "grad_norm": 3.290997266769409, + "learning_rate": 1.2371526456033499e-06, + "loss": 1.3471, + "step": 2286 + }, + { + "epoch": 2.3492552645095017, + "grad_norm": 3.1327056884765625, + "learning_rate": 1.2352493338408832e-06, + "loss": 1.3108, + "step": 2287 + }, + { + "epoch": 2.3502824858757063, + "grad_norm": 3.295562744140625, + "learning_rate": 1.2333460220784166e-06, + "loss": 1.4226, + "step": 2288 + }, + { + "epoch": 2.3513097072419105, + "grad_norm": 3.3461313247680664, + "learning_rate": 1.23144271031595e-06, + "loss": 1.3494, + "step": 2289 + }, + { + "epoch": 2.352336928608115, + "grad_norm": 3.4178872108459473, + "learning_rate": 1.2295393985534831e-06, + "loss": 1.3876, + "step": 2290 + }, + { + "epoch": 2.352336928608115, + "eval_loss": 1.855657935142517, + "eval_runtime": 37.2052, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 2290 + }, + { + "epoch": 2.3533641499743196, + "grad_norm": 3.0909383296966553, + "learning_rate": 1.2276360867910165e-06, + "loss": 1.2625, + "step": 2291 + }, + { + "epoch": 2.354391371340524, + "grad_norm": 3.2940239906311035, + "learning_rate": 1.2257327750285498e-06, + "loss": 1.3836, + "step": 2292 + }, + { + "epoch": 2.3554185927067284, + "grad_norm": 3.166198253631592, + "learning_rate": 1.2238294632660832e-06, + "loss": 1.3014, + "step": 2293 + }, + { + "epoch": 2.3564458140729325, + "grad_norm": 3.1372265815734863, + "learning_rate": 1.2219261515036164e-06, + "loss": 1.3133, + "step": 2294 + }, + { + "epoch": 2.357473035439137, + "grad_norm": 3.2904369831085205, + "learning_rate": 1.2200228397411497e-06, + "loss": 1.3578, + "step": 2295 + }, + { + "epoch": 2.3585002568053417, + "grad_norm": 3.1268763542175293, + "learning_rate": 1.2181195279786829e-06, + "loss": 1.391, + "step": 2296 + }, + { + "epoch": 2.359527478171546, + "grad_norm": 3.1357498168945312, + "learning_rate": 1.2162162162162164e-06, + "loss": 1.327, + "step": 2297 + }, + { + "epoch": 2.3605546995377504, + "grad_norm": 3.1766469478607178, + "learning_rate": 1.2143129044537496e-06, + "loss": 1.3146, + "step": 2298 + }, + { + "epoch": 2.361581920903955, + "grad_norm": 3.1096513271331787, + "learning_rate": 1.212409592691283e-06, + "loss": 1.3235, + "step": 2299 + }, + { + "epoch": 2.362609142270159, + "grad_norm": 3.161182403564453, + "learning_rate": 1.2105062809288163e-06, + "loss": 1.3411, + "step": 2300 + }, + { + "epoch": 2.362609142270159, + "eval_loss": 1.852128267288208, + "eval_runtime": 37.4484, + "eval_samples_per_second": 26.703, + "eval_steps_per_second": 1.122, + "step": 2300 + }, + { + "epoch": 2.3636363636363638, + "grad_norm": 3.2012717723846436, + "learning_rate": 1.2086029691663495e-06, + "loss": 1.3425, + "step": 2301 + }, + { + "epoch": 2.364663585002568, + "grad_norm": 3.0783276557922363, + "learning_rate": 1.2066996574038828e-06, + "loss": 1.2886, + "step": 2302 + }, + { + "epoch": 2.3656908063687725, + "grad_norm": 3.2996575832366943, + "learning_rate": 1.2047963456414162e-06, + "loss": 1.36, + "step": 2303 + }, + { + "epoch": 2.366718027734977, + "grad_norm": 3.2035586833953857, + "learning_rate": 1.2028930338789496e-06, + "loss": 1.3454, + "step": 2304 + }, + { + "epoch": 2.3677452491011812, + "grad_norm": 3.0552821159362793, + "learning_rate": 1.2009897221164827e-06, + "loss": 1.2788, + "step": 2305 + }, + { + "epoch": 2.368772470467386, + "grad_norm": 3.2013018131256104, + "learning_rate": 1.199086410354016e-06, + "loss": 1.3467, + "step": 2306 + }, + { + "epoch": 2.36979969183359, + "grad_norm": 3.226489305496216, + "learning_rate": 1.1971830985915492e-06, + "loss": 1.3177, + "step": 2307 + }, + { + "epoch": 2.3708269131997946, + "grad_norm": 3.320812940597534, + "learning_rate": 1.1952797868290828e-06, + "loss": 1.3118, + "step": 2308 + }, + { + "epoch": 2.371854134565999, + "grad_norm": 3.223085641860962, + "learning_rate": 1.193376475066616e-06, + "loss": 1.3363, + "step": 2309 + }, + { + "epoch": 2.3728813559322033, + "grad_norm": 3.3175759315490723, + "learning_rate": 1.1914731633041493e-06, + "loss": 1.3476, + "step": 2310 + }, + { + "epoch": 2.3728813559322033, + "eval_loss": 1.855944037437439, + "eval_runtime": 36.9815, + "eval_samples_per_second": 27.041, + "eval_steps_per_second": 1.136, + "step": 2310 + }, + { + "epoch": 2.373908577298408, + "grad_norm": 3.1612298488616943, + "learning_rate": 1.1895698515416827e-06, + "loss": 1.3083, + "step": 2311 + }, + { + "epoch": 2.3749357986646125, + "grad_norm": 3.18595814704895, + "learning_rate": 1.1876665397792158e-06, + "loss": 1.3048, + "step": 2312 + }, + { + "epoch": 2.3759630200308166, + "grad_norm": 3.3328146934509277, + "learning_rate": 1.1857632280167492e-06, + "loss": 1.3496, + "step": 2313 + }, + { + "epoch": 2.376990241397021, + "grad_norm": 3.1661181449890137, + "learning_rate": 1.1838599162542825e-06, + "loss": 1.284, + "step": 2314 + }, + { + "epoch": 2.3780174627632253, + "grad_norm": 3.298358201980591, + "learning_rate": 1.181956604491816e-06, + "loss": 1.359, + "step": 2315 + }, + { + "epoch": 2.37904468412943, + "grad_norm": 3.24114727973938, + "learning_rate": 1.180053292729349e-06, + "loss": 1.3264, + "step": 2316 + }, + { + "epoch": 2.380071905495634, + "grad_norm": 3.4791877269744873, + "learning_rate": 1.1781499809668824e-06, + "loss": 1.4022, + "step": 2317 + }, + { + "epoch": 2.3810991268618387, + "grad_norm": 3.366711139678955, + "learning_rate": 1.1762466692044158e-06, + "loss": 1.3341, + "step": 2318 + }, + { + "epoch": 2.3821263482280433, + "grad_norm": 3.3023927211761475, + "learning_rate": 1.1743433574419491e-06, + "loss": 1.3693, + "step": 2319 + }, + { + "epoch": 2.3831535695942474, + "grad_norm": 3.175189256668091, + "learning_rate": 1.1724400456794823e-06, + "loss": 1.3068, + "step": 2320 + }, + { + "epoch": 2.3831535695942474, + "eval_loss": 1.8537958860397339, + "eval_runtime": 37.079, + "eval_samples_per_second": 26.969, + "eval_steps_per_second": 1.133, + "step": 2320 + }, + { + "epoch": 2.384180790960452, + "grad_norm": 3.326911211013794, + "learning_rate": 1.1705367339170157e-06, + "loss": 1.3565, + "step": 2321 + }, + { + "epoch": 2.3852080123266566, + "grad_norm": 3.4324638843536377, + "learning_rate": 1.168633422154549e-06, + "loss": 1.3412, + "step": 2322 + }, + { + "epoch": 2.3862352336928607, + "grad_norm": 3.170397996902466, + "learning_rate": 1.1667301103920824e-06, + "loss": 1.3, + "step": 2323 + }, + { + "epoch": 2.3872624550590653, + "grad_norm": 3.1137685775756836, + "learning_rate": 1.1648267986296157e-06, + "loss": 1.2722, + "step": 2324 + }, + { + "epoch": 2.3882896764252695, + "grad_norm": 3.2298166751861572, + "learning_rate": 1.162923486867149e-06, + "loss": 1.326, + "step": 2325 + }, + { + "epoch": 2.389316897791474, + "grad_norm": 3.2462759017944336, + "learning_rate": 1.1610201751046823e-06, + "loss": 1.344, + "step": 2326 + }, + { + "epoch": 2.3903441191576786, + "grad_norm": 3.3631975650787354, + "learning_rate": 1.1591168633422154e-06, + "loss": 1.3909, + "step": 2327 + }, + { + "epoch": 2.391371340523883, + "grad_norm": 3.279407024383545, + "learning_rate": 1.157213551579749e-06, + "loss": 1.3544, + "step": 2328 + }, + { + "epoch": 2.3923985618900874, + "grad_norm": 3.1387593746185303, + "learning_rate": 1.1553102398172821e-06, + "loss": 1.33, + "step": 2329 + }, + { + "epoch": 2.3934257832562915, + "grad_norm": 3.218921661376953, + "learning_rate": 1.1534069280548155e-06, + "loss": 1.3117, + "step": 2330 + }, + { + "epoch": 2.3934257832562915, + "eval_loss": 1.8487199544906616, + "eval_runtime": 37.1303, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 2330 + }, + { + "epoch": 2.394453004622496, + "grad_norm": 3.221266746520996, + "learning_rate": 1.1515036162923487e-06, + "loss": 1.3521, + "step": 2331 + }, + { + "epoch": 2.3954802259887007, + "grad_norm": 3.112572431564331, + "learning_rate": 1.149600304529882e-06, + "loss": 1.3161, + "step": 2332 + }, + { + "epoch": 2.396507447354905, + "grad_norm": 3.2089130878448486, + "learning_rate": 1.1476969927674154e-06, + "loss": 1.3236, + "step": 2333 + }, + { + "epoch": 2.3975346687211094, + "grad_norm": 2.985661029815674, + "learning_rate": 1.1457936810049487e-06, + "loss": 1.3295, + "step": 2334 + }, + { + "epoch": 2.398561890087314, + "grad_norm": 3.189842939376831, + "learning_rate": 1.143890369242482e-06, + "loss": 1.3685, + "step": 2335 + }, + { + "epoch": 2.399589111453518, + "grad_norm": 3.124861478805542, + "learning_rate": 1.1419870574800153e-06, + "loss": 1.293, + "step": 2336 + }, + { + "epoch": 2.4006163328197228, + "grad_norm": 3.064199209213257, + "learning_rate": 1.1400837457175486e-06, + "loss": 1.3229, + "step": 2337 + }, + { + "epoch": 2.401643554185927, + "grad_norm": 3.1462392807006836, + "learning_rate": 1.138180433955082e-06, + "loss": 1.2931, + "step": 2338 + }, + { + "epoch": 2.4026707755521315, + "grad_norm": 3.192121982574463, + "learning_rate": 1.1362771221926153e-06, + "loss": 1.3569, + "step": 2339 + }, + { + "epoch": 2.403697996918336, + "grad_norm": 3.2683119773864746, + "learning_rate": 1.1343738104301485e-06, + "loss": 1.3437, + "step": 2340 + }, + { + "epoch": 2.403697996918336, + "eval_loss": 1.8514460325241089, + "eval_runtime": 37.185, + "eval_samples_per_second": 26.893, + "eval_steps_per_second": 1.129, + "step": 2340 + }, + { + "epoch": 2.4047252182845402, + "grad_norm": 3.168116331100464, + "learning_rate": 1.1324704986676819e-06, + "loss": 1.3177, + "step": 2341 + }, + { + "epoch": 2.405752439650745, + "grad_norm": 3.291511058807373, + "learning_rate": 1.130567186905215e-06, + "loss": 1.3731, + "step": 2342 + }, + { + "epoch": 2.406779661016949, + "grad_norm": 3.0953369140625, + "learning_rate": 1.1286638751427486e-06, + "loss": 1.3231, + "step": 2343 + }, + { + "epoch": 2.4078068823831535, + "grad_norm": 3.1431541442871094, + "learning_rate": 1.1267605633802817e-06, + "loss": 1.3227, + "step": 2344 + }, + { + "epoch": 2.408834103749358, + "grad_norm": 3.2409775257110596, + "learning_rate": 1.124857251617815e-06, + "loss": 1.3603, + "step": 2345 + }, + { + "epoch": 2.4098613251155623, + "grad_norm": 3.240900993347168, + "learning_rate": 1.1229539398553485e-06, + "loss": 1.3912, + "step": 2346 + }, + { + "epoch": 2.410888546481767, + "grad_norm": 3.1512467861175537, + "learning_rate": 1.1210506280928816e-06, + "loss": 1.3211, + "step": 2347 + }, + { + "epoch": 2.4119157678479715, + "grad_norm": 3.2743258476257324, + "learning_rate": 1.119147316330415e-06, + "loss": 1.3704, + "step": 2348 + }, + { + "epoch": 2.4129429892141756, + "grad_norm": 3.381438732147217, + "learning_rate": 1.1172440045679483e-06, + "loss": 1.335, + "step": 2349 + }, + { + "epoch": 2.41397021058038, + "grad_norm": 3.19822096824646, + "learning_rate": 1.1153406928054817e-06, + "loss": 1.3355, + "step": 2350 + }, + { + "epoch": 2.41397021058038, + "eval_loss": 1.8520244359970093, + "eval_runtime": 37.1581, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2350 + }, + { + "epoch": 2.4149974319465843, + "grad_norm": 3.3650894165039062, + "learning_rate": 1.1134373810430148e-06, + "loss": 1.3576, + "step": 2351 + }, + { + "epoch": 2.416024653312789, + "grad_norm": 3.3263492584228516, + "learning_rate": 1.1115340692805482e-06, + "loss": 1.3715, + "step": 2352 + }, + { + "epoch": 2.4170518746789935, + "grad_norm": 3.280782699584961, + "learning_rate": 1.1096307575180816e-06, + "loss": 1.3282, + "step": 2353 + }, + { + "epoch": 2.4180790960451977, + "grad_norm": 3.1976306438446045, + "learning_rate": 1.107727445755615e-06, + "loss": 1.3013, + "step": 2354 + }, + { + "epoch": 2.4191063174114023, + "grad_norm": 3.1435351371765137, + "learning_rate": 1.105824133993148e-06, + "loss": 1.3163, + "step": 2355 + }, + { + "epoch": 2.4201335387776064, + "grad_norm": 3.2856905460357666, + "learning_rate": 1.1039208222306814e-06, + "loss": 1.3378, + "step": 2356 + }, + { + "epoch": 2.421160760143811, + "grad_norm": 3.241748809814453, + "learning_rate": 1.1020175104682148e-06, + "loss": 1.3285, + "step": 2357 + }, + { + "epoch": 2.4221879815100156, + "grad_norm": 3.1425986289978027, + "learning_rate": 1.1001141987057482e-06, + "loss": 1.3155, + "step": 2358 + }, + { + "epoch": 2.4232152028762197, + "grad_norm": 3.311344623565674, + "learning_rate": 1.0982108869432813e-06, + "loss": 1.3291, + "step": 2359 + }, + { + "epoch": 2.4242424242424243, + "grad_norm": 3.2412517070770264, + "learning_rate": 1.0963075751808147e-06, + "loss": 1.3655, + "step": 2360 + }, + { + "epoch": 2.4242424242424243, + "eval_loss": 1.853851318359375, + "eval_runtime": 37.1577, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2360 + }, + { + "epoch": 2.425269645608629, + "grad_norm": 3.166773796081543, + "learning_rate": 1.094404263418348e-06, + "loss": 1.3316, + "step": 2361 + }, + { + "epoch": 2.426296866974833, + "grad_norm": 3.1892364025115967, + "learning_rate": 1.0925009516558812e-06, + "loss": 1.3304, + "step": 2362 + }, + { + "epoch": 2.4273240883410376, + "grad_norm": 3.129615306854248, + "learning_rate": 1.0905976398934148e-06, + "loss": 1.3211, + "step": 2363 + }, + { + "epoch": 2.428351309707242, + "grad_norm": 3.2534453868865967, + "learning_rate": 1.088694328130948e-06, + "loss": 1.3387, + "step": 2364 + }, + { + "epoch": 2.4293785310734464, + "grad_norm": 3.071871757507324, + "learning_rate": 1.0867910163684813e-06, + "loss": 1.2949, + "step": 2365 + }, + { + "epoch": 2.4304057524396505, + "grad_norm": 3.2920479774475098, + "learning_rate": 1.0848877046060144e-06, + "loss": 1.3345, + "step": 2366 + }, + { + "epoch": 2.431432973805855, + "grad_norm": 3.2313623428344727, + "learning_rate": 1.0829843928435478e-06, + "loss": 1.3308, + "step": 2367 + }, + { + "epoch": 2.4324601951720597, + "grad_norm": 3.1129872798919678, + "learning_rate": 1.0810810810810812e-06, + "loss": 1.3333, + "step": 2368 + }, + { + "epoch": 2.433487416538264, + "grad_norm": 3.278503894805908, + "learning_rate": 1.0791777693186145e-06, + "loss": 1.3458, + "step": 2369 + }, + { + "epoch": 2.4345146379044684, + "grad_norm": 3.390359401702881, + "learning_rate": 1.0772744575561479e-06, + "loss": 1.3262, + "step": 2370 + }, + { + "epoch": 2.4345146379044684, + "eval_loss": 1.8534605503082275, + "eval_runtime": 37.1424, + "eval_samples_per_second": 26.923, + "eval_steps_per_second": 1.131, + "step": 2370 + }, + { + "epoch": 2.435541859270673, + "grad_norm": 3.2505109310150146, + "learning_rate": 1.075371145793681e-06, + "loss": 1.3144, + "step": 2371 + }, + { + "epoch": 2.436569080636877, + "grad_norm": 3.1556410789489746, + "learning_rate": 1.0734678340312144e-06, + "loss": 1.3213, + "step": 2372 + }, + { + "epoch": 2.4375963020030817, + "grad_norm": 3.1231777667999268, + "learning_rate": 1.0715645222687478e-06, + "loss": 1.3124, + "step": 2373 + }, + { + "epoch": 2.438623523369286, + "grad_norm": 3.1159915924072266, + "learning_rate": 1.0696612105062811e-06, + "loss": 1.3191, + "step": 2374 + }, + { + "epoch": 2.4396507447354905, + "grad_norm": 3.2663285732269287, + "learning_rate": 1.0677578987438143e-06, + "loss": 1.3491, + "step": 2375 + }, + { + "epoch": 2.440677966101695, + "grad_norm": 3.0827524662017822, + "learning_rate": 1.0658545869813476e-06, + "loss": 1.3057, + "step": 2376 + }, + { + "epoch": 2.441705187467899, + "grad_norm": 3.1982173919677734, + "learning_rate": 1.0639512752188808e-06, + "loss": 1.312, + "step": 2377 + }, + { + "epoch": 2.442732408834104, + "grad_norm": 3.078274965286255, + "learning_rate": 1.0620479634564144e-06, + "loss": 1.3261, + "step": 2378 + }, + { + "epoch": 2.443759630200308, + "grad_norm": 3.0313515663146973, + "learning_rate": 1.0601446516939475e-06, + "loss": 1.2953, + "step": 2379 + }, + { + "epoch": 2.4447868515665125, + "grad_norm": 3.1637635231018066, + "learning_rate": 1.0582413399314809e-06, + "loss": 1.2973, + "step": 2380 + }, + { + "epoch": 2.4447868515665125, + "eval_loss": 1.8524926900863647, + "eval_runtime": 37.1576, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 2380 + }, + { + "epoch": 2.445814072932717, + "grad_norm": 3.2800862789154053, + "learning_rate": 1.0563380281690142e-06, + "loss": 1.3048, + "step": 2381 + }, + { + "epoch": 2.4468412942989213, + "grad_norm": 3.113825559616089, + "learning_rate": 1.0544347164065474e-06, + "loss": 1.3081, + "step": 2382 + }, + { + "epoch": 2.447868515665126, + "grad_norm": 3.201627254486084, + "learning_rate": 1.0525314046440808e-06, + "loss": 1.3362, + "step": 2383 + }, + { + "epoch": 2.4488957370313305, + "grad_norm": 3.1768431663513184, + "learning_rate": 1.0506280928816141e-06, + "loss": 1.3135, + "step": 2384 + }, + { + "epoch": 2.4499229583975346, + "grad_norm": 3.2254347801208496, + "learning_rate": 1.0487247811191475e-06, + "loss": 1.3168, + "step": 2385 + }, + { + "epoch": 2.450950179763739, + "grad_norm": 3.2468769550323486, + "learning_rate": 1.0468214693566806e-06, + "loss": 1.3301, + "step": 2386 + }, + { + "epoch": 2.4519774011299433, + "grad_norm": 3.214177370071411, + "learning_rate": 1.044918157594214e-06, + "loss": 1.3481, + "step": 2387 + }, + { + "epoch": 2.453004622496148, + "grad_norm": 3.1789238452911377, + "learning_rate": 1.0430148458317474e-06, + "loss": 1.3065, + "step": 2388 + }, + { + "epoch": 2.4540318438623525, + "grad_norm": 3.082336187362671, + "learning_rate": 1.0411115340692807e-06, + "loss": 1.2923, + "step": 2389 + }, + { + "epoch": 2.4550590652285567, + "grad_norm": 3.305126905441284, + "learning_rate": 1.0392082223068139e-06, + "loss": 1.3777, + "step": 2390 + }, + { + "epoch": 2.4550590652285567, + "eval_loss": 1.8477189540863037, + "eval_runtime": 37.3455, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 2390 + }, + { + "epoch": 2.4560862865947612, + "grad_norm": 3.322448492050171, + "learning_rate": 1.0373049105443472e-06, + "loss": 1.3309, + "step": 2391 + }, + { + "epoch": 2.4571135079609654, + "grad_norm": 3.2111802101135254, + "learning_rate": 1.0354015987818806e-06, + "loss": 1.3376, + "step": 2392 + }, + { + "epoch": 2.45814072932717, + "grad_norm": 3.0796451568603516, + "learning_rate": 1.033498287019414e-06, + "loss": 1.3093, + "step": 2393 + }, + { + "epoch": 2.4591679506933746, + "grad_norm": 3.265315055847168, + "learning_rate": 1.0315949752569471e-06, + "loss": 1.3652, + "step": 2394 + }, + { + "epoch": 2.4601951720595787, + "grad_norm": 3.1291234493255615, + "learning_rate": 1.0296916634944805e-06, + "loss": 1.2947, + "step": 2395 + }, + { + "epoch": 2.4612223934257833, + "grad_norm": 3.314732789993286, + "learning_rate": 1.0277883517320138e-06, + "loss": 1.3356, + "step": 2396 + }, + { + "epoch": 2.462249614791988, + "grad_norm": 3.2572879791259766, + "learning_rate": 1.025885039969547e-06, + "loss": 1.3665, + "step": 2397 + }, + { + "epoch": 2.463276836158192, + "grad_norm": 3.065584897994995, + "learning_rate": 1.0239817282070806e-06, + "loss": 1.2614, + "step": 2398 + }, + { + "epoch": 2.4643040575243966, + "grad_norm": 3.2398078441619873, + "learning_rate": 1.0220784164446137e-06, + "loss": 1.2986, + "step": 2399 + }, + { + "epoch": 2.4653312788906008, + "grad_norm": 3.1384856700897217, + "learning_rate": 1.020175104682147e-06, + "loss": 1.3157, + "step": 2400 + }, + { + "epoch": 2.4653312788906008, + "eval_loss": 1.8481135368347168, + "eval_runtime": 37.4463, + "eval_samples_per_second": 26.705, + "eval_steps_per_second": 1.122, + "step": 2400 + }, + { + "epoch": 2.4663585002568054, + "grad_norm": 3.2159786224365234, + "learning_rate": 1.0182717929196802e-06, + "loss": 1.3289, + "step": 2401 + }, + { + "epoch": 2.4673857216230095, + "grad_norm": 3.353900909423828, + "learning_rate": 1.0163684811572136e-06, + "loss": 1.3565, + "step": 2402 + }, + { + "epoch": 2.468412942989214, + "grad_norm": 3.1520562171936035, + "learning_rate": 1.014465169394747e-06, + "loss": 1.2855, + "step": 2403 + }, + { + "epoch": 2.4694401643554187, + "grad_norm": 3.2022621631622314, + "learning_rate": 1.0125618576322803e-06, + "loss": 1.3257, + "step": 2404 + }, + { + "epoch": 2.470467385721623, + "grad_norm": 3.1945600509643555, + "learning_rate": 1.0106585458698137e-06, + "loss": 1.3101, + "step": 2405 + }, + { + "epoch": 2.4714946070878274, + "grad_norm": 3.234997510910034, + "learning_rate": 1.0087552341073468e-06, + "loss": 1.3037, + "step": 2406 + }, + { + "epoch": 2.472521828454032, + "grad_norm": 3.3020741939544678, + "learning_rate": 1.0068519223448802e-06, + "loss": 1.3819, + "step": 2407 + }, + { + "epoch": 2.473549049820236, + "grad_norm": 3.4102628231048584, + "learning_rate": 1.0049486105824133e-06, + "loss": 1.3646, + "step": 2408 + }, + { + "epoch": 2.4745762711864407, + "grad_norm": 3.191983222961426, + "learning_rate": 1.003045298819947e-06, + "loss": 1.3036, + "step": 2409 + }, + { + "epoch": 2.4756034925526453, + "grad_norm": 3.3082430362701416, + "learning_rate": 1.00114198705748e-06, + "loss": 1.2973, + "step": 2410 + }, + { + "epoch": 2.4756034925526453, + "eval_loss": 1.8559728860855103, + "eval_runtime": 36.9665, + "eval_samples_per_second": 27.052, + "eval_steps_per_second": 1.136, + "step": 2410 + }, + { + "epoch": 2.4766307139188495, + "grad_norm": 3.3207387924194336, + "learning_rate": 9.992386752950134e-07, + "loss": 1.302, + "step": 2411 + }, + { + "epoch": 2.477657935285054, + "grad_norm": 3.2640819549560547, + "learning_rate": 9.973353635325466e-07, + "loss": 1.2963, + "step": 2412 + }, + { + "epoch": 2.478685156651258, + "grad_norm": 3.3404572010040283, + "learning_rate": 9.9543205177008e-07, + "loss": 1.2979, + "step": 2413 + }, + { + "epoch": 2.479712378017463, + "grad_norm": 3.2177815437316895, + "learning_rate": 9.935287400076133e-07, + "loss": 1.3022, + "step": 2414 + }, + { + "epoch": 2.480739599383667, + "grad_norm": 3.1873295307159424, + "learning_rate": 9.916254282451467e-07, + "loss": 1.2808, + "step": 2415 + }, + { + "epoch": 2.4817668207498715, + "grad_norm": 3.3681764602661133, + "learning_rate": 9.8972211648268e-07, + "loss": 1.3237, + "step": 2416 + }, + { + "epoch": 2.482794042116076, + "grad_norm": 3.2793691158294678, + "learning_rate": 9.878188047202132e-07, + "loss": 1.3437, + "step": 2417 + }, + { + "epoch": 2.4838212634822803, + "grad_norm": 3.1983675956726074, + "learning_rate": 9.859154929577465e-07, + "loss": 1.3211, + "step": 2418 + }, + { + "epoch": 2.484848484848485, + "grad_norm": 3.121095895767212, + "learning_rate": 9.8401218119528e-07, + "loss": 1.2957, + "step": 2419 + }, + { + "epoch": 2.4858757062146895, + "grad_norm": 3.3415417671203613, + "learning_rate": 9.821088694328133e-07, + "loss": 1.3116, + "step": 2420 + }, + { + "epoch": 2.4858757062146895, + "eval_loss": 1.8509575128555298, + "eval_runtime": 37.0683, + "eval_samples_per_second": 26.977, + "eval_steps_per_second": 1.133, + "step": 2420 + }, + { + "epoch": 2.4869029275808936, + "grad_norm": 3.196071147918701, + "learning_rate": 9.802055576703464e-07, + "loss": 1.3574, + "step": 2421 + }, + { + "epoch": 2.487930148947098, + "grad_norm": 3.260653018951416, + "learning_rate": 9.783022459078798e-07, + "loss": 1.3297, + "step": 2422 + }, + { + "epoch": 2.4889573703133023, + "grad_norm": 3.1694345474243164, + "learning_rate": 9.76398934145413e-07, + "loss": 1.3266, + "step": 2423 + }, + { + "epoch": 2.489984591679507, + "grad_norm": 3.227717161178589, + "learning_rate": 9.744956223829465e-07, + "loss": 1.3408, + "step": 2424 + }, + { + "epoch": 2.4910118130457115, + "grad_norm": 3.2542760372161865, + "learning_rate": 9.725923106204797e-07, + "loss": 1.3524, + "step": 2425 + }, + { + "epoch": 2.4920390344119157, + "grad_norm": 3.308953285217285, + "learning_rate": 9.70688998858013e-07, + "loss": 1.3758, + "step": 2426 + }, + { + "epoch": 2.4930662557781202, + "grad_norm": 3.467383623123169, + "learning_rate": 9.687856870955464e-07, + "loss": 1.3549, + "step": 2427 + }, + { + "epoch": 2.4940934771443244, + "grad_norm": 3.2643022537231445, + "learning_rate": 9.668823753330795e-07, + "loss": 1.3438, + "step": 2428 + }, + { + "epoch": 2.495120698510529, + "grad_norm": 3.142022132873535, + "learning_rate": 9.64979063570613e-07, + "loss": 1.2852, + "step": 2429 + }, + { + "epoch": 2.4961479198767336, + "grad_norm": 3.220409393310547, + "learning_rate": 9.630757518081463e-07, + "loss": 1.3315, + "step": 2430 + }, + { + "epoch": 2.4961479198767336, + "eval_loss": 1.8497542142868042, + "eval_runtime": 37.1119, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 2430 + }, + { + "epoch": 2.4971751412429377, + "grad_norm": 3.296198844909668, + "learning_rate": 9.611724400456796e-07, + "loss": 1.3241, + "step": 2431 + }, + { + "epoch": 2.4982023626091423, + "grad_norm": 3.197559356689453, + "learning_rate": 9.592691282832128e-07, + "loss": 1.2976, + "step": 2432 + }, + { + "epoch": 2.499229583975347, + "grad_norm": 3.261249542236328, + "learning_rate": 9.573658165207461e-07, + "loss": 1.3264, + "step": 2433 + }, + { + "epoch": 2.500256805341551, + "grad_norm": 3.235084295272827, + "learning_rate": 9.554625047582795e-07, + "loss": 1.3465, + "step": 2434 + }, + { + "epoch": 2.5012840267077556, + "grad_norm": 3.1297430992126465, + "learning_rate": 9.535591929958129e-07, + "loss": 1.299, + "step": 2435 + }, + { + "epoch": 2.50231124807396, + "grad_norm": 3.2298195362091064, + "learning_rate": 9.516558812333461e-07, + "loss": 1.3374, + "step": 2436 + }, + { + "epoch": 2.5033384694401644, + "grad_norm": 3.2262182235717773, + "learning_rate": 9.497525694708794e-07, + "loss": 1.3432, + "step": 2437 + }, + { + "epoch": 2.5043656908063685, + "grad_norm": 3.227480888366699, + "learning_rate": 9.478492577084126e-07, + "loss": 1.3076, + "step": 2438 + }, + { + "epoch": 2.505392912172573, + "grad_norm": 3.0988104343414307, + "learning_rate": 9.459459459459461e-07, + "loss": 1.3167, + "step": 2439 + }, + { + "epoch": 2.5064201335387777, + "grad_norm": 3.2276222705841064, + "learning_rate": 9.440426341834794e-07, + "loss": 1.3069, + "step": 2440 + }, + { + "epoch": 2.5064201335387777, + "eval_loss": 1.8424607515335083, + "eval_runtime": 37.1264, + "eval_samples_per_second": 26.935, + "eval_steps_per_second": 1.131, + "step": 2440 + }, + { + "epoch": 2.507447354904982, + "grad_norm": 3.1694467067718506, + "learning_rate": 9.421393224210126e-07, + "loss": 1.3194, + "step": 2441 + }, + { + "epoch": 2.5084745762711864, + "grad_norm": 3.2202742099761963, + "learning_rate": 9.402360106585459e-07, + "loss": 1.3288, + "step": 2442 + }, + { + "epoch": 2.509501797637391, + "grad_norm": 3.156813621520996, + "learning_rate": 9.383326988960792e-07, + "loss": 1.2853, + "step": 2443 + }, + { + "epoch": 2.510529019003595, + "grad_norm": 3.199050188064575, + "learning_rate": 9.364293871336126e-07, + "loss": 1.2818, + "step": 2444 + }, + { + "epoch": 2.5115562403697997, + "grad_norm": 3.1687068939208984, + "learning_rate": 9.345260753711458e-07, + "loss": 1.3115, + "step": 2445 + }, + { + "epoch": 2.5125834617360043, + "grad_norm": 3.224169969558716, + "learning_rate": 9.326227636086792e-07, + "loss": 1.3357, + "step": 2446 + }, + { + "epoch": 2.5136106831022085, + "grad_norm": 3.1867403984069824, + "learning_rate": 9.307194518462125e-07, + "loss": 1.277, + "step": 2447 + }, + { + "epoch": 2.514637904468413, + "grad_norm": 3.4736502170562744, + "learning_rate": 9.288161400837457e-07, + "loss": 1.3972, + "step": 2448 + }, + { + "epoch": 2.515665125834617, + "grad_norm": 3.147759437561035, + "learning_rate": 9.269128283212792e-07, + "loss": 1.287, + "step": 2449 + }, + { + "epoch": 2.516692347200822, + "grad_norm": 3.1714866161346436, + "learning_rate": 9.250095165588125e-07, + "loss": 1.3214, + "step": 2450 + }, + { + "epoch": 2.516692347200822, + "eval_loss": 1.8418748378753662, + "eval_runtime": 37.1781, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 2450 + }, + { + "epoch": 2.517719568567026, + "grad_norm": 3.308077573776245, + "learning_rate": 9.231062047963457e-07, + "loss": 1.3745, + "step": 2451 + }, + { + "epoch": 2.5187467899332305, + "grad_norm": 3.318955421447754, + "learning_rate": 9.21202893033879e-07, + "loss": 1.3767, + "step": 2452 + }, + { + "epoch": 2.519774011299435, + "grad_norm": 3.236506462097168, + "learning_rate": 9.192995812714123e-07, + "loss": 1.3311, + "step": 2453 + }, + { + "epoch": 2.5208012326656393, + "grad_norm": 3.3041205406188965, + "learning_rate": 9.173962695089457e-07, + "loss": 1.3243, + "step": 2454 + }, + { + "epoch": 2.521828454031844, + "grad_norm": 3.1637344360351562, + "learning_rate": 9.154929577464789e-07, + "loss": 1.3216, + "step": 2455 + }, + { + "epoch": 2.5228556753980484, + "grad_norm": 3.151911735534668, + "learning_rate": 9.135896459840123e-07, + "loss": 1.3494, + "step": 2456 + }, + { + "epoch": 2.5238828967642526, + "grad_norm": 3.335695743560791, + "learning_rate": 9.116863342215456e-07, + "loss": 1.3225, + "step": 2457 + }, + { + "epoch": 2.524910118130457, + "grad_norm": 3.2160415649414062, + "learning_rate": 9.097830224590788e-07, + "loss": 1.3693, + "step": 2458 + }, + { + "epoch": 2.5259373394966618, + "grad_norm": 3.0803115367889404, + "learning_rate": 9.078797106966122e-07, + "loss": 1.2844, + "step": 2459 + }, + { + "epoch": 2.526964560862866, + "grad_norm": 3.2147631645202637, + "learning_rate": 9.059763989341455e-07, + "loss": 1.3602, + "step": 2460 + }, + { + "epoch": 2.526964560862866, + "eval_loss": 1.8384426832199097, + "eval_runtime": 37.1832, + "eval_samples_per_second": 26.894, + "eval_steps_per_second": 1.13, + "step": 2460 + }, + { + "epoch": 2.5279917822290705, + "grad_norm": 3.198880672454834, + "learning_rate": 9.040730871716788e-07, + "loss": 1.2999, + "step": 2461 + }, + { + "epoch": 2.5290190035952747, + "grad_norm": 3.095944404602051, + "learning_rate": 9.021697754092121e-07, + "loss": 1.324, + "step": 2462 + }, + { + "epoch": 2.5300462249614792, + "grad_norm": 3.150750160217285, + "learning_rate": 9.002664636467453e-07, + "loss": 1.3259, + "step": 2463 + }, + { + "epoch": 2.5310734463276834, + "grad_norm": 3.1914174556732178, + "learning_rate": 8.983631518842788e-07, + "loss": 1.3401, + "step": 2464 + }, + { + "epoch": 2.532100667693888, + "grad_norm": 3.2229790687561035, + "learning_rate": 8.96459840121812e-07, + "loss": 1.3624, + "step": 2465 + }, + { + "epoch": 2.5331278890600926, + "grad_norm": 3.161130666732788, + "learning_rate": 8.945565283593453e-07, + "loss": 1.3183, + "step": 2466 + }, + { + "epoch": 2.5341551104262967, + "grad_norm": 3.0963833332061768, + "learning_rate": 8.926532165968787e-07, + "loss": 1.3446, + "step": 2467 + }, + { + "epoch": 2.5351823317925013, + "grad_norm": 3.324404001235962, + "learning_rate": 8.907499048344119e-07, + "loss": 1.3394, + "step": 2468 + }, + { + "epoch": 2.536209553158706, + "grad_norm": 3.122046709060669, + "learning_rate": 8.888465930719453e-07, + "loss": 1.2868, + "step": 2469 + }, + { + "epoch": 2.53723677452491, + "grad_norm": 3.1900618076324463, + "learning_rate": 8.869432813094786e-07, + "loss": 1.2872, + "step": 2470 + }, + { + "epoch": 2.53723677452491, + "eval_loss": 1.8466377258300781, + "eval_runtime": 37.1472, + "eval_samples_per_second": 26.92, + "eval_steps_per_second": 1.131, + "step": 2470 + }, + { + "epoch": 2.5382639958911146, + "grad_norm": 3.353994369506836, + "learning_rate": 8.850399695470119e-07, + "loss": 1.3304, + "step": 2471 + }, + { + "epoch": 2.539291217257319, + "grad_norm": 3.3058688640594482, + "learning_rate": 8.831366577845452e-07, + "loss": 1.344, + "step": 2472 + }, + { + "epoch": 2.5403184386235234, + "grad_norm": 3.3035926818847656, + "learning_rate": 8.812333460220784e-07, + "loss": 1.3029, + "step": 2473 + }, + { + "epoch": 2.541345659989728, + "grad_norm": 3.173894166946411, + "learning_rate": 8.793300342596119e-07, + "loss": 1.2742, + "step": 2474 + }, + { + "epoch": 2.542372881355932, + "grad_norm": 3.1931140422821045, + "learning_rate": 8.774267224971451e-07, + "loss": 1.3369, + "step": 2475 + }, + { + "epoch": 2.5434001027221367, + "grad_norm": 3.1834166049957275, + "learning_rate": 8.755234107346784e-07, + "loss": 1.3605, + "step": 2476 + }, + { + "epoch": 2.544427324088341, + "grad_norm": 3.244100332260132, + "learning_rate": 8.736200989722117e-07, + "loss": 1.2971, + "step": 2477 + }, + { + "epoch": 2.5454545454545454, + "grad_norm": 3.223937749862671, + "learning_rate": 8.71716787209745e-07, + "loss": 1.3174, + "step": 2478 + }, + { + "epoch": 2.54648176682075, + "grad_norm": 3.1803457736968994, + "learning_rate": 8.698134754472784e-07, + "loss": 1.3486, + "step": 2479 + }, + { + "epoch": 2.547508988186954, + "grad_norm": 2.991408109664917, + "learning_rate": 8.679101636848116e-07, + "loss": 1.2632, + "step": 2480 + }, + { + "epoch": 2.547508988186954, + "eval_loss": 1.8453017473220825, + "eval_runtime": 37.1608, + "eval_samples_per_second": 26.91, + "eval_steps_per_second": 1.13, + "step": 2480 + }, + { + "epoch": 2.5485362095531587, + "grad_norm": 3.2398955821990967, + "learning_rate": 8.66006851922345e-07, + "loss": 1.3418, + "step": 2481 + }, + { + "epoch": 2.5495634309193633, + "grad_norm": 3.0708181858062744, + "learning_rate": 8.641035401598783e-07, + "loss": 1.324, + "step": 2482 + }, + { + "epoch": 2.5505906522855675, + "grad_norm": 3.213524103164673, + "learning_rate": 8.622002283974115e-07, + "loss": 1.3071, + "step": 2483 + }, + { + "epoch": 2.551617873651772, + "grad_norm": 3.27577805519104, + "learning_rate": 8.60296916634945e-07, + "loss": 1.3458, + "step": 2484 + }, + { + "epoch": 2.5526450950179767, + "grad_norm": 3.110621213912964, + "learning_rate": 8.583936048724782e-07, + "loss": 1.3027, + "step": 2485 + }, + { + "epoch": 2.553672316384181, + "grad_norm": 3.2169458866119385, + "learning_rate": 8.564902931100115e-07, + "loss": 1.307, + "step": 2486 + }, + { + "epoch": 2.554699537750385, + "grad_norm": 3.26141357421875, + "learning_rate": 8.545869813475448e-07, + "loss": 1.3232, + "step": 2487 + }, + { + "epoch": 2.5557267591165895, + "grad_norm": 3.294462203979492, + "learning_rate": 8.526836695850781e-07, + "loss": 1.3328, + "step": 2488 + }, + { + "epoch": 2.556753980482794, + "grad_norm": 3.301452159881592, + "learning_rate": 8.507803578226115e-07, + "loss": 1.3408, + "step": 2489 + }, + { + "epoch": 2.5577812018489983, + "grad_norm": 3.342216968536377, + "learning_rate": 8.488770460601447e-07, + "loss": 1.3315, + "step": 2490 + }, + { + "epoch": 2.5577812018489983, + "eval_loss": 1.8501962423324585, + "eval_runtime": 37.2524, + "eval_samples_per_second": 26.844, + "eval_steps_per_second": 1.127, + "step": 2490 + }, + { + "epoch": 2.558808423215203, + "grad_norm": 3.3099851608276367, + "learning_rate": 8.46973734297678e-07, + "loss": 1.2939, + "step": 2491 + }, + { + "epoch": 2.5598356445814074, + "grad_norm": 3.141737461090088, + "learning_rate": 8.450704225352114e-07, + "loss": 1.3091, + "step": 2492 + }, + { + "epoch": 2.5608628659476116, + "grad_norm": 3.344872236251831, + "learning_rate": 8.431671107727446e-07, + "loss": 1.3001, + "step": 2493 + }, + { + "epoch": 2.561890087313816, + "grad_norm": 3.3284311294555664, + "learning_rate": 8.41263799010278e-07, + "loss": 1.3393, + "step": 2494 + }, + { + "epoch": 2.5629173086800208, + "grad_norm": 3.349339485168457, + "learning_rate": 8.393604872478113e-07, + "loss": 1.3197, + "step": 2495 + }, + { + "epoch": 2.563944530046225, + "grad_norm": 3.331051826477051, + "learning_rate": 8.374571754853446e-07, + "loss": 1.3074, + "step": 2496 + }, + { + "epoch": 2.5649717514124295, + "grad_norm": 3.2611076831817627, + "learning_rate": 8.355538637228778e-07, + "loss": 1.3172, + "step": 2497 + }, + { + "epoch": 2.5659989727786336, + "grad_norm": 3.284789800643921, + "learning_rate": 8.336505519604111e-07, + "loss": 1.3147, + "step": 2498 + }, + { + "epoch": 2.5670261941448382, + "grad_norm": 3.1650099754333496, + "learning_rate": 8.317472401979445e-07, + "loss": 1.3439, + "step": 2499 + }, + { + "epoch": 2.5680534155110424, + "grad_norm": 3.436192512512207, + "learning_rate": 8.298439284354778e-07, + "loss": 1.3644, + "step": 2500 + }, + { + "epoch": 2.5680534155110424, + "eval_loss": 1.8527119159698486, + "eval_runtime": 37.4654, + "eval_samples_per_second": 26.691, + "eval_steps_per_second": 1.121, + "step": 2500 + }, + { + "epoch": 2.569080636877247, + "grad_norm": 3.253854751586914, + "learning_rate": 8.279406166730111e-07, + "loss": 1.3392, + "step": 2501 + }, + { + "epoch": 2.5701078582434516, + "grad_norm": 3.260098695755005, + "learning_rate": 8.260373049105444e-07, + "loss": 1.3452, + "step": 2502 + }, + { + "epoch": 2.5711350796096557, + "grad_norm": 3.2392895221710205, + "learning_rate": 8.241339931480777e-07, + "loss": 1.3312, + "step": 2503 + }, + { + "epoch": 2.5721623009758603, + "grad_norm": 3.2497167587280273, + "learning_rate": 8.22230681385611e-07, + "loss": 1.374, + "step": 2504 + }, + { + "epoch": 2.573189522342065, + "grad_norm": 3.3461084365844727, + "learning_rate": 8.203273696231444e-07, + "loss": 1.3306, + "step": 2505 + }, + { + "epoch": 2.574216743708269, + "grad_norm": 3.3338277339935303, + "learning_rate": 8.184240578606777e-07, + "loss": 1.2846, + "step": 2506 + }, + { + "epoch": 2.5752439650744736, + "grad_norm": 3.235508918762207, + "learning_rate": 8.165207460982109e-07, + "loss": 1.3315, + "step": 2507 + }, + { + "epoch": 2.576271186440678, + "grad_norm": 3.091784715652466, + "learning_rate": 8.146174343357442e-07, + "loss": 1.2891, + "step": 2508 + }, + { + "epoch": 2.5772984078068824, + "grad_norm": 3.3658840656280518, + "learning_rate": 8.127141225732775e-07, + "loss": 1.3375, + "step": 2509 + }, + { + "epoch": 2.578325629173087, + "grad_norm": 3.244807720184326, + "learning_rate": 8.108108108108109e-07, + "loss": 1.3361, + "step": 2510 + }, + { + "epoch": 2.578325629173087, + "eval_loss": 1.8489638566970825, + "eval_runtime": 36.9423, + "eval_samples_per_second": 27.069, + "eval_steps_per_second": 1.137, + "step": 2510 + }, + { + "epoch": 2.579352850539291, + "grad_norm": 3.242994546890259, + "learning_rate": 8.089074990483442e-07, + "loss": 1.322, + "step": 2511 + }, + { + "epoch": 2.5803800719054957, + "grad_norm": 3.2104480266571045, + "learning_rate": 8.070041872858774e-07, + "loss": 1.3194, + "step": 2512 + }, + { + "epoch": 2.5814072932717, + "grad_norm": 3.294405221939087, + "learning_rate": 8.051008755234108e-07, + "loss": 1.3305, + "step": 2513 + }, + { + "epoch": 2.5824345146379044, + "grad_norm": 3.112826108932495, + "learning_rate": 8.031975637609441e-07, + "loss": 1.2705, + "step": 2514 + }, + { + "epoch": 2.583461736004109, + "grad_norm": 3.28314208984375, + "learning_rate": 8.012942519984774e-07, + "loss": 1.3185, + "step": 2515 + }, + { + "epoch": 2.584488957370313, + "grad_norm": 3.2840018272399902, + "learning_rate": 7.993909402360108e-07, + "loss": 1.3486, + "step": 2516 + }, + { + "epoch": 2.5855161787365177, + "grad_norm": 3.3145153522491455, + "learning_rate": 7.97487628473544e-07, + "loss": 1.3396, + "step": 2517 + }, + { + "epoch": 2.5865434001027223, + "grad_norm": 3.334235429763794, + "learning_rate": 7.955843167110773e-07, + "loss": 1.3218, + "step": 2518 + }, + { + "epoch": 2.5875706214689265, + "grad_norm": 3.12007999420166, + "learning_rate": 7.936810049486106e-07, + "loss": 1.2869, + "step": 2519 + }, + { + "epoch": 2.588597842835131, + "grad_norm": 3.074021816253662, + "learning_rate": 7.91777693186144e-07, + "loss": 1.3042, + "step": 2520 + }, + { + "epoch": 2.588597842835131, + "eval_loss": 1.8512887954711914, + "eval_runtime": 37.0714, + "eval_samples_per_second": 26.975, + "eval_steps_per_second": 1.133, + "step": 2520 + }, + { + "epoch": 2.5896250642013356, + "grad_norm": 3.2101500034332275, + "learning_rate": 7.898743814236773e-07, + "loss": 1.3303, + "step": 2521 + }, + { + "epoch": 2.59065228556754, + "grad_norm": 3.178086042404175, + "learning_rate": 7.879710696612105e-07, + "loss": 1.3095, + "step": 2522 + }, + { + "epoch": 2.5916795069337444, + "grad_norm": 3.319711923599243, + "learning_rate": 7.860677578987438e-07, + "loss": 1.3661, + "step": 2523 + }, + { + "epoch": 2.5927067282999485, + "grad_norm": 3.135955572128296, + "learning_rate": 7.841644461362772e-07, + "loss": 1.3263, + "step": 2524 + }, + { + "epoch": 2.593733949666153, + "grad_norm": 3.272080659866333, + "learning_rate": 7.822611343738105e-07, + "loss": 1.3034, + "step": 2525 + }, + { + "epoch": 2.5947611710323573, + "grad_norm": 3.161445140838623, + "learning_rate": 7.803578226113438e-07, + "loss": 1.3595, + "step": 2526 + }, + { + "epoch": 2.595788392398562, + "grad_norm": 3.1976101398468018, + "learning_rate": 7.784545108488771e-07, + "loss": 1.302, + "step": 2527 + }, + { + "epoch": 2.5968156137647664, + "grad_norm": 3.1619246006011963, + "learning_rate": 7.765511990864104e-07, + "loss": 1.326, + "step": 2528 + }, + { + "epoch": 2.5978428351309706, + "grad_norm": 3.375790596008301, + "learning_rate": 7.746478873239437e-07, + "loss": 1.3786, + "step": 2529 + }, + { + "epoch": 2.598870056497175, + "grad_norm": 3.1595287322998047, + "learning_rate": 7.727445755614771e-07, + "loss": 1.3252, + "step": 2530 + }, + { + "epoch": 2.598870056497175, + "eval_loss": 1.8481647968292236, + "eval_runtime": 37.0379, + "eval_samples_per_second": 26.999, + "eval_steps_per_second": 1.134, + "step": 2530 + }, + { + "epoch": 2.5998972778633798, + "grad_norm": 3.304802894592285, + "learning_rate": 7.708412637990104e-07, + "loss": 1.3358, + "step": 2531 + }, + { + "epoch": 2.600924499229584, + "grad_norm": 3.1958820819854736, + "learning_rate": 7.689379520365436e-07, + "loss": 1.3508, + "step": 2532 + }, + { + "epoch": 2.6019517205957885, + "grad_norm": 3.117140054702759, + "learning_rate": 7.670346402740769e-07, + "loss": 1.305, + "step": 2533 + }, + { + "epoch": 2.602978941961993, + "grad_norm": 3.147562026977539, + "learning_rate": 7.651313285116103e-07, + "loss": 1.3408, + "step": 2534 + }, + { + "epoch": 2.6040061633281972, + "grad_norm": 3.2008414268493652, + "learning_rate": 7.632280167491436e-07, + "loss": 1.359, + "step": 2535 + }, + { + "epoch": 2.6050333846944014, + "grad_norm": 3.063007354736328, + "learning_rate": 7.613247049866769e-07, + "loss": 1.2846, + "step": 2536 + }, + { + "epoch": 2.606060606060606, + "grad_norm": 3.1810076236724854, + "learning_rate": 7.594213932242102e-07, + "loss": 1.262, + "step": 2537 + }, + { + "epoch": 2.6070878274268106, + "grad_norm": 3.1805059909820557, + "learning_rate": 7.575180814617435e-07, + "loss": 1.3017, + "step": 2538 + }, + { + "epoch": 2.6081150487930147, + "grad_norm": 3.1886227130889893, + "learning_rate": 7.556147696992767e-07, + "loss": 1.3215, + "step": 2539 + }, + { + "epoch": 2.6091422701592193, + "grad_norm": 3.317014217376709, + "learning_rate": 7.537114579368102e-07, + "loss": 1.329, + "step": 2540 + }, + { + "epoch": 2.6091422701592193, + "eval_loss": 1.8478699922561646, + "eval_runtime": 37.059, + "eval_samples_per_second": 26.984, + "eval_steps_per_second": 1.133, + "step": 2540 + }, + { + "epoch": 2.610169491525424, + "grad_norm": 3.147467851638794, + "learning_rate": 7.518081461743435e-07, + "loss": 1.295, + "step": 2541 + }, + { + "epoch": 2.611196712891628, + "grad_norm": 3.2641215324401855, + "learning_rate": 7.499048344118767e-07, + "loss": 1.3129, + "step": 2542 + }, + { + "epoch": 2.6122239342578326, + "grad_norm": 3.3371410369873047, + "learning_rate": 7.4800152264941e-07, + "loss": 1.3384, + "step": 2543 + }, + { + "epoch": 2.613251155624037, + "grad_norm": 3.193312406539917, + "learning_rate": 7.460982108869432e-07, + "loss": 1.3082, + "step": 2544 + }, + { + "epoch": 2.6142783769902413, + "grad_norm": 3.2359626293182373, + "learning_rate": 7.441948991244767e-07, + "loss": 1.2993, + "step": 2545 + }, + { + "epoch": 2.615305598356446, + "grad_norm": 3.3347742557525635, + "learning_rate": 7.4229158736201e-07, + "loss": 1.3439, + "step": 2546 + }, + { + "epoch": 2.61633281972265, + "grad_norm": 3.2009949684143066, + "learning_rate": 7.403882755995432e-07, + "loss": 1.3431, + "step": 2547 + }, + { + "epoch": 2.6173600410888547, + "grad_norm": 3.2062506675720215, + "learning_rate": 7.384849638370766e-07, + "loss": 1.2976, + "step": 2548 + }, + { + "epoch": 2.618387262455059, + "grad_norm": 3.2773313522338867, + "learning_rate": 7.365816520746098e-07, + "loss": 1.256, + "step": 2549 + }, + { + "epoch": 2.6194144838212634, + "grad_norm": 3.2360665798187256, + "learning_rate": 7.346783403121432e-07, + "loss": 1.3198, + "step": 2550 + }, + { + "epoch": 2.6194144838212634, + "eval_loss": 1.8466824293136597, + "eval_runtime": 37.0917, + "eval_samples_per_second": 26.96, + "eval_steps_per_second": 1.132, + "step": 2550 + }, + { + "epoch": 2.620441705187468, + "grad_norm": 3.159076690673828, + "learning_rate": 7.327750285496766e-07, + "loss": 1.3037, + "step": 2551 + }, + { + "epoch": 2.621468926553672, + "grad_norm": 3.3330905437469482, + "learning_rate": 7.308717167872098e-07, + "loss": 1.3099, + "step": 2552 + }, + { + "epoch": 2.6224961479198767, + "grad_norm": 3.04374361038208, + "learning_rate": 7.289684050247431e-07, + "loss": 1.28, + "step": 2553 + }, + { + "epoch": 2.6235233692860813, + "grad_norm": 3.2761824131011963, + "learning_rate": 7.270650932622763e-07, + "loss": 1.3726, + "step": 2554 + }, + { + "epoch": 2.6245505906522855, + "grad_norm": 3.1171300411224365, + "learning_rate": 7.251617814998098e-07, + "loss": 1.2626, + "step": 2555 + }, + { + "epoch": 2.62557781201849, + "grad_norm": 3.1138417720794678, + "learning_rate": 7.232584697373431e-07, + "loss": 1.3294, + "step": 2556 + }, + { + "epoch": 2.6266050333846946, + "grad_norm": 3.174699068069458, + "learning_rate": 7.213551579748763e-07, + "loss": 1.2998, + "step": 2557 + }, + { + "epoch": 2.627632254750899, + "grad_norm": 3.1825549602508545, + "learning_rate": 7.194518462124096e-07, + "loss": 1.3227, + "step": 2558 + }, + { + "epoch": 2.6286594761171034, + "grad_norm": 3.4744699001312256, + "learning_rate": 7.175485344499429e-07, + "loss": 1.4021, + "step": 2559 + }, + { + "epoch": 2.6296866974833075, + "grad_norm": 3.3134055137634277, + "learning_rate": 7.156452226874763e-07, + "loss": 1.2968, + "step": 2560 + }, + { + "epoch": 2.6296866974833075, + "eval_loss": 1.8447562456130981, + "eval_runtime": 37.1051, + "eval_samples_per_second": 26.951, + "eval_steps_per_second": 1.132, + "step": 2560 + }, + { + "epoch": 2.630713918849512, + "grad_norm": 3.340637683868408, + "learning_rate": 7.137419109250096e-07, + "loss": 1.3952, + "step": 2561 + }, + { + "epoch": 2.6317411402157163, + "grad_norm": 3.2392141819000244, + "learning_rate": 7.118385991625429e-07, + "loss": 1.3035, + "step": 2562 + }, + { + "epoch": 2.632768361581921, + "grad_norm": 3.281599521636963, + "learning_rate": 7.099352874000762e-07, + "loss": 1.2963, + "step": 2563 + }, + { + "epoch": 2.6337955829481254, + "grad_norm": 3.1986238956451416, + "learning_rate": 7.080319756376094e-07, + "loss": 1.3125, + "step": 2564 + }, + { + "epoch": 2.6348228043143296, + "grad_norm": 3.180161476135254, + "learning_rate": 7.061286638751429e-07, + "loss": 1.2844, + "step": 2565 + }, + { + "epoch": 2.635850025680534, + "grad_norm": 3.2440733909606934, + "learning_rate": 7.042253521126762e-07, + "loss": 1.3215, + "step": 2566 + }, + { + "epoch": 2.6368772470467388, + "grad_norm": 3.274649143218994, + "learning_rate": 7.023220403502094e-07, + "loss": 1.2705, + "step": 2567 + }, + { + "epoch": 2.637904468412943, + "grad_norm": 3.3886029720306396, + "learning_rate": 7.004187285877427e-07, + "loss": 1.3542, + "step": 2568 + }, + { + "epoch": 2.6389316897791475, + "grad_norm": 3.2831993103027344, + "learning_rate": 6.98515416825276e-07, + "loss": 1.3462, + "step": 2569 + }, + { + "epoch": 2.639958911145352, + "grad_norm": 3.1610167026519775, + "learning_rate": 6.966121050628094e-07, + "loss": 1.3155, + "step": 2570 + }, + { + "epoch": 2.639958911145352, + "eval_loss": 1.8500503301620483, + "eval_runtime": 37.1167, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 2570 + }, + { + "epoch": 2.6409861325115562, + "grad_norm": 3.226743698120117, + "learning_rate": 6.947087933003427e-07, + "loss": 1.3681, + "step": 2571 + }, + { + "epoch": 2.642013353877761, + "grad_norm": 3.297621011734009, + "learning_rate": 6.92805481537876e-07, + "loss": 1.3279, + "step": 2572 + }, + { + "epoch": 2.643040575243965, + "grad_norm": 3.3724985122680664, + "learning_rate": 6.909021697754093e-07, + "loss": 1.3627, + "step": 2573 + }, + { + "epoch": 2.6440677966101696, + "grad_norm": 3.1355743408203125, + "learning_rate": 6.889988580129425e-07, + "loss": 1.3426, + "step": 2574 + }, + { + "epoch": 2.6450950179763737, + "grad_norm": 3.174088716506958, + "learning_rate": 6.870955462504759e-07, + "loss": 1.2999, + "step": 2575 + }, + { + "epoch": 2.6461222393425783, + "grad_norm": 3.369877338409424, + "learning_rate": 6.851922344880093e-07, + "loss": 1.3423, + "step": 2576 + }, + { + "epoch": 2.647149460708783, + "grad_norm": 3.1381266117095947, + "learning_rate": 6.832889227255425e-07, + "loss": 1.3273, + "step": 2577 + }, + { + "epoch": 2.648176682074987, + "grad_norm": 3.2909109592437744, + "learning_rate": 6.813856109630758e-07, + "loss": 1.3339, + "step": 2578 + }, + { + "epoch": 2.6492039034411916, + "grad_norm": 3.284780502319336, + "learning_rate": 6.79482299200609e-07, + "loss": 1.3344, + "step": 2579 + }, + { + "epoch": 2.650231124807396, + "grad_norm": 3.060544729232788, + "learning_rate": 6.775789874381425e-07, + "loss": 1.2978, + "step": 2580 + }, + { + "epoch": 2.650231124807396, + "eval_loss": 1.8496923446655273, + "eval_runtime": 37.1508, + "eval_samples_per_second": 26.917, + "eval_steps_per_second": 1.131, + "step": 2580 + }, + { + "epoch": 2.6512583461736003, + "grad_norm": 3.4877448081970215, + "learning_rate": 6.756756756756758e-07, + "loss": 1.3615, + "step": 2581 + }, + { + "epoch": 2.652285567539805, + "grad_norm": 3.3131983280181885, + "learning_rate": 6.73772363913209e-07, + "loss": 1.3498, + "step": 2582 + }, + { + "epoch": 2.653312788906009, + "grad_norm": 3.285895347595215, + "learning_rate": 6.718690521507424e-07, + "loss": 1.3375, + "step": 2583 + }, + { + "epoch": 2.6543400102722137, + "grad_norm": 3.271651029586792, + "learning_rate": 6.699657403882756e-07, + "loss": 1.2863, + "step": 2584 + }, + { + "epoch": 2.655367231638418, + "grad_norm": 3.2834060192108154, + "learning_rate": 6.68062428625809e-07, + "loss": 1.3478, + "step": 2585 + }, + { + "epoch": 2.6563944530046224, + "grad_norm": 3.21610951423645, + "learning_rate": 6.661591168633424e-07, + "loss": 1.3564, + "step": 2586 + }, + { + "epoch": 2.657421674370827, + "grad_norm": 3.2195615768432617, + "learning_rate": 6.642558051008756e-07, + "loss": 1.3404, + "step": 2587 + }, + { + "epoch": 2.658448895737031, + "grad_norm": 3.295682191848755, + "learning_rate": 6.623524933384089e-07, + "loss": 1.3367, + "step": 2588 + }, + { + "epoch": 2.6594761171032357, + "grad_norm": 3.3580493927001953, + "learning_rate": 6.604491815759421e-07, + "loss": 1.3307, + "step": 2589 + }, + { + "epoch": 2.6605033384694403, + "grad_norm": 3.1793158054351807, + "learning_rate": 6.585458698134756e-07, + "loss": 1.3038, + "step": 2590 + }, + { + "epoch": 2.6605033384694403, + "eval_loss": 1.8480147123336792, + "eval_runtime": 37.2352, + "eval_samples_per_second": 26.856, + "eval_steps_per_second": 1.128, + "step": 2590 + }, + { + "epoch": 2.6615305598356445, + "grad_norm": 3.136833429336548, + "learning_rate": 6.566425580510089e-07, + "loss": 1.2706, + "step": 2591 + }, + { + "epoch": 2.662557781201849, + "grad_norm": 3.2903378009796143, + "learning_rate": 6.547392462885421e-07, + "loss": 1.3364, + "step": 2592 + }, + { + "epoch": 2.6635850025680536, + "grad_norm": 3.220923900604248, + "learning_rate": 6.528359345260754e-07, + "loss": 1.3526, + "step": 2593 + }, + { + "epoch": 2.664612223934258, + "grad_norm": 3.1533079147338867, + "learning_rate": 6.509326227636087e-07, + "loss": 1.315, + "step": 2594 + }, + { + "epoch": 2.6656394453004624, + "grad_norm": 3.3369879722595215, + "learning_rate": 6.490293110011421e-07, + "loss": 1.335, + "step": 2595 + }, + { + "epoch": 2.6666666666666665, + "grad_norm": 3.1055495738983154, + "learning_rate": 6.471259992386753e-07, + "loss": 1.3285, + "step": 2596 + }, + { + "epoch": 2.667693888032871, + "grad_norm": 3.1696672439575195, + "learning_rate": 6.452226874762087e-07, + "loss": 1.3665, + "step": 2597 + }, + { + "epoch": 2.6687211093990753, + "grad_norm": 3.243185043334961, + "learning_rate": 6.43319375713742e-07, + "loss": 1.32, + "step": 2598 + }, + { + "epoch": 2.66974833076528, + "grad_norm": 3.2200284004211426, + "learning_rate": 6.414160639512752e-07, + "loss": 1.3371, + "step": 2599 + }, + { + "epoch": 2.6707755521314844, + "grad_norm": 3.225383758544922, + "learning_rate": 6.395127521888085e-07, + "loss": 1.3217, + "step": 2600 + }, + { + "epoch": 2.6707755521314844, + "eval_loss": 1.8515571355819702, + "eval_runtime": 37.4223, + "eval_samples_per_second": 26.722, + "eval_steps_per_second": 1.122, + "step": 2600 + } + ], + "logging_steps": 1, + "max_steps": 2919, + "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": 8.105004464340992e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-2600/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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_0.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_0.pth new file mode 100644 index 0000000000000000000000000000000000000000..d5f37008b65b2b6b774ba8ca3e08ea7f2886beeb --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_1.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_1.pth new file mode 100644 index 0000000000000000000000000000000000000000..43698f33ac8bdb91266f12645eda8b36787da828 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_2.pth b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/rng_state_2.pth new file mode 100644 index 0000000000000000000000000000000000000000..90d0b3f139f4cb1368a680fad6f12df4c66f078f --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_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/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/scheduler.pt b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/scheduler.pt new file mode 100644 index 0000000000000000000000000000000000000000..67baa84b8152ff3010768663c012ec867209faeb --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/scheduler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af6cd6397670b1cedae20439b5ba35a00a237ec5bba71a4c5f6d95e2d59d045b +size 627 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/tokenizer.json b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/tokenizer.json new file mode 100644 index 0000000000000000000000000000000000000000..c76f0701115b50c463b66cc3645d7a3654ab5205 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fda2ecf7b5daf3cb77324ea6c5b9cb8bf47119249a6d756ef2dd4d62540ac6f +size 17210554 diff --git a/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/training_args.bin b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/training_args.bin new file mode 100644 index 0000000000000000000000000000000000000000..35c83e0aeb67b51f9041a0775518d818bd4514c1 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_reverse_control_10M_seed0/runs/checkpoint-600/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:892167fd00638a9a336920c633c1535594e6726eea0c8de0982206892bd302fd +size 6011 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/latest b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/latest new file mode 100644 index 0000000000000000000000000000000000000000..744ae7dbad571b6f37ec6c7066549494261bb59e --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/latest @@ -0,0 +1 @@ +global_step100 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..5684b14e2756b56aa397c1324e239b442162b34a --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/tokenizer_config.json @@ -0,0 +1,2070 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..bb4b1a9f949b53f1282c4de0c74239a1c5044bc7 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/trainer_state.json @@ -0,0 +1,813 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.1095890410958904, + "eval_steps": 10, + "global_step": 100, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.001095890410958904, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0476, + "step": 1 + }, + { + "epoch": 0.002191780821917808, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0289, + "step": 2 + }, + { + "epoch": 0.0032876712328767125, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9935, + "step": 3 + }, + { + "epoch": 0.004383561643835616, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0139, + "step": 4 + }, + { + "epoch": 0.005479452054794521, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9892, + "step": 5 + }, + { + "epoch": 0.006575342465753425, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.1252, + "step": 6 + }, + { + "epoch": 0.007671232876712329, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0291, + "step": 7 + }, + { + "epoch": 0.008767123287671232, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0607, + "step": 8 + }, + { + "epoch": 0.009863013698630137, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0306, + "step": 9 + }, + { + "epoch": 0.010958904109589041, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0811, + "step": 10 + }, + { + "epoch": 0.010958904109589041, + "eval_loss": 3.0768930912017822, + "eval_runtime": 38.0377, + "eval_samples_per_second": 26.29, + "eval_steps_per_second": 1.104, + "step": 10 + }, + { + "epoch": 0.012054794520547946, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0693, + "step": 11 + }, + { + "epoch": 0.01315068493150685, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9668, + "step": 12 + }, + { + "epoch": 0.014246575342465753, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.052, + "step": 13 + }, + { + "epoch": 0.015342465753424657, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0363, + "step": 14 + }, + { + "epoch": 0.01643835616438356, + "grad_norm": 5.58175802230835, + "learning_rate": 1.8248175182481753e-08, + "loss": 3.0778, + "step": 15 + }, + { + "epoch": 0.017534246575342465, + "grad_norm": 6.360531806945801, + "learning_rate": 3.6496350364963505e-08, + "loss": 3.0257, + "step": 16 + }, + { + "epoch": 0.01863013698630137, + "grad_norm": 6.453113555908203, + "learning_rate": 5.4744525547445264e-08, + "loss": 3.0262, + "step": 17 + }, + { + "epoch": 0.019726027397260273, + "grad_norm": 6.789405345916748, + "learning_rate": 7.299270072992701e-08, + "loss": 3.0671, + "step": 18 + }, + { + "epoch": 0.020821917808219178, + "grad_norm": 6.185514450073242, + "learning_rate": 9.124087591240876e-08, + "loss": 3.0077, + "step": 19 + }, + { + "epoch": 0.021917808219178082, + "grad_norm": 5.719798564910889, + "learning_rate": 1.0948905109489053e-07, + "loss": 3.089, + "step": 20 + }, + { + "epoch": 0.021917808219178082, + "eval_loss": 3.07546329498291, + "eval_runtime": 37.5318, + "eval_samples_per_second": 26.644, + "eval_steps_per_second": 1.119, + "step": 20 + }, + { + "epoch": 0.023013698630136987, + "grad_norm": 6.307768821716309, + "learning_rate": 1.2773722627737227e-07, + "loss": 3.0641, + "step": 21 + }, + { + "epoch": 0.02410958904109589, + "grad_norm": 6.255112171173096, + "learning_rate": 1.4598540145985402e-07, + "loss": 2.9849, + "step": 22 + }, + { + "epoch": 0.025205479452054796, + "grad_norm": 5.812101364135742, + "learning_rate": 1.642335766423358e-07, + "loss": 3.0211, + "step": 23 + }, + { + "epoch": 0.0263013698630137, + "grad_norm": 6.262075901031494, + "learning_rate": 1.8248175182481753e-07, + "loss": 3.0654, + "step": 24 + }, + { + "epoch": 0.0273972602739726, + "grad_norm": 5.419229507446289, + "learning_rate": 2.0072992700729928e-07, + "loss": 3.0661, + "step": 25 + }, + { + "epoch": 0.028493150684931506, + "grad_norm": 4.750737190246582, + "learning_rate": 2.1897810218978106e-07, + "loss": 3.0494, + "step": 26 + }, + { + "epoch": 0.02958904109589041, + "grad_norm": 4.732361316680908, + "learning_rate": 2.3722627737226278e-07, + "loss": 3.0471, + "step": 27 + }, + { + "epoch": 0.030684931506849315, + "grad_norm": 4.749786853790283, + "learning_rate": 2.5547445255474454e-07, + "loss": 3.0107, + "step": 28 + }, + { + "epoch": 0.03178082191780822, + "grad_norm": 4.451244831085205, + "learning_rate": 2.737226277372263e-07, + "loss": 2.9987, + "step": 29 + }, + { + "epoch": 0.03287671232876712, + "grad_norm": 4.5158257484436035, + "learning_rate": 2.9197080291970804e-07, + "loss": 2.9686, + "step": 30 + }, + { + "epoch": 0.03287671232876712, + "eval_loss": 3.031872034072876, + "eval_runtime": 37.2608, + "eval_samples_per_second": 26.838, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03397260273972603, + "grad_norm": 3.8239498138427734, + "learning_rate": 3.102189781021898e-07, + "loss": 3.0057, + "step": 31 + }, + { + "epoch": 0.03506849315068493, + "grad_norm": 4.037254810333252, + "learning_rate": 3.284671532846716e-07, + "loss": 2.9733, + "step": 32 + }, + { + "epoch": 0.03616438356164384, + "grad_norm": 4.283931732177734, + "learning_rate": 3.467153284671533e-07, + "loss": 2.9515, + "step": 33 + }, + { + "epoch": 0.03726027397260274, + "grad_norm": 4.1784987449646, + "learning_rate": 3.6496350364963505e-07, + "loss": 2.9483, + "step": 34 + }, + { + "epoch": 0.038356164383561646, + "grad_norm": 3.9210753440856934, + "learning_rate": 3.8321167883211686e-07, + "loss": 2.9679, + "step": 35 + }, + { + "epoch": 0.03945205479452055, + "grad_norm": 3.72884202003479, + "learning_rate": 4.0145985401459856e-07, + "loss": 2.9989, + "step": 36 + }, + { + "epoch": 0.040547945205479455, + "grad_norm": 3.2882330417633057, + "learning_rate": 4.197080291970803e-07, + "loss": 2.9384, + "step": 37 + }, + { + "epoch": 0.041643835616438356, + "grad_norm": 3.515293836593628, + "learning_rate": 4.379562043795621e-07, + "loss": 2.8986, + "step": 38 + }, + { + "epoch": 0.042739726027397264, + "grad_norm": 3.568453073501587, + "learning_rate": 4.562043795620438e-07, + "loss": 2.9854, + "step": 39 + }, + { + "epoch": 0.043835616438356165, + "grad_norm": 3.389789342880249, + "learning_rate": 4.7445255474452557e-07, + "loss": 2.8891, + "step": 40 + }, + { + "epoch": 0.043835616438356165, + "eval_loss": 2.964235544204712, + "eval_runtime": 37.3406, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.044931506849315066, + "grad_norm": 3.1826283931732178, + "learning_rate": 4.927007299270073e-07, + "loss": 2.9271, + "step": 41 + }, + { + "epoch": 0.046027397260273974, + "grad_norm": 2.941539764404297, + "learning_rate": 5.109489051094891e-07, + "loss": 2.9486, + "step": 42 + }, + { + "epoch": 0.047123287671232875, + "grad_norm": 3.0076773166656494, + "learning_rate": 5.291970802919708e-07, + "loss": 2.9252, + "step": 43 + }, + { + "epoch": 0.04821917808219178, + "grad_norm": 2.79028582572937, + "learning_rate": 5.474452554744526e-07, + "loss": 2.9277, + "step": 44 + }, + { + "epoch": 0.049315068493150684, + "grad_norm": 2.603456735610962, + "learning_rate": 5.656934306569343e-07, + "loss": 2.9144, + "step": 45 + }, + { + "epoch": 0.05041095890410959, + "grad_norm": 2.674888849258423, + "learning_rate": 5.839416058394161e-07, + "loss": 2.8732, + "step": 46 + }, + { + "epoch": 0.05150684931506849, + "grad_norm": 2.972093343734741, + "learning_rate": 6.021897810218979e-07, + "loss": 2.8623, + "step": 47 + }, + { + "epoch": 0.0526027397260274, + "grad_norm": 2.757766008377075, + "learning_rate": 6.204379562043796e-07, + "loss": 2.8941, + "step": 48 + }, + { + "epoch": 0.0536986301369863, + "grad_norm": 2.910634994506836, + "learning_rate": 6.386861313868614e-07, + "loss": 2.8101, + "step": 49 + }, + { + "epoch": 0.0547945205479452, + "grad_norm": 2.5542638301849365, + "learning_rate": 6.569343065693432e-07, + "loss": 2.9262, + "step": 50 + }, + { + "epoch": 0.0547945205479452, + "eval_loss": 2.883579969406128, + "eval_runtime": 37.326, + "eval_samples_per_second": 26.791, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05589041095890411, + "grad_norm": 2.333974599838257, + "learning_rate": 6.751824817518248e-07, + "loss": 2.8217, + "step": 51 + }, + { + "epoch": 0.05698630136986301, + "grad_norm": 2.181022882461548, + "learning_rate": 6.934306569343066e-07, + "loss": 2.8483, + "step": 52 + }, + { + "epoch": 0.05808219178082192, + "grad_norm": 2.12200927734375, + "learning_rate": 7.116788321167883e-07, + "loss": 2.7788, + "step": 53 + }, + { + "epoch": 0.05917808219178082, + "grad_norm": 2.052626371383667, + "learning_rate": 7.299270072992701e-07, + "loss": 2.7774, + "step": 54 + }, + { + "epoch": 0.06027397260273973, + "grad_norm": 2.11487078666687, + "learning_rate": 7.48175182481752e-07, + "loss": 2.8045, + "step": 55 + }, + { + "epoch": 0.06136986301369863, + "grad_norm": 2.083588123321533, + "learning_rate": 7.664233576642337e-07, + "loss": 2.8119, + "step": 56 + }, + { + "epoch": 0.06246575342465754, + "grad_norm": 2.2079808712005615, + "learning_rate": 7.846715328467154e-07, + "loss": 2.82, + "step": 57 + }, + { + "epoch": 0.06356164383561644, + "grad_norm": 2.100802421569824, + "learning_rate": 8.029197080291971e-07, + "loss": 2.794, + "step": 58 + }, + { + "epoch": 0.06465753424657535, + "grad_norm": 1.8322066068649292, + "learning_rate": 8.211678832116789e-07, + "loss": 2.7444, + "step": 59 + }, + { + "epoch": 0.06575342465753424, + "grad_norm": 2.077005386352539, + "learning_rate": 8.394160583941606e-07, + "loss": 2.7093, + "step": 60 + }, + { + "epoch": 0.06575342465753424, + "eval_loss": 2.8177874088287354, + "eval_runtime": 37.2903, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 60 + }, + { + "epoch": 0.06684931506849315, + "grad_norm": 2.1688592433929443, + "learning_rate": 8.576642335766425e-07, + "loss": 2.7446, + "step": 61 + }, + { + "epoch": 0.06794520547945206, + "grad_norm": 2.1709237098693848, + "learning_rate": 8.759124087591242e-07, + "loss": 2.736, + "step": 62 + }, + { + "epoch": 0.06904109589041096, + "grad_norm": 2.3578364849090576, + "learning_rate": 8.94160583941606e-07, + "loss": 2.7583, + "step": 63 + }, + { + "epoch": 0.07013698630136986, + "grad_norm": 1.9025483131408691, + "learning_rate": 9.124087591240876e-07, + "loss": 2.7651, + "step": 64 + }, + { + "epoch": 0.07123287671232877, + "grad_norm": 1.9450963735580444, + "learning_rate": 9.306569343065694e-07, + "loss": 2.7186, + "step": 65 + }, + { + "epoch": 0.07232876712328767, + "grad_norm": 1.957723617553711, + "learning_rate": 9.489051094890511e-07, + "loss": 2.7421, + "step": 66 + }, + { + "epoch": 0.07342465753424658, + "grad_norm": 1.9823631048202515, + "learning_rate": 9.67153284671533e-07, + "loss": 2.7258, + "step": 67 + }, + { + "epoch": 0.07452054794520548, + "grad_norm": 1.945495843887329, + "learning_rate": 9.854014598540146e-07, + "loss": 2.7733, + "step": 68 + }, + { + "epoch": 0.07561643835616438, + "grad_norm": 1.9684759378433228, + "learning_rate": 1.0036496350364965e-06, + "loss": 2.7041, + "step": 69 + }, + { + "epoch": 0.07671232876712329, + "grad_norm": 2.087336540222168, + "learning_rate": 1.0218978102189781e-06, + "loss": 2.6662, + "step": 70 + }, + { + "epoch": 0.07671232876712329, + "eval_loss": 2.7529091835021973, + "eval_runtime": 37.3122, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 70 + }, + { + "epoch": 0.07780821917808219, + "grad_norm": 1.7790117263793945, + "learning_rate": 1.04014598540146e-06, + "loss": 2.7379, + "step": 71 + }, + { + "epoch": 0.0789041095890411, + "grad_norm": 2.2780861854553223, + "learning_rate": 1.0583941605839416e-06, + "loss": 2.6234, + "step": 72 + }, + { + "epoch": 0.08, + "grad_norm": 2.1144700050354004, + "learning_rate": 1.0766423357664233e-06, + "loss": 2.6248, + "step": 73 + }, + { + "epoch": 0.08109589041095891, + "grad_norm": 2.0247490406036377, + "learning_rate": 1.0948905109489052e-06, + "loss": 2.6456, + "step": 74 + }, + { + "epoch": 0.0821917808219178, + "grad_norm": 2.702547311782837, + "learning_rate": 1.113138686131387e-06, + "loss": 2.6759, + "step": 75 + }, + { + "epoch": 0.08328767123287671, + "grad_norm": 1.8208750486373901, + "learning_rate": 1.1313868613138687e-06, + "loss": 2.6845, + "step": 76 + }, + { + "epoch": 0.08438356164383562, + "grad_norm": 2.3829221725463867, + "learning_rate": 1.1496350364963505e-06, + "loss": 2.61, + "step": 77 + }, + { + "epoch": 0.08547945205479453, + "grad_norm": 2.580885887145996, + "learning_rate": 1.1678832116788322e-06, + "loss": 2.5896, + "step": 78 + }, + { + "epoch": 0.08657534246575342, + "grad_norm": 2.2477641105651855, + "learning_rate": 1.1861313868613138e-06, + "loss": 2.596, + "step": 79 + }, + { + "epoch": 0.08767123287671233, + "grad_norm": 2.1895623207092285, + "learning_rate": 1.2043795620437959e-06, + "loss": 2.5906, + "step": 80 + }, + { + "epoch": 0.08767123287671233, + "eval_loss": 2.6815176010131836, + "eval_runtime": 37.3342, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 80 + }, + { + "epoch": 0.08876712328767124, + "grad_norm": 2.217090606689453, + "learning_rate": 1.2226277372262775e-06, + "loss": 2.6005, + "step": 81 + }, + { + "epoch": 0.08986301369863013, + "grad_norm": 2.4679362773895264, + "learning_rate": 1.2408759124087592e-06, + "loss": 2.6333, + "step": 82 + }, + { + "epoch": 0.09095890410958904, + "grad_norm": 2.2590575218200684, + "learning_rate": 1.259124087591241e-06, + "loss": 2.5814, + "step": 83 + }, + { + "epoch": 0.09205479452054795, + "grad_norm": 1.877977728843689, + "learning_rate": 1.2773722627737229e-06, + "loss": 2.5948, + "step": 84 + }, + { + "epoch": 0.09315068493150686, + "grad_norm": 2.1535534858703613, + "learning_rate": 1.2956204379562045e-06, + "loss": 2.5781, + "step": 85 + }, + { + "epoch": 0.09424657534246575, + "grad_norm": 2.5001144409179688, + "learning_rate": 1.3138686131386864e-06, + "loss": 2.5339, + "step": 86 + }, + { + "epoch": 0.09534246575342466, + "grad_norm": 2.3279049396514893, + "learning_rate": 1.332116788321168e-06, + "loss": 2.545, + "step": 87 + }, + { + "epoch": 0.09643835616438357, + "grad_norm": 2.2230734825134277, + "learning_rate": 1.3503649635036497e-06, + "loss": 2.4745, + "step": 88 + }, + { + "epoch": 0.09753424657534246, + "grad_norm": 2.8762636184692383, + "learning_rate": 1.3686131386861315e-06, + "loss": 2.516, + "step": 89 + }, + { + "epoch": 0.09863013698630137, + "grad_norm": 2.153506278991699, + "learning_rate": 1.3868613138686132e-06, + "loss": 2.4547, + "step": 90 + }, + { + "epoch": 0.09863013698630137, + "eval_loss": 2.603238344192505, + "eval_runtime": 37.4162, + "eval_samples_per_second": 26.726, + "eval_steps_per_second": 1.123, + "step": 90 + }, + { + "epoch": 0.09972602739726028, + "grad_norm": 2.4143614768981934, + "learning_rate": 1.405109489051095e-06, + "loss": 2.547, + "step": 91 + }, + { + "epoch": 0.10082191780821918, + "grad_norm": 2.354632616043091, + "learning_rate": 1.4233576642335767e-06, + "loss": 2.5009, + "step": 92 + }, + { + "epoch": 0.10191780821917808, + "grad_norm": 2.6605563163757324, + "learning_rate": 1.4416058394160583e-06, + "loss": 2.5304, + "step": 93 + }, + { + "epoch": 0.10301369863013699, + "grad_norm": 2.3724122047424316, + "learning_rate": 1.4598540145985402e-06, + "loss": 2.4925, + "step": 94 + }, + { + "epoch": 0.10410958904109589, + "grad_norm": 3.6323022842407227, + "learning_rate": 1.478102189781022e-06, + "loss": 2.4592, + "step": 95 + }, + { + "epoch": 0.1052054794520548, + "grad_norm": 2.3407089710235596, + "learning_rate": 1.496350364963504e-06, + "loss": 2.5223, + "step": 96 + }, + { + "epoch": 0.1063013698630137, + "grad_norm": 2.8238413333892822, + "learning_rate": 1.5145985401459856e-06, + "loss": 2.489, + "step": 97 + }, + { + "epoch": 0.1073972602739726, + "grad_norm": 2.6433658599853516, + "learning_rate": 1.5328467153284674e-06, + "loss": 2.5101, + "step": 98 + }, + { + "epoch": 0.10849315068493151, + "grad_norm": 2.562761068344116, + "learning_rate": 1.551094890510949e-06, + "loss": 2.4642, + "step": 99 + }, + { + "epoch": 0.1095890410958904, + "grad_norm": 2.4501149654388428, + "learning_rate": 1.5693430656934307e-06, + "loss": 2.4102, + "step": 100 + }, + { + "epoch": 0.1095890410958904, + "eval_loss": 2.5452804565429688, + "eval_runtime": 37.4224, + "eval_samples_per_second": 26.722, + "eval_steps_per_second": 1.122, + "step": 100 + } + ], + "logging_steps": 1, + "max_steps": 2736, + "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.11730940936192e+16, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-100/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/latest b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/latest new file mode 100644 index 0000000000000000000000000000000000000000..873b08355ae55e303606c474ab5fba25f9e8262a --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/latest @@ -0,0 +1 @@ +global_step1200 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..5684b14e2756b56aa397c1324e239b442162b34a --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/tokenizer_config.json @@ -0,0 +1,2070 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1200/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/latest b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/latest new file mode 100644 index 0000000000000000000000000000000000000000..cb24338af451be6d3ccddc18a950b58a70589761 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/latest @@ -0,0 +1 @@ +global_step1700 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..5684b14e2756b56aa397c1324e239b442162b34a --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/tokenizer_config.json @@ -0,0 +1,2070 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..16b343c2aef97b70c9cd17e4ae9af919e968f4a4 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/trainer_state.json @@ -0,0 +1,13293 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 1.8630136986301369, + "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.001095890410958904, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0476, + "step": 1 + }, + { + "epoch": 0.002191780821917808, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0289, + "step": 2 + }, + { + "epoch": 0.0032876712328767125, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9935, + "step": 3 + }, + { + "epoch": 0.004383561643835616, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0139, + "step": 4 + }, + { + "epoch": 0.005479452054794521, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9892, + "step": 5 + }, + { + "epoch": 0.006575342465753425, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.1252, + "step": 6 + }, + { + "epoch": 0.007671232876712329, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0291, + "step": 7 + }, + { + "epoch": 0.008767123287671232, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0607, + "step": 8 + }, + { + "epoch": 0.009863013698630137, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0306, + "step": 9 + }, + { + "epoch": 0.010958904109589041, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0811, + "step": 10 + }, + { + "epoch": 0.010958904109589041, + "eval_loss": 3.0768930912017822, + "eval_runtime": 38.0377, + "eval_samples_per_second": 26.29, + "eval_steps_per_second": 1.104, + "step": 10 + }, + { + "epoch": 0.012054794520547946, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0693, + "step": 11 + }, + { + "epoch": 0.01315068493150685, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9668, + "step": 12 + }, + { + "epoch": 0.014246575342465753, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.052, + "step": 13 + }, + { + "epoch": 0.015342465753424657, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0363, + "step": 14 + }, + { + "epoch": 0.01643835616438356, + "grad_norm": 5.58175802230835, + "learning_rate": 1.8248175182481753e-08, + "loss": 3.0778, + "step": 15 + }, + { + "epoch": 0.017534246575342465, + "grad_norm": 6.360531806945801, + "learning_rate": 3.6496350364963505e-08, + "loss": 3.0257, + "step": 16 + }, + { + "epoch": 0.01863013698630137, + "grad_norm": 6.453113555908203, + "learning_rate": 5.4744525547445264e-08, + "loss": 3.0262, + "step": 17 + }, + { + "epoch": 0.019726027397260273, + "grad_norm": 6.789405345916748, + "learning_rate": 7.299270072992701e-08, + "loss": 3.0671, + "step": 18 + }, + { + "epoch": 0.020821917808219178, + "grad_norm": 6.185514450073242, + "learning_rate": 9.124087591240876e-08, + "loss": 3.0077, + "step": 19 + }, + { + "epoch": 0.021917808219178082, + "grad_norm": 5.719798564910889, + "learning_rate": 1.0948905109489053e-07, + "loss": 3.089, + "step": 20 + }, + { + "epoch": 0.021917808219178082, + "eval_loss": 3.07546329498291, + "eval_runtime": 37.5318, + "eval_samples_per_second": 26.644, + "eval_steps_per_second": 1.119, + "step": 20 + }, + { + "epoch": 0.023013698630136987, + "grad_norm": 6.307768821716309, + "learning_rate": 1.2773722627737227e-07, + "loss": 3.0641, + "step": 21 + }, + { + "epoch": 0.02410958904109589, + "grad_norm": 6.255112171173096, + "learning_rate": 1.4598540145985402e-07, + "loss": 2.9849, + "step": 22 + }, + { + "epoch": 0.025205479452054796, + "grad_norm": 5.812101364135742, + "learning_rate": 1.642335766423358e-07, + "loss": 3.0211, + "step": 23 + }, + { + "epoch": 0.0263013698630137, + "grad_norm": 6.262075901031494, + "learning_rate": 1.8248175182481753e-07, + "loss": 3.0654, + "step": 24 + }, + { + "epoch": 0.0273972602739726, + "grad_norm": 5.419229507446289, + "learning_rate": 2.0072992700729928e-07, + "loss": 3.0661, + "step": 25 + }, + { + "epoch": 0.028493150684931506, + "grad_norm": 4.750737190246582, + "learning_rate": 2.1897810218978106e-07, + "loss": 3.0494, + "step": 26 + }, + { + "epoch": 0.02958904109589041, + "grad_norm": 4.732361316680908, + "learning_rate": 2.3722627737226278e-07, + "loss": 3.0471, + "step": 27 + }, + { + "epoch": 0.030684931506849315, + "grad_norm": 4.749786853790283, + "learning_rate": 2.5547445255474454e-07, + "loss": 3.0107, + "step": 28 + }, + { + "epoch": 0.03178082191780822, + "grad_norm": 4.451244831085205, + "learning_rate": 2.737226277372263e-07, + "loss": 2.9987, + "step": 29 + }, + { + "epoch": 0.03287671232876712, + "grad_norm": 4.5158257484436035, + "learning_rate": 2.9197080291970804e-07, + "loss": 2.9686, + "step": 30 + }, + { + "epoch": 0.03287671232876712, + "eval_loss": 3.031872034072876, + "eval_runtime": 37.2608, + "eval_samples_per_second": 26.838, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03397260273972603, + "grad_norm": 3.8239498138427734, + "learning_rate": 3.102189781021898e-07, + "loss": 3.0057, + "step": 31 + }, + { + "epoch": 0.03506849315068493, + "grad_norm": 4.037254810333252, + "learning_rate": 3.284671532846716e-07, + "loss": 2.9733, + "step": 32 + }, + { + "epoch": 0.03616438356164384, + "grad_norm": 4.283931732177734, + "learning_rate": 3.467153284671533e-07, + "loss": 2.9515, + "step": 33 + }, + { + "epoch": 0.03726027397260274, + "grad_norm": 4.1784987449646, + "learning_rate": 3.6496350364963505e-07, + "loss": 2.9483, + "step": 34 + }, + { + "epoch": 0.038356164383561646, + "grad_norm": 3.9210753440856934, + "learning_rate": 3.8321167883211686e-07, + "loss": 2.9679, + "step": 35 + }, + { + "epoch": 0.03945205479452055, + "grad_norm": 3.72884202003479, + "learning_rate": 4.0145985401459856e-07, + "loss": 2.9989, + "step": 36 + }, + { + "epoch": 0.040547945205479455, + "grad_norm": 3.2882330417633057, + "learning_rate": 4.197080291970803e-07, + "loss": 2.9384, + "step": 37 + }, + { + "epoch": 0.041643835616438356, + "grad_norm": 3.515293836593628, + "learning_rate": 4.379562043795621e-07, + "loss": 2.8986, + "step": 38 + }, + { + "epoch": 0.042739726027397264, + "grad_norm": 3.568453073501587, + "learning_rate": 4.562043795620438e-07, + "loss": 2.9854, + "step": 39 + }, + { + "epoch": 0.043835616438356165, + "grad_norm": 3.389789342880249, + "learning_rate": 4.7445255474452557e-07, + "loss": 2.8891, + "step": 40 + }, + { + "epoch": 0.043835616438356165, + "eval_loss": 2.964235544204712, + "eval_runtime": 37.3406, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.044931506849315066, + "grad_norm": 3.1826283931732178, + "learning_rate": 4.927007299270073e-07, + "loss": 2.9271, + "step": 41 + }, + { + "epoch": 0.046027397260273974, + "grad_norm": 2.941539764404297, + "learning_rate": 5.109489051094891e-07, + "loss": 2.9486, + "step": 42 + }, + { + "epoch": 0.047123287671232875, + "grad_norm": 3.0076773166656494, + "learning_rate": 5.291970802919708e-07, + "loss": 2.9252, + "step": 43 + }, + { + "epoch": 0.04821917808219178, + "grad_norm": 2.79028582572937, + "learning_rate": 5.474452554744526e-07, + "loss": 2.9277, + "step": 44 + }, + { + "epoch": 0.049315068493150684, + "grad_norm": 2.603456735610962, + "learning_rate": 5.656934306569343e-07, + "loss": 2.9144, + "step": 45 + }, + { + "epoch": 0.05041095890410959, + "grad_norm": 2.674888849258423, + "learning_rate": 5.839416058394161e-07, + "loss": 2.8732, + "step": 46 + }, + { + "epoch": 0.05150684931506849, + "grad_norm": 2.972093343734741, + "learning_rate": 6.021897810218979e-07, + "loss": 2.8623, + "step": 47 + }, + { + "epoch": 0.0526027397260274, + "grad_norm": 2.757766008377075, + "learning_rate": 6.204379562043796e-07, + "loss": 2.8941, + "step": 48 + }, + { + "epoch": 0.0536986301369863, + "grad_norm": 2.910634994506836, + "learning_rate": 6.386861313868614e-07, + "loss": 2.8101, + "step": 49 + }, + { + "epoch": 0.0547945205479452, + "grad_norm": 2.5542638301849365, + "learning_rate": 6.569343065693432e-07, + "loss": 2.9262, + "step": 50 + }, + { + "epoch": 0.0547945205479452, + "eval_loss": 2.883579969406128, + "eval_runtime": 37.326, + "eval_samples_per_second": 26.791, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05589041095890411, + "grad_norm": 2.333974599838257, + "learning_rate": 6.751824817518248e-07, + "loss": 2.8217, + "step": 51 + }, + { + "epoch": 0.05698630136986301, + "grad_norm": 2.181022882461548, + "learning_rate": 6.934306569343066e-07, + "loss": 2.8483, + "step": 52 + }, + { + "epoch": 0.05808219178082192, + "grad_norm": 2.12200927734375, + "learning_rate": 7.116788321167883e-07, + "loss": 2.7788, + "step": 53 + }, + { + "epoch": 0.05917808219178082, + "grad_norm": 2.052626371383667, + "learning_rate": 7.299270072992701e-07, + "loss": 2.7774, + "step": 54 + }, + { + "epoch": 0.06027397260273973, + "grad_norm": 2.11487078666687, + "learning_rate": 7.48175182481752e-07, + "loss": 2.8045, + "step": 55 + }, + { + "epoch": 0.06136986301369863, + "grad_norm": 2.083588123321533, + "learning_rate": 7.664233576642337e-07, + "loss": 2.8119, + "step": 56 + }, + { + "epoch": 0.06246575342465754, + "grad_norm": 2.2079808712005615, + "learning_rate": 7.846715328467154e-07, + "loss": 2.82, + "step": 57 + }, + { + "epoch": 0.06356164383561644, + "grad_norm": 2.100802421569824, + "learning_rate": 8.029197080291971e-07, + "loss": 2.794, + "step": 58 + }, + { + "epoch": 0.06465753424657535, + "grad_norm": 1.8322066068649292, + "learning_rate": 8.211678832116789e-07, + "loss": 2.7444, + "step": 59 + }, + { + "epoch": 0.06575342465753424, + "grad_norm": 2.077005386352539, + "learning_rate": 8.394160583941606e-07, + "loss": 2.7093, + "step": 60 + }, + { + "epoch": 0.06575342465753424, + "eval_loss": 2.8177874088287354, + "eval_runtime": 37.2903, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 60 + }, + { + "epoch": 0.06684931506849315, + "grad_norm": 2.1688592433929443, + "learning_rate": 8.576642335766425e-07, + "loss": 2.7446, + "step": 61 + }, + { + "epoch": 0.06794520547945206, + "grad_norm": 2.1709237098693848, + "learning_rate": 8.759124087591242e-07, + "loss": 2.736, + "step": 62 + }, + { + "epoch": 0.06904109589041096, + "grad_norm": 2.3578364849090576, + "learning_rate": 8.94160583941606e-07, + "loss": 2.7583, + "step": 63 + }, + { + "epoch": 0.07013698630136986, + "grad_norm": 1.9025483131408691, + "learning_rate": 9.124087591240876e-07, + "loss": 2.7651, + "step": 64 + }, + { + "epoch": 0.07123287671232877, + "grad_norm": 1.9450963735580444, + "learning_rate": 9.306569343065694e-07, + "loss": 2.7186, + "step": 65 + }, + { + "epoch": 0.07232876712328767, + "grad_norm": 1.957723617553711, + "learning_rate": 9.489051094890511e-07, + "loss": 2.7421, + "step": 66 + }, + { + "epoch": 0.07342465753424658, + "grad_norm": 1.9823631048202515, + "learning_rate": 9.67153284671533e-07, + "loss": 2.7258, + "step": 67 + }, + { + "epoch": 0.07452054794520548, + "grad_norm": 1.945495843887329, + "learning_rate": 9.854014598540146e-07, + "loss": 2.7733, + "step": 68 + }, + { + "epoch": 0.07561643835616438, + "grad_norm": 1.9684759378433228, + "learning_rate": 1.0036496350364965e-06, + "loss": 2.7041, + "step": 69 + }, + { + "epoch": 0.07671232876712329, + "grad_norm": 2.087336540222168, + "learning_rate": 1.0218978102189781e-06, + "loss": 2.6662, + "step": 70 + }, + { + "epoch": 0.07671232876712329, + "eval_loss": 2.7529091835021973, + "eval_runtime": 37.3122, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 70 + }, + { + "epoch": 0.07780821917808219, + "grad_norm": 1.7790117263793945, + "learning_rate": 1.04014598540146e-06, + "loss": 2.7379, + "step": 71 + }, + { + "epoch": 0.0789041095890411, + "grad_norm": 2.2780861854553223, + "learning_rate": 1.0583941605839416e-06, + "loss": 2.6234, + "step": 72 + }, + { + "epoch": 0.08, + "grad_norm": 2.1144700050354004, + "learning_rate": 1.0766423357664233e-06, + "loss": 2.6248, + "step": 73 + }, + { + "epoch": 0.08109589041095891, + "grad_norm": 2.0247490406036377, + "learning_rate": 1.0948905109489052e-06, + "loss": 2.6456, + "step": 74 + }, + { + "epoch": 0.0821917808219178, + "grad_norm": 2.702547311782837, + "learning_rate": 1.113138686131387e-06, + "loss": 2.6759, + "step": 75 + }, + { + "epoch": 0.08328767123287671, + "grad_norm": 1.8208750486373901, + "learning_rate": 1.1313868613138687e-06, + "loss": 2.6845, + "step": 76 + }, + { + "epoch": 0.08438356164383562, + "grad_norm": 2.3829221725463867, + "learning_rate": 1.1496350364963505e-06, + "loss": 2.61, + "step": 77 + }, + { + "epoch": 0.08547945205479453, + "grad_norm": 2.580885887145996, + "learning_rate": 1.1678832116788322e-06, + "loss": 2.5896, + "step": 78 + }, + { + "epoch": 0.08657534246575342, + "grad_norm": 2.2477641105651855, + "learning_rate": 1.1861313868613138e-06, + "loss": 2.596, + "step": 79 + }, + { + "epoch": 0.08767123287671233, + "grad_norm": 2.1895623207092285, + "learning_rate": 1.2043795620437959e-06, + "loss": 2.5906, + "step": 80 + }, + { + "epoch": 0.08767123287671233, + "eval_loss": 2.6815176010131836, + "eval_runtime": 37.3342, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 80 + }, + { + "epoch": 0.08876712328767124, + "grad_norm": 2.217090606689453, + "learning_rate": 1.2226277372262775e-06, + "loss": 2.6005, + "step": 81 + }, + { + "epoch": 0.08986301369863013, + "grad_norm": 2.4679362773895264, + "learning_rate": 1.2408759124087592e-06, + "loss": 2.6333, + "step": 82 + }, + { + "epoch": 0.09095890410958904, + "grad_norm": 2.2590575218200684, + "learning_rate": 1.259124087591241e-06, + "loss": 2.5814, + "step": 83 + }, + { + "epoch": 0.09205479452054795, + "grad_norm": 1.877977728843689, + "learning_rate": 1.2773722627737229e-06, + "loss": 2.5948, + "step": 84 + }, + { + "epoch": 0.09315068493150686, + "grad_norm": 2.1535534858703613, + "learning_rate": 1.2956204379562045e-06, + "loss": 2.5781, + "step": 85 + }, + { + "epoch": 0.09424657534246575, + "grad_norm": 2.5001144409179688, + "learning_rate": 1.3138686131386864e-06, + "loss": 2.5339, + "step": 86 + }, + { + "epoch": 0.09534246575342466, + "grad_norm": 2.3279049396514893, + "learning_rate": 1.332116788321168e-06, + "loss": 2.545, + "step": 87 + }, + { + "epoch": 0.09643835616438357, + "grad_norm": 2.2230734825134277, + "learning_rate": 1.3503649635036497e-06, + "loss": 2.4745, + "step": 88 + }, + { + "epoch": 0.09753424657534246, + "grad_norm": 2.8762636184692383, + "learning_rate": 1.3686131386861315e-06, + "loss": 2.516, + "step": 89 + }, + { + "epoch": 0.09863013698630137, + "grad_norm": 2.153506278991699, + "learning_rate": 1.3868613138686132e-06, + "loss": 2.4547, + "step": 90 + }, + { + "epoch": 0.09863013698630137, + "eval_loss": 2.603238344192505, + "eval_runtime": 37.4162, + "eval_samples_per_second": 26.726, + "eval_steps_per_second": 1.123, + "step": 90 + }, + { + "epoch": 0.09972602739726028, + "grad_norm": 2.4143614768981934, + "learning_rate": 1.405109489051095e-06, + "loss": 2.547, + "step": 91 + }, + { + "epoch": 0.10082191780821918, + "grad_norm": 2.354632616043091, + "learning_rate": 1.4233576642335767e-06, + "loss": 2.5009, + "step": 92 + }, + { + "epoch": 0.10191780821917808, + "grad_norm": 2.6605563163757324, + "learning_rate": 1.4416058394160583e-06, + "loss": 2.5304, + "step": 93 + }, + { + "epoch": 0.10301369863013699, + "grad_norm": 2.3724122047424316, + "learning_rate": 1.4598540145985402e-06, + "loss": 2.4925, + "step": 94 + }, + { + "epoch": 0.10410958904109589, + "grad_norm": 3.6323022842407227, + "learning_rate": 1.478102189781022e-06, + "loss": 2.4592, + "step": 95 + }, + { + "epoch": 0.1052054794520548, + "grad_norm": 2.3407089710235596, + "learning_rate": 1.496350364963504e-06, + "loss": 2.5223, + "step": 96 + }, + { + "epoch": 0.1063013698630137, + "grad_norm": 2.8238413333892822, + "learning_rate": 1.5145985401459856e-06, + "loss": 2.489, + "step": 97 + }, + { + "epoch": 0.1073972602739726, + "grad_norm": 2.6433658599853516, + "learning_rate": 1.5328467153284674e-06, + "loss": 2.5101, + "step": 98 + }, + { + "epoch": 0.10849315068493151, + "grad_norm": 2.562761068344116, + "learning_rate": 1.551094890510949e-06, + "loss": 2.4642, + "step": 99 + }, + { + "epoch": 0.1095890410958904, + "grad_norm": 2.4501149654388428, + "learning_rate": 1.5693430656934307e-06, + "loss": 2.4102, + "step": 100 + }, + { + "epoch": 0.1095890410958904, + "eval_loss": 2.5452804565429688, + "eval_runtime": 37.4224, + "eval_samples_per_second": 26.722, + "eval_steps_per_second": 1.122, + "step": 100 + }, + { + "epoch": 0.11068493150684931, + "grad_norm": 2.475864887237549, + "learning_rate": 1.5875912408759126e-06, + "loss": 2.4147, + "step": 101 + }, + { + "epoch": 0.11178082191780822, + "grad_norm": 2.558875322341919, + "learning_rate": 1.6058394160583942e-06, + "loss": 2.4694, + "step": 102 + }, + { + "epoch": 0.11287671232876713, + "grad_norm": 2.1085734367370605, + "learning_rate": 1.624087591240876e-06, + "loss": 2.5099, + "step": 103 + }, + { + "epoch": 0.11397260273972602, + "grad_norm": 2.410731792449951, + "learning_rate": 1.6423357664233577e-06, + "loss": 2.432, + "step": 104 + }, + { + "epoch": 0.11506849315068493, + "grad_norm": 2.2800745964050293, + "learning_rate": 1.6605839416058394e-06, + "loss": 2.3669, + "step": 105 + }, + { + "epoch": 0.11616438356164384, + "grad_norm": 2.2234933376312256, + "learning_rate": 1.6788321167883212e-06, + "loss": 2.4145, + "step": 106 + }, + { + "epoch": 0.11726027397260275, + "grad_norm": 2.3367040157318115, + "learning_rate": 1.6970802919708029e-06, + "loss": 2.4272, + "step": 107 + }, + { + "epoch": 0.11835616438356164, + "grad_norm": 2.7002811431884766, + "learning_rate": 1.715328467153285e-06, + "loss": 2.3478, + "step": 108 + }, + { + "epoch": 0.11945205479452055, + "grad_norm": 2.6628332138061523, + "learning_rate": 1.7335766423357666e-06, + "loss": 2.4096, + "step": 109 + }, + { + "epoch": 0.12054794520547946, + "grad_norm": 2.4174230098724365, + "learning_rate": 1.7518248175182485e-06, + "loss": 2.407, + "step": 110 + }, + { + "epoch": 0.12054794520547946, + "eval_loss": 2.491746664047241, + "eval_runtime": 37.2555, + "eval_samples_per_second": 26.842, + "eval_steps_per_second": 1.127, + "step": 110 + }, + { + "epoch": 0.12164383561643835, + "grad_norm": 2.24796986579895, + "learning_rate": 1.77007299270073e-06, + "loss": 2.4369, + "step": 111 + }, + { + "epoch": 0.12273972602739726, + "grad_norm": 2.720939874649048, + "learning_rate": 1.788321167883212e-06, + "loss": 2.4116, + "step": 112 + }, + { + "epoch": 0.12383561643835617, + "grad_norm": 2.6907670497894287, + "learning_rate": 1.8065693430656936e-06, + "loss": 2.4577, + "step": 113 + }, + { + "epoch": 0.12493150684931507, + "grad_norm": 2.559636354446411, + "learning_rate": 1.8248175182481753e-06, + "loss": 2.3999, + "step": 114 + }, + { + "epoch": 0.12602739726027398, + "grad_norm": 2.6730551719665527, + "learning_rate": 1.8430656934306571e-06, + "loss": 2.3428, + "step": 115 + }, + { + "epoch": 0.12712328767123288, + "grad_norm": 3.172179937362671, + "learning_rate": 1.8613138686131388e-06, + "loss": 2.3722, + "step": 116 + }, + { + "epoch": 0.12821917808219177, + "grad_norm": 2.6757171154022217, + "learning_rate": 1.8795620437956206e-06, + "loss": 2.3572, + "step": 117 + }, + { + "epoch": 0.1293150684931507, + "grad_norm": 2.5840766429901123, + "learning_rate": 1.8978102189781023e-06, + "loss": 2.353, + "step": 118 + }, + { + "epoch": 0.1304109589041096, + "grad_norm": 2.6248581409454346, + "learning_rate": 1.916058394160584e-06, + "loss": 2.36, + "step": 119 + }, + { + "epoch": 0.13150684931506848, + "grad_norm": 2.428516387939453, + "learning_rate": 1.934306569343066e-06, + "loss": 2.3821, + "step": 120 + }, + { + "epoch": 0.13150684931506848, + "eval_loss": 2.451565980911255, + "eval_runtime": 37.3669, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.1326027397260274, + "grad_norm": 2.9956233501434326, + "learning_rate": 1.952554744525548e-06, + "loss": 2.2778, + "step": 121 + }, + { + "epoch": 0.1336986301369863, + "grad_norm": 2.4120893478393555, + "learning_rate": 1.9708029197080293e-06, + "loss": 2.325, + "step": 122 + }, + { + "epoch": 0.13479452054794522, + "grad_norm": 2.1410024166107178, + "learning_rate": 1.989051094890511e-06, + "loss": 2.3248, + "step": 123 + }, + { + "epoch": 0.1358904109589041, + "grad_norm": 2.923799514770508, + "learning_rate": 2.007299270072993e-06, + "loss": 2.327, + "step": 124 + }, + { + "epoch": 0.136986301369863, + "grad_norm": 3.0770339965820312, + "learning_rate": 2.025547445255475e-06, + "loss": 2.3847, + "step": 125 + }, + { + "epoch": 0.13808219178082193, + "grad_norm": 2.6408181190490723, + "learning_rate": 2.0437956204379563e-06, + "loss": 2.3196, + "step": 126 + }, + { + "epoch": 0.13917808219178082, + "grad_norm": 2.9127118587493896, + "learning_rate": 2.062043795620438e-06, + "loss": 2.3287, + "step": 127 + }, + { + "epoch": 0.14027397260273972, + "grad_norm": 2.9805052280426025, + "learning_rate": 2.08029197080292e-06, + "loss": 2.3243, + "step": 128 + }, + { + "epoch": 0.14136986301369864, + "grad_norm": 3.1342010498046875, + "learning_rate": 2.0985401459854014e-06, + "loss": 2.38, + "step": 129 + }, + { + "epoch": 0.14246575342465753, + "grad_norm": 2.678422451019287, + "learning_rate": 2.1167883211678833e-06, + "loss": 2.2727, + "step": 130 + }, + { + "epoch": 0.14246575342465753, + "eval_loss": 2.4187021255493164, + "eval_runtime": 37.3117, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 130 + }, + { + "epoch": 0.14356164383561643, + "grad_norm": 2.9281492233276367, + "learning_rate": 2.135036496350365e-06, + "loss": 2.3327, + "step": 131 + }, + { + "epoch": 0.14465753424657535, + "grad_norm": 2.9592933654785156, + "learning_rate": 2.1532846715328466e-06, + "loss": 2.3674, + "step": 132 + }, + { + "epoch": 0.14575342465753424, + "grad_norm": 3.404761552810669, + "learning_rate": 2.171532846715329e-06, + "loss": 2.2679, + "step": 133 + }, + { + "epoch": 0.14684931506849316, + "grad_norm": 3.635571241378784, + "learning_rate": 2.1897810218978103e-06, + "loss": 2.2799, + "step": 134 + }, + { + "epoch": 0.14794520547945206, + "grad_norm": 3.9130170345306396, + "learning_rate": 2.208029197080292e-06, + "loss": 2.3263, + "step": 135 + }, + { + "epoch": 0.14904109589041095, + "grad_norm": 4.300438404083252, + "learning_rate": 2.226277372262774e-06, + "loss": 2.2969, + "step": 136 + }, + { + "epoch": 0.15013698630136987, + "grad_norm": 2.863511085510254, + "learning_rate": 2.244525547445256e-06, + "loss": 2.297, + "step": 137 + }, + { + "epoch": 0.15123287671232877, + "grad_norm": 4.910321235656738, + "learning_rate": 2.2627737226277373e-06, + "loss": 2.3269, + "step": 138 + }, + { + "epoch": 0.15232876712328766, + "grad_norm": 3.3323862552642822, + "learning_rate": 2.281021897810219e-06, + "loss": 2.3195, + "step": 139 + }, + { + "epoch": 0.15342465753424658, + "grad_norm": 3.2672693729400635, + "learning_rate": 2.299270072992701e-06, + "loss": 2.2797, + "step": 140 + }, + { + "epoch": 0.15342465753424658, + "eval_loss": 2.3943886756896973, + "eval_runtime": 37.3137, + "eval_samples_per_second": 26.8, + "eval_steps_per_second": 1.126, + "step": 140 + }, + { + "epoch": 0.15452054794520548, + "grad_norm": 3.9229774475097656, + "learning_rate": 2.3175182481751825e-06, + "loss": 2.2718, + "step": 141 + }, + { + "epoch": 0.15561643835616437, + "grad_norm": 2.6409823894500732, + "learning_rate": 2.3357664233576643e-06, + "loss": 2.2828, + "step": 142 + }, + { + "epoch": 0.1567123287671233, + "grad_norm": 4.6122612953186035, + "learning_rate": 2.354014598540146e-06, + "loss": 2.247, + "step": 143 + }, + { + "epoch": 0.1578082191780822, + "grad_norm": 3.391357898712158, + "learning_rate": 2.3722627737226276e-06, + "loss": 2.2863, + "step": 144 + }, + { + "epoch": 0.1589041095890411, + "grad_norm": 3.3550853729248047, + "learning_rate": 2.39051094890511e-06, + "loss": 2.2789, + "step": 145 + }, + { + "epoch": 0.16, + "grad_norm": 3.7376883029937744, + "learning_rate": 2.4087591240875918e-06, + "loss": 2.2297, + "step": 146 + }, + { + "epoch": 0.1610958904109589, + "grad_norm": 3.005157709121704, + "learning_rate": 2.427007299270073e-06, + "loss": 2.2467, + "step": 147 + }, + { + "epoch": 0.16219178082191782, + "grad_norm": 3.4616103172302246, + "learning_rate": 2.445255474452555e-06, + "loss": 2.2612, + "step": 148 + }, + { + "epoch": 0.1632876712328767, + "grad_norm": 3.3595101833343506, + "learning_rate": 2.463503649635037e-06, + "loss": 2.3021, + "step": 149 + }, + { + "epoch": 0.1643835616438356, + "grad_norm": 4.831241607666016, + "learning_rate": 2.4817518248175183e-06, + "loss": 2.2174, + "step": 150 + }, + { + "epoch": 0.1643835616438356, + "eval_loss": 2.3677749633789062, + "eval_runtime": 37.3092, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 150 + }, + { + "epoch": 0.16547945205479453, + "grad_norm": 2.778820037841797, + "learning_rate": 2.5e-06, + "loss": 2.1856, + "step": 151 + }, + { + "epoch": 0.16657534246575342, + "grad_norm": 4.566643714904785, + "learning_rate": 2.518248175182482e-06, + "loss": 2.3186, + "step": 152 + }, + { + "epoch": 0.16767123287671232, + "grad_norm": 2.9908089637756348, + "learning_rate": 2.5364963503649635e-06, + "loss": 2.2127, + "step": 153 + }, + { + "epoch": 0.16876712328767124, + "grad_norm": 2.9702792167663574, + "learning_rate": 2.5547445255474458e-06, + "loss": 2.2071, + "step": 154 + }, + { + "epoch": 0.16986301369863013, + "grad_norm": 2.850044012069702, + "learning_rate": 2.5729927007299272e-06, + "loss": 2.2585, + "step": 155 + }, + { + "epoch": 0.17095890410958905, + "grad_norm": 2.926091432571411, + "learning_rate": 2.591240875912409e-06, + "loss": 2.2397, + "step": 156 + }, + { + "epoch": 0.17205479452054795, + "grad_norm": 2.8720169067382812, + "learning_rate": 2.6094890510948905e-06, + "loss": 2.2306, + "step": 157 + }, + { + "epoch": 0.17315068493150684, + "grad_norm": 2.8837177753448486, + "learning_rate": 2.627737226277373e-06, + "loss": 2.2666, + "step": 158 + }, + { + "epoch": 0.17424657534246576, + "grad_norm": 3.119621753692627, + "learning_rate": 2.6459854014598542e-06, + "loss": 2.2722, + "step": 159 + }, + { + "epoch": 0.17534246575342466, + "grad_norm": 2.7085700035095215, + "learning_rate": 2.664233576642336e-06, + "loss": 2.2346, + "step": 160 + }, + { + "epoch": 0.17534246575342466, + "eval_loss": 2.352083921432495, + "eval_runtime": 37.245, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 160 + }, + { + "epoch": 0.17643835616438355, + "grad_norm": 2.9964346885681152, + "learning_rate": 2.6824817518248175e-06, + "loss": 2.2155, + "step": 161 + }, + { + "epoch": 0.17753424657534247, + "grad_norm": 3.1535208225250244, + "learning_rate": 2.7007299270072994e-06, + "loss": 2.235, + "step": 162 + }, + { + "epoch": 0.17863013698630137, + "grad_norm": 2.7240943908691406, + "learning_rate": 2.718978102189781e-06, + "loss": 2.2095, + "step": 163 + }, + { + "epoch": 0.17972602739726026, + "grad_norm": 3.122654676437378, + "learning_rate": 2.737226277372263e-06, + "loss": 2.2667, + "step": 164 + }, + { + "epoch": 0.18082191780821918, + "grad_norm": 2.9718410968780518, + "learning_rate": 2.755474452554745e-06, + "loss": 2.2499, + "step": 165 + }, + { + "epoch": 0.18191780821917808, + "grad_norm": 3.0081896781921387, + "learning_rate": 2.7737226277372264e-06, + "loss": 2.2089, + "step": 166 + }, + { + "epoch": 0.18301369863013697, + "grad_norm": 2.829509973526001, + "learning_rate": 2.7919708029197082e-06, + "loss": 2.3139, + "step": 167 + }, + { + "epoch": 0.1841095890410959, + "grad_norm": 3.4480397701263428, + "learning_rate": 2.81021897810219e-06, + "loss": 2.2705, + "step": 168 + }, + { + "epoch": 0.1852054794520548, + "grad_norm": 3.1107025146484375, + "learning_rate": 2.828467153284672e-06, + "loss": 2.1895, + "step": 169 + }, + { + "epoch": 0.1863013698630137, + "grad_norm": 2.916790008544922, + "learning_rate": 2.8467153284671534e-06, + "loss": 2.2317, + "step": 170 + }, + { + "epoch": 0.1863013698630137, + "eval_loss": 2.336153030395508, + "eval_runtime": 37.2401, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 170 + }, + { + "epoch": 0.1873972602739726, + "grad_norm": 3.2582287788391113, + "learning_rate": 2.8649635036496353e-06, + "loss": 2.143, + "step": 171 + }, + { + "epoch": 0.1884931506849315, + "grad_norm": 2.430973768234253, + "learning_rate": 2.8832116788321167e-06, + "loss": 2.1977, + "step": 172 + }, + { + "epoch": 0.18958904109589042, + "grad_norm": 2.8763599395751953, + "learning_rate": 2.901459854014599e-06, + "loss": 2.1551, + "step": 173 + }, + { + "epoch": 0.19068493150684931, + "grad_norm": 2.9242653846740723, + "learning_rate": 2.9197080291970804e-06, + "loss": 2.2569, + "step": 174 + }, + { + "epoch": 0.1917808219178082, + "grad_norm": 2.623551607131958, + "learning_rate": 2.9379562043795623e-06, + "loss": 2.2329, + "step": 175 + }, + { + "epoch": 0.19287671232876713, + "grad_norm": 2.9511337280273438, + "learning_rate": 2.956204379562044e-06, + "loss": 2.2154, + "step": 176 + }, + { + "epoch": 0.19397260273972602, + "grad_norm": 2.415118932723999, + "learning_rate": 2.9744525547445256e-06, + "loss": 2.2637, + "step": 177 + }, + { + "epoch": 0.19506849315068492, + "grad_norm": 2.9664924144744873, + "learning_rate": 2.992700729927008e-06, + "loss": 2.2263, + "step": 178 + }, + { + "epoch": 0.19616438356164384, + "grad_norm": 2.7214572429656982, + "learning_rate": 3.0109489051094893e-06, + "loss": 2.1667, + "step": 179 + }, + { + "epoch": 0.19726027397260273, + "grad_norm": 2.8712823390960693, + "learning_rate": 3.029197080291971e-06, + "loss": 2.1661, + "step": 180 + }, + { + "epoch": 0.19726027397260273, + "eval_loss": 2.3191800117492676, + "eval_runtime": 37.24, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 180 + }, + { + "epoch": 0.19835616438356166, + "grad_norm": 2.959709405899048, + "learning_rate": 3.0474452554744526e-06, + "loss": 2.2483, + "step": 181 + }, + { + "epoch": 0.19945205479452055, + "grad_norm": 2.567089319229126, + "learning_rate": 3.065693430656935e-06, + "loss": 2.2382, + "step": 182 + }, + { + "epoch": 0.20054794520547944, + "grad_norm": 2.47188401222229, + "learning_rate": 3.0839416058394163e-06, + "loss": 2.2148, + "step": 183 + }, + { + "epoch": 0.20164383561643837, + "grad_norm": 2.3435261249542236, + "learning_rate": 3.102189781021898e-06, + "loss": 2.2334, + "step": 184 + }, + { + "epoch": 0.20273972602739726, + "grad_norm": 2.5280468463897705, + "learning_rate": 3.1204379562043796e-06, + "loss": 2.2214, + "step": 185 + }, + { + "epoch": 0.20383561643835615, + "grad_norm": 2.4808971881866455, + "learning_rate": 3.1386861313868614e-06, + "loss": 2.1554, + "step": 186 + }, + { + "epoch": 0.20493150684931508, + "grad_norm": 2.9937026500701904, + "learning_rate": 3.156934306569343e-06, + "loss": 2.1389, + "step": 187 + }, + { + "epoch": 0.20602739726027397, + "grad_norm": 2.9220476150512695, + "learning_rate": 3.175182481751825e-06, + "loss": 2.2515, + "step": 188 + }, + { + "epoch": 0.20712328767123286, + "grad_norm": 3.0449001789093018, + "learning_rate": 3.193430656934307e-06, + "loss": 2.2016, + "step": 189 + }, + { + "epoch": 0.20821917808219179, + "grad_norm": 2.8027074337005615, + "learning_rate": 3.2116788321167884e-06, + "loss": 2.1823, + "step": 190 + }, + { + "epoch": 0.20821917808219179, + "eval_loss": 2.309677839279175, + "eval_runtime": 37.2442, + "eval_samples_per_second": 26.85, + "eval_steps_per_second": 1.128, + "step": 190 + }, + { + "epoch": 0.20931506849315068, + "grad_norm": 2.9158318042755127, + "learning_rate": 3.2299270072992707e-06, + "loss": 2.1928, + "step": 191 + }, + { + "epoch": 0.2104109589041096, + "grad_norm": 2.5328080654144287, + "learning_rate": 3.248175182481752e-06, + "loss": 2.1828, + "step": 192 + }, + { + "epoch": 0.2115068493150685, + "grad_norm": 3.1709704399108887, + "learning_rate": 3.266423357664234e-06, + "loss": 2.1052, + "step": 193 + }, + { + "epoch": 0.2126027397260274, + "grad_norm": 2.9156367778778076, + "learning_rate": 3.2846715328467155e-06, + "loss": 2.1943, + "step": 194 + }, + { + "epoch": 0.2136986301369863, + "grad_norm": 2.879408836364746, + "learning_rate": 3.3029197080291973e-06, + "loss": 2.1782, + "step": 195 + }, + { + "epoch": 0.2147945205479452, + "grad_norm": 3.748483180999756, + "learning_rate": 3.3211678832116788e-06, + "loss": 2.2107, + "step": 196 + }, + { + "epoch": 0.2158904109589041, + "grad_norm": 3.197510004043579, + "learning_rate": 3.339416058394161e-06, + "loss": 2.1845, + "step": 197 + }, + { + "epoch": 0.21698630136986302, + "grad_norm": 3.2596049308776855, + "learning_rate": 3.3576642335766425e-06, + "loss": 2.1816, + "step": 198 + }, + { + "epoch": 0.21808219178082192, + "grad_norm": 3.3329780101776123, + "learning_rate": 3.3759124087591243e-06, + "loss": 2.181, + "step": 199 + }, + { + "epoch": 0.2191780821917808, + "grad_norm": 2.7302327156066895, + "learning_rate": 3.3941605839416058e-06, + "loss": 2.1761, + "step": 200 + }, + { + "epoch": 0.2191780821917808, + "eval_loss": 2.2909653186798096, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 200 + }, + { + "epoch": 0.22027397260273973, + "grad_norm": 3.002255916595459, + "learning_rate": 3.412408759124088e-06, + "loss": 2.1868, + "step": 201 + }, + { + "epoch": 0.22136986301369863, + "grad_norm": 2.6694891452789307, + "learning_rate": 3.43065693430657e-06, + "loss": 2.192, + "step": 202 + }, + { + "epoch": 0.22246575342465755, + "grad_norm": 2.9225986003875732, + "learning_rate": 3.4489051094890513e-06, + "loss": 2.1303, + "step": 203 + }, + { + "epoch": 0.22356164383561644, + "grad_norm": 2.7844080924987793, + "learning_rate": 3.467153284671533e-06, + "loss": 2.1788, + "step": 204 + }, + { + "epoch": 0.22465753424657534, + "grad_norm": 2.726123332977295, + "learning_rate": 3.4854014598540146e-06, + "loss": 2.1193, + "step": 205 + }, + { + "epoch": 0.22575342465753426, + "grad_norm": 2.838040590286255, + "learning_rate": 3.503649635036497e-06, + "loss": 2.1674, + "step": 206 + }, + { + "epoch": 0.22684931506849315, + "grad_norm": 2.6863937377929688, + "learning_rate": 3.5218978102189783e-06, + "loss": 2.161, + "step": 207 + }, + { + "epoch": 0.22794520547945205, + "grad_norm": 2.5116159915924072, + "learning_rate": 3.54014598540146e-06, + "loss": 2.179, + "step": 208 + }, + { + "epoch": 0.22904109589041097, + "grad_norm": 3.381863594055176, + "learning_rate": 3.5583941605839416e-06, + "loss": 2.2035, + "step": 209 + }, + { + "epoch": 0.23013698630136986, + "grad_norm": 3.2741382122039795, + "learning_rate": 3.576642335766424e-06, + "loss": 2.2518, + "step": 210 + }, + { + "epoch": 0.23013698630136986, + "eval_loss": 2.2773635387420654, + "eval_runtime": 37.2658, + "eval_samples_per_second": 26.834, + "eval_steps_per_second": 1.127, + "step": 210 + }, + { + "epoch": 0.23123287671232876, + "grad_norm": 2.917370319366455, + "learning_rate": 3.5948905109489054e-06, + "loss": 2.2265, + "step": 211 + }, + { + "epoch": 0.23232876712328768, + "grad_norm": 2.8392834663391113, + "learning_rate": 3.6131386861313872e-06, + "loss": 2.0709, + "step": 212 + }, + { + "epoch": 0.23342465753424657, + "grad_norm": 2.753612518310547, + "learning_rate": 3.6313868613138687e-06, + "loss": 2.0752, + "step": 213 + }, + { + "epoch": 0.2345205479452055, + "grad_norm": 3.1477746963500977, + "learning_rate": 3.6496350364963505e-06, + "loss": 2.1169, + "step": 214 + }, + { + "epoch": 0.2356164383561644, + "grad_norm": 2.5448100566864014, + "learning_rate": 3.667883211678833e-06, + "loss": 2.1338, + "step": 215 + }, + { + "epoch": 0.23671232876712328, + "grad_norm": 3.1622560024261475, + "learning_rate": 3.6861313868613142e-06, + "loss": 2.1065, + "step": 216 + }, + { + "epoch": 0.2378082191780822, + "grad_norm": 2.7054336071014404, + "learning_rate": 3.704379562043796e-06, + "loss": 2.1104, + "step": 217 + }, + { + "epoch": 0.2389041095890411, + "grad_norm": 2.964141845703125, + "learning_rate": 3.7226277372262775e-06, + "loss": 2.1949, + "step": 218 + }, + { + "epoch": 0.24, + "grad_norm": 2.7586112022399902, + "learning_rate": 3.7408759124087594e-06, + "loss": 2.165, + "step": 219 + }, + { + "epoch": 0.2410958904109589, + "grad_norm": 2.9035964012145996, + "learning_rate": 3.7591240875912412e-06, + "loss": 2.1192, + "step": 220 + }, + { + "epoch": 0.2410958904109589, + "eval_loss": 2.2649471759796143, + "eval_runtime": 37.4311, + "eval_samples_per_second": 26.716, + "eval_steps_per_second": 1.122, + "step": 220 + }, + { + "epoch": 0.2421917808219178, + "grad_norm": 2.4212076663970947, + "learning_rate": 3.777372262773723e-06, + "loss": 2.1389, + "step": 221 + }, + { + "epoch": 0.2432876712328767, + "grad_norm": 2.973227024078369, + "learning_rate": 3.7956204379562045e-06, + "loss": 2.132, + "step": 222 + }, + { + "epoch": 0.24438356164383562, + "grad_norm": 2.9831430912017822, + "learning_rate": 3.8138686131386864e-06, + "loss": 2.1142, + "step": 223 + }, + { + "epoch": 0.24547945205479452, + "grad_norm": 3.2880353927612305, + "learning_rate": 3.832116788321168e-06, + "loss": 2.168, + "step": 224 + }, + { + "epoch": 0.2465753424657534, + "grad_norm": 3.3920998573303223, + "learning_rate": 3.85036496350365e-06, + "loss": 2.1277, + "step": 225 + }, + { + "epoch": 0.24767123287671233, + "grad_norm": 2.7170026302337646, + "learning_rate": 3.868613138686132e-06, + "loss": 2.0936, + "step": 226 + }, + { + "epoch": 0.24876712328767123, + "grad_norm": 3.6220011711120605, + "learning_rate": 3.886861313868613e-06, + "loss": 2.1259, + "step": 227 + }, + { + "epoch": 0.24986301369863015, + "grad_norm": 3.4229273796081543, + "learning_rate": 3.905109489051096e-06, + "loss": 2.1421, + "step": 228 + }, + { + "epoch": 0.25095890410958904, + "grad_norm": 3.0358338356018066, + "learning_rate": 3.923357664233577e-06, + "loss": 2.1747, + "step": 229 + }, + { + "epoch": 0.25205479452054796, + "grad_norm": 2.611771821975708, + "learning_rate": 3.9416058394160585e-06, + "loss": 2.1282, + "step": 230 + }, + { + "epoch": 0.25205479452054796, + "eval_loss": 2.254887580871582, + "eval_runtime": 37.5943, + "eval_samples_per_second": 26.6, + "eval_steps_per_second": 1.117, + "step": 230 + }, + { + "epoch": 0.25315068493150683, + "grad_norm": 3.846412181854248, + "learning_rate": 3.95985401459854e-06, + "loss": 2.1892, + "step": 231 + }, + { + "epoch": 0.25424657534246575, + "grad_norm": 2.488525390625, + "learning_rate": 3.978102189781022e-06, + "loss": 2.2064, + "step": 232 + }, + { + "epoch": 0.2553424657534247, + "grad_norm": 3.114964485168457, + "learning_rate": 3.996350364963504e-06, + "loss": 2.0653, + "step": 233 + }, + { + "epoch": 0.25643835616438354, + "grad_norm": 2.5559768676757812, + "learning_rate": 4.014598540145986e-06, + "loss": 2.1503, + "step": 234 + }, + { + "epoch": 0.25753424657534246, + "grad_norm": 3.2347540855407715, + "learning_rate": 4.032846715328467e-06, + "loss": 2.0523, + "step": 235 + }, + { + "epoch": 0.2586301369863014, + "grad_norm": 2.8086111545562744, + "learning_rate": 4.05109489051095e-06, + "loss": 2.1206, + "step": 236 + }, + { + "epoch": 0.25972602739726025, + "grad_norm": 2.909231185913086, + "learning_rate": 4.069343065693431e-06, + "loss": 2.139, + "step": 237 + }, + { + "epoch": 0.2608219178082192, + "grad_norm": 3.2171082496643066, + "learning_rate": 4.0875912408759126e-06, + "loss": 2.0908, + "step": 238 + }, + { + "epoch": 0.2619178082191781, + "grad_norm": 2.3851265907287598, + "learning_rate": 4.105839416058395e-06, + "loss": 2.1133, + "step": 239 + }, + { + "epoch": 0.26301369863013696, + "grad_norm": 3.3912737369537354, + "learning_rate": 4.124087591240876e-06, + "loss": 2.1911, + "step": 240 + }, + { + "epoch": 0.26301369863013696, + "eval_loss": 2.2436957359313965, + "eval_runtime": 37.2947, + "eval_samples_per_second": 26.813, + "eval_steps_per_second": 1.126, + "step": 240 + }, + { + "epoch": 0.2641095890410959, + "grad_norm": 2.82906174659729, + "learning_rate": 4.1423357664233586e-06, + "loss": 2.11, + "step": 241 + }, + { + "epoch": 0.2652054794520548, + "grad_norm": 3.0317952632904053, + "learning_rate": 4.16058394160584e-06, + "loss": 2.0953, + "step": 242 + }, + { + "epoch": 0.2663013698630137, + "grad_norm": 3.6438329219818115, + "learning_rate": 4.1788321167883214e-06, + "loss": 2.1746, + "step": 243 + }, + { + "epoch": 0.2673972602739726, + "grad_norm": 2.6690592765808105, + "learning_rate": 4.197080291970803e-06, + "loss": 2.0555, + "step": 244 + }, + { + "epoch": 0.2684931506849315, + "grad_norm": 3.135376453399658, + "learning_rate": 4.215328467153285e-06, + "loss": 2.1394, + "step": 245 + }, + { + "epoch": 0.26958904109589044, + "grad_norm": 2.5474987030029297, + "learning_rate": 4.233576642335767e-06, + "loss": 2.0822, + "step": 246 + }, + { + "epoch": 0.2706849315068493, + "grad_norm": 2.883108377456665, + "learning_rate": 4.251824817518249e-06, + "loss": 2.0869, + "step": 247 + }, + { + "epoch": 0.2717808219178082, + "grad_norm": 2.9801037311553955, + "learning_rate": 4.27007299270073e-06, + "loss": 2.1194, + "step": 248 + }, + { + "epoch": 0.27287671232876715, + "grad_norm": 2.7688217163085938, + "learning_rate": 4.288321167883212e-06, + "loss": 2.0908, + "step": 249 + }, + { + "epoch": 0.273972602739726, + "grad_norm": 3.205148458480835, + "learning_rate": 4.306569343065693e-06, + "loss": 2.0762, + "step": 250 + }, + { + "epoch": 0.273972602739726, + "eval_loss": 2.238699197769165, + "eval_runtime": 37.2428, + "eval_samples_per_second": 26.851, + "eval_steps_per_second": 1.128, + "step": 250 + }, + { + "epoch": 0.27506849315068493, + "grad_norm": 2.8262946605682373, + "learning_rate": 4.3248175182481755e-06, + "loss": 2.0509, + "step": 251 + }, + { + "epoch": 0.27616438356164386, + "grad_norm": 3.156261682510376, + "learning_rate": 4.343065693430658e-06, + "loss": 2.1213, + "step": 252 + }, + { + "epoch": 0.2772602739726027, + "grad_norm": 2.6367998123168945, + "learning_rate": 4.361313868613139e-06, + "loss": 2.1071, + "step": 253 + }, + { + "epoch": 0.27835616438356164, + "grad_norm": 2.871534824371338, + "learning_rate": 4.379562043795621e-06, + "loss": 2.1008, + "step": 254 + }, + { + "epoch": 0.27945205479452057, + "grad_norm": 2.839432716369629, + "learning_rate": 4.397810218978102e-06, + "loss": 2.1293, + "step": 255 + }, + { + "epoch": 0.28054794520547943, + "grad_norm": 2.627143383026123, + "learning_rate": 4.416058394160584e-06, + "loss": 2.1141, + "step": 256 + }, + { + "epoch": 0.28164383561643835, + "grad_norm": 2.8292996883392334, + "learning_rate": 4.434306569343066e-06, + "loss": 2.0981, + "step": 257 + }, + { + "epoch": 0.2827397260273973, + "grad_norm": 2.5711069107055664, + "learning_rate": 4.452554744525548e-06, + "loss": 2.0707, + "step": 258 + }, + { + "epoch": 0.28383561643835614, + "grad_norm": 2.488954544067383, + "learning_rate": 4.4708029197080295e-06, + "loss": 2.1172, + "step": 259 + }, + { + "epoch": 0.28493150684931506, + "grad_norm": 2.442209005355835, + "learning_rate": 4.489051094890512e-06, + "loss": 2.1119, + "step": 260 + }, + { + "epoch": 0.28493150684931506, + "eval_loss": 2.2306911945343018, + "eval_runtime": 37.2412, + "eval_samples_per_second": 26.852, + "eval_steps_per_second": 1.128, + "step": 260 + }, + { + "epoch": 0.286027397260274, + "grad_norm": 2.9147934913635254, + "learning_rate": 4.507299270072993e-06, + "loss": 2.1517, + "step": 261 + }, + { + "epoch": 0.28712328767123285, + "grad_norm": 2.549833297729492, + "learning_rate": 4.525547445255475e-06, + "loss": 2.1047, + "step": 262 + }, + { + "epoch": 0.2882191780821918, + "grad_norm": 3.3405206203460693, + "learning_rate": 4.543795620437956e-06, + "loss": 2.0975, + "step": 263 + }, + { + "epoch": 0.2893150684931507, + "grad_norm": 2.554431676864624, + "learning_rate": 4.562043795620438e-06, + "loss": 2.0622, + "step": 264 + }, + { + "epoch": 0.29041095890410956, + "grad_norm": 3.5529873371124268, + "learning_rate": 4.580291970802921e-06, + "loss": 2.1567, + "step": 265 + }, + { + "epoch": 0.2915068493150685, + "grad_norm": 2.421682119369507, + "learning_rate": 4.598540145985402e-06, + "loss": 2.1619, + "step": 266 + }, + { + "epoch": 0.2926027397260274, + "grad_norm": 2.918288230895996, + "learning_rate": 4.6167883211678835e-06, + "loss": 2.1238, + "step": 267 + }, + { + "epoch": 0.2936986301369863, + "grad_norm": 3.0737621784210205, + "learning_rate": 4.635036496350365e-06, + "loss": 2.0728, + "step": 268 + }, + { + "epoch": 0.2947945205479452, + "grad_norm": 2.5862410068511963, + "learning_rate": 4.653284671532847e-06, + "loss": 2.0675, + "step": 269 + }, + { + "epoch": 0.2958904109589041, + "grad_norm": 2.9004979133605957, + "learning_rate": 4.671532846715329e-06, + "loss": 2.1248, + "step": 270 + }, + { + "epoch": 0.2958904109589041, + "eval_loss": 2.2179365158081055, + "eval_runtime": 37.2352, + "eval_samples_per_second": 26.856, + "eval_steps_per_second": 1.128, + "step": 270 + }, + { + "epoch": 0.29698630136986304, + "grad_norm": 2.4784982204437256, + "learning_rate": 4.689781021897811e-06, + "loss": 2.1276, + "step": 271 + }, + { + "epoch": 0.2980821917808219, + "grad_norm": 2.7971739768981934, + "learning_rate": 4.708029197080292e-06, + "loss": 2.0931, + "step": 272 + }, + { + "epoch": 0.2991780821917808, + "grad_norm": 2.5839481353759766, + "learning_rate": 4.726277372262774e-06, + "loss": 2.1371, + "step": 273 + }, + { + "epoch": 0.30027397260273975, + "grad_norm": 2.7168564796447754, + "learning_rate": 4.744525547445255e-06, + "loss": 2.1086, + "step": 274 + }, + { + "epoch": 0.3013698630136986, + "grad_norm": 2.6566879749298096, + "learning_rate": 4.7627737226277375e-06, + "loss": 2.0368, + "step": 275 + }, + { + "epoch": 0.30246575342465754, + "grad_norm": 2.5068323612213135, + "learning_rate": 4.78102189781022e-06, + "loss": 2.1077, + "step": 276 + }, + { + "epoch": 0.30356164383561646, + "grad_norm": 2.631408214569092, + "learning_rate": 4.799270072992701e-06, + "loss": 2.1013, + "step": 277 + }, + { + "epoch": 0.3046575342465753, + "grad_norm": 2.5642812252044678, + "learning_rate": 4.8175182481751835e-06, + "loss": 2.0739, + "step": 278 + }, + { + "epoch": 0.30575342465753425, + "grad_norm": 2.3326497077941895, + "learning_rate": 4.835766423357665e-06, + "loss": 2.0971, + "step": 279 + }, + { + "epoch": 0.30684931506849317, + "grad_norm": 2.70701265335083, + "learning_rate": 4.854014598540146e-06, + "loss": 2.0904, + "step": 280 + }, + { + "epoch": 0.30684931506849317, + "eval_loss": 2.215972900390625, + "eval_runtime": 37.2196, + "eval_samples_per_second": 26.868, + "eval_steps_per_second": 1.128, + "step": 280 + }, + { + "epoch": 0.30794520547945203, + "grad_norm": 2.615201234817505, + "learning_rate": 4.872262773722628e-06, + "loss": 2.039, + "step": 281 + }, + { + "epoch": 0.30904109589041096, + "grad_norm": 2.2821998596191406, + "learning_rate": 4.89051094890511e-06, + "loss": 2.0529, + "step": 282 + }, + { + "epoch": 0.3101369863013699, + "grad_norm": 2.767153024673462, + "learning_rate": 4.9087591240875915e-06, + "loss": 2.0351, + "step": 283 + }, + { + "epoch": 0.31123287671232874, + "grad_norm": 2.537121295928955, + "learning_rate": 4.927007299270074e-06, + "loss": 2.0752, + "step": 284 + }, + { + "epoch": 0.31232876712328766, + "grad_norm": 2.835052013397217, + "learning_rate": 4.945255474452555e-06, + "loss": 2.0892, + "step": 285 + }, + { + "epoch": 0.3134246575342466, + "grad_norm": 2.5154924392700195, + "learning_rate": 4.963503649635037e-06, + "loss": 2.0569, + "step": 286 + }, + { + "epoch": 0.31452054794520545, + "grad_norm": 2.8212568759918213, + "learning_rate": 4.981751824817518e-06, + "loss": 2.1588, + "step": 287 + }, + { + "epoch": 0.3156164383561644, + "grad_norm": 2.3840994834899902, + "learning_rate": 5e-06, + "loss": 2.1409, + "step": 288 + }, + { + "epoch": 0.3167123287671233, + "grad_norm": 2.9641315937042236, + "learning_rate": 4.997969130787978e-06, + "loss": 2.0892, + "step": 289 + }, + { + "epoch": 0.3178082191780822, + "grad_norm": 2.4516749382019043, + "learning_rate": 4.995938261575955e-06, + "loss": 2.1105, + "step": 290 + }, + { + "epoch": 0.3178082191780822, + "eval_loss": 2.207075834274292, + "eval_runtime": 37.1304, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 290 + }, + { + "epoch": 0.3189041095890411, + "grad_norm": 2.768869400024414, + "learning_rate": 4.993907392363932e-06, + "loss": 2.0231, + "step": 291 + }, + { + "epoch": 0.32, + "grad_norm": 2.771475315093994, + "learning_rate": 4.9918765231519095e-06, + "loss": 2.1018, + "step": 292 + }, + { + "epoch": 0.32109589041095893, + "grad_norm": 2.773983955383301, + "learning_rate": 4.989845653939887e-06, + "loss": 2.0317, + "step": 293 + }, + { + "epoch": 0.3221917808219178, + "grad_norm": 2.4447343349456787, + "learning_rate": 4.987814784727864e-06, + "loss": 2.0739, + "step": 294 + }, + { + "epoch": 0.3232876712328767, + "grad_norm": 2.444261312484741, + "learning_rate": 4.985783915515841e-06, + "loss": 2.073, + "step": 295 + }, + { + "epoch": 0.32438356164383564, + "grad_norm": 2.871792793273926, + "learning_rate": 4.983753046303819e-06, + "loss": 2.058, + "step": 296 + }, + { + "epoch": 0.3254794520547945, + "grad_norm": 2.3651537895202637, + "learning_rate": 4.981722177091796e-06, + "loss": 2.0749, + "step": 297 + }, + { + "epoch": 0.3265753424657534, + "grad_norm": 2.3395845890045166, + "learning_rate": 4.979691307879773e-06, + "loss": 2.065, + "step": 298 + }, + { + "epoch": 0.32767123287671235, + "grad_norm": 2.6275217533111572, + "learning_rate": 4.9776604386677504e-06, + "loss": 2.0945, + "step": 299 + }, + { + "epoch": 0.3287671232876712, + "grad_norm": 2.866285562515259, + "learning_rate": 4.975629569455728e-06, + "loss": 2.0702, + "step": 300 + }, + { + "epoch": 0.3287671232876712, + "eval_loss": 2.200012683868408, + "eval_runtime": 37.2082, + "eval_samples_per_second": 26.876, + "eval_steps_per_second": 1.129, + "step": 300 + }, + { + "epoch": 0.32986301369863014, + "grad_norm": 2.268401861190796, + "learning_rate": 4.973598700243705e-06, + "loss": 2.1257, + "step": 301 + }, + { + "epoch": 0.33095890410958906, + "grad_norm": 2.6197733879089355, + "learning_rate": 4.971567831031681e-06, + "loss": 2.0347, + "step": 302 + }, + { + "epoch": 0.3320547945205479, + "grad_norm": 2.4738848209381104, + "learning_rate": 4.9695369618196595e-06, + "loss": 2.0368, + "step": 303 + }, + { + "epoch": 0.33315068493150685, + "grad_norm": 2.705670118331909, + "learning_rate": 4.967506092607636e-06, + "loss": 1.9256, + "step": 304 + }, + { + "epoch": 0.33424657534246577, + "grad_norm": 2.813279628753662, + "learning_rate": 4.965475223395614e-06, + "loss": 2.0214, + "step": 305 + }, + { + "epoch": 0.33534246575342463, + "grad_norm": 2.320960283279419, + "learning_rate": 4.9634443541835905e-06, + "loss": 2.0342, + "step": 306 + }, + { + "epoch": 0.33643835616438356, + "grad_norm": 2.4495227336883545, + "learning_rate": 4.961413484971569e-06, + "loss": 2.0421, + "step": 307 + }, + { + "epoch": 0.3375342465753425, + "grad_norm": 2.5384018421173096, + "learning_rate": 4.959382615759545e-06, + "loss": 2.0572, + "step": 308 + }, + { + "epoch": 0.33863013698630134, + "grad_norm": 2.4018006324768066, + "learning_rate": 4.957351746547523e-06, + "loss": 2.0617, + "step": 309 + }, + { + "epoch": 0.33972602739726027, + "grad_norm": 2.552424907684326, + "learning_rate": 4.9553208773355e-06, + "loss": 2.0384, + "step": 310 + }, + { + "epoch": 0.33972602739726027, + "eval_loss": 2.1974849700927734, + "eval_runtime": 37.2308, + "eval_samples_per_second": 26.859, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3408219178082192, + "grad_norm": 2.4587671756744385, + "learning_rate": 4.953290008123478e-06, + "loss": 2.1701, + "step": 311 + }, + { + "epoch": 0.3419178082191781, + "grad_norm": 2.4486496448516846, + "learning_rate": 4.951259138911454e-06, + "loss": 2.0654, + "step": 312 + }, + { + "epoch": 0.343013698630137, + "grad_norm": 2.123568058013916, + "learning_rate": 4.9492282696994315e-06, + "loss": 1.9961, + "step": 313 + }, + { + "epoch": 0.3441095890410959, + "grad_norm": 2.549788475036621, + "learning_rate": 4.947197400487409e-06, + "loss": 2.1265, + "step": 314 + }, + { + "epoch": 0.3452054794520548, + "grad_norm": 2.3453609943389893, + "learning_rate": 4.945166531275386e-06, + "loss": 2.0501, + "step": 315 + }, + { + "epoch": 0.3463013698630137, + "grad_norm": 2.3584256172180176, + "learning_rate": 4.943135662063363e-06, + "loss": 2.0402, + "step": 316 + }, + { + "epoch": 0.3473972602739726, + "grad_norm": 3.3100943565368652, + "learning_rate": 4.9411047928513406e-06, + "loss": 2.0545, + "step": 317 + }, + { + "epoch": 0.34849315068493153, + "grad_norm": 2.4266037940979004, + "learning_rate": 4.939073923639318e-06, + "loss": 2.0926, + "step": 318 + }, + { + "epoch": 0.3495890410958904, + "grad_norm": 3.101090908050537, + "learning_rate": 4.937043054427295e-06, + "loss": 2.0987, + "step": 319 + }, + { + "epoch": 0.3506849315068493, + "grad_norm": 2.743619203567505, + "learning_rate": 4.935012185215272e-06, + "loss": 2.0315, + "step": 320 + }, + { + "epoch": 0.3506849315068493, + "eval_loss": 2.188877820968628, + "eval_runtime": 37.4609, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 320 + }, + { + "epoch": 0.35178082191780824, + "grad_norm": 2.6065986156463623, + "learning_rate": 4.93298131600325e-06, + "loss": 2.0166, + "step": 321 + }, + { + "epoch": 0.3528767123287671, + "grad_norm": 2.7554283142089844, + "learning_rate": 4.930950446791227e-06, + "loss": 2.0862, + "step": 322 + }, + { + "epoch": 0.353972602739726, + "grad_norm": 2.4773144721984863, + "learning_rate": 4.928919577579204e-06, + "loss": 2.0472, + "step": 323 + }, + { + "epoch": 0.35506849315068495, + "grad_norm": 2.695289134979248, + "learning_rate": 4.9268887083671815e-06, + "loss": 2.0567, + "step": 324 + }, + { + "epoch": 0.3561643835616438, + "grad_norm": 2.408220052719116, + "learning_rate": 4.924857839155159e-06, + "loss": 2.046, + "step": 325 + }, + { + "epoch": 0.35726027397260274, + "grad_norm": 2.3851559162139893, + "learning_rate": 4.922826969943136e-06, + "loss": 2.054, + "step": 326 + }, + { + "epoch": 0.35835616438356166, + "grad_norm": 2.8667280673980713, + "learning_rate": 4.920796100731113e-06, + "loss": 2.0717, + "step": 327 + }, + { + "epoch": 0.3594520547945205, + "grad_norm": 2.3814942836761475, + "learning_rate": 4.918765231519091e-06, + "loss": 1.9884, + "step": 328 + }, + { + "epoch": 0.36054794520547945, + "grad_norm": 2.341076612472534, + "learning_rate": 4.916734362307068e-06, + "loss": 2.0474, + "step": 329 + }, + { + "epoch": 0.36164383561643837, + "grad_norm": 2.3571853637695312, + "learning_rate": 4.914703493095045e-06, + "loss": 2.0009, + "step": 330 + }, + { + "epoch": 0.36164383561643837, + "eval_loss": 2.186048746109009, + "eval_runtime": 37.5746, + "eval_samples_per_second": 26.614, + "eval_steps_per_second": 1.118, + "step": 330 + }, + { + "epoch": 0.36273972602739724, + "grad_norm": 2.3770196437835693, + "learning_rate": 4.912672623883022e-06, + "loss": 2.0167, + "step": 331 + }, + { + "epoch": 0.36383561643835616, + "grad_norm": 2.3357701301574707, + "learning_rate": 4.910641754671e-06, + "loss": 1.9972, + "step": 332 + }, + { + "epoch": 0.3649315068493151, + "grad_norm": 2.4100499153137207, + "learning_rate": 4.908610885458977e-06, + "loss": 2.0319, + "step": 333 + }, + { + "epoch": 0.36602739726027395, + "grad_norm": 2.193018913269043, + "learning_rate": 4.906580016246954e-06, + "loss": 2.0285, + "step": 334 + }, + { + "epoch": 0.36712328767123287, + "grad_norm": 2.275637149810791, + "learning_rate": 4.9045491470349315e-06, + "loss": 2.0749, + "step": 335 + }, + { + "epoch": 0.3682191780821918, + "grad_norm": 2.347121238708496, + "learning_rate": 4.902518277822909e-06, + "loss": 2.0502, + "step": 336 + }, + { + "epoch": 0.3693150684931507, + "grad_norm": 2.210817575454712, + "learning_rate": 4.900487408610886e-06, + "loss": 2.0612, + "step": 337 + }, + { + "epoch": 0.3704109589041096, + "grad_norm": 2.15594482421875, + "learning_rate": 4.898456539398863e-06, + "loss": 2.0653, + "step": 338 + }, + { + "epoch": 0.3715068493150685, + "grad_norm": 2.3996052742004395, + "learning_rate": 4.89642567018684e-06, + "loss": 2.074, + "step": 339 + }, + { + "epoch": 0.3726027397260274, + "grad_norm": 2.384172201156616, + "learning_rate": 4.894394800974818e-06, + "loss": 2.1533, + "step": 340 + }, + { + "epoch": 0.3726027397260274, + "eval_loss": 2.175034284591675, + "eval_runtime": 37.2455, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 340 + }, + { + "epoch": 0.3736986301369863, + "grad_norm": 2.3431336879730225, + "learning_rate": 4.892363931762794e-06, + "loss": 1.9919, + "step": 341 + }, + { + "epoch": 0.3747945205479452, + "grad_norm": 2.1828389167785645, + "learning_rate": 4.8903330625507725e-06, + "loss": 2.0703, + "step": 342 + }, + { + "epoch": 0.37589041095890413, + "grad_norm": 2.2117249965667725, + "learning_rate": 4.888302193338749e-06, + "loss": 2.0071, + "step": 343 + }, + { + "epoch": 0.376986301369863, + "grad_norm": 2.3595967292785645, + "learning_rate": 4.886271324126727e-06, + "loss": 1.9879, + "step": 344 + }, + { + "epoch": 0.3780821917808219, + "grad_norm": 2.5211799144744873, + "learning_rate": 4.8842404549147034e-06, + "loss": 2.0861, + "step": 345 + }, + { + "epoch": 0.37917808219178084, + "grad_norm": 2.1551895141601562, + "learning_rate": 4.8822095857026816e-06, + "loss": 2.0683, + "step": 346 + }, + { + "epoch": 0.3802739726027397, + "grad_norm": 2.4145829677581787, + "learning_rate": 4.880178716490658e-06, + "loss": 2.043, + "step": 347 + }, + { + "epoch": 0.38136986301369863, + "grad_norm": 2.4273521900177, + "learning_rate": 4.878147847278636e-06, + "loss": 1.9928, + "step": 348 + }, + { + "epoch": 0.38246575342465755, + "grad_norm": 2.261054754257202, + "learning_rate": 4.8761169780666125e-06, + "loss": 2.0065, + "step": 349 + }, + { + "epoch": 0.3835616438356164, + "grad_norm": 2.4703147411346436, + "learning_rate": 4.874086108854591e-06, + "loss": 1.9766, + "step": 350 + }, + { + "epoch": 0.3835616438356164, + "eval_loss": 2.176915168762207, + "eval_runtime": 37.3223, + "eval_samples_per_second": 26.794, + "eval_steps_per_second": 1.125, + "step": 350 + }, + { + "epoch": 0.38465753424657534, + "grad_norm": 2.3353374004364014, + "learning_rate": 4.872055239642567e-06, + "loss": 1.9874, + "step": 351 + }, + { + "epoch": 0.38575342465753426, + "grad_norm": 2.227834463119507, + "learning_rate": 4.870024370430544e-06, + "loss": 1.9644, + "step": 352 + }, + { + "epoch": 0.3868493150684931, + "grad_norm": 2.3155159950256348, + "learning_rate": 4.867993501218522e-06, + "loss": 2.0227, + "step": 353 + }, + { + "epoch": 0.38794520547945205, + "grad_norm": 2.2456047534942627, + "learning_rate": 4.865962632006499e-06, + "loss": 2.0566, + "step": 354 + }, + { + "epoch": 0.38904109589041097, + "grad_norm": 2.3364651203155518, + "learning_rate": 4.863931762794476e-06, + "loss": 1.9941, + "step": 355 + }, + { + "epoch": 0.39013698630136984, + "grad_norm": 2.2886059284210205, + "learning_rate": 4.8619008935824535e-06, + "loss": 2.0163, + "step": 356 + }, + { + "epoch": 0.39123287671232876, + "grad_norm": 2.3796911239624023, + "learning_rate": 4.859870024370431e-06, + "loss": 2.0525, + "step": 357 + }, + { + "epoch": 0.3923287671232877, + "grad_norm": 2.381986379623413, + "learning_rate": 4.857839155158408e-06, + "loss": 2.0682, + "step": 358 + }, + { + "epoch": 0.3934246575342466, + "grad_norm": 2.2048425674438477, + "learning_rate": 4.855808285946385e-06, + "loss": 2.0748, + "step": 359 + }, + { + "epoch": 0.39452054794520547, + "grad_norm": 2.162846088409424, + "learning_rate": 4.8537774167343626e-06, + "loss": 2.0512, + "step": 360 + }, + { + "epoch": 0.39452054794520547, + "eval_loss": 2.1645994186401367, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.3956164383561644, + "grad_norm": 2.4070873260498047, + "learning_rate": 4.85174654752234e-06, + "loss": 2.042, + "step": 361 + }, + { + "epoch": 0.3967123287671233, + "grad_norm": 2.11716628074646, + "learning_rate": 4.849715678310317e-06, + "loss": 2.0398, + "step": 362 + }, + { + "epoch": 0.3978082191780822, + "grad_norm": 2.265479803085327, + "learning_rate": 4.847684809098294e-06, + "loss": 2.0334, + "step": 363 + }, + { + "epoch": 0.3989041095890411, + "grad_norm": 2.2270994186401367, + "learning_rate": 4.845653939886272e-06, + "loss": 2.0292, + "step": 364 + }, + { + "epoch": 0.4, + "grad_norm": 2.406064033508301, + "learning_rate": 4.843623070674249e-06, + "loss": 2.0351, + "step": 365 + }, + { + "epoch": 0.4010958904109589, + "grad_norm": 2.216654062271118, + "learning_rate": 4.841592201462226e-06, + "loss": 2.005, + "step": 366 + }, + { + "epoch": 0.4021917808219178, + "grad_norm": 2.4241385459899902, + "learning_rate": 4.8395613322502035e-06, + "loss": 2.0647, + "step": 367 + }, + { + "epoch": 0.40328767123287673, + "grad_norm": 2.4315145015716553, + "learning_rate": 4.837530463038181e-06, + "loss": 1.9679, + "step": 368 + }, + { + "epoch": 0.4043835616438356, + "grad_norm": 2.3034439086914062, + "learning_rate": 4.835499593826158e-06, + "loss": 2.0259, + "step": 369 + }, + { + "epoch": 0.4054794520547945, + "grad_norm": 2.493436098098755, + "learning_rate": 4.833468724614135e-06, + "loss": 2.0468, + "step": 370 + }, + { + "epoch": 0.4054794520547945, + "eval_loss": 2.161508083343506, + "eval_runtime": 37.2692, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 370 + }, + { + "epoch": 0.40657534246575344, + "grad_norm": 2.3887157440185547, + "learning_rate": 4.831437855402113e-06, + "loss": 2.0313, + "step": 371 + }, + { + "epoch": 0.4076712328767123, + "grad_norm": 2.563095808029175, + "learning_rate": 4.82940698619009e-06, + "loss": 1.9288, + "step": 372 + }, + { + "epoch": 0.40876712328767123, + "grad_norm": 2.2609968185424805, + "learning_rate": 4.827376116978067e-06, + "loss": 1.9975, + "step": 373 + }, + { + "epoch": 0.40986301369863015, + "grad_norm": 2.624257802963257, + "learning_rate": 4.8253452477660444e-06, + "loss": 2.0639, + "step": 374 + }, + { + "epoch": 0.410958904109589, + "grad_norm": 2.502197265625, + "learning_rate": 4.823314378554022e-06, + "loss": 2.0424, + "step": 375 + }, + { + "epoch": 0.41205479452054794, + "grad_norm": 2.36297345161438, + "learning_rate": 4.821283509341998e-06, + "loss": 2.027, + "step": 376 + }, + { + "epoch": 0.41315068493150686, + "grad_norm": 2.388918161392212, + "learning_rate": 4.819252640129976e-06, + "loss": 2.0002, + "step": 377 + }, + { + "epoch": 0.41424657534246573, + "grad_norm": 2.4262335300445557, + "learning_rate": 4.817221770917953e-06, + "loss": 2.0215, + "step": 378 + }, + { + "epoch": 0.41534246575342465, + "grad_norm": 2.8953094482421875, + "learning_rate": 4.815190901705931e-06, + "loss": 1.988, + "step": 379 + }, + { + "epoch": 0.41643835616438357, + "grad_norm": 2.3092124462127686, + "learning_rate": 4.813160032493907e-06, + "loss": 2.0199, + "step": 380 + }, + { + "epoch": 0.41643835616438357, + "eval_loss": 2.1589064598083496, + "eval_runtime": 37.2649, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.41753424657534244, + "grad_norm": 3.0842413902282715, + "learning_rate": 4.811129163281885e-06, + "loss": 2.0134, + "step": 381 + }, + { + "epoch": 0.41863013698630136, + "grad_norm": 2.2015092372894287, + "learning_rate": 4.809098294069862e-06, + "loss": 2.0447, + "step": 382 + }, + { + "epoch": 0.4197260273972603, + "grad_norm": 2.93471097946167, + "learning_rate": 4.80706742485784e-06, + "loss": 2.0329, + "step": 383 + }, + { + "epoch": 0.4208219178082192, + "grad_norm": 2.3277227878570557, + "learning_rate": 4.805036555645816e-06, + "loss": 2.0805, + "step": 384 + }, + { + "epoch": 0.42191780821917807, + "grad_norm": 2.357966184616089, + "learning_rate": 4.8030056864337945e-06, + "loss": 2.0008, + "step": 385 + }, + { + "epoch": 0.423013698630137, + "grad_norm": 2.7505595684051514, + "learning_rate": 4.800974817221771e-06, + "loss": 2.059, + "step": 386 + }, + { + "epoch": 0.4241095890410959, + "grad_norm": 2.144338369369507, + "learning_rate": 4.798943948009749e-06, + "loss": 2.0125, + "step": 387 + }, + { + "epoch": 0.4252054794520548, + "grad_norm": 2.608407735824585, + "learning_rate": 4.7969130787977254e-06, + "loss": 2.0263, + "step": 388 + }, + { + "epoch": 0.4263013698630137, + "grad_norm": 2.4293761253356934, + "learning_rate": 4.794882209585703e-06, + "loss": 2.0107, + "step": 389 + }, + { + "epoch": 0.4273972602739726, + "grad_norm": 2.2136595249176025, + "learning_rate": 4.79285134037368e-06, + "loss": 2.0421, + "step": 390 + }, + { + "epoch": 0.4273972602739726, + "eval_loss": 2.1473641395568848, + "eval_runtime": 37.2895, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 390 + }, + { + "epoch": 0.4284931506849315, + "grad_norm": 2.3706328868865967, + "learning_rate": 4.790820471161657e-06, + "loss": 1.9828, + "step": 391 + }, + { + "epoch": 0.4295890410958904, + "grad_norm": 2.1628777980804443, + "learning_rate": 4.7887896019496345e-06, + "loss": 2.0006, + "step": 392 + }, + { + "epoch": 0.43068493150684933, + "grad_norm": 2.5572149753570557, + "learning_rate": 4.786758732737612e-06, + "loss": 2.0046, + "step": 393 + }, + { + "epoch": 0.4317808219178082, + "grad_norm": 2.44661808013916, + "learning_rate": 4.784727863525589e-06, + "loss": 2.0513, + "step": 394 + }, + { + "epoch": 0.4328767123287671, + "grad_norm": 2.429487466812134, + "learning_rate": 4.782696994313566e-06, + "loss": 1.9619, + "step": 395 + }, + { + "epoch": 0.43397260273972604, + "grad_norm": 2.233353614807129, + "learning_rate": 4.780666125101544e-06, + "loss": 2.0141, + "step": 396 + }, + { + "epoch": 0.4350684931506849, + "grad_norm": 2.137549638748169, + "learning_rate": 4.778635255889521e-06, + "loss": 1.9512, + "step": 397 + }, + { + "epoch": 0.43616438356164383, + "grad_norm": 2.187638759613037, + "learning_rate": 4.776604386677498e-06, + "loss": 1.9802, + "step": 398 + }, + { + "epoch": 0.43726027397260275, + "grad_norm": 2.3084137439727783, + "learning_rate": 4.7745735174654755e-06, + "loss": 1.9887, + "step": 399 + }, + { + "epoch": 0.4383561643835616, + "grad_norm": 2.16654109954834, + "learning_rate": 4.772542648253453e-06, + "loss": 2.0535, + "step": 400 + }, + { + "epoch": 0.4383561643835616, + "eval_loss": 2.1444413661956787, + "eval_runtime": 37.2158, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 400 + }, + { + "epoch": 0.43945205479452054, + "grad_norm": 2.0973072052001953, + "learning_rate": 4.77051177904143e-06, + "loss": 2.0529, + "step": 401 + }, + { + "epoch": 0.44054794520547946, + "grad_norm": 2.421881675720215, + "learning_rate": 4.768480909829407e-06, + "loss": 1.9893, + "step": 402 + }, + { + "epoch": 0.44164383561643833, + "grad_norm": 2.349296808242798, + "learning_rate": 4.766450040617385e-06, + "loss": 2.0237, + "step": 403 + }, + { + "epoch": 0.44273972602739725, + "grad_norm": 2.2154288291931152, + "learning_rate": 4.764419171405362e-06, + "loss": 1.996, + "step": 404 + }, + { + "epoch": 0.4438356164383562, + "grad_norm": 2.6355648040771484, + "learning_rate": 4.762388302193339e-06, + "loss": 2.0345, + "step": 405 + }, + { + "epoch": 0.4449315068493151, + "grad_norm": 2.352238655090332, + "learning_rate": 4.760357432981316e-06, + "loss": 2.0219, + "step": 406 + }, + { + "epoch": 0.44602739726027396, + "grad_norm": 2.1886510848999023, + "learning_rate": 4.758326563769294e-06, + "loss": 1.9923, + "step": 407 + }, + { + "epoch": 0.4471232876712329, + "grad_norm": 2.6483092308044434, + "learning_rate": 4.756295694557271e-06, + "loss": 2.0236, + "step": 408 + }, + { + "epoch": 0.4482191780821918, + "grad_norm": 2.382389545440674, + "learning_rate": 4.754264825345248e-06, + "loss": 2.0249, + "step": 409 + }, + { + "epoch": 0.44931506849315067, + "grad_norm": 2.4824411869049072, + "learning_rate": 4.7522339561332255e-06, + "loss": 1.9571, + "step": 410 + }, + { + "epoch": 0.44931506849315067, + "eval_loss": 2.1502621173858643, + "eval_runtime": 37.1789, + "eval_samples_per_second": 26.897, + "eval_steps_per_second": 1.13, + "step": 410 + }, + { + "epoch": 0.4504109589041096, + "grad_norm": 2.325589656829834, + "learning_rate": 4.750203086921203e-06, + "loss": 1.9753, + "step": 411 + }, + { + "epoch": 0.4515068493150685, + "grad_norm": 2.3618829250335693, + "learning_rate": 4.74817221770918e-06, + "loss": 1.9397, + "step": 412 + }, + { + "epoch": 0.4526027397260274, + "grad_norm": 2.3922839164733887, + "learning_rate": 4.7461413484971565e-06, + "loss": 2.0119, + "step": 413 + }, + { + "epoch": 0.4536986301369863, + "grad_norm": 2.403475046157837, + "learning_rate": 4.744110479285135e-06, + "loss": 2.0664, + "step": 414 + }, + { + "epoch": 0.4547945205479452, + "grad_norm": 2.225860834121704, + "learning_rate": 4.742079610073111e-06, + "loss": 1.9525, + "step": 415 + }, + { + "epoch": 0.4558904109589041, + "grad_norm": 2.344074010848999, + "learning_rate": 4.740048740861089e-06, + "loss": 1.9785, + "step": 416 + }, + { + "epoch": 0.456986301369863, + "grad_norm": 2.255382776260376, + "learning_rate": 4.738017871649066e-06, + "loss": 1.9902, + "step": 417 + }, + { + "epoch": 0.45808219178082193, + "grad_norm": 2.160628318786621, + "learning_rate": 4.735987002437044e-06, + "loss": 2.0075, + "step": 418 + }, + { + "epoch": 0.4591780821917808, + "grad_norm": 2.5667307376861572, + "learning_rate": 4.73395613322502e-06, + "loss": 2.0625, + "step": 419 + }, + { + "epoch": 0.4602739726027397, + "grad_norm": 2.3771040439605713, + "learning_rate": 4.731925264012998e-06, + "loss": 2.0724, + "step": 420 + }, + { + "epoch": 0.4602739726027397, + "eval_loss": 2.138829231262207, + "eval_runtime": 37.4361, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 420 + }, + { + "epoch": 0.46136986301369864, + "grad_norm": 2.336482048034668, + "learning_rate": 4.729894394800975e-06, + "loss": 2.0234, + "step": 421 + }, + { + "epoch": 0.4624657534246575, + "grad_norm": 2.4206933975219727, + "learning_rate": 4.727863525588953e-06, + "loss": 1.9693, + "step": 422 + }, + { + "epoch": 0.46356164383561643, + "grad_norm": 2.4845645427703857, + "learning_rate": 4.725832656376929e-06, + "loss": 1.9239, + "step": 423 + }, + { + "epoch": 0.46465753424657535, + "grad_norm": 2.5199646949768066, + "learning_rate": 4.723801787164907e-06, + "loss": 1.9801, + "step": 424 + }, + { + "epoch": 0.4657534246575342, + "grad_norm": 2.2969748973846436, + "learning_rate": 4.721770917952885e-06, + "loss": 2.0192, + "step": 425 + }, + { + "epoch": 0.46684931506849314, + "grad_norm": 2.423503875732422, + "learning_rate": 4.719740048740861e-06, + "loss": 1.9684, + "step": 426 + }, + { + "epoch": 0.46794520547945206, + "grad_norm": 2.3681061267852783, + "learning_rate": 4.717709179528839e-06, + "loss": 1.9934, + "step": 427 + }, + { + "epoch": 0.469041095890411, + "grad_norm": 2.277965784072876, + "learning_rate": 4.715678310316816e-06, + "loss": 2.0166, + "step": 428 + }, + { + "epoch": 0.47013698630136985, + "grad_norm": 2.2730283737182617, + "learning_rate": 4.713647441104794e-06, + "loss": 2.011, + "step": 429 + }, + { + "epoch": 0.4712328767123288, + "grad_norm": 2.6694860458374023, + "learning_rate": 4.71161657189277e-06, + "loss": 1.9645, + "step": 430 + }, + { + "epoch": 0.4712328767123288, + "eval_loss": 2.1322011947631836, + "eval_runtime": 37.5723, + "eval_samples_per_second": 26.615, + "eval_steps_per_second": 1.118, + "step": 430 + }, + { + "epoch": 0.4723287671232877, + "grad_norm": 2.4100632667541504, + "learning_rate": 4.709585702680748e-06, + "loss": 1.9478, + "step": 431 + }, + { + "epoch": 0.47342465753424656, + "grad_norm": 2.2779541015625, + "learning_rate": 4.707554833468725e-06, + "loss": 2.0274, + "step": 432 + }, + { + "epoch": 0.4745205479452055, + "grad_norm": 2.55690336227417, + "learning_rate": 4.705523964256703e-06, + "loss": 2.0831, + "step": 433 + }, + { + "epoch": 0.4756164383561644, + "grad_norm": 2.455623149871826, + "learning_rate": 4.703493095044679e-06, + "loss": 1.9673, + "step": 434 + }, + { + "epoch": 0.4767123287671233, + "grad_norm": 2.2613699436187744, + "learning_rate": 4.701462225832657e-06, + "loss": 1.992, + "step": 435 + }, + { + "epoch": 0.4778082191780822, + "grad_norm": 2.4467933177948, + "learning_rate": 4.699431356620634e-06, + "loss": 1.9105, + "step": 436 + }, + { + "epoch": 0.4789041095890411, + "grad_norm": 2.4194023609161377, + "learning_rate": 4.697400487408612e-06, + "loss": 2.074, + "step": 437 + }, + { + "epoch": 0.48, + "grad_norm": 2.2165699005126953, + "learning_rate": 4.695369618196588e-06, + "loss": 2.0152, + "step": 438 + }, + { + "epoch": 0.4810958904109589, + "grad_norm": 2.1709225177764893, + "learning_rate": 4.693338748984566e-06, + "loss": 1.9888, + "step": 439 + }, + { + "epoch": 0.4821917808219178, + "grad_norm": 2.2325828075408936, + "learning_rate": 4.691307879772543e-06, + "loss": 1.9838, + "step": 440 + }, + { + "epoch": 0.4821917808219178, + "eval_loss": 2.128732919692993, + "eval_runtime": 37.205, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.4832876712328767, + "grad_norm": 2.446331739425659, + "learning_rate": 4.68927701056052e-06, + "loss": 1.9295, + "step": 441 + }, + { + "epoch": 0.4843835616438356, + "grad_norm": 2.5459415912628174, + "learning_rate": 4.6872461413484975e-06, + "loss": 2.0351, + "step": 442 + }, + { + "epoch": 0.48547945205479454, + "grad_norm": 2.2877118587493896, + "learning_rate": 4.685215272136475e-06, + "loss": 2.0, + "step": 443 + }, + { + "epoch": 0.4865753424657534, + "grad_norm": 2.346625328063965, + "learning_rate": 4.683184402924452e-06, + "loss": 1.9928, + "step": 444 + }, + { + "epoch": 0.4876712328767123, + "grad_norm": 2.268467664718628, + "learning_rate": 4.681153533712429e-06, + "loss": 1.9969, + "step": 445 + }, + { + "epoch": 0.48876712328767125, + "grad_norm": 2.3699793815612793, + "learning_rate": 4.679122664500407e-06, + "loss": 2.0636, + "step": 446 + }, + { + "epoch": 0.4898630136986301, + "grad_norm": 2.2871041297912598, + "learning_rate": 4.677091795288384e-06, + "loss": 1.9535, + "step": 447 + }, + { + "epoch": 0.49095890410958903, + "grad_norm": 2.249393939971924, + "learning_rate": 4.675060926076361e-06, + "loss": 1.9358, + "step": 448 + }, + { + "epoch": 0.49205479452054796, + "grad_norm": 2.299863815307617, + "learning_rate": 4.673030056864338e-06, + "loss": 2.0299, + "step": 449 + }, + { + "epoch": 0.4931506849315068, + "grad_norm": 2.31239652633667, + "learning_rate": 4.670999187652316e-06, + "loss": 1.9372, + "step": 450 + }, + { + "epoch": 0.4931506849315068, + "eval_loss": 2.1267426013946533, + "eval_runtime": 37.1884, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 450 + }, + { + "epoch": 0.49424657534246574, + "grad_norm": 2.2824623584747314, + "learning_rate": 4.668968318440293e-06, + "loss": 1.9435, + "step": 451 + }, + { + "epoch": 0.49534246575342467, + "grad_norm": 2.285339593887329, + "learning_rate": 4.66693744922827e-06, + "loss": 1.9455, + "step": 452 + }, + { + "epoch": 0.4964383561643836, + "grad_norm": 2.346592903137207, + "learning_rate": 4.6649065800162475e-06, + "loss": 2.0212, + "step": 453 + }, + { + "epoch": 0.49753424657534245, + "grad_norm": 2.280669689178467, + "learning_rate": 4.662875710804225e-06, + "loss": 1.9136, + "step": 454 + }, + { + "epoch": 0.4986301369863014, + "grad_norm": 2.178434133529663, + "learning_rate": 4.660844841592202e-06, + "loss": 1.9952, + "step": 455 + }, + { + "epoch": 0.4997260273972603, + "grad_norm": 2.108492851257324, + "learning_rate": 4.658813972380179e-06, + "loss": 1.9022, + "step": 456 + }, + { + "epoch": 0.5008219178082192, + "grad_norm": 2.3862147331237793, + "learning_rate": 4.656783103168157e-06, + "loss": 2.0332, + "step": 457 + }, + { + "epoch": 0.5019178082191781, + "grad_norm": 2.2017860412597656, + "learning_rate": 4.654752233956134e-06, + "loss": 1.8998, + "step": 458 + }, + { + "epoch": 0.503013698630137, + "grad_norm": 2.222134828567505, + "learning_rate": 4.652721364744111e-06, + "loss": 1.9836, + "step": 459 + }, + { + "epoch": 0.5041095890410959, + "grad_norm": 2.2974510192871094, + "learning_rate": 4.6506904955320885e-06, + "loss": 1.9626, + "step": 460 + }, + { + "epoch": 0.5041095890410959, + "eval_loss": 2.1345443725585938, + "eval_runtime": 37.2785, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 460 + }, + { + "epoch": 0.5052054794520547, + "grad_norm": 2.107480764389038, + "learning_rate": 4.648659626320066e-06, + "loss": 2.0061, + "step": 461 + }, + { + "epoch": 0.5063013698630137, + "grad_norm": 2.3541181087493896, + "learning_rate": 4.646628757108043e-06, + "loss": 2.0133, + "step": 462 + }, + { + "epoch": 0.5073972602739726, + "grad_norm": 2.215571880340576, + "learning_rate": 4.6445978878960194e-06, + "loss": 1.9727, + "step": 463 + }, + { + "epoch": 0.5084931506849315, + "grad_norm": 2.2263615131378174, + "learning_rate": 4.6425670186839976e-06, + "loss": 1.9477, + "step": 464 + }, + { + "epoch": 0.5095890410958904, + "grad_norm": 2.178900957107544, + "learning_rate": 4.640536149471974e-06, + "loss": 1.9842, + "step": 465 + }, + { + "epoch": 0.5106849315068493, + "grad_norm": 2.3703105449676514, + "learning_rate": 4.638505280259952e-06, + "loss": 2.0082, + "step": 466 + }, + { + "epoch": 0.5117808219178083, + "grad_norm": 2.1587533950805664, + "learning_rate": 4.6364744110479285e-06, + "loss": 2.0271, + "step": 467 + }, + { + "epoch": 0.5128767123287671, + "grad_norm": 2.1626017093658447, + "learning_rate": 4.634443541835907e-06, + "loss": 1.9858, + "step": 468 + }, + { + "epoch": 0.513972602739726, + "grad_norm": 2.326869487762451, + "learning_rate": 4.632412672623883e-06, + "loss": 2.0413, + "step": 469 + }, + { + "epoch": 0.5150684931506849, + "grad_norm": 2.266502857208252, + "learning_rate": 4.630381803411861e-06, + "loss": 1.9584, + "step": 470 + }, + { + "epoch": 0.5150684931506849, + "eval_loss": 2.1273915767669678, + "eval_runtime": 37.2571, + "eval_samples_per_second": 26.841, + "eval_steps_per_second": 1.127, + "step": 470 + }, + { + "epoch": 0.5161643835616438, + "grad_norm": 2.131348133087158, + "learning_rate": 4.628350934199838e-06, + "loss": 1.863, + "step": 471 + }, + { + "epoch": 0.5172602739726028, + "grad_norm": 2.2635257244110107, + "learning_rate": 4.626320064987816e-06, + "loss": 1.9699, + "step": 472 + }, + { + "epoch": 0.5183561643835617, + "grad_norm": 2.1303744316101074, + "learning_rate": 4.624289195775792e-06, + "loss": 2.0001, + "step": 473 + }, + { + "epoch": 0.5194520547945205, + "grad_norm": 2.1736788749694824, + "learning_rate": 4.62225832656377e-06, + "loss": 1.9684, + "step": 474 + }, + { + "epoch": 0.5205479452054794, + "grad_norm": 2.1996278762817383, + "learning_rate": 4.620227457351747e-06, + "loss": 1.9739, + "step": 475 + }, + { + "epoch": 0.5216438356164383, + "grad_norm": 2.225456953048706, + "learning_rate": 4.618196588139724e-06, + "loss": 1.9649, + "step": 476 + }, + { + "epoch": 0.5227397260273973, + "grad_norm": 2.118537187576294, + "learning_rate": 4.616165718927701e-06, + "loss": 1.9911, + "step": 477 + }, + { + "epoch": 0.5238356164383562, + "grad_norm": 2.232940435409546, + "learning_rate": 4.6141348497156786e-06, + "loss": 1.944, + "step": 478 + }, + { + "epoch": 0.5249315068493151, + "grad_norm": 2.1776034832000732, + "learning_rate": 4.612103980503656e-06, + "loss": 2.0342, + "step": 479 + }, + { + "epoch": 0.5260273972602739, + "grad_norm": 2.042645215988159, + "learning_rate": 4.610073111291633e-06, + "loss": 2.0124, + "step": 480 + }, + { + "epoch": 0.5260273972602739, + "eval_loss": 2.118508815765381, + "eval_runtime": 37.2764, + "eval_samples_per_second": 26.827, + "eval_steps_per_second": 1.127, + "step": 480 + }, + { + "epoch": 0.5271232876712328, + "grad_norm": 2.302661895751953, + "learning_rate": 4.60804224207961e-06, + "loss": 1.9874, + "step": 481 + }, + { + "epoch": 0.5282191780821918, + "grad_norm": 2.1251118183135986, + "learning_rate": 4.606011372867588e-06, + "loss": 1.9594, + "step": 482 + }, + { + "epoch": 0.5293150684931507, + "grad_norm": 2.157043933868408, + "learning_rate": 4.603980503655565e-06, + "loss": 1.9955, + "step": 483 + }, + { + "epoch": 0.5304109589041096, + "grad_norm": 2.291428804397583, + "learning_rate": 4.601949634443542e-06, + "loss": 1.9695, + "step": 484 + }, + { + "epoch": 0.5315068493150685, + "grad_norm": 2.2341537475585938, + "learning_rate": 4.5999187652315195e-06, + "loss": 1.899, + "step": 485 + }, + { + "epoch": 0.5326027397260275, + "grad_norm": 2.1881206035614014, + "learning_rate": 4.597887896019497e-06, + "loss": 1.9066, + "step": 486 + }, + { + "epoch": 0.5336986301369863, + "grad_norm": 2.3057658672332764, + "learning_rate": 4.595857026807474e-06, + "loss": 1.9646, + "step": 487 + }, + { + "epoch": 0.5347945205479452, + "grad_norm": 2.2318356037139893, + "learning_rate": 4.593826157595451e-06, + "loss": 1.9662, + "step": 488 + }, + { + "epoch": 0.5358904109589041, + "grad_norm": 2.170509099960327, + "learning_rate": 4.591795288383429e-06, + "loss": 1.9522, + "step": 489 + }, + { + "epoch": 0.536986301369863, + "grad_norm": 2.101646661758423, + "learning_rate": 4.589764419171406e-06, + "loss": 2.0315, + "step": 490 + }, + { + "epoch": 0.536986301369863, + "eval_loss": 2.113968849182129, + "eval_runtime": 37.2751, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 490 + }, + { + "epoch": 0.538082191780822, + "grad_norm": 2.1563706398010254, + "learning_rate": 4.587733549959383e-06, + "loss": 1.9843, + "step": 491 + }, + { + "epoch": 0.5391780821917809, + "grad_norm": 2.295722246170044, + "learning_rate": 4.5857026807473604e-06, + "loss": 1.9499, + "step": 492 + }, + { + "epoch": 0.5402739726027397, + "grad_norm": 1.974424123764038, + "learning_rate": 4.583671811535338e-06, + "loss": 1.9346, + "step": 493 + }, + { + "epoch": 0.5413698630136986, + "grad_norm": 2.4402668476104736, + "learning_rate": 4.581640942323315e-06, + "loss": 1.9053, + "step": 494 + }, + { + "epoch": 0.5424657534246575, + "grad_norm": 2.269000768661499, + "learning_rate": 4.579610073111292e-06, + "loss": 2.0108, + "step": 495 + }, + { + "epoch": 0.5435616438356164, + "grad_norm": 2.1182594299316406, + "learning_rate": 4.5775792038992695e-06, + "loss": 2.0155, + "step": 496 + }, + { + "epoch": 0.5446575342465754, + "grad_norm": 2.2181780338287354, + "learning_rate": 4.575548334687247e-06, + "loss": 2.0112, + "step": 497 + }, + { + "epoch": 0.5457534246575343, + "grad_norm": 2.2806756496429443, + "learning_rate": 4.573517465475224e-06, + "loss": 2.0418, + "step": 498 + }, + { + "epoch": 0.5468493150684931, + "grad_norm": 2.2690632343292236, + "learning_rate": 4.571486596263201e-06, + "loss": 1.9457, + "step": 499 + }, + { + "epoch": 0.547945205479452, + "grad_norm": 2.361335515975952, + "learning_rate": 4.569455727051178e-06, + "loss": 1.9949, + "step": 500 + }, + { + "epoch": 0.547945205479452, + "eval_loss": 2.1165614128112793, + "eval_runtime": 37.2664, + "eval_samples_per_second": 26.834, + "eval_steps_per_second": 1.127, + "step": 500 + }, + { + "epoch": 0.549041095890411, + "grad_norm": 2.3833138942718506, + "learning_rate": 4.567424857839156e-06, + "loss": 1.9932, + "step": 501 + }, + { + "epoch": 0.5501369863013699, + "grad_norm": 2.328500509262085, + "learning_rate": 4.565393988627132e-06, + "loss": 1.9773, + "step": 502 + }, + { + "epoch": 0.5512328767123288, + "grad_norm": 2.4191694259643555, + "learning_rate": 4.5633631194151105e-06, + "loss": 2.0389, + "step": 503 + }, + { + "epoch": 0.5523287671232877, + "grad_norm": 2.4106884002685547, + "learning_rate": 4.561332250203087e-06, + "loss": 1.965, + "step": 504 + }, + { + "epoch": 0.5534246575342465, + "grad_norm": 2.1452605724334717, + "learning_rate": 4.559301380991065e-06, + "loss": 1.9713, + "step": 505 + }, + { + "epoch": 0.5545205479452054, + "grad_norm": 2.2753357887268066, + "learning_rate": 4.5572705117790414e-06, + "loss": 1.9116, + "step": 506 + }, + { + "epoch": 0.5556164383561644, + "grad_norm": 2.6925032138824463, + "learning_rate": 4.5552396425670196e-06, + "loss": 1.9593, + "step": 507 + }, + { + "epoch": 0.5567123287671233, + "grad_norm": 2.3136658668518066, + "learning_rate": 4.553208773354996e-06, + "loss": 2.0072, + "step": 508 + }, + { + "epoch": 0.5578082191780822, + "grad_norm": 2.254849672317505, + "learning_rate": 4.551177904142974e-06, + "loss": 1.9193, + "step": 509 + }, + { + "epoch": 0.5589041095890411, + "grad_norm": 2.352755546569824, + "learning_rate": 4.5491470349309505e-06, + "loss": 2.0228, + "step": 510 + }, + { + "epoch": 0.5589041095890411, + "eval_loss": 2.114323377609253, + "eval_runtime": 37.3629, + "eval_samples_per_second": 26.764, + "eval_steps_per_second": 1.124, + "step": 510 + }, + { + "epoch": 0.56, + "grad_norm": 2.31339955329895, + "learning_rate": 4.547116165718929e-06, + "loss": 1.9416, + "step": 511 + }, + { + "epoch": 0.5610958904109589, + "grad_norm": 2.191164970397949, + "learning_rate": 4.545085296506905e-06, + "loss": 1.9709, + "step": 512 + }, + { + "epoch": 0.5621917808219178, + "grad_norm": 2.193551778793335, + "learning_rate": 4.543054427294882e-06, + "loss": 1.9475, + "step": 513 + }, + { + "epoch": 0.5632876712328767, + "grad_norm": 2.222710371017456, + "learning_rate": 4.54102355808286e-06, + "loss": 1.9283, + "step": 514 + }, + { + "epoch": 0.5643835616438356, + "grad_norm": 2.078907012939453, + "learning_rate": 4.538992688870837e-06, + "loss": 1.929, + "step": 515 + }, + { + "epoch": 0.5654794520547946, + "grad_norm": 2.2043654918670654, + "learning_rate": 4.536961819658814e-06, + "loss": 1.9591, + "step": 516 + }, + { + "epoch": 0.5665753424657535, + "grad_norm": 2.192481517791748, + "learning_rate": 4.5349309504467915e-06, + "loss": 1.9618, + "step": 517 + }, + { + "epoch": 0.5676712328767123, + "grad_norm": 2.3509292602539062, + "learning_rate": 4.532900081234769e-06, + "loss": 1.9782, + "step": 518 + }, + { + "epoch": 0.5687671232876712, + "grad_norm": 2.2415924072265625, + "learning_rate": 4.530869212022746e-06, + "loss": 1.9456, + "step": 519 + }, + { + "epoch": 0.5698630136986301, + "grad_norm": 2.2135009765625, + "learning_rate": 4.528838342810723e-06, + "loss": 1.9671, + "step": 520 + }, + { + "epoch": 0.5698630136986301, + "eval_loss": 2.109738826751709, + "eval_runtime": 37.5253, + "eval_samples_per_second": 26.649, + "eval_steps_per_second": 1.119, + "step": 520 + }, + { + "epoch": 0.570958904109589, + "grad_norm": 2.3513574600219727, + "learning_rate": 4.526807473598701e-06, + "loss": 1.9625, + "step": 521 + }, + { + "epoch": 0.572054794520548, + "grad_norm": 1.9949424266815186, + "learning_rate": 4.524776604386678e-06, + "loss": 1.9621, + "step": 522 + }, + { + "epoch": 0.5731506849315069, + "grad_norm": 2.199488401412964, + "learning_rate": 4.522745735174655e-06, + "loss": 1.9389, + "step": 523 + }, + { + "epoch": 0.5742465753424657, + "grad_norm": 2.2119452953338623, + "learning_rate": 4.520714865962632e-06, + "loss": 1.9936, + "step": 524 + }, + { + "epoch": 0.5753424657534246, + "grad_norm": 2.242812156677246, + "learning_rate": 4.51868399675061e-06, + "loss": 1.9728, + "step": 525 + }, + { + "epoch": 0.5764383561643835, + "grad_norm": 2.1713850498199463, + "learning_rate": 4.516653127538587e-06, + "loss": 1.98, + "step": 526 + }, + { + "epoch": 0.5775342465753425, + "grad_norm": 2.261742353439331, + "learning_rate": 4.514622258326564e-06, + "loss": 1.9412, + "step": 527 + }, + { + "epoch": 0.5786301369863014, + "grad_norm": 2.222663402557373, + "learning_rate": 4.5125913891145415e-06, + "loss": 1.9266, + "step": 528 + }, + { + "epoch": 0.5797260273972603, + "grad_norm": 2.1108906269073486, + "learning_rate": 4.510560519902519e-06, + "loss": 1.9539, + "step": 529 + }, + { + "epoch": 0.5808219178082191, + "grad_norm": 1.9594205617904663, + "learning_rate": 4.508529650690496e-06, + "loss": 1.9842, + "step": 530 + }, + { + "epoch": 0.5808219178082191, + "eval_loss": 2.1073970794677734, + "eval_runtime": 37.6167, + "eval_samples_per_second": 26.584, + "eval_steps_per_second": 1.117, + "step": 530 + }, + { + "epoch": 0.581917808219178, + "grad_norm": 2.1339495182037354, + "learning_rate": 4.506498781478473e-06, + "loss": 1.9095, + "step": 531 + }, + { + "epoch": 0.583013698630137, + "grad_norm": 2.116248607635498, + "learning_rate": 4.504467912266451e-06, + "loss": 1.9091, + "step": 532 + }, + { + "epoch": 0.5841095890410959, + "grad_norm": 2.063962459564209, + "learning_rate": 4.502437043054428e-06, + "loss": 1.9537, + "step": 533 + }, + { + "epoch": 0.5852054794520548, + "grad_norm": 2.162569522857666, + "learning_rate": 4.500406173842405e-06, + "loss": 2.0253, + "step": 534 + }, + { + "epoch": 0.5863013698630137, + "grad_norm": 2.1783549785614014, + "learning_rate": 4.4983753046303824e-06, + "loss": 1.9714, + "step": 535 + }, + { + "epoch": 0.5873972602739727, + "grad_norm": 2.0791046619415283, + "learning_rate": 4.49634443541836e-06, + "loss": 1.9629, + "step": 536 + }, + { + "epoch": 0.5884931506849315, + "grad_norm": 2.3923251628875732, + "learning_rate": 4.494313566206336e-06, + "loss": 1.9445, + "step": 537 + }, + { + "epoch": 0.5895890410958904, + "grad_norm": 2.1848347187042236, + "learning_rate": 4.492282696994314e-06, + "loss": 2.007, + "step": 538 + }, + { + "epoch": 0.5906849315068493, + "grad_norm": 2.0824925899505615, + "learning_rate": 4.490251827782291e-06, + "loss": 1.9409, + "step": 539 + }, + { + "epoch": 0.5917808219178082, + "grad_norm": 2.057790756225586, + "learning_rate": 4.488220958570269e-06, + "loss": 1.9061, + "step": 540 + }, + { + "epoch": 0.5917808219178082, + "eval_loss": 2.1101415157318115, + "eval_runtime": 37.2235, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5928767123287672, + "grad_norm": 2.3652148246765137, + "learning_rate": 4.486190089358245e-06, + "loss": 1.9379, + "step": 541 + }, + { + "epoch": 0.5939726027397261, + "grad_norm": 2.195535659790039, + "learning_rate": 4.484159220146223e-06, + "loss": 1.8881, + "step": 542 + }, + { + "epoch": 0.5950684931506849, + "grad_norm": 2.258000135421753, + "learning_rate": 4.4821283509342e-06, + "loss": 1.9923, + "step": 543 + }, + { + "epoch": 0.5961643835616438, + "grad_norm": 2.097041606903076, + "learning_rate": 4.480097481722178e-06, + "loss": 1.9715, + "step": 544 + }, + { + "epoch": 0.5972602739726027, + "grad_norm": 2.1741156578063965, + "learning_rate": 4.478066612510154e-06, + "loss": 1.9487, + "step": 545 + }, + { + "epoch": 0.5983561643835617, + "grad_norm": 2.135338306427002, + "learning_rate": 4.4760357432981325e-06, + "loss": 1.9172, + "step": 546 + }, + { + "epoch": 0.5994520547945206, + "grad_norm": 2.3435418605804443, + "learning_rate": 4.474004874086109e-06, + "loss": 1.945, + "step": 547 + }, + { + "epoch": 0.6005479452054795, + "grad_norm": 2.2242488861083984, + "learning_rate": 4.471974004874087e-06, + "loss": 2.0717, + "step": 548 + }, + { + "epoch": 0.6016438356164383, + "grad_norm": 2.3371622562408447, + "learning_rate": 4.4699431356620635e-06, + "loss": 1.9082, + "step": 549 + }, + { + "epoch": 0.6027397260273972, + "grad_norm": 2.176064968109131, + "learning_rate": 4.467912266450041e-06, + "loss": 2.0154, + "step": 550 + }, + { + "epoch": 0.6027397260273972, + "eval_loss": 2.105236291885376, + "eval_runtime": 37.1885, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.6038356164383561, + "grad_norm": 2.131166934967041, + "learning_rate": 4.465881397238018e-06, + "loss": 1.9575, + "step": 551 + }, + { + "epoch": 0.6049315068493151, + "grad_norm": 2.094514846801758, + "learning_rate": 4.463850528025995e-06, + "loss": 1.9544, + "step": 552 + }, + { + "epoch": 0.606027397260274, + "grad_norm": 2.0827765464782715, + "learning_rate": 4.4618196588139726e-06, + "loss": 1.9539, + "step": 553 + }, + { + "epoch": 0.6071232876712329, + "grad_norm": 2.225749969482422, + "learning_rate": 4.45978878960195e-06, + "loss": 1.983, + "step": 554 + }, + { + "epoch": 0.6082191780821918, + "grad_norm": 2.1363680362701416, + "learning_rate": 4.457757920389927e-06, + "loss": 1.9769, + "step": 555 + }, + { + "epoch": 0.6093150684931506, + "grad_norm": 2.238170623779297, + "learning_rate": 4.455727051177904e-06, + "loss": 1.9094, + "step": 556 + }, + { + "epoch": 0.6104109589041096, + "grad_norm": 2.228384494781494, + "learning_rate": 4.453696181965882e-06, + "loss": 1.9464, + "step": 557 + }, + { + "epoch": 0.6115068493150685, + "grad_norm": 2.479428768157959, + "learning_rate": 4.451665312753859e-06, + "loss": 2.0017, + "step": 558 + }, + { + "epoch": 0.6126027397260274, + "grad_norm": 2.2331902980804443, + "learning_rate": 4.449634443541836e-06, + "loss": 1.9728, + "step": 559 + }, + { + "epoch": 0.6136986301369863, + "grad_norm": 2.1728556156158447, + "learning_rate": 4.4476035743298135e-06, + "loss": 2.0153, + "step": 560 + }, + { + "epoch": 0.6136986301369863, + "eval_loss": 2.0978519916534424, + "eval_runtime": 37.233, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.6147945205479453, + "grad_norm": 2.377814292907715, + "learning_rate": 4.445572705117791e-06, + "loss": 1.9242, + "step": 561 + }, + { + "epoch": 0.6158904109589041, + "grad_norm": 2.190436601638794, + "learning_rate": 4.443541835905768e-06, + "loss": 1.9454, + "step": 562 + }, + { + "epoch": 0.616986301369863, + "grad_norm": 2.052201986312866, + "learning_rate": 4.441510966693745e-06, + "loss": 2.0048, + "step": 563 + }, + { + "epoch": 0.6180821917808219, + "grad_norm": 2.2706589698791504, + "learning_rate": 4.439480097481723e-06, + "loss": 1.9426, + "step": 564 + }, + { + "epoch": 0.6191780821917808, + "grad_norm": 2.3478105068206787, + "learning_rate": 4.4374492282697e-06, + "loss": 2.0047, + "step": 565 + }, + { + "epoch": 0.6202739726027398, + "grad_norm": 2.465078115463257, + "learning_rate": 4.435418359057677e-06, + "loss": 1.8739, + "step": 566 + }, + { + "epoch": 0.6213698630136987, + "grad_norm": 2.0979490280151367, + "learning_rate": 4.433387489845654e-06, + "loss": 1.9588, + "step": 567 + }, + { + "epoch": 0.6224657534246575, + "grad_norm": 2.1239588260650635, + "learning_rate": 4.431356620633632e-06, + "loss": 1.9695, + "step": 568 + }, + { + "epoch": 0.6235616438356164, + "grad_norm": 2.014007806777954, + "learning_rate": 4.429325751421609e-06, + "loss": 1.9232, + "step": 569 + }, + { + "epoch": 0.6246575342465753, + "grad_norm": 2.212414026260376, + "learning_rate": 4.427294882209586e-06, + "loss": 1.928, + "step": 570 + }, + { + "epoch": 0.6246575342465753, + "eval_loss": 2.103149890899658, + "eval_runtime": 37.2641, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 570 + }, + { + "epoch": 0.6257534246575343, + "grad_norm": 2.0944411754608154, + "learning_rate": 4.4252640129975635e-06, + "loss": 1.9568, + "step": 571 + }, + { + "epoch": 0.6268493150684932, + "grad_norm": 2.1068055629730225, + "learning_rate": 4.423233143785541e-06, + "loss": 1.8917, + "step": 572 + }, + { + "epoch": 0.6279452054794521, + "grad_norm": 2.1992151737213135, + "learning_rate": 4.421202274573518e-06, + "loss": 1.9536, + "step": 573 + }, + { + "epoch": 0.6290410958904109, + "grad_norm": 2.2079834938049316, + "learning_rate": 4.4191714053614945e-06, + "loss": 1.9603, + "step": 574 + }, + { + "epoch": 0.6301369863013698, + "grad_norm": 2.1230316162109375, + "learning_rate": 4.417140536149473e-06, + "loss": 1.8529, + "step": 575 + }, + { + "epoch": 0.6312328767123287, + "grad_norm": 2.1069579124450684, + "learning_rate": 4.415109666937449e-06, + "loss": 1.8997, + "step": 576 + }, + { + "epoch": 0.6323287671232877, + "grad_norm": 2.1683409214019775, + "learning_rate": 4.413078797725427e-06, + "loss": 1.971, + "step": 577 + }, + { + "epoch": 0.6334246575342466, + "grad_norm": 2.0599024295806885, + "learning_rate": 4.411047928513404e-06, + "loss": 1.9154, + "step": 578 + }, + { + "epoch": 0.6345205479452055, + "grad_norm": 2.1605849266052246, + "learning_rate": 4.409017059301382e-06, + "loss": 1.9562, + "step": 579 + }, + { + "epoch": 0.6356164383561644, + "grad_norm": 2.151249408721924, + "learning_rate": 4.406986190089358e-06, + "loss": 1.9571, + "step": 580 + }, + { + "epoch": 0.6356164383561644, + "eval_loss": 2.098177671432495, + "eval_runtime": 37.2578, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.6367123287671232, + "grad_norm": 2.1819236278533936, + "learning_rate": 4.404955320877336e-06, + "loss": 1.9816, + "step": 581 + }, + { + "epoch": 0.6378082191780822, + "grad_norm": 2.1131205558776855, + "learning_rate": 4.402924451665313e-06, + "loss": 1.9738, + "step": 582 + }, + { + "epoch": 0.6389041095890411, + "grad_norm": 2.0852456092834473, + "learning_rate": 4.400893582453291e-06, + "loss": 1.924, + "step": 583 + }, + { + "epoch": 0.64, + "grad_norm": 2.0997941493988037, + "learning_rate": 4.398862713241267e-06, + "loss": 1.9067, + "step": 584 + }, + { + "epoch": 0.6410958904109589, + "grad_norm": 2.0848538875579834, + "learning_rate": 4.396831844029245e-06, + "loss": 1.9393, + "step": 585 + }, + { + "epoch": 0.6421917808219179, + "grad_norm": 2.2121758460998535, + "learning_rate": 4.394800974817222e-06, + "loss": 1.9458, + "step": 586 + }, + { + "epoch": 0.6432876712328767, + "grad_norm": 2.194462299346924, + "learning_rate": 4.392770105605199e-06, + "loss": 1.9528, + "step": 587 + }, + { + "epoch": 0.6443835616438356, + "grad_norm": 2.1936254501342773, + "learning_rate": 4.390739236393176e-06, + "loss": 1.9584, + "step": 588 + }, + { + "epoch": 0.6454794520547945, + "grad_norm": 2.1685783863067627, + "learning_rate": 4.388708367181154e-06, + "loss": 1.9666, + "step": 589 + }, + { + "epoch": 0.6465753424657534, + "grad_norm": 2.334571123123169, + "learning_rate": 4.386677497969131e-06, + "loss": 1.9748, + "step": 590 + }, + { + "epoch": 0.6465753424657534, + "eval_loss": 2.0937106609344482, + "eval_runtime": 37.2877, + "eval_samples_per_second": 26.818, + "eval_steps_per_second": 1.126, + "step": 590 + }, + { + "epoch": 0.6476712328767124, + "grad_norm": 2.2901110649108887, + "learning_rate": 4.384646628757108e-06, + "loss": 1.9597, + "step": 591 + }, + { + "epoch": 0.6487671232876713, + "grad_norm": 2.2364838123321533, + "learning_rate": 4.3826157595450855e-06, + "loss": 2.004, + "step": 592 + }, + { + "epoch": 0.6498630136986301, + "grad_norm": 2.1088364124298096, + "learning_rate": 4.380584890333063e-06, + "loss": 1.9319, + "step": 593 + }, + { + "epoch": 0.650958904109589, + "grad_norm": 2.164210319519043, + "learning_rate": 4.37855402112104e-06, + "loss": 1.8906, + "step": 594 + }, + { + "epoch": 0.6520547945205479, + "grad_norm": 2.205200672149658, + "learning_rate": 4.376523151909017e-06, + "loss": 1.8474, + "step": 595 + }, + { + "epoch": 0.6531506849315069, + "grad_norm": 2.239530086517334, + "learning_rate": 4.3744922826969946e-06, + "loss": 2.0033, + "step": 596 + }, + { + "epoch": 0.6542465753424658, + "grad_norm": 2.214747190475464, + "learning_rate": 4.372461413484972e-06, + "loss": 1.9258, + "step": 597 + }, + { + "epoch": 0.6553424657534247, + "grad_norm": 2.1746792793273926, + "learning_rate": 4.370430544272949e-06, + "loss": 1.9479, + "step": 598 + }, + { + "epoch": 0.6564383561643835, + "grad_norm": 2.194131851196289, + "learning_rate": 4.368399675060926e-06, + "loss": 1.9825, + "step": 599 + }, + { + "epoch": 0.6575342465753424, + "grad_norm": 2.3334717750549316, + "learning_rate": 4.366368805848904e-06, + "loss": 1.9831, + "step": 600 + }, + { + "epoch": 0.6575342465753424, + "eval_loss": 2.0881216526031494, + "eval_runtime": 37.3226, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 600 + }, + { + "epoch": 0.6586301369863014, + "grad_norm": 2.269484758377075, + "learning_rate": 4.364337936636881e-06, + "loss": 1.9407, + "step": 601 + }, + { + "epoch": 0.6597260273972603, + "grad_norm": 2.470773935317993, + "learning_rate": 4.362307067424858e-06, + "loss": 1.8905, + "step": 602 + }, + { + "epoch": 0.6608219178082192, + "grad_norm": 2.485574960708618, + "learning_rate": 4.3602761982128355e-06, + "loss": 1.953, + "step": 603 + }, + { + "epoch": 0.6619178082191781, + "grad_norm": 2.2443618774414062, + "learning_rate": 4.358245329000813e-06, + "loss": 1.9244, + "step": 604 + }, + { + "epoch": 0.663013698630137, + "grad_norm": 2.039600372314453, + "learning_rate": 4.35621445978879e-06, + "loss": 1.9364, + "step": 605 + }, + { + "epoch": 0.6641095890410958, + "grad_norm": 2.093001127243042, + "learning_rate": 4.354183590576767e-06, + "loss": 1.9439, + "step": 606 + }, + { + "epoch": 0.6652054794520548, + "grad_norm": 2.3433260917663574, + "learning_rate": 4.352152721364745e-06, + "loss": 1.9322, + "step": 607 + }, + { + "epoch": 0.6663013698630137, + "grad_norm": 2.2403812408447266, + "learning_rate": 4.350121852152722e-06, + "loss": 1.8829, + "step": 608 + }, + { + "epoch": 0.6673972602739726, + "grad_norm": 2.1542811393737793, + "learning_rate": 4.348090982940699e-06, + "loss": 1.9549, + "step": 609 + }, + { + "epoch": 0.6684931506849315, + "grad_norm": 2.1526548862457275, + "learning_rate": 4.3460601137286764e-06, + "loss": 1.9433, + "step": 610 + }, + { + "epoch": 0.6684931506849315, + "eval_loss": 2.083638906478882, + "eval_runtime": 37.2677, + "eval_samples_per_second": 26.833, + "eval_steps_per_second": 1.127, + "step": 610 + }, + { + "epoch": 0.6695890410958905, + "grad_norm": 2.093421459197998, + "learning_rate": 4.344029244516654e-06, + "loss": 1.9643, + "step": 611 + }, + { + "epoch": 0.6706849315068493, + "grad_norm": 2.2205088138580322, + "learning_rate": 4.341998375304631e-06, + "loss": 1.9309, + "step": 612 + }, + { + "epoch": 0.6717808219178082, + "grad_norm": 2.059763193130493, + "learning_rate": 4.339967506092607e-06, + "loss": 2.0064, + "step": 613 + }, + { + "epoch": 0.6728767123287671, + "grad_norm": 2.0340707302093506, + "learning_rate": 4.3379366368805855e-06, + "loss": 1.9394, + "step": 614 + }, + { + "epoch": 0.673972602739726, + "grad_norm": 2.0716137886047363, + "learning_rate": 4.335905767668562e-06, + "loss": 1.9421, + "step": 615 + }, + { + "epoch": 0.675068493150685, + "grad_norm": 2.176809549331665, + "learning_rate": 4.33387489845654e-06, + "loss": 1.9261, + "step": 616 + }, + { + "epoch": 0.6761643835616439, + "grad_norm": 2.144385814666748, + "learning_rate": 4.3318440292445165e-06, + "loss": 1.9635, + "step": 617 + }, + { + "epoch": 0.6772602739726027, + "grad_norm": 2.0949580669403076, + "learning_rate": 4.329813160032495e-06, + "loss": 1.9417, + "step": 618 + }, + { + "epoch": 0.6783561643835616, + "grad_norm": 2.1727259159088135, + "learning_rate": 4.327782290820471e-06, + "loss": 1.9041, + "step": 619 + }, + { + "epoch": 0.6794520547945205, + "grad_norm": 2.012868881225586, + "learning_rate": 4.325751421608449e-06, + "loss": 1.8973, + "step": 620 + }, + { + "epoch": 0.6794520547945205, + "eval_loss": 2.0839755535125732, + "eval_runtime": 37.5302, + "eval_samples_per_second": 26.645, + "eval_steps_per_second": 1.119, + "step": 620 + }, + { + "epoch": 0.6805479452054795, + "grad_norm": 2.204153299331665, + "learning_rate": 4.323720552396426e-06, + "loss": 1.9719, + "step": 621 + }, + { + "epoch": 0.6816438356164384, + "grad_norm": 2.2421770095825195, + "learning_rate": 4.321689683184404e-06, + "loss": 1.9207, + "step": 622 + }, + { + "epoch": 0.6827397260273973, + "grad_norm": 2.2263410091400146, + "learning_rate": 4.31965881397238e-06, + "loss": 1.9146, + "step": 623 + }, + { + "epoch": 0.6838356164383562, + "grad_norm": 2.2015221118927, + "learning_rate": 4.3176279447603574e-06, + "loss": 1.9311, + "step": 624 + }, + { + "epoch": 0.684931506849315, + "grad_norm": 2.1459643840789795, + "learning_rate": 4.315597075548335e-06, + "loss": 1.9673, + "step": 625 + }, + { + "epoch": 0.686027397260274, + "grad_norm": 2.278928756713867, + "learning_rate": 4.313566206336312e-06, + "loss": 1.954, + "step": 626 + }, + { + "epoch": 0.6871232876712329, + "grad_norm": 2.05305552482605, + "learning_rate": 4.311535337124289e-06, + "loss": 1.9789, + "step": 627 + }, + { + "epoch": 0.6882191780821918, + "grad_norm": 2.3979289531707764, + "learning_rate": 4.3095044679122665e-06, + "loss": 1.926, + "step": 628 + }, + { + "epoch": 0.6893150684931507, + "grad_norm": 2.1635828018188477, + "learning_rate": 4.307473598700244e-06, + "loss": 1.9316, + "step": 629 + }, + { + "epoch": 0.6904109589041096, + "grad_norm": 2.237713098526001, + "learning_rate": 4.305442729488221e-06, + "loss": 1.9409, + "step": 630 + }, + { + "epoch": 0.6904109589041096, + "eval_loss": 2.084226131439209, + "eval_runtime": 37.6039, + "eval_samples_per_second": 26.593, + "eval_steps_per_second": 1.117, + "step": 630 + }, + { + "epoch": 0.6915068493150685, + "grad_norm": 2.157435894012451, + "learning_rate": 4.303411860276198e-06, + "loss": 1.8777, + "step": 631 + }, + { + "epoch": 0.6926027397260274, + "grad_norm": 2.20766019821167, + "learning_rate": 4.301380991064176e-06, + "loss": 1.9803, + "step": 632 + }, + { + "epoch": 0.6936986301369863, + "grad_norm": 2.173079490661621, + "learning_rate": 4.299350121852153e-06, + "loss": 1.9541, + "step": 633 + }, + { + "epoch": 0.6947945205479452, + "grad_norm": 1.9727106094360352, + "learning_rate": 4.29731925264013e-06, + "loss": 1.9766, + "step": 634 + }, + { + "epoch": 0.6958904109589041, + "grad_norm": 2.268538475036621, + "learning_rate": 4.2952883834281075e-06, + "loss": 1.9737, + "step": 635 + }, + { + "epoch": 0.6969863013698631, + "grad_norm": 2.2118406295776367, + "learning_rate": 4.293257514216085e-06, + "loss": 1.9291, + "step": 636 + }, + { + "epoch": 0.6980821917808219, + "grad_norm": 2.2939369678497314, + "learning_rate": 4.291226645004062e-06, + "loss": 1.8868, + "step": 637 + }, + { + "epoch": 0.6991780821917808, + "grad_norm": 2.4940083026885986, + "learning_rate": 4.289195775792039e-06, + "loss": 1.9694, + "step": 638 + }, + { + "epoch": 0.7002739726027397, + "grad_norm": 2.1988351345062256, + "learning_rate": 4.287164906580017e-06, + "loss": 1.9781, + "step": 639 + }, + { + "epoch": 0.7013698630136986, + "grad_norm": 2.099578619003296, + "learning_rate": 4.285134037367994e-06, + "loss": 1.9145, + "step": 640 + }, + { + "epoch": 0.7013698630136986, + "eval_loss": 2.07913875579834, + "eval_runtime": 37.3458, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 640 + }, + { + "epoch": 0.7024657534246576, + "grad_norm": 2.2437307834625244, + "learning_rate": 4.283103168155971e-06, + "loss": 1.9075, + "step": 641 + }, + { + "epoch": 0.7035616438356165, + "grad_norm": 2.320880174636841, + "learning_rate": 4.281072298943948e-06, + "loss": 1.8929, + "step": 642 + }, + { + "epoch": 0.7046575342465753, + "grad_norm": 2.1174306869506836, + "learning_rate": 4.279041429731926e-06, + "loss": 1.8872, + "step": 643 + }, + { + "epoch": 0.7057534246575342, + "grad_norm": 2.2826361656188965, + "learning_rate": 4.277010560519903e-06, + "loss": 1.9287, + "step": 644 + }, + { + "epoch": 0.7068493150684931, + "grad_norm": 2.7543559074401855, + "learning_rate": 4.27497969130788e-06, + "loss": 1.9119, + "step": 645 + }, + { + "epoch": 0.707945205479452, + "grad_norm": 2.2582201957702637, + "learning_rate": 4.2729488220958575e-06, + "loss": 1.9118, + "step": 646 + }, + { + "epoch": 0.709041095890411, + "grad_norm": 2.3121371269226074, + "learning_rate": 4.270917952883835e-06, + "loss": 1.959, + "step": 647 + }, + { + "epoch": 0.7101369863013699, + "grad_norm": 2.419102430343628, + "learning_rate": 4.268887083671812e-06, + "loss": 1.9679, + "step": 648 + }, + { + "epoch": 0.7112328767123288, + "grad_norm": 2.285771369934082, + "learning_rate": 4.266856214459789e-06, + "loss": 1.8939, + "step": 649 + }, + { + "epoch": 0.7123287671232876, + "grad_norm": 2.193603754043579, + "learning_rate": 4.264825345247766e-06, + "loss": 2.0097, + "step": 650 + }, + { + "epoch": 0.7123287671232876, + "eval_loss": 2.0785164833068848, + "eval_runtime": 37.2595, + "eval_samples_per_second": 26.839, + "eval_steps_per_second": 1.127, + "step": 650 + }, + { + "epoch": 0.7134246575342466, + "grad_norm": 2.2645511627197266, + "learning_rate": 4.262794476035744e-06, + "loss": 1.8865, + "step": 651 + }, + { + "epoch": 0.7145205479452055, + "grad_norm": 2.5553910732269287, + "learning_rate": 4.26076360682372e-06, + "loss": 1.9331, + "step": 652 + }, + { + "epoch": 0.7156164383561644, + "grad_norm": 2.2770845890045166, + "learning_rate": 4.2587327376116984e-06, + "loss": 1.847, + "step": 653 + }, + { + "epoch": 0.7167123287671233, + "grad_norm": 2.1486358642578125, + "learning_rate": 4.256701868399675e-06, + "loss": 1.9058, + "step": 654 + }, + { + "epoch": 0.7178082191780822, + "grad_norm": 2.168975353240967, + "learning_rate": 4.254670999187653e-06, + "loss": 1.9408, + "step": 655 + }, + { + "epoch": 0.718904109589041, + "grad_norm": 2.2156167030334473, + "learning_rate": 4.252640129975629e-06, + "loss": 1.932, + "step": 656 + }, + { + "epoch": 0.72, + "grad_norm": 2.148214101791382, + "learning_rate": 4.2506092607636075e-06, + "loss": 1.9327, + "step": 657 + }, + { + "epoch": 0.7210958904109589, + "grad_norm": 2.3860998153686523, + "learning_rate": 4.248578391551584e-06, + "loss": 1.9191, + "step": 658 + }, + { + "epoch": 0.7221917808219178, + "grad_norm": 2.3917558193206787, + "learning_rate": 4.246547522339562e-06, + "loss": 1.8607, + "step": 659 + }, + { + "epoch": 0.7232876712328767, + "grad_norm": 2.3024795055389404, + "learning_rate": 4.2445166531275385e-06, + "loss": 1.9083, + "step": 660 + }, + { + "epoch": 0.7232876712328767, + "eval_loss": 2.07985782623291, + "eval_runtime": 37.1583, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.7243835616438357, + "grad_norm": 2.261615037918091, + "learning_rate": 4.242485783915517e-06, + "loss": 1.9155, + "step": 661 + }, + { + "epoch": 0.7254794520547945, + "grad_norm": 2.190124034881592, + "learning_rate": 4.240454914703493e-06, + "loss": 1.858, + "step": 662 + }, + { + "epoch": 0.7265753424657534, + "grad_norm": 2.3396036624908447, + "learning_rate": 4.23842404549147e-06, + "loss": 1.878, + "step": 663 + }, + { + "epoch": 0.7276712328767123, + "grad_norm": 2.198410987854004, + "learning_rate": 4.236393176279448e-06, + "loss": 1.8755, + "step": 664 + }, + { + "epoch": 0.7287671232876712, + "grad_norm": 2.108875274658203, + "learning_rate": 4.234362307067425e-06, + "loss": 1.8709, + "step": 665 + }, + { + "epoch": 0.7298630136986302, + "grad_norm": 2.0908193588256836, + "learning_rate": 4.232331437855402e-06, + "loss": 1.9395, + "step": 666 + }, + { + "epoch": 0.7309589041095891, + "grad_norm": 2.208134651184082, + "learning_rate": 4.2303005686433795e-06, + "loss": 1.9675, + "step": 667 + }, + { + "epoch": 0.7320547945205479, + "grad_norm": 2.2779526710510254, + "learning_rate": 4.228269699431357e-06, + "loss": 1.9661, + "step": 668 + }, + { + "epoch": 0.7331506849315068, + "grad_norm": 2.053279161453247, + "learning_rate": 4.226238830219334e-06, + "loss": 1.9244, + "step": 669 + }, + { + "epoch": 0.7342465753424657, + "grad_norm": 2.2060821056365967, + "learning_rate": 4.224207961007311e-06, + "loss": 1.9252, + "step": 670 + }, + { + "epoch": 0.7342465753424657, + "eval_loss": 2.0804648399353027, + "eval_runtime": 37.1852, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 670 + }, + { + "epoch": 0.7353424657534247, + "grad_norm": 2.1996090412139893, + "learning_rate": 4.2221770917952886e-06, + "loss": 1.926, + "step": 671 + }, + { + "epoch": 0.7364383561643836, + "grad_norm": 2.0817034244537354, + "learning_rate": 4.220146222583266e-06, + "loss": 1.9123, + "step": 672 + }, + { + "epoch": 0.7375342465753425, + "grad_norm": 2.2505035400390625, + "learning_rate": 4.218115353371243e-06, + "loss": 1.9369, + "step": 673 + }, + { + "epoch": 0.7386301369863014, + "grad_norm": 2.2053000926971436, + "learning_rate": 4.21608448415922e-06, + "loss": 1.9225, + "step": 674 + }, + { + "epoch": 0.7397260273972602, + "grad_norm": 2.039905071258545, + "learning_rate": 4.214053614947198e-06, + "loss": 1.8876, + "step": 675 + }, + { + "epoch": 0.7408219178082192, + "grad_norm": 2.047445297241211, + "learning_rate": 4.212022745735175e-06, + "loss": 1.9724, + "step": 676 + }, + { + "epoch": 0.7419178082191781, + "grad_norm": 2.277803897857666, + "learning_rate": 4.209991876523152e-06, + "loss": 1.8687, + "step": 677 + }, + { + "epoch": 0.743013698630137, + "grad_norm": 2.0048105716705322, + "learning_rate": 4.2079610073111295e-06, + "loss": 1.9349, + "step": 678 + }, + { + "epoch": 0.7441095890410959, + "grad_norm": 2.121614456176758, + "learning_rate": 4.205930138099107e-06, + "loss": 1.8723, + "step": 679 + }, + { + "epoch": 0.7452054794520548, + "grad_norm": 2.169165849685669, + "learning_rate": 4.203899268887084e-06, + "loss": 1.9713, + "step": 680 + }, + { + "epoch": 0.7452054794520548, + "eval_loss": 2.0704824924468994, + "eval_runtime": 37.2015, + "eval_samples_per_second": 26.881, + "eval_steps_per_second": 1.129, + "step": 680 + }, + { + "epoch": 0.7463013698630137, + "grad_norm": 2.004974603652954, + "learning_rate": 4.201868399675061e-06, + "loss": 1.9237, + "step": 681 + }, + { + "epoch": 0.7473972602739726, + "grad_norm": 2.042402505874634, + "learning_rate": 4.199837530463039e-06, + "loss": 1.9686, + "step": 682 + }, + { + "epoch": 0.7484931506849315, + "grad_norm": 2.152040719985962, + "learning_rate": 4.197806661251016e-06, + "loss": 1.8498, + "step": 683 + }, + { + "epoch": 0.7495890410958904, + "grad_norm": 2.116483211517334, + "learning_rate": 4.195775792038993e-06, + "loss": 1.9241, + "step": 684 + }, + { + "epoch": 0.7506849315068493, + "grad_norm": 2.1628384590148926, + "learning_rate": 4.19374492282697e-06, + "loss": 1.9019, + "step": 685 + }, + { + "epoch": 0.7517808219178083, + "grad_norm": 2.1851415634155273, + "learning_rate": 4.191714053614948e-06, + "loss": 1.836, + "step": 686 + }, + { + "epoch": 0.7528767123287671, + "grad_norm": 2.173215866088867, + "learning_rate": 4.189683184402924e-06, + "loss": 1.9412, + "step": 687 + }, + { + "epoch": 0.753972602739726, + "grad_norm": 2.1002466678619385, + "learning_rate": 4.187652315190902e-06, + "loss": 1.8871, + "step": 688 + }, + { + "epoch": 0.7550684931506849, + "grad_norm": 2.1932122707366943, + "learning_rate": 4.185621445978879e-06, + "loss": 1.9671, + "step": 689 + }, + { + "epoch": 0.7561643835616438, + "grad_norm": 2.2915258407592773, + "learning_rate": 4.183590576766857e-06, + "loss": 1.8957, + "step": 690 + }, + { + "epoch": 0.7561643835616438, + "eval_loss": 2.067124128341675, + "eval_runtime": 37.2362, + "eval_samples_per_second": 26.856, + "eval_steps_per_second": 1.128, + "step": 690 + }, + { + "epoch": 0.7572602739726028, + "grad_norm": 2.060879707336426, + "learning_rate": 4.181559707554833e-06, + "loss": 1.9091, + "step": 691 + }, + { + "epoch": 0.7583561643835617, + "grad_norm": 2.031156539916992, + "learning_rate": 4.179528838342811e-06, + "loss": 1.9606, + "step": 692 + }, + { + "epoch": 0.7594520547945205, + "grad_norm": 2.121628999710083, + "learning_rate": 4.177497969130788e-06, + "loss": 1.88, + "step": 693 + }, + { + "epoch": 0.7605479452054794, + "grad_norm": 2.2009618282318115, + "learning_rate": 4.175467099918766e-06, + "loss": 1.8916, + "step": 694 + }, + { + "epoch": 0.7616438356164383, + "grad_norm": 2.124082326889038, + "learning_rate": 4.173436230706742e-06, + "loss": 1.9805, + "step": 695 + }, + { + "epoch": 0.7627397260273973, + "grad_norm": 2.3914906978607178, + "learning_rate": 4.1714053614947205e-06, + "loss": 1.9052, + "step": 696 + }, + { + "epoch": 0.7638356164383562, + "grad_norm": 2.2398853302001953, + "learning_rate": 4.169374492282697e-06, + "loss": 1.8965, + "step": 697 + }, + { + "epoch": 0.7649315068493151, + "grad_norm": 2.224860191345215, + "learning_rate": 4.167343623070675e-06, + "loss": 1.8962, + "step": 698 + }, + { + "epoch": 0.766027397260274, + "grad_norm": 2.3356244564056396, + "learning_rate": 4.165312753858652e-06, + "loss": 1.9543, + "step": 699 + }, + { + "epoch": 0.7671232876712328, + "grad_norm": 2.1308751106262207, + "learning_rate": 4.163281884646629e-06, + "loss": 1.9095, + "step": 700 + }, + { + "epoch": 0.7671232876712328, + "eval_loss": 2.066145896911621, + "eval_runtime": 37.2629, + "eval_samples_per_second": 26.836, + "eval_steps_per_second": 1.127, + "step": 700 + }, + { + "epoch": 0.7682191780821918, + "grad_norm": 2.1655938625335693, + "learning_rate": 4.161251015434607e-06, + "loss": 1.8923, + "step": 701 + }, + { + "epoch": 0.7693150684931507, + "grad_norm": 2.1985316276550293, + "learning_rate": 4.159220146222583e-06, + "loss": 1.9216, + "step": 702 + }, + { + "epoch": 0.7704109589041096, + "grad_norm": 2.1537251472473145, + "learning_rate": 4.157189277010561e-06, + "loss": 1.879, + "step": 703 + }, + { + "epoch": 0.7715068493150685, + "grad_norm": 2.1867849826812744, + "learning_rate": 4.155158407798538e-06, + "loss": 1.905, + "step": 704 + }, + { + "epoch": 0.7726027397260274, + "grad_norm": 2.342320203781128, + "learning_rate": 4.153127538586516e-06, + "loss": 1.9403, + "step": 705 + }, + { + "epoch": 0.7736986301369863, + "grad_norm": 2.2250113487243652, + "learning_rate": 4.151096669374492e-06, + "loss": 1.9109, + "step": 706 + }, + { + "epoch": 0.7747945205479452, + "grad_norm": 2.179103374481201, + "learning_rate": 4.1490658001624705e-06, + "loss": 1.9288, + "step": 707 + }, + { + "epoch": 0.7758904109589041, + "grad_norm": 2.3618645668029785, + "learning_rate": 4.147034930950447e-06, + "loss": 1.8982, + "step": 708 + }, + { + "epoch": 0.776986301369863, + "grad_norm": 2.150402307510376, + "learning_rate": 4.145004061738425e-06, + "loss": 1.9, + "step": 709 + }, + { + "epoch": 0.7780821917808219, + "grad_norm": 2.295571804046631, + "learning_rate": 4.1429731925264015e-06, + "loss": 1.9003, + "step": 710 + }, + { + "epoch": 0.7780821917808219, + "eval_loss": 2.067948341369629, + "eval_runtime": 37.2379, + "eval_samples_per_second": 26.854, + "eval_steps_per_second": 1.128, + "step": 710 + }, + { + "epoch": 0.7791780821917809, + "grad_norm": 2.3279545307159424, + "learning_rate": 4.140942323314379e-06, + "loss": 1.8307, + "step": 711 + }, + { + "epoch": 0.7802739726027397, + "grad_norm": 2.3993380069732666, + "learning_rate": 4.138911454102356e-06, + "loss": 1.8755, + "step": 712 + }, + { + "epoch": 0.7813698630136986, + "grad_norm": 2.139653444290161, + "learning_rate": 4.136880584890333e-06, + "loss": 1.8914, + "step": 713 + }, + { + "epoch": 0.7824657534246575, + "grad_norm": 2.239544153213501, + "learning_rate": 4.1348497156783106e-06, + "loss": 1.9289, + "step": 714 + }, + { + "epoch": 0.7835616438356164, + "grad_norm": 2.2039642333984375, + "learning_rate": 4.132818846466288e-06, + "loss": 1.9237, + "step": 715 + }, + { + "epoch": 0.7846575342465754, + "grad_norm": 2.1453804969787598, + "learning_rate": 4.130787977254265e-06, + "loss": 1.9291, + "step": 716 + }, + { + "epoch": 0.7857534246575343, + "grad_norm": 2.2350032329559326, + "learning_rate": 4.128757108042242e-06, + "loss": 1.9413, + "step": 717 + }, + { + "epoch": 0.7868493150684932, + "grad_norm": 2.1718318462371826, + "learning_rate": 4.12672623883022e-06, + "loss": 1.9318, + "step": 718 + }, + { + "epoch": 0.787945205479452, + "grad_norm": 2.1860601902008057, + "learning_rate": 4.124695369618197e-06, + "loss": 1.9934, + "step": 719 + }, + { + "epoch": 0.7890410958904109, + "grad_norm": 2.205166816711426, + "learning_rate": 4.122664500406174e-06, + "loss": 1.9038, + "step": 720 + }, + { + "epoch": 0.7890410958904109, + "eval_loss": 2.06229829788208, + "eval_runtime": 37.6121, + "eval_samples_per_second": 26.587, + "eval_steps_per_second": 1.117, + "step": 720 + }, + { + "epoch": 0.7901369863013699, + "grad_norm": 2.2094533443450928, + "learning_rate": 4.1206336311941515e-06, + "loss": 1.906, + "step": 721 + }, + { + "epoch": 0.7912328767123288, + "grad_norm": 2.178891658782959, + "learning_rate": 4.118602761982129e-06, + "loss": 1.936, + "step": 722 + }, + { + "epoch": 0.7923287671232877, + "grad_norm": 2.051077365875244, + "learning_rate": 4.116571892770106e-06, + "loss": 1.8736, + "step": 723 + }, + { + "epoch": 0.7934246575342466, + "grad_norm": 2.2430427074432373, + "learning_rate": 4.114541023558083e-06, + "loss": 1.8396, + "step": 724 + }, + { + "epoch": 0.7945205479452054, + "grad_norm": 2.2255303859710693, + "learning_rate": 4.112510154346061e-06, + "loss": 1.9668, + "step": 725 + }, + { + "epoch": 0.7956164383561644, + "grad_norm": 2.280313491821289, + "learning_rate": 4.110479285134038e-06, + "loss": 1.8945, + "step": 726 + }, + { + "epoch": 0.7967123287671233, + "grad_norm": 2.2516801357269287, + "learning_rate": 4.108448415922015e-06, + "loss": 1.9157, + "step": 727 + }, + { + "epoch": 0.7978082191780822, + "grad_norm": 2.1379809379577637, + "learning_rate": 4.1064175467099924e-06, + "loss": 1.935, + "step": 728 + }, + { + "epoch": 0.7989041095890411, + "grad_norm": 2.149425983428955, + "learning_rate": 4.10438667749797e-06, + "loss": 1.935, + "step": 729 + }, + { + "epoch": 0.8, + "grad_norm": 2.275270700454712, + "learning_rate": 4.102355808285947e-06, + "loss": 1.7889, + "step": 730 + }, + { + "epoch": 0.8, + "eval_loss": 2.0633974075317383, + "eval_runtime": 37.6895, + "eval_samples_per_second": 26.533, + "eval_steps_per_second": 1.114, + "step": 730 + }, + { + "epoch": 0.8010958904109589, + "grad_norm": 2.1550278663635254, + "learning_rate": 4.100324939073924e-06, + "loss": 1.9833, + "step": 731 + }, + { + "epoch": 0.8021917808219178, + "grad_norm": 2.3050060272216797, + "learning_rate": 4.0982940698619015e-06, + "loss": 1.8969, + "step": 732 + }, + { + "epoch": 0.8032876712328767, + "grad_norm": 2.4548778533935547, + "learning_rate": 4.096263200649879e-06, + "loss": 1.8945, + "step": 733 + }, + { + "epoch": 0.8043835616438356, + "grad_norm": 2.2611639499664307, + "learning_rate": 4.094232331437856e-06, + "loss": 1.9071, + "step": 734 + }, + { + "epoch": 0.8054794520547945, + "grad_norm": 2.1806774139404297, + "learning_rate": 4.092201462225833e-06, + "loss": 1.8787, + "step": 735 + }, + { + "epoch": 0.8065753424657535, + "grad_norm": 2.4374301433563232, + "learning_rate": 4.090170593013811e-06, + "loss": 1.88, + "step": 736 + }, + { + "epoch": 0.8076712328767123, + "grad_norm": 2.328333854675293, + "learning_rate": 4.088139723801787e-06, + "loss": 1.9178, + "step": 737 + }, + { + "epoch": 0.8087671232876712, + "grad_norm": 2.33552885055542, + "learning_rate": 4.086108854589765e-06, + "loss": 1.8587, + "step": 738 + }, + { + "epoch": 0.8098630136986301, + "grad_norm": 2.3472790718078613, + "learning_rate": 4.084077985377742e-06, + "loss": 1.877, + "step": 739 + }, + { + "epoch": 0.810958904109589, + "grad_norm": 2.307123899459839, + "learning_rate": 4.08204711616572e-06, + "loss": 1.8888, + "step": 740 + }, + { + "epoch": 0.810958904109589, + "eval_loss": 2.065826892852783, + "eval_runtime": 37.2747, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 740 + }, + { + "epoch": 0.812054794520548, + "grad_norm": 2.3688747882843018, + "learning_rate": 4.080016246953696e-06, + "loss": 1.9571, + "step": 741 + }, + { + "epoch": 0.8131506849315069, + "grad_norm": 2.385007381439209, + "learning_rate": 4.077985377741674e-06, + "loss": 1.8657, + "step": 742 + }, + { + "epoch": 0.8142465753424658, + "grad_norm": 2.156766414642334, + "learning_rate": 4.075954508529651e-06, + "loss": 1.9421, + "step": 743 + }, + { + "epoch": 0.8153424657534246, + "grad_norm": 2.2074286937713623, + "learning_rate": 4.073923639317629e-06, + "loss": 1.9396, + "step": 744 + }, + { + "epoch": 0.8164383561643835, + "grad_norm": 2.185458183288574, + "learning_rate": 4.071892770105605e-06, + "loss": 1.9612, + "step": 745 + }, + { + "epoch": 0.8175342465753425, + "grad_norm": 2.324209451675415, + "learning_rate": 4.069861900893583e-06, + "loss": 1.892, + "step": 746 + }, + { + "epoch": 0.8186301369863014, + "grad_norm": 2.1312127113342285, + "learning_rate": 4.06783103168156e-06, + "loss": 1.8522, + "step": 747 + }, + { + "epoch": 0.8197260273972603, + "grad_norm": 2.0959460735321045, + "learning_rate": 4.065800162469538e-06, + "loss": 1.8732, + "step": 748 + }, + { + "epoch": 0.8208219178082192, + "grad_norm": 2.085519552230835, + "learning_rate": 4.063769293257514e-06, + "loss": 1.8832, + "step": 749 + }, + { + "epoch": 0.821917808219178, + "grad_norm": 2.310145854949951, + "learning_rate": 4.061738424045492e-06, + "loss": 1.919, + "step": 750 + }, + { + "epoch": 0.821917808219178, + "eval_loss": 2.0637967586517334, + "eval_runtime": 37.2246, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 750 + }, + { + "epoch": 0.823013698630137, + "grad_norm": 2.3171160221099854, + "learning_rate": 4.059707554833469e-06, + "loss": 1.8874, + "step": 751 + }, + { + "epoch": 0.8241095890410959, + "grad_norm": 2.1065480709075928, + "learning_rate": 4.057676685621446e-06, + "loss": 1.9064, + "step": 752 + }, + { + "epoch": 0.8252054794520548, + "grad_norm": 2.0539677143096924, + "learning_rate": 4.0556458164094235e-06, + "loss": 1.9376, + "step": 753 + }, + { + "epoch": 0.8263013698630137, + "grad_norm": 2.4440360069274902, + "learning_rate": 4.053614947197401e-06, + "loss": 1.9129, + "step": 754 + }, + { + "epoch": 0.8273972602739726, + "grad_norm": 2.2324604988098145, + "learning_rate": 4.051584077985378e-06, + "loss": 1.8281, + "step": 755 + }, + { + "epoch": 0.8284931506849315, + "grad_norm": 2.262627363204956, + "learning_rate": 4.049553208773355e-06, + "loss": 1.9036, + "step": 756 + }, + { + "epoch": 0.8295890410958904, + "grad_norm": 2.2473418712615967, + "learning_rate": 4.047522339561333e-06, + "loss": 1.9455, + "step": 757 + }, + { + "epoch": 0.8306849315068493, + "grad_norm": 2.473444938659668, + "learning_rate": 4.04549147034931e-06, + "loss": 1.9003, + "step": 758 + }, + { + "epoch": 0.8317808219178082, + "grad_norm": 2.3454110622406006, + "learning_rate": 4.043460601137287e-06, + "loss": 1.8941, + "step": 759 + }, + { + "epoch": 0.8328767123287671, + "grad_norm": 2.4391231536865234, + "learning_rate": 4.041429731925264e-06, + "loss": 2.0146, + "step": 760 + }, + { + "epoch": 0.8328767123287671, + "eval_loss": 2.0618460178375244, + "eval_runtime": 37.1704, + "eval_samples_per_second": 26.903, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.8339726027397261, + "grad_norm": 2.3129498958587646, + "learning_rate": 4.039398862713242e-06, + "loss": 1.9, + "step": 761 + }, + { + "epoch": 0.8350684931506849, + "grad_norm": 2.3766934871673584, + "learning_rate": 4.037367993501219e-06, + "loss": 1.9017, + "step": 762 + }, + { + "epoch": 0.8361643835616438, + "grad_norm": 2.1698460578918457, + "learning_rate": 4.035337124289196e-06, + "loss": 1.9657, + "step": 763 + }, + { + "epoch": 0.8372602739726027, + "grad_norm": 2.384904146194458, + "learning_rate": 4.0333062550771735e-06, + "loss": 1.8867, + "step": 764 + }, + { + "epoch": 0.8383561643835616, + "grad_norm": 2.572391986846924, + "learning_rate": 4.031275385865151e-06, + "loss": 1.8561, + "step": 765 + }, + { + "epoch": 0.8394520547945206, + "grad_norm": 2.216606616973877, + "learning_rate": 4.029244516653128e-06, + "loss": 1.8926, + "step": 766 + }, + { + "epoch": 0.8405479452054795, + "grad_norm": 2.215756893157959, + "learning_rate": 4.027213647441105e-06, + "loss": 1.928, + "step": 767 + }, + { + "epoch": 0.8416438356164384, + "grad_norm": 2.2713284492492676, + "learning_rate": 4.025182778229083e-06, + "loss": 1.8286, + "step": 768 + }, + { + "epoch": 0.8427397260273972, + "grad_norm": 2.341747283935547, + "learning_rate": 4.02315190901706e-06, + "loss": 1.8789, + "step": 769 + }, + { + "epoch": 0.8438356164383561, + "grad_norm": 2.1485848426818848, + "learning_rate": 4.021121039805037e-06, + "loss": 1.8998, + "step": 770 + }, + { + "epoch": 0.8438356164383561, + "eval_loss": 2.0596234798431396, + "eval_runtime": 37.1854, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 770 + }, + { + "epoch": 0.8449315068493151, + "grad_norm": 2.234192132949829, + "learning_rate": 4.0190901705930144e-06, + "loss": 1.8995, + "step": 771 + }, + { + "epoch": 0.846027397260274, + "grad_norm": 2.289328098297119, + "learning_rate": 4.017059301380992e-06, + "loss": 1.9882, + "step": 772 + }, + { + "epoch": 0.8471232876712329, + "grad_norm": 2.1001455783843994, + "learning_rate": 4.015028432168969e-06, + "loss": 1.8706, + "step": 773 + }, + { + "epoch": 0.8482191780821918, + "grad_norm": 2.2200522422790527, + "learning_rate": 4.012997562956945e-06, + "loss": 1.9443, + "step": 774 + }, + { + "epoch": 0.8493150684931506, + "grad_norm": 2.2540805339813232, + "learning_rate": 4.0109666937449235e-06, + "loss": 1.8552, + "step": 775 + }, + { + "epoch": 0.8504109589041096, + "grad_norm": 2.1214053630828857, + "learning_rate": 4.0089358245329e-06, + "loss": 1.8741, + "step": 776 + }, + { + "epoch": 0.8515068493150685, + "grad_norm": 2.0221869945526123, + "learning_rate": 4.006904955320878e-06, + "loss": 1.8798, + "step": 777 + }, + { + "epoch": 0.8526027397260274, + "grad_norm": 2.217261791229248, + "learning_rate": 4.0048740861088545e-06, + "loss": 1.8542, + "step": 778 + }, + { + "epoch": 0.8536986301369863, + "grad_norm": 2.099202871322632, + "learning_rate": 4.002843216896833e-06, + "loss": 1.9342, + "step": 779 + }, + { + "epoch": 0.8547945205479452, + "grad_norm": 2.139526844024658, + "learning_rate": 4.000812347684809e-06, + "loss": 1.9252, + "step": 780 + }, + { + "epoch": 0.8547945205479452, + "eval_loss": 2.054051399230957, + "eval_runtime": 37.1125, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 780 + }, + { + "epoch": 0.8558904109589041, + "grad_norm": 2.2231476306915283, + "learning_rate": 3.998781478472787e-06, + "loss": 1.8671, + "step": 781 + }, + { + "epoch": 0.856986301369863, + "grad_norm": 2.143979072570801, + "learning_rate": 3.996750609260764e-06, + "loss": 1.8759, + "step": 782 + }, + { + "epoch": 0.8580821917808219, + "grad_norm": 2.1353135108947754, + "learning_rate": 3.994719740048742e-06, + "loss": 1.8559, + "step": 783 + }, + { + "epoch": 0.8591780821917808, + "grad_norm": 2.222564697265625, + "learning_rate": 3.992688870836718e-06, + "loss": 1.999, + "step": 784 + }, + { + "epoch": 0.8602739726027397, + "grad_norm": 1.958031177520752, + "learning_rate": 3.990658001624696e-06, + "loss": 1.9019, + "step": 785 + }, + { + "epoch": 0.8613698630136987, + "grad_norm": 2.2048962116241455, + "learning_rate": 3.988627132412673e-06, + "loss": 1.8561, + "step": 786 + }, + { + "epoch": 0.8624657534246576, + "grad_norm": 2.150332450866699, + "learning_rate": 3.98659626320065e-06, + "loss": 1.9399, + "step": 787 + }, + { + "epoch": 0.8635616438356164, + "grad_norm": 2.182742118835449, + "learning_rate": 3.984565393988627e-06, + "loss": 1.8933, + "step": 788 + }, + { + "epoch": 0.8646575342465753, + "grad_norm": 2.19435715675354, + "learning_rate": 3.9825345247766046e-06, + "loss": 1.8918, + "step": 789 + }, + { + "epoch": 0.8657534246575342, + "grad_norm": 2.3382859230041504, + "learning_rate": 3.980503655564582e-06, + "loss": 1.8937, + "step": 790 + }, + { + "epoch": 0.8657534246575342, + "eval_loss": 2.051997423171997, + "eval_runtime": 37.1866, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 790 + }, + { + "epoch": 0.8668493150684932, + "grad_norm": 2.2700726985931396, + "learning_rate": 3.978472786352559e-06, + "loss": 1.9148, + "step": 791 + }, + { + "epoch": 0.8679452054794521, + "grad_norm": 2.2998759746551514, + "learning_rate": 3.976441917140536e-06, + "loss": 1.9268, + "step": 792 + }, + { + "epoch": 0.869041095890411, + "grad_norm": 2.2367377281188965, + "learning_rate": 3.974411047928514e-06, + "loss": 1.901, + "step": 793 + }, + { + "epoch": 0.8701369863013698, + "grad_norm": 2.0868773460388184, + "learning_rate": 3.972380178716491e-06, + "loss": 1.9677, + "step": 794 + }, + { + "epoch": 0.8712328767123287, + "grad_norm": 2.183732509613037, + "learning_rate": 3.970349309504468e-06, + "loss": 1.9271, + "step": 795 + }, + { + "epoch": 0.8723287671232877, + "grad_norm": 2.187282085418701, + "learning_rate": 3.9683184402924455e-06, + "loss": 1.9437, + "step": 796 + }, + { + "epoch": 0.8734246575342466, + "grad_norm": 2.1906070709228516, + "learning_rate": 3.966287571080423e-06, + "loss": 1.8437, + "step": 797 + }, + { + "epoch": 0.8745205479452055, + "grad_norm": 2.16979718208313, + "learning_rate": 3.9642567018684e-06, + "loss": 1.829, + "step": 798 + }, + { + "epoch": 0.8756164383561644, + "grad_norm": 2.2387869358062744, + "learning_rate": 3.962225832656377e-06, + "loss": 1.9143, + "step": 799 + }, + { + "epoch": 0.8767123287671232, + "grad_norm": 2.208221197128296, + "learning_rate": 3.960194963444355e-06, + "loss": 1.8894, + "step": 800 + }, + { + "epoch": 0.8767123287671232, + "eval_loss": 2.047945976257324, + "eval_runtime": 37.1778, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 800 + }, + { + "epoch": 0.8778082191780822, + "grad_norm": 2.335108518600464, + "learning_rate": 3.958164094232332e-06, + "loss": 1.8155, + "step": 801 + }, + { + "epoch": 0.8789041095890411, + "grad_norm": 2.168006181716919, + "learning_rate": 3.956133225020309e-06, + "loss": 1.8919, + "step": 802 + }, + { + "epoch": 0.88, + "grad_norm": 2.3029897212982178, + "learning_rate": 3.954102355808286e-06, + "loss": 1.8996, + "step": 803 + }, + { + "epoch": 0.8810958904109589, + "grad_norm": 2.156141519546509, + "learning_rate": 3.952071486596264e-06, + "loss": 1.8965, + "step": 804 + }, + { + "epoch": 0.8821917808219178, + "grad_norm": 2.1522068977355957, + "learning_rate": 3.950040617384241e-06, + "loss": 1.8922, + "step": 805 + }, + { + "epoch": 0.8832876712328767, + "grad_norm": 2.153430700302124, + "learning_rate": 3.948009748172218e-06, + "loss": 1.8839, + "step": 806 + }, + { + "epoch": 0.8843835616438356, + "grad_norm": 2.1891674995422363, + "learning_rate": 3.9459788789601955e-06, + "loss": 1.8756, + "step": 807 + }, + { + "epoch": 0.8854794520547945, + "grad_norm": 2.1092092990875244, + "learning_rate": 3.943948009748173e-06, + "loss": 1.9452, + "step": 808 + }, + { + "epoch": 0.8865753424657534, + "grad_norm": 2.0465025901794434, + "learning_rate": 3.94191714053615e-06, + "loss": 1.9067, + "step": 809 + }, + { + "epoch": 0.8876712328767123, + "grad_norm": 2.1150474548339844, + "learning_rate": 3.939886271324127e-06, + "loss": 1.8874, + "step": 810 + }, + { + "epoch": 0.8876712328767123, + "eval_loss": 2.046292304992676, + "eval_runtime": 37.2421, + "eval_samples_per_second": 26.851, + "eval_steps_per_second": 1.128, + "step": 810 + }, + { + "epoch": 0.8887671232876713, + "grad_norm": 2.039989948272705, + "learning_rate": 3.937855402112104e-06, + "loss": 1.9212, + "step": 811 + }, + { + "epoch": 0.8898630136986302, + "grad_norm": 2.17354679107666, + "learning_rate": 3.935824532900082e-06, + "loss": 1.8473, + "step": 812 + }, + { + "epoch": 0.890958904109589, + "grad_norm": 2.3101940155029297, + "learning_rate": 3.933793663688058e-06, + "loss": 1.8413, + "step": 813 + }, + { + "epoch": 0.8920547945205479, + "grad_norm": 2.1300928592681885, + "learning_rate": 3.9317627944760365e-06, + "loss": 1.9126, + "step": 814 + }, + { + "epoch": 0.8931506849315068, + "grad_norm": 2.2317357063293457, + "learning_rate": 3.929731925264013e-06, + "loss": 1.938, + "step": 815 + }, + { + "epoch": 0.8942465753424658, + "grad_norm": 2.1097419261932373, + "learning_rate": 3.927701056051991e-06, + "loss": 1.8426, + "step": 816 + }, + { + "epoch": 0.8953424657534247, + "grad_norm": 2.1709141731262207, + "learning_rate": 3.9256701868399674e-06, + "loss": 1.8779, + "step": 817 + }, + { + "epoch": 0.8964383561643836, + "grad_norm": 2.104193925857544, + "learning_rate": 3.9236393176279456e-06, + "loss": 1.8736, + "step": 818 + }, + { + "epoch": 0.8975342465753424, + "grad_norm": 2.094637632369995, + "learning_rate": 3.921608448415922e-06, + "loss": 1.8556, + "step": 819 + }, + { + "epoch": 0.8986301369863013, + "grad_norm": 2.217918872833252, + "learning_rate": 3.9195775792039e-06, + "loss": 1.8429, + "step": 820 + }, + { + "epoch": 0.8986301369863013, + "eval_loss": 2.046095848083496, + "eval_runtime": 37.5659, + "eval_samples_per_second": 26.62, + "eval_steps_per_second": 1.118, + "step": 820 + }, + { + "epoch": 0.8997260273972603, + "grad_norm": 2.220355272293091, + "learning_rate": 3.9175467099918765e-06, + "loss": 1.8973, + "step": 821 + }, + { + "epoch": 0.9008219178082192, + "grad_norm": 2.0766689777374268, + "learning_rate": 3.915515840779855e-06, + "loss": 1.8728, + "step": 822 + }, + { + "epoch": 0.9019178082191781, + "grad_norm": 2.2046821117401123, + "learning_rate": 3.913484971567831e-06, + "loss": 1.9211, + "step": 823 + }, + { + "epoch": 0.903013698630137, + "grad_norm": 2.286024332046509, + "learning_rate": 3.911454102355808e-06, + "loss": 1.913, + "step": 824 + }, + { + "epoch": 0.9041095890410958, + "grad_norm": 2.2495479583740234, + "learning_rate": 3.909423233143786e-06, + "loss": 1.8214, + "step": 825 + }, + { + "epoch": 0.9052054794520548, + "grad_norm": 2.1300203800201416, + "learning_rate": 3.907392363931763e-06, + "loss": 1.9033, + "step": 826 + }, + { + "epoch": 0.9063013698630137, + "grad_norm": 2.092094898223877, + "learning_rate": 3.90536149471974e-06, + "loss": 1.9007, + "step": 827 + }, + { + "epoch": 0.9073972602739726, + "grad_norm": 2.1571428775787354, + "learning_rate": 3.9033306255077175e-06, + "loss": 1.9606, + "step": 828 + }, + { + "epoch": 0.9084931506849315, + "grad_norm": 2.1052281856536865, + "learning_rate": 3.901299756295695e-06, + "loss": 1.9916, + "step": 829 + }, + { + "epoch": 0.9095890410958904, + "grad_norm": 2.113666534423828, + "learning_rate": 3.899268887083672e-06, + "loss": 2.0136, + "step": 830 + }, + { + "epoch": 0.9095890410958904, + "eval_loss": 2.0450072288513184, + "eval_runtime": 37.6746, + "eval_samples_per_second": 26.543, + "eval_steps_per_second": 1.115, + "step": 830 + }, + { + "epoch": 0.9106849315068493, + "grad_norm": 2.112591028213501, + "learning_rate": 3.897238017871649e-06, + "loss": 1.8784, + "step": 831 + }, + { + "epoch": 0.9117808219178082, + "grad_norm": 2.1584672927856445, + "learning_rate": 3.8952071486596266e-06, + "loss": 1.8876, + "step": 832 + }, + { + "epoch": 0.9128767123287671, + "grad_norm": 2.2744317054748535, + "learning_rate": 3.893176279447604e-06, + "loss": 1.8981, + "step": 833 + }, + { + "epoch": 0.913972602739726, + "grad_norm": 2.21824312210083, + "learning_rate": 3.891145410235581e-06, + "loss": 1.85, + "step": 834 + }, + { + "epoch": 0.915068493150685, + "grad_norm": 2.2203729152679443, + "learning_rate": 3.889114541023558e-06, + "loss": 1.9649, + "step": 835 + }, + { + "epoch": 0.9161643835616439, + "grad_norm": 2.1767687797546387, + "learning_rate": 3.887083671811536e-06, + "loss": 1.9044, + "step": 836 + }, + { + "epoch": 0.9172602739726028, + "grad_norm": 2.0724353790283203, + "learning_rate": 3.885052802599513e-06, + "loss": 1.8787, + "step": 837 + }, + { + "epoch": 0.9183561643835616, + "grad_norm": 2.1283740997314453, + "learning_rate": 3.88302193338749e-06, + "loss": 1.8284, + "step": 838 + }, + { + "epoch": 0.9194520547945205, + "grad_norm": 2.5087451934814453, + "learning_rate": 3.8809910641754675e-06, + "loss": 1.9546, + "step": 839 + }, + { + "epoch": 0.9205479452054794, + "grad_norm": 2.0633344650268555, + "learning_rate": 3.878960194963445e-06, + "loss": 1.8969, + "step": 840 + }, + { + "epoch": 0.9205479452054794, + "eval_loss": 2.047938346862793, + "eval_runtime": 37.2293, + "eval_samples_per_second": 26.861, + "eval_steps_per_second": 1.128, + "step": 840 + }, + { + "epoch": 0.9216438356164384, + "grad_norm": 2.178361654281616, + "learning_rate": 3.876929325751422e-06, + "loss": 1.8855, + "step": 841 + }, + { + "epoch": 0.9227397260273973, + "grad_norm": 2.3223721981048584, + "learning_rate": 3.874898456539399e-06, + "loss": 1.9363, + "step": 842 + }, + { + "epoch": 0.9238356164383562, + "grad_norm": 2.3330960273742676, + "learning_rate": 3.872867587327377e-06, + "loss": 1.8374, + "step": 843 + }, + { + "epoch": 0.924931506849315, + "grad_norm": 2.0584356784820557, + "learning_rate": 3.870836718115354e-06, + "loss": 1.911, + "step": 844 + }, + { + "epoch": 0.9260273972602739, + "grad_norm": 2.1834609508514404, + "learning_rate": 3.868805848903331e-06, + "loss": 1.8645, + "step": 845 + }, + { + "epoch": 0.9271232876712329, + "grad_norm": 2.2576122283935547, + "learning_rate": 3.8667749796913084e-06, + "loss": 1.8662, + "step": 846 + }, + { + "epoch": 0.9282191780821918, + "grad_norm": 2.1271183490753174, + "learning_rate": 3.864744110479286e-06, + "loss": 1.8935, + "step": 847 + }, + { + "epoch": 0.9293150684931507, + "grad_norm": 2.226727247238159, + "learning_rate": 3.862713241267262e-06, + "loss": 1.8969, + "step": 848 + }, + { + "epoch": 0.9304109589041096, + "grad_norm": 2.3301939964294434, + "learning_rate": 3.86068237205524e-06, + "loss": 1.9393, + "step": 849 + }, + { + "epoch": 0.9315068493150684, + "grad_norm": 2.180250883102417, + "learning_rate": 3.858651502843217e-06, + "loss": 1.9331, + "step": 850 + }, + { + "epoch": 0.9315068493150684, + "eval_loss": 2.0509681701660156, + "eval_runtime": 37.2111, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 850 + }, + { + "epoch": 0.9326027397260274, + "grad_norm": 2.223548173904419, + "learning_rate": 3.856620633631195e-06, + "loss": 1.894, + "step": 851 + }, + { + "epoch": 0.9336986301369863, + "grad_norm": 2.1904265880584717, + "learning_rate": 3.854589764419171e-06, + "loss": 1.8265, + "step": 852 + }, + { + "epoch": 0.9347945205479452, + "grad_norm": 2.1374316215515137, + "learning_rate": 3.852558895207149e-06, + "loss": 1.8755, + "step": 853 + }, + { + "epoch": 0.9358904109589041, + "grad_norm": 2.2175488471984863, + "learning_rate": 3.850528025995126e-06, + "loss": 1.9649, + "step": 854 + }, + { + "epoch": 0.936986301369863, + "grad_norm": 2.192126989364624, + "learning_rate": 3.848497156783104e-06, + "loss": 1.919, + "step": 855 + }, + { + "epoch": 0.938082191780822, + "grad_norm": 2.219954013824463, + "learning_rate": 3.84646628757108e-06, + "loss": 1.8651, + "step": 856 + }, + { + "epoch": 0.9391780821917808, + "grad_norm": 2.2897071838378906, + "learning_rate": 3.8444354183590585e-06, + "loss": 1.8744, + "step": 857 + }, + { + "epoch": 0.9402739726027397, + "grad_norm": 2.1656205654144287, + "learning_rate": 3.842404549147035e-06, + "loss": 1.8622, + "step": 858 + }, + { + "epoch": 0.9413698630136986, + "grad_norm": 2.20072603225708, + "learning_rate": 3.840373679935013e-06, + "loss": 1.8678, + "step": 859 + }, + { + "epoch": 0.9424657534246575, + "grad_norm": 2.1474406719207764, + "learning_rate": 3.8383428107229894e-06, + "loss": 1.8462, + "step": 860 + }, + { + "epoch": 0.9424657534246575, + "eval_loss": 2.0451602935791016, + "eval_runtime": 37.1383, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.9435616438356165, + "grad_norm": 1.9542036056518555, + "learning_rate": 3.836311941510967e-06, + "loss": 1.8739, + "step": 861 + }, + { + "epoch": 0.9446575342465754, + "grad_norm": 2.3951354026794434, + "learning_rate": 3.834281072298944e-06, + "loss": 1.8565, + "step": 862 + }, + { + "epoch": 0.9457534246575342, + "grad_norm": 2.4916961193084717, + "learning_rate": 3.832250203086921e-06, + "loss": 1.9138, + "step": 863 + }, + { + "epoch": 0.9468493150684931, + "grad_norm": 2.1650452613830566, + "learning_rate": 3.8302193338748985e-06, + "loss": 1.8499, + "step": 864 + }, + { + "epoch": 0.947945205479452, + "grad_norm": 2.2215073108673096, + "learning_rate": 3.828188464662876e-06, + "loss": 1.9376, + "step": 865 + }, + { + "epoch": 0.949041095890411, + "grad_norm": 2.3827078342437744, + "learning_rate": 3.826157595450853e-06, + "loss": 1.9726, + "step": 866 + }, + { + "epoch": 0.9501369863013699, + "grad_norm": 2.37194561958313, + "learning_rate": 3.82412672623883e-06, + "loss": 1.9009, + "step": 867 + }, + { + "epoch": 0.9512328767123288, + "grad_norm": 2.116562604904175, + "learning_rate": 3.822095857026808e-06, + "loss": 1.8342, + "step": 868 + }, + { + "epoch": 0.9523287671232876, + "grad_norm": 2.3410849571228027, + "learning_rate": 3.820064987814785e-06, + "loss": 1.9035, + "step": 869 + }, + { + "epoch": 0.9534246575342465, + "grad_norm": 2.207545280456543, + "learning_rate": 3.818034118602762e-06, + "loss": 1.8539, + "step": 870 + }, + { + "epoch": 0.9534246575342465, + "eval_loss": 2.0433826446533203, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.9545205479452055, + "grad_norm": 2.1716196537017822, + "learning_rate": 3.8160032493907395e-06, + "loss": 1.8557, + "step": 871 + }, + { + "epoch": 0.9556164383561644, + "grad_norm": 2.213124990463257, + "learning_rate": 3.813972380178717e-06, + "loss": 1.8638, + "step": 872 + }, + { + "epoch": 0.9567123287671233, + "grad_norm": 2.3185672760009766, + "learning_rate": 3.811941510966694e-06, + "loss": 1.8645, + "step": 873 + }, + { + "epoch": 0.9578082191780822, + "grad_norm": 2.453956127166748, + "learning_rate": 3.809910641754671e-06, + "loss": 1.9183, + "step": 874 + }, + { + "epoch": 0.958904109589041, + "grad_norm": 2.1971592903137207, + "learning_rate": 3.8078797725426486e-06, + "loss": 1.8445, + "step": 875 + }, + { + "epoch": 0.96, + "grad_norm": 2.273247241973877, + "learning_rate": 3.8058489033306254e-06, + "loss": 1.8024, + "step": 876 + }, + { + "epoch": 0.9610958904109589, + "grad_norm": 2.1687841415405273, + "learning_rate": 3.803818034118603e-06, + "loss": 1.8931, + "step": 877 + }, + { + "epoch": 0.9621917808219178, + "grad_norm": 2.326498508453369, + "learning_rate": 3.80178716490658e-06, + "loss": 1.8587, + "step": 878 + }, + { + "epoch": 0.9632876712328767, + "grad_norm": 2.0317249298095703, + "learning_rate": 3.7997562956945577e-06, + "loss": 1.9394, + "step": 879 + }, + { + "epoch": 0.9643835616438357, + "grad_norm": 2.3171892166137695, + "learning_rate": 3.7977254264825345e-06, + "loss": 1.92, + "step": 880 + }, + { + "epoch": 0.9643835616438357, + "eval_loss": 2.0431556701660156, + "eval_runtime": 37.1174, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 0.9654794520547946, + "grad_norm": 2.2215981483459473, + "learning_rate": 3.7956945572705122e-06, + "loss": 1.8658, + "step": 881 + }, + { + "epoch": 0.9665753424657534, + "grad_norm": 2.2940685749053955, + "learning_rate": 3.793663688058489e-06, + "loss": 1.8528, + "step": 882 + }, + { + "epoch": 0.9676712328767123, + "grad_norm": 2.2218081951141357, + "learning_rate": 3.791632818846467e-06, + "loss": 1.9446, + "step": 883 + }, + { + "epoch": 0.9687671232876712, + "grad_norm": 2.2608420848846436, + "learning_rate": 3.7896019496344436e-06, + "loss": 1.9433, + "step": 884 + }, + { + "epoch": 0.9698630136986301, + "grad_norm": 2.0247788429260254, + "learning_rate": 3.787571080422421e-06, + "loss": 1.9286, + "step": 885 + }, + { + "epoch": 0.9709589041095891, + "grad_norm": 2.164071798324585, + "learning_rate": 3.785540211210398e-06, + "loss": 1.9349, + "step": 886 + }, + { + "epoch": 0.972054794520548, + "grad_norm": 2.089215040206909, + "learning_rate": 3.7835093419983755e-06, + "loss": 1.9077, + "step": 887 + }, + { + "epoch": 0.9731506849315068, + "grad_norm": 2.245256185531616, + "learning_rate": 3.7814784727863527e-06, + "loss": 1.8911, + "step": 888 + }, + { + "epoch": 0.9742465753424657, + "grad_norm": 2.1227362155914307, + "learning_rate": 3.77944760357433e-06, + "loss": 1.8761, + "step": 889 + }, + { + "epoch": 0.9753424657534246, + "grad_norm": 2.1797730922698975, + "learning_rate": 3.7774167343623073e-06, + "loss": 1.9954, + "step": 890 + }, + { + "epoch": 0.9753424657534246, + "eval_loss": 2.034895896911621, + "eval_runtime": 37.1682, + "eval_samples_per_second": 26.905, + "eval_steps_per_second": 1.13, + "step": 890 + }, + { + "epoch": 0.9764383561643836, + "grad_norm": 2.171062707901001, + "learning_rate": 3.7753858651502846e-06, + "loss": 1.9287, + "step": 891 + }, + { + "epoch": 0.9775342465753425, + "grad_norm": 2.1950814723968506, + "learning_rate": 3.773354995938262e-06, + "loss": 1.8587, + "step": 892 + }, + { + "epoch": 0.9786301369863014, + "grad_norm": 2.2582995891571045, + "learning_rate": 3.771324126726239e-06, + "loss": 1.9269, + "step": 893 + }, + { + "epoch": 0.9797260273972602, + "grad_norm": 2.218113660812378, + "learning_rate": 3.7692932575142164e-06, + "loss": 1.9007, + "step": 894 + }, + { + "epoch": 0.9808219178082191, + "grad_norm": 2.187178134918213, + "learning_rate": 3.7672623883021937e-06, + "loss": 1.9326, + "step": 895 + }, + { + "epoch": 0.9819178082191781, + "grad_norm": 2.20451283454895, + "learning_rate": 3.765231519090171e-06, + "loss": 1.8644, + "step": 896 + }, + { + "epoch": 0.983013698630137, + "grad_norm": 2.1399290561676025, + "learning_rate": 3.7632006498781482e-06, + "loss": 1.7875, + "step": 897 + }, + { + "epoch": 0.9841095890410959, + "grad_norm": 2.049473524093628, + "learning_rate": 3.761169780666125e-06, + "loss": 1.8987, + "step": 898 + }, + { + "epoch": 0.9852054794520548, + "grad_norm": 2.154444456100464, + "learning_rate": 3.7591389114541028e-06, + "loss": 1.8475, + "step": 899 + }, + { + "epoch": 0.9863013698630136, + "grad_norm": 2.1840224266052246, + "learning_rate": 3.7571080422420796e-06, + "loss": 1.8966, + "step": 900 + }, + { + "epoch": 0.9863013698630136, + "eval_loss": 2.0326430797576904, + "eval_runtime": 37.1551, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 900 + }, + { + "epoch": 0.9873972602739726, + "grad_norm": 2.0891942977905273, + "learning_rate": 3.7550771730300573e-06, + "loss": 1.9113, + "step": 901 + }, + { + "epoch": 0.9884931506849315, + "grad_norm": 2.2646753787994385, + "learning_rate": 3.753046303818034e-06, + "loss": 1.8624, + "step": 902 + }, + { + "epoch": 0.9895890410958904, + "grad_norm": 2.15173602104187, + "learning_rate": 3.751015434606012e-06, + "loss": 1.933, + "step": 903 + }, + { + "epoch": 0.9906849315068493, + "grad_norm": 2.1751456260681152, + "learning_rate": 3.7489845653939887e-06, + "loss": 1.9114, + "step": 904 + }, + { + "epoch": 0.9917808219178083, + "grad_norm": 2.161860942840576, + "learning_rate": 3.7469536961819664e-06, + "loss": 1.9508, + "step": 905 + }, + { + "epoch": 0.9928767123287672, + "grad_norm": 2.0921733379364014, + "learning_rate": 3.7449228269699433e-06, + "loss": 1.8482, + "step": 906 + }, + { + "epoch": 0.993972602739726, + "grad_norm": 2.0823893547058105, + "learning_rate": 3.742891957757921e-06, + "loss": 1.8744, + "step": 907 + }, + { + "epoch": 0.9950684931506849, + "grad_norm": 2.136939525604248, + "learning_rate": 3.740861088545898e-06, + "loss": 1.9037, + "step": 908 + }, + { + "epoch": 0.9961643835616438, + "grad_norm": 2.0600836277008057, + "learning_rate": 3.7388302193338755e-06, + "loss": 1.8922, + "step": 909 + }, + { + "epoch": 0.9972602739726028, + "grad_norm": 2.235780715942383, + "learning_rate": 3.7367993501218524e-06, + "loss": 1.9222, + "step": 910 + }, + { + "epoch": 0.9972602739726028, + "eval_loss": 2.034031629562378, + "eval_runtime": 37.2911, + "eval_samples_per_second": 26.816, + "eval_steps_per_second": 1.126, + "step": 910 + }, + { + "epoch": 0.9983561643835617, + "grad_norm": 2.124662160873413, + "learning_rate": 3.7347684809098297e-06, + "loss": 1.8584, + "step": 911 + }, + { + "epoch": 0.9994520547945206, + "grad_norm": 2.1237940788269043, + "learning_rate": 3.732737611697807e-06, + "loss": 1.8579, + "step": 912 + }, + { + "epoch": 1.0005479452054795, + "grad_norm": 2.1175124645233154, + "learning_rate": 3.7307067424857842e-06, + "loss": 1.8348, + "step": 913 + }, + { + "epoch": 1.0016438356164383, + "grad_norm": 2.349686861038208, + "learning_rate": 3.7286758732737615e-06, + "loss": 1.7822, + "step": 914 + }, + { + "epoch": 1.0027397260273974, + "grad_norm": 2.3278446197509766, + "learning_rate": 3.7266450040617388e-06, + "loss": 1.8346, + "step": 915 + }, + { + "epoch": 1.0038356164383562, + "grad_norm": 2.1633124351501465, + "learning_rate": 3.724614134849716e-06, + "loss": 1.8137, + "step": 916 + }, + { + "epoch": 1.004931506849315, + "grad_norm": 2.5075809955596924, + "learning_rate": 3.7225832656376933e-06, + "loss": 1.955, + "step": 917 + }, + { + "epoch": 1.006027397260274, + "grad_norm": 2.3370068073272705, + "learning_rate": 3.7205523964256706e-06, + "loss": 1.7985, + "step": 918 + }, + { + "epoch": 1.0071232876712328, + "grad_norm": 2.288229465484619, + "learning_rate": 3.718521527213648e-06, + "loss": 1.923, + "step": 919 + }, + { + "epoch": 1.0082191780821919, + "grad_norm": 2.1989684104919434, + "learning_rate": 3.716490658001625e-06, + "loss": 1.8876, + "step": 920 + }, + { + "epoch": 1.0082191780821919, + "eval_loss": 2.03442120552063, + "eval_runtime": 37.5659, + "eval_samples_per_second": 26.62, + "eval_steps_per_second": 1.118, + "step": 920 + }, + { + "epoch": 1.0093150684931507, + "grad_norm": 2.202122211456299, + "learning_rate": 3.7144597887896024e-06, + "loss": 1.879, + "step": 921 + }, + { + "epoch": 1.0104109589041095, + "grad_norm": 2.325223445892334, + "learning_rate": 3.7124289195775797e-06, + "loss": 1.7632, + "step": 922 + }, + { + "epoch": 1.0115068493150685, + "grad_norm": 2.2964577674865723, + "learning_rate": 3.710398050365557e-06, + "loss": 1.8983, + "step": 923 + }, + { + "epoch": 1.0126027397260273, + "grad_norm": 2.548537015914917, + "learning_rate": 3.708367181153534e-06, + "loss": 1.8666, + "step": 924 + }, + { + "epoch": 1.0136986301369864, + "grad_norm": 2.229745626449585, + "learning_rate": 3.7063363119415115e-06, + "loss": 1.8395, + "step": 925 + }, + { + "epoch": 1.0147945205479452, + "grad_norm": 2.3639235496520996, + "learning_rate": 3.7043054427294884e-06, + "loss": 1.8864, + "step": 926 + }, + { + "epoch": 1.0158904109589042, + "grad_norm": 2.3403055667877197, + "learning_rate": 3.702274573517466e-06, + "loss": 1.8385, + "step": 927 + }, + { + "epoch": 1.016986301369863, + "grad_norm": 2.2838001251220703, + "learning_rate": 3.700243704305443e-06, + "loss": 1.8135, + "step": 928 + }, + { + "epoch": 1.0180821917808218, + "grad_norm": 2.4730923175811768, + "learning_rate": 3.6982128350934206e-06, + "loss": 1.837, + "step": 929 + }, + { + "epoch": 1.0191780821917809, + "grad_norm": 2.5711073875427246, + "learning_rate": 3.6961819658813975e-06, + "loss": 1.8276, + "step": 930 + }, + { + "epoch": 1.0191780821917809, + "eval_loss": 2.0346033573150635, + "eval_runtime": 37.6741, + "eval_samples_per_second": 26.543, + "eval_steps_per_second": 1.115, + "step": 930 + }, + { + "epoch": 1.0202739726027397, + "grad_norm": 2.296823740005493, + "learning_rate": 3.694151096669375e-06, + "loss": 1.8054, + "step": 931 + }, + { + "epoch": 1.0213698630136987, + "grad_norm": 2.321107864379883, + "learning_rate": 3.692120227457352e-06, + "loss": 1.9296, + "step": 932 + }, + { + "epoch": 1.0224657534246575, + "grad_norm": 2.3478338718414307, + "learning_rate": 3.6900893582453297e-06, + "loss": 1.8503, + "step": 933 + }, + { + "epoch": 1.0235616438356165, + "grad_norm": 2.3698856830596924, + "learning_rate": 3.6880584890333066e-06, + "loss": 1.8766, + "step": 934 + }, + { + "epoch": 1.0246575342465754, + "grad_norm": 2.2751212120056152, + "learning_rate": 3.6860276198212834e-06, + "loss": 1.7575, + "step": 935 + }, + { + "epoch": 1.0257534246575342, + "grad_norm": 2.368741035461426, + "learning_rate": 3.683996750609261e-06, + "loss": 1.8555, + "step": 936 + }, + { + "epoch": 1.0268493150684932, + "grad_norm": 2.4507620334625244, + "learning_rate": 3.681965881397238e-06, + "loss": 1.8765, + "step": 937 + }, + { + "epoch": 1.027945205479452, + "grad_norm": 2.1907308101654053, + "learning_rate": 3.6799350121852157e-06, + "loss": 1.8652, + "step": 938 + }, + { + "epoch": 1.029041095890411, + "grad_norm": 2.2696914672851562, + "learning_rate": 3.6779041429731925e-06, + "loss": 1.8854, + "step": 939 + }, + { + "epoch": 1.0301369863013699, + "grad_norm": 2.3696796894073486, + "learning_rate": 3.6758732737611702e-06, + "loss": 1.8303, + "step": 940 + }, + { + "epoch": 1.0301369863013699, + "eval_loss": 2.030118703842163, + "eval_runtime": 37.1755, + "eval_samples_per_second": 26.899, + "eval_steps_per_second": 1.13, + "step": 940 + }, + { + "epoch": 1.0312328767123287, + "grad_norm": 2.440284013748169, + "learning_rate": 3.673842404549147e-06, + "loss": 1.8498, + "step": 941 + }, + { + "epoch": 1.0323287671232877, + "grad_norm": 2.250570297241211, + "learning_rate": 3.671811535337125e-06, + "loss": 1.8361, + "step": 942 + }, + { + "epoch": 1.0334246575342465, + "grad_norm": 2.5550990104675293, + "learning_rate": 3.6697806661251016e-06, + "loss": 1.8464, + "step": 943 + }, + { + "epoch": 1.0345205479452055, + "grad_norm": 2.502474784851074, + "learning_rate": 3.6677497969130793e-06, + "loss": 1.8471, + "step": 944 + }, + { + "epoch": 1.0356164383561643, + "grad_norm": 2.3815231323242188, + "learning_rate": 3.665718927701056e-06, + "loss": 1.9786, + "step": 945 + }, + { + "epoch": 1.0367123287671234, + "grad_norm": 2.3053736686706543, + "learning_rate": 3.663688058489034e-06, + "loss": 1.8625, + "step": 946 + }, + { + "epoch": 1.0378082191780822, + "grad_norm": 2.2623229026794434, + "learning_rate": 3.6616571892770107e-06, + "loss": 1.9206, + "step": 947 + }, + { + "epoch": 1.038904109589041, + "grad_norm": 2.634559392929077, + "learning_rate": 3.659626320064988e-06, + "loss": 1.8475, + "step": 948 + }, + { + "epoch": 1.04, + "grad_norm": 2.264615297317505, + "learning_rate": 3.6575954508529653e-06, + "loss": 1.7857, + "step": 949 + }, + { + "epoch": 1.0410958904109588, + "grad_norm": 2.4125282764434814, + "learning_rate": 3.6555645816409426e-06, + "loss": 1.841, + "step": 950 + }, + { + "epoch": 1.0410958904109588, + "eval_loss": 2.0278165340423584, + "eval_runtime": 37.1503, + "eval_samples_per_second": 26.918, + "eval_steps_per_second": 1.131, + "step": 950 + }, + { + "epoch": 1.0421917808219179, + "grad_norm": 2.365743637084961, + "learning_rate": 3.65353371242892e-06, + "loss": 1.8352, + "step": 951 + }, + { + "epoch": 1.0432876712328767, + "grad_norm": 2.6469883918762207, + "learning_rate": 3.651502843216897e-06, + "loss": 1.8814, + "step": 952 + }, + { + "epoch": 1.0443835616438357, + "grad_norm": 2.3646206855773926, + "learning_rate": 3.6494719740048744e-06, + "loss": 1.7282, + "step": 953 + }, + { + "epoch": 1.0454794520547945, + "grad_norm": 2.250856876373291, + "learning_rate": 3.6474411047928517e-06, + "loss": 1.8647, + "step": 954 + }, + { + "epoch": 1.0465753424657533, + "grad_norm": 2.5404837131500244, + "learning_rate": 3.645410235580829e-06, + "loss": 1.8175, + "step": 955 + }, + { + "epoch": 1.0476712328767124, + "grad_norm": 2.4388837814331055, + "learning_rate": 3.6433793663688062e-06, + "loss": 1.8643, + "step": 956 + }, + { + "epoch": 1.0487671232876712, + "grad_norm": 2.285959005355835, + "learning_rate": 3.6413484971567835e-06, + "loss": 1.8367, + "step": 957 + }, + { + "epoch": 1.0498630136986302, + "grad_norm": 2.1190242767333984, + "learning_rate": 3.6393176279447608e-06, + "loss": 1.8242, + "step": 958 + }, + { + "epoch": 1.050958904109589, + "grad_norm": 2.397578716278076, + "learning_rate": 3.637286758732738e-06, + "loss": 1.8681, + "step": 959 + }, + { + "epoch": 1.0520547945205478, + "grad_norm": 2.510599374771118, + "learning_rate": 3.6352558895207153e-06, + "loss": 1.8804, + "step": 960 + }, + { + "epoch": 1.0520547945205478, + "eval_loss": 2.0267090797424316, + "eval_runtime": 37.1226, + "eval_samples_per_second": 26.938, + "eval_steps_per_second": 1.131, + "step": 960 + }, + { + "epoch": 1.0531506849315069, + "grad_norm": 2.4743072986602783, + "learning_rate": 3.633225020308692e-06, + "loss": 1.8895, + "step": 961 + }, + { + "epoch": 1.0542465753424657, + "grad_norm": 2.3666775226593018, + "learning_rate": 3.63119415109667e-06, + "loss": 1.8836, + "step": 962 + }, + { + "epoch": 1.0553424657534247, + "grad_norm": 2.301119327545166, + "learning_rate": 3.6291632818846467e-06, + "loss": 1.8541, + "step": 963 + }, + { + "epoch": 1.0564383561643835, + "grad_norm": 2.482811212539673, + "learning_rate": 3.6271324126726244e-06, + "loss": 1.88, + "step": 964 + }, + { + "epoch": 1.0575342465753426, + "grad_norm": 2.2962276935577393, + "learning_rate": 3.6251015434606013e-06, + "loss": 1.7834, + "step": 965 + }, + { + "epoch": 1.0586301369863014, + "grad_norm": 2.365130662918091, + "learning_rate": 3.623070674248579e-06, + "loss": 1.7946, + "step": 966 + }, + { + "epoch": 1.0597260273972602, + "grad_norm": 2.4644675254821777, + "learning_rate": 3.621039805036556e-06, + "loss": 1.8379, + "step": 967 + }, + { + "epoch": 1.0608219178082192, + "grad_norm": 2.31742787361145, + "learning_rate": 3.6190089358245335e-06, + "loss": 1.8192, + "step": 968 + }, + { + "epoch": 1.061917808219178, + "grad_norm": 2.1614131927490234, + "learning_rate": 3.6169780666125104e-06, + "loss": 1.8794, + "step": 969 + }, + { + "epoch": 1.063013698630137, + "grad_norm": 2.4367032051086426, + "learning_rate": 3.614947197400488e-06, + "loss": 1.8673, + "step": 970 + }, + { + "epoch": 1.063013698630137, + "eval_loss": 2.0261292457580566, + "eval_runtime": 37.1116, + "eval_samples_per_second": 26.946, + "eval_steps_per_second": 1.132, + "step": 970 + }, + { + "epoch": 1.0641095890410959, + "grad_norm": 2.262789726257324, + "learning_rate": 3.612916328188465e-06, + "loss": 1.8808, + "step": 971 + }, + { + "epoch": 1.0652054794520547, + "grad_norm": 2.316316843032837, + "learning_rate": 3.610885458976442e-06, + "loss": 1.7803, + "step": 972 + }, + { + "epoch": 1.0663013698630137, + "grad_norm": 2.2936418056488037, + "learning_rate": 3.6088545897644195e-06, + "loss": 1.8783, + "step": 973 + }, + { + "epoch": 1.0673972602739725, + "grad_norm": 2.667069673538208, + "learning_rate": 3.6068237205523963e-06, + "loss": 1.8748, + "step": 974 + }, + { + "epoch": 1.0684931506849316, + "grad_norm": 2.31111216545105, + "learning_rate": 3.604792851340374e-06, + "loss": 1.8119, + "step": 975 + }, + { + "epoch": 1.0695890410958904, + "grad_norm": 2.268258571624756, + "learning_rate": 3.602761982128351e-06, + "loss": 1.8055, + "step": 976 + }, + { + "epoch": 1.0706849315068494, + "grad_norm": 2.307137966156006, + "learning_rate": 3.6007311129163286e-06, + "loss": 1.8976, + "step": 977 + }, + { + "epoch": 1.0717808219178082, + "grad_norm": 2.406036853790283, + "learning_rate": 3.5987002437043054e-06, + "loss": 1.8608, + "step": 978 + }, + { + "epoch": 1.072876712328767, + "grad_norm": 2.3303062915802, + "learning_rate": 3.596669374492283e-06, + "loss": 1.8038, + "step": 979 + }, + { + "epoch": 1.073972602739726, + "grad_norm": 2.2280168533325195, + "learning_rate": 3.59463850528026e-06, + "loss": 1.8342, + "step": 980 + }, + { + "epoch": 1.073972602739726, + "eval_loss": 2.0273854732513428, + "eval_runtime": 37.1546, + "eval_samples_per_second": 26.915, + "eval_steps_per_second": 1.13, + "step": 980 + }, + { + "epoch": 1.0750684931506849, + "grad_norm": 2.4218759536743164, + "learning_rate": 3.5926076360682377e-06, + "loss": 1.9104, + "step": 981 + }, + { + "epoch": 1.076164383561644, + "grad_norm": 2.2140634059906006, + "learning_rate": 3.5905767668562145e-06, + "loss": 1.7927, + "step": 982 + }, + { + "epoch": 1.0772602739726027, + "grad_norm": 2.179877996444702, + "learning_rate": 3.5885458976441922e-06, + "loss": 1.8868, + "step": 983 + }, + { + "epoch": 1.0783561643835617, + "grad_norm": 2.1988537311553955, + "learning_rate": 3.586515028432169e-06, + "loss": 1.8516, + "step": 984 + }, + { + "epoch": 1.0794520547945206, + "grad_norm": 2.1724448204040527, + "learning_rate": 3.5844841592201464e-06, + "loss": 1.8869, + "step": 985 + }, + { + "epoch": 1.0805479452054794, + "grad_norm": 2.2685718536376953, + "learning_rate": 3.5824532900081237e-06, + "loss": 1.8955, + "step": 986 + }, + { + "epoch": 1.0816438356164384, + "grad_norm": 2.39284348487854, + "learning_rate": 3.580422420796101e-06, + "loss": 1.7769, + "step": 987 + }, + { + "epoch": 1.0827397260273972, + "grad_norm": 2.3448750972747803, + "learning_rate": 3.578391551584078e-06, + "loss": 1.8417, + "step": 988 + }, + { + "epoch": 1.0838356164383562, + "grad_norm": 2.3518283367156982, + "learning_rate": 3.5763606823720555e-06, + "loss": 1.8524, + "step": 989 + }, + { + "epoch": 1.084931506849315, + "grad_norm": 2.1696958541870117, + "learning_rate": 3.5743298131600328e-06, + "loss": 1.7668, + "step": 990 + }, + { + "epoch": 1.084931506849315, + "eval_loss": 2.0244593620300293, + "eval_runtime": 37.1929, + "eval_samples_per_second": 26.887, + "eval_steps_per_second": 1.129, + "step": 990 + }, + { + "epoch": 1.086027397260274, + "grad_norm": 2.18300724029541, + "learning_rate": 3.57229894394801e-06, + "loss": 1.852, + "step": 991 + }, + { + "epoch": 1.087123287671233, + "grad_norm": 2.2075037956237793, + "learning_rate": 3.5702680747359873e-06, + "loss": 1.8799, + "step": 992 + }, + { + "epoch": 1.0882191780821917, + "grad_norm": 2.2172231674194336, + "learning_rate": 3.5682372055239646e-06, + "loss": 1.8573, + "step": 993 + }, + { + "epoch": 1.0893150684931507, + "grad_norm": 2.25169038772583, + "learning_rate": 3.566206336311942e-06, + "loss": 1.8125, + "step": 994 + }, + { + "epoch": 1.0904109589041096, + "grad_norm": 2.168386697769165, + "learning_rate": 3.564175467099919e-06, + "loss": 1.857, + "step": 995 + }, + { + "epoch": 1.0915068493150686, + "grad_norm": 2.2610065937042236, + "learning_rate": 3.5621445978878964e-06, + "loss": 1.817, + "step": 996 + }, + { + "epoch": 1.0926027397260274, + "grad_norm": 2.2561864852905273, + "learning_rate": 3.5601137286758737e-06, + "loss": 1.7754, + "step": 997 + }, + { + "epoch": 1.0936986301369862, + "grad_norm": 2.143625497817993, + "learning_rate": 3.5580828594638505e-06, + "loss": 1.8734, + "step": 998 + }, + { + "epoch": 1.0947945205479452, + "grad_norm": 2.433831214904785, + "learning_rate": 3.5560519902518282e-06, + "loss": 1.8248, + "step": 999 + }, + { + "epoch": 1.095890410958904, + "grad_norm": 2.237497091293335, + "learning_rate": 3.554021121039805e-06, + "loss": 1.8543, + "step": 1000 + }, + { + "epoch": 1.095890410958904, + "eval_loss": 2.020202159881592, + "eval_runtime": 37.2413, + "eval_samples_per_second": 26.852, + "eval_steps_per_second": 1.128, + "step": 1000 + }, + { + "epoch": 1.096986301369863, + "grad_norm": 2.253833532333374, + "learning_rate": 3.551990251827783e-06, + "loss": 1.8799, + "step": 1001 + }, + { + "epoch": 1.098082191780822, + "grad_norm": 2.205322504043579, + "learning_rate": 3.5499593826157596e-06, + "loss": 1.834, + "step": 1002 + }, + { + "epoch": 1.099178082191781, + "grad_norm": 2.3098037242889404, + "learning_rate": 3.5479285134037373e-06, + "loss": 1.8184, + "step": 1003 + }, + { + "epoch": 1.1002739726027397, + "grad_norm": 2.361888885498047, + "learning_rate": 3.545897644191714e-06, + "loss": 1.827, + "step": 1004 + }, + { + "epoch": 1.1013698630136985, + "grad_norm": 2.341947555541992, + "learning_rate": 3.543866774979692e-06, + "loss": 1.8184, + "step": 1005 + }, + { + "epoch": 1.1024657534246576, + "grad_norm": 2.2927780151367188, + "learning_rate": 3.5418359057676687e-06, + "loss": 1.805, + "step": 1006 + }, + { + "epoch": 1.1035616438356164, + "grad_norm": 2.432988166809082, + "learning_rate": 3.5398050365556464e-06, + "loss": 1.8465, + "step": 1007 + }, + { + "epoch": 1.1046575342465754, + "grad_norm": 2.428126335144043, + "learning_rate": 3.5377741673436233e-06, + "loss": 1.8115, + "step": 1008 + }, + { + "epoch": 1.1057534246575342, + "grad_norm": 2.32551646232605, + "learning_rate": 3.535743298131601e-06, + "loss": 1.8522, + "step": 1009 + }, + { + "epoch": 1.106849315068493, + "grad_norm": 2.297980308532715, + "learning_rate": 3.533712428919578e-06, + "loss": 1.8137, + "step": 1010 + }, + { + "epoch": 1.106849315068493, + "eval_loss": 2.0190494060516357, + "eval_runtime": 37.2149, + "eval_samples_per_second": 26.871, + "eval_steps_per_second": 1.129, + "step": 1010 + }, + { + "epoch": 1.107945205479452, + "grad_norm": 2.364342451095581, + "learning_rate": 3.5316815597075547e-06, + "loss": 1.8677, + "step": 1011 + }, + { + "epoch": 1.1090410958904109, + "grad_norm": 2.355008125305176, + "learning_rate": 3.5296506904955324e-06, + "loss": 1.8937, + "step": 1012 + }, + { + "epoch": 1.11013698630137, + "grad_norm": 2.2288317680358887, + "learning_rate": 3.5276198212835093e-06, + "loss": 1.8188, + "step": 1013 + }, + { + "epoch": 1.1112328767123287, + "grad_norm": 2.2801361083984375, + "learning_rate": 3.525588952071487e-06, + "loss": 1.8386, + "step": 1014 + }, + { + "epoch": 1.1123287671232878, + "grad_norm": 2.2801361083984375, + "learning_rate": 3.525588952071487e-06, + "loss": 1.8845, + "step": 1015 + }, + { + "epoch": 1.1134246575342466, + "grad_norm": 2.4301352500915527, + "learning_rate": 3.523558082859464e-06, + "loss": 1.8898, + "step": 1016 + }, + { + "epoch": 1.1145205479452054, + "grad_norm": 2.3492274284362793, + "learning_rate": 3.5215272136474415e-06, + "loss": 1.8711, + "step": 1017 + }, + { + "epoch": 1.1156164383561644, + "grad_norm": 2.1788229942321777, + "learning_rate": 3.5194963444354184e-06, + "loss": 1.8177, + "step": 1018 + }, + { + "epoch": 1.1167123287671232, + "grad_norm": 2.269767999649048, + "learning_rate": 3.517465475223396e-06, + "loss": 1.8259, + "step": 1019 + }, + { + "epoch": 1.1178082191780823, + "grad_norm": 2.2890219688415527, + "learning_rate": 3.515434606011373e-06, + "loss": 1.7873, + "step": 1020 + }, + { + "epoch": 1.1178082191780823, + "eval_loss": 2.023195266723633, + "eval_runtime": 37.5701, + "eval_samples_per_second": 26.617, + "eval_steps_per_second": 1.118, + "step": 1020 + }, + { + "epoch": 1.118904109589041, + "grad_norm": 2.5528533458709717, + "learning_rate": 3.5134037367993506e-06, + "loss": 1.8624, + "step": 1021 + }, + { + "epoch": 1.12, + "grad_norm": 2.5109457969665527, + "learning_rate": 3.5113728675873275e-06, + "loss": 1.9178, + "step": 1022 + }, + { + "epoch": 1.121095890410959, + "grad_norm": 2.288877010345459, + "learning_rate": 3.5093419983753047e-06, + "loss": 1.8604, + "step": 1023 + }, + { + "epoch": 1.1221917808219177, + "grad_norm": 2.460153818130493, + "learning_rate": 3.507311129163282e-06, + "loss": 1.8551, + "step": 1024 + }, + { + "epoch": 1.1232876712328768, + "grad_norm": 2.3520290851593018, + "learning_rate": 3.5052802599512593e-06, + "loss": 1.7818, + "step": 1025 + }, + { + "epoch": 1.1243835616438356, + "grad_norm": 2.32812237739563, + "learning_rate": 3.5032493907392366e-06, + "loss": 1.8255, + "step": 1026 + }, + { + "epoch": 1.1254794520547946, + "grad_norm": 2.3118042945861816, + "learning_rate": 3.501218521527214e-06, + "loss": 1.8493, + "step": 1027 + }, + { + "epoch": 1.1265753424657534, + "grad_norm": 2.2750277519226074, + "learning_rate": 3.499187652315191e-06, + "loss": 1.7759, + "step": 1028 + }, + { + "epoch": 1.1276712328767124, + "grad_norm": 2.272120475769043, + "learning_rate": 3.4971567831031684e-06, + "loss": 1.8399, + "step": 1029 + }, + { + "epoch": 1.1287671232876713, + "grad_norm": 2.2075254917144775, + "learning_rate": 3.4951259138911457e-06, + "loss": 1.8774, + "step": 1030 + }, + { + "epoch": 1.1287671232876713, + "eval_loss": 2.0199804306030273, + "eval_runtime": 37.6753, + "eval_samples_per_second": 26.543, + "eval_steps_per_second": 1.115, + "step": 1030 + }, + { + "epoch": 1.12986301369863, + "grad_norm": 2.359877586364746, + "learning_rate": 3.493095044679123e-06, + "loss": 1.7942, + "step": 1031 + }, + { + "epoch": 1.130958904109589, + "grad_norm": 2.3057703971862793, + "learning_rate": 3.4910641754671002e-06, + "loss": 1.8996, + "step": 1032 + }, + { + "epoch": 1.132054794520548, + "grad_norm": 2.3109569549560547, + "learning_rate": 3.4890333062550775e-06, + "loss": 1.7336, + "step": 1033 + }, + { + "epoch": 1.133150684931507, + "grad_norm": 2.3021819591522217, + "learning_rate": 3.4870024370430548e-06, + "loss": 1.869, + "step": 1034 + }, + { + "epoch": 1.1342465753424658, + "grad_norm": 2.4741852283477783, + "learning_rate": 3.484971567831032e-06, + "loss": 1.7652, + "step": 1035 + }, + { + "epoch": 1.1353424657534246, + "grad_norm": 2.326500415802002, + "learning_rate": 3.482940698619009e-06, + "loss": 1.809, + "step": 1036 + }, + { + "epoch": 1.1364383561643836, + "grad_norm": 2.2562193870544434, + "learning_rate": 3.4809098294069866e-06, + "loss": 1.8806, + "step": 1037 + }, + { + "epoch": 1.1375342465753424, + "grad_norm": 2.2849345207214355, + "learning_rate": 3.4788789601949634e-06, + "loss": 1.8329, + "step": 1038 + }, + { + "epoch": 1.1386301369863014, + "grad_norm": 2.333707332611084, + "learning_rate": 3.476848090982941e-06, + "loss": 1.8541, + "step": 1039 + }, + { + "epoch": 1.1397260273972603, + "grad_norm": 2.234175682067871, + "learning_rate": 3.474817221770918e-06, + "loss": 1.8126, + "step": 1040 + }, + { + "epoch": 1.1397260273972603, + "eval_loss": 2.0176141262054443, + "eval_runtime": 37.232, + "eval_samples_per_second": 26.859, + "eval_steps_per_second": 1.128, + "step": 1040 + }, + { + "epoch": 1.140821917808219, + "grad_norm": 2.4212377071380615, + "learning_rate": 3.4727863525588957e-06, + "loss": 1.9325, + "step": 1041 + }, + { + "epoch": 1.141917808219178, + "grad_norm": 2.1526010036468506, + "learning_rate": 3.4707554833468725e-06, + "loss": 1.8674, + "step": 1042 + }, + { + "epoch": 1.143013698630137, + "grad_norm": 2.336186408996582, + "learning_rate": 3.4687246141348502e-06, + "loss": 1.8275, + "step": 1043 + }, + { + "epoch": 1.144109589041096, + "grad_norm": 2.2991745471954346, + "learning_rate": 3.466693744922827e-06, + "loss": 1.8057, + "step": 1044 + }, + { + "epoch": 1.1452054794520548, + "grad_norm": 2.3816370964050293, + "learning_rate": 3.464662875710805e-06, + "loss": 1.8212, + "step": 1045 + }, + { + "epoch": 1.1463013698630138, + "grad_norm": 2.2383193969726562, + "learning_rate": 3.4626320064987817e-06, + "loss": 1.8444, + "step": 1046 + }, + { + "epoch": 1.1473972602739726, + "grad_norm": 2.377574920654297, + "learning_rate": 3.4606011372867594e-06, + "loss": 1.8319, + "step": 1047 + }, + { + "epoch": 1.1484931506849314, + "grad_norm": 2.1803700923919678, + "learning_rate": 3.458570268074736e-06, + "loss": 1.8484, + "step": 1048 + }, + { + "epoch": 1.1495890410958904, + "grad_norm": 2.1810641288757324, + "learning_rate": 3.4565393988627135e-06, + "loss": 1.8187, + "step": 1049 + }, + { + "epoch": 1.1506849315068493, + "grad_norm": 2.1565394401550293, + "learning_rate": 3.4545085296506908e-06, + "loss": 1.8117, + "step": 1050 + }, + { + "epoch": 1.1506849315068493, + "eval_loss": 2.0156562328338623, + "eval_runtime": 37.2072, + "eval_samples_per_second": 26.877, + "eval_steps_per_second": 1.129, + "step": 1050 + }, + { + "epoch": 1.1517808219178083, + "grad_norm": 2.1753759384155273, + "learning_rate": 3.452477660438668e-06, + "loss": 1.7822, + "step": 1051 + }, + { + "epoch": 1.152876712328767, + "grad_norm": 2.250089645385742, + "learning_rate": 3.4504467912266453e-06, + "loss": 1.8634, + "step": 1052 + }, + { + "epoch": 1.1539726027397261, + "grad_norm": 2.3043293952941895, + "learning_rate": 3.4484159220146226e-06, + "loss": 1.7743, + "step": 1053 + }, + { + "epoch": 1.155068493150685, + "grad_norm": 2.2193732261657715, + "learning_rate": 3.4463850528026e-06, + "loss": 1.8328, + "step": 1054 + }, + { + "epoch": 1.1561643835616437, + "grad_norm": 2.4204866886138916, + "learning_rate": 3.444354183590577e-06, + "loss": 1.8133, + "step": 1055 + }, + { + "epoch": 1.1572602739726028, + "grad_norm": 2.3088018894195557, + "learning_rate": 3.4423233143785544e-06, + "loss": 1.8523, + "step": 1056 + }, + { + "epoch": 1.1583561643835616, + "grad_norm": 2.38910174369812, + "learning_rate": 3.4402924451665317e-06, + "loss": 1.7846, + "step": 1057 + }, + { + "epoch": 1.1594520547945206, + "grad_norm": 2.2473702430725098, + "learning_rate": 3.438261575954509e-06, + "loss": 1.7632, + "step": 1058 + }, + { + "epoch": 1.1605479452054794, + "grad_norm": 2.3356645107269287, + "learning_rate": 3.4362307067424862e-06, + "loss": 1.8461, + "step": 1059 + }, + { + "epoch": 1.1616438356164385, + "grad_norm": 2.2460737228393555, + "learning_rate": 3.434199837530463e-06, + "loss": 1.8853, + "step": 1060 + }, + { + "epoch": 1.1616438356164385, + "eval_loss": 2.016077995300293, + "eval_runtime": 37.1943, + "eval_samples_per_second": 26.886, + "eval_steps_per_second": 1.129, + "step": 1060 + }, + { + "epoch": 1.1627397260273973, + "grad_norm": 2.28434157371521, + "learning_rate": 3.432168968318441e-06, + "loss": 1.8451, + "step": 1061 + }, + { + "epoch": 1.163835616438356, + "grad_norm": 2.299049139022827, + "learning_rate": 3.4301380991064176e-06, + "loss": 1.7777, + "step": 1062 + }, + { + "epoch": 1.1649315068493151, + "grad_norm": 2.364074468612671, + "learning_rate": 3.4281072298943953e-06, + "loss": 1.8517, + "step": 1063 + }, + { + "epoch": 1.166027397260274, + "grad_norm": 2.2381322383880615, + "learning_rate": 3.426076360682372e-06, + "loss": 1.889, + "step": 1064 + }, + { + "epoch": 1.167123287671233, + "grad_norm": 2.2984025478363037, + "learning_rate": 3.42404549147035e-06, + "loss": 1.8907, + "step": 1065 + }, + { + "epoch": 1.1682191780821918, + "grad_norm": 2.2215473651885986, + "learning_rate": 3.4220146222583267e-06, + "loss": 1.8313, + "step": 1066 + }, + { + "epoch": 1.1693150684931506, + "grad_norm": 2.3414270877838135, + "learning_rate": 3.4199837530463044e-06, + "loss": 1.783, + "step": 1067 + }, + { + "epoch": 1.1704109589041096, + "grad_norm": 2.0636470317840576, + "learning_rate": 3.4179528838342813e-06, + "loss": 1.8185, + "step": 1068 + }, + { + "epoch": 1.1715068493150684, + "grad_norm": 2.2158873081207275, + "learning_rate": 3.415922014622259e-06, + "loss": 1.9319, + "step": 1069 + }, + { + "epoch": 1.1726027397260275, + "grad_norm": 2.244051694869995, + "learning_rate": 3.413891145410236e-06, + "loss": 1.8732, + "step": 1070 + }, + { + "epoch": 1.1726027397260275, + "eval_loss": 2.0180234909057617, + "eval_runtime": 37.1822, + "eval_samples_per_second": 26.895, + "eval_steps_per_second": 1.13, + "step": 1070 + }, + { + "epoch": 1.1736986301369863, + "grad_norm": 2.1608433723449707, + "learning_rate": 3.4118602761982135e-06, + "loss": 1.8709, + "step": 1071 + }, + { + "epoch": 1.174794520547945, + "grad_norm": 2.2612338066101074, + "learning_rate": 3.4098294069861904e-06, + "loss": 1.8882, + "step": 1072 + }, + { + "epoch": 1.1758904109589041, + "grad_norm": 2.3257741928100586, + "learning_rate": 3.4077985377741673e-06, + "loss": 1.8377, + "step": 1073 + }, + { + "epoch": 1.176986301369863, + "grad_norm": 2.2486586570739746, + "learning_rate": 3.405767668562145e-06, + "loss": 1.8105, + "step": 1074 + }, + { + "epoch": 1.178082191780822, + "grad_norm": 2.2840020656585693, + "learning_rate": 3.403736799350122e-06, + "loss": 1.8868, + "step": 1075 + }, + { + "epoch": 1.1791780821917808, + "grad_norm": 2.3264284133911133, + "learning_rate": 3.4017059301380995e-06, + "loss": 1.8701, + "step": 1076 + }, + { + "epoch": 1.1802739726027398, + "grad_norm": 2.311424493789673, + "learning_rate": 3.3996750609260764e-06, + "loss": 1.801, + "step": 1077 + }, + { + "epoch": 1.1813698630136986, + "grad_norm": 2.2177181243896484, + "learning_rate": 3.397644191714054e-06, + "loss": 1.9188, + "step": 1078 + }, + { + "epoch": 1.1824657534246574, + "grad_norm": 2.413118362426758, + "learning_rate": 3.395613322502031e-06, + "loss": 1.8198, + "step": 1079 + }, + { + "epoch": 1.1835616438356165, + "grad_norm": 2.334688186645508, + "learning_rate": 3.3935824532900086e-06, + "loss": 1.822, + "step": 1080 + }, + { + "epoch": 1.1835616438356165, + "eval_loss": 2.012676954269409, + "eval_runtime": 37.2415, + "eval_samples_per_second": 26.852, + "eval_steps_per_second": 1.128, + "step": 1080 + }, + { + "epoch": 1.1846575342465753, + "grad_norm": 2.3901584148406982, + "learning_rate": 3.3915515840779855e-06, + "loss": 1.791, + "step": 1081 + }, + { + "epoch": 1.1857534246575343, + "grad_norm": 2.2731425762176514, + "learning_rate": 3.389520714865963e-06, + "loss": 1.8367, + "step": 1082 + }, + { + "epoch": 1.1868493150684931, + "grad_norm": 2.311553478240967, + "learning_rate": 3.38748984565394e-06, + "loss": 1.8302, + "step": 1083 + }, + { + "epoch": 1.1879452054794521, + "grad_norm": 2.3030776977539062, + "learning_rate": 3.3854589764419177e-06, + "loss": 1.8652, + "step": 1084 + }, + { + "epoch": 1.189041095890411, + "grad_norm": 2.341188430786133, + "learning_rate": 3.3834281072298946e-06, + "loss": 1.8094, + "step": 1085 + }, + { + "epoch": 1.1901369863013698, + "grad_norm": 2.331962823867798, + "learning_rate": 3.381397238017872e-06, + "loss": 1.827, + "step": 1086 + }, + { + "epoch": 1.1912328767123288, + "grad_norm": 2.275923728942871, + "learning_rate": 3.379366368805849e-06, + "loss": 1.7935, + "step": 1087 + }, + { + "epoch": 1.1923287671232876, + "grad_norm": 2.4608163833618164, + "learning_rate": 3.3773354995938264e-06, + "loss": 1.8965, + "step": 1088 + }, + { + "epoch": 1.1934246575342466, + "grad_norm": 2.2208592891693115, + "learning_rate": 3.3753046303818037e-06, + "loss": 1.8447, + "step": 1089 + }, + { + "epoch": 1.1945205479452055, + "grad_norm": 2.4284770488739014, + "learning_rate": 3.373273761169781e-06, + "loss": 1.8306, + "step": 1090 + }, + { + "epoch": 1.1945205479452055, + "eval_loss": 2.010664939880371, + "eval_runtime": 37.237, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 1090 + }, + { + "epoch": 1.1956164383561645, + "grad_norm": 2.329601764678955, + "learning_rate": 3.3712428919577582e-06, + "loss": 1.8275, + "step": 1091 + }, + { + "epoch": 1.1967123287671233, + "grad_norm": 2.277939796447754, + "learning_rate": 3.3692120227457355e-06, + "loss": 1.8161, + "step": 1092 + }, + { + "epoch": 1.1978082191780821, + "grad_norm": 2.446780204772949, + "learning_rate": 3.3671811535337128e-06, + "loss": 1.8511, + "step": 1093 + }, + { + "epoch": 1.1989041095890411, + "grad_norm": 2.382477283477783, + "learning_rate": 3.36515028432169e-06, + "loss": 1.7834, + "step": 1094 + }, + { + "epoch": 1.2, + "grad_norm": 2.366361379623413, + "learning_rate": 3.3631194151096673e-06, + "loss": 1.8105, + "step": 1095 + }, + { + "epoch": 1.201095890410959, + "grad_norm": 2.3439717292785645, + "learning_rate": 3.3610885458976446e-06, + "loss": 1.8418, + "step": 1096 + }, + { + "epoch": 1.2021917808219178, + "grad_norm": 2.2739830017089844, + "learning_rate": 3.359057676685622e-06, + "loss": 1.8818, + "step": 1097 + }, + { + "epoch": 1.2032876712328768, + "grad_norm": 2.1636738777160645, + "learning_rate": 3.357026807473599e-06, + "loss": 1.8757, + "step": 1098 + }, + { + "epoch": 1.2043835616438356, + "grad_norm": 2.386441946029663, + "learning_rate": 3.354995938261576e-06, + "loss": 1.756, + "step": 1099 + }, + { + "epoch": 1.2054794520547945, + "grad_norm": 2.2241456508636475, + "learning_rate": 3.3529650690495537e-06, + "loss": 1.8034, + "step": 1100 + }, + { + "epoch": 1.2054794520547945, + "eval_loss": 2.008504867553711, + "eval_runtime": 37.2855, + "eval_samples_per_second": 26.82, + "eval_steps_per_second": 1.126, + "step": 1100 + }, + { + "epoch": 1.2065753424657535, + "grad_norm": 2.196992874145508, + "learning_rate": 3.3509341998375305e-06, + "loss": 1.7722, + "step": 1101 + }, + { + "epoch": 1.2076712328767123, + "grad_norm": 2.3572094440460205, + "learning_rate": 3.3489033306255082e-06, + "loss": 1.8109, + "step": 1102 + }, + { + "epoch": 1.208767123287671, + "grad_norm": 2.443281650543213, + "learning_rate": 3.346872461413485e-06, + "loss": 1.843, + "step": 1103 + }, + { + "epoch": 1.2098630136986301, + "grad_norm": 2.2613234519958496, + "learning_rate": 3.344841592201463e-06, + "loss": 1.8023, + "step": 1104 + }, + { + "epoch": 1.210958904109589, + "grad_norm": 2.2466726303100586, + "learning_rate": 3.3428107229894397e-06, + "loss": 1.7838, + "step": 1105 + }, + { + "epoch": 1.212054794520548, + "grad_norm": 2.2314462661743164, + "learning_rate": 3.3407798537774174e-06, + "loss": 1.8077, + "step": 1106 + }, + { + "epoch": 1.2131506849315068, + "grad_norm": 2.3288400173187256, + "learning_rate": 3.338748984565394e-06, + "loss": 1.8319, + "step": 1107 + }, + { + "epoch": 1.2142465753424658, + "grad_norm": 2.3477249145507812, + "learning_rate": 3.336718115353372e-06, + "loss": 1.8772, + "step": 1108 + }, + { + "epoch": 1.2153424657534246, + "grad_norm": 2.3588409423828125, + "learning_rate": 3.3346872461413488e-06, + "loss": 1.8832, + "step": 1109 + }, + { + "epoch": 1.2164383561643834, + "grad_norm": 2.327796220779419, + "learning_rate": 3.3326563769293256e-06, + "loss": 1.8033, + "step": 1110 + }, + { + "epoch": 1.2164383561643834, + "eval_loss": 2.0094382762908936, + "eval_runtime": 37.2666, + "eval_samples_per_second": 26.834, + "eval_steps_per_second": 1.127, + "step": 1110 + }, + { + "epoch": 1.2175342465753425, + "grad_norm": 2.257244825363159, + "learning_rate": 3.3306255077173033e-06, + "loss": 1.7903, + "step": 1111 + }, + { + "epoch": 1.2186301369863013, + "grad_norm": 2.3321614265441895, + "learning_rate": 3.32859463850528e-06, + "loss": 1.8295, + "step": 1112 + }, + { + "epoch": 1.2197260273972603, + "grad_norm": 2.4073803424835205, + "learning_rate": 3.326563769293258e-06, + "loss": 1.8395, + "step": 1113 + }, + { + "epoch": 1.2208219178082191, + "grad_norm": 2.3631434440612793, + "learning_rate": 3.3245329000812347e-06, + "loss": 1.7548, + "step": 1114 + }, + { + "epoch": 1.2219178082191782, + "grad_norm": 2.275766134262085, + "learning_rate": 3.3225020308692124e-06, + "loss": 1.808, + "step": 1115 + }, + { + "epoch": 1.223013698630137, + "grad_norm": 2.438011407852173, + "learning_rate": 3.3204711616571893e-06, + "loss": 1.8274, + "step": 1116 + }, + { + "epoch": 1.2241095890410958, + "grad_norm": 2.371826171875, + "learning_rate": 3.318440292445167e-06, + "loss": 1.7742, + "step": 1117 + }, + { + "epoch": 1.2252054794520548, + "grad_norm": 2.422004461288452, + "learning_rate": 3.316409423233144e-06, + "loss": 1.8268, + "step": 1118 + }, + { + "epoch": 1.2263013698630136, + "grad_norm": 2.377979278564453, + "learning_rate": 3.3143785540211215e-06, + "loss": 1.7857, + "step": 1119 + }, + { + "epoch": 1.2273972602739727, + "grad_norm": 2.288482904434204, + "learning_rate": 3.3123476848090984e-06, + "loss": 1.876, + "step": 1120 + }, + { + "epoch": 1.2273972602739727, + "eval_loss": 2.0104730129241943, + "eval_runtime": 37.5365, + "eval_samples_per_second": 26.641, + "eval_steps_per_second": 1.119, + "step": 1120 + }, + { + "epoch": 1.2284931506849315, + "grad_norm": 2.451533555984497, + "learning_rate": 3.310316815597076e-06, + "loss": 1.8324, + "step": 1121 + }, + { + "epoch": 1.2295890410958905, + "grad_norm": 2.3785181045532227, + "learning_rate": 3.308285946385053e-06, + "loss": 1.8239, + "step": 1122 + }, + { + "epoch": 1.2306849315068493, + "grad_norm": 2.2276721000671387, + "learning_rate": 3.30625507717303e-06, + "loss": 1.9132, + "step": 1123 + }, + { + "epoch": 1.2317808219178081, + "grad_norm": 2.2575314044952393, + "learning_rate": 3.3042242079610075e-06, + "loss": 1.769, + "step": 1124 + }, + { + "epoch": 1.2328767123287672, + "grad_norm": 2.5311717987060547, + "learning_rate": 3.3021933387489847e-06, + "loss": 1.8605, + "step": 1125 + }, + { + "epoch": 1.233972602739726, + "grad_norm": 2.2395896911621094, + "learning_rate": 3.300162469536962e-06, + "loss": 1.8725, + "step": 1126 + }, + { + "epoch": 1.235068493150685, + "grad_norm": 2.2402031421661377, + "learning_rate": 3.2981316003249393e-06, + "loss": 1.7955, + "step": 1127 + }, + { + "epoch": 1.2361643835616438, + "grad_norm": 2.308950424194336, + "learning_rate": 3.2961007311129166e-06, + "loss": 1.8812, + "step": 1128 + }, + { + "epoch": 1.2372602739726029, + "grad_norm": 2.437767267227173, + "learning_rate": 3.294069861900894e-06, + "loss": 1.8489, + "step": 1129 + }, + { + "epoch": 1.2383561643835617, + "grad_norm": 2.3188693523406982, + "learning_rate": 3.292038992688871e-06, + "loss": 1.9082, + "step": 1130 + }, + { + "epoch": 1.2383561643835617, + "eval_loss": 2.0110676288604736, + "eval_runtime": 37.6437, + "eval_samples_per_second": 26.565, + "eval_steps_per_second": 1.116, + "step": 1130 + }, + { + "epoch": 1.2394520547945205, + "grad_norm": 2.312760591506958, + "learning_rate": 3.2900081234768484e-06, + "loss": 1.8344, + "step": 1131 + }, + { + "epoch": 1.2405479452054795, + "grad_norm": 2.5837206840515137, + "learning_rate": 3.2879772542648257e-06, + "loss": 1.8503, + "step": 1132 + }, + { + "epoch": 1.2416438356164383, + "grad_norm": 2.1864733695983887, + "learning_rate": 3.285946385052803e-06, + "loss": 1.8822, + "step": 1133 + }, + { + "epoch": 1.2427397260273974, + "grad_norm": 2.2644550800323486, + "learning_rate": 3.2839155158407802e-06, + "loss": 1.7541, + "step": 1134 + }, + { + "epoch": 1.2438356164383562, + "grad_norm": 2.292646646499634, + "learning_rate": 3.2818846466287575e-06, + "loss": 1.8575, + "step": 1135 + }, + { + "epoch": 1.244931506849315, + "grad_norm": 2.2162070274353027, + "learning_rate": 3.2798537774167344e-06, + "loss": 1.7525, + "step": 1136 + }, + { + "epoch": 1.246027397260274, + "grad_norm": 2.32688307762146, + "learning_rate": 3.277822908204712e-06, + "loss": 1.8955, + "step": 1137 + }, + { + "epoch": 1.2471232876712328, + "grad_norm": 2.3226473331451416, + "learning_rate": 3.275792038992689e-06, + "loss": 1.8255, + "step": 1138 + }, + { + "epoch": 1.2482191780821918, + "grad_norm": 2.3533706665039062, + "learning_rate": 3.2737611697806666e-06, + "loss": 1.783, + "step": 1139 + }, + { + "epoch": 1.2493150684931507, + "grad_norm": 2.237661123275757, + "learning_rate": 3.2717303005686435e-06, + "loss": 1.8367, + "step": 1140 + }, + { + "epoch": 1.2493150684931507, + "eval_loss": 2.0123534202575684, + "eval_runtime": 37.3573, + "eval_samples_per_second": 26.769, + "eval_steps_per_second": 1.124, + "step": 1140 + }, + { + "epoch": 1.2504109589041095, + "grad_norm": 2.283047676086426, + "learning_rate": 3.269699431356621e-06, + "loss": 1.7987, + "step": 1141 + }, + { + "epoch": 1.2515068493150685, + "grad_norm": 2.2264456748962402, + "learning_rate": 3.267668562144598e-06, + "loss": 1.8307, + "step": 1142 + }, + { + "epoch": 1.2526027397260273, + "grad_norm": 2.2724905014038086, + "learning_rate": 3.2656376929325757e-06, + "loss": 1.8237, + "step": 1143 + }, + { + "epoch": 1.2536986301369863, + "grad_norm": 2.270782947540283, + "learning_rate": 3.2636068237205526e-06, + "loss": 1.7846, + "step": 1144 + }, + { + "epoch": 1.2547945205479452, + "grad_norm": 2.2032551765441895, + "learning_rate": 3.2615759545085303e-06, + "loss": 1.8438, + "step": 1145 + }, + { + "epoch": 1.2558904109589042, + "grad_norm": 2.199150800704956, + "learning_rate": 3.259545085296507e-06, + "loss": 1.8199, + "step": 1146 + }, + { + "epoch": 1.256986301369863, + "grad_norm": 2.4404170513153076, + "learning_rate": 3.257514216084484e-06, + "loss": 1.8817, + "step": 1147 + }, + { + "epoch": 1.2580821917808218, + "grad_norm": 2.281348943710327, + "learning_rate": 3.2554833468724617e-06, + "loss": 1.771, + "step": 1148 + }, + { + "epoch": 1.2591780821917808, + "grad_norm": 2.19482159614563, + "learning_rate": 3.2534524776604385e-06, + "loss": 1.8684, + "step": 1149 + }, + { + "epoch": 1.2602739726027397, + "grad_norm": 2.221618413925171, + "learning_rate": 3.2514216084484162e-06, + "loss": 1.8398, + "step": 1150 + }, + { + "epoch": 1.2602739726027397, + "eval_loss": 2.007969617843628, + "eval_runtime": 37.3679, + "eval_samples_per_second": 26.761, + "eval_steps_per_second": 1.124, + "step": 1150 + }, + { + "epoch": 1.2613698630136987, + "grad_norm": 2.193432569503784, + "learning_rate": 3.249390739236393e-06, + "loss": 1.8122, + "step": 1151 + }, + { + "epoch": 1.2624657534246575, + "grad_norm": 2.246462106704712, + "learning_rate": 3.2473598700243708e-06, + "loss": 1.8447, + "step": 1152 + }, + { + "epoch": 1.2635616438356165, + "grad_norm": 2.3411879539489746, + "learning_rate": 3.2453290008123476e-06, + "loss": 1.7988, + "step": 1153 + }, + { + "epoch": 1.2646575342465753, + "grad_norm": 2.245286464691162, + "learning_rate": 3.2432981316003253e-06, + "loss": 1.8209, + "step": 1154 + }, + { + "epoch": 1.2657534246575342, + "grad_norm": 2.292067527770996, + "learning_rate": 3.241267262388302e-06, + "loss": 1.8285, + "step": 1155 + }, + { + "epoch": 1.2668493150684932, + "grad_norm": 2.3058111667633057, + "learning_rate": 3.23923639317628e-06, + "loss": 1.7842, + "step": 1156 + }, + { + "epoch": 1.267945205479452, + "grad_norm": 2.306243896484375, + "learning_rate": 3.2372055239642567e-06, + "loss": 1.8221, + "step": 1157 + }, + { + "epoch": 1.269041095890411, + "grad_norm": 2.399759531021118, + "learning_rate": 3.2351746547522344e-06, + "loss": 1.7836, + "step": 1158 + }, + { + "epoch": 1.2701369863013698, + "grad_norm": 2.3405187129974365, + "learning_rate": 3.2331437855402113e-06, + "loss": 1.8365, + "step": 1159 + }, + { + "epoch": 1.2712328767123289, + "grad_norm": 2.3855013847351074, + "learning_rate": 3.2311129163281885e-06, + "loss": 1.8973, + "step": 1160 + }, + { + "epoch": 1.2712328767123289, + "eval_loss": 2.005983591079712, + "eval_runtime": 37.3061, + "eval_samples_per_second": 26.805, + "eval_steps_per_second": 1.126, + "step": 1160 + }, + { + "epoch": 1.2723287671232877, + "grad_norm": 2.2767140865325928, + "learning_rate": 3.229082047116166e-06, + "loss": 1.7791, + "step": 1161 + }, + { + "epoch": 1.2734246575342465, + "grad_norm": 2.289494276046753, + "learning_rate": 3.227051177904143e-06, + "loss": 1.8508, + "step": 1162 + }, + { + "epoch": 1.2745205479452055, + "grad_norm": 2.1836023330688477, + "learning_rate": 3.2250203086921204e-06, + "loss": 1.8275, + "step": 1163 + }, + { + "epoch": 1.2756164383561643, + "grad_norm": 2.1917505264282227, + "learning_rate": 3.2229894394800977e-06, + "loss": 1.8289, + "step": 1164 + }, + { + "epoch": 1.2767123287671232, + "grad_norm": 2.315983295440674, + "learning_rate": 3.220958570268075e-06, + "loss": 1.8655, + "step": 1165 + }, + { + "epoch": 1.2778082191780822, + "grad_norm": 2.3628008365631104, + "learning_rate": 3.218927701056052e-06, + "loss": 1.7396, + "step": 1166 + }, + { + "epoch": 1.2789041095890412, + "grad_norm": 2.3200488090515137, + "learning_rate": 3.2168968318440295e-06, + "loss": 1.8196, + "step": 1167 + }, + { + "epoch": 1.28, + "grad_norm": 2.2591991424560547, + "learning_rate": 3.2148659626320068e-06, + "loss": 1.7323, + "step": 1168 + }, + { + "epoch": 1.2810958904109588, + "grad_norm": 2.2458770275115967, + "learning_rate": 3.212835093419984e-06, + "loss": 1.8881, + "step": 1169 + }, + { + "epoch": 1.2821917808219179, + "grad_norm": 2.202010154724121, + "learning_rate": 3.2108042242079613e-06, + "loss": 1.7776, + "step": 1170 + }, + { + "epoch": 1.2821917808219179, + "eval_loss": 2.005174398422241, + "eval_runtime": 37.3597, + "eval_samples_per_second": 26.767, + "eval_steps_per_second": 1.124, + "step": 1170 + }, + { + "epoch": 1.2832876712328767, + "grad_norm": 2.2375359535217285, + "learning_rate": 3.2087733549959386e-06, + "loss": 1.7994, + "step": 1171 + }, + { + "epoch": 1.2843835616438355, + "grad_norm": 2.3152217864990234, + "learning_rate": 3.206742485783916e-06, + "loss": 1.8767, + "step": 1172 + }, + { + "epoch": 1.2854794520547945, + "grad_norm": 2.3317618370056152, + "learning_rate": 3.2047116165718927e-06, + "loss": 1.772, + "step": 1173 + }, + { + "epoch": 1.2865753424657536, + "grad_norm": 2.249943971633911, + "learning_rate": 3.2026807473598704e-06, + "loss": 1.7793, + "step": 1174 + }, + { + "epoch": 1.2876712328767124, + "grad_norm": 2.320481538772583, + "learning_rate": 3.2006498781478473e-06, + "loss": 1.8275, + "step": 1175 + }, + { + "epoch": 1.2887671232876712, + "grad_norm": 2.21785569190979, + "learning_rate": 3.198619008935825e-06, + "loss": 1.8449, + "step": 1176 + }, + { + "epoch": 1.2898630136986302, + "grad_norm": 2.469965934753418, + "learning_rate": 3.196588139723802e-06, + "loss": 1.779, + "step": 1177 + }, + { + "epoch": 1.290958904109589, + "grad_norm": 2.1989049911499023, + "learning_rate": 3.1945572705117795e-06, + "loss": 1.8188, + "step": 1178 + }, + { + "epoch": 1.2920547945205478, + "grad_norm": 2.293513298034668, + "learning_rate": 3.1925264012997564e-06, + "loss": 1.8377, + "step": 1179 + }, + { + "epoch": 1.2931506849315069, + "grad_norm": 2.3462443351745605, + "learning_rate": 3.190495532087734e-06, + "loss": 1.8418, + "step": 1180 + }, + { + "epoch": 1.2931506849315069, + "eval_loss": 2.0061001777648926, + "eval_runtime": 37.2951, + "eval_samples_per_second": 26.813, + "eval_steps_per_second": 1.126, + "step": 1180 + }, + { + "epoch": 1.2942465753424657, + "grad_norm": 2.2071478366851807, + "learning_rate": 3.188464662875711e-06, + "loss": 1.8419, + "step": 1181 + }, + { + "epoch": 1.2953424657534247, + "grad_norm": 2.233333110809326, + "learning_rate": 3.1864337936636886e-06, + "loss": 1.8931, + "step": 1182 + }, + { + "epoch": 1.2964383561643835, + "grad_norm": 2.383085250854492, + "learning_rate": 3.1844029244516655e-06, + "loss": 1.8047, + "step": 1183 + }, + { + "epoch": 1.2975342465753426, + "grad_norm": 2.473808765411377, + "learning_rate": 3.182372055239643e-06, + "loss": 1.7794, + "step": 1184 + }, + { + "epoch": 1.2986301369863014, + "grad_norm": 2.3357670307159424, + "learning_rate": 3.18034118602762e-06, + "loss": 1.7717, + "step": 1185 + }, + { + "epoch": 1.2997260273972602, + "grad_norm": 2.376835823059082, + "learning_rate": 3.1783103168155973e-06, + "loss": 1.8534, + "step": 1186 + }, + { + "epoch": 1.3008219178082192, + "grad_norm": 2.447415590286255, + "learning_rate": 3.1762794476035746e-06, + "loss": 1.8251, + "step": 1187 + }, + { + "epoch": 1.301917808219178, + "grad_norm": 2.5145719051361084, + "learning_rate": 3.174248578391552e-06, + "loss": 1.8117, + "step": 1188 + }, + { + "epoch": 1.303013698630137, + "grad_norm": 2.3134653568267822, + "learning_rate": 3.172217709179529e-06, + "loss": 1.8111, + "step": 1189 + }, + { + "epoch": 1.3041095890410959, + "grad_norm": 2.293311834335327, + "learning_rate": 3.1701868399675064e-06, + "loss": 1.791, + "step": 1190 + }, + { + "epoch": 1.3041095890410959, + "eval_loss": 2.0038933753967285, + "eval_runtime": 37.2282, + "eval_samples_per_second": 26.861, + "eval_steps_per_second": 1.128, + "step": 1190 + }, + { + "epoch": 1.305205479452055, + "grad_norm": 2.298020124435425, + "learning_rate": 3.1681559707554837e-06, + "loss": 1.8301, + "step": 1191 + }, + { + "epoch": 1.3063013698630137, + "grad_norm": 2.5261130332946777, + "learning_rate": 3.166125101543461e-06, + "loss": 1.7714, + "step": 1192 + }, + { + "epoch": 1.3073972602739725, + "grad_norm": 2.270820379257202, + "learning_rate": 3.1640942323314382e-06, + "loss": 1.8246, + "step": 1193 + }, + { + "epoch": 1.3084931506849315, + "grad_norm": 2.373260021209717, + "learning_rate": 3.1620633631194155e-06, + "loss": 1.8316, + "step": 1194 + }, + { + "epoch": 1.3095890410958904, + "grad_norm": 2.3265364170074463, + "learning_rate": 3.1600324939073928e-06, + "loss": 1.7679, + "step": 1195 + }, + { + "epoch": 1.3106849315068494, + "grad_norm": 2.2378532886505127, + "learning_rate": 3.15800162469537e-06, + "loss": 1.8149, + "step": 1196 + }, + { + "epoch": 1.3117808219178082, + "grad_norm": 2.2459309101104736, + "learning_rate": 3.155970755483347e-06, + "loss": 1.8086, + "step": 1197 + }, + { + "epoch": 1.3128767123287672, + "grad_norm": 2.2769811153411865, + "learning_rate": 3.1539398862713246e-06, + "loss": 1.8109, + "step": 1198 + }, + { + "epoch": 1.313972602739726, + "grad_norm": 2.310483694076538, + "learning_rate": 3.1519090170593015e-06, + "loss": 1.8297, + "step": 1199 + }, + { + "epoch": 1.3150684931506849, + "grad_norm": 2.302478313446045, + "learning_rate": 3.149878147847279e-06, + "loss": 1.8237, + "step": 1200 + }, + { + "epoch": 1.3150684931506849, + "eval_loss": 2.0011062622070312, + "eval_runtime": 37.2077, + "eval_samples_per_second": 26.876, + "eval_steps_per_second": 1.129, + "step": 1200 + }, + { + "epoch": 1.316164383561644, + "grad_norm": 2.2240796089172363, + "learning_rate": 3.147847278635256e-06, + "loss": 1.8745, + "step": 1201 + }, + { + "epoch": 1.3172602739726027, + "grad_norm": 2.303537130355835, + "learning_rate": 3.1458164094232337e-06, + "loss": 1.9051, + "step": 1202 + }, + { + "epoch": 1.3183561643835615, + "grad_norm": 2.4038636684417725, + "learning_rate": 3.1437855402112106e-06, + "loss": 1.7974, + "step": 1203 + }, + { + "epoch": 1.3194520547945205, + "grad_norm": 2.3458874225616455, + "learning_rate": 3.1417546709991883e-06, + "loss": 1.7601, + "step": 1204 + }, + { + "epoch": 1.3205479452054796, + "grad_norm": 2.3399100303649902, + "learning_rate": 3.139723801787165e-06, + "loss": 1.8314, + "step": 1205 + }, + { + "epoch": 1.3216438356164384, + "grad_norm": 2.293229818344116, + "learning_rate": 3.137692932575143e-06, + "loss": 1.8117, + "step": 1206 + }, + { + "epoch": 1.3227397260273972, + "grad_norm": 2.5378005504608154, + "learning_rate": 3.1356620633631197e-06, + "loss": 1.8614, + "step": 1207 + }, + { + "epoch": 1.3238356164383562, + "grad_norm": 2.3644745349884033, + "learning_rate": 3.1336311941510974e-06, + "loss": 1.7843, + "step": 1208 + }, + { + "epoch": 1.324931506849315, + "grad_norm": 2.1592626571655273, + "learning_rate": 3.1316003249390742e-06, + "loss": 1.8128, + "step": 1209 + }, + { + "epoch": 1.3260273972602739, + "grad_norm": 2.155649185180664, + "learning_rate": 3.129569455727051e-06, + "loss": 1.8636, + "step": 1210 + }, + { + "epoch": 1.3260273972602739, + "eval_loss": 1.9952574968338013, + "eval_runtime": 37.3004, + "eval_samples_per_second": 26.809, + "eval_steps_per_second": 1.126, + "step": 1210 + }, + { + "epoch": 1.3271232876712329, + "grad_norm": 2.1906397342681885, + "learning_rate": 3.1275385865150288e-06, + "loss": 1.8583, + "step": 1211 + }, + { + "epoch": 1.3282191780821917, + "grad_norm": 2.2708194255828857, + "learning_rate": 3.1255077173030056e-06, + "loss": 1.8363, + "step": 1212 + }, + { + "epoch": 1.3293150684931507, + "grad_norm": 2.1753969192504883, + "learning_rate": 3.1234768480909833e-06, + "loss": 1.8372, + "step": 1213 + }, + { + "epoch": 1.3304109589041095, + "grad_norm": 2.229363441467285, + "learning_rate": 3.12144597887896e-06, + "loss": 1.8179, + "step": 1214 + }, + { + "epoch": 1.3315068493150686, + "grad_norm": 2.362152338027954, + "learning_rate": 3.119415109666938e-06, + "loss": 1.8596, + "step": 1215 + }, + { + "epoch": 1.3326027397260274, + "grad_norm": 2.3278255462646484, + "learning_rate": 3.1173842404549147e-06, + "loss": 1.8778, + "step": 1216 + }, + { + "epoch": 1.3336986301369862, + "grad_norm": 2.477710247039795, + "learning_rate": 3.1153533712428924e-06, + "loss": 1.8297, + "step": 1217 + }, + { + "epoch": 1.3347945205479452, + "grad_norm": 2.2611043453216553, + "learning_rate": 3.1133225020308693e-06, + "loss": 1.8374, + "step": 1218 + }, + { + "epoch": 1.335890410958904, + "grad_norm": 2.198275566101074, + "learning_rate": 3.111291632818847e-06, + "loss": 1.8686, + "step": 1219 + }, + { + "epoch": 1.336986301369863, + "grad_norm": 2.227508068084717, + "learning_rate": 3.109260763606824e-06, + "loss": 1.8642, + "step": 1220 + }, + { + "epoch": 1.336986301369863, + "eval_loss": 1.9998422861099243, + "eval_runtime": 37.5152, + "eval_samples_per_second": 26.656, + "eval_steps_per_second": 1.12, + "step": 1220 + }, + { + "epoch": 1.3380821917808219, + "grad_norm": 2.386310338973999, + "learning_rate": 3.1072298943948015e-06, + "loss": 1.8347, + "step": 1221 + }, + { + "epoch": 1.339178082191781, + "grad_norm": 2.290313482284546, + "learning_rate": 3.1051990251827784e-06, + "loss": 1.8455, + "step": 1222 + }, + { + "epoch": 1.3402739726027397, + "grad_norm": 2.3063457012176514, + "learning_rate": 3.1031681559707557e-06, + "loss": 1.811, + "step": 1223 + }, + { + "epoch": 1.3413698630136985, + "grad_norm": 2.297840118408203, + "learning_rate": 3.101137286758733e-06, + "loss": 1.8558, + "step": 1224 + }, + { + "epoch": 1.3424657534246576, + "grad_norm": 2.40864896774292, + "learning_rate": 3.09910641754671e-06, + "loss": 1.8604, + "step": 1225 + }, + { + "epoch": 1.3435616438356164, + "grad_norm": 2.397599458694458, + "learning_rate": 3.0970755483346875e-06, + "loss": 1.8556, + "step": 1226 + }, + { + "epoch": 1.3446575342465754, + "grad_norm": 2.409029245376587, + "learning_rate": 3.0950446791226648e-06, + "loss": 1.8424, + "step": 1227 + }, + { + "epoch": 1.3457534246575342, + "grad_norm": 2.2789814472198486, + "learning_rate": 3.093013809910642e-06, + "loss": 1.8945, + "step": 1228 + }, + { + "epoch": 1.3468493150684933, + "grad_norm": 2.3749048709869385, + "learning_rate": 3.0909829406986193e-06, + "loss": 1.8387, + "step": 1229 + }, + { + "epoch": 1.347945205479452, + "grad_norm": 2.44486141204834, + "learning_rate": 3.0889520714865966e-06, + "loss": 1.8838, + "step": 1230 + }, + { + "epoch": 1.347945205479452, + "eval_loss": 1.9961321353912354, + "eval_runtime": 37.6273, + "eval_samples_per_second": 26.576, + "eval_steps_per_second": 1.116, + "step": 1230 + }, + { + "epoch": 1.3490410958904109, + "grad_norm": 2.2852954864501953, + "learning_rate": 3.086921202274574e-06, + "loss": 1.8551, + "step": 1231 + }, + { + "epoch": 1.35013698630137, + "grad_norm": 2.3131206035614014, + "learning_rate": 3.084890333062551e-06, + "loss": 1.8183, + "step": 1232 + }, + { + "epoch": 1.3512328767123287, + "grad_norm": 2.4095048904418945, + "learning_rate": 3.0828594638505284e-06, + "loss": 1.7946, + "step": 1233 + }, + { + "epoch": 1.3523287671232875, + "grad_norm": 2.429971933364868, + "learning_rate": 3.0808285946385057e-06, + "loss": 1.8349, + "step": 1234 + }, + { + "epoch": 1.3534246575342466, + "grad_norm": 2.2360565662384033, + "learning_rate": 3.078797725426483e-06, + "loss": 1.794, + "step": 1235 + }, + { + "epoch": 1.3545205479452056, + "grad_norm": 2.3630571365356445, + "learning_rate": 3.07676685621446e-06, + "loss": 1.7377, + "step": 1236 + }, + { + "epoch": 1.3556164383561644, + "grad_norm": 2.2604660987854004, + "learning_rate": 3.0747359870024375e-06, + "loss": 1.8239, + "step": 1237 + }, + { + "epoch": 1.3567123287671232, + "grad_norm": 2.209717035293579, + "learning_rate": 3.0727051177904144e-06, + "loss": 1.7835, + "step": 1238 + }, + { + "epoch": 1.3578082191780823, + "grad_norm": 2.3221793174743652, + "learning_rate": 3.070674248578392e-06, + "loss": 1.8192, + "step": 1239 + }, + { + "epoch": 1.358904109589041, + "grad_norm": 2.357313394546509, + "learning_rate": 3.068643379366369e-06, + "loss": 1.8246, + "step": 1240 + }, + { + "epoch": 1.358904109589041, + "eval_loss": 1.9935922622680664, + "eval_runtime": 37.2832, + "eval_samples_per_second": 26.822, + "eval_steps_per_second": 1.127, + "step": 1240 + }, + { + "epoch": 1.3599999999999999, + "grad_norm": 2.2563390731811523, + "learning_rate": 3.0666125101543466e-06, + "loss": 1.8175, + "step": 1241 + }, + { + "epoch": 1.361095890410959, + "grad_norm": 2.2531445026397705, + "learning_rate": 3.0645816409423235e-06, + "loss": 1.7713, + "step": 1242 + }, + { + "epoch": 1.362191780821918, + "grad_norm": 2.2369470596313477, + "learning_rate": 3.062550771730301e-06, + "loss": 1.8377, + "step": 1243 + }, + { + "epoch": 1.3632876712328768, + "grad_norm": 2.4063796997070312, + "learning_rate": 3.060519902518278e-06, + "loss": 1.8116, + "step": 1244 + }, + { + "epoch": 1.3643835616438356, + "grad_norm": 2.4628145694732666, + "learning_rate": 3.0584890333062557e-06, + "loss": 1.7998, + "step": 1245 + }, + { + "epoch": 1.3654794520547946, + "grad_norm": 2.3018743991851807, + "learning_rate": 3.0564581640942326e-06, + "loss": 1.7413, + "step": 1246 + }, + { + "epoch": 1.3665753424657534, + "grad_norm": 2.2432262897491455, + "learning_rate": 3.0544272948822094e-06, + "loss": 1.82, + "step": 1247 + }, + { + "epoch": 1.3676712328767122, + "grad_norm": 2.368435859680176, + "learning_rate": 3.052396425670187e-06, + "loss": 1.7976, + "step": 1248 + }, + { + "epoch": 1.3687671232876713, + "grad_norm": 2.2668025493621826, + "learning_rate": 3.050365556458164e-06, + "loss": 1.8379, + "step": 1249 + }, + { + "epoch": 1.36986301369863, + "grad_norm": 2.2622179985046387, + "learning_rate": 3.0483346872461417e-06, + "loss": 1.794, + "step": 1250 + }, + { + "epoch": 1.36986301369863, + "eval_loss": 1.9959099292755127, + "eval_runtime": 37.3358, + "eval_samples_per_second": 26.784, + "eval_steps_per_second": 1.125, + "step": 1250 + }, + { + "epoch": 1.370958904109589, + "grad_norm": 2.3414738178253174, + "learning_rate": 3.0463038180341185e-06, + "loss": 1.7993, + "step": 1251 + }, + { + "epoch": 1.372054794520548, + "grad_norm": 2.2678418159484863, + "learning_rate": 3.0442729488220962e-06, + "loss": 1.8581, + "step": 1252 + }, + { + "epoch": 1.373150684931507, + "grad_norm": 2.268195867538452, + "learning_rate": 3.042242079610073e-06, + "loss": 1.8486, + "step": 1253 + }, + { + "epoch": 1.3742465753424657, + "grad_norm": 2.230173110961914, + "learning_rate": 3.0402112103980508e-06, + "loss": 1.8468, + "step": 1254 + }, + { + "epoch": 1.3753424657534246, + "grad_norm": 2.3020238876342773, + "learning_rate": 3.0381803411860276e-06, + "loss": 1.7623, + "step": 1255 + }, + { + "epoch": 1.3764383561643836, + "grad_norm": 2.385343313217163, + "learning_rate": 3.0361494719740053e-06, + "loss": 1.7976, + "step": 1256 + }, + { + "epoch": 1.3775342465753424, + "grad_norm": 2.344768524169922, + "learning_rate": 3.034118602761982e-06, + "loss": 1.807, + "step": 1257 + }, + { + "epoch": 1.3786301369863014, + "grad_norm": 2.411760091781616, + "learning_rate": 3.03208773354996e-06, + "loss": 1.8079, + "step": 1258 + }, + { + "epoch": 1.3797260273972602, + "grad_norm": 2.3160128593444824, + "learning_rate": 3.0300568643379367e-06, + "loss": 1.7868, + "step": 1259 + }, + { + "epoch": 1.3808219178082193, + "grad_norm": 2.334632635116577, + "learning_rate": 3.028025995125914e-06, + "loss": 1.8466, + "step": 1260 + }, + { + "epoch": 1.3808219178082193, + "eval_loss": 1.9903327226638794, + "eval_runtime": 37.2256, + "eval_samples_per_second": 26.863, + "eval_steps_per_second": 1.128, + "step": 1260 + }, + { + "epoch": 1.381917808219178, + "grad_norm": 2.3443334102630615, + "learning_rate": 3.0259951259138913e-06, + "loss": 1.8302, + "step": 1261 + }, + { + "epoch": 1.383013698630137, + "grad_norm": 2.311563014984131, + "learning_rate": 3.0239642567018686e-06, + "loss": 1.7727, + "step": 1262 + }, + { + "epoch": 1.384109589041096, + "grad_norm": 2.2501776218414307, + "learning_rate": 3.021933387489846e-06, + "loss": 1.8058, + "step": 1263 + }, + { + "epoch": 1.3852054794520547, + "grad_norm": 2.259157657623291, + "learning_rate": 3.019902518277823e-06, + "loss": 1.7627, + "step": 1264 + }, + { + "epoch": 1.3863013698630138, + "grad_norm": 2.4380314350128174, + "learning_rate": 3.0178716490658004e-06, + "loss": 1.8237, + "step": 1265 + }, + { + "epoch": 1.3873972602739726, + "grad_norm": 2.3203837871551514, + "learning_rate": 3.0158407798537777e-06, + "loss": 1.8557, + "step": 1266 + }, + { + "epoch": 1.3884931506849316, + "grad_norm": 2.220996141433716, + "learning_rate": 3.013809910641755e-06, + "loss": 1.7641, + "step": 1267 + }, + { + "epoch": 1.3895890410958904, + "grad_norm": 2.190006971359253, + "learning_rate": 3.0117790414297322e-06, + "loss": 1.7705, + "step": 1268 + }, + { + "epoch": 1.3906849315068492, + "grad_norm": 2.229748249053955, + "learning_rate": 3.0097481722177095e-06, + "loss": 1.7709, + "step": 1269 + }, + { + "epoch": 1.3917808219178083, + "grad_norm": 2.3755619525909424, + "learning_rate": 3.0077173030056868e-06, + "loss": 1.8172, + "step": 1270 + }, + { + "epoch": 1.3917808219178083, + "eval_loss": 1.992952585220337, + "eval_runtime": 37.2403, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 1270 + }, + { + "epoch": 1.392876712328767, + "grad_norm": 2.2514004707336426, + "learning_rate": 3.005686433793664e-06, + "loss": 1.7719, + "step": 1271 + }, + { + "epoch": 1.393972602739726, + "grad_norm": 2.3010191917419434, + "learning_rate": 3.0036555645816413e-06, + "loss": 1.8369, + "step": 1272 + }, + { + "epoch": 1.395068493150685, + "grad_norm": 2.68790602684021, + "learning_rate": 3.001624695369618e-06, + "loss": 1.7974, + "step": 1273 + }, + { + "epoch": 1.396164383561644, + "grad_norm": 2.454904794692993, + "learning_rate": 2.999593826157596e-06, + "loss": 1.8509, + "step": 1274 + }, + { + "epoch": 1.3972602739726028, + "grad_norm": 2.251556396484375, + "learning_rate": 2.9975629569455727e-06, + "loss": 1.8011, + "step": 1275 + }, + { + "epoch": 1.3983561643835616, + "grad_norm": 2.3789098262786865, + "learning_rate": 2.9955320877335504e-06, + "loss": 1.8488, + "step": 1276 + }, + { + "epoch": 1.3994520547945206, + "grad_norm": 2.2659196853637695, + "learning_rate": 2.9935012185215273e-06, + "loss": 1.7946, + "step": 1277 + }, + { + "epoch": 1.4005479452054794, + "grad_norm": 2.2148971557617188, + "learning_rate": 2.991470349309505e-06, + "loss": 1.8388, + "step": 1278 + }, + { + "epoch": 1.4016438356164382, + "grad_norm": 2.1883089542388916, + "learning_rate": 2.989439480097482e-06, + "loss": 1.7925, + "step": 1279 + }, + { + "epoch": 1.4027397260273973, + "grad_norm": 2.290208339691162, + "learning_rate": 2.9874086108854595e-06, + "loss": 1.8099, + "step": 1280 + }, + { + "epoch": 1.4027397260273973, + "eval_loss": 1.9924782514572144, + "eval_runtime": 37.1891, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 1280 + }, + { + "epoch": 1.403835616438356, + "grad_norm": 2.121649980545044, + "learning_rate": 2.9853777416734364e-06, + "loss": 1.7991, + "step": 1281 + }, + { + "epoch": 1.4049315068493151, + "grad_norm": 2.3528690338134766, + "learning_rate": 2.983346872461414e-06, + "loss": 1.8487, + "step": 1282 + }, + { + "epoch": 1.406027397260274, + "grad_norm": 2.391491174697876, + "learning_rate": 2.981316003249391e-06, + "loss": 1.8564, + "step": 1283 + }, + { + "epoch": 1.407123287671233, + "grad_norm": 2.336615800857544, + "learning_rate": 2.9792851340373678e-06, + "loss": 1.8216, + "step": 1284 + }, + { + "epoch": 1.4082191780821918, + "grad_norm": 2.2958548069000244, + "learning_rate": 2.9772542648253455e-06, + "loss": 1.8003, + "step": 1285 + }, + { + "epoch": 1.4093150684931506, + "grad_norm": 2.2062184810638428, + "learning_rate": 2.9752233956133223e-06, + "loss": 1.8488, + "step": 1286 + }, + { + "epoch": 1.4104109589041096, + "grad_norm": 2.486133575439453, + "learning_rate": 2.9731925264013e-06, + "loss": 1.7914, + "step": 1287 + }, + { + "epoch": 1.4115068493150684, + "grad_norm": 2.2933201789855957, + "learning_rate": 2.971161657189277e-06, + "loss": 1.8164, + "step": 1288 + }, + { + "epoch": 1.4126027397260275, + "grad_norm": 2.3301920890808105, + "learning_rate": 2.9691307879772546e-06, + "loss": 1.7883, + "step": 1289 + }, + { + "epoch": 1.4136986301369863, + "grad_norm": 2.248832941055298, + "learning_rate": 2.9670999187652314e-06, + "loss": 1.7916, + "step": 1290 + }, + { + "epoch": 1.4136986301369863, + "eval_loss": 1.9919097423553467, + "eval_runtime": 37.1283, + "eval_samples_per_second": 26.934, + "eval_steps_per_second": 1.131, + "step": 1290 + }, + { + "epoch": 1.4147945205479453, + "grad_norm": 2.37760853767395, + "learning_rate": 2.965069049553209e-06, + "loss": 1.7854, + "step": 1291 + }, + { + "epoch": 1.415890410958904, + "grad_norm": 2.327718734741211, + "learning_rate": 2.963038180341186e-06, + "loss": 1.874, + "step": 1292 + }, + { + "epoch": 1.416986301369863, + "grad_norm": 2.2413878440856934, + "learning_rate": 2.9610073111291637e-06, + "loss": 1.7591, + "step": 1293 + }, + { + "epoch": 1.418082191780822, + "grad_norm": 2.2203164100646973, + "learning_rate": 2.9589764419171405e-06, + "loss": 1.8466, + "step": 1294 + }, + { + "epoch": 1.4191780821917808, + "grad_norm": 2.2070505619049072, + "learning_rate": 2.9569455727051182e-06, + "loss": 1.8689, + "step": 1295 + }, + { + "epoch": 1.4202739726027398, + "grad_norm": 2.2122223377227783, + "learning_rate": 2.954914703493095e-06, + "loss": 1.7689, + "step": 1296 + }, + { + "epoch": 1.4213698630136986, + "grad_norm": 2.331303358078003, + "learning_rate": 2.9528838342810724e-06, + "loss": 1.8089, + "step": 1297 + }, + { + "epoch": 1.4224657534246576, + "grad_norm": 2.408184289932251, + "learning_rate": 2.9508529650690496e-06, + "loss": 1.7847, + "step": 1298 + }, + { + "epoch": 1.4235616438356165, + "grad_norm": 2.212766170501709, + "learning_rate": 2.948822095857027e-06, + "loss": 1.7899, + "step": 1299 + }, + { + "epoch": 1.4246575342465753, + "grad_norm": 2.259953737258911, + "learning_rate": 2.946791226645004e-06, + "loss": 1.8435, + "step": 1300 + }, + { + "epoch": 1.4246575342465753, + "eval_loss": 1.9893293380737305, + "eval_runtime": 37.1675, + "eval_samples_per_second": 26.905, + "eval_steps_per_second": 1.13, + "step": 1300 + }, + { + "epoch": 1.4257534246575343, + "grad_norm": 2.342400312423706, + "learning_rate": 2.9447603574329815e-06, + "loss": 1.7779, + "step": 1301 + }, + { + "epoch": 1.426849315068493, + "grad_norm": 2.237618923187256, + "learning_rate": 2.9427294882209587e-06, + "loss": 1.7509, + "step": 1302 + }, + { + "epoch": 1.427945205479452, + "grad_norm": 2.3418633937835693, + "learning_rate": 2.940698619008936e-06, + "loss": 1.7963, + "step": 1303 + }, + { + "epoch": 1.429041095890411, + "grad_norm": 2.470827579498291, + "learning_rate": 2.9386677497969133e-06, + "loss": 1.8045, + "step": 1304 + }, + { + "epoch": 1.43013698630137, + "grad_norm": 2.508345365524292, + "learning_rate": 2.9366368805848906e-06, + "loss": 1.7926, + "step": 1305 + }, + { + "epoch": 1.4312328767123288, + "grad_norm": 2.261319160461426, + "learning_rate": 2.934606011372868e-06, + "loss": 1.831, + "step": 1306 + }, + { + "epoch": 1.4323287671232876, + "grad_norm": 2.2703914642333984, + "learning_rate": 2.932575142160845e-06, + "loss": 1.8352, + "step": 1307 + }, + { + "epoch": 1.4334246575342466, + "grad_norm": 2.243626117706299, + "learning_rate": 2.9305442729488224e-06, + "loss": 1.8162, + "step": 1308 + }, + { + "epoch": 1.4345205479452054, + "grad_norm": 2.3859238624572754, + "learning_rate": 2.9285134037367997e-06, + "loss": 1.8171, + "step": 1309 + }, + { + "epoch": 1.4356164383561643, + "grad_norm": 2.3350934982299805, + "learning_rate": 2.9264825345247765e-06, + "loss": 1.8066, + "step": 1310 + }, + { + "epoch": 1.4356164383561643, + "eval_loss": 1.9905259609222412, + "eval_runtime": 37.2009, + "eval_samples_per_second": 26.881, + "eval_steps_per_second": 1.129, + "step": 1310 + }, + { + "epoch": 1.4367123287671233, + "grad_norm": 2.1943347454071045, + "learning_rate": 2.9244516653127542e-06, + "loss": 1.8435, + "step": 1311 + }, + { + "epoch": 1.4378082191780823, + "grad_norm": 2.429128885269165, + "learning_rate": 2.922420796100731e-06, + "loss": 1.8179, + "step": 1312 + }, + { + "epoch": 1.4389041095890411, + "grad_norm": 2.3677234649658203, + "learning_rate": 2.9203899268887088e-06, + "loss": 1.7245, + "step": 1313 + }, + { + "epoch": 1.44, + "grad_norm": 2.255570888519287, + "learning_rate": 2.9183590576766856e-06, + "loss": 1.8545, + "step": 1314 + }, + { + "epoch": 1.441095890410959, + "grad_norm": 2.5176682472229004, + "learning_rate": 2.9163281884646633e-06, + "loss": 1.7509, + "step": 1315 + }, + { + "epoch": 1.4421917808219178, + "grad_norm": 2.29972767829895, + "learning_rate": 2.91429731925264e-06, + "loss": 1.8116, + "step": 1316 + }, + { + "epoch": 1.4432876712328766, + "grad_norm": 2.2853286266326904, + "learning_rate": 2.912266450040618e-06, + "loss": 1.8552, + "step": 1317 + }, + { + "epoch": 1.4443835616438356, + "grad_norm": 2.505075216293335, + "learning_rate": 2.9102355808285947e-06, + "loss": 1.811, + "step": 1318 + }, + { + "epoch": 1.4454794520547944, + "grad_norm": 2.229064464569092, + "learning_rate": 2.9082047116165724e-06, + "loss": 1.8068, + "step": 1319 + }, + { + "epoch": 1.4465753424657535, + "grad_norm": 2.199187994003296, + "learning_rate": 2.9061738424045493e-06, + "loss": 1.7825, + "step": 1320 + }, + { + "epoch": 1.4465753424657535, + "eval_loss": 1.9851222038269043, + "eval_runtime": 37.4614, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 1320 + }, + { + "epoch": 1.4476712328767123, + "grad_norm": 2.4318959712982178, + "learning_rate": 2.904142973192527e-06, + "loss": 1.77, + "step": 1321 + }, + { + "epoch": 1.4487671232876713, + "grad_norm": 2.480703592300415, + "learning_rate": 2.902112103980504e-06, + "loss": 1.8774, + "step": 1322 + }, + { + "epoch": 1.4498630136986301, + "grad_norm": 2.367624521255493, + "learning_rate": 2.900081234768481e-06, + "loss": 1.818, + "step": 1323 + }, + { + "epoch": 1.450958904109589, + "grad_norm": 2.245974540710449, + "learning_rate": 2.8980503655564584e-06, + "loss": 1.8054, + "step": 1324 + }, + { + "epoch": 1.452054794520548, + "grad_norm": 2.352687120437622, + "learning_rate": 2.8960194963444357e-06, + "loss": 1.7922, + "step": 1325 + }, + { + "epoch": 1.4531506849315068, + "grad_norm": 2.308253526687622, + "learning_rate": 2.893988627132413e-06, + "loss": 1.7637, + "step": 1326 + }, + { + "epoch": 1.4542465753424658, + "grad_norm": 2.309495687484741, + "learning_rate": 2.8919577579203902e-06, + "loss": 1.7905, + "step": 1327 + }, + { + "epoch": 1.4553424657534246, + "grad_norm": 2.354365348815918, + "learning_rate": 2.8899268887083675e-06, + "loss": 1.7321, + "step": 1328 + }, + { + "epoch": 1.4564383561643837, + "grad_norm": 2.3886210918426514, + "learning_rate": 2.8878960194963448e-06, + "loss": 1.7932, + "step": 1329 + }, + { + "epoch": 1.4575342465753425, + "grad_norm": 2.395714044570923, + "learning_rate": 2.885865150284322e-06, + "loss": 1.813, + "step": 1330 + }, + { + "epoch": 1.4575342465753425, + "eval_loss": 1.9851990938186646, + "eval_runtime": 37.2966, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 1330 + }, + { + "epoch": 1.4586301369863013, + "grad_norm": 2.433069944381714, + "learning_rate": 2.8838342810722993e-06, + "loss": 1.8346, + "step": 1331 + }, + { + "epoch": 1.4597260273972603, + "grad_norm": 2.295894145965576, + "learning_rate": 2.8818034118602766e-06, + "loss": 1.7855, + "step": 1332 + }, + { + "epoch": 1.4608219178082191, + "grad_norm": 2.495742082595825, + "learning_rate": 2.879772542648254e-06, + "loss": 1.7601, + "step": 1333 + }, + { + "epoch": 1.4619178082191782, + "grad_norm": 2.35984206199646, + "learning_rate": 2.8777416734362307e-06, + "loss": 1.8329, + "step": 1334 + }, + { + "epoch": 1.463013698630137, + "grad_norm": 2.39632248878479, + "learning_rate": 2.8757108042242084e-06, + "loss": 1.7426, + "step": 1335 + }, + { + "epoch": 1.464109589041096, + "grad_norm": 2.4151153564453125, + "learning_rate": 2.8736799350121853e-06, + "loss": 1.8899, + "step": 1336 + }, + { + "epoch": 1.4652054794520548, + "grad_norm": 2.43914532661438, + "learning_rate": 2.871649065800163e-06, + "loss": 1.8464, + "step": 1337 + }, + { + "epoch": 1.4663013698630136, + "grad_norm": 2.44490122795105, + "learning_rate": 2.86961819658814e-06, + "loss": 1.8643, + "step": 1338 + }, + { + "epoch": 1.4673972602739727, + "grad_norm": 2.2639644145965576, + "learning_rate": 2.8675873273761175e-06, + "loss": 1.792, + "step": 1339 + }, + { + "epoch": 1.4684931506849315, + "grad_norm": 2.2803847789764404, + "learning_rate": 2.8655564581640944e-06, + "loss": 1.8109, + "step": 1340 + }, + { + "epoch": 1.4684931506849315, + "eval_loss": 1.9856781959533691, + "eval_runtime": 37.2519, + "eval_samples_per_second": 26.844, + "eval_steps_per_second": 1.127, + "step": 1340 + }, + { + "epoch": 1.4695890410958903, + "grad_norm": 2.494727849960327, + "learning_rate": 2.863525588952072e-06, + "loss": 1.8032, + "step": 1341 + }, + { + "epoch": 1.4706849315068493, + "grad_norm": 2.239595413208008, + "learning_rate": 2.861494719740049e-06, + "loss": 1.8659, + "step": 1342 + }, + { + "epoch": 1.4717808219178083, + "grad_norm": 2.2922589778900146, + "learning_rate": 2.8594638505280266e-06, + "loss": 1.7587, + "step": 1343 + }, + { + "epoch": 1.4728767123287672, + "grad_norm": 2.345224142074585, + "learning_rate": 2.8574329813160035e-06, + "loss": 1.7602, + "step": 1344 + }, + { + "epoch": 1.473972602739726, + "grad_norm": 2.3415307998657227, + "learning_rate": 2.855402112103981e-06, + "loss": 1.8044, + "step": 1345 + }, + { + "epoch": 1.475068493150685, + "grad_norm": 2.331557035446167, + "learning_rate": 2.853371242891958e-06, + "loss": 1.7808, + "step": 1346 + }, + { + "epoch": 1.4761643835616438, + "grad_norm": 2.293572425842285, + "learning_rate": 2.851340373679935e-06, + "loss": 1.8017, + "step": 1347 + }, + { + "epoch": 1.4772602739726026, + "grad_norm": 2.3477773666381836, + "learning_rate": 2.8493095044679126e-06, + "loss": 1.8113, + "step": 1348 + }, + { + "epoch": 1.4783561643835617, + "grad_norm": 2.3740994930267334, + "learning_rate": 2.8472786352558894e-06, + "loss": 1.8801, + "step": 1349 + }, + { + "epoch": 1.4794520547945205, + "grad_norm": 2.4202206134796143, + "learning_rate": 2.845247766043867e-06, + "loss": 1.821, + "step": 1350 + }, + { + "epoch": 1.4794520547945205, + "eval_loss": 1.9867810010910034, + "eval_runtime": 37.2597, + "eval_samples_per_second": 26.839, + "eval_steps_per_second": 1.127, + "step": 1350 + }, + { + "epoch": 1.4805479452054795, + "grad_norm": 2.2323198318481445, + "learning_rate": 2.843216896831844e-06, + "loss": 1.7574, + "step": 1351 + }, + { + "epoch": 1.4816438356164383, + "grad_norm": 2.2192625999450684, + "learning_rate": 2.8411860276198217e-06, + "loss": 1.7662, + "step": 1352 + }, + { + "epoch": 1.4827397260273973, + "grad_norm": 2.276507616043091, + "learning_rate": 2.8391551584077985e-06, + "loss": 1.7748, + "step": 1353 + }, + { + "epoch": 1.4838356164383562, + "grad_norm": 2.292773723602295, + "learning_rate": 2.8371242891957762e-06, + "loss": 1.8388, + "step": 1354 + }, + { + "epoch": 1.484931506849315, + "grad_norm": 2.3127009868621826, + "learning_rate": 2.835093419983753e-06, + "loss": 1.861, + "step": 1355 + }, + { + "epoch": 1.486027397260274, + "grad_norm": 2.2752184867858887, + "learning_rate": 2.833062550771731e-06, + "loss": 1.7883, + "step": 1356 + }, + { + "epoch": 1.4871232876712328, + "grad_norm": 2.288158416748047, + "learning_rate": 2.8310316815597076e-06, + "loss": 1.7963, + "step": 1357 + }, + { + "epoch": 1.4882191780821918, + "grad_norm": 2.3287482261657715, + "learning_rate": 2.8290008123476853e-06, + "loss": 1.7534, + "step": 1358 + }, + { + "epoch": 1.4893150684931507, + "grad_norm": 2.441279888153076, + "learning_rate": 2.826969943135662e-06, + "loss": 1.8073, + "step": 1359 + }, + { + "epoch": 1.4904109589041097, + "grad_norm": 2.3399839401245117, + "learning_rate": 2.8249390739236395e-06, + "loss": 1.7876, + "step": 1360 + }, + { + "epoch": 1.4904109589041097, + "eval_loss": 1.9868030548095703, + "eval_runtime": 37.2891, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 1360 + }, + { + "epoch": 1.4915068493150685, + "grad_norm": 2.234246253967285, + "learning_rate": 2.8229082047116167e-06, + "loss": 1.8031, + "step": 1361 + }, + { + "epoch": 1.4926027397260273, + "grad_norm": 2.3136820793151855, + "learning_rate": 2.820877335499594e-06, + "loss": 1.7669, + "step": 1362 + }, + { + "epoch": 1.4936986301369863, + "grad_norm": 2.389420747756958, + "learning_rate": 2.8188464662875713e-06, + "loss": 1.7879, + "step": 1363 + }, + { + "epoch": 1.4947945205479451, + "grad_norm": 2.222306489944458, + "learning_rate": 2.8168155970755486e-06, + "loss": 1.7412, + "step": 1364 + }, + { + "epoch": 1.4958904109589042, + "grad_norm": 2.1689319610595703, + "learning_rate": 2.814784727863526e-06, + "loss": 1.7651, + "step": 1365 + }, + { + "epoch": 1.496986301369863, + "grad_norm": 2.262047529220581, + "learning_rate": 2.812753858651503e-06, + "loss": 1.7798, + "step": 1366 + }, + { + "epoch": 1.498082191780822, + "grad_norm": 2.266894817352295, + "learning_rate": 2.8107229894394804e-06, + "loss": 1.8172, + "step": 1367 + }, + { + "epoch": 1.4991780821917808, + "grad_norm": 2.6400444507598877, + "learning_rate": 2.8086921202274577e-06, + "loss": 1.7871, + "step": 1368 + }, + { + "epoch": 1.5002739726027396, + "grad_norm": 2.3868865966796875, + "learning_rate": 2.806661251015435e-06, + "loss": 1.7427, + "step": 1369 + }, + { + "epoch": 1.5013698630136987, + "grad_norm": 2.5206542015075684, + "learning_rate": 2.8046303818034122e-06, + "loss": 1.817, + "step": 1370 + }, + { + "epoch": 1.5013698630136987, + "eval_loss": 1.9870396852493286, + "eval_runtime": 37.2372, + "eval_samples_per_second": 26.855, + "eval_steps_per_second": 1.128, + "step": 1370 + }, + { + "epoch": 1.5024657534246575, + "grad_norm": 2.5236382484436035, + "learning_rate": 2.802599512591389e-06, + "loss": 1.8662, + "step": 1371 + }, + { + "epoch": 1.5035616438356163, + "grad_norm": 2.5095479488372803, + "learning_rate": 2.8005686433793668e-06, + "loss": 1.9168, + "step": 1372 + }, + { + "epoch": 1.5046575342465753, + "grad_norm": 2.307508707046509, + "learning_rate": 2.7985377741673436e-06, + "loss": 1.799, + "step": 1373 + }, + { + "epoch": 1.5057534246575344, + "grad_norm": 2.410879373550415, + "learning_rate": 2.7965069049553213e-06, + "loss": 1.7447, + "step": 1374 + }, + { + "epoch": 1.5068493150684932, + "grad_norm": 2.738558530807495, + "learning_rate": 2.794476035743298e-06, + "loss": 1.8073, + "step": 1375 + }, + { + "epoch": 1.507945205479452, + "grad_norm": 2.3826518058776855, + "learning_rate": 2.792445166531276e-06, + "loss": 1.8506, + "step": 1376 + }, + { + "epoch": 1.509041095890411, + "grad_norm": 2.334123373031616, + "learning_rate": 2.7904142973192527e-06, + "loss": 1.8209, + "step": 1377 + }, + { + "epoch": 1.5101369863013698, + "grad_norm": 2.668513298034668, + "learning_rate": 2.7883834281072304e-06, + "loss": 1.8285, + "step": 1378 + }, + { + "epoch": 1.5112328767123286, + "grad_norm": 2.471938133239746, + "learning_rate": 2.7863525588952073e-06, + "loss": 1.7668, + "step": 1379 + }, + { + "epoch": 1.5123287671232877, + "grad_norm": 2.3712313175201416, + "learning_rate": 2.784321689683185e-06, + "loss": 1.7565, + "step": 1380 + }, + { + "epoch": 1.5123287671232877, + "eval_loss": 1.9826414585113525, + "eval_runtime": 37.2426, + "eval_samples_per_second": 26.851, + "eval_steps_per_second": 1.128, + "step": 1380 + }, + { + "epoch": 1.5134246575342467, + "grad_norm": 2.3364150524139404, + "learning_rate": 2.782290820471162e-06, + "loss": 1.7596, + "step": 1381 + }, + { + "epoch": 1.5145205479452055, + "grad_norm": 2.3088393211364746, + "learning_rate": 2.7802599512591395e-06, + "loss": 1.7568, + "step": 1382 + }, + { + "epoch": 1.5156164383561643, + "grad_norm": 2.2584595680236816, + "learning_rate": 2.7782290820471164e-06, + "loss": 1.833, + "step": 1383 + }, + { + "epoch": 1.5167123287671234, + "grad_norm": 2.2440273761749268, + "learning_rate": 2.7761982128350932e-06, + "loss": 1.8237, + "step": 1384 + }, + { + "epoch": 1.5178082191780822, + "grad_norm": 2.3552212715148926, + "learning_rate": 2.774167343623071e-06, + "loss": 1.8288, + "step": 1385 + }, + { + "epoch": 1.518904109589041, + "grad_norm": 2.3001790046691895, + "learning_rate": 2.772136474411048e-06, + "loss": 1.8519, + "step": 1386 + }, + { + "epoch": 1.52, + "grad_norm": 2.3691201210021973, + "learning_rate": 2.7701056051990255e-06, + "loss": 1.7759, + "step": 1387 + }, + { + "epoch": 1.521095890410959, + "grad_norm": 2.327988624572754, + "learning_rate": 2.7680747359870023e-06, + "loss": 1.7132, + "step": 1388 + }, + { + "epoch": 1.5221917808219179, + "grad_norm": 2.30245304107666, + "learning_rate": 2.76604386677498e-06, + "loss": 1.8167, + "step": 1389 + }, + { + "epoch": 1.5232876712328767, + "grad_norm": 2.290121078491211, + "learning_rate": 2.764012997562957e-06, + "loss": 1.7913, + "step": 1390 + }, + { + "epoch": 1.5232876712328767, + "eval_loss": 1.98329758644104, + "eval_runtime": 37.2462, + "eval_samples_per_second": 26.848, + "eval_steps_per_second": 1.128, + "step": 1390 + }, + { + "epoch": 1.5243835616438357, + "grad_norm": 2.294076681137085, + "learning_rate": 2.7619821283509346e-06, + "loss": 1.8598, + "step": 1391 + }, + { + "epoch": 1.5254794520547945, + "grad_norm": 2.3325250148773193, + "learning_rate": 2.7599512591389114e-06, + "loss": 1.7392, + "step": 1392 + }, + { + "epoch": 1.5265753424657533, + "grad_norm": 2.2802610397338867, + "learning_rate": 2.757920389926889e-06, + "loss": 1.8332, + "step": 1393 + }, + { + "epoch": 1.5276712328767124, + "grad_norm": 2.284207344055176, + "learning_rate": 2.755889520714866e-06, + "loss": 1.7875, + "step": 1394 + }, + { + "epoch": 1.5287671232876714, + "grad_norm": 2.325998306274414, + "learning_rate": 2.7538586515028437e-06, + "loss": 1.7657, + "step": 1395 + }, + { + "epoch": 1.52986301369863, + "grad_norm": 2.285393476486206, + "learning_rate": 2.7518277822908205e-06, + "loss": 1.7416, + "step": 1396 + }, + { + "epoch": 1.530958904109589, + "grad_norm": 2.3110482692718506, + "learning_rate": 2.749796913078798e-06, + "loss": 1.7931, + "step": 1397 + }, + { + "epoch": 1.532054794520548, + "grad_norm": 2.4234025478363037, + "learning_rate": 2.747766043866775e-06, + "loss": 1.8472, + "step": 1398 + }, + { + "epoch": 1.5331506849315069, + "grad_norm": 2.384767532348633, + "learning_rate": 2.7457351746547524e-06, + "loss": 1.7961, + "step": 1399 + }, + { + "epoch": 1.5342465753424657, + "grad_norm": 2.1857433319091797, + "learning_rate": 2.7437043054427297e-06, + "loss": 1.8654, + "step": 1400 + }, + { + "epoch": 1.5342465753424657, + "eval_loss": 1.9802658557891846, + "eval_runtime": 37.1443, + "eval_samples_per_second": 26.922, + "eval_steps_per_second": 1.131, + "step": 1400 + }, + { + "epoch": 1.5353424657534247, + "grad_norm": 2.1302895545959473, + "learning_rate": 2.741673436230707e-06, + "loss": 1.9051, + "step": 1401 + }, + { + "epoch": 1.5364383561643835, + "grad_norm": 2.377307415008545, + "learning_rate": 2.739642567018684e-06, + "loss": 1.7942, + "step": 1402 + }, + { + "epoch": 1.5375342465753423, + "grad_norm": 2.316549062728882, + "learning_rate": 2.7376116978066615e-06, + "loss": 1.7734, + "step": 1403 + }, + { + "epoch": 1.5386301369863014, + "grad_norm": 2.3452532291412354, + "learning_rate": 2.7355808285946388e-06, + "loss": 1.8243, + "step": 1404 + }, + { + "epoch": 1.5397260273972604, + "grad_norm": 2.3139419555664062, + "learning_rate": 2.733549959382616e-06, + "loss": 1.812, + "step": 1405 + }, + { + "epoch": 1.5408219178082192, + "grad_norm": 2.3135571479797363, + "learning_rate": 2.7315190901705933e-06, + "loss": 1.7751, + "step": 1406 + }, + { + "epoch": 1.541917808219178, + "grad_norm": 2.403231620788574, + "learning_rate": 2.7294882209585706e-06, + "loss": 1.8561, + "step": 1407 + }, + { + "epoch": 1.543013698630137, + "grad_norm": 2.4028689861297607, + "learning_rate": 2.727457351746548e-06, + "loss": 1.8249, + "step": 1408 + }, + { + "epoch": 1.5441095890410959, + "grad_norm": 2.4028851985931396, + "learning_rate": 2.725426482534525e-06, + "loss": 1.779, + "step": 1409 + }, + { + "epoch": 1.5452054794520547, + "grad_norm": 2.270709753036499, + "learning_rate": 2.723395613322502e-06, + "loss": 1.7754, + "step": 1410 + }, + { + "epoch": 1.5452054794520547, + "eval_loss": 1.983272910118103, + "eval_runtime": 37.2405, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 1410 + }, + { + "epoch": 1.5463013698630137, + "grad_norm": 2.2672812938690186, + "learning_rate": 2.7213647441104797e-06, + "loss": 1.814, + "step": 1411 + }, + { + "epoch": 1.5473972602739727, + "grad_norm": 2.3539397716522217, + "learning_rate": 2.7193338748984565e-06, + "loss": 1.7907, + "step": 1412 + }, + { + "epoch": 1.5484931506849315, + "grad_norm": 2.3229010105133057, + "learning_rate": 2.7173030056864342e-06, + "loss": 1.8061, + "step": 1413 + }, + { + "epoch": 1.5495890410958904, + "grad_norm": 2.482874870300293, + "learning_rate": 2.715272136474411e-06, + "loss": 1.9577, + "step": 1414 + }, + { + "epoch": 1.5506849315068494, + "grad_norm": 2.211798906326294, + "learning_rate": 2.713241267262389e-06, + "loss": 1.7309, + "step": 1415 + }, + { + "epoch": 1.5517808219178082, + "grad_norm": 2.3159289360046387, + "learning_rate": 2.7112103980503656e-06, + "loss": 1.8132, + "step": 1416 + }, + { + "epoch": 1.552876712328767, + "grad_norm": 2.486035108566284, + "learning_rate": 2.7091795288383433e-06, + "loss": 1.7734, + "step": 1417 + }, + { + "epoch": 1.553972602739726, + "grad_norm": 2.3025195598602295, + "learning_rate": 2.70714865962632e-06, + "loss": 1.8232, + "step": 1418 + }, + { + "epoch": 1.555068493150685, + "grad_norm": 2.234581470489502, + "learning_rate": 2.705117790414298e-06, + "loss": 1.7814, + "step": 1419 + }, + { + "epoch": 1.5561643835616439, + "grad_norm": 2.2855048179626465, + "learning_rate": 2.7030869212022747e-06, + "loss": 1.777, + "step": 1420 + }, + { + "epoch": 1.5561643835616439, + "eval_loss": 1.9851160049438477, + "eval_runtime": 37.4606, + "eval_samples_per_second": 26.695, + "eval_steps_per_second": 1.121, + "step": 1420 + }, + { + "epoch": 1.5572602739726027, + "grad_norm": 2.4190328121185303, + "learning_rate": 2.7010560519902516e-06, + "loss": 1.8456, + "step": 1421 + }, + { + "epoch": 1.5583561643835617, + "grad_norm": 2.225623369216919, + "learning_rate": 2.6990251827782293e-06, + "loss": 1.7638, + "step": 1422 + }, + { + "epoch": 1.5594520547945205, + "grad_norm": 2.1516661643981934, + "learning_rate": 2.696994313566206e-06, + "loss": 1.7861, + "step": 1423 + }, + { + "epoch": 1.5605479452054793, + "grad_norm": 2.2461135387420654, + "learning_rate": 2.694963444354184e-06, + "loss": 1.7763, + "step": 1424 + }, + { + "epoch": 1.5616438356164384, + "grad_norm": 2.280869245529175, + "learning_rate": 2.6929325751421607e-06, + "loss": 1.8589, + "step": 1425 + }, + { + "epoch": 1.5627397260273974, + "grad_norm": 2.2034990787506104, + "learning_rate": 2.6909017059301384e-06, + "loss": 1.7705, + "step": 1426 + }, + { + "epoch": 1.563835616438356, + "grad_norm": 2.1346969604492188, + "learning_rate": 2.6888708367181153e-06, + "loss": 1.8318, + "step": 1427 + }, + { + "epoch": 1.564931506849315, + "grad_norm": 2.37400221824646, + "learning_rate": 2.686839967506093e-06, + "loss": 1.746, + "step": 1428 + }, + { + "epoch": 1.566027397260274, + "grad_norm": 2.392561435699463, + "learning_rate": 2.68480909829407e-06, + "loss": 1.8105, + "step": 1429 + }, + { + "epoch": 1.5671232876712329, + "grad_norm": 2.299830198287964, + "learning_rate": 2.6827782290820475e-06, + "loss": 1.7103, + "step": 1430 + }, + { + "epoch": 1.5671232876712329, + "eval_loss": 1.9877182245254517, + "eval_runtime": 37.4813, + "eval_samples_per_second": 26.68, + "eval_steps_per_second": 1.121, + "step": 1430 + }, + { + "epoch": 1.5682191780821917, + "grad_norm": 2.3431026935577393, + "learning_rate": 2.6807473598700244e-06, + "loss": 1.7217, + "step": 1431 + }, + { + "epoch": 1.5693150684931507, + "grad_norm": 2.3474628925323486, + "learning_rate": 2.678716490658002e-06, + "loss": 1.8419, + "step": 1432 + }, + { + "epoch": 1.5704109589041098, + "grad_norm": 2.2877044677734375, + "learning_rate": 2.676685621445979e-06, + "loss": 1.8031, + "step": 1433 + }, + { + "epoch": 1.5715068493150683, + "grad_norm": 2.278613805770874, + "learning_rate": 2.674654752233956e-06, + "loss": 1.8066, + "step": 1434 + }, + { + "epoch": 1.5726027397260274, + "grad_norm": 2.3357837200164795, + "learning_rate": 2.6726238830219335e-06, + "loss": 1.8079, + "step": 1435 + }, + { + "epoch": 1.5736986301369864, + "grad_norm": 2.1932482719421387, + "learning_rate": 2.6705930138099107e-06, + "loss": 1.8032, + "step": 1436 + }, + { + "epoch": 1.5747945205479452, + "grad_norm": 2.298553705215454, + "learning_rate": 2.668562144597888e-06, + "loss": 1.8439, + "step": 1437 + }, + { + "epoch": 1.575890410958904, + "grad_norm": 2.2906084060668945, + "learning_rate": 2.6665312753858653e-06, + "loss": 1.7144, + "step": 1438 + }, + { + "epoch": 1.576986301369863, + "grad_norm": 2.2512989044189453, + "learning_rate": 2.6645004061738426e-06, + "loss": 1.8031, + "step": 1439 + }, + { + "epoch": 1.5780821917808219, + "grad_norm": 2.177103281021118, + "learning_rate": 2.66246953696182e-06, + "loss": 1.8073, + "step": 1440 + }, + { + "epoch": 1.5780821917808219, + "eval_loss": 1.9914618730545044, + "eval_runtime": 37.2706, + "eval_samples_per_second": 26.831, + "eval_steps_per_second": 1.127, + "step": 1440 + }, + { + "epoch": 1.5791780821917807, + "grad_norm": 2.4129281044006348, + "learning_rate": 2.660438667749797e-06, + "loss": 1.7552, + "step": 1441 + }, + { + "epoch": 1.5802739726027397, + "grad_norm": 2.2975244522094727, + "learning_rate": 2.6584077985377744e-06, + "loss": 1.7318, + "step": 1442 + }, + { + "epoch": 1.5813698630136988, + "grad_norm": 2.4061625003814697, + "learning_rate": 2.6563769293257517e-06, + "loss": 1.7839, + "step": 1443 + }, + { + "epoch": 1.5824657534246576, + "grad_norm": 2.236353635787964, + "learning_rate": 2.654346060113729e-06, + "loss": 1.7995, + "step": 1444 + }, + { + "epoch": 1.5835616438356164, + "grad_norm": 2.2464141845703125, + "learning_rate": 2.6523151909017062e-06, + "loss": 1.7496, + "step": 1445 + }, + { + "epoch": 1.5846575342465754, + "grad_norm": 2.2744123935699463, + "learning_rate": 2.6502843216896835e-06, + "loss": 1.7703, + "step": 1446 + }, + { + "epoch": 1.5857534246575342, + "grad_norm": 2.1578054428100586, + "learning_rate": 2.6482534524776603e-06, + "loss": 1.7561, + "step": 1447 + }, + { + "epoch": 1.586849315068493, + "grad_norm": 2.1924850940704346, + "learning_rate": 2.646222583265638e-06, + "loss": 1.867, + "step": 1448 + }, + { + "epoch": 1.587945205479452, + "grad_norm": 2.303420305252075, + "learning_rate": 2.644191714053615e-06, + "loss": 1.8345, + "step": 1449 + }, + { + "epoch": 1.589041095890411, + "grad_norm": 2.308837413787842, + "learning_rate": 2.6421608448415926e-06, + "loss": 1.7979, + "step": 1450 + }, + { + "epoch": 1.589041095890411, + "eval_loss": 1.9836338758468628, + "eval_runtime": 37.253, + "eval_samples_per_second": 26.843, + "eval_steps_per_second": 1.127, + "step": 1450 + }, + { + "epoch": 1.59013698630137, + "grad_norm": 2.2896456718444824, + "learning_rate": 2.6401299756295694e-06, + "loss": 1.7679, + "step": 1451 + }, + { + "epoch": 1.5912328767123287, + "grad_norm": 2.2482378482818604, + "learning_rate": 2.638099106417547e-06, + "loss": 1.7481, + "step": 1452 + }, + { + "epoch": 1.5923287671232877, + "grad_norm": 2.2620620727539062, + "learning_rate": 2.636068237205524e-06, + "loss": 1.801, + "step": 1453 + }, + { + "epoch": 1.5934246575342466, + "grad_norm": 2.216510772705078, + "learning_rate": 2.6340373679935017e-06, + "loss": 1.7485, + "step": 1454 + }, + { + "epoch": 1.5945205479452054, + "grad_norm": 2.354858160018921, + "learning_rate": 2.6320064987814785e-06, + "loss": 1.8636, + "step": 1455 + }, + { + "epoch": 1.5956164383561644, + "grad_norm": 2.251880168914795, + "learning_rate": 2.6299756295694562e-06, + "loss": 1.7087, + "step": 1456 + }, + { + "epoch": 1.5967123287671234, + "grad_norm": 2.2605879306793213, + "learning_rate": 2.627944760357433e-06, + "loss": 1.8141, + "step": 1457 + }, + { + "epoch": 1.5978082191780822, + "grad_norm": 2.3315021991729736, + "learning_rate": 2.6259138911454104e-06, + "loss": 1.8565, + "step": 1458 + }, + { + "epoch": 1.598904109589041, + "grad_norm": 2.1999425888061523, + "learning_rate": 2.6238830219333877e-06, + "loss": 1.8683, + "step": 1459 + }, + { + "epoch": 1.6, + "grad_norm": 2.2652416229248047, + "learning_rate": 2.621852152721365e-06, + "loss": 1.8489, + "step": 1460 + }, + { + "epoch": 1.6, + "eval_loss": 1.9816856384277344, + "eval_runtime": 37.2727, + "eval_samples_per_second": 26.829, + "eval_steps_per_second": 1.127, + "step": 1460 + }, + { + "epoch": 1.601095890410959, + "grad_norm": 2.469440221786499, + "learning_rate": 2.619821283509342e-06, + "loss": 1.7335, + "step": 1461 + }, + { + "epoch": 1.6021917808219177, + "grad_norm": 2.434267044067383, + "learning_rate": 2.6177904142973195e-06, + "loss": 1.8291, + "step": 1462 + }, + { + "epoch": 1.6032876712328767, + "grad_norm": 2.300492286682129, + "learning_rate": 2.6157595450852968e-06, + "loss": 1.7504, + "step": 1463 + }, + { + "epoch": 1.6043835616438358, + "grad_norm": 2.3906843662261963, + "learning_rate": 2.613728675873274e-06, + "loss": 1.8114, + "step": 1464 + }, + { + "epoch": 1.6054794520547944, + "grad_norm": 2.5111773014068604, + "learning_rate": 2.6116978066612513e-06, + "loss": 1.836, + "step": 1465 + }, + { + "epoch": 1.6065753424657534, + "grad_norm": 2.3834846019744873, + "learning_rate": 2.6096669374492286e-06, + "loss": 1.8589, + "step": 1466 + }, + { + "epoch": 1.6076712328767124, + "grad_norm": 2.2709615230560303, + "learning_rate": 2.607636068237206e-06, + "loss": 1.8052, + "step": 1467 + }, + { + "epoch": 1.6087671232876712, + "grad_norm": 2.3351972103118896, + "learning_rate": 2.605605199025183e-06, + "loss": 1.7559, + "step": 1468 + }, + { + "epoch": 1.60986301369863, + "grad_norm": 2.3994698524475098, + "learning_rate": 2.6035743298131604e-06, + "loss": 1.8099, + "step": 1469 + }, + { + "epoch": 1.610958904109589, + "grad_norm": 2.352590322494507, + "learning_rate": 2.6015434606011377e-06, + "loss": 1.8043, + "step": 1470 + }, + { + "epoch": 1.610958904109589, + "eval_loss": 1.9790339469909668, + "eval_runtime": 37.2619, + "eval_samples_per_second": 26.837, + "eval_steps_per_second": 1.127, + "step": 1470 + }, + { + "epoch": 1.612054794520548, + "grad_norm": 2.35754656791687, + "learning_rate": 2.5995125913891145e-06, + "loss": 1.8252, + "step": 1471 + }, + { + "epoch": 1.6131506849315067, + "grad_norm": 2.4011359214782715, + "learning_rate": 2.5974817221770922e-06, + "loss": 1.8005, + "step": 1472 + }, + { + "epoch": 1.6142465753424657, + "grad_norm": 3.143979787826538, + "learning_rate": 2.595450852965069e-06, + "loss": 1.7952, + "step": 1473 + }, + { + "epoch": 1.6153424657534248, + "grad_norm": 2.3928465843200684, + "learning_rate": 2.593419983753047e-06, + "loss": 1.8344, + "step": 1474 + }, + { + "epoch": 1.6164383561643836, + "grad_norm": 2.370368480682373, + "learning_rate": 2.5913891145410236e-06, + "loss": 1.8358, + "step": 1475 + }, + { + "epoch": 1.6175342465753424, + "grad_norm": 2.6163346767425537, + "learning_rate": 2.5893582453290013e-06, + "loss": 1.7734, + "step": 1476 + }, + { + "epoch": 1.6186301369863014, + "grad_norm": 2.370539903640747, + "learning_rate": 2.587327376116978e-06, + "loss": 1.8383, + "step": 1477 + }, + { + "epoch": 1.6197260273972602, + "grad_norm": 2.397007465362549, + "learning_rate": 2.585296506904956e-06, + "loss": 1.7702, + "step": 1478 + }, + { + "epoch": 1.620821917808219, + "grad_norm": 2.4692702293395996, + "learning_rate": 2.5832656376929327e-06, + "loss": 1.7887, + "step": 1479 + }, + { + "epoch": 1.621917808219178, + "grad_norm": 2.5779969692230225, + "learning_rate": 2.5812347684809104e-06, + "loss": 1.7751, + "step": 1480 + }, + { + "epoch": 1.621917808219178, + "eval_loss": 1.9826369285583496, + "eval_runtime": 37.2779, + "eval_samples_per_second": 26.826, + "eval_steps_per_second": 1.127, + "step": 1480 + }, + { + "epoch": 1.6230136986301371, + "grad_norm": 2.3182761669158936, + "learning_rate": 2.5792038992688873e-06, + "loss": 1.7136, + "step": 1481 + }, + { + "epoch": 1.624109589041096, + "grad_norm": 2.4109292030334473, + "learning_rate": 2.577173030056865e-06, + "loss": 1.8804, + "step": 1482 + }, + { + "epoch": 1.6252054794520547, + "grad_norm": 2.426166534423828, + "learning_rate": 2.575142160844842e-06, + "loss": 1.6996, + "step": 1483 + }, + { + "epoch": 1.6263013698630138, + "grad_norm": 2.5459423065185547, + "learning_rate": 2.5731112916328187e-06, + "loss": 1.8186, + "step": 1484 + }, + { + "epoch": 1.6273972602739726, + "grad_norm": 2.583578586578369, + "learning_rate": 2.5710804224207964e-06, + "loss": 1.8275, + "step": 1485 + }, + { + "epoch": 1.6284931506849314, + "grad_norm": 2.442936658859253, + "learning_rate": 2.5690495532087733e-06, + "loss": 1.7553, + "step": 1486 + }, + { + "epoch": 1.6295890410958904, + "grad_norm": 2.587864637374878, + "learning_rate": 2.567018683996751e-06, + "loss": 1.7171, + "step": 1487 + }, + { + "epoch": 1.6306849315068495, + "grad_norm": 2.4119534492492676, + "learning_rate": 2.564987814784728e-06, + "loss": 1.8296, + "step": 1488 + }, + { + "epoch": 1.6317808219178083, + "grad_norm": 2.3582253456115723, + "learning_rate": 2.5629569455727055e-06, + "loss": 1.7595, + "step": 1489 + }, + { + "epoch": 1.632876712328767, + "grad_norm": 2.522674083709717, + "learning_rate": 2.5609260763606824e-06, + "loss": 1.7908, + "step": 1490 + }, + { + "epoch": 1.632876712328767, + "eval_loss": 1.9791436195373535, + "eval_runtime": 37.2967, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 1490 + }, + { + "epoch": 1.633972602739726, + "grad_norm": 2.2927470207214355, + "learning_rate": 2.55889520714866e-06, + "loss": 1.7789, + "step": 1491 + }, + { + "epoch": 1.635068493150685, + "grad_norm": 2.4469518661499023, + "learning_rate": 2.556864337936637e-06, + "loss": 1.8937, + "step": 1492 + }, + { + "epoch": 1.6361643835616437, + "grad_norm": 2.36860728263855, + "learning_rate": 2.5548334687246146e-06, + "loss": 1.778, + "step": 1493 + }, + { + "epoch": 1.6372602739726028, + "grad_norm": 2.4953925609588623, + "learning_rate": 2.5528025995125915e-06, + "loss": 1.7894, + "step": 1494 + }, + { + "epoch": 1.6383561643835618, + "grad_norm": 2.3478338718414307, + "learning_rate": 2.550771730300569e-06, + "loss": 1.8327, + "step": 1495 + }, + { + "epoch": 1.6394520547945204, + "grad_norm": 2.4399492740631104, + "learning_rate": 2.548740861088546e-06, + "loss": 1.8062, + "step": 1496 + }, + { + "epoch": 1.6405479452054794, + "grad_norm": 2.4457216262817383, + "learning_rate": 2.5467099918765233e-06, + "loss": 1.8268, + "step": 1497 + }, + { + "epoch": 1.6416438356164385, + "grad_norm": 2.365879535675049, + "learning_rate": 2.5446791226645006e-06, + "loss": 1.7288, + "step": 1498 + }, + { + "epoch": 1.6427397260273973, + "grad_norm": 2.4942879676818848, + "learning_rate": 2.542648253452478e-06, + "loss": 1.7111, + "step": 1499 + }, + { + "epoch": 1.643835616438356, + "grad_norm": 2.275982618331909, + "learning_rate": 2.540617384240455e-06, + "loss": 1.8292, + "step": 1500 + }, + { + "epoch": 1.643835616438356, + "eval_loss": 1.9746516942977905, + "eval_runtime": 37.2247, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 1500 + }, + { + "epoch": 1.644931506849315, + "grad_norm": 2.4430429935455322, + "learning_rate": 2.5385865150284324e-06, + "loss": 1.8034, + "step": 1501 + }, + { + "epoch": 1.6460273972602741, + "grad_norm": 2.2778332233428955, + "learning_rate": 2.5365556458164097e-06, + "loss": 1.7589, + "step": 1502 + }, + { + "epoch": 1.6471232876712327, + "grad_norm": 2.2818830013275146, + "learning_rate": 2.534524776604387e-06, + "loss": 1.7743, + "step": 1503 + }, + { + "epoch": 1.6482191780821918, + "grad_norm": 2.321328639984131, + "learning_rate": 2.5324939073923642e-06, + "loss": 1.7922, + "step": 1504 + }, + { + "epoch": 1.6493150684931508, + "grad_norm": 2.4312188625335693, + "learning_rate": 2.5304630381803415e-06, + "loss": 1.6784, + "step": 1505 + }, + { + "epoch": 1.6504109589041096, + "grad_norm": 2.298741102218628, + "learning_rate": 2.5284321689683188e-06, + "loss": 1.7745, + "step": 1506 + }, + { + "epoch": 1.6515068493150684, + "grad_norm": 2.2882816791534424, + "learning_rate": 2.526401299756296e-06, + "loss": 1.7896, + "step": 1507 + }, + { + "epoch": 1.6526027397260274, + "grad_norm": 2.2336251735687256, + "learning_rate": 2.524370430544273e-06, + "loss": 1.7548, + "step": 1508 + }, + { + "epoch": 1.6536986301369863, + "grad_norm": 2.4369094371795654, + "learning_rate": 2.5223395613322506e-06, + "loss": 1.8049, + "step": 1509 + }, + { + "epoch": 1.654794520547945, + "grad_norm": 2.2978999614715576, + "learning_rate": 2.5203086921202274e-06, + "loss": 1.7489, + "step": 1510 + }, + { + "epoch": 1.654794520547945, + "eval_loss": 1.9760501384735107, + "eval_runtime": 37.3461, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 1510 + }, + { + "epoch": 1.655890410958904, + "grad_norm": 2.331296920776367, + "learning_rate": 2.518277822908205e-06, + "loss": 1.8293, + "step": 1511 + }, + { + "epoch": 1.6569863013698631, + "grad_norm": 2.4544663429260254, + "learning_rate": 2.516246953696182e-06, + "loss": 1.7617, + "step": 1512 + }, + { + "epoch": 1.658082191780822, + "grad_norm": 2.4682042598724365, + "learning_rate": 2.5142160844841597e-06, + "loss": 1.8054, + "step": 1513 + }, + { + "epoch": 1.6591780821917808, + "grad_norm": 2.2827956676483154, + "learning_rate": 2.5121852152721365e-06, + "loss": 1.8137, + "step": 1514 + }, + { + "epoch": 1.6602739726027398, + "grad_norm": 2.2961761951446533, + "learning_rate": 2.5101543460601142e-06, + "loss": 1.7938, + "step": 1515 + }, + { + "epoch": 1.6613698630136986, + "grad_norm": 2.3464298248291016, + "learning_rate": 2.508123476848091e-06, + "loss": 1.8026, + "step": 1516 + }, + { + "epoch": 1.6624657534246574, + "grad_norm": 2.4975106716156006, + "learning_rate": 2.506092607636069e-06, + "loss": 1.7457, + "step": 1517 + }, + { + "epoch": 1.6635616438356164, + "grad_norm": 2.4823005199432373, + "learning_rate": 2.5040617384240457e-06, + "loss": 1.8282, + "step": 1518 + }, + { + "epoch": 1.6646575342465755, + "grad_norm": 2.2875800132751465, + "learning_rate": 2.5020308692120234e-06, + "loss": 1.7569, + "step": 1519 + }, + { + "epoch": 1.6657534246575343, + "grad_norm": 2.451022148132324, + "learning_rate": 2.5e-06, + "loss": 1.7483, + "step": 1520 + }, + { + "epoch": 1.6657534246575343, + "eval_loss": 1.9748642444610596, + "eval_runtime": 37.4512, + "eval_samples_per_second": 26.701, + "eval_steps_per_second": 1.121, + "step": 1520 + }, + { + "epoch": 1.666849315068493, + "grad_norm": 2.3427352905273438, + "learning_rate": 2.4979691307879775e-06, + "loss": 1.8493, + "step": 1521 + }, + { + "epoch": 1.6679452054794521, + "grad_norm": 2.4496946334838867, + "learning_rate": 2.4959382615759548e-06, + "loss": 1.7569, + "step": 1522 + }, + { + "epoch": 1.669041095890411, + "grad_norm": 2.141963481903076, + "learning_rate": 2.493907392363932e-06, + "loss": 1.7553, + "step": 1523 + }, + { + "epoch": 1.6701369863013698, + "grad_norm": 2.4115428924560547, + "learning_rate": 2.4918765231519093e-06, + "loss": 1.8606, + "step": 1524 + }, + { + "epoch": 1.6712328767123288, + "grad_norm": 2.614616632461548, + "learning_rate": 2.4898456539398866e-06, + "loss": 1.8217, + "step": 1525 + }, + { + "epoch": 1.6723287671232878, + "grad_norm": 2.338557243347168, + "learning_rate": 2.487814784727864e-06, + "loss": 1.7672, + "step": 1526 + }, + { + "epoch": 1.6734246575342466, + "grad_norm": 2.4309446811676025, + "learning_rate": 2.4857839155158407e-06, + "loss": 1.8476, + "step": 1527 + }, + { + "epoch": 1.6745205479452054, + "grad_norm": 2.25883412361145, + "learning_rate": 2.483753046303818e-06, + "loss": 1.7827, + "step": 1528 + }, + { + "epoch": 1.6756164383561645, + "grad_norm": 2.264845371246338, + "learning_rate": 2.4817221770917953e-06, + "loss": 1.7877, + "step": 1529 + }, + { + "epoch": 1.6767123287671233, + "grad_norm": 2.3584723472595215, + "learning_rate": 2.4796913078797725e-06, + "loss": 1.7891, + "step": 1530 + }, + { + "epoch": 1.6767123287671233, + "eval_loss": 1.9751919507980347, + "eval_runtime": 37.558, + "eval_samples_per_second": 26.625, + "eval_steps_per_second": 1.118, + "step": 1530 + }, + { + "epoch": 1.677808219178082, + "grad_norm": 2.3555185794830322, + "learning_rate": 2.47766043866775e-06, + "loss": 1.8235, + "step": 1531 + }, + { + "epoch": 1.6789041095890411, + "grad_norm": 2.3086600303649902, + "learning_rate": 2.475629569455727e-06, + "loss": 1.8036, + "step": 1532 + }, + { + "epoch": 1.6800000000000002, + "grad_norm": 2.286921262741089, + "learning_rate": 2.4735987002437044e-06, + "loss": 1.7885, + "step": 1533 + }, + { + "epoch": 1.6810958904109587, + "grad_norm": 2.2671141624450684, + "learning_rate": 2.4715678310316816e-06, + "loss": 1.758, + "step": 1534 + }, + { + "epoch": 1.6821917808219178, + "grad_norm": 2.4795749187469482, + "learning_rate": 2.469536961819659e-06, + "loss": 1.7791, + "step": 1535 + }, + { + "epoch": 1.6832876712328768, + "grad_norm": 2.201650857925415, + "learning_rate": 2.467506092607636e-06, + "loss": 1.7802, + "step": 1536 + }, + { + "epoch": 1.6843835616438356, + "grad_norm": 2.3804843425750732, + "learning_rate": 2.4654752233956135e-06, + "loss": 1.7739, + "step": 1537 + }, + { + "epoch": 1.6854794520547944, + "grad_norm": 2.322216510772705, + "learning_rate": 2.4634443541835907e-06, + "loss": 1.758, + "step": 1538 + }, + { + "epoch": 1.6865753424657535, + "grad_norm": 2.3582146167755127, + "learning_rate": 2.461413484971568e-06, + "loss": 1.8439, + "step": 1539 + }, + { + "epoch": 1.6876712328767123, + "grad_norm": 2.3597848415374756, + "learning_rate": 2.4593826157595453e-06, + "loss": 1.7772, + "step": 1540 + }, + { + "epoch": 1.6876712328767123, + "eval_loss": 1.9731802940368652, + "eval_runtime": 37.2439, + "eval_samples_per_second": 26.85, + "eval_steps_per_second": 1.128, + "step": 1540 + }, + { + "epoch": 1.688767123287671, + "grad_norm": 2.404569149017334, + "learning_rate": 2.4573517465475226e-06, + "loss": 1.8683, + "step": 1541 + }, + { + "epoch": 1.6898630136986301, + "grad_norm": 2.318737268447876, + "learning_rate": 2.4553208773355e-06, + "loss": 1.7223, + "step": 1542 + }, + { + "epoch": 1.6909589041095892, + "grad_norm": 2.2535340785980225, + "learning_rate": 2.453290008123477e-06, + "loss": 1.7693, + "step": 1543 + }, + { + "epoch": 1.692054794520548, + "grad_norm": 2.2533493041992188, + "learning_rate": 2.4512591389114544e-06, + "loss": 1.7876, + "step": 1544 + }, + { + "epoch": 1.6931506849315068, + "grad_norm": 2.253278970718384, + "learning_rate": 2.4492282696994317e-06, + "loss": 1.788, + "step": 1545 + }, + { + "epoch": 1.6942465753424658, + "grad_norm": 2.296807289123535, + "learning_rate": 2.447197400487409e-06, + "loss": 1.7877, + "step": 1546 + }, + { + "epoch": 1.6953424657534246, + "grad_norm": 2.4022111892700195, + "learning_rate": 2.4451665312753862e-06, + "loss": 1.7986, + "step": 1547 + }, + { + "epoch": 1.6964383561643834, + "grad_norm": 2.2671430110931396, + "learning_rate": 2.4431356620633635e-06, + "loss": 1.8269, + "step": 1548 + }, + { + "epoch": 1.6975342465753425, + "grad_norm": 2.338252067565918, + "learning_rate": 2.4411047928513408e-06, + "loss": 1.8279, + "step": 1549 + }, + { + "epoch": 1.6986301369863015, + "grad_norm": 2.365365505218506, + "learning_rate": 2.439073923639318e-06, + "loss": 1.7895, + "step": 1550 + }, + { + "epoch": 1.6986301369863015, + "eval_loss": 1.9711860418319702, + "eval_runtime": 37.2757, + "eval_samples_per_second": 26.827, + "eval_steps_per_second": 1.127, + "step": 1550 + }, + { + "epoch": 1.6997260273972603, + "grad_norm": 2.407580614089966, + "learning_rate": 2.4370430544272953e-06, + "loss": 1.7985, + "step": 1551 + }, + { + "epoch": 1.7008219178082191, + "grad_norm": 2.3145346641540527, + "learning_rate": 2.435012185215272e-06, + "loss": 1.7589, + "step": 1552 + }, + { + "epoch": 1.7019178082191782, + "grad_norm": 2.3809280395507812, + "learning_rate": 2.4329813160032495e-06, + "loss": 1.7334, + "step": 1553 + }, + { + "epoch": 1.703013698630137, + "grad_norm": 2.4224488735198975, + "learning_rate": 2.4309504467912267e-06, + "loss": 1.7549, + "step": 1554 + }, + { + "epoch": 1.7041095890410958, + "grad_norm": 2.374133348464966, + "learning_rate": 2.428919577579204e-06, + "loss": 1.7923, + "step": 1555 + }, + { + "epoch": 1.7052054794520548, + "grad_norm": 2.258222818374634, + "learning_rate": 2.4268887083671813e-06, + "loss": 1.7683, + "step": 1556 + }, + { + "epoch": 1.7063013698630138, + "grad_norm": 2.4481396675109863, + "learning_rate": 2.4248578391551586e-06, + "loss": 1.8609, + "step": 1557 + }, + { + "epoch": 1.7073972602739727, + "grad_norm": 2.3932790756225586, + "learning_rate": 2.422826969943136e-06, + "loss": 1.7475, + "step": 1558 + }, + { + "epoch": 1.7084931506849315, + "grad_norm": 2.3034355640411377, + "learning_rate": 2.420796100731113e-06, + "loss": 1.7664, + "step": 1559 + }, + { + "epoch": 1.7095890410958905, + "grad_norm": 2.2018799781799316, + "learning_rate": 2.4187652315190904e-06, + "loss": 1.7328, + "step": 1560 + }, + { + "epoch": 1.7095890410958905, + "eval_loss": 1.9715769290924072, + "eval_runtime": 37.2507, + "eval_samples_per_second": 26.845, + "eval_steps_per_second": 1.127, + "step": 1560 + }, + { + "epoch": 1.7106849315068493, + "grad_norm": 2.396462917327881, + "learning_rate": 2.4167343623070677e-06, + "loss": 1.7773, + "step": 1561 + }, + { + "epoch": 1.7117808219178081, + "grad_norm": 2.3212695121765137, + "learning_rate": 2.414703493095045e-06, + "loss": 1.7966, + "step": 1562 + }, + { + "epoch": 1.7128767123287671, + "grad_norm": 2.4157631397247314, + "learning_rate": 2.4126726238830222e-06, + "loss": 1.7797, + "step": 1563 + }, + { + "epoch": 1.7139726027397262, + "grad_norm": 2.404907464981079, + "learning_rate": 2.410641754670999e-06, + "loss": 1.8093, + "step": 1564 + }, + { + "epoch": 1.7150684931506848, + "grad_norm": 2.3798625469207764, + "learning_rate": 2.4086108854589763e-06, + "loss": 1.7635, + "step": 1565 + }, + { + "epoch": 1.7161643835616438, + "grad_norm": 2.325932025909424, + "learning_rate": 2.4065800162469536e-06, + "loss": 1.8281, + "step": 1566 + }, + { + "epoch": 1.7172602739726028, + "grad_norm": 2.413975238800049, + "learning_rate": 2.404549147034931e-06, + "loss": 1.7609, + "step": 1567 + }, + { + "epoch": 1.7183561643835616, + "grad_norm": 2.512775421142578, + "learning_rate": 2.402518277822908e-06, + "loss": 1.7793, + "step": 1568 + }, + { + "epoch": 1.7194520547945205, + "grad_norm": 2.581679582595825, + "learning_rate": 2.4004874086108854e-06, + "loss": 1.8302, + "step": 1569 + }, + { + "epoch": 1.7205479452054795, + "grad_norm": 2.389822244644165, + "learning_rate": 2.3984565393988627e-06, + "loss": 1.7573, + "step": 1570 + }, + { + "epoch": 1.7205479452054795, + "eval_loss": 1.9670331478118896, + "eval_runtime": 37.2901, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 1570 + }, + { + "epoch": 1.7216438356164385, + "grad_norm": 2.6191556453704834, + "learning_rate": 2.39642567018684e-06, + "loss": 1.8252, + "step": 1571 + }, + { + "epoch": 1.7227397260273971, + "grad_norm": 2.394347906112671, + "learning_rate": 2.3943948009748173e-06, + "loss": 1.7561, + "step": 1572 + }, + { + "epoch": 1.7238356164383561, + "grad_norm": 2.3016326427459717, + "learning_rate": 2.3923639317627945e-06, + "loss": 1.7619, + "step": 1573 + }, + { + "epoch": 1.7249315068493152, + "grad_norm": 2.3663971424102783, + "learning_rate": 2.390333062550772e-06, + "loss": 1.7922, + "step": 1574 + }, + { + "epoch": 1.726027397260274, + "grad_norm": 2.3328938484191895, + "learning_rate": 2.388302193338749e-06, + "loss": 1.7374, + "step": 1575 + }, + { + "epoch": 1.7271232876712328, + "grad_norm": 2.313525676727295, + "learning_rate": 2.3862713241267264e-06, + "loss": 1.6585, + "step": 1576 + }, + { + "epoch": 1.7282191780821918, + "grad_norm": 2.4609642028808594, + "learning_rate": 2.3842404549147037e-06, + "loss": 1.8757, + "step": 1577 + }, + { + "epoch": 1.7293150684931506, + "grad_norm": 2.3502120971679688, + "learning_rate": 2.382209585702681e-06, + "loss": 1.7786, + "step": 1578 + }, + { + "epoch": 1.7304109589041095, + "grad_norm": 2.383025884628296, + "learning_rate": 2.380178716490658e-06, + "loss": 1.7502, + "step": 1579 + }, + { + "epoch": 1.7315068493150685, + "grad_norm": 2.280890464782715, + "learning_rate": 2.3781478472786355e-06, + "loss": 1.8192, + "step": 1580 + }, + { + "epoch": 1.7315068493150685, + "eval_loss": 1.9658632278442383, + "eval_runtime": 37.3714, + "eval_samples_per_second": 26.758, + "eval_steps_per_second": 1.124, + "step": 1580 + }, + { + "epoch": 1.7326027397260275, + "grad_norm": 2.342240333557129, + "learning_rate": 2.3761169780666128e-06, + "loss": 1.7786, + "step": 1581 + }, + { + "epoch": 1.7336986301369863, + "grad_norm": 2.2916715145111084, + "learning_rate": 2.37408610885459e-06, + "loss": 1.8161, + "step": 1582 + }, + { + "epoch": 1.7347945205479451, + "grad_norm": 2.24537992477417, + "learning_rate": 2.3720552396425673e-06, + "loss": 1.8442, + "step": 1583 + }, + { + "epoch": 1.7358904109589042, + "grad_norm": 2.3003411293029785, + "learning_rate": 2.3700243704305446e-06, + "loss": 1.7775, + "step": 1584 + }, + { + "epoch": 1.736986301369863, + "grad_norm": 2.3062400817871094, + "learning_rate": 2.367993501218522e-06, + "loss": 1.8374, + "step": 1585 + }, + { + "epoch": 1.7380821917808218, + "grad_norm": 2.3534083366394043, + "learning_rate": 2.365962632006499e-06, + "loss": 1.7829, + "step": 1586 + }, + { + "epoch": 1.7391780821917808, + "grad_norm": 2.2612221240997314, + "learning_rate": 2.3639317627944764e-06, + "loss": 1.7934, + "step": 1587 + }, + { + "epoch": 1.7402739726027399, + "grad_norm": 2.4197304248809814, + "learning_rate": 2.3619008935824537e-06, + "loss": 1.8226, + "step": 1588 + }, + { + "epoch": 1.7413698630136987, + "grad_norm": 2.435908079147339, + "learning_rate": 2.3598700243704305e-06, + "loss": 1.8148, + "step": 1589 + }, + { + "epoch": 1.7424657534246575, + "grad_norm": 2.3617560863494873, + "learning_rate": 2.357839155158408e-06, + "loss": 1.8085, + "step": 1590 + }, + { + "epoch": 1.7424657534246575, + "eval_loss": 1.9636695384979248, + "eval_runtime": 37.2968, + "eval_samples_per_second": 26.812, + "eval_steps_per_second": 1.126, + "step": 1590 + }, + { + "epoch": 1.7435616438356165, + "grad_norm": 2.295182228088379, + "learning_rate": 2.355808285946385e-06, + "loss": 1.8091, + "step": 1591 + }, + { + "epoch": 1.7446575342465753, + "grad_norm": 2.3347294330596924, + "learning_rate": 2.3537774167343624e-06, + "loss": 1.8086, + "step": 1592 + }, + { + "epoch": 1.7457534246575341, + "grad_norm": 2.4760799407958984, + "learning_rate": 2.3517465475223396e-06, + "loss": 1.8227, + "step": 1593 + }, + { + "epoch": 1.7468493150684932, + "grad_norm": 2.4243907928466797, + "learning_rate": 2.349715678310317e-06, + "loss": 1.7921, + "step": 1594 + }, + { + "epoch": 1.7479452054794522, + "grad_norm": 2.271735668182373, + "learning_rate": 2.347684809098294e-06, + "loss": 1.7987, + "step": 1595 + }, + { + "epoch": 1.749041095890411, + "grad_norm": 2.5412354469299316, + "learning_rate": 2.3456539398862715e-06, + "loss": 1.7558, + "step": 1596 + }, + { + "epoch": 1.7501369863013698, + "grad_norm": 2.464944362640381, + "learning_rate": 2.3436230706742487e-06, + "loss": 1.8087, + "step": 1597 + }, + { + "epoch": 1.7512328767123289, + "grad_norm": 2.4210381507873535, + "learning_rate": 2.341592201462226e-06, + "loss": 1.756, + "step": 1598 + }, + { + "epoch": 1.7523287671232877, + "grad_norm": 2.284945011138916, + "learning_rate": 2.3395613322502033e-06, + "loss": 1.7909, + "step": 1599 + }, + { + "epoch": 1.7534246575342465, + "grad_norm": 2.3740732669830322, + "learning_rate": 2.3375304630381806e-06, + "loss": 1.7683, + "step": 1600 + }, + { + "epoch": 1.7534246575342465, + "eval_loss": 1.9646408557891846, + "eval_runtime": 37.3266, + "eval_samples_per_second": 26.791, + "eval_steps_per_second": 1.125, + "step": 1600 + }, + { + "epoch": 1.7545205479452055, + "grad_norm": 2.399583578109741, + "learning_rate": 2.335499593826158e-06, + "loss": 1.7342, + "step": 1601 + }, + { + "epoch": 1.7556164383561645, + "grad_norm": 2.4731597900390625, + "learning_rate": 2.333468724614135e-06, + "loss": 1.7507, + "step": 1602 + }, + { + "epoch": 1.7567123287671231, + "grad_norm": 2.365433692932129, + "learning_rate": 2.3314378554021124e-06, + "loss": 1.7944, + "step": 1603 + }, + { + "epoch": 1.7578082191780822, + "grad_norm": 2.4500601291656494, + "learning_rate": 2.3294069861900897e-06, + "loss": 1.822, + "step": 1604 + }, + { + "epoch": 1.7589041095890412, + "grad_norm": 2.3339014053344727, + "learning_rate": 2.327376116978067e-06, + "loss": 1.7464, + "step": 1605 + }, + { + "epoch": 1.76, + "grad_norm": 2.343973159790039, + "learning_rate": 2.3253452477660442e-06, + "loss": 1.7408, + "step": 1606 + }, + { + "epoch": 1.7610958904109588, + "grad_norm": 2.453462839126587, + "learning_rate": 2.3233143785540215e-06, + "loss": 1.8261, + "step": 1607 + }, + { + "epoch": 1.7621917808219179, + "grad_norm": 2.357698678970337, + "learning_rate": 2.3212835093419988e-06, + "loss": 1.762, + "step": 1608 + }, + { + "epoch": 1.7632876712328767, + "grad_norm": 2.353257417678833, + "learning_rate": 2.319252640129976e-06, + "loss": 1.8076, + "step": 1609 + }, + { + "epoch": 1.7643835616438355, + "grad_norm": 2.3269948959350586, + "learning_rate": 2.3172217709179533e-06, + "loss": 1.7516, + "step": 1610 + }, + { + "epoch": 1.7643835616438355, + "eval_loss": 1.9678465127944946, + "eval_runtime": 37.2052, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 1610 + }, + { + "epoch": 1.7654794520547945, + "grad_norm": 2.4023501873016357, + "learning_rate": 2.3151909017059306e-06, + "loss": 1.8477, + "step": 1611 + }, + { + "epoch": 1.7665753424657535, + "grad_norm": 2.3356990814208984, + "learning_rate": 2.313160032493908e-06, + "loss": 1.8236, + "step": 1612 + }, + { + "epoch": 1.7676712328767124, + "grad_norm": 2.3571391105651855, + "learning_rate": 2.311129163281885e-06, + "loss": 1.7335, + "step": 1613 + }, + { + "epoch": 1.7687671232876712, + "grad_norm": 2.3510208129882812, + "learning_rate": 2.309098294069862e-06, + "loss": 1.8088, + "step": 1614 + }, + { + "epoch": 1.7698630136986302, + "grad_norm": 2.3852577209472656, + "learning_rate": 2.3070674248578393e-06, + "loss": 1.7312, + "step": 1615 + }, + { + "epoch": 1.770958904109589, + "grad_norm": 2.3085618019104004, + "learning_rate": 2.3050365556458166e-06, + "loss": 1.7379, + "step": 1616 + }, + { + "epoch": 1.7720547945205478, + "grad_norm": 2.3454065322875977, + "learning_rate": 2.303005686433794e-06, + "loss": 1.8054, + "step": 1617 + }, + { + "epoch": 1.7731506849315068, + "grad_norm": 2.4034247398376465, + "learning_rate": 2.300974817221771e-06, + "loss": 1.7955, + "step": 1618 + }, + { + "epoch": 1.7742465753424659, + "grad_norm": 2.4181671142578125, + "learning_rate": 2.2989439480097484e-06, + "loss": 1.7732, + "step": 1619 + }, + { + "epoch": 1.7753424657534247, + "grad_norm": 2.3408010005950928, + "learning_rate": 2.2969130787977257e-06, + "loss": 1.7231, + "step": 1620 + }, + { + "epoch": 1.7753424657534247, + "eval_loss": 1.9655163288116455, + "eval_runtime": 37.4878, + "eval_samples_per_second": 26.675, + "eval_steps_per_second": 1.12, + "step": 1620 + }, + { + "epoch": 1.7764383561643835, + "grad_norm": 2.3113718032836914, + "learning_rate": 2.294882209585703e-06, + "loss": 1.7904, + "step": 1621 + }, + { + "epoch": 1.7775342465753425, + "grad_norm": 2.3037896156311035, + "learning_rate": 2.2928513403736802e-06, + "loss": 1.8069, + "step": 1622 + }, + { + "epoch": 1.7786301369863013, + "grad_norm": 2.3962693214416504, + "learning_rate": 2.2908204711616575e-06, + "loss": 1.7354, + "step": 1623 + }, + { + "epoch": 1.7797260273972602, + "grad_norm": 2.3499252796173096, + "learning_rate": 2.2887896019496348e-06, + "loss": 1.779, + "step": 1624 + }, + { + "epoch": 1.7808219178082192, + "grad_norm": 2.390958070755005, + "learning_rate": 2.286758732737612e-06, + "loss": 1.8257, + "step": 1625 + }, + { + "epoch": 1.7819178082191782, + "grad_norm": 2.365626573562622, + "learning_rate": 2.284727863525589e-06, + "loss": 1.7618, + "step": 1626 + }, + { + "epoch": 1.783013698630137, + "grad_norm": 2.3558530807495117, + "learning_rate": 2.282696994313566e-06, + "loss": 1.7729, + "step": 1627 + }, + { + "epoch": 1.7841095890410958, + "grad_norm": 2.3276045322418213, + "learning_rate": 2.2806661251015434e-06, + "loss": 1.8231, + "step": 1628 + }, + { + "epoch": 1.7852054794520549, + "grad_norm": 2.3369953632354736, + "learning_rate": 2.2786352558895207e-06, + "loss": 1.8011, + "step": 1629 + }, + { + "epoch": 1.7863013698630137, + "grad_norm": 2.3588311672210693, + "learning_rate": 2.276604386677498e-06, + "loss": 1.7682, + "step": 1630 + }, + { + "epoch": 1.7863013698630137, + "eval_loss": 1.963529348373413, + "eval_runtime": 37.5589, + "eval_samples_per_second": 26.625, + "eval_steps_per_second": 1.118, + "step": 1630 + }, + { + "epoch": 1.7873972602739725, + "grad_norm": 2.2918803691864014, + "learning_rate": 2.2745735174654753e-06, + "loss": 1.7256, + "step": 1631 + }, + { + "epoch": 1.7884931506849315, + "grad_norm": 2.423738479614258, + "learning_rate": 2.2725426482534526e-06, + "loss": 1.7582, + "step": 1632 + }, + { + "epoch": 1.7895890410958906, + "grad_norm": 2.3500850200653076, + "learning_rate": 2.27051177904143e-06, + "loss": 1.8255, + "step": 1633 + }, + { + "epoch": 1.7906849315068492, + "grad_norm": 2.3280837535858154, + "learning_rate": 2.268480909829407e-06, + "loss": 1.7845, + "step": 1634 + }, + { + "epoch": 1.7917808219178082, + "grad_norm": 2.5055465698242188, + "learning_rate": 2.2664500406173844e-06, + "loss": 1.78, + "step": 1635 + }, + { + "epoch": 1.7928767123287672, + "grad_norm": 2.3132143020629883, + "learning_rate": 2.2644191714053617e-06, + "loss": 1.7814, + "step": 1636 + }, + { + "epoch": 1.793972602739726, + "grad_norm": 2.316361427307129, + "learning_rate": 2.262388302193339e-06, + "loss": 1.8161, + "step": 1637 + }, + { + "epoch": 1.7950684931506848, + "grad_norm": 2.337318181991577, + "learning_rate": 2.260357432981316e-06, + "loss": 1.7942, + "step": 1638 + }, + { + "epoch": 1.7961643835616439, + "grad_norm": 2.362031936645508, + "learning_rate": 2.2583265637692935e-06, + "loss": 1.7771, + "step": 1639 + }, + { + "epoch": 1.7972602739726027, + "grad_norm": 2.3260183334350586, + "learning_rate": 2.2562956945572708e-06, + "loss": 1.7975, + "step": 1640 + }, + { + "epoch": 1.7972602739726027, + "eval_loss": 1.9619382619857788, + "eval_runtime": 37.2606, + "eval_samples_per_second": 26.838, + "eval_steps_per_second": 1.127, + "step": 1640 + }, + { + "epoch": 1.7983561643835615, + "grad_norm": 2.4260714054107666, + "learning_rate": 2.254264825345248e-06, + "loss": 1.7654, + "step": 1641 + }, + { + "epoch": 1.7994520547945205, + "grad_norm": 2.297428846359253, + "learning_rate": 2.2522339561332253e-06, + "loss": 1.775, + "step": 1642 + }, + { + "epoch": 1.8005479452054796, + "grad_norm": 2.5153353214263916, + "learning_rate": 2.2502030869212026e-06, + "loss": 1.7796, + "step": 1643 + }, + { + "epoch": 1.8016438356164384, + "grad_norm": 2.5102577209472656, + "learning_rate": 2.24817221770918e-06, + "loss": 1.7917, + "step": 1644 + }, + { + "epoch": 1.8027397260273972, + "grad_norm": 2.3693106174468994, + "learning_rate": 2.246141348497157e-06, + "loss": 1.8459, + "step": 1645 + }, + { + "epoch": 1.8038356164383562, + "grad_norm": 2.2353153228759766, + "learning_rate": 2.2441104792851344e-06, + "loss": 1.8095, + "step": 1646 + }, + { + "epoch": 1.804931506849315, + "grad_norm": 2.265465497970581, + "learning_rate": 2.2420796100731117e-06, + "loss": 1.7623, + "step": 1647 + }, + { + "epoch": 1.8060273972602738, + "grad_norm": 2.2820940017700195, + "learning_rate": 2.240048740861089e-06, + "loss": 1.7946, + "step": 1648 + }, + { + "epoch": 1.8071232876712329, + "grad_norm": 2.378035068511963, + "learning_rate": 2.2380178716490662e-06, + "loss": 1.8059, + "step": 1649 + }, + { + "epoch": 1.808219178082192, + "grad_norm": 2.328341245651245, + "learning_rate": 2.2359870024370435e-06, + "loss": 1.6611, + "step": 1650 + }, + { + "epoch": 1.808219178082192, + "eval_loss": 1.9600144624710083, + "eval_runtime": 37.2601, + "eval_samples_per_second": 26.838, + "eval_steps_per_second": 1.127, + "step": 1650 + }, + { + "epoch": 1.8093150684931507, + "grad_norm": 2.2819113731384277, + "learning_rate": 2.2339561332250204e-06, + "loss": 1.8058, + "step": 1651 + }, + { + "epoch": 1.8104109589041095, + "grad_norm": 2.4178121089935303, + "learning_rate": 2.2319252640129976e-06, + "loss": 1.8204, + "step": 1652 + }, + { + "epoch": 1.8115068493150686, + "grad_norm": 2.497008800506592, + "learning_rate": 2.229894394800975e-06, + "loss": 1.7968, + "step": 1653 + }, + { + "epoch": 1.8126027397260274, + "grad_norm": 2.4681501388549805, + "learning_rate": 2.227863525588952e-06, + "loss": 1.7292, + "step": 1654 + }, + { + "epoch": 1.8136986301369862, + "grad_norm": 2.2983527183532715, + "learning_rate": 2.2258326563769295e-06, + "loss": 1.7097, + "step": 1655 + }, + { + "epoch": 1.8147945205479452, + "grad_norm": 2.349454164505005, + "learning_rate": 2.2238017871649067e-06, + "loss": 1.7256, + "step": 1656 + }, + { + "epoch": 1.8158904109589042, + "grad_norm": 2.4247677326202393, + "learning_rate": 2.221770917952884e-06, + "loss": 1.7165, + "step": 1657 + }, + { + "epoch": 1.816986301369863, + "grad_norm": 2.3035342693328857, + "learning_rate": 2.2197400487408613e-06, + "loss": 1.7811, + "step": 1658 + }, + { + "epoch": 1.8180821917808219, + "grad_norm": 2.318772554397583, + "learning_rate": 2.2177091795288386e-06, + "loss": 1.8082, + "step": 1659 + }, + { + "epoch": 1.819178082191781, + "grad_norm": 2.2477188110351562, + "learning_rate": 2.215678310316816e-06, + "loss": 1.7206, + "step": 1660 + }, + { + "epoch": 1.819178082191781, + "eval_loss": 1.962982416152954, + "eval_runtime": 37.2709, + "eval_samples_per_second": 26.831, + "eval_steps_per_second": 1.127, + "step": 1660 + }, + { + "epoch": 1.8202739726027397, + "grad_norm": 2.355264186859131, + "learning_rate": 2.213647441104793e-06, + "loss": 1.7634, + "step": 1661 + }, + { + "epoch": 1.8213698630136985, + "grad_norm": 2.267458200454712, + "learning_rate": 2.2116165718927704e-06, + "loss": 1.7819, + "step": 1662 + }, + { + "epoch": 1.8224657534246576, + "grad_norm": 2.2487895488739014, + "learning_rate": 2.2095857026807473e-06, + "loss": 1.759, + "step": 1663 + }, + { + "epoch": 1.8235616438356166, + "grad_norm": 2.4253578186035156, + "learning_rate": 2.2075548334687245e-06, + "loss": 1.7937, + "step": 1664 + }, + { + "epoch": 1.8246575342465754, + "grad_norm": 2.478050470352173, + "learning_rate": 2.205523964256702e-06, + "loss": 1.7241, + "step": 1665 + }, + { + "epoch": 1.8257534246575342, + "grad_norm": 2.289905309677124, + "learning_rate": 2.203493095044679e-06, + "loss": 1.7531, + "step": 1666 + }, + { + "epoch": 1.8268493150684932, + "grad_norm": 2.303421974182129, + "learning_rate": 2.2014622258326564e-06, + "loss": 1.7711, + "step": 1667 + }, + { + "epoch": 1.827945205479452, + "grad_norm": 2.3670248985290527, + "learning_rate": 2.1994313566206336e-06, + "loss": 1.8093, + "step": 1668 + }, + { + "epoch": 1.8290410958904109, + "grad_norm": 2.3721048831939697, + "learning_rate": 2.197400487408611e-06, + "loss": 1.8273, + "step": 1669 + }, + { + "epoch": 1.83013698630137, + "grad_norm": 2.3000190258026123, + "learning_rate": 2.195369618196588e-06, + "loss": 1.75, + "step": 1670 + }, + { + "epoch": 1.83013698630137, + "eval_loss": 1.9577263593673706, + "eval_runtime": 37.2953, + "eval_samples_per_second": 26.813, + "eval_steps_per_second": 1.126, + "step": 1670 + }, + { + "epoch": 1.831232876712329, + "grad_norm": 2.303584575653076, + "learning_rate": 2.1933387489845655e-06, + "loss": 1.7278, + "step": 1671 + }, + { + "epoch": 1.8323287671232875, + "grad_norm": 2.260579824447632, + "learning_rate": 2.1913078797725427e-06, + "loss": 1.7524, + "step": 1672 + }, + { + "epoch": 1.8334246575342465, + "grad_norm": 2.311967372894287, + "learning_rate": 2.18927701056052e-06, + "loss": 1.7799, + "step": 1673 + }, + { + "epoch": 1.8345205479452056, + "grad_norm": 2.338989496231079, + "learning_rate": 2.1872461413484973e-06, + "loss": 1.8468, + "step": 1674 + }, + { + "epoch": 1.8356164383561644, + "grad_norm": 2.338989496231079, + "learning_rate": 2.1872461413484973e-06, + "loss": 1.7853, + "step": 1675 + }, + { + "epoch": 1.8367123287671232, + "grad_norm": 2.526724100112915, + "learning_rate": 2.1852152721364746e-06, + "loss": 1.78, + "step": 1676 + }, + { + "epoch": 1.8378082191780822, + "grad_norm": 2.4637107849121094, + "learning_rate": 2.183184402924452e-06, + "loss": 1.7328, + "step": 1677 + }, + { + "epoch": 1.838904109589041, + "grad_norm": 2.34454345703125, + "learning_rate": 2.181153533712429e-06, + "loss": 1.8058, + "step": 1678 + }, + { + "epoch": 1.8399999999999999, + "grad_norm": 2.5518760681152344, + "learning_rate": 2.1791226645004064e-06, + "loss": 1.7746, + "step": 1679 + }, + { + "epoch": 1.841095890410959, + "grad_norm": 2.437443971633911, + "learning_rate": 2.1770917952883837e-06, + "loss": 1.7517, + "step": 1680 + }, + { + "epoch": 1.841095890410959, + "eval_loss": 1.9572285413742065, + "eval_runtime": 37.3402, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 1680 + }, + { + "epoch": 1.842191780821918, + "grad_norm": 2.344287157058716, + "learning_rate": 2.175060926076361e-06, + "loss": 1.8229, + "step": 1681 + }, + { + "epoch": 1.8432876712328767, + "grad_norm": 2.333840847015381, + "learning_rate": 2.1730300568643382e-06, + "loss": 1.8534, + "step": 1682 + }, + { + "epoch": 1.8443835616438355, + "grad_norm": 2.372926950454712, + "learning_rate": 2.1709991876523155e-06, + "loss": 1.7609, + "step": 1683 + }, + { + "epoch": 1.8454794520547946, + "grad_norm": 2.2858645915985107, + "learning_rate": 2.1689683184402928e-06, + "loss": 1.7508, + "step": 1684 + }, + { + "epoch": 1.8465753424657534, + "grad_norm": 2.326586961746216, + "learning_rate": 2.16693744922827e-06, + "loss": 1.8122, + "step": 1685 + }, + { + "epoch": 1.8476712328767122, + "grad_norm": 2.3605616092681885, + "learning_rate": 2.1649065800162473e-06, + "loss": 1.7794, + "step": 1686 + }, + { + "epoch": 1.8487671232876712, + "grad_norm": 2.4188039302825928, + "learning_rate": 2.1628757108042246e-06, + "loss": 1.7552, + "step": 1687 + }, + { + "epoch": 1.8498630136986303, + "grad_norm": 2.335617780685425, + "learning_rate": 2.160844841592202e-06, + "loss": 1.7259, + "step": 1688 + }, + { + "epoch": 1.850958904109589, + "grad_norm": 2.245882272720337, + "learning_rate": 2.1588139723801787e-06, + "loss": 1.7259, + "step": 1689 + }, + { + "epoch": 1.8520547945205479, + "grad_norm": 2.4447500705718994, + "learning_rate": 2.156783103168156e-06, + "loss": 1.7605, + "step": 1690 + }, + { + "epoch": 1.8520547945205479, + "eval_loss": 1.9608982801437378, + "eval_runtime": 37.3756, + "eval_samples_per_second": 26.755, + "eval_steps_per_second": 1.124, + "step": 1690 + }, + { + "epoch": 1.853150684931507, + "grad_norm": 2.363443613052368, + "learning_rate": 2.1547522339561333e-06, + "loss": 1.7336, + "step": 1691 + }, + { + "epoch": 1.8542465753424657, + "grad_norm": 2.3573989868164062, + "learning_rate": 2.1527213647441106e-06, + "loss": 1.7953, + "step": 1692 + }, + { + "epoch": 1.8553424657534245, + "grad_norm": 2.4292593002319336, + "learning_rate": 2.150690495532088e-06, + "loss": 1.8276, + "step": 1693 + }, + { + "epoch": 1.8564383561643836, + "grad_norm": 2.374563694000244, + "learning_rate": 2.148659626320065e-06, + "loss": 1.7975, + "step": 1694 + }, + { + "epoch": 1.8575342465753426, + "grad_norm": 2.3935306072235107, + "learning_rate": 2.1466287571080424e-06, + "loss": 1.7824, + "step": 1695 + }, + { + "epoch": 1.8586301369863014, + "grad_norm": 2.343773365020752, + "learning_rate": 2.1445978878960197e-06, + "loss": 1.7574, + "step": 1696 + }, + { + "epoch": 1.8597260273972602, + "grad_norm": 2.2765426635742188, + "learning_rate": 2.142567018683997e-06, + "loss": 1.7682, + "step": 1697 + }, + { + "epoch": 1.8608219178082193, + "grad_norm": 2.339470863342285, + "learning_rate": 2.140536149471974e-06, + "loss": 1.8229, + "step": 1698 + }, + { + "epoch": 1.861917808219178, + "grad_norm": 2.341775894165039, + "learning_rate": 2.1385052802599515e-06, + "loss": 1.7421, + "step": 1699 + }, + { + "epoch": 1.8630136986301369, + "grad_norm": 2.343876838684082, + "learning_rate": 2.1364744110479288e-06, + "loss": 1.7519, + "step": 1700 + }, + { + "epoch": 1.8630136986301369, + "eval_loss": 1.9587067365646362, + "eval_runtime": 37.4192, + "eval_samples_per_second": 26.724, + "eval_steps_per_second": 1.122, + "step": 1700 + } + ], + "logging_steps": 1, + "max_steps": 2736, + "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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-1700/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb0f9a53a507ffcdb60410deb71d60ef801bf350 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/generation_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/generation_config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c389b841a9a29ddd904e02d1eb0e08dcce82ad9 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/latest b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/latest new file mode 100644 index 0000000000000000000000000000000000000000..4b10acccf3e8395339ff8799cea202bbc54d7f7d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/latest @@ -0,0 +1 @@ +global_step900 \ No newline at end of file diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/model.safetensors.index.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/model.safetensors.index.json new file mode 100644 index 0000000000000000000000000000000000000000..ed64de846d720b9a7859dc20575fea8e8ca51940 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/special_tokens_map.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/special_tokens_map.json new file mode 100644 index 0000000000000000000000000000000000000000..6949c5975ee0e961ef61cf31010dce04df0a03f8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/tokenizer_config.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/tokenizer_config.json new file mode 100644 index 0000000000000000000000000000000000000000..5684b14e2756b56aa397c1324e239b442162b34a --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/tokenizer_config.json @@ -0,0 +1,2070 @@ +{ + "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 + } + }, + "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/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/trainer_state.json b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/trainer_state.json new file mode 100644 index 0000000000000000000000000000000000000000..e4e356dcb090a84e62cf07a36969e5352cac9f85 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/trainer_state.json @@ -0,0 +1,7053 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.9863013698630136, + "eval_steps": 10, + "global_step": 900, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.001095890410958904, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0476, + "step": 1 + }, + { + "epoch": 0.002191780821917808, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0289, + "step": 2 + }, + { + "epoch": 0.0032876712328767125, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9935, + "step": 3 + }, + { + "epoch": 0.004383561643835616, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0139, + "step": 4 + }, + { + "epoch": 0.005479452054794521, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9892, + "step": 5 + }, + { + "epoch": 0.006575342465753425, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.1252, + "step": 6 + }, + { + "epoch": 0.007671232876712329, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0291, + "step": 7 + }, + { + "epoch": 0.008767123287671232, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0607, + "step": 8 + }, + { + "epoch": 0.009863013698630137, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0306, + "step": 9 + }, + { + "epoch": 0.010958904109589041, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0811, + "step": 10 + }, + { + "epoch": 0.010958904109589041, + "eval_loss": 3.0768930912017822, + "eval_runtime": 38.0377, + "eval_samples_per_second": 26.29, + "eval_steps_per_second": 1.104, + "step": 10 + }, + { + "epoch": 0.012054794520547946, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0693, + "step": 11 + }, + { + "epoch": 0.01315068493150685, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 2.9668, + "step": 12 + }, + { + "epoch": 0.014246575342465753, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.052, + "step": 13 + }, + { + "epoch": 0.015342465753424657, + "grad_norm": 0.0, + "learning_rate": 0.0, + "loss": 3.0363, + "step": 14 + }, + { + "epoch": 0.01643835616438356, + "grad_norm": 5.58175802230835, + "learning_rate": 1.8248175182481753e-08, + "loss": 3.0778, + "step": 15 + }, + { + "epoch": 0.017534246575342465, + "grad_norm": 6.360531806945801, + "learning_rate": 3.6496350364963505e-08, + "loss": 3.0257, + "step": 16 + }, + { + "epoch": 0.01863013698630137, + "grad_norm": 6.453113555908203, + "learning_rate": 5.4744525547445264e-08, + "loss": 3.0262, + "step": 17 + }, + { + "epoch": 0.019726027397260273, + "grad_norm": 6.789405345916748, + "learning_rate": 7.299270072992701e-08, + "loss": 3.0671, + "step": 18 + }, + { + "epoch": 0.020821917808219178, + "grad_norm": 6.185514450073242, + "learning_rate": 9.124087591240876e-08, + "loss": 3.0077, + "step": 19 + }, + { + "epoch": 0.021917808219178082, + "grad_norm": 5.719798564910889, + "learning_rate": 1.0948905109489053e-07, + "loss": 3.089, + "step": 20 + }, + { + "epoch": 0.021917808219178082, + "eval_loss": 3.07546329498291, + "eval_runtime": 37.5318, + "eval_samples_per_second": 26.644, + "eval_steps_per_second": 1.119, + "step": 20 + }, + { + "epoch": 0.023013698630136987, + "grad_norm": 6.307768821716309, + "learning_rate": 1.2773722627737227e-07, + "loss": 3.0641, + "step": 21 + }, + { + "epoch": 0.02410958904109589, + "grad_norm": 6.255112171173096, + "learning_rate": 1.4598540145985402e-07, + "loss": 2.9849, + "step": 22 + }, + { + "epoch": 0.025205479452054796, + "grad_norm": 5.812101364135742, + "learning_rate": 1.642335766423358e-07, + "loss": 3.0211, + "step": 23 + }, + { + "epoch": 0.0263013698630137, + "grad_norm": 6.262075901031494, + "learning_rate": 1.8248175182481753e-07, + "loss": 3.0654, + "step": 24 + }, + { + "epoch": 0.0273972602739726, + "grad_norm": 5.419229507446289, + "learning_rate": 2.0072992700729928e-07, + "loss": 3.0661, + "step": 25 + }, + { + "epoch": 0.028493150684931506, + "grad_norm": 4.750737190246582, + "learning_rate": 2.1897810218978106e-07, + "loss": 3.0494, + "step": 26 + }, + { + "epoch": 0.02958904109589041, + "grad_norm": 4.732361316680908, + "learning_rate": 2.3722627737226278e-07, + "loss": 3.0471, + "step": 27 + }, + { + "epoch": 0.030684931506849315, + "grad_norm": 4.749786853790283, + "learning_rate": 2.5547445255474454e-07, + "loss": 3.0107, + "step": 28 + }, + { + "epoch": 0.03178082191780822, + "grad_norm": 4.451244831085205, + "learning_rate": 2.737226277372263e-07, + "loss": 2.9987, + "step": 29 + }, + { + "epoch": 0.03287671232876712, + "grad_norm": 4.5158257484436035, + "learning_rate": 2.9197080291970804e-07, + "loss": 2.9686, + "step": 30 + }, + { + "epoch": 0.03287671232876712, + "eval_loss": 3.031872034072876, + "eval_runtime": 37.2608, + "eval_samples_per_second": 26.838, + "eval_steps_per_second": 1.127, + "step": 30 + }, + { + "epoch": 0.03397260273972603, + "grad_norm": 3.8239498138427734, + "learning_rate": 3.102189781021898e-07, + "loss": 3.0057, + "step": 31 + }, + { + "epoch": 0.03506849315068493, + "grad_norm": 4.037254810333252, + "learning_rate": 3.284671532846716e-07, + "loss": 2.9733, + "step": 32 + }, + { + "epoch": 0.03616438356164384, + "grad_norm": 4.283931732177734, + "learning_rate": 3.467153284671533e-07, + "loss": 2.9515, + "step": 33 + }, + { + "epoch": 0.03726027397260274, + "grad_norm": 4.1784987449646, + "learning_rate": 3.6496350364963505e-07, + "loss": 2.9483, + "step": 34 + }, + { + "epoch": 0.038356164383561646, + "grad_norm": 3.9210753440856934, + "learning_rate": 3.8321167883211686e-07, + "loss": 2.9679, + "step": 35 + }, + { + "epoch": 0.03945205479452055, + "grad_norm": 3.72884202003479, + "learning_rate": 4.0145985401459856e-07, + "loss": 2.9989, + "step": 36 + }, + { + "epoch": 0.040547945205479455, + "grad_norm": 3.2882330417633057, + "learning_rate": 4.197080291970803e-07, + "loss": 2.9384, + "step": 37 + }, + { + "epoch": 0.041643835616438356, + "grad_norm": 3.515293836593628, + "learning_rate": 4.379562043795621e-07, + "loss": 2.8986, + "step": 38 + }, + { + "epoch": 0.042739726027397264, + "grad_norm": 3.568453073501587, + "learning_rate": 4.562043795620438e-07, + "loss": 2.9854, + "step": 39 + }, + { + "epoch": 0.043835616438356165, + "grad_norm": 3.389789342880249, + "learning_rate": 4.7445255474452557e-07, + "loss": 2.8891, + "step": 40 + }, + { + "epoch": 0.043835616438356165, + "eval_loss": 2.964235544204712, + "eval_runtime": 37.3406, + "eval_samples_per_second": 26.781, + "eval_steps_per_second": 1.125, + "step": 40 + }, + { + "epoch": 0.044931506849315066, + "grad_norm": 3.1826283931732178, + "learning_rate": 4.927007299270073e-07, + "loss": 2.9271, + "step": 41 + }, + { + "epoch": 0.046027397260273974, + "grad_norm": 2.941539764404297, + "learning_rate": 5.109489051094891e-07, + "loss": 2.9486, + "step": 42 + }, + { + "epoch": 0.047123287671232875, + "grad_norm": 3.0076773166656494, + "learning_rate": 5.291970802919708e-07, + "loss": 2.9252, + "step": 43 + }, + { + "epoch": 0.04821917808219178, + "grad_norm": 2.79028582572937, + "learning_rate": 5.474452554744526e-07, + "loss": 2.9277, + "step": 44 + }, + { + "epoch": 0.049315068493150684, + "grad_norm": 2.603456735610962, + "learning_rate": 5.656934306569343e-07, + "loss": 2.9144, + "step": 45 + }, + { + "epoch": 0.05041095890410959, + "grad_norm": 2.674888849258423, + "learning_rate": 5.839416058394161e-07, + "loss": 2.8732, + "step": 46 + }, + { + "epoch": 0.05150684931506849, + "grad_norm": 2.972093343734741, + "learning_rate": 6.021897810218979e-07, + "loss": 2.8623, + "step": 47 + }, + { + "epoch": 0.0526027397260274, + "grad_norm": 2.757766008377075, + "learning_rate": 6.204379562043796e-07, + "loss": 2.8941, + "step": 48 + }, + { + "epoch": 0.0536986301369863, + "grad_norm": 2.910634994506836, + "learning_rate": 6.386861313868614e-07, + "loss": 2.8101, + "step": 49 + }, + { + "epoch": 0.0547945205479452, + "grad_norm": 2.5542638301849365, + "learning_rate": 6.569343065693432e-07, + "loss": 2.9262, + "step": 50 + }, + { + "epoch": 0.0547945205479452, + "eval_loss": 2.883579969406128, + "eval_runtime": 37.326, + "eval_samples_per_second": 26.791, + "eval_steps_per_second": 1.125, + "step": 50 + }, + { + "epoch": 0.05589041095890411, + "grad_norm": 2.333974599838257, + "learning_rate": 6.751824817518248e-07, + "loss": 2.8217, + "step": 51 + }, + { + "epoch": 0.05698630136986301, + "grad_norm": 2.181022882461548, + "learning_rate": 6.934306569343066e-07, + "loss": 2.8483, + "step": 52 + }, + { + "epoch": 0.05808219178082192, + "grad_norm": 2.12200927734375, + "learning_rate": 7.116788321167883e-07, + "loss": 2.7788, + "step": 53 + }, + { + "epoch": 0.05917808219178082, + "grad_norm": 2.052626371383667, + "learning_rate": 7.299270072992701e-07, + "loss": 2.7774, + "step": 54 + }, + { + "epoch": 0.06027397260273973, + "grad_norm": 2.11487078666687, + "learning_rate": 7.48175182481752e-07, + "loss": 2.8045, + "step": 55 + }, + { + "epoch": 0.06136986301369863, + "grad_norm": 2.083588123321533, + "learning_rate": 7.664233576642337e-07, + "loss": 2.8119, + "step": 56 + }, + { + "epoch": 0.06246575342465754, + "grad_norm": 2.2079808712005615, + "learning_rate": 7.846715328467154e-07, + "loss": 2.82, + "step": 57 + }, + { + "epoch": 0.06356164383561644, + "grad_norm": 2.100802421569824, + "learning_rate": 8.029197080291971e-07, + "loss": 2.794, + "step": 58 + }, + { + "epoch": 0.06465753424657535, + "grad_norm": 1.8322066068649292, + "learning_rate": 8.211678832116789e-07, + "loss": 2.7444, + "step": 59 + }, + { + "epoch": 0.06575342465753424, + "grad_norm": 2.077005386352539, + "learning_rate": 8.394160583941606e-07, + "loss": 2.7093, + "step": 60 + }, + { + "epoch": 0.06575342465753424, + "eval_loss": 2.8177874088287354, + "eval_runtime": 37.2903, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 60 + }, + { + "epoch": 0.06684931506849315, + "grad_norm": 2.1688592433929443, + "learning_rate": 8.576642335766425e-07, + "loss": 2.7446, + "step": 61 + }, + { + "epoch": 0.06794520547945206, + "grad_norm": 2.1709237098693848, + "learning_rate": 8.759124087591242e-07, + "loss": 2.736, + "step": 62 + }, + { + "epoch": 0.06904109589041096, + "grad_norm": 2.3578364849090576, + "learning_rate": 8.94160583941606e-07, + "loss": 2.7583, + "step": 63 + }, + { + "epoch": 0.07013698630136986, + "grad_norm": 1.9025483131408691, + "learning_rate": 9.124087591240876e-07, + "loss": 2.7651, + "step": 64 + }, + { + "epoch": 0.07123287671232877, + "grad_norm": 1.9450963735580444, + "learning_rate": 9.306569343065694e-07, + "loss": 2.7186, + "step": 65 + }, + { + "epoch": 0.07232876712328767, + "grad_norm": 1.957723617553711, + "learning_rate": 9.489051094890511e-07, + "loss": 2.7421, + "step": 66 + }, + { + "epoch": 0.07342465753424658, + "grad_norm": 1.9823631048202515, + "learning_rate": 9.67153284671533e-07, + "loss": 2.7258, + "step": 67 + }, + { + "epoch": 0.07452054794520548, + "grad_norm": 1.945495843887329, + "learning_rate": 9.854014598540146e-07, + "loss": 2.7733, + "step": 68 + }, + { + "epoch": 0.07561643835616438, + "grad_norm": 1.9684759378433228, + "learning_rate": 1.0036496350364965e-06, + "loss": 2.7041, + "step": 69 + }, + { + "epoch": 0.07671232876712329, + "grad_norm": 2.087336540222168, + "learning_rate": 1.0218978102189781e-06, + "loss": 2.6662, + "step": 70 + }, + { + "epoch": 0.07671232876712329, + "eval_loss": 2.7529091835021973, + "eval_runtime": 37.3122, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 70 + }, + { + "epoch": 0.07780821917808219, + "grad_norm": 1.7790117263793945, + "learning_rate": 1.04014598540146e-06, + "loss": 2.7379, + "step": 71 + }, + { + "epoch": 0.0789041095890411, + "grad_norm": 2.2780861854553223, + "learning_rate": 1.0583941605839416e-06, + "loss": 2.6234, + "step": 72 + }, + { + "epoch": 0.08, + "grad_norm": 2.1144700050354004, + "learning_rate": 1.0766423357664233e-06, + "loss": 2.6248, + "step": 73 + }, + { + "epoch": 0.08109589041095891, + "grad_norm": 2.0247490406036377, + "learning_rate": 1.0948905109489052e-06, + "loss": 2.6456, + "step": 74 + }, + { + "epoch": 0.0821917808219178, + "grad_norm": 2.702547311782837, + "learning_rate": 1.113138686131387e-06, + "loss": 2.6759, + "step": 75 + }, + { + "epoch": 0.08328767123287671, + "grad_norm": 1.8208750486373901, + "learning_rate": 1.1313868613138687e-06, + "loss": 2.6845, + "step": 76 + }, + { + "epoch": 0.08438356164383562, + "grad_norm": 2.3829221725463867, + "learning_rate": 1.1496350364963505e-06, + "loss": 2.61, + "step": 77 + }, + { + "epoch": 0.08547945205479453, + "grad_norm": 2.580885887145996, + "learning_rate": 1.1678832116788322e-06, + "loss": 2.5896, + "step": 78 + }, + { + "epoch": 0.08657534246575342, + "grad_norm": 2.2477641105651855, + "learning_rate": 1.1861313868613138e-06, + "loss": 2.596, + "step": 79 + }, + { + "epoch": 0.08767123287671233, + "grad_norm": 2.1895623207092285, + "learning_rate": 1.2043795620437959e-06, + "loss": 2.5906, + "step": 80 + }, + { + "epoch": 0.08767123287671233, + "eval_loss": 2.6815176010131836, + "eval_runtime": 37.3342, + "eval_samples_per_second": 26.785, + "eval_steps_per_second": 1.125, + "step": 80 + }, + { + "epoch": 0.08876712328767124, + "grad_norm": 2.217090606689453, + "learning_rate": 1.2226277372262775e-06, + "loss": 2.6005, + "step": 81 + }, + { + "epoch": 0.08986301369863013, + "grad_norm": 2.4679362773895264, + "learning_rate": 1.2408759124087592e-06, + "loss": 2.6333, + "step": 82 + }, + { + "epoch": 0.09095890410958904, + "grad_norm": 2.2590575218200684, + "learning_rate": 1.259124087591241e-06, + "loss": 2.5814, + "step": 83 + }, + { + "epoch": 0.09205479452054795, + "grad_norm": 1.877977728843689, + "learning_rate": 1.2773722627737229e-06, + "loss": 2.5948, + "step": 84 + }, + { + "epoch": 0.09315068493150686, + "grad_norm": 2.1535534858703613, + "learning_rate": 1.2956204379562045e-06, + "loss": 2.5781, + "step": 85 + }, + { + "epoch": 0.09424657534246575, + "grad_norm": 2.5001144409179688, + "learning_rate": 1.3138686131386864e-06, + "loss": 2.5339, + "step": 86 + }, + { + "epoch": 0.09534246575342466, + "grad_norm": 2.3279049396514893, + "learning_rate": 1.332116788321168e-06, + "loss": 2.545, + "step": 87 + }, + { + "epoch": 0.09643835616438357, + "grad_norm": 2.2230734825134277, + "learning_rate": 1.3503649635036497e-06, + "loss": 2.4745, + "step": 88 + }, + { + "epoch": 0.09753424657534246, + "grad_norm": 2.8762636184692383, + "learning_rate": 1.3686131386861315e-06, + "loss": 2.516, + "step": 89 + }, + { + "epoch": 0.09863013698630137, + "grad_norm": 2.153506278991699, + "learning_rate": 1.3868613138686132e-06, + "loss": 2.4547, + "step": 90 + }, + { + "epoch": 0.09863013698630137, + "eval_loss": 2.603238344192505, + "eval_runtime": 37.4162, + "eval_samples_per_second": 26.726, + "eval_steps_per_second": 1.123, + "step": 90 + }, + { + "epoch": 0.09972602739726028, + "grad_norm": 2.4143614768981934, + "learning_rate": 1.405109489051095e-06, + "loss": 2.547, + "step": 91 + }, + { + "epoch": 0.10082191780821918, + "grad_norm": 2.354632616043091, + "learning_rate": 1.4233576642335767e-06, + "loss": 2.5009, + "step": 92 + }, + { + "epoch": 0.10191780821917808, + "grad_norm": 2.6605563163757324, + "learning_rate": 1.4416058394160583e-06, + "loss": 2.5304, + "step": 93 + }, + { + "epoch": 0.10301369863013699, + "grad_norm": 2.3724122047424316, + "learning_rate": 1.4598540145985402e-06, + "loss": 2.4925, + "step": 94 + }, + { + "epoch": 0.10410958904109589, + "grad_norm": 3.6323022842407227, + "learning_rate": 1.478102189781022e-06, + "loss": 2.4592, + "step": 95 + }, + { + "epoch": 0.1052054794520548, + "grad_norm": 2.3407089710235596, + "learning_rate": 1.496350364963504e-06, + "loss": 2.5223, + "step": 96 + }, + { + "epoch": 0.1063013698630137, + "grad_norm": 2.8238413333892822, + "learning_rate": 1.5145985401459856e-06, + "loss": 2.489, + "step": 97 + }, + { + "epoch": 0.1073972602739726, + "grad_norm": 2.6433658599853516, + "learning_rate": 1.5328467153284674e-06, + "loss": 2.5101, + "step": 98 + }, + { + "epoch": 0.10849315068493151, + "grad_norm": 2.562761068344116, + "learning_rate": 1.551094890510949e-06, + "loss": 2.4642, + "step": 99 + }, + { + "epoch": 0.1095890410958904, + "grad_norm": 2.4501149654388428, + "learning_rate": 1.5693430656934307e-06, + "loss": 2.4102, + "step": 100 + }, + { + "epoch": 0.1095890410958904, + "eval_loss": 2.5452804565429688, + "eval_runtime": 37.4224, + "eval_samples_per_second": 26.722, + "eval_steps_per_second": 1.122, + "step": 100 + }, + { + "epoch": 0.11068493150684931, + "grad_norm": 2.475864887237549, + "learning_rate": 1.5875912408759126e-06, + "loss": 2.4147, + "step": 101 + }, + { + "epoch": 0.11178082191780822, + "grad_norm": 2.558875322341919, + "learning_rate": 1.6058394160583942e-06, + "loss": 2.4694, + "step": 102 + }, + { + "epoch": 0.11287671232876713, + "grad_norm": 2.1085734367370605, + "learning_rate": 1.624087591240876e-06, + "loss": 2.5099, + "step": 103 + }, + { + "epoch": 0.11397260273972602, + "grad_norm": 2.410731792449951, + "learning_rate": 1.6423357664233577e-06, + "loss": 2.432, + "step": 104 + }, + { + "epoch": 0.11506849315068493, + "grad_norm": 2.2800745964050293, + "learning_rate": 1.6605839416058394e-06, + "loss": 2.3669, + "step": 105 + }, + { + "epoch": 0.11616438356164384, + "grad_norm": 2.2234933376312256, + "learning_rate": 1.6788321167883212e-06, + "loss": 2.4145, + "step": 106 + }, + { + "epoch": 0.11726027397260275, + "grad_norm": 2.3367040157318115, + "learning_rate": 1.6970802919708029e-06, + "loss": 2.4272, + "step": 107 + }, + { + "epoch": 0.11835616438356164, + "grad_norm": 2.7002811431884766, + "learning_rate": 1.715328467153285e-06, + "loss": 2.3478, + "step": 108 + }, + { + "epoch": 0.11945205479452055, + "grad_norm": 2.6628332138061523, + "learning_rate": 1.7335766423357666e-06, + "loss": 2.4096, + "step": 109 + }, + { + "epoch": 0.12054794520547946, + "grad_norm": 2.4174230098724365, + "learning_rate": 1.7518248175182485e-06, + "loss": 2.407, + "step": 110 + }, + { + "epoch": 0.12054794520547946, + "eval_loss": 2.491746664047241, + "eval_runtime": 37.2555, + "eval_samples_per_second": 26.842, + "eval_steps_per_second": 1.127, + "step": 110 + }, + { + "epoch": 0.12164383561643835, + "grad_norm": 2.24796986579895, + "learning_rate": 1.77007299270073e-06, + "loss": 2.4369, + "step": 111 + }, + { + "epoch": 0.12273972602739726, + "grad_norm": 2.720939874649048, + "learning_rate": 1.788321167883212e-06, + "loss": 2.4116, + "step": 112 + }, + { + "epoch": 0.12383561643835617, + "grad_norm": 2.6907670497894287, + "learning_rate": 1.8065693430656936e-06, + "loss": 2.4577, + "step": 113 + }, + { + "epoch": 0.12493150684931507, + "grad_norm": 2.559636354446411, + "learning_rate": 1.8248175182481753e-06, + "loss": 2.3999, + "step": 114 + }, + { + "epoch": 0.12602739726027398, + "grad_norm": 2.6730551719665527, + "learning_rate": 1.8430656934306571e-06, + "loss": 2.3428, + "step": 115 + }, + { + "epoch": 0.12712328767123288, + "grad_norm": 3.172179937362671, + "learning_rate": 1.8613138686131388e-06, + "loss": 2.3722, + "step": 116 + }, + { + "epoch": 0.12821917808219177, + "grad_norm": 2.6757171154022217, + "learning_rate": 1.8795620437956206e-06, + "loss": 2.3572, + "step": 117 + }, + { + "epoch": 0.1293150684931507, + "grad_norm": 2.5840766429901123, + "learning_rate": 1.8978102189781023e-06, + "loss": 2.353, + "step": 118 + }, + { + "epoch": 0.1304109589041096, + "grad_norm": 2.6248581409454346, + "learning_rate": 1.916058394160584e-06, + "loss": 2.36, + "step": 119 + }, + { + "epoch": 0.13150684931506848, + "grad_norm": 2.428516387939453, + "learning_rate": 1.934306569343066e-06, + "loss": 2.3821, + "step": 120 + }, + { + "epoch": 0.13150684931506848, + "eval_loss": 2.451565980911255, + "eval_runtime": 37.3669, + "eval_samples_per_second": 26.762, + "eval_steps_per_second": 1.124, + "step": 120 + }, + { + "epoch": 0.1326027397260274, + "grad_norm": 2.9956233501434326, + "learning_rate": 1.952554744525548e-06, + "loss": 2.2778, + "step": 121 + }, + { + "epoch": 0.1336986301369863, + "grad_norm": 2.4120893478393555, + "learning_rate": 1.9708029197080293e-06, + "loss": 2.325, + "step": 122 + }, + { + "epoch": 0.13479452054794522, + "grad_norm": 2.1410024166107178, + "learning_rate": 1.989051094890511e-06, + "loss": 2.3248, + "step": 123 + }, + { + "epoch": 0.1358904109589041, + "grad_norm": 2.923799514770508, + "learning_rate": 2.007299270072993e-06, + "loss": 2.327, + "step": 124 + }, + { + "epoch": 0.136986301369863, + "grad_norm": 3.0770339965820312, + "learning_rate": 2.025547445255475e-06, + "loss": 2.3847, + "step": 125 + }, + { + "epoch": 0.13808219178082193, + "grad_norm": 2.6408181190490723, + "learning_rate": 2.0437956204379563e-06, + "loss": 2.3196, + "step": 126 + }, + { + "epoch": 0.13917808219178082, + "grad_norm": 2.9127118587493896, + "learning_rate": 2.062043795620438e-06, + "loss": 2.3287, + "step": 127 + }, + { + "epoch": 0.14027397260273972, + "grad_norm": 2.9805052280426025, + "learning_rate": 2.08029197080292e-06, + "loss": 2.3243, + "step": 128 + }, + { + "epoch": 0.14136986301369864, + "grad_norm": 3.1342010498046875, + "learning_rate": 2.0985401459854014e-06, + "loss": 2.38, + "step": 129 + }, + { + "epoch": 0.14246575342465753, + "grad_norm": 2.678422451019287, + "learning_rate": 2.1167883211678833e-06, + "loss": 2.2727, + "step": 130 + }, + { + "epoch": 0.14246575342465753, + "eval_loss": 2.4187021255493164, + "eval_runtime": 37.3117, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 130 + }, + { + "epoch": 0.14356164383561643, + "grad_norm": 2.9281492233276367, + "learning_rate": 2.135036496350365e-06, + "loss": 2.3327, + "step": 131 + }, + { + "epoch": 0.14465753424657535, + "grad_norm": 2.9592933654785156, + "learning_rate": 2.1532846715328466e-06, + "loss": 2.3674, + "step": 132 + }, + { + "epoch": 0.14575342465753424, + "grad_norm": 3.404761552810669, + "learning_rate": 2.171532846715329e-06, + "loss": 2.2679, + "step": 133 + }, + { + "epoch": 0.14684931506849316, + "grad_norm": 3.635571241378784, + "learning_rate": 2.1897810218978103e-06, + "loss": 2.2799, + "step": 134 + }, + { + "epoch": 0.14794520547945206, + "grad_norm": 3.9130170345306396, + "learning_rate": 2.208029197080292e-06, + "loss": 2.3263, + "step": 135 + }, + { + "epoch": 0.14904109589041095, + "grad_norm": 4.300438404083252, + "learning_rate": 2.226277372262774e-06, + "loss": 2.2969, + "step": 136 + }, + { + "epoch": 0.15013698630136987, + "grad_norm": 2.863511085510254, + "learning_rate": 2.244525547445256e-06, + "loss": 2.297, + "step": 137 + }, + { + "epoch": 0.15123287671232877, + "grad_norm": 4.910321235656738, + "learning_rate": 2.2627737226277373e-06, + "loss": 2.3269, + "step": 138 + }, + { + "epoch": 0.15232876712328766, + "grad_norm": 3.3323862552642822, + "learning_rate": 2.281021897810219e-06, + "loss": 2.3195, + "step": 139 + }, + { + "epoch": 0.15342465753424658, + "grad_norm": 3.2672693729400635, + "learning_rate": 2.299270072992701e-06, + "loss": 2.2797, + "step": 140 + }, + { + "epoch": 0.15342465753424658, + "eval_loss": 2.3943886756896973, + "eval_runtime": 37.3137, + "eval_samples_per_second": 26.8, + "eval_steps_per_second": 1.126, + "step": 140 + }, + { + "epoch": 0.15452054794520548, + "grad_norm": 3.9229774475097656, + "learning_rate": 2.3175182481751825e-06, + "loss": 2.2718, + "step": 141 + }, + { + "epoch": 0.15561643835616437, + "grad_norm": 2.6409823894500732, + "learning_rate": 2.3357664233576643e-06, + "loss": 2.2828, + "step": 142 + }, + { + "epoch": 0.1567123287671233, + "grad_norm": 4.6122612953186035, + "learning_rate": 2.354014598540146e-06, + "loss": 2.247, + "step": 143 + }, + { + "epoch": 0.1578082191780822, + "grad_norm": 3.391357898712158, + "learning_rate": 2.3722627737226276e-06, + "loss": 2.2863, + "step": 144 + }, + { + "epoch": 0.1589041095890411, + "grad_norm": 3.3550853729248047, + "learning_rate": 2.39051094890511e-06, + "loss": 2.2789, + "step": 145 + }, + { + "epoch": 0.16, + "grad_norm": 3.7376883029937744, + "learning_rate": 2.4087591240875918e-06, + "loss": 2.2297, + "step": 146 + }, + { + "epoch": 0.1610958904109589, + "grad_norm": 3.005157709121704, + "learning_rate": 2.427007299270073e-06, + "loss": 2.2467, + "step": 147 + }, + { + "epoch": 0.16219178082191782, + "grad_norm": 3.4616103172302246, + "learning_rate": 2.445255474452555e-06, + "loss": 2.2612, + "step": 148 + }, + { + "epoch": 0.1632876712328767, + "grad_norm": 3.3595101833343506, + "learning_rate": 2.463503649635037e-06, + "loss": 2.3021, + "step": 149 + }, + { + "epoch": 0.1643835616438356, + "grad_norm": 4.831241607666016, + "learning_rate": 2.4817518248175183e-06, + "loss": 2.2174, + "step": 150 + }, + { + "epoch": 0.1643835616438356, + "eval_loss": 2.3677749633789062, + "eval_runtime": 37.3092, + "eval_samples_per_second": 26.803, + "eval_steps_per_second": 1.126, + "step": 150 + }, + { + "epoch": 0.16547945205479453, + "grad_norm": 2.778820037841797, + "learning_rate": 2.5e-06, + "loss": 2.1856, + "step": 151 + }, + { + "epoch": 0.16657534246575342, + "grad_norm": 4.566643714904785, + "learning_rate": 2.518248175182482e-06, + "loss": 2.3186, + "step": 152 + }, + { + "epoch": 0.16767123287671232, + "grad_norm": 2.9908089637756348, + "learning_rate": 2.5364963503649635e-06, + "loss": 2.2127, + "step": 153 + }, + { + "epoch": 0.16876712328767124, + "grad_norm": 2.9702792167663574, + "learning_rate": 2.5547445255474458e-06, + "loss": 2.2071, + "step": 154 + }, + { + "epoch": 0.16986301369863013, + "grad_norm": 2.850044012069702, + "learning_rate": 2.5729927007299272e-06, + "loss": 2.2585, + "step": 155 + }, + { + "epoch": 0.17095890410958905, + "grad_norm": 2.926091432571411, + "learning_rate": 2.591240875912409e-06, + "loss": 2.2397, + "step": 156 + }, + { + "epoch": 0.17205479452054795, + "grad_norm": 2.8720169067382812, + "learning_rate": 2.6094890510948905e-06, + "loss": 2.2306, + "step": 157 + }, + { + "epoch": 0.17315068493150684, + "grad_norm": 2.8837177753448486, + "learning_rate": 2.627737226277373e-06, + "loss": 2.2666, + "step": 158 + }, + { + "epoch": 0.17424657534246576, + "grad_norm": 3.119621753692627, + "learning_rate": 2.6459854014598542e-06, + "loss": 2.2722, + "step": 159 + }, + { + "epoch": 0.17534246575342466, + "grad_norm": 2.7085700035095215, + "learning_rate": 2.664233576642336e-06, + "loss": 2.2346, + "step": 160 + }, + { + "epoch": 0.17534246575342466, + "eval_loss": 2.352083921432495, + "eval_runtime": 37.245, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 160 + }, + { + "epoch": 0.17643835616438355, + "grad_norm": 2.9964346885681152, + "learning_rate": 2.6824817518248175e-06, + "loss": 2.2155, + "step": 161 + }, + { + "epoch": 0.17753424657534247, + "grad_norm": 3.1535208225250244, + "learning_rate": 2.7007299270072994e-06, + "loss": 2.235, + "step": 162 + }, + { + "epoch": 0.17863013698630137, + "grad_norm": 2.7240943908691406, + "learning_rate": 2.718978102189781e-06, + "loss": 2.2095, + "step": 163 + }, + { + "epoch": 0.17972602739726026, + "grad_norm": 3.122654676437378, + "learning_rate": 2.737226277372263e-06, + "loss": 2.2667, + "step": 164 + }, + { + "epoch": 0.18082191780821918, + "grad_norm": 2.9718410968780518, + "learning_rate": 2.755474452554745e-06, + "loss": 2.2499, + "step": 165 + }, + { + "epoch": 0.18191780821917808, + "grad_norm": 3.0081896781921387, + "learning_rate": 2.7737226277372264e-06, + "loss": 2.2089, + "step": 166 + }, + { + "epoch": 0.18301369863013697, + "grad_norm": 2.829509973526001, + "learning_rate": 2.7919708029197082e-06, + "loss": 2.3139, + "step": 167 + }, + { + "epoch": 0.1841095890410959, + "grad_norm": 3.4480397701263428, + "learning_rate": 2.81021897810219e-06, + "loss": 2.2705, + "step": 168 + }, + { + "epoch": 0.1852054794520548, + "grad_norm": 3.1107025146484375, + "learning_rate": 2.828467153284672e-06, + "loss": 2.1895, + "step": 169 + }, + { + "epoch": 0.1863013698630137, + "grad_norm": 2.916790008544922, + "learning_rate": 2.8467153284671534e-06, + "loss": 2.2317, + "step": 170 + }, + { + "epoch": 0.1863013698630137, + "eval_loss": 2.336153030395508, + "eval_runtime": 37.2401, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 170 + }, + { + "epoch": 0.1873972602739726, + "grad_norm": 3.2582287788391113, + "learning_rate": 2.8649635036496353e-06, + "loss": 2.143, + "step": 171 + }, + { + "epoch": 0.1884931506849315, + "grad_norm": 2.430973768234253, + "learning_rate": 2.8832116788321167e-06, + "loss": 2.1977, + "step": 172 + }, + { + "epoch": 0.18958904109589042, + "grad_norm": 2.8763599395751953, + "learning_rate": 2.901459854014599e-06, + "loss": 2.1551, + "step": 173 + }, + { + "epoch": 0.19068493150684931, + "grad_norm": 2.9242653846740723, + "learning_rate": 2.9197080291970804e-06, + "loss": 2.2569, + "step": 174 + }, + { + "epoch": 0.1917808219178082, + "grad_norm": 2.623551607131958, + "learning_rate": 2.9379562043795623e-06, + "loss": 2.2329, + "step": 175 + }, + { + "epoch": 0.19287671232876713, + "grad_norm": 2.9511337280273438, + "learning_rate": 2.956204379562044e-06, + "loss": 2.2154, + "step": 176 + }, + { + "epoch": 0.19397260273972602, + "grad_norm": 2.415118932723999, + "learning_rate": 2.9744525547445256e-06, + "loss": 2.2637, + "step": 177 + }, + { + "epoch": 0.19506849315068492, + "grad_norm": 2.9664924144744873, + "learning_rate": 2.992700729927008e-06, + "loss": 2.2263, + "step": 178 + }, + { + "epoch": 0.19616438356164384, + "grad_norm": 2.7214572429656982, + "learning_rate": 3.0109489051094893e-06, + "loss": 2.1667, + "step": 179 + }, + { + "epoch": 0.19726027397260273, + "grad_norm": 2.8712823390960693, + "learning_rate": 3.029197080291971e-06, + "loss": 2.1661, + "step": 180 + }, + { + "epoch": 0.19726027397260273, + "eval_loss": 2.3191800117492676, + "eval_runtime": 37.24, + "eval_samples_per_second": 26.853, + "eval_steps_per_second": 1.128, + "step": 180 + }, + { + "epoch": 0.19835616438356166, + "grad_norm": 2.959709405899048, + "learning_rate": 3.0474452554744526e-06, + "loss": 2.2483, + "step": 181 + }, + { + "epoch": 0.19945205479452055, + "grad_norm": 2.567089319229126, + "learning_rate": 3.065693430656935e-06, + "loss": 2.2382, + "step": 182 + }, + { + "epoch": 0.20054794520547944, + "grad_norm": 2.47188401222229, + "learning_rate": 3.0839416058394163e-06, + "loss": 2.2148, + "step": 183 + }, + { + "epoch": 0.20164383561643837, + "grad_norm": 2.3435261249542236, + "learning_rate": 3.102189781021898e-06, + "loss": 2.2334, + "step": 184 + }, + { + "epoch": 0.20273972602739726, + "grad_norm": 2.5280468463897705, + "learning_rate": 3.1204379562043796e-06, + "loss": 2.2214, + "step": 185 + }, + { + "epoch": 0.20383561643835615, + "grad_norm": 2.4808971881866455, + "learning_rate": 3.1386861313868614e-06, + "loss": 2.1554, + "step": 186 + }, + { + "epoch": 0.20493150684931508, + "grad_norm": 2.9937026500701904, + "learning_rate": 3.156934306569343e-06, + "loss": 2.1389, + "step": 187 + }, + { + "epoch": 0.20602739726027397, + "grad_norm": 2.9220476150512695, + "learning_rate": 3.175182481751825e-06, + "loss": 2.2515, + "step": 188 + }, + { + "epoch": 0.20712328767123286, + "grad_norm": 3.0449001789093018, + "learning_rate": 3.193430656934307e-06, + "loss": 2.2016, + "step": 189 + }, + { + "epoch": 0.20821917808219179, + "grad_norm": 2.8027074337005615, + "learning_rate": 3.2116788321167884e-06, + "loss": 2.1823, + "step": 190 + }, + { + "epoch": 0.20821917808219179, + "eval_loss": 2.309677839279175, + "eval_runtime": 37.2442, + "eval_samples_per_second": 26.85, + "eval_steps_per_second": 1.128, + "step": 190 + }, + { + "epoch": 0.20931506849315068, + "grad_norm": 2.9158318042755127, + "learning_rate": 3.2299270072992707e-06, + "loss": 2.1928, + "step": 191 + }, + { + "epoch": 0.2104109589041096, + "grad_norm": 2.5328080654144287, + "learning_rate": 3.248175182481752e-06, + "loss": 2.1828, + "step": 192 + }, + { + "epoch": 0.2115068493150685, + "grad_norm": 3.1709704399108887, + "learning_rate": 3.266423357664234e-06, + "loss": 2.1052, + "step": 193 + }, + { + "epoch": 0.2126027397260274, + "grad_norm": 2.9156367778778076, + "learning_rate": 3.2846715328467155e-06, + "loss": 2.1943, + "step": 194 + }, + { + "epoch": 0.2136986301369863, + "grad_norm": 2.879408836364746, + "learning_rate": 3.3029197080291973e-06, + "loss": 2.1782, + "step": 195 + }, + { + "epoch": 0.2147945205479452, + "grad_norm": 3.748483180999756, + "learning_rate": 3.3211678832116788e-06, + "loss": 2.2107, + "step": 196 + }, + { + "epoch": 0.2158904109589041, + "grad_norm": 3.197510004043579, + "learning_rate": 3.339416058394161e-06, + "loss": 2.1845, + "step": 197 + }, + { + "epoch": 0.21698630136986302, + "grad_norm": 3.2596049308776855, + "learning_rate": 3.3576642335766425e-06, + "loss": 2.1816, + "step": 198 + }, + { + "epoch": 0.21808219178082192, + "grad_norm": 3.3329780101776123, + "learning_rate": 3.3759124087591243e-06, + "loss": 2.181, + "step": 199 + }, + { + "epoch": 0.2191780821917808, + "grad_norm": 2.7302327156066895, + "learning_rate": 3.3941605839416058e-06, + "loss": 2.1761, + "step": 200 + }, + { + "epoch": 0.2191780821917808, + "eval_loss": 2.2909653186798096, + "eval_runtime": 37.284, + "eval_samples_per_second": 26.821, + "eval_steps_per_second": 1.126, + "step": 200 + }, + { + "epoch": 0.22027397260273973, + "grad_norm": 3.002255916595459, + "learning_rate": 3.412408759124088e-06, + "loss": 2.1868, + "step": 201 + }, + { + "epoch": 0.22136986301369863, + "grad_norm": 2.6694891452789307, + "learning_rate": 3.43065693430657e-06, + "loss": 2.192, + "step": 202 + }, + { + "epoch": 0.22246575342465755, + "grad_norm": 2.9225986003875732, + "learning_rate": 3.4489051094890513e-06, + "loss": 2.1303, + "step": 203 + }, + { + "epoch": 0.22356164383561644, + "grad_norm": 2.7844080924987793, + "learning_rate": 3.467153284671533e-06, + "loss": 2.1788, + "step": 204 + }, + { + "epoch": 0.22465753424657534, + "grad_norm": 2.726123332977295, + "learning_rate": 3.4854014598540146e-06, + "loss": 2.1193, + "step": 205 + }, + { + "epoch": 0.22575342465753426, + "grad_norm": 2.838040590286255, + "learning_rate": 3.503649635036497e-06, + "loss": 2.1674, + "step": 206 + }, + { + "epoch": 0.22684931506849315, + "grad_norm": 2.6863937377929688, + "learning_rate": 3.5218978102189783e-06, + "loss": 2.161, + "step": 207 + }, + { + "epoch": 0.22794520547945205, + "grad_norm": 2.5116159915924072, + "learning_rate": 3.54014598540146e-06, + "loss": 2.179, + "step": 208 + }, + { + "epoch": 0.22904109589041097, + "grad_norm": 3.381863594055176, + "learning_rate": 3.5583941605839416e-06, + "loss": 2.2035, + "step": 209 + }, + { + "epoch": 0.23013698630136986, + "grad_norm": 3.2741382122039795, + "learning_rate": 3.576642335766424e-06, + "loss": 2.2518, + "step": 210 + }, + { + "epoch": 0.23013698630136986, + "eval_loss": 2.2773635387420654, + "eval_runtime": 37.2658, + "eval_samples_per_second": 26.834, + "eval_steps_per_second": 1.127, + "step": 210 + }, + { + "epoch": 0.23123287671232876, + "grad_norm": 2.917370319366455, + "learning_rate": 3.5948905109489054e-06, + "loss": 2.2265, + "step": 211 + }, + { + "epoch": 0.23232876712328768, + "grad_norm": 2.8392834663391113, + "learning_rate": 3.6131386861313872e-06, + "loss": 2.0709, + "step": 212 + }, + { + "epoch": 0.23342465753424657, + "grad_norm": 2.753612518310547, + "learning_rate": 3.6313868613138687e-06, + "loss": 2.0752, + "step": 213 + }, + { + "epoch": 0.2345205479452055, + "grad_norm": 3.1477746963500977, + "learning_rate": 3.6496350364963505e-06, + "loss": 2.1169, + "step": 214 + }, + { + "epoch": 0.2356164383561644, + "grad_norm": 2.5448100566864014, + "learning_rate": 3.667883211678833e-06, + "loss": 2.1338, + "step": 215 + }, + { + "epoch": 0.23671232876712328, + "grad_norm": 3.1622560024261475, + "learning_rate": 3.6861313868613142e-06, + "loss": 2.1065, + "step": 216 + }, + { + "epoch": 0.2378082191780822, + "grad_norm": 2.7054336071014404, + "learning_rate": 3.704379562043796e-06, + "loss": 2.1104, + "step": 217 + }, + { + "epoch": 0.2389041095890411, + "grad_norm": 2.964141845703125, + "learning_rate": 3.7226277372262775e-06, + "loss": 2.1949, + "step": 218 + }, + { + "epoch": 0.24, + "grad_norm": 2.7586112022399902, + "learning_rate": 3.7408759124087594e-06, + "loss": 2.165, + "step": 219 + }, + { + "epoch": 0.2410958904109589, + "grad_norm": 2.9035964012145996, + "learning_rate": 3.7591240875912412e-06, + "loss": 2.1192, + "step": 220 + }, + { + "epoch": 0.2410958904109589, + "eval_loss": 2.2649471759796143, + "eval_runtime": 37.4311, + "eval_samples_per_second": 26.716, + "eval_steps_per_second": 1.122, + "step": 220 + }, + { + "epoch": 0.2421917808219178, + "grad_norm": 2.4212076663970947, + "learning_rate": 3.777372262773723e-06, + "loss": 2.1389, + "step": 221 + }, + { + "epoch": 0.2432876712328767, + "grad_norm": 2.973227024078369, + "learning_rate": 3.7956204379562045e-06, + "loss": 2.132, + "step": 222 + }, + { + "epoch": 0.24438356164383562, + "grad_norm": 2.9831430912017822, + "learning_rate": 3.8138686131386864e-06, + "loss": 2.1142, + "step": 223 + }, + { + "epoch": 0.24547945205479452, + "grad_norm": 3.2880353927612305, + "learning_rate": 3.832116788321168e-06, + "loss": 2.168, + "step": 224 + }, + { + "epoch": 0.2465753424657534, + "grad_norm": 3.3920998573303223, + "learning_rate": 3.85036496350365e-06, + "loss": 2.1277, + "step": 225 + }, + { + "epoch": 0.24767123287671233, + "grad_norm": 2.7170026302337646, + "learning_rate": 3.868613138686132e-06, + "loss": 2.0936, + "step": 226 + }, + { + "epoch": 0.24876712328767123, + "grad_norm": 3.6220011711120605, + "learning_rate": 3.886861313868613e-06, + "loss": 2.1259, + "step": 227 + }, + { + "epoch": 0.24986301369863015, + "grad_norm": 3.4229273796081543, + "learning_rate": 3.905109489051096e-06, + "loss": 2.1421, + "step": 228 + }, + { + "epoch": 0.25095890410958904, + "grad_norm": 3.0358338356018066, + "learning_rate": 3.923357664233577e-06, + "loss": 2.1747, + "step": 229 + }, + { + "epoch": 0.25205479452054796, + "grad_norm": 2.611771821975708, + "learning_rate": 3.9416058394160585e-06, + "loss": 2.1282, + "step": 230 + }, + { + "epoch": 0.25205479452054796, + "eval_loss": 2.254887580871582, + "eval_runtime": 37.5943, + "eval_samples_per_second": 26.6, + "eval_steps_per_second": 1.117, + "step": 230 + }, + { + "epoch": 0.25315068493150683, + "grad_norm": 3.846412181854248, + "learning_rate": 3.95985401459854e-06, + "loss": 2.1892, + "step": 231 + }, + { + "epoch": 0.25424657534246575, + "grad_norm": 2.488525390625, + "learning_rate": 3.978102189781022e-06, + "loss": 2.2064, + "step": 232 + }, + { + "epoch": 0.2553424657534247, + "grad_norm": 3.114964485168457, + "learning_rate": 3.996350364963504e-06, + "loss": 2.0653, + "step": 233 + }, + { + "epoch": 0.25643835616438354, + "grad_norm": 2.5559768676757812, + "learning_rate": 4.014598540145986e-06, + "loss": 2.1503, + "step": 234 + }, + { + "epoch": 0.25753424657534246, + "grad_norm": 3.2347540855407715, + "learning_rate": 4.032846715328467e-06, + "loss": 2.0523, + "step": 235 + }, + { + "epoch": 0.2586301369863014, + "grad_norm": 2.8086111545562744, + "learning_rate": 4.05109489051095e-06, + "loss": 2.1206, + "step": 236 + }, + { + "epoch": 0.25972602739726025, + "grad_norm": 2.909231185913086, + "learning_rate": 4.069343065693431e-06, + "loss": 2.139, + "step": 237 + }, + { + "epoch": 0.2608219178082192, + "grad_norm": 3.2171082496643066, + "learning_rate": 4.0875912408759126e-06, + "loss": 2.0908, + "step": 238 + }, + { + "epoch": 0.2619178082191781, + "grad_norm": 2.3851265907287598, + "learning_rate": 4.105839416058395e-06, + "loss": 2.1133, + "step": 239 + }, + { + "epoch": 0.26301369863013696, + "grad_norm": 3.3912737369537354, + "learning_rate": 4.124087591240876e-06, + "loss": 2.1911, + "step": 240 + }, + { + "epoch": 0.26301369863013696, + "eval_loss": 2.2436957359313965, + "eval_runtime": 37.2947, + "eval_samples_per_second": 26.813, + "eval_steps_per_second": 1.126, + "step": 240 + }, + { + "epoch": 0.2641095890410959, + "grad_norm": 2.82906174659729, + "learning_rate": 4.1423357664233586e-06, + "loss": 2.11, + "step": 241 + }, + { + "epoch": 0.2652054794520548, + "grad_norm": 3.0317952632904053, + "learning_rate": 4.16058394160584e-06, + "loss": 2.0953, + "step": 242 + }, + { + "epoch": 0.2663013698630137, + "grad_norm": 3.6438329219818115, + "learning_rate": 4.1788321167883214e-06, + "loss": 2.1746, + "step": 243 + }, + { + "epoch": 0.2673972602739726, + "grad_norm": 2.6690592765808105, + "learning_rate": 4.197080291970803e-06, + "loss": 2.0555, + "step": 244 + }, + { + "epoch": 0.2684931506849315, + "grad_norm": 3.135376453399658, + "learning_rate": 4.215328467153285e-06, + "loss": 2.1394, + "step": 245 + }, + { + "epoch": 0.26958904109589044, + "grad_norm": 2.5474987030029297, + "learning_rate": 4.233576642335767e-06, + "loss": 2.0822, + "step": 246 + }, + { + "epoch": 0.2706849315068493, + "grad_norm": 2.883108377456665, + "learning_rate": 4.251824817518249e-06, + "loss": 2.0869, + "step": 247 + }, + { + "epoch": 0.2717808219178082, + "grad_norm": 2.9801037311553955, + "learning_rate": 4.27007299270073e-06, + "loss": 2.1194, + "step": 248 + }, + { + "epoch": 0.27287671232876715, + "grad_norm": 2.7688217163085938, + "learning_rate": 4.288321167883212e-06, + "loss": 2.0908, + "step": 249 + }, + { + "epoch": 0.273972602739726, + "grad_norm": 3.205148458480835, + "learning_rate": 4.306569343065693e-06, + "loss": 2.0762, + "step": 250 + }, + { + "epoch": 0.273972602739726, + "eval_loss": 2.238699197769165, + "eval_runtime": 37.2428, + "eval_samples_per_second": 26.851, + "eval_steps_per_second": 1.128, + "step": 250 + }, + { + "epoch": 0.27506849315068493, + "grad_norm": 2.8262946605682373, + "learning_rate": 4.3248175182481755e-06, + "loss": 2.0509, + "step": 251 + }, + { + "epoch": 0.27616438356164386, + "grad_norm": 3.156261682510376, + "learning_rate": 4.343065693430658e-06, + "loss": 2.1213, + "step": 252 + }, + { + "epoch": 0.2772602739726027, + "grad_norm": 2.6367998123168945, + "learning_rate": 4.361313868613139e-06, + "loss": 2.1071, + "step": 253 + }, + { + "epoch": 0.27835616438356164, + "grad_norm": 2.871534824371338, + "learning_rate": 4.379562043795621e-06, + "loss": 2.1008, + "step": 254 + }, + { + "epoch": 0.27945205479452057, + "grad_norm": 2.839432716369629, + "learning_rate": 4.397810218978102e-06, + "loss": 2.1293, + "step": 255 + }, + { + "epoch": 0.28054794520547943, + "grad_norm": 2.627143383026123, + "learning_rate": 4.416058394160584e-06, + "loss": 2.1141, + "step": 256 + }, + { + "epoch": 0.28164383561643835, + "grad_norm": 2.8292996883392334, + "learning_rate": 4.434306569343066e-06, + "loss": 2.0981, + "step": 257 + }, + { + "epoch": 0.2827397260273973, + "grad_norm": 2.5711069107055664, + "learning_rate": 4.452554744525548e-06, + "loss": 2.0707, + "step": 258 + }, + { + "epoch": 0.28383561643835614, + "grad_norm": 2.488954544067383, + "learning_rate": 4.4708029197080295e-06, + "loss": 2.1172, + "step": 259 + }, + { + "epoch": 0.28493150684931506, + "grad_norm": 2.442209005355835, + "learning_rate": 4.489051094890512e-06, + "loss": 2.1119, + "step": 260 + }, + { + "epoch": 0.28493150684931506, + "eval_loss": 2.2306911945343018, + "eval_runtime": 37.2412, + "eval_samples_per_second": 26.852, + "eval_steps_per_second": 1.128, + "step": 260 + }, + { + "epoch": 0.286027397260274, + "grad_norm": 2.9147934913635254, + "learning_rate": 4.507299270072993e-06, + "loss": 2.1517, + "step": 261 + }, + { + "epoch": 0.28712328767123285, + "grad_norm": 2.549833297729492, + "learning_rate": 4.525547445255475e-06, + "loss": 2.1047, + "step": 262 + }, + { + "epoch": 0.2882191780821918, + "grad_norm": 3.3405206203460693, + "learning_rate": 4.543795620437956e-06, + "loss": 2.0975, + "step": 263 + }, + { + "epoch": 0.2893150684931507, + "grad_norm": 2.554431676864624, + "learning_rate": 4.562043795620438e-06, + "loss": 2.0622, + "step": 264 + }, + { + "epoch": 0.29041095890410956, + "grad_norm": 3.5529873371124268, + "learning_rate": 4.580291970802921e-06, + "loss": 2.1567, + "step": 265 + }, + { + "epoch": 0.2915068493150685, + "grad_norm": 2.421682119369507, + "learning_rate": 4.598540145985402e-06, + "loss": 2.1619, + "step": 266 + }, + { + "epoch": 0.2926027397260274, + "grad_norm": 2.918288230895996, + "learning_rate": 4.6167883211678835e-06, + "loss": 2.1238, + "step": 267 + }, + { + "epoch": 0.2936986301369863, + "grad_norm": 3.0737621784210205, + "learning_rate": 4.635036496350365e-06, + "loss": 2.0728, + "step": 268 + }, + { + "epoch": 0.2947945205479452, + "grad_norm": 2.5862410068511963, + "learning_rate": 4.653284671532847e-06, + "loss": 2.0675, + "step": 269 + }, + { + "epoch": 0.2958904109589041, + "grad_norm": 2.9004979133605957, + "learning_rate": 4.671532846715329e-06, + "loss": 2.1248, + "step": 270 + }, + { + "epoch": 0.2958904109589041, + "eval_loss": 2.2179365158081055, + "eval_runtime": 37.2352, + "eval_samples_per_second": 26.856, + "eval_steps_per_second": 1.128, + "step": 270 + }, + { + "epoch": 0.29698630136986304, + "grad_norm": 2.4784982204437256, + "learning_rate": 4.689781021897811e-06, + "loss": 2.1276, + "step": 271 + }, + { + "epoch": 0.2980821917808219, + "grad_norm": 2.7971739768981934, + "learning_rate": 4.708029197080292e-06, + "loss": 2.0931, + "step": 272 + }, + { + "epoch": 0.2991780821917808, + "grad_norm": 2.5839481353759766, + "learning_rate": 4.726277372262774e-06, + "loss": 2.1371, + "step": 273 + }, + { + "epoch": 0.30027397260273975, + "grad_norm": 2.7168564796447754, + "learning_rate": 4.744525547445255e-06, + "loss": 2.1086, + "step": 274 + }, + { + "epoch": 0.3013698630136986, + "grad_norm": 2.6566879749298096, + "learning_rate": 4.7627737226277375e-06, + "loss": 2.0368, + "step": 275 + }, + { + "epoch": 0.30246575342465754, + "grad_norm": 2.5068323612213135, + "learning_rate": 4.78102189781022e-06, + "loss": 2.1077, + "step": 276 + }, + { + "epoch": 0.30356164383561646, + "grad_norm": 2.631408214569092, + "learning_rate": 4.799270072992701e-06, + "loss": 2.1013, + "step": 277 + }, + { + "epoch": 0.3046575342465753, + "grad_norm": 2.5642812252044678, + "learning_rate": 4.8175182481751835e-06, + "loss": 2.0739, + "step": 278 + }, + { + "epoch": 0.30575342465753425, + "grad_norm": 2.3326497077941895, + "learning_rate": 4.835766423357665e-06, + "loss": 2.0971, + "step": 279 + }, + { + "epoch": 0.30684931506849317, + "grad_norm": 2.70701265335083, + "learning_rate": 4.854014598540146e-06, + "loss": 2.0904, + "step": 280 + }, + { + "epoch": 0.30684931506849317, + "eval_loss": 2.215972900390625, + "eval_runtime": 37.2196, + "eval_samples_per_second": 26.868, + "eval_steps_per_second": 1.128, + "step": 280 + }, + { + "epoch": 0.30794520547945203, + "grad_norm": 2.615201234817505, + "learning_rate": 4.872262773722628e-06, + "loss": 2.039, + "step": 281 + }, + { + "epoch": 0.30904109589041096, + "grad_norm": 2.2821998596191406, + "learning_rate": 4.89051094890511e-06, + "loss": 2.0529, + "step": 282 + }, + { + "epoch": 0.3101369863013699, + "grad_norm": 2.767153024673462, + "learning_rate": 4.9087591240875915e-06, + "loss": 2.0351, + "step": 283 + }, + { + "epoch": 0.31123287671232874, + "grad_norm": 2.537121295928955, + "learning_rate": 4.927007299270074e-06, + "loss": 2.0752, + "step": 284 + }, + { + "epoch": 0.31232876712328766, + "grad_norm": 2.835052013397217, + "learning_rate": 4.945255474452555e-06, + "loss": 2.0892, + "step": 285 + }, + { + "epoch": 0.3134246575342466, + "grad_norm": 2.5154924392700195, + "learning_rate": 4.963503649635037e-06, + "loss": 2.0569, + "step": 286 + }, + { + "epoch": 0.31452054794520545, + "grad_norm": 2.8212568759918213, + "learning_rate": 4.981751824817518e-06, + "loss": 2.1588, + "step": 287 + }, + { + "epoch": 0.3156164383561644, + "grad_norm": 2.3840994834899902, + "learning_rate": 5e-06, + "loss": 2.1409, + "step": 288 + }, + { + "epoch": 0.3167123287671233, + "grad_norm": 2.9641315937042236, + "learning_rate": 4.997969130787978e-06, + "loss": 2.0892, + "step": 289 + }, + { + "epoch": 0.3178082191780822, + "grad_norm": 2.4516749382019043, + "learning_rate": 4.995938261575955e-06, + "loss": 2.1105, + "step": 290 + }, + { + "epoch": 0.3178082191780822, + "eval_loss": 2.207075834274292, + "eval_runtime": 37.1304, + "eval_samples_per_second": 26.932, + "eval_steps_per_second": 1.131, + "step": 290 + }, + { + "epoch": 0.3189041095890411, + "grad_norm": 2.768869400024414, + "learning_rate": 4.993907392363932e-06, + "loss": 2.0231, + "step": 291 + }, + { + "epoch": 0.32, + "grad_norm": 2.771475315093994, + "learning_rate": 4.9918765231519095e-06, + "loss": 2.1018, + "step": 292 + }, + { + "epoch": 0.32109589041095893, + "grad_norm": 2.773983955383301, + "learning_rate": 4.989845653939887e-06, + "loss": 2.0317, + "step": 293 + }, + { + "epoch": 0.3221917808219178, + "grad_norm": 2.4447343349456787, + "learning_rate": 4.987814784727864e-06, + "loss": 2.0739, + "step": 294 + }, + { + "epoch": 0.3232876712328767, + "grad_norm": 2.444261312484741, + "learning_rate": 4.985783915515841e-06, + "loss": 2.073, + "step": 295 + }, + { + "epoch": 0.32438356164383564, + "grad_norm": 2.871792793273926, + "learning_rate": 4.983753046303819e-06, + "loss": 2.058, + "step": 296 + }, + { + "epoch": 0.3254794520547945, + "grad_norm": 2.3651537895202637, + "learning_rate": 4.981722177091796e-06, + "loss": 2.0749, + "step": 297 + }, + { + "epoch": 0.3265753424657534, + "grad_norm": 2.3395845890045166, + "learning_rate": 4.979691307879773e-06, + "loss": 2.065, + "step": 298 + }, + { + "epoch": 0.32767123287671235, + "grad_norm": 2.6275217533111572, + "learning_rate": 4.9776604386677504e-06, + "loss": 2.0945, + "step": 299 + }, + { + "epoch": 0.3287671232876712, + "grad_norm": 2.866285562515259, + "learning_rate": 4.975629569455728e-06, + "loss": 2.0702, + "step": 300 + }, + { + "epoch": 0.3287671232876712, + "eval_loss": 2.200012683868408, + "eval_runtime": 37.2082, + "eval_samples_per_second": 26.876, + "eval_steps_per_second": 1.129, + "step": 300 + }, + { + "epoch": 0.32986301369863014, + "grad_norm": 2.268401861190796, + "learning_rate": 4.973598700243705e-06, + "loss": 2.1257, + "step": 301 + }, + { + "epoch": 0.33095890410958906, + "grad_norm": 2.6197733879089355, + "learning_rate": 4.971567831031681e-06, + "loss": 2.0347, + "step": 302 + }, + { + "epoch": 0.3320547945205479, + "grad_norm": 2.4738848209381104, + "learning_rate": 4.9695369618196595e-06, + "loss": 2.0368, + "step": 303 + }, + { + "epoch": 0.33315068493150685, + "grad_norm": 2.705670118331909, + "learning_rate": 4.967506092607636e-06, + "loss": 1.9256, + "step": 304 + }, + { + "epoch": 0.33424657534246577, + "grad_norm": 2.813279628753662, + "learning_rate": 4.965475223395614e-06, + "loss": 2.0214, + "step": 305 + }, + { + "epoch": 0.33534246575342463, + "grad_norm": 2.320960283279419, + "learning_rate": 4.9634443541835905e-06, + "loss": 2.0342, + "step": 306 + }, + { + "epoch": 0.33643835616438356, + "grad_norm": 2.4495227336883545, + "learning_rate": 4.961413484971569e-06, + "loss": 2.0421, + "step": 307 + }, + { + "epoch": 0.3375342465753425, + "grad_norm": 2.5384018421173096, + "learning_rate": 4.959382615759545e-06, + "loss": 2.0572, + "step": 308 + }, + { + "epoch": 0.33863013698630134, + "grad_norm": 2.4018006324768066, + "learning_rate": 4.957351746547523e-06, + "loss": 2.0617, + "step": 309 + }, + { + "epoch": 0.33972602739726027, + "grad_norm": 2.552424907684326, + "learning_rate": 4.9553208773355e-06, + "loss": 2.0384, + "step": 310 + }, + { + "epoch": 0.33972602739726027, + "eval_loss": 2.1974849700927734, + "eval_runtime": 37.2308, + "eval_samples_per_second": 26.859, + "eval_steps_per_second": 1.128, + "step": 310 + }, + { + "epoch": 0.3408219178082192, + "grad_norm": 2.4587671756744385, + "learning_rate": 4.953290008123478e-06, + "loss": 2.1701, + "step": 311 + }, + { + "epoch": 0.3419178082191781, + "grad_norm": 2.4486496448516846, + "learning_rate": 4.951259138911454e-06, + "loss": 2.0654, + "step": 312 + }, + { + "epoch": 0.343013698630137, + "grad_norm": 2.123568058013916, + "learning_rate": 4.9492282696994315e-06, + "loss": 1.9961, + "step": 313 + }, + { + "epoch": 0.3441095890410959, + "grad_norm": 2.549788475036621, + "learning_rate": 4.947197400487409e-06, + "loss": 2.1265, + "step": 314 + }, + { + "epoch": 0.3452054794520548, + "grad_norm": 2.3453609943389893, + "learning_rate": 4.945166531275386e-06, + "loss": 2.0501, + "step": 315 + }, + { + "epoch": 0.3463013698630137, + "grad_norm": 2.3584256172180176, + "learning_rate": 4.943135662063363e-06, + "loss": 2.0402, + "step": 316 + }, + { + "epoch": 0.3473972602739726, + "grad_norm": 3.3100943565368652, + "learning_rate": 4.9411047928513406e-06, + "loss": 2.0545, + "step": 317 + }, + { + "epoch": 0.34849315068493153, + "grad_norm": 2.4266037940979004, + "learning_rate": 4.939073923639318e-06, + "loss": 2.0926, + "step": 318 + }, + { + "epoch": 0.3495890410958904, + "grad_norm": 3.101090908050537, + "learning_rate": 4.937043054427295e-06, + "loss": 2.0987, + "step": 319 + }, + { + "epoch": 0.3506849315068493, + "grad_norm": 2.743619203567505, + "learning_rate": 4.935012185215272e-06, + "loss": 2.0315, + "step": 320 + }, + { + "epoch": 0.3506849315068493, + "eval_loss": 2.188877820968628, + "eval_runtime": 37.4609, + "eval_samples_per_second": 26.694, + "eval_steps_per_second": 1.121, + "step": 320 + }, + { + "epoch": 0.35178082191780824, + "grad_norm": 2.6065986156463623, + "learning_rate": 4.93298131600325e-06, + "loss": 2.0166, + "step": 321 + }, + { + "epoch": 0.3528767123287671, + "grad_norm": 2.7554283142089844, + "learning_rate": 4.930950446791227e-06, + "loss": 2.0862, + "step": 322 + }, + { + "epoch": 0.353972602739726, + "grad_norm": 2.4773144721984863, + "learning_rate": 4.928919577579204e-06, + "loss": 2.0472, + "step": 323 + }, + { + "epoch": 0.35506849315068495, + "grad_norm": 2.695289134979248, + "learning_rate": 4.9268887083671815e-06, + "loss": 2.0567, + "step": 324 + }, + { + "epoch": 0.3561643835616438, + "grad_norm": 2.408220052719116, + "learning_rate": 4.924857839155159e-06, + "loss": 2.046, + "step": 325 + }, + { + "epoch": 0.35726027397260274, + "grad_norm": 2.3851559162139893, + "learning_rate": 4.922826969943136e-06, + "loss": 2.054, + "step": 326 + }, + { + "epoch": 0.35835616438356166, + "grad_norm": 2.8667280673980713, + "learning_rate": 4.920796100731113e-06, + "loss": 2.0717, + "step": 327 + }, + { + "epoch": 0.3594520547945205, + "grad_norm": 2.3814942836761475, + "learning_rate": 4.918765231519091e-06, + "loss": 1.9884, + "step": 328 + }, + { + "epoch": 0.36054794520547945, + "grad_norm": 2.341076612472534, + "learning_rate": 4.916734362307068e-06, + "loss": 2.0474, + "step": 329 + }, + { + "epoch": 0.36164383561643837, + "grad_norm": 2.3571853637695312, + "learning_rate": 4.914703493095045e-06, + "loss": 2.0009, + "step": 330 + }, + { + "epoch": 0.36164383561643837, + "eval_loss": 2.186048746109009, + "eval_runtime": 37.5746, + "eval_samples_per_second": 26.614, + "eval_steps_per_second": 1.118, + "step": 330 + }, + { + "epoch": 0.36273972602739724, + "grad_norm": 2.3770196437835693, + "learning_rate": 4.912672623883022e-06, + "loss": 2.0167, + "step": 331 + }, + { + "epoch": 0.36383561643835616, + "grad_norm": 2.3357701301574707, + "learning_rate": 4.910641754671e-06, + "loss": 1.9972, + "step": 332 + }, + { + "epoch": 0.3649315068493151, + "grad_norm": 2.4100499153137207, + "learning_rate": 4.908610885458977e-06, + "loss": 2.0319, + "step": 333 + }, + { + "epoch": 0.36602739726027395, + "grad_norm": 2.193018913269043, + "learning_rate": 4.906580016246954e-06, + "loss": 2.0285, + "step": 334 + }, + { + "epoch": 0.36712328767123287, + "grad_norm": 2.275637149810791, + "learning_rate": 4.9045491470349315e-06, + "loss": 2.0749, + "step": 335 + }, + { + "epoch": 0.3682191780821918, + "grad_norm": 2.347121238708496, + "learning_rate": 4.902518277822909e-06, + "loss": 2.0502, + "step": 336 + }, + { + "epoch": 0.3693150684931507, + "grad_norm": 2.210817575454712, + "learning_rate": 4.900487408610886e-06, + "loss": 2.0612, + "step": 337 + }, + { + "epoch": 0.3704109589041096, + "grad_norm": 2.15594482421875, + "learning_rate": 4.898456539398863e-06, + "loss": 2.0653, + "step": 338 + }, + { + "epoch": 0.3715068493150685, + "grad_norm": 2.3996052742004395, + "learning_rate": 4.89642567018684e-06, + "loss": 2.074, + "step": 339 + }, + { + "epoch": 0.3726027397260274, + "grad_norm": 2.384172201156616, + "learning_rate": 4.894394800974818e-06, + "loss": 2.1533, + "step": 340 + }, + { + "epoch": 0.3726027397260274, + "eval_loss": 2.175034284591675, + "eval_runtime": 37.2455, + "eval_samples_per_second": 26.849, + "eval_steps_per_second": 1.128, + "step": 340 + }, + { + "epoch": 0.3736986301369863, + "grad_norm": 2.3431336879730225, + "learning_rate": 4.892363931762794e-06, + "loss": 1.9919, + "step": 341 + }, + { + "epoch": 0.3747945205479452, + "grad_norm": 2.1828389167785645, + "learning_rate": 4.8903330625507725e-06, + "loss": 2.0703, + "step": 342 + }, + { + "epoch": 0.37589041095890413, + "grad_norm": 2.2117249965667725, + "learning_rate": 4.888302193338749e-06, + "loss": 2.0071, + "step": 343 + }, + { + "epoch": 0.376986301369863, + "grad_norm": 2.3595967292785645, + "learning_rate": 4.886271324126727e-06, + "loss": 1.9879, + "step": 344 + }, + { + "epoch": 0.3780821917808219, + "grad_norm": 2.5211799144744873, + "learning_rate": 4.8842404549147034e-06, + "loss": 2.0861, + "step": 345 + }, + { + "epoch": 0.37917808219178084, + "grad_norm": 2.1551895141601562, + "learning_rate": 4.8822095857026816e-06, + "loss": 2.0683, + "step": 346 + }, + { + "epoch": 0.3802739726027397, + "grad_norm": 2.4145829677581787, + "learning_rate": 4.880178716490658e-06, + "loss": 2.043, + "step": 347 + }, + { + "epoch": 0.38136986301369863, + "grad_norm": 2.4273521900177, + "learning_rate": 4.878147847278636e-06, + "loss": 1.9928, + "step": 348 + }, + { + "epoch": 0.38246575342465755, + "grad_norm": 2.261054754257202, + "learning_rate": 4.8761169780666125e-06, + "loss": 2.0065, + "step": 349 + }, + { + "epoch": 0.3835616438356164, + "grad_norm": 2.4703147411346436, + "learning_rate": 4.874086108854591e-06, + "loss": 1.9766, + "step": 350 + }, + { + "epoch": 0.3835616438356164, + "eval_loss": 2.176915168762207, + "eval_runtime": 37.3223, + "eval_samples_per_second": 26.794, + "eval_steps_per_second": 1.125, + "step": 350 + }, + { + "epoch": 0.38465753424657534, + "grad_norm": 2.3353374004364014, + "learning_rate": 4.872055239642567e-06, + "loss": 1.9874, + "step": 351 + }, + { + "epoch": 0.38575342465753426, + "grad_norm": 2.227834463119507, + "learning_rate": 4.870024370430544e-06, + "loss": 1.9644, + "step": 352 + }, + { + "epoch": 0.3868493150684931, + "grad_norm": 2.3155159950256348, + "learning_rate": 4.867993501218522e-06, + "loss": 2.0227, + "step": 353 + }, + { + "epoch": 0.38794520547945205, + "grad_norm": 2.2456047534942627, + "learning_rate": 4.865962632006499e-06, + "loss": 2.0566, + "step": 354 + }, + { + "epoch": 0.38904109589041097, + "grad_norm": 2.3364651203155518, + "learning_rate": 4.863931762794476e-06, + "loss": 1.9941, + "step": 355 + }, + { + "epoch": 0.39013698630136984, + "grad_norm": 2.2886059284210205, + "learning_rate": 4.8619008935824535e-06, + "loss": 2.0163, + "step": 356 + }, + { + "epoch": 0.39123287671232876, + "grad_norm": 2.3796911239624023, + "learning_rate": 4.859870024370431e-06, + "loss": 2.0525, + "step": 357 + }, + { + "epoch": 0.3923287671232877, + "grad_norm": 2.381986379623413, + "learning_rate": 4.857839155158408e-06, + "loss": 2.0682, + "step": 358 + }, + { + "epoch": 0.3934246575342466, + "grad_norm": 2.2048425674438477, + "learning_rate": 4.855808285946385e-06, + "loss": 2.0748, + "step": 359 + }, + { + "epoch": 0.39452054794520547, + "grad_norm": 2.162846088409424, + "learning_rate": 4.8537774167343626e-06, + "loss": 2.0512, + "step": 360 + }, + { + "epoch": 0.39452054794520547, + "eval_loss": 2.1645994186401367, + "eval_runtime": 37.3126, + "eval_samples_per_second": 26.801, + "eval_steps_per_second": 1.126, + "step": 360 + }, + { + "epoch": 0.3956164383561644, + "grad_norm": 2.4070873260498047, + "learning_rate": 4.85174654752234e-06, + "loss": 2.042, + "step": 361 + }, + { + "epoch": 0.3967123287671233, + "grad_norm": 2.11716628074646, + "learning_rate": 4.849715678310317e-06, + "loss": 2.0398, + "step": 362 + }, + { + "epoch": 0.3978082191780822, + "grad_norm": 2.265479803085327, + "learning_rate": 4.847684809098294e-06, + "loss": 2.0334, + "step": 363 + }, + { + "epoch": 0.3989041095890411, + "grad_norm": 2.2270994186401367, + "learning_rate": 4.845653939886272e-06, + "loss": 2.0292, + "step": 364 + }, + { + "epoch": 0.4, + "grad_norm": 2.406064033508301, + "learning_rate": 4.843623070674249e-06, + "loss": 2.0351, + "step": 365 + }, + { + "epoch": 0.4010958904109589, + "grad_norm": 2.216654062271118, + "learning_rate": 4.841592201462226e-06, + "loss": 2.005, + "step": 366 + }, + { + "epoch": 0.4021917808219178, + "grad_norm": 2.4241385459899902, + "learning_rate": 4.8395613322502035e-06, + "loss": 2.0647, + "step": 367 + }, + { + "epoch": 0.40328767123287673, + "grad_norm": 2.4315145015716553, + "learning_rate": 4.837530463038181e-06, + "loss": 1.9679, + "step": 368 + }, + { + "epoch": 0.4043835616438356, + "grad_norm": 2.3034439086914062, + "learning_rate": 4.835499593826158e-06, + "loss": 2.0259, + "step": 369 + }, + { + "epoch": 0.4054794520547945, + "grad_norm": 2.493436098098755, + "learning_rate": 4.833468724614135e-06, + "loss": 2.0468, + "step": 370 + }, + { + "epoch": 0.4054794520547945, + "eval_loss": 2.161508083343506, + "eval_runtime": 37.2692, + "eval_samples_per_second": 26.832, + "eval_steps_per_second": 1.127, + "step": 370 + }, + { + "epoch": 0.40657534246575344, + "grad_norm": 2.3887157440185547, + "learning_rate": 4.831437855402113e-06, + "loss": 2.0313, + "step": 371 + }, + { + "epoch": 0.4076712328767123, + "grad_norm": 2.563095808029175, + "learning_rate": 4.82940698619009e-06, + "loss": 1.9288, + "step": 372 + }, + { + "epoch": 0.40876712328767123, + "grad_norm": 2.2609968185424805, + "learning_rate": 4.827376116978067e-06, + "loss": 1.9975, + "step": 373 + }, + { + "epoch": 0.40986301369863015, + "grad_norm": 2.624257802963257, + "learning_rate": 4.8253452477660444e-06, + "loss": 2.0639, + "step": 374 + }, + { + "epoch": 0.410958904109589, + "grad_norm": 2.502197265625, + "learning_rate": 4.823314378554022e-06, + "loss": 2.0424, + "step": 375 + }, + { + "epoch": 0.41205479452054794, + "grad_norm": 2.36297345161438, + "learning_rate": 4.821283509341998e-06, + "loss": 2.027, + "step": 376 + }, + { + "epoch": 0.41315068493150686, + "grad_norm": 2.388918161392212, + "learning_rate": 4.819252640129976e-06, + "loss": 2.0002, + "step": 377 + }, + { + "epoch": 0.41424657534246573, + "grad_norm": 2.4262335300445557, + "learning_rate": 4.817221770917953e-06, + "loss": 2.0215, + "step": 378 + }, + { + "epoch": 0.41534246575342465, + "grad_norm": 2.8953094482421875, + "learning_rate": 4.815190901705931e-06, + "loss": 1.988, + "step": 379 + }, + { + "epoch": 0.41643835616438357, + "grad_norm": 2.3092124462127686, + "learning_rate": 4.813160032493907e-06, + "loss": 2.0199, + "step": 380 + }, + { + "epoch": 0.41643835616438357, + "eval_loss": 2.1589064598083496, + "eval_runtime": 37.2649, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 380 + }, + { + "epoch": 0.41753424657534244, + "grad_norm": 3.0842413902282715, + "learning_rate": 4.811129163281885e-06, + "loss": 2.0134, + "step": 381 + }, + { + "epoch": 0.41863013698630136, + "grad_norm": 2.2015092372894287, + "learning_rate": 4.809098294069862e-06, + "loss": 2.0447, + "step": 382 + }, + { + "epoch": 0.4197260273972603, + "grad_norm": 2.93471097946167, + "learning_rate": 4.80706742485784e-06, + "loss": 2.0329, + "step": 383 + }, + { + "epoch": 0.4208219178082192, + "grad_norm": 2.3277227878570557, + "learning_rate": 4.805036555645816e-06, + "loss": 2.0805, + "step": 384 + }, + { + "epoch": 0.42191780821917807, + "grad_norm": 2.357966184616089, + "learning_rate": 4.8030056864337945e-06, + "loss": 2.0008, + "step": 385 + }, + { + "epoch": 0.423013698630137, + "grad_norm": 2.7505595684051514, + "learning_rate": 4.800974817221771e-06, + "loss": 2.059, + "step": 386 + }, + { + "epoch": 0.4241095890410959, + "grad_norm": 2.144338369369507, + "learning_rate": 4.798943948009749e-06, + "loss": 2.0125, + "step": 387 + }, + { + "epoch": 0.4252054794520548, + "grad_norm": 2.608407735824585, + "learning_rate": 4.7969130787977254e-06, + "loss": 2.0263, + "step": 388 + }, + { + "epoch": 0.4263013698630137, + "grad_norm": 2.4293761253356934, + "learning_rate": 4.794882209585703e-06, + "loss": 2.0107, + "step": 389 + }, + { + "epoch": 0.4273972602739726, + "grad_norm": 2.2136595249176025, + "learning_rate": 4.79285134037368e-06, + "loss": 2.0421, + "step": 390 + }, + { + "epoch": 0.4273972602739726, + "eval_loss": 2.1473641395568848, + "eval_runtime": 37.2895, + "eval_samples_per_second": 26.817, + "eval_steps_per_second": 1.126, + "step": 390 + }, + { + "epoch": 0.4284931506849315, + "grad_norm": 2.3706328868865967, + "learning_rate": 4.790820471161657e-06, + "loss": 1.9828, + "step": 391 + }, + { + "epoch": 0.4295890410958904, + "grad_norm": 2.1628777980804443, + "learning_rate": 4.7887896019496345e-06, + "loss": 2.0006, + "step": 392 + }, + { + "epoch": 0.43068493150684933, + "grad_norm": 2.5572149753570557, + "learning_rate": 4.786758732737612e-06, + "loss": 2.0046, + "step": 393 + }, + { + "epoch": 0.4317808219178082, + "grad_norm": 2.44661808013916, + "learning_rate": 4.784727863525589e-06, + "loss": 2.0513, + "step": 394 + }, + { + "epoch": 0.4328767123287671, + "grad_norm": 2.429487466812134, + "learning_rate": 4.782696994313566e-06, + "loss": 1.9619, + "step": 395 + }, + { + "epoch": 0.43397260273972604, + "grad_norm": 2.233353614807129, + "learning_rate": 4.780666125101544e-06, + "loss": 2.0141, + "step": 396 + }, + { + "epoch": 0.4350684931506849, + "grad_norm": 2.137549638748169, + "learning_rate": 4.778635255889521e-06, + "loss": 1.9512, + "step": 397 + }, + { + "epoch": 0.43616438356164383, + "grad_norm": 2.187638759613037, + "learning_rate": 4.776604386677498e-06, + "loss": 1.9802, + "step": 398 + }, + { + "epoch": 0.43726027397260275, + "grad_norm": 2.3084137439727783, + "learning_rate": 4.7745735174654755e-06, + "loss": 1.9887, + "step": 399 + }, + { + "epoch": 0.4383561643835616, + "grad_norm": 2.16654109954834, + "learning_rate": 4.772542648253453e-06, + "loss": 2.0535, + "step": 400 + }, + { + "epoch": 0.4383561643835616, + "eval_loss": 2.1444413661956787, + "eval_runtime": 37.2158, + "eval_samples_per_second": 26.87, + "eval_steps_per_second": 1.129, + "step": 400 + }, + { + "epoch": 0.43945205479452054, + "grad_norm": 2.0973072052001953, + "learning_rate": 4.77051177904143e-06, + "loss": 2.0529, + "step": 401 + }, + { + "epoch": 0.44054794520547946, + "grad_norm": 2.421881675720215, + "learning_rate": 4.768480909829407e-06, + "loss": 1.9893, + "step": 402 + }, + { + "epoch": 0.44164383561643833, + "grad_norm": 2.349296808242798, + "learning_rate": 4.766450040617385e-06, + "loss": 2.0237, + "step": 403 + }, + { + "epoch": 0.44273972602739725, + "grad_norm": 2.2154288291931152, + "learning_rate": 4.764419171405362e-06, + "loss": 1.996, + "step": 404 + }, + { + "epoch": 0.4438356164383562, + "grad_norm": 2.6355648040771484, + "learning_rate": 4.762388302193339e-06, + "loss": 2.0345, + "step": 405 + }, + { + "epoch": 0.4449315068493151, + "grad_norm": 2.352238655090332, + "learning_rate": 4.760357432981316e-06, + "loss": 2.0219, + "step": 406 + }, + { + "epoch": 0.44602739726027396, + "grad_norm": 2.1886510848999023, + "learning_rate": 4.758326563769294e-06, + "loss": 1.9923, + "step": 407 + }, + { + "epoch": 0.4471232876712329, + "grad_norm": 2.6483092308044434, + "learning_rate": 4.756295694557271e-06, + "loss": 2.0236, + "step": 408 + }, + { + "epoch": 0.4482191780821918, + "grad_norm": 2.382389545440674, + "learning_rate": 4.754264825345248e-06, + "loss": 2.0249, + "step": 409 + }, + { + "epoch": 0.44931506849315067, + "grad_norm": 2.4824411869049072, + "learning_rate": 4.7522339561332255e-06, + "loss": 1.9571, + "step": 410 + }, + { + "epoch": 0.44931506849315067, + "eval_loss": 2.1502621173858643, + "eval_runtime": 37.1789, + "eval_samples_per_second": 26.897, + "eval_steps_per_second": 1.13, + "step": 410 + }, + { + "epoch": 0.4504109589041096, + "grad_norm": 2.325589656829834, + "learning_rate": 4.750203086921203e-06, + "loss": 1.9753, + "step": 411 + }, + { + "epoch": 0.4515068493150685, + "grad_norm": 2.3618829250335693, + "learning_rate": 4.74817221770918e-06, + "loss": 1.9397, + "step": 412 + }, + { + "epoch": 0.4526027397260274, + "grad_norm": 2.3922839164733887, + "learning_rate": 4.7461413484971565e-06, + "loss": 2.0119, + "step": 413 + }, + { + "epoch": 0.4536986301369863, + "grad_norm": 2.403475046157837, + "learning_rate": 4.744110479285135e-06, + "loss": 2.0664, + "step": 414 + }, + { + "epoch": 0.4547945205479452, + "grad_norm": 2.225860834121704, + "learning_rate": 4.742079610073111e-06, + "loss": 1.9525, + "step": 415 + }, + { + "epoch": 0.4558904109589041, + "grad_norm": 2.344074010848999, + "learning_rate": 4.740048740861089e-06, + "loss": 1.9785, + "step": 416 + }, + { + "epoch": 0.456986301369863, + "grad_norm": 2.255382776260376, + "learning_rate": 4.738017871649066e-06, + "loss": 1.9902, + "step": 417 + }, + { + "epoch": 0.45808219178082193, + "grad_norm": 2.160628318786621, + "learning_rate": 4.735987002437044e-06, + "loss": 2.0075, + "step": 418 + }, + { + "epoch": 0.4591780821917808, + "grad_norm": 2.5667307376861572, + "learning_rate": 4.73395613322502e-06, + "loss": 2.0625, + "step": 419 + }, + { + "epoch": 0.4602739726027397, + "grad_norm": 2.3771040439605713, + "learning_rate": 4.731925264012998e-06, + "loss": 2.0724, + "step": 420 + }, + { + "epoch": 0.4602739726027397, + "eval_loss": 2.138829231262207, + "eval_runtime": 37.4361, + "eval_samples_per_second": 26.712, + "eval_steps_per_second": 1.122, + "step": 420 + }, + { + "epoch": 0.46136986301369864, + "grad_norm": 2.336482048034668, + "learning_rate": 4.729894394800975e-06, + "loss": 2.0234, + "step": 421 + }, + { + "epoch": 0.4624657534246575, + "grad_norm": 2.4206933975219727, + "learning_rate": 4.727863525588953e-06, + "loss": 1.9693, + "step": 422 + }, + { + "epoch": 0.46356164383561643, + "grad_norm": 2.4845645427703857, + "learning_rate": 4.725832656376929e-06, + "loss": 1.9239, + "step": 423 + }, + { + "epoch": 0.46465753424657535, + "grad_norm": 2.5199646949768066, + "learning_rate": 4.723801787164907e-06, + "loss": 1.9801, + "step": 424 + }, + { + "epoch": 0.4657534246575342, + "grad_norm": 2.2969748973846436, + "learning_rate": 4.721770917952885e-06, + "loss": 2.0192, + "step": 425 + }, + { + "epoch": 0.46684931506849314, + "grad_norm": 2.423503875732422, + "learning_rate": 4.719740048740861e-06, + "loss": 1.9684, + "step": 426 + }, + { + "epoch": 0.46794520547945206, + "grad_norm": 2.3681061267852783, + "learning_rate": 4.717709179528839e-06, + "loss": 1.9934, + "step": 427 + }, + { + "epoch": 0.469041095890411, + "grad_norm": 2.277965784072876, + "learning_rate": 4.715678310316816e-06, + "loss": 2.0166, + "step": 428 + }, + { + "epoch": 0.47013698630136985, + "grad_norm": 2.2730283737182617, + "learning_rate": 4.713647441104794e-06, + "loss": 2.011, + "step": 429 + }, + { + "epoch": 0.4712328767123288, + "grad_norm": 2.6694860458374023, + "learning_rate": 4.71161657189277e-06, + "loss": 1.9645, + "step": 430 + }, + { + "epoch": 0.4712328767123288, + "eval_loss": 2.1322011947631836, + "eval_runtime": 37.5723, + "eval_samples_per_second": 26.615, + "eval_steps_per_second": 1.118, + "step": 430 + }, + { + "epoch": 0.4723287671232877, + "grad_norm": 2.4100632667541504, + "learning_rate": 4.709585702680748e-06, + "loss": 1.9478, + "step": 431 + }, + { + "epoch": 0.47342465753424656, + "grad_norm": 2.2779541015625, + "learning_rate": 4.707554833468725e-06, + "loss": 2.0274, + "step": 432 + }, + { + "epoch": 0.4745205479452055, + "grad_norm": 2.55690336227417, + "learning_rate": 4.705523964256703e-06, + "loss": 2.0831, + "step": 433 + }, + { + "epoch": 0.4756164383561644, + "grad_norm": 2.455623149871826, + "learning_rate": 4.703493095044679e-06, + "loss": 1.9673, + "step": 434 + }, + { + "epoch": 0.4767123287671233, + "grad_norm": 2.2613699436187744, + "learning_rate": 4.701462225832657e-06, + "loss": 1.992, + "step": 435 + }, + { + "epoch": 0.4778082191780822, + "grad_norm": 2.4467933177948, + "learning_rate": 4.699431356620634e-06, + "loss": 1.9105, + "step": 436 + }, + { + "epoch": 0.4789041095890411, + "grad_norm": 2.4194023609161377, + "learning_rate": 4.697400487408612e-06, + "loss": 2.074, + "step": 437 + }, + { + "epoch": 0.48, + "grad_norm": 2.2165699005126953, + "learning_rate": 4.695369618196588e-06, + "loss": 2.0152, + "step": 438 + }, + { + "epoch": 0.4810958904109589, + "grad_norm": 2.1709225177764893, + "learning_rate": 4.693338748984566e-06, + "loss": 1.9888, + "step": 439 + }, + { + "epoch": 0.4821917808219178, + "grad_norm": 2.2325828075408936, + "learning_rate": 4.691307879772543e-06, + "loss": 1.9838, + "step": 440 + }, + { + "epoch": 0.4821917808219178, + "eval_loss": 2.128732919692993, + "eval_runtime": 37.205, + "eval_samples_per_second": 26.878, + "eval_steps_per_second": 1.129, + "step": 440 + }, + { + "epoch": 0.4832876712328767, + "grad_norm": 2.446331739425659, + "learning_rate": 4.68927701056052e-06, + "loss": 1.9295, + "step": 441 + }, + { + "epoch": 0.4843835616438356, + "grad_norm": 2.5459415912628174, + "learning_rate": 4.6872461413484975e-06, + "loss": 2.0351, + "step": 442 + }, + { + "epoch": 0.48547945205479454, + "grad_norm": 2.2877118587493896, + "learning_rate": 4.685215272136475e-06, + "loss": 2.0, + "step": 443 + }, + { + "epoch": 0.4865753424657534, + "grad_norm": 2.346625328063965, + "learning_rate": 4.683184402924452e-06, + "loss": 1.9928, + "step": 444 + }, + { + "epoch": 0.4876712328767123, + "grad_norm": 2.268467664718628, + "learning_rate": 4.681153533712429e-06, + "loss": 1.9969, + "step": 445 + }, + { + "epoch": 0.48876712328767125, + "grad_norm": 2.3699793815612793, + "learning_rate": 4.679122664500407e-06, + "loss": 2.0636, + "step": 446 + }, + { + "epoch": 0.4898630136986301, + "grad_norm": 2.2871041297912598, + "learning_rate": 4.677091795288384e-06, + "loss": 1.9535, + "step": 447 + }, + { + "epoch": 0.49095890410958903, + "grad_norm": 2.249393939971924, + "learning_rate": 4.675060926076361e-06, + "loss": 1.9358, + "step": 448 + }, + { + "epoch": 0.49205479452054796, + "grad_norm": 2.299863815307617, + "learning_rate": 4.673030056864338e-06, + "loss": 2.0299, + "step": 449 + }, + { + "epoch": 0.4931506849315068, + "grad_norm": 2.31239652633667, + "learning_rate": 4.670999187652316e-06, + "loss": 1.9372, + "step": 450 + }, + { + "epoch": 0.4931506849315068, + "eval_loss": 2.1267426013946533, + "eval_runtime": 37.1884, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 450 + }, + { + "epoch": 0.49424657534246574, + "grad_norm": 2.2824623584747314, + "learning_rate": 4.668968318440293e-06, + "loss": 1.9435, + "step": 451 + }, + { + "epoch": 0.49534246575342467, + "grad_norm": 2.285339593887329, + "learning_rate": 4.66693744922827e-06, + "loss": 1.9455, + "step": 452 + }, + { + "epoch": 0.4964383561643836, + "grad_norm": 2.346592903137207, + "learning_rate": 4.6649065800162475e-06, + "loss": 2.0212, + "step": 453 + }, + { + "epoch": 0.49753424657534245, + "grad_norm": 2.280669689178467, + "learning_rate": 4.662875710804225e-06, + "loss": 1.9136, + "step": 454 + }, + { + "epoch": 0.4986301369863014, + "grad_norm": 2.178434133529663, + "learning_rate": 4.660844841592202e-06, + "loss": 1.9952, + "step": 455 + }, + { + "epoch": 0.4997260273972603, + "grad_norm": 2.108492851257324, + "learning_rate": 4.658813972380179e-06, + "loss": 1.9022, + "step": 456 + }, + { + "epoch": 0.5008219178082192, + "grad_norm": 2.3862147331237793, + "learning_rate": 4.656783103168157e-06, + "loss": 2.0332, + "step": 457 + }, + { + "epoch": 0.5019178082191781, + "grad_norm": 2.2017860412597656, + "learning_rate": 4.654752233956134e-06, + "loss": 1.8998, + "step": 458 + }, + { + "epoch": 0.503013698630137, + "grad_norm": 2.222134828567505, + "learning_rate": 4.652721364744111e-06, + "loss": 1.9836, + "step": 459 + }, + { + "epoch": 0.5041095890410959, + "grad_norm": 2.2974510192871094, + "learning_rate": 4.6506904955320885e-06, + "loss": 1.9626, + "step": 460 + }, + { + "epoch": 0.5041095890410959, + "eval_loss": 2.1345443725585938, + "eval_runtime": 37.2785, + "eval_samples_per_second": 26.825, + "eval_steps_per_second": 1.127, + "step": 460 + }, + { + "epoch": 0.5052054794520547, + "grad_norm": 2.107480764389038, + "learning_rate": 4.648659626320066e-06, + "loss": 2.0061, + "step": 461 + }, + { + "epoch": 0.5063013698630137, + "grad_norm": 2.3541181087493896, + "learning_rate": 4.646628757108043e-06, + "loss": 2.0133, + "step": 462 + }, + { + "epoch": 0.5073972602739726, + "grad_norm": 2.215571880340576, + "learning_rate": 4.6445978878960194e-06, + "loss": 1.9727, + "step": 463 + }, + { + "epoch": 0.5084931506849315, + "grad_norm": 2.2263615131378174, + "learning_rate": 4.6425670186839976e-06, + "loss": 1.9477, + "step": 464 + }, + { + "epoch": 0.5095890410958904, + "grad_norm": 2.178900957107544, + "learning_rate": 4.640536149471974e-06, + "loss": 1.9842, + "step": 465 + }, + { + "epoch": 0.5106849315068493, + "grad_norm": 2.3703105449676514, + "learning_rate": 4.638505280259952e-06, + "loss": 2.0082, + "step": 466 + }, + { + "epoch": 0.5117808219178083, + "grad_norm": 2.1587533950805664, + "learning_rate": 4.6364744110479285e-06, + "loss": 2.0271, + "step": 467 + }, + { + "epoch": 0.5128767123287671, + "grad_norm": 2.1626017093658447, + "learning_rate": 4.634443541835907e-06, + "loss": 1.9858, + "step": 468 + }, + { + "epoch": 0.513972602739726, + "grad_norm": 2.326869487762451, + "learning_rate": 4.632412672623883e-06, + "loss": 2.0413, + "step": 469 + }, + { + "epoch": 0.5150684931506849, + "grad_norm": 2.266502857208252, + "learning_rate": 4.630381803411861e-06, + "loss": 1.9584, + "step": 470 + }, + { + "epoch": 0.5150684931506849, + "eval_loss": 2.1273915767669678, + "eval_runtime": 37.2571, + "eval_samples_per_second": 26.841, + "eval_steps_per_second": 1.127, + "step": 470 + }, + { + "epoch": 0.5161643835616438, + "grad_norm": 2.131348133087158, + "learning_rate": 4.628350934199838e-06, + "loss": 1.863, + "step": 471 + }, + { + "epoch": 0.5172602739726028, + "grad_norm": 2.2635257244110107, + "learning_rate": 4.626320064987816e-06, + "loss": 1.9699, + "step": 472 + }, + { + "epoch": 0.5183561643835617, + "grad_norm": 2.1303744316101074, + "learning_rate": 4.624289195775792e-06, + "loss": 2.0001, + "step": 473 + }, + { + "epoch": 0.5194520547945205, + "grad_norm": 2.1736788749694824, + "learning_rate": 4.62225832656377e-06, + "loss": 1.9684, + "step": 474 + }, + { + "epoch": 0.5205479452054794, + "grad_norm": 2.1996278762817383, + "learning_rate": 4.620227457351747e-06, + "loss": 1.9739, + "step": 475 + }, + { + "epoch": 0.5216438356164383, + "grad_norm": 2.225456953048706, + "learning_rate": 4.618196588139724e-06, + "loss": 1.9649, + "step": 476 + }, + { + "epoch": 0.5227397260273973, + "grad_norm": 2.118537187576294, + "learning_rate": 4.616165718927701e-06, + "loss": 1.9911, + "step": 477 + }, + { + "epoch": 0.5238356164383562, + "grad_norm": 2.232940435409546, + "learning_rate": 4.6141348497156786e-06, + "loss": 1.944, + "step": 478 + }, + { + "epoch": 0.5249315068493151, + "grad_norm": 2.1776034832000732, + "learning_rate": 4.612103980503656e-06, + "loss": 2.0342, + "step": 479 + }, + { + "epoch": 0.5260273972602739, + "grad_norm": 2.042645215988159, + "learning_rate": 4.610073111291633e-06, + "loss": 2.0124, + "step": 480 + }, + { + "epoch": 0.5260273972602739, + "eval_loss": 2.118508815765381, + "eval_runtime": 37.2764, + "eval_samples_per_second": 26.827, + "eval_steps_per_second": 1.127, + "step": 480 + }, + { + "epoch": 0.5271232876712328, + "grad_norm": 2.302661895751953, + "learning_rate": 4.60804224207961e-06, + "loss": 1.9874, + "step": 481 + }, + { + "epoch": 0.5282191780821918, + "grad_norm": 2.1251118183135986, + "learning_rate": 4.606011372867588e-06, + "loss": 1.9594, + "step": 482 + }, + { + "epoch": 0.5293150684931507, + "grad_norm": 2.157043933868408, + "learning_rate": 4.603980503655565e-06, + "loss": 1.9955, + "step": 483 + }, + { + "epoch": 0.5304109589041096, + "grad_norm": 2.291428804397583, + "learning_rate": 4.601949634443542e-06, + "loss": 1.9695, + "step": 484 + }, + { + "epoch": 0.5315068493150685, + "grad_norm": 2.2341537475585938, + "learning_rate": 4.5999187652315195e-06, + "loss": 1.899, + "step": 485 + }, + { + "epoch": 0.5326027397260275, + "grad_norm": 2.1881206035614014, + "learning_rate": 4.597887896019497e-06, + "loss": 1.9066, + "step": 486 + }, + { + "epoch": 0.5336986301369863, + "grad_norm": 2.3057658672332764, + "learning_rate": 4.595857026807474e-06, + "loss": 1.9646, + "step": 487 + }, + { + "epoch": 0.5347945205479452, + "grad_norm": 2.2318356037139893, + "learning_rate": 4.593826157595451e-06, + "loss": 1.9662, + "step": 488 + }, + { + "epoch": 0.5358904109589041, + "grad_norm": 2.170509099960327, + "learning_rate": 4.591795288383429e-06, + "loss": 1.9522, + "step": 489 + }, + { + "epoch": 0.536986301369863, + "grad_norm": 2.101646661758423, + "learning_rate": 4.589764419171406e-06, + "loss": 2.0315, + "step": 490 + }, + { + "epoch": 0.536986301369863, + "eval_loss": 2.113968849182129, + "eval_runtime": 37.2751, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 490 + }, + { + "epoch": 0.538082191780822, + "grad_norm": 2.1563706398010254, + "learning_rate": 4.587733549959383e-06, + "loss": 1.9843, + "step": 491 + }, + { + "epoch": 0.5391780821917809, + "grad_norm": 2.295722246170044, + "learning_rate": 4.5857026807473604e-06, + "loss": 1.9499, + "step": 492 + }, + { + "epoch": 0.5402739726027397, + "grad_norm": 1.974424123764038, + "learning_rate": 4.583671811535338e-06, + "loss": 1.9346, + "step": 493 + }, + { + "epoch": 0.5413698630136986, + "grad_norm": 2.4402668476104736, + "learning_rate": 4.581640942323315e-06, + "loss": 1.9053, + "step": 494 + }, + { + "epoch": 0.5424657534246575, + "grad_norm": 2.269000768661499, + "learning_rate": 4.579610073111292e-06, + "loss": 2.0108, + "step": 495 + }, + { + "epoch": 0.5435616438356164, + "grad_norm": 2.1182594299316406, + "learning_rate": 4.5775792038992695e-06, + "loss": 2.0155, + "step": 496 + }, + { + "epoch": 0.5446575342465754, + "grad_norm": 2.2181780338287354, + "learning_rate": 4.575548334687247e-06, + "loss": 2.0112, + "step": 497 + }, + { + "epoch": 0.5457534246575343, + "grad_norm": 2.2806756496429443, + "learning_rate": 4.573517465475224e-06, + "loss": 2.0418, + "step": 498 + }, + { + "epoch": 0.5468493150684931, + "grad_norm": 2.2690632343292236, + "learning_rate": 4.571486596263201e-06, + "loss": 1.9457, + "step": 499 + }, + { + "epoch": 0.547945205479452, + "grad_norm": 2.361335515975952, + "learning_rate": 4.569455727051178e-06, + "loss": 1.9949, + "step": 500 + }, + { + "epoch": 0.547945205479452, + "eval_loss": 2.1165614128112793, + "eval_runtime": 37.2664, + "eval_samples_per_second": 26.834, + "eval_steps_per_second": 1.127, + "step": 500 + }, + { + "epoch": 0.549041095890411, + "grad_norm": 2.3833138942718506, + "learning_rate": 4.567424857839156e-06, + "loss": 1.9932, + "step": 501 + }, + { + "epoch": 0.5501369863013699, + "grad_norm": 2.328500509262085, + "learning_rate": 4.565393988627132e-06, + "loss": 1.9773, + "step": 502 + }, + { + "epoch": 0.5512328767123288, + "grad_norm": 2.4191694259643555, + "learning_rate": 4.5633631194151105e-06, + "loss": 2.0389, + "step": 503 + }, + { + "epoch": 0.5523287671232877, + "grad_norm": 2.4106884002685547, + "learning_rate": 4.561332250203087e-06, + "loss": 1.965, + "step": 504 + }, + { + "epoch": 0.5534246575342465, + "grad_norm": 2.1452605724334717, + "learning_rate": 4.559301380991065e-06, + "loss": 1.9713, + "step": 505 + }, + { + "epoch": 0.5545205479452054, + "grad_norm": 2.2753357887268066, + "learning_rate": 4.5572705117790414e-06, + "loss": 1.9116, + "step": 506 + }, + { + "epoch": 0.5556164383561644, + "grad_norm": 2.6925032138824463, + "learning_rate": 4.5552396425670196e-06, + "loss": 1.9593, + "step": 507 + }, + { + "epoch": 0.5567123287671233, + "grad_norm": 2.3136658668518066, + "learning_rate": 4.553208773354996e-06, + "loss": 2.0072, + "step": 508 + }, + { + "epoch": 0.5578082191780822, + "grad_norm": 2.254849672317505, + "learning_rate": 4.551177904142974e-06, + "loss": 1.9193, + "step": 509 + }, + { + "epoch": 0.5589041095890411, + "grad_norm": 2.352755546569824, + "learning_rate": 4.5491470349309505e-06, + "loss": 2.0228, + "step": 510 + }, + { + "epoch": 0.5589041095890411, + "eval_loss": 2.114323377609253, + "eval_runtime": 37.3629, + "eval_samples_per_second": 26.764, + "eval_steps_per_second": 1.124, + "step": 510 + }, + { + "epoch": 0.56, + "grad_norm": 2.31339955329895, + "learning_rate": 4.547116165718929e-06, + "loss": 1.9416, + "step": 511 + }, + { + "epoch": 0.5610958904109589, + "grad_norm": 2.191164970397949, + "learning_rate": 4.545085296506905e-06, + "loss": 1.9709, + "step": 512 + }, + { + "epoch": 0.5621917808219178, + "grad_norm": 2.193551778793335, + "learning_rate": 4.543054427294882e-06, + "loss": 1.9475, + "step": 513 + }, + { + "epoch": 0.5632876712328767, + "grad_norm": 2.222710371017456, + "learning_rate": 4.54102355808286e-06, + "loss": 1.9283, + "step": 514 + }, + { + "epoch": 0.5643835616438356, + "grad_norm": 2.078907012939453, + "learning_rate": 4.538992688870837e-06, + "loss": 1.929, + "step": 515 + }, + { + "epoch": 0.5654794520547946, + "grad_norm": 2.2043654918670654, + "learning_rate": 4.536961819658814e-06, + "loss": 1.9591, + "step": 516 + }, + { + "epoch": 0.5665753424657535, + "grad_norm": 2.192481517791748, + "learning_rate": 4.5349309504467915e-06, + "loss": 1.9618, + "step": 517 + }, + { + "epoch": 0.5676712328767123, + "grad_norm": 2.3509292602539062, + "learning_rate": 4.532900081234769e-06, + "loss": 1.9782, + "step": 518 + }, + { + "epoch": 0.5687671232876712, + "grad_norm": 2.2415924072265625, + "learning_rate": 4.530869212022746e-06, + "loss": 1.9456, + "step": 519 + }, + { + "epoch": 0.5698630136986301, + "grad_norm": 2.2135009765625, + "learning_rate": 4.528838342810723e-06, + "loss": 1.9671, + "step": 520 + }, + { + "epoch": 0.5698630136986301, + "eval_loss": 2.109738826751709, + "eval_runtime": 37.5253, + "eval_samples_per_second": 26.649, + "eval_steps_per_second": 1.119, + "step": 520 + }, + { + "epoch": 0.570958904109589, + "grad_norm": 2.3513574600219727, + "learning_rate": 4.526807473598701e-06, + "loss": 1.9625, + "step": 521 + }, + { + "epoch": 0.572054794520548, + "grad_norm": 1.9949424266815186, + "learning_rate": 4.524776604386678e-06, + "loss": 1.9621, + "step": 522 + }, + { + "epoch": 0.5731506849315069, + "grad_norm": 2.199488401412964, + "learning_rate": 4.522745735174655e-06, + "loss": 1.9389, + "step": 523 + }, + { + "epoch": 0.5742465753424657, + "grad_norm": 2.2119452953338623, + "learning_rate": 4.520714865962632e-06, + "loss": 1.9936, + "step": 524 + }, + { + "epoch": 0.5753424657534246, + "grad_norm": 2.242812156677246, + "learning_rate": 4.51868399675061e-06, + "loss": 1.9728, + "step": 525 + }, + { + "epoch": 0.5764383561643835, + "grad_norm": 2.1713850498199463, + "learning_rate": 4.516653127538587e-06, + "loss": 1.98, + "step": 526 + }, + { + "epoch": 0.5775342465753425, + "grad_norm": 2.261742353439331, + "learning_rate": 4.514622258326564e-06, + "loss": 1.9412, + "step": 527 + }, + { + "epoch": 0.5786301369863014, + "grad_norm": 2.222663402557373, + "learning_rate": 4.5125913891145415e-06, + "loss": 1.9266, + "step": 528 + }, + { + "epoch": 0.5797260273972603, + "grad_norm": 2.1108906269073486, + "learning_rate": 4.510560519902519e-06, + "loss": 1.9539, + "step": 529 + }, + { + "epoch": 0.5808219178082191, + "grad_norm": 1.9594205617904663, + "learning_rate": 4.508529650690496e-06, + "loss": 1.9842, + "step": 530 + }, + { + "epoch": 0.5808219178082191, + "eval_loss": 2.1073970794677734, + "eval_runtime": 37.6167, + "eval_samples_per_second": 26.584, + "eval_steps_per_second": 1.117, + "step": 530 + }, + { + "epoch": 0.581917808219178, + "grad_norm": 2.1339495182037354, + "learning_rate": 4.506498781478473e-06, + "loss": 1.9095, + "step": 531 + }, + { + "epoch": 0.583013698630137, + "grad_norm": 2.116248607635498, + "learning_rate": 4.504467912266451e-06, + "loss": 1.9091, + "step": 532 + }, + { + "epoch": 0.5841095890410959, + "grad_norm": 2.063962459564209, + "learning_rate": 4.502437043054428e-06, + "loss": 1.9537, + "step": 533 + }, + { + "epoch": 0.5852054794520548, + "grad_norm": 2.162569522857666, + "learning_rate": 4.500406173842405e-06, + "loss": 2.0253, + "step": 534 + }, + { + "epoch": 0.5863013698630137, + "grad_norm": 2.1783549785614014, + "learning_rate": 4.4983753046303824e-06, + "loss": 1.9714, + "step": 535 + }, + { + "epoch": 0.5873972602739727, + "grad_norm": 2.0791046619415283, + "learning_rate": 4.49634443541836e-06, + "loss": 1.9629, + "step": 536 + }, + { + "epoch": 0.5884931506849315, + "grad_norm": 2.3923251628875732, + "learning_rate": 4.494313566206336e-06, + "loss": 1.9445, + "step": 537 + }, + { + "epoch": 0.5895890410958904, + "grad_norm": 2.1848347187042236, + "learning_rate": 4.492282696994314e-06, + "loss": 2.007, + "step": 538 + }, + { + "epoch": 0.5906849315068493, + "grad_norm": 2.0824925899505615, + "learning_rate": 4.490251827782291e-06, + "loss": 1.9409, + "step": 539 + }, + { + "epoch": 0.5917808219178082, + "grad_norm": 2.057790756225586, + "learning_rate": 4.488220958570269e-06, + "loss": 1.9061, + "step": 540 + }, + { + "epoch": 0.5917808219178082, + "eval_loss": 2.1101415157318115, + "eval_runtime": 37.2235, + "eval_samples_per_second": 26.865, + "eval_steps_per_second": 1.128, + "step": 540 + }, + { + "epoch": 0.5928767123287672, + "grad_norm": 2.3652148246765137, + "learning_rate": 4.486190089358245e-06, + "loss": 1.9379, + "step": 541 + }, + { + "epoch": 0.5939726027397261, + "grad_norm": 2.195535659790039, + "learning_rate": 4.484159220146223e-06, + "loss": 1.8881, + "step": 542 + }, + { + "epoch": 0.5950684931506849, + "grad_norm": 2.258000135421753, + "learning_rate": 4.4821283509342e-06, + "loss": 1.9923, + "step": 543 + }, + { + "epoch": 0.5961643835616438, + "grad_norm": 2.097041606903076, + "learning_rate": 4.480097481722178e-06, + "loss": 1.9715, + "step": 544 + }, + { + "epoch": 0.5972602739726027, + "grad_norm": 2.1741156578063965, + "learning_rate": 4.478066612510154e-06, + "loss": 1.9487, + "step": 545 + }, + { + "epoch": 0.5983561643835617, + "grad_norm": 2.135338306427002, + "learning_rate": 4.4760357432981325e-06, + "loss": 1.9172, + "step": 546 + }, + { + "epoch": 0.5994520547945206, + "grad_norm": 2.3435418605804443, + "learning_rate": 4.474004874086109e-06, + "loss": 1.945, + "step": 547 + }, + { + "epoch": 0.6005479452054795, + "grad_norm": 2.2242488861083984, + "learning_rate": 4.471974004874087e-06, + "loss": 2.0717, + "step": 548 + }, + { + "epoch": 0.6016438356164383, + "grad_norm": 2.3371622562408447, + "learning_rate": 4.4699431356620635e-06, + "loss": 1.9082, + "step": 549 + }, + { + "epoch": 0.6027397260273972, + "grad_norm": 2.176064968109131, + "learning_rate": 4.467912266450041e-06, + "loss": 2.0154, + "step": 550 + }, + { + "epoch": 0.6027397260273972, + "eval_loss": 2.105236291885376, + "eval_runtime": 37.1885, + "eval_samples_per_second": 26.89, + "eval_steps_per_second": 1.129, + "step": 550 + }, + { + "epoch": 0.6038356164383561, + "grad_norm": 2.131166934967041, + "learning_rate": 4.465881397238018e-06, + "loss": 1.9575, + "step": 551 + }, + { + "epoch": 0.6049315068493151, + "grad_norm": 2.094514846801758, + "learning_rate": 4.463850528025995e-06, + "loss": 1.9544, + "step": 552 + }, + { + "epoch": 0.606027397260274, + "grad_norm": 2.0827765464782715, + "learning_rate": 4.4618196588139726e-06, + "loss": 1.9539, + "step": 553 + }, + { + "epoch": 0.6071232876712329, + "grad_norm": 2.225749969482422, + "learning_rate": 4.45978878960195e-06, + "loss": 1.983, + "step": 554 + }, + { + "epoch": 0.6082191780821918, + "grad_norm": 2.1363680362701416, + "learning_rate": 4.457757920389927e-06, + "loss": 1.9769, + "step": 555 + }, + { + "epoch": 0.6093150684931506, + "grad_norm": 2.238170623779297, + "learning_rate": 4.455727051177904e-06, + "loss": 1.9094, + "step": 556 + }, + { + "epoch": 0.6104109589041096, + "grad_norm": 2.228384494781494, + "learning_rate": 4.453696181965882e-06, + "loss": 1.9464, + "step": 557 + }, + { + "epoch": 0.6115068493150685, + "grad_norm": 2.479428768157959, + "learning_rate": 4.451665312753859e-06, + "loss": 2.0017, + "step": 558 + }, + { + "epoch": 0.6126027397260274, + "grad_norm": 2.2331902980804443, + "learning_rate": 4.449634443541836e-06, + "loss": 1.9728, + "step": 559 + }, + { + "epoch": 0.6136986301369863, + "grad_norm": 2.1728556156158447, + "learning_rate": 4.4476035743298135e-06, + "loss": 2.0153, + "step": 560 + }, + { + "epoch": 0.6136986301369863, + "eval_loss": 2.0978519916534424, + "eval_runtime": 37.233, + "eval_samples_per_second": 26.858, + "eval_steps_per_second": 1.128, + "step": 560 + }, + { + "epoch": 0.6147945205479453, + "grad_norm": 2.377814292907715, + "learning_rate": 4.445572705117791e-06, + "loss": 1.9242, + "step": 561 + }, + { + "epoch": 0.6158904109589041, + "grad_norm": 2.190436601638794, + "learning_rate": 4.443541835905768e-06, + "loss": 1.9454, + "step": 562 + }, + { + "epoch": 0.616986301369863, + "grad_norm": 2.052201986312866, + "learning_rate": 4.441510966693745e-06, + "loss": 2.0048, + "step": 563 + }, + { + "epoch": 0.6180821917808219, + "grad_norm": 2.2706589698791504, + "learning_rate": 4.439480097481723e-06, + "loss": 1.9426, + "step": 564 + }, + { + "epoch": 0.6191780821917808, + "grad_norm": 2.3478105068206787, + "learning_rate": 4.4374492282697e-06, + "loss": 2.0047, + "step": 565 + }, + { + "epoch": 0.6202739726027398, + "grad_norm": 2.465078115463257, + "learning_rate": 4.435418359057677e-06, + "loss": 1.8739, + "step": 566 + }, + { + "epoch": 0.6213698630136987, + "grad_norm": 2.0979490280151367, + "learning_rate": 4.433387489845654e-06, + "loss": 1.9588, + "step": 567 + }, + { + "epoch": 0.6224657534246575, + "grad_norm": 2.1239588260650635, + "learning_rate": 4.431356620633632e-06, + "loss": 1.9695, + "step": 568 + }, + { + "epoch": 0.6235616438356164, + "grad_norm": 2.014007806777954, + "learning_rate": 4.429325751421609e-06, + "loss": 1.9232, + "step": 569 + }, + { + "epoch": 0.6246575342465753, + "grad_norm": 2.212414026260376, + "learning_rate": 4.427294882209586e-06, + "loss": 1.928, + "step": 570 + }, + { + "epoch": 0.6246575342465753, + "eval_loss": 2.103149890899658, + "eval_runtime": 37.2641, + "eval_samples_per_second": 26.835, + "eval_steps_per_second": 1.127, + "step": 570 + }, + { + "epoch": 0.6257534246575343, + "grad_norm": 2.0944411754608154, + "learning_rate": 4.4252640129975635e-06, + "loss": 1.9568, + "step": 571 + }, + { + "epoch": 0.6268493150684932, + "grad_norm": 2.1068055629730225, + "learning_rate": 4.423233143785541e-06, + "loss": 1.8917, + "step": 572 + }, + { + "epoch": 0.6279452054794521, + "grad_norm": 2.1992151737213135, + "learning_rate": 4.421202274573518e-06, + "loss": 1.9536, + "step": 573 + }, + { + "epoch": 0.6290410958904109, + "grad_norm": 2.2079834938049316, + "learning_rate": 4.4191714053614945e-06, + "loss": 1.9603, + "step": 574 + }, + { + "epoch": 0.6301369863013698, + "grad_norm": 2.1230316162109375, + "learning_rate": 4.417140536149473e-06, + "loss": 1.8529, + "step": 575 + }, + { + "epoch": 0.6312328767123287, + "grad_norm": 2.1069579124450684, + "learning_rate": 4.415109666937449e-06, + "loss": 1.8997, + "step": 576 + }, + { + "epoch": 0.6323287671232877, + "grad_norm": 2.1683409214019775, + "learning_rate": 4.413078797725427e-06, + "loss": 1.971, + "step": 577 + }, + { + "epoch": 0.6334246575342466, + "grad_norm": 2.0599024295806885, + "learning_rate": 4.411047928513404e-06, + "loss": 1.9154, + "step": 578 + }, + { + "epoch": 0.6345205479452055, + "grad_norm": 2.1605849266052246, + "learning_rate": 4.409017059301382e-06, + "loss": 1.9562, + "step": 579 + }, + { + "epoch": 0.6356164383561644, + "grad_norm": 2.151249408721924, + "learning_rate": 4.406986190089358e-06, + "loss": 1.9571, + "step": 580 + }, + { + "epoch": 0.6356164383561644, + "eval_loss": 2.098177671432495, + "eval_runtime": 37.2578, + "eval_samples_per_second": 26.84, + "eval_steps_per_second": 1.127, + "step": 580 + }, + { + "epoch": 0.6367123287671232, + "grad_norm": 2.1819236278533936, + "learning_rate": 4.404955320877336e-06, + "loss": 1.9816, + "step": 581 + }, + { + "epoch": 0.6378082191780822, + "grad_norm": 2.1131205558776855, + "learning_rate": 4.402924451665313e-06, + "loss": 1.9738, + "step": 582 + }, + { + "epoch": 0.6389041095890411, + "grad_norm": 2.0852456092834473, + "learning_rate": 4.400893582453291e-06, + "loss": 1.924, + "step": 583 + }, + { + "epoch": 0.64, + "grad_norm": 2.0997941493988037, + "learning_rate": 4.398862713241267e-06, + "loss": 1.9067, + "step": 584 + }, + { + "epoch": 0.6410958904109589, + "grad_norm": 2.0848538875579834, + "learning_rate": 4.396831844029245e-06, + "loss": 1.9393, + "step": 585 + }, + { + "epoch": 0.6421917808219179, + "grad_norm": 2.2121758460998535, + "learning_rate": 4.394800974817222e-06, + "loss": 1.9458, + "step": 586 + }, + { + "epoch": 0.6432876712328767, + "grad_norm": 2.194462299346924, + "learning_rate": 4.392770105605199e-06, + "loss": 1.9528, + "step": 587 + }, + { + "epoch": 0.6443835616438356, + "grad_norm": 2.1936254501342773, + "learning_rate": 4.390739236393176e-06, + "loss": 1.9584, + "step": 588 + }, + { + "epoch": 0.6454794520547945, + "grad_norm": 2.1685783863067627, + "learning_rate": 4.388708367181154e-06, + "loss": 1.9666, + "step": 589 + }, + { + "epoch": 0.6465753424657534, + "grad_norm": 2.334571123123169, + "learning_rate": 4.386677497969131e-06, + "loss": 1.9748, + "step": 590 + }, + { + "epoch": 0.6465753424657534, + "eval_loss": 2.0937106609344482, + "eval_runtime": 37.2877, + "eval_samples_per_second": 26.818, + "eval_steps_per_second": 1.126, + "step": 590 + }, + { + "epoch": 0.6476712328767124, + "grad_norm": 2.2901110649108887, + "learning_rate": 4.384646628757108e-06, + "loss": 1.9597, + "step": 591 + }, + { + "epoch": 0.6487671232876713, + "grad_norm": 2.2364838123321533, + "learning_rate": 4.3826157595450855e-06, + "loss": 2.004, + "step": 592 + }, + { + "epoch": 0.6498630136986301, + "grad_norm": 2.1088364124298096, + "learning_rate": 4.380584890333063e-06, + "loss": 1.9319, + "step": 593 + }, + { + "epoch": 0.650958904109589, + "grad_norm": 2.164210319519043, + "learning_rate": 4.37855402112104e-06, + "loss": 1.8906, + "step": 594 + }, + { + "epoch": 0.6520547945205479, + "grad_norm": 2.205200672149658, + "learning_rate": 4.376523151909017e-06, + "loss": 1.8474, + "step": 595 + }, + { + "epoch": 0.6531506849315069, + "grad_norm": 2.239530086517334, + "learning_rate": 4.3744922826969946e-06, + "loss": 2.0033, + "step": 596 + }, + { + "epoch": 0.6542465753424658, + "grad_norm": 2.214747190475464, + "learning_rate": 4.372461413484972e-06, + "loss": 1.9258, + "step": 597 + }, + { + "epoch": 0.6553424657534247, + "grad_norm": 2.1746792793273926, + "learning_rate": 4.370430544272949e-06, + "loss": 1.9479, + "step": 598 + }, + { + "epoch": 0.6564383561643835, + "grad_norm": 2.194131851196289, + "learning_rate": 4.368399675060926e-06, + "loss": 1.9825, + "step": 599 + }, + { + "epoch": 0.6575342465753424, + "grad_norm": 2.3334717750549316, + "learning_rate": 4.366368805848904e-06, + "loss": 1.9831, + "step": 600 + }, + { + "epoch": 0.6575342465753424, + "eval_loss": 2.0881216526031494, + "eval_runtime": 37.3226, + "eval_samples_per_second": 26.793, + "eval_steps_per_second": 1.125, + "step": 600 + }, + { + "epoch": 0.6586301369863014, + "grad_norm": 2.269484758377075, + "learning_rate": 4.364337936636881e-06, + "loss": 1.9407, + "step": 601 + }, + { + "epoch": 0.6597260273972603, + "grad_norm": 2.470773935317993, + "learning_rate": 4.362307067424858e-06, + "loss": 1.8905, + "step": 602 + }, + { + "epoch": 0.6608219178082192, + "grad_norm": 2.485574960708618, + "learning_rate": 4.3602761982128355e-06, + "loss": 1.953, + "step": 603 + }, + { + "epoch": 0.6619178082191781, + "grad_norm": 2.2443618774414062, + "learning_rate": 4.358245329000813e-06, + "loss": 1.9244, + "step": 604 + }, + { + "epoch": 0.663013698630137, + "grad_norm": 2.039600372314453, + "learning_rate": 4.35621445978879e-06, + "loss": 1.9364, + "step": 605 + }, + { + "epoch": 0.6641095890410958, + "grad_norm": 2.093001127243042, + "learning_rate": 4.354183590576767e-06, + "loss": 1.9439, + "step": 606 + }, + { + "epoch": 0.6652054794520548, + "grad_norm": 2.3433260917663574, + "learning_rate": 4.352152721364745e-06, + "loss": 1.9322, + "step": 607 + }, + { + "epoch": 0.6663013698630137, + "grad_norm": 2.2403812408447266, + "learning_rate": 4.350121852152722e-06, + "loss": 1.8829, + "step": 608 + }, + { + "epoch": 0.6673972602739726, + "grad_norm": 2.1542811393737793, + "learning_rate": 4.348090982940699e-06, + "loss": 1.9549, + "step": 609 + }, + { + "epoch": 0.6684931506849315, + "grad_norm": 2.1526548862457275, + "learning_rate": 4.3460601137286764e-06, + "loss": 1.9433, + "step": 610 + }, + { + "epoch": 0.6684931506849315, + "eval_loss": 2.083638906478882, + "eval_runtime": 37.2677, + "eval_samples_per_second": 26.833, + "eval_steps_per_second": 1.127, + "step": 610 + }, + { + "epoch": 0.6695890410958905, + "grad_norm": 2.093421459197998, + "learning_rate": 4.344029244516654e-06, + "loss": 1.9643, + "step": 611 + }, + { + "epoch": 0.6706849315068493, + "grad_norm": 2.2205088138580322, + "learning_rate": 4.341998375304631e-06, + "loss": 1.9309, + "step": 612 + }, + { + "epoch": 0.6717808219178082, + "grad_norm": 2.059763193130493, + "learning_rate": 4.339967506092607e-06, + "loss": 2.0064, + "step": 613 + }, + { + "epoch": 0.6728767123287671, + "grad_norm": 2.0340707302093506, + "learning_rate": 4.3379366368805855e-06, + "loss": 1.9394, + "step": 614 + }, + { + "epoch": 0.673972602739726, + "grad_norm": 2.0716137886047363, + "learning_rate": 4.335905767668562e-06, + "loss": 1.9421, + "step": 615 + }, + { + "epoch": 0.675068493150685, + "grad_norm": 2.176809549331665, + "learning_rate": 4.33387489845654e-06, + "loss": 1.9261, + "step": 616 + }, + { + "epoch": 0.6761643835616439, + "grad_norm": 2.144385814666748, + "learning_rate": 4.3318440292445165e-06, + "loss": 1.9635, + "step": 617 + }, + { + "epoch": 0.6772602739726027, + "grad_norm": 2.0949580669403076, + "learning_rate": 4.329813160032495e-06, + "loss": 1.9417, + "step": 618 + }, + { + "epoch": 0.6783561643835616, + "grad_norm": 2.1727259159088135, + "learning_rate": 4.327782290820471e-06, + "loss": 1.9041, + "step": 619 + }, + { + "epoch": 0.6794520547945205, + "grad_norm": 2.012868881225586, + "learning_rate": 4.325751421608449e-06, + "loss": 1.8973, + "step": 620 + }, + { + "epoch": 0.6794520547945205, + "eval_loss": 2.0839755535125732, + "eval_runtime": 37.5302, + "eval_samples_per_second": 26.645, + "eval_steps_per_second": 1.119, + "step": 620 + }, + { + "epoch": 0.6805479452054795, + "grad_norm": 2.204153299331665, + "learning_rate": 4.323720552396426e-06, + "loss": 1.9719, + "step": 621 + }, + { + "epoch": 0.6816438356164384, + "grad_norm": 2.2421770095825195, + "learning_rate": 4.321689683184404e-06, + "loss": 1.9207, + "step": 622 + }, + { + "epoch": 0.6827397260273973, + "grad_norm": 2.2263410091400146, + "learning_rate": 4.31965881397238e-06, + "loss": 1.9146, + "step": 623 + }, + { + "epoch": 0.6838356164383562, + "grad_norm": 2.2015221118927, + "learning_rate": 4.3176279447603574e-06, + "loss": 1.9311, + "step": 624 + }, + { + "epoch": 0.684931506849315, + "grad_norm": 2.1459643840789795, + "learning_rate": 4.315597075548335e-06, + "loss": 1.9673, + "step": 625 + }, + { + "epoch": 0.686027397260274, + "grad_norm": 2.278928756713867, + "learning_rate": 4.313566206336312e-06, + "loss": 1.954, + "step": 626 + }, + { + "epoch": 0.6871232876712329, + "grad_norm": 2.05305552482605, + "learning_rate": 4.311535337124289e-06, + "loss": 1.9789, + "step": 627 + }, + { + "epoch": 0.6882191780821918, + "grad_norm": 2.3979289531707764, + "learning_rate": 4.3095044679122665e-06, + "loss": 1.926, + "step": 628 + }, + { + "epoch": 0.6893150684931507, + "grad_norm": 2.1635828018188477, + "learning_rate": 4.307473598700244e-06, + "loss": 1.9316, + "step": 629 + }, + { + "epoch": 0.6904109589041096, + "grad_norm": 2.237713098526001, + "learning_rate": 4.305442729488221e-06, + "loss": 1.9409, + "step": 630 + }, + { + "epoch": 0.6904109589041096, + "eval_loss": 2.084226131439209, + "eval_runtime": 37.6039, + "eval_samples_per_second": 26.593, + "eval_steps_per_second": 1.117, + "step": 630 + }, + { + "epoch": 0.6915068493150685, + "grad_norm": 2.157435894012451, + "learning_rate": 4.303411860276198e-06, + "loss": 1.8777, + "step": 631 + }, + { + "epoch": 0.6926027397260274, + "grad_norm": 2.20766019821167, + "learning_rate": 4.301380991064176e-06, + "loss": 1.9803, + "step": 632 + }, + { + "epoch": 0.6936986301369863, + "grad_norm": 2.173079490661621, + "learning_rate": 4.299350121852153e-06, + "loss": 1.9541, + "step": 633 + }, + { + "epoch": 0.6947945205479452, + "grad_norm": 1.9727106094360352, + "learning_rate": 4.29731925264013e-06, + "loss": 1.9766, + "step": 634 + }, + { + "epoch": 0.6958904109589041, + "grad_norm": 2.268538475036621, + "learning_rate": 4.2952883834281075e-06, + "loss": 1.9737, + "step": 635 + }, + { + "epoch": 0.6969863013698631, + "grad_norm": 2.2118406295776367, + "learning_rate": 4.293257514216085e-06, + "loss": 1.9291, + "step": 636 + }, + { + "epoch": 0.6980821917808219, + "grad_norm": 2.2939369678497314, + "learning_rate": 4.291226645004062e-06, + "loss": 1.8868, + "step": 637 + }, + { + "epoch": 0.6991780821917808, + "grad_norm": 2.4940083026885986, + "learning_rate": 4.289195775792039e-06, + "loss": 1.9694, + "step": 638 + }, + { + "epoch": 0.7002739726027397, + "grad_norm": 2.1988351345062256, + "learning_rate": 4.287164906580017e-06, + "loss": 1.9781, + "step": 639 + }, + { + "epoch": 0.7013698630136986, + "grad_norm": 2.099578619003296, + "learning_rate": 4.285134037367994e-06, + "loss": 1.9145, + "step": 640 + }, + { + "epoch": 0.7013698630136986, + "eval_loss": 2.07913875579834, + "eval_runtime": 37.3458, + "eval_samples_per_second": 26.777, + "eval_steps_per_second": 1.125, + "step": 640 + }, + { + "epoch": 0.7024657534246576, + "grad_norm": 2.2437307834625244, + "learning_rate": 4.283103168155971e-06, + "loss": 1.9075, + "step": 641 + }, + { + "epoch": 0.7035616438356165, + "grad_norm": 2.320880174636841, + "learning_rate": 4.281072298943948e-06, + "loss": 1.8929, + "step": 642 + }, + { + "epoch": 0.7046575342465753, + "grad_norm": 2.1174306869506836, + "learning_rate": 4.279041429731926e-06, + "loss": 1.8872, + "step": 643 + }, + { + "epoch": 0.7057534246575342, + "grad_norm": 2.2826361656188965, + "learning_rate": 4.277010560519903e-06, + "loss": 1.9287, + "step": 644 + }, + { + "epoch": 0.7068493150684931, + "grad_norm": 2.7543559074401855, + "learning_rate": 4.27497969130788e-06, + "loss": 1.9119, + "step": 645 + }, + { + "epoch": 0.707945205479452, + "grad_norm": 2.2582201957702637, + "learning_rate": 4.2729488220958575e-06, + "loss": 1.9118, + "step": 646 + }, + { + "epoch": 0.709041095890411, + "grad_norm": 2.3121371269226074, + "learning_rate": 4.270917952883835e-06, + "loss": 1.959, + "step": 647 + }, + { + "epoch": 0.7101369863013699, + "grad_norm": 2.419102430343628, + "learning_rate": 4.268887083671812e-06, + "loss": 1.9679, + "step": 648 + }, + { + "epoch": 0.7112328767123288, + "grad_norm": 2.285771369934082, + "learning_rate": 4.266856214459789e-06, + "loss": 1.8939, + "step": 649 + }, + { + "epoch": 0.7123287671232876, + "grad_norm": 2.193603754043579, + "learning_rate": 4.264825345247766e-06, + "loss": 2.0097, + "step": 650 + }, + { + "epoch": 0.7123287671232876, + "eval_loss": 2.0785164833068848, + "eval_runtime": 37.2595, + "eval_samples_per_second": 26.839, + "eval_steps_per_second": 1.127, + "step": 650 + }, + { + "epoch": 0.7134246575342466, + "grad_norm": 2.2645511627197266, + "learning_rate": 4.262794476035744e-06, + "loss": 1.8865, + "step": 651 + }, + { + "epoch": 0.7145205479452055, + "grad_norm": 2.5553910732269287, + "learning_rate": 4.26076360682372e-06, + "loss": 1.9331, + "step": 652 + }, + { + "epoch": 0.7156164383561644, + "grad_norm": 2.2770845890045166, + "learning_rate": 4.2587327376116984e-06, + "loss": 1.847, + "step": 653 + }, + { + "epoch": 0.7167123287671233, + "grad_norm": 2.1486358642578125, + "learning_rate": 4.256701868399675e-06, + "loss": 1.9058, + "step": 654 + }, + { + "epoch": 0.7178082191780822, + "grad_norm": 2.168975353240967, + "learning_rate": 4.254670999187653e-06, + "loss": 1.9408, + "step": 655 + }, + { + "epoch": 0.718904109589041, + "grad_norm": 2.2156167030334473, + "learning_rate": 4.252640129975629e-06, + "loss": 1.932, + "step": 656 + }, + { + "epoch": 0.72, + "grad_norm": 2.148214101791382, + "learning_rate": 4.2506092607636075e-06, + "loss": 1.9327, + "step": 657 + }, + { + "epoch": 0.7210958904109589, + "grad_norm": 2.3860998153686523, + "learning_rate": 4.248578391551584e-06, + "loss": 1.9191, + "step": 658 + }, + { + "epoch": 0.7221917808219178, + "grad_norm": 2.3917558193206787, + "learning_rate": 4.246547522339562e-06, + "loss": 1.8607, + "step": 659 + }, + { + "epoch": 0.7232876712328767, + "grad_norm": 2.3024795055389404, + "learning_rate": 4.2445166531275385e-06, + "loss": 1.9083, + "step": 660 + }, + { + "epoch": 0.7232876712328767, + "eval_loss": 2.07985782623291, + "eval_runtime": 37.1583, + "eval_samples_per_second": 26.912, + "eval_steps_per_second": 1.13, + "step": 660 + }, + { + "epoch": 0.7243835616438357, + "grad_norm": 2.261615037918091, + "learning_rate": 4.242485783915517e-06, + "loss": 1.9155, + "step": 661 + }, + { + "epoch": 0.7254794520547945, + "grad_norm": 2.190124034881592, + "learning_rate": 4.240454914703493e-06, + "loss": 1.858, + "step": 662 + }, + { + "epoch": 0.7265753424657534, + "grad_norm": 2.3396036624908447, + "learning_rate": 4.23842404549147e-06, + "loss": 1.878, + "step": 663 + }, + { + "epoch": 0.7276712328767123, + "grad_norm": 2.198410987854004, + "learning_rate": 4.236393176279448e-06, + "loss": 1.8755, + "step": 664 + }, + { + "epoch": 0.7287671232876712, + "grad_norm": 2.108875274658203, + "learning_rate": 4.234362307067425e-06, + "loss": 1.8709, + "step": 665 + }, + { + "epoch": 0.7298630136986302, + "grad_norm": 2.0908193588256836, + "learning_rate": 4.232331437855402e-06, + "loss": 1.9395, + "step": 666 + }, + { + "epoch": 0.7309589041095891, + "grad_norm": 2.208134651184082, + "learning_rate": 4.2303005686433795e-06, + "loss": 1.9675, + "step": 667 + }, + { + "epoch": 0.7320547945205479, + "grad_norm": 2.2779526710510254, + "learning_rate": 4.228269699431357e-06, + "loss": 1.9661, + "step": 668 + }, + { + "epoch": 0.7331506849315068, + "grad_norm": 2.053279161453247, + "learning_rate": 4.226238830219334e-06, + "loss": 1.9244, + "step": 669 + }, + { + "epoch": 0.7342465753424657, + "grad_norm": 2.2060821056365967, + "learning_rate": 4.224207961007311e-06, + "loss": 1.9252, + "step": 670 + }, + { + "epoch": 0.7342465753424657, + "eval_loss": 2.0804648399353027, + "eval_runtime": 37.1852, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 670 + }, + { + "epoch": 0.7353424657534247, + "grad_norm": 2.1996090412139893, + "learning_rate": 4.2221770917952886e-06, + "loss": 1.926, + "step": 671 + }, + { + "epoch": 0.7364383561643836, + "grad_norm": 2.0817034244537354, + "learning_rate": 4.220146222583266e-06, + "loss": 1.9123, + "step": 672 + }, + { + "epoch": 0.7375342465753425, + "grad_norm": 2.2505035400390625, + "learning_rate": 4.218115353371243e-06, + "loss": 1.9369, + "step": 673 + }, + { + "epoch": 0.7386301369863014, + "grad_norm": 2.2053000926971436, + "learning_rate": 4.21608448415922e-06, + "loss": 1.9225, + "step": 674 + }, + { + "epoch": 0.7397260273972602, + "grad_norm": 2.039905071258545, + "learning_rate": 4.214053614947198e-06, + "loss": 1.8876, + "step": 675 + }, + { + "epoch": 0.7408219178082192, + "grad_norm": 2.047445297241211, + "learning_rate": 4.212022745735175e-06, + "loss": 1.9724, + "step": 676 + }, + { + "epoch": 0.7419178082191781, + "grad_norm": 2.277803897857666, + "learning_rate": 4.209991876523152e-06, + "loss": 1.8687, + "step": 677 + }, + { + "epoch": 0.743013698630137, + "grad_norm": 2.0048105716705322, + "learning_rate": 4.2079610073111295e-06, + "loss": 1.9349, + "step": 678 + }, + { + "epoch": 0.7441095890410959, + "grad_norm": 2.121614456176758, + "learning_rate": 4.205930138099107e-06, + "loss": 1.8723, + "step": 679 + }, + { + "epoch": 0.7452054794520548, + "grad_norm": 2.169165849685669, + "learning_rate": 4.203899268887084e-06, + "loss": 1.9713, + "step": 680 + }, + { + "epoch": 0.7452054794520548, + "eval_loss": 2.0704824924468994, + "eval_runtime": 37.2015, + "eval_samples_per_second": 26.881, + "eval_steps_per_second": 1.129, + "step": 680 + }, + { + "epoch": 0.7463013698630137, + "grad_norm": 2.004974603652954, + "learning_rate": 4.201868399675061e-06, + "loss": 1.9237, + "step": 681 + }, + { + "epoch": 0.7473972602739726, + "grad_norm": 2.042402505874634, + "learning_rate": 4.199837530463039e-06, + "loss": 1.9686, + "step": 682 + }, + { + "epoch": 0.7484931506849315, + "grad_norm": 2.152040719985962, + "learning_rate": 4.197806661251016e-06, + "loss": 1.8498, + "step": 683 + }, + { + "epoch": 0.7495890410958904, + "grad_norm": 2.116483211517334, + "learning_rate": 4.195775792038993e-06, + "loss": 1.9241, + "step": 684 + }, + { + "epoch": 0.7506849315068493, + "grad_norm": 2.1628384590148926, + "learning_rate": 4.19374492282697e-06, + "loss": 1.9019, + "step": 685 + }, + { + "epoch": 0.7517808219178083, + "grad_norm": 2.1851415634155273, + "learning_rate": 4.191714053614948e-06, + "loss": 1.836, + "step": 686 + }, + { + "epoch": 0.7528767123287671, + "grad_norm": 2.173215866088867, + "learning_rate": 4.189683184402924e-06, + "loss": 1.9412, + "step": 687 + }, + { + "epoch": 0.753972602739726, + "grad_norm": 2.1002466678619385, + "learning_rate": 4.187652315190902e-06, + "loss": 1.8871, + "step": 688 + }, + { + "epoch": 0.7550684931506849, + "grad_norm": 2.1932122707366943, + "learning_rate": 4.185621445978879e-06, + "loss": 1.9671, + "step": 689 + }, + { + "epoch": 0.7561643835616438, + "grad_norm": 2.2915258407592773, + "learning_rate": 4.183590576766857e-06, + "loss": 1.8957, + "step": 690 + }, + { + "epoch": 0.7561643835616438, + "eval_loss": 2.067124128341675, + "eval_runtime": 37.2362, + "eval_samples_per_second": 26.856, + "eval_steps_per_second": 1.128, + "step": 690 + }, + { + "epoch": 0.7572602739726028, + "grad_norm": 2.060879707336426, + "learning_rate": 4.181559707554833e-06, + "loss": 1.9091, + "step": 691 + }, + { + "epoch": 0.7583561643835617, + "grad_norm": 2.031156539916992, + "learning_rate": 4.179528838342811e-06, + "loss": 1.9606, + "step": 692 + }, + { + "epoch": 0.7594520547945205, + "grad_norm": 2.121628999710083, + "learning_rate": 4.177497969130788e-06, + "loss": 1.88, + "step": 693 + }, + { + "epoch": 0.7605479452054794, + "grad_norm": 2.2009618282318115, + "learning_rate": 4.175467099918766e-06, + "loss": 1.8916, + "step": 694 + }, + { + "epoch": 0.7616438356164383, + "grad_norm": 2.124082326889038, + "learning_rate": 4.173436230706742e-06, + "loss": 1.9805, + "step": 695 + }, + { + "epoch": 0.7627397260273973, + "grad_norm": 2.3914906978607178, + "learning_rate": 4.1714053614947205e-06, + "loss": 1.9052, + "step": 696 + }, + { + "epoch": 0.7638356164383562, + "grad_norm": 2.2398853302001953, + "learning_rate": 4.169374492282697e-06, + "loss": 1.8965, + "step": 697 + }, + { + "epoch": 0.7649315068493151, + "grad_norm": 2.224860191345215, + "learning_rate": 4.167343623070675e-06, + "loss": 1.8962, + "step": 698 + }, + { + "epoch": 0.766027397260274, + "grad_norm": 2.3356244564056396, + "learning_rate": 4.165312753858652e-06, + "loss": 1.9543, + "step": 699 + }, + { + "epoch": 0.7671232876712328, + "grad_norm": 2.1308751106262207, + "learning_rate": 4.163281884646629e-06, + "loss": 1.9095, + "step": 700 + }, + { + "epoch": 0.7671232876712328, + "eval_loss": 2.066145896911621, + "eval_runtime": 37.2629, + "eval_samples_per_second": 26.836, + "eval_steps_per_second": 1.127, + "step": 700 + }, + { + "epoch": 0.7682191780821918, + "grad_norm": 2.1655938625335693, + "learning_rate": 4.161251015434607e-06, + "loss": 1.8923, + "step": 701 + }, + { + "epoch": 0.7693150684931507, + "grad_norm": 2.1985316276550293, + "learning_rate": 4.159220146222583e-06, + "loss": 1.9216, + "step": 702 + }, + { + "epoch": 0.7704109589041096, + "grad_norm": 2.1537251472473145, + "learning_rate": 4.157189277010561e-06, + "loss": 1.879, + "step": 703 + }, + { + "epoch": 0.7715068493150685, + "grad_norm": 2.1867849826812744, + "learning_rate": 4.155158407798538e-06, + "loss": 1.905, + "step": 704 + }, + { + "epoch": 0.7726027397260274, + "grad_norm": 2.342320203781128, + "learning_rate": 4.153127538586516e-06, + "loss": 1.9403, + "step": 705 + }, + { + "epoch": 0.7736986301369863, + "grad_norm": 2.2250113487243652, + "learning_rate": 4.151096669374492e-06, + "loss": 1.9109, + "step": 706 + }, + { + "epoch": 0.7747945205479452, + "grad_norm": 2.179103374481201, + "learning_rate": 4.1490658001624705e-06, + "loss": 1.9288, + "step": 707 + }, + { + "epoch": 0.7758904109589041, + "grad_norm": 2.3618645668029785, + "learning_rate": 4.147034930950447e-06, + "loss": 1.8982, + "step": 708 + }, + { + "epoch": 0.776986301369863, + "grad_norm": 2.150402307510376, + "learning_rate": 4.145004061738425e-06, + "loss": 1.9, + "step": 709 + }, + { + "epoch": 0.7780821917808219, + "grad_norm": 2.295571804046631, + "learning_rate": 4.1429731925264015e-06, + "loss": 1.9003, + "step": 710 + }, + { + "epoch": 0.7780821917808219, + "eval_loss": 2.067948341369629, + "eval_runtime": 37.2379, + "eval_samples_per_second": 26.854, + "eval_steps_per_second": 1.128, + "step": 710 + }, + { + "epoch": 0.7791780821917809, + "grad_norm": 2.3279545307159424, + "learning_rate": 4.140942323314379e-06, + "loss": 1.8307, + "step": 711 + }, + { + "epoch": 0.7802739726027397, + "grad_norm": 2.3993380069732666, + "learning_rate": 4.138911454102356e-06, + "loss": 1.8755, + "step": 712 + }, + { + "epoch": 0.7813698630136986, + "grad_norm": 2.139653444290161, + "learning_rate": 4.136880584890333e-06, + "loss": 1.8914, + "step": 713 + }, + { + "epoch": 0.7824657534246575, + "grad_norm": 2.239544153213501, + "learning_rate": 4.1348497156783106e-06, + "loss": 1.9289, + "step": 714 + }, + { + "epoch": 0.7835616438356164, + "grad_norm": 2.2039642333984375, + "learning_rate": 4.132818846466288e-06, + "loss": 1.9237, + "step": 715 + }, + { + "epoch": 0.7846575342465754, + "grad_norm": 2.1453804969787598, + "learning_rate": 4.130787977254265e-06, + "loss": 1.9291, + "step": 716 + }, + { + "epoch": 0.7857534246575343, + "grad_norm": 2.2350032329559326, + "learning_rate": 4.128757108042242e-06, + "loss": 1.9413, + "step": 717 + }, + { + "epoch": 0.7868493150684932, + "grad_norm": 2.1718318462371826, + "learning_rate": 4.12672623883022e-06, + "loss": 1.9318, + "step": 718 + }, + { + "epoch": 0.787945205479452, + "grad_norm": 2.1860601902008057, + "learning_rate": 4.124695369618197e-06, + "loss": 1.9934, + "step": 719 + }, + { + "epoch": 0.7890410958904109, + "grad_norm": 2.205166816711426, + "learning_rate": 4.122664500406174e-06, + "loss": 1.9038, + "step": 720 + }, + { + "epoch": 0.7890410958904109, + "eval_loss": 2.06229829788208, + "eval_runtime": 37.6121, + "eval_samples_per_second": 26.587, + "eval_steps_per_second": 1.117, + "step": 720 + }, + { + "epoch": 0.7901369863013699, + "grad_norm": 2.2094533443450928, + "learning_rate": 4.1206336311941515e-06, + "loss": 1.906, + "step": 721 + }, + { + "epoch": 0.7912328767123288, + "grad_norm": 2.178891658782959, + "learning_rate": 4.118602761982129e-06, + "loss": 1.936, + "step": 722 + }, + { + "epoch": 0.7923287671232877, + "grad_norm": 2.051077365875244, + "learning_rate": 4.116571892770106e-06, + "loss": 1.8736, + "step": 723 + }, + { + "epoch": 0.7934246575342466, + "grad_norm": 2.2430427074432373, + "learning_rate": 4.114541023558083e-06, + "loss": 1.8396, + "step": 724 + }, + { + "epoch": 0.7945205479452054, + "grad_norm": 2.2255303859710693, + "learning_rate": 4.112510154346061e-06, + "loss": 1.9668, + "step": 725 + }, + { + "epoch": 0.7956164383561644, + "grad_norm": 2.280313491821289, + "learning_rate": 4.110479285134038e-06, + "loss": 1.8945, + "step": 726 + }, + { + "epoch": 0.7967123287671233, + "grad_norm": 2.2516801357269287, + "learning_rate": 4.108448415922015e-06, + "loss": 1.9157, + "step": 727 + }, + { + "epoch": 0.7978082191780822, + "grad_norm": 2.1379809379577637, + "learning_rate": 4.1064175467099924e-06, + "loss": 1.935, + "step": 728 + }, + { + "epoch": 0.7989041095890411, + "grad_norm": 2.149425983428955, + "learning_rate": 4.10438667749797e-06, + "loss": 1.935, + "step": 729 + }, + { + "epoch": 0.8, + "grad_norm": 2.275270700454712, + "learning_rate": 4.102355808285947e-06, + "loss": 1.7889, + "step": 730 + }, + { + "epoch": 0.8, + "eval_loss": 2.0633974075317383, + "eval_runtime": 37.6895, + "eval_samples_per_second": 26.533, + "eval_steps_per_second": 1.114, + "step": 730 + }, + { + "epoch": 0.8010958904109589, + "grad_norm": 2.1550278663635254, + "learning_rate": 4.100324939073924e-06, + "loss": 1.9833, + "step": 731 + }, + { + "epoch": 0.8021917808219178, + "grad_norm": 2.3050060272216797, + "learning_rate": 4.0982940698619015e-06, + "loss": 1.8969, + "step": 732 + }, + { + "epoch": 0.8032876712328767, + "grad_norm": 2.4548778533935547, + "learning_rate": 4.096263200649879e-06, + "loss": 1.8945, + "step": 733 + }, + { + "epoch": 0.8043835616438356, + "grad_norm": 2.2611639499664307, + "learning_rate": 4.094232331437856e-06, + "loss": 1.9071, + "step": 734 + }, + { + "epoch": 0.8054794520547945, + "grad_norm": 2.1806774139404297, + "learning_rate": 4.092201462225833e-06, + "loss": 1.8787, + "step": 735 + }, + { + "epoch": 0.8065753424657535, + "grad_norm": 2.4374301433563232, + "learning_rate": 4.090170593013811e-06, + "loss": 1.88, + "step": 736 + }, + { + "epoch": 0.8076712328767123, + "grad_norm": 2.328333854675293, + "learning_rate": 4.088139723801787e-06, + "loss": 1.9178, + "step": 737 + }, + { + "epoch": 0.8087671232876712, + "grad_norm": 2.33552885055542, + "learning_rate": 4.086108854589765e-06, + "loss": 1.8587, + "step": 738 + }, + { + "epoch": 0.8098630136986301, + "grad_norm": 2.3472790718078613, + "learning_rate": 4.084077985377742e-06, + "loss": 1.877, + "step": 739 + }, + { + "epoch": 0.810958904109589, + "grad_norm": 2.307123899459839, + "learning_rate": 4.08204711616572e-06, + "loss": 1.8888, + "step": 740 + }, + { + "epoch": 0.810958904109589, + "eval_loss": 2.065826892852783, + "eval_runtime": 37.2747, + "eval_samples_per_second": 26.828, + "eval_steps_per_second": 1.127, + "step": 740 + }, + { + "epoch": 0.812054794520548, + "grad_norm": 2.3688747882843018, + "learning_rate": 4.080016246953696e-06, + "loss": 1.9571, + "step": 741 + }, + { + "epoch": 0.8131506849315069, + "grad_norm": 2.385007381439209, + "learning_rate": 4.077985377741674e-06, + "loss": 1.8657, + "step": 742 + }, + { + "epoch": 0.8142465753424658, + "grad_norm": 2.156766414642334, + "learning_rate": 4.075954508529651e-06, + "loss": 1.9421, + "step": 743 + }, + { + "epoch": 0.8153424657534246, + "grad_norm": 2.2074286937713623, + "learning_rate": 4.073923639317629e-06, + "loss": 1.9396, + "step": 744 + }, + { + "epoch": 0.8164383561643835, + "grad_norm": 2.185458183288574, + "learning_rate": 4.071892770105605e-06, + "loss": 1.9612, + "step": 745 + }, + { + "epoch": 0.8175342465753425, + "grad_norm": 2.324209451675415, + "learning_rate": 4.069861900893583e-06, + "loss": 1.892, + "step": 746 + }, + { + "epoch": 0.8186301369863014, + "grad_norm": 2.1312127113342285, + "learning_rate": 4.06783103168156e-06, + "loss": 1.8522, + "step": 747 + }, + { + "epoch": 0.8197260273972603, + "grad_norm": 2.0959460735321045, + "learning_rate": 4.065800162469538e-06, + "loss": 1.8732, + "step": 748 + }, + { + "epoch": 0.8208219178082192, + "grad_norm": 2.085519552230835, + "learning_rate": 4.063769293257514e-06, + "loss": 1.8832, + "step": 749 + }, + { + "epoch": 0.821917808219178, + "grad_norm": 2.310145854949951, + "learning_rate": 4.061738424045492e-06, + "loss": 1.919, + "step": 750 + }, + { + "epoch": 0.821917808219178, + "eval_loss": 2.0637967586517334, + "eval_runtime": 37.2246, + "eval_samples_per_second": 26.864, + "eval_steps_per_second": 1.128, + "step": 750 + }, + { + "epoch": 0.823013698630137, + "grad_norm": 2.3171160221099854, + "learning_rate": 4.059707554833469e-06, + "loss": 1.8874, + "step": 751 + }, + { + "epoch": 0.8241095890410959, + "grad_norm": 2.1065480709075928, + "learning_rate": 4.057676685621446e-06, + "loss": 1.9064, + "step": 752 + }, + { + "epoch": 0.8252054794520548, + "grad_norm": 2.0539677143096924, + "learning_rate": 4.0556458164094235e-06, + "loss": 1.9376, + "step": 753 + }, + { + "epoch": 0.8263013698630137, + "grad_norm": 2.4440360069274902, + "learning_rate": 4.053614947197401e-06, + "loss": 1.9129, + "step": 754 + }, + { + "epoch": 0.8273972602739726, + "grad_norm": 2.2324604988098145, + "learning_rate": 4.051584077985378e-06, + "loss": 1.8281, + "step": 755 + }, + { + "epoch": 0.8284931506849315, + "grad_norm": 2.262627363204956, + "learning_rate": 4.049553208773355e-06, + "loss": 1.9036, + "step": 756 + }, + { + "epoch": 0.8295890410958904, + "grad_norm": 2.2473418712615967, + "learning_rate": 4.047522339561333e-06, + "loss": 1.9455, + "step": 757 + }, + { + "epoch": 0.8306849315068493, + "grad_norm": 2.473444938659668, + "learning_rate": 4.04549147034931e-06, + "loss": 1.9003, + "step": 758 + }, + { + "epoch": 0.8317808219178082, + "grad_norm": 2.3454110622406006, + "learning_rate": 4.043460601137287e-06, + "loss": 1.8941, + "step": 759 + }, + { + "epoch": 0.8328767123287671, + "grad_norm": 2.4391231536865234, + "learning_rate": 4.041429731925264e-06, + "loss": 2.0146, + "step": 760 + }, + { + "epoch": 0.8328767123287671, + "eval_loss": 2.0618460178375244, + "eval_runtime": 37.1704, + "eval_samples_per_second": 26.903, + "eval_steps_per_second": 1.13, + "step": 760 + }, + { + "epoch": 0.8339726027397261, + "grad_norm": 2.3129498958587646, + "learning_rate": 4.039398862713242e-06, + "loss": 1.9, + "step": 761 + }, + { + "epoch": 0.8350684931506849, + "grad_norm": 2.3766934871673584, + "learning_rate": 4.037367993501219e-06, + "loss": 1.9017, + "step": 762 + }, + { + "epoch": 0.8361643835616438, + "grad_norm": 2.1698460578918457, + "learning_rate": 4.035337124289196e-06, + "loss": 1.9657, + "step": 763 + }, + { + "epoch": 0.8372602739726027, + "grad_norm": 2.384904146194458, + "learning_rate": 4.0333062550771735e-06, + "loss": 1.8867, + "step": 764 + }, + { + "epoch": 0.8383561643835616, + "grad_norm": 2.572391986846924, + "learning_rate": 4.031275385865151e-06, + "loss": 1.8561, + "step": 765 + }, + { + "epoch": 0.8394520547945206, + "grad_norm": 2.216606616973877, + "learning_rate": 4.029244516653128e-06, + "loss": 1.8926, + "step": 766 + }, + { + "epoch": 0.8405479452054795, + "grad_norm": 2.215756893157959, + "learning_rate": 4.027213647441105e-06, + "loss": 1.928, + "step": 767 + }, + { + "epoch": 0.8416438356164384, + "grad_norm": 2.2713284492492676, + "learning_rate": 4.025182778229083e-06, + "loss": 1.8286, + "step": 768 + }, + { + "epoch": 0.8427397260273972, + "grad_norm": 2.341747283935547, + "learning_rate": 4.02315190901706e-06, + "loss": 1.8789, + "step": 769 + }, + { + "epoch": 0.8438356164383561, + "grad_norm": 2.1485848426818848, + "learning_rate": 4.021121039805037e-06, + "loss": 1.8998, + "step": 770 + }, + { + "epoch": 0.8438356164383561, + "eval_loss": 2.0596234798431396, + "eval_runtime": 37.1854, + "eval_samples_per_second": 26.892, + "eval_steps_per_second": 1.129, + "step": 770 + }, + { + "epoch": 0.8449315068493151, + "grad_norm": 2.234192132949829, + "learning_rate": 4.0190901705930144e-06, + "loss": 1.8995, + "step": 771 + }, + { + "epoch": 0.846027397260274, + "grad_norm": 2.289328098297119, + "learning_rate": 4.017059301380992e-06, + "loss": 1.9882, + "step": 772 + }, + { + "epoch": 0.8471232876712329, + "grad_norm": 2.1001455783843994, + "learning_rate": 4.015028432168969e-06, + "loss": 1.8706, + "step": 773 + }, + { + "epoch": 0.8482191780821918, + "grad_norm": 2.2200522422790527, + "learning_rate": 4.012997562956945e-06, + "loss": 1.9443, + "step": 774 + }, + { + "epoch": 0.8493150684931506, + "grad_norm": 2.2540805339813232, + "learning_rate": 4.0109666937449235e-06, + "loss": 1.8552, + "step": 775 + }, + { + "epoch": 0.8504109589041096, + "grad_norm": 2.1214053630828857, + "learning_rate": 4.0089358245329e-06, + "loss": 1.8741, + "step": 776 + }, + { + "epoch": 0.8515068493150685, + "grad_norm": 2.0221869945526123, + "learning_rate": 4.006904955320878e-06, + "loss": 1.8798, + "step": 777 + }, + { + "epoch": 0.8526027397260274, + "grad_norm": 2.217261791229248, + "learning_rate": 4.0048740861088545e-06, + "loss": 1.8542, + "step": 778 + }, + { + "epoch": 0.8536986301369863, + "grad_norm": 2.099202871322632, + "learning_rate": 4.002843216896833e-06, + "loss": 1.9342, + "step": 779 + }, + { + "epoch": 0.8547945205479452, + "grad_norm": 2.139526844024658, + "learning_rate": 4.000812347684809e-06, + "loss": 1.9252, + "step": 780 + }, + { + "epoch": 0.8547945205479452, + "eval_loss": 2.054051399230957, + "eval_runtime": 37.1125, + "eval_samples_per_second": 26.945, + "eval_steps_per_second": 1.132, + "step": 780 + }, + { + "epoch": 0.8558904109589041, + "grad_norm": 2.2231476306915283, + "learning_rate": 3.998781478472787e-06, + "loss": 1.8671, + "step": 781 + }, + { + "epoch": 0.856986301369863, + "grad_norm": 2.143979072570801, + "learning_rate": 3.996750609260764e-06, + "loss": 1.8759, + "step": 782 + }, + { + "epoch": 0.8580821917808219, + "grad_norm": 2.1353135108947754, + "learning_rate": 3.994719740048742e-06, + "loss": 1.8559, + "step": 783 + }, + { + "epoch": 0.8591780821917808, + "grad_norm": 2.222564697265625, + "learning_rate": 3.992688870836718e-06, + "loss": 1.999, + "step": 784 + }, + { + "epoch": 0.8602739726027397, + "grad_norm": 1.958031177520752, + "learning_rate": 3.990658001624696e-06, + "loss": 1.9019, + "step": 785 + }, + { + "epoch": 0.8613698630136987, + "grad_norm": 2.2048962116241455, + "learning_rate": 3.988627132412673e-06, + "loss": 1.8561, + "step": 786 + }, + { + "epoch": 0.8624657534246576, + "grad_norm": 2.150332450866699, + "learning_rate": 3.98659626320065e-06, + "loss": 1.9399, + "step": 787 + }, + { + "epoch": 0.8635616438356164, + "grad_norm": 2.182742118835449, + "learning_rate": 3.984565393988627e-06, + "loss": 1.8933, + "step": 788 + }, + { + "epoch": 0.8646575342465753, + "grad_norm": 2.19435715675354, + "learning_rate": 3.9825345247766046e-06, + "loss": 1.8918, + "step": 789 + }, + { + "epoch": 0.8657534246575342, + "grad_norm": 2.3382859230041504, + "learning_rate": 3.980503655564582e-06, + "loss": 1.8937, + "step": 790 + }, + { + "epoch": 0.8657534246575342, + "eval_loss": 2.051997423171997, + "eval_runtime": 37.1866, + "eval_samples_per_second": 26.891, + "eval_steps_per_second": 1.129, + "step": 790 + }, + { + "epoch": 0.8668493150684932, + "grad_norm": 2.2700726985931396, + "learning_rate": 3.978472786352559e-06, + "loss": 1.9148, + "step": 791 + }, + { + "epoch": 0.8679452054794521, + "grad_norm": 2.2998759746551514, + "learning_rate": 3.976441917140536e-06, + "loss": 1.9268, + "step": 792 + }, + { + "epoch": 0.869041095890411, + "grad_norm": 2.2367377281188965, + "learning_rate": 3.974411047928514e-06, + "loss": 1.901, + "step": 793 + }, + { + "epoch": 0.8701369863013698, + "grad_norm": 2.0868773460388184, + "learning_rate": 3.972380178716491e-06, + "loss": 1.9677, + "step": 794 + }, + { + "epoch": 0.8712328767123287, + "grad_norm": 2.183732509613037, + "learning_rate": 3.970349309504468e-06, + "loss": 1.9271, + "step": 795 + }, + { + "epoch": 0.8723287671232877, + "grad_norm": 2.187282085418701, + "learning_rate": 3.9683184402924455e-06, + "loss": 1.9437, + "step": 796 + }, + { + "epoch": 0.8734246575342466, + "grad_norm": 2.1906070709228516, + "learning_rate": 3.966287571080423e-06, + "loss": 1.8437, + "step": 797 + }, + { + "epoch": 0.8745205479452055, + "grad_norm": 2.16979718208313, + "learning_rate": 3.9642567018684e-06, + "loss": 1.829, + "step": 798 + }, + { + "epoch": 0.8756164383561644, + "grad_norm": 2.2387869358062744, + "learning_rate": 3.962225832656377e-06, + "loss": 1.9143, + "step": 799 + }, + { + "epoch": 0.8767123287671232, + "grad_norm": 2.208221197128296, + "learning_rate": 3.960194963444355e-06, + "loss": 1.8894, + "step": 800 + }, + { + "epoch": 0.8767123287671232, + "eval_loss": 2.047945976257324, + "eval_runtime": 37.1778, + "eval_samples_per_second": 26.898, + "eval_steps_per_second": 1.13, + "step": 800 + }, + { + "epoch": 0.8778082191780822, + "grad_norm": 2.335108518600464, + "learning_rate": 3.958164094232332e-06, + "loss": 1.8155, + "step": 801 + }, + { + "epoch": 0.8789041095890411, + "grad_norm": 2.168006181716919, + "learning_rate": 3.956133225020309e-06, + "loss": 1.8919, + "step": 802 + }, + { + "epoch": 0.88, + "grad_norm": 2.3029897212982178, + "learning_rate": 3.954102355808286e-06, + "loss": 1.8996, + "step": 803 + }, + { + "epoch": 0.8810958904109589, + "grad_norm": 2.156141519546509, + "learning_rate": 3.952071486596264e-06, + "loss": 1.8965, + "step": 804 + }, + { + "epoch": 0.8821917808219178, + "grad_norm": 2.1522068977355957, + "learning_rate": 3.950040617384241e-06, + "loss": 1.8922, + "step": 805 + }, + { + "epoch": 0.8832876712328767, + "grad_norm": 2.153430700302124, + "learning_rate": 3.948009748172218e-06, + "loss": 1.8839, + "step": 806 + }, + { + "epoch": 0.8843835616438356, + "grad_norm": 2.1891674995422363, + "learning_rate": 3.9459788789601955e-06, + "loss": 1.8756, + "step": 807 + }, + { + "epoch": 0.8854794520547945, + "grad_norm": 2.1092092990875244, + "learning_rate": 3.943948009748173e-06, + "loss": 1.9452, + "step": 808 + }, + { + "epoch": 0.8865753424657534, + "grad_norm": 2.0465025901794434, + "learning_rate": 3.94191714053615e-06, + "loss": 1.9067, + "step": 809 + }, + { + "epoch": 0.8876712328767123, + "grad_norm": 2.1150474548339844, + "learning_rate": 3.939886271324127e-06, + "loss": 1.8874, + "step": 810 + }, + { + "epoch": 0.8876712328767123, + "eval_loss": 2.046292304992676, + "eval_runtime": 37.2421, + "eval_samples_per_second": 26.851, + "eval_steps_per_second": 1.128, + "step": 810 + }, + { + "epoch": 0.8887671232876713, + "grad_norm": 2.039989948272705, + "learning_rate": 3.937855402112104e-06, + "loss": 1.9212, + "step": 811 + }, + { + "epoch": 0.8898630136986302, + "grad_norm": 2.17354679107666, + "learning_rate": 3.935824532900082e-06, + "loss": 1.8473, + "step": 812 + }, + { + "epoch": 0.890958904109589, + "grad_norm": 2.3101940155029297, + "learning_rate": 3.933793663688058e-06, + "loss": 1.8413, + "step": 813 + }, + { + "epoch": 0.8920547945205479, + "grad_norm": 2.1300928592681885, + "learning_rate": 3.9317627944760365e-06, + "loss": 1.9126, + "step": 814 + }, + { + "epoch": 0.8931506849315068, + "grad_norm": 2.2317357063293457, + "learning_rate": 3.929731925264013e-06, + "loss": 1.938, + "step": 815 + }, + { + "epoch": 0.8942465753424658, + "grad_norm": 2.1097419261932373, + "learning_rate": 3.927701056051991e-06, + "loss": 1.8426, + "step": 816 + }, + { + "epoch": 0.8953424657534247, + "grad_norm": 2.1709141731262207, + "learning_rate": 3.9256701868399674e-06, + "loss": 1.8779, + "step": 817 + }, + { + "epoch": 0.8964383561643836, + "grad_norm": 2.104193925857544, + "learning_rate": 3.9236393176279456e-06, + "loss": 1.8736, + "step": 818 + }, + { + "epoch": 0.8975342465753424, + "grad_norm": 2.094637632369995, + "learning_rate": 3.921608448415922e-06, + "loss": 1.8556, + "step": 819 + }, + { + "epoch": 0.8986301369863013, + "grad_norm": 2.217918872833252, + "learning_rate": 3.9195775792039e-06, + "loss": 1.8429, + "step": 820 + }, + { + "epoch": 0.8986301369863013, + "eval_loss": 2.046095848083496, + "eval_runtime": 37.5659, + "eval_samples_per_second": 26.62, + "eval_steps_per_second": 1.118, + "step": 820 + }, + { + "epoch": 0.8997260273972603, + "grad_norm": 2.220355272293091, + "learning_rate": 3.9175467099918765e-06, + "loss": 1.8973, + "step": 821 + }, + { + "epoch": 0.9008219178082192, + "grad_norm": 2.0766689777374268, + "learning_rate": 3.915515840779855e-06, + "loss": 1.8728, + "step": 822 + }, + { + "epoch": 0.9019178082191781, + "grad_norm": 2.2046821117401123, + "learning_rate": 3.913484971567831e-06, + "loss": 1.9211, + "step": 823 + }, + { + "epoch": 0.903013698630137, + "grad_norm": 2.286024332046509, + "learning_rate": 3.911454102355808e-06, + "loss": 1.913, + "step": 824 + }, + { + "epoch": 0.9041095890410958, + "grad_norm": 2.2495479583740234, + "learning_rate": 3.909423233143786e-06, + "loss": 1.8214, + "step": 825 + }, + { + "epoch": 0.9052054794520548, + "grad_norm": 2.1300203800201416, + "learning_rate": 3.907392363931763e-06, + "loss": 1.9033, + "step": 826 + }, + { + "epoch": 0.9063013698630137, + "grad_norm": 2.092094898223877, + "learning_rate": 3.90536149471974e-06, + "loss": 1.9007, + "step": 827 + }, + { + "epoch": 0.9073972602739726, + "grad_norm": 2.1571428775787354, + "learning_rate": 3.9033306255077175e-06, + "loss": 1.9606, + "step": 828 + }, + { + "epoch": 0.9084931506849315, + "grad_norm": 2.1052281856536865, + "learning_rate": 3.901299756295695e-06, + "loss": 1.9916, + "step": 829 + }, + { + "epoch": 0.9095890410958904, + "grad_norm": 2.113666534423828, + "learning_rate": 3.899268887083672e-06, + "loss": 2.0136, + "step": 830 + }, + { + "epoch": 0.9095890410958904, + "eval_loss": 2.0450072288513184, + "eval_runtime": 37.6746, + "eval_samples_per_second": 26.543, + "eval_steps_per_second": 1.115, + "step": 830 + }, + { + "epoch": 0.9106849315068493, + "grad_norm": 2.112591028213501, + "learning_rate": 3.897238017871649e-06, + "loss": 1.8784, + "step": 831 + }, + { + "epoch": 0.9117808219178082, + "grad_norm": 2.1584672927856445, + "learning_rate": 3.8952071486596266e-06, + "loss": 1.8876, + "step": 832 + }, + { + "epoch": 0.9128767123287671, + "grad_norm": 2.2744317054748535, + "learning_rate": 3.893176279447604e-06, + "loss": 1.8981, + "step": 833 + }, + { + "epoch": 0.913972602739726, + "grad_norm": 2.21824312210083, + "learning_rate": 3.891145410235581e-06, + "loss": 1.85, + "step": 834 + }, + { + "epoch": 0.915068493150685, + "grad_norm": 2.2203729152679443, + "learning_rate": 3.889114541023558e-06, + "loss": 1.9649, + "step": 835 + }, + { + "epoch": 0.9161643835616439, + "grad_norm": 2.1767687797546387, + "learning_rate": 3.887083671811536e-06, + "loss": 1.9044, + "step": 836 + }, + { + "epoch": 0.9172602739726028, + "grad_norm": 2.0724353790283203, + "learning_rate": 3.885052802599513e-06, + "loss": 1.8787, + "step": 837 + }, + { + "epoch": 0.9183561643835616, + "grad_norm": 2.1283740997314453, + "learning_rate": 3.88302193338749e-06, + "loss": 1.8284, + "step": 838 + }, + { + "epoch": 0.9194520547945205, + "grad_norm": 2.5087451934814453, + "learning_rate": 3.8809910641754675e-06, + "loss": 1.9546, + "step": 839 + }, + { + "epoch": 0.9205479452054794, + "grad_norm": 2.0633344650268555, + "learning_rate": 3.878960194963445e-06, + "loss": 1.8969, + "step": 840 + }, + { + "epoch": 0.9205479452054794, + "eval_loss": 2.047938346862793, + "eval_runtime": 37.2293, + "eval_samples_per_second": 26.861, + "eval_steps_per_second": 1.128, + "step": 840 + }, + { + "epoch": 0.9216438356164384, + "grad_norm": 2.178361654281616, + "learning_rate": 3.876929325751422e-06, + "loss": 1.8855, + "step": 841 + }, + { + "epoch": 0.9227397260273973, + "grad_norm": 2.3223721981048584, + "learning_rate": 3.874898456539399e-06, + "loss": 1.9363, + "step": 842 + }, + { + "epoch": 0.9238356164383562, + "grad_norm": 2.3330960273742676, + "learning_rate": 3.872867587327377e-06, + "loss": 1.8374, + "step": 843 + }, + { + "epoch": 0.924931506849315, + "grad_norm": 2.0584356784820557, + "learning_rate": 3.870836718115354e-06, + "loss": 1.911, + "step": 844 + }, + { + "epoch": 0.9260273972602739, + "grad_norm": 2.1834609508514404, + "learning_rate": 3.868805848903331e-06, + "loss": 1.8645, + "step": 845 + }, + { + "epoch": 0.9271232876712329, + "grad_norm": 2.2576122283935547, + "learning_rate": 3.8667749796913084e-06, + "loss": 1.8662, + "step": 846 + }, + { + "epoch": 0.9282191780821918, + "grad_norm": 2.1271183490753174, + "learning_rate": 3.864744110479286e-06, + "loss": 1.8935, + "step": 847 + }, + { + "epoch": 0.9293150684931507, + "grad_norm": 2.226727247238159, + "learning_rate": 3.862713241267262e-06, + "loss": 1.8969, + "step": 848 + }, + { + "epoch": 0.9304109589041096, + "grad_norm": 2.3301939964294434, + "learning_rate": 3.86068237205524e-06, + "loss": 1.9393, + "step": 849 + }, + { + "epoch": 0.9315068493150684, + "grad_norm": 2.180250883102417, + "learning_rate": 3.858651502843217e-06, + "loss": 1.9331, + "step": 850 + }, + { + "epoch": 0.9315068493150684, + "eval_loss": 2.0509681701660156, + "eval_runtime": 37.2111, + "eval_samples_per_second": 26.874, + "eval_steps_per_second": 1.129, + "step": 850 + }, + { + "epoch": 0.9326027397260274, + "grad_norm": 2.223548173904419, + "learning_rate": 3.856620633631195e-06, + "loss": 1.894, + "step": 851 + }, + { + "epoch": 0.9336986301369863, + "grad_norm": 2.1904265880584717, + "learning_rate": 3.854589764419171e-06, + "loss": 1.8265, + "step": 852 + }, + { + "epoch": 0.9347945205479452, + "grad_norm": 2.1374316215515137, + "learning_rate": 3.852558895207149e-06, + "loss": 1.8755, + "step": 853 + }, + { + "epoch": 0.9358904109589041, + "grad_norm": 2.2175488471984863, + "learning_rate": 3.850528025995126e-06, + "loss": 1.9649, + "step": 854 + }, + { + "epoch": 0.936986301369863, + "grad_norm": 2.192126989364624, + "learning_rate": 3.848497156783104e-06, + "loss": 1.919, + "step": 855 + }, + { + "epoch": 0.938082191780822, + "grad_norm": 2.219954013824463, + "learning_rate": 3.84646628757108e-06, + "loss": 1.8651, + "step": 856 + }, + { + "epoch": 0.9391780821917808, + "grad_norm": 2.2897071838378906, + "learning_rate": 3.8444354183590585e-06, + "loss": 1.8744, + "step": 857 + }, + { + "epoch": 0.9402739726027397, + "grad_norm": 2.1656205654144287, + "learning_rate": 3.842404549147035e-06, + "loss": 1.8622, + "step": 858 + }, + { + "epoch": 0.9413698630136986, + "grad_norm": 2.20072603225708, + "learning_rate": 3.840373679935013e-06, + "loss": 1.8678, + "step": 859 + }, + { + "epoch": 0.9424657534246575, + "grad_norm": 2.1474406719207764, + "learning_rate": 3.8383428107229894e-06, + "loss": 1.8462, + "step": 860 + }, + { + "epoch": 0.9424657534246575, + "eval_loss": 2.0451602935791016, + "eval_runtime": 37.1383, + "eval_samples_per_second": 26.926, + "eval_steps_per_second": 1.131, + "step": 860 + }, + { + "epoch": 0.9435616438356165, + "grad_norm": 1.9542036056518555, + "learning_rate": 3.836311941510967e-06, + "loss": 1.8739, + "step": 861 + }, + { + "epoch": 0.9446575342465754, + "grad_norm": 2.3951354026794434, + "learning_rate": 3.834281072298944e-06, + "loss": 1.8565, + "step": 862 + }, + { + "epoch": 0.9457534246575342, + "grad_norm": 2.4916961193084717, + "learning_rate": 3.832250203086921e-06, + "loss": 1.9138, + "step": 863 + }, + { + "epoch": 0.9468493150684931, + "grad_norm": 2.1650452613830566, + "learning_rate": 3.8302193338748985e-06, + "loss": 1.8499, + "step": 864 + }, + { + "epoch": 0.947945205479452, + "grad_norm": 2.2215073108673096, + "learning_rate": 3.828188464662876e-06, + "loss": 1.9376, + "step": 865 + }, + { + "epoch": 0.949041095890411, + "grad_norm": 2.3827078342437744, + "learning_rate": 3.826157595450853e-06, + "loss": 1.9726, + "step": 866 + }, + { + "epoch": 0.9501369863013699, + "grad_norm": 2.37194561958313, + "learning_rate": 3.82412672623883e-06, + "loss": 1.9009, + "step": 867 + }, + { + "epoch": 0.9512328767123288, + "grad_norm": 2.116562604904175, + "learning_rate": 3.822095857026808e-06, + "loss": 1.8342, + "step": 868 + }, + { + "epoch": 0.9523287671232876, + "grad_norm": 2.3410849571228027, + "learning_rate": 3.820064987814785e-06, + "loss": 1.9035, + "step": 869 + }, + { + "epoch": 0.9534246575342465, + "grad_norm": 2.207545280456543, + "learning_rate": 3.818034118602762e-06, + "loss": 1.8539, + "step": 870 + }, + { + "epoch": 0.9534246575342465, + "eval_loss": 2.0433826446533203, + "eval_runtime": 37.114, + "eval_samples_per_second": 26.944, + "eval_steps_per_second": 1.132, + "step": 870 + }, + { + "epoch": 0.9545205479452055, + "grad_norm": 2.1716196537017822, + "learning_rate": 3.8160032493907395e-06, + "loss": 1.8557, + "step": 871 + }, + { + "epoch": 0.9556164383561644, + "grad_norm": 2.213124990463257, + "learning_rate": 3.813972380178717e-06, + "loss": 1.8638, + "step": 872 + }, + { + "epoch": 0.9567123287671233, + "grad_norm": 2.3185672760009766, + "learning_rate": 3.811941510966694e-06, + "loss": 1.8645, + "step": 873 + }, + { + "epoch": 0.9578082191780822, + "grad_norm": 2.453956127166748, + "learning_rate": 3.809910641754671e-06, + "loss": 1.9183, + "step": 874 + }, + { + "epoch": 0.958904109589041, + "grad_norm": 2.1971592903137207, + "learning_rate": 3.8078797725426486e-06, + "loss": 1.8445, + "step": 875 + }, + { + "epoch": 0.96, + "grad_norm": 2.273247241973877, + "learning_rate": 3.8058489033306254e-06, + "loss": 1.8024, + "step": 876 + }, + { + "epoch": 0.9610958904109589, + "grad_norm": 2.1687841415405273, + "learning_rate": 3.803818034118603e-06, + "loss": 1.8931, + "step": 877 + }, + { + "epoch": 0.9621917808219178, + "grad_norm": 2.326498508453369, + "learning_rate": 3.80178716490658e-06, + "loss": 1.8587, + "step": 878 + }, + { + "epoch": 0.9632876712328767, + "grad_norm": 2.0317249298095703, + "learning_rate": 3.7997562956945577e-06, + "loss": 1.9394, + "step": 879 + }, + { + "epoch": 0.9643835616438357, + "grad_norm": 2.3171892166137695, + "learning_rate": 3.7977254264825345e-06, + "loss": 1.92, + "step": 880 + }, + { + "epoch": 0.9643835616438357, + "eval_loss": 2.0431556701660156, + "eval_runtime": 37.1174, + "eval_samples_per_second": 26.942, + "eval_steps_per_second": 1.132, + "step": 880 + }, + { + "epoch": 0.9654794520547946, + "grad_norm": 2.2215981483459473, + "learning_rate": 3.7956945572705122e-06, + "loss": 1.8658, + "step": 881 + }, + { + "epoch": 0.9665753424657534, + "grad_norm": 2.2940685749053955, + "learning_rate": 3.793663688058489e-06, + "loss": 1.8528, + "step": 882 + }, + { + "epoch": 0.9676712328767123, + "grad_norm": 2.2218081951141357, + "learning_rate": 3.791632818846467e-06, + "loss": 1.9446, + "step": 883 + }, + { + "epoch": 0.9687671232876712, + "grad_norm": 2.2608420848846436, + "learning_rate": 3.7896019496344436e-06, + "loss": 1.9433, + "step": 884 + }, + { + "epoch": 0.9698630136986301, + "grad_norm": 2.0247788429260254, + "learning_rate": 3.787571080422421e-06, + "loss": 1.9286, + "step": 885 + }, + { + "epoch": 0.9709589041095891, + "grad_norm": 2.164071798324585, + "learning_rate": 3.785540211210398e-06, + "loss": 1.9349, + "step": 886 + }, + { + "epoch": 0.972054794520548, + "grad_norm": 2.089215040206909, + "learning_rate": 3.7835093419983755e-06, + "loss": 1.9077, + "step": 887 + }, + { + "epoch": 0.9731506849315068, + "grad_norm": 2.245256185531616, + "learning_rate": 3.7814784727863527e-06, + "loss": 1.8911, + "step": 888 + }, + { + "epoch": 0.9742465753424657, + "grad_norm": 2.1227362155914307, + "learning_rate": 3.77944760357433e-06, + "loss": 1.8761, + "step": 889 + }, + { + "epoch": 0.9753424657534246, + "grad_norm": 2.1797730922698975, + "learning_rate": 3.7774167343623073e-06, + "loss": 1.9954, + "step": 890 + }, + { + "epoch": 0.9753424657534246, + "eval_loss": 2.034895896911621, + "eval_runtime": 37.1682, + "eval_samples_per_second": 26.905, + "eval_steps_per_second": 1.13, + "step": 890 + }, + { + "epoch": 0.9764383561643836, + "grad_norm": 2.171062707901001, + "learning_rate": 3.7753858651502846e-06, + "loss": 1.9287, + "step": 891 + }, + { + "epoch": 0.9775342465753425, + "grad_norm": 2.1950814723968506, + "learning_rate": 3.773354995938262e-06, + "loss": 1.8587, + "step": 892 + }, + { + "epoch": 0.9786301369863014, + "grad_norm": 2.2582995891571045, + "learning_rate": 3.771324126726239e-06, + "loss": 1.9269, + "step": 893 + }, + { + "epoch": 0.9797260273972602, + "grad_norm": 2.218113660812378, + "learning_rate": 3.7692932575142164e-06, + "loss": 1.9007, + "step": 894 + }, + { + "epoch": 0.9808219178082191, + "grad_norm": 2.187178134918213, + "learning_rate": 3.7672623883021937e-06, + "loss": 1.9326, + "step": 895 + }, + { + "epoch": 0.9819178082191781, + "grad_norm": 2.20451283454895, + "learning_rate": 3.765231519090171e-06, + "loss": 1.8644, + "step": 896 + }, + { + "epoch": 0.983013698630137, + "grad_norm": 2.1399290561676025, + "learning_rate": 3.7632006498781482e-06, + "loss": 1.7875, + "step": 897 + }, + { + "epoch": 0.9841095890410959, + "grad_norm": 2.049473524093628, + "learning_rate": 3.761169780666125e-06, + "loss": 1.8987, + "step": 898 + }, + { + "epoch": 0.9852054794520548, + "grad_norm": 2.154444456100464, + "learning_rate": 3.7591389114541028e-06, + "loss": 1.8475, + "step": 899 + }, + { + "epoch": 0.9863013698630136, + "grad_norm": 2.1840224266052246, + "learning_rate": 3.7571080422420796e-06, + "loss": 1.8966, + "step": 900 + }, + { + "epoch": 0.9863013698630136, + "eval_loss": 2.0326430797576904, + "eval_runtime": 37.1551, + "eval_samples_per_second": 26.914, + "eval_steps_per_second": 1.13, + "step": 900 + } + ], + "logging_steps": 1, + "max_steps": 2736, + "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.805578468425728e+17, + "train_batch_size": 3, + "trial_name": null, + "trial_params": null +} diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/zero_to_fp32.py b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/zero_to_fp32.py new file mode 100644 index 0000000000000000000000000000000000000000..24cc342e78d1a006c782b3a4cd68d9ce786d8fd8 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_deterministic21_10M_seed0/runs/checkpoint-900/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/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5924756ba0fa0a338016d10f03c2cbb9d5b2fbdd --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-100/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ffcdea323ecb3de20f12357407fae43013f3ebf4837af7284ed753bae0d0a31b +size 4965798912 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1000/model-00002-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1000/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..788a1032c3578393d78c999e5e70be0fe914315d --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1000/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bede7c4ca00c814122345581851fe70d1ffd39eb584d8c2506155c3f7ed13614 +size 2247734920 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1200/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1200/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f61b7c6870edf23a90d7b900e7df2e0c86d2c388 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1200/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29c59df358820d3ebd015c17f3562637fb8d4583f8de0ee694387e046695fe21 +size 4965798912 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1700/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1700/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..613ab4ce5fe9bdb41ade030fb082b4fa14eaab72 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-1700/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0a9cab626cc484f93bce919a985bd53bb285f3ebec81e1504f06571b99d35b0f +size 4965798912 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-200/model-00002-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-200/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b8821292fa9f9cdce09e757081c6ac03d29483d6 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-200/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5169b15f32e37b995efc90a596068f38664ac73c954aedb77043d90546717487 +size 2247734920 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2000/model-00002-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2000/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..86772c8e9fe589e22200f715ee5b22dd8f410183 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2000/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e0dfdc6cc00e6f89319398feb02823ee0580de8ecdb5434f2049d63fa94af9e +size 2247734920 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2736/model-00002-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2736/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..53f28757cabfd43a0be53dafc844fdead23fecdb --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-2736/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:87c492f350d29ebe8b915939528ecd49beb37d98042b050ec518a2f4de4999cc +size 2247734920 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-500/model-00002-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-500/model-00002-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..656e35e3fdbaf5f4b88baf544eb5e839f25c6bcd --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-500/model-00002-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a07687d6097a20b115959a1dd134c21f3c036450fd6748062f073d26e962415b +size 2247734920 diff --git a/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..640b2d94d969a3b9f7d2b7db52efcb13370b81a6 --- /dev/null +++ b/checkpoints/Llama-3.2-3B/babylm_shuffle_even_odd_10M_seed0/runs/checkpoint-900/model-00001-of-00002.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e34092eb08bc0732e44e51d902e8effd9dfeeb74fd344a4ad1fbb795594046d7 +size 4965798912