rabiulawal commited on
Commit
7048b2e
·
verified ·
1 Parent(s): 1dbdee5

Add files using upload-large-folder tool

Browse files
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/mnt/stardoc/escher/checkpoints/Emu3-Base-SFT-without_cot-Mar04_256_lr1e-4/checkpoint-11600",
3
+ "architectures": [
4
+ "Emu3ForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.1,
7
+ "auto_map": {
8
+ "AutoConfig": "BAAI/Emu3-Stage1--configuration_emu3.Emu3Config",
9
+ "AutoModelForCausalLM": "BAAI/Emu3-Stage1--modeling_emu3.Emu3ForCausalLM"
10
+ },
11
+ "boi_token_id": 151852,
12
+ "bos_token_id": 151849,
13
+ "eof_token_id": 151847,
14
+ "eoi_token_id": 151853,
15
+ "eol_token_id": 151846,
16
+ "eos_token_id": 151850,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 4096,
19
+ "image_area": 65536,
20
+ "img_token_id": 151851,
21
+ "initializer_range": 0.02,
22
+ "intermediate_size": 14336,
23
+ "max_position_embeddings": 4300,
24
+ "model_type": "Emu3",
25
+ "num_attention_heads": 32,
26
+ "num_hidden_layers": 32,
27
+ "num_key_value_heads": 8,
28
+ "pad_token_id": 151643,
29
+ "pretraining_tp": 1,
30
+ "rms_norm_eps": 1e-05,
31
+ "rope_scaling": null,
32
+ "rope_theta": 1000000.0,
33
+ "tie_word_embeddings": false,
34
+ "torch_dtype": "bfloat16",
35
+ "transformers_version": "4.44.0",
36
+ "use_cache": false,
37
+ "vocab_size": 184622
38
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 151849,
4
+ "eos_token_id": 151850,
5
+ "pad_token_id": 151643,
6
+ "transformers_version": "4.44.0"
7
+ }
model-00001-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0e93ef690d217f7b0b83b6f45f6708fa75015fed943fe724e80934a6a2c3e11
3
+ size 4884766656
model-00002-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e348d9ca31a8ddbbf733dfc66c8300dd5df40205ca9376fa3577178cd9030b5
3
+ size 4999819320
model-00003-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1aa240be163d8ad7859ccf6f75023f0411a52c1f0914a95af58bd86210d70fc
3
+ size 4915916184
model-00004-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d986ee90a8f7dc9f9c6cdb38f481cbf50308d43ea1e3f5854f5bd858aac67aea
3
+ size 2183554760
model.safetensors.index.json ADDED
@@ -0,0 +1,298 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 16984023040
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00004-of-00004.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00004.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
15
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
16
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
17
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
18
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
19
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
20
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
21
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
22
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
23
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
24
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
25
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
26
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
27
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
28
+ "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
29
+ "model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
30
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
31
+ "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
32
+ "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
33
+ "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
34
+ "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
35
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
36
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
37
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
38
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
39
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
40
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
41
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
42
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
43
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
44
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
45
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
46
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
47
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
48
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
49
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
50
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
51
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
52
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
53
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
54
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
55
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
56
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
57
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
58
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
59
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
60
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
61
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
62
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
63
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
64
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
65
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
66
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
67
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
68
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
69
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
70
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
71
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
72
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
73
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
74
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
75
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
76
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
77
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
78
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
79
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
80
+ "model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
81
+ "model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
82
+ "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
83
+ "model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
84
+ "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
85
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
86
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
87
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
88
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
89
+ "model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
90
+ "model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
91
+ "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
92
+ "model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
93
+ "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
94
+ "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
95
+ "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
96
+ "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
97
+ "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
98
+ "model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
99
+ "model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
100
+ "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
101
+ "model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
102
+ "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
103
+ "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
104
+ "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
105
+ "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
106
+ "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
107
+ "model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
108
+ "model.layers.19.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
109
+ "model.layers.19.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
110
+ "model.layers.19.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
111
+ "model.layers.19.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
112
+ "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
113
+ "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
114
+ "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
115
+ "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
116
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
117
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
118
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
119
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
120
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
121
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
122
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
123
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
124
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
125
+ "model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
126
+ "model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
127
+ "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
128
+ "model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
129
+ "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
130
+ "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
131
+ "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
132
+ "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
133
+ "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
134
+ "model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
135
+ "model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
136
+ "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
137
+ "model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
138
+ "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
139
+ "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
140
+ "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
141
+ "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
142
+ "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
143
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
144
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
145
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
146
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
147
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
148
+ "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
149
+ "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
150
+ "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
151
+ "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
152
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
153
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
154
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
155
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
156
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
157
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
158
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
159
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
160
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
161
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
162
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
163
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
164
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
165
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
166
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
167
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
168
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
169
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
170
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
171
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
172
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
173
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
174
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
175
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
176
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
177
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
178
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
179
+ "model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
180
+ "model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
181
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
182
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
183
+ "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
184
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
185
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
186
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
187
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
188
+ "model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
189
+ "model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
190
+ "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
191
+ "model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
192
+ "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
193
+ "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
194
+ "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
195
+ "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
196
+ "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
197
+ "model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
198
+ "model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
199
+ "model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
200
+ "model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
201
+ "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
202
+ "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
203
+ "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
204
+ "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
205
+ "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
206
+ "model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
207
+ "model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
208
+ "model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
209
+ "model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
210
+ "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
211
+ "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
212
+ "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
213
+ "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
214
+ "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
215
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
216
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
217
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
218
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
219
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
220
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
221
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
222
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
223
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
224
+ "model.layers.30.input_layernorm.weight": "model-00004-of-00004.safetensors",
225
+ "model.layers.30.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
226
+ "model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
227
+ "model.layers.30.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
228
+ "model.layers.30.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
229
+ "model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
230
+ "model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
231
+ "model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
232
+ "model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
233
+ "model.layers.31.input_layernorm.weight": "model-00004-of-00004.safetensors",
234
+ "model.layers.31.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
235
+ "model.layers.31.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
236
+ "model.layers.31.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
237
+ "model.layers.31.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
238
+ "model.layers.31.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
239
+ "model.layers.31.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
240
+ "model.layers.31.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
241
+ "model.layers.31.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
242
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
243
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
244
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
245
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
246
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
247
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
248
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
249
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
250
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
251
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
252
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
253
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
254
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
255
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
256
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
257
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
258
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
259
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
260
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
261
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
262
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
263
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
264
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
265
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
266
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
267
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
268
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
269
+ "model.layers.7.input_layernorm.weight": "model-00002-of-00004.safetensors",
270
+ "model.layers.7.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
271
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
272
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
273
+ "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
274
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
275
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
276
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
277
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
278
+ "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
279
+ "model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
280
+ "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
281
+ "model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
282
+ "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
283
+ "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
284
+ "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
285
+ "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
286
+ "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
287
+ "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
288
+ "model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
289
+ "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
290
+ "model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
291
+ "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
292
+ "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
293
+ "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
294
+ "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
295
+ "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
296
+ "model.norm.weight": "model-00004-of-00004.safetensors"
297
+ }
298
+ }
rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a38be4bafd19c9709c916cbdf3bb0d24a4d54b7beee42a689b90be58f9d0a254
3
+ size 15984
rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df39f271d9dc598368618fdf84ac293e1df8ce296d8c6e60a46dbebde29f1613
3
+ size 15984
rng_state_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d63a96ea9dc866ad74d3a0841e3476e08435b922c44451c0e4f0daf7f1785f5
3
+ size 15984
rng_state_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:711417848bf8a7a1fe368e0e584e27e55cbfba2b4d5c04f272ad77c58b1420a9
3
+ size 15984
rng_state_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:981cfdd4bf58e48d68f23c4a220ba02012fbdb05e47f222ece3cfcd5b4680c46
3
+ size 15984
rng_state_5.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f1fe6a05ea90223c8fd2e6717b22c6d267d3d9cfba7851086a9f35c538ddfbc
3
+ size 15984
rng_state_6.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2996adfe835967a8da93c7c664ecc50dbd0461e36bc10f1f7baaaa0801fb68b0
3
+ size 15984
rng_state_7.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45e68c55c0008bdeed2eacae0df016e3b20226f6f1862f952c00d0c64296c337
3
+ size 15984
trainer_state.json ADDED
@@ -0,0 +1,1089 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.596745243508436,
5
+ "eval_steps": 100,
6
+ "global_step": 4800,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.047864066052411156,
13
+ "grad_norm": 1.8448210254110051,
14
+ "learning_rate": 0.0001,
15
+ "loss": 3.0105,
16
+ "step": 50
17
+ },
18
+ {
19
+ "epoch": 0.09572813210482231,
20
+ "grad_norm": 3.031350218009488,
21
+ "learning_rate": 9.999434313792203e-05,
22
+ "loss": 2.1128,
23
+ "step": 100
24
+ },
25
+ {
26
+ "epoch": 0.09572813210482231,
27
+ "eval_loss": 2.215362310409546,
28
+ "eval_runtime": 21.1651,
29
+ "eval_samples_per_second": 78.856,
30
+ "eval_steps_per_second": 2.504,
31
+ "step": 100
32
+ },
33
+ {
34
+ "epoch": 0.14359219815723345,
35
+ "grad_norm": 1.6850632845975553,
36
+ "learning_rate": 9.997737384462097e-05,
37
+ "loss": 1.9469,
38
+ "step": 150
39
+ },
40
+ {
41
+ "epoch": 0.19145626420964462,
42
+ "grad_norm": 1.2243584455167913,
43
+ "learning_rate": 9.994909599859993e-05,
44
+ "loss": 1.867,
45
+ "step": 200
46
+ },
47
+ {
48
+ "epoch": 0.19145626420964462,
49
+ "eval_loss": 2.0263774394989014,
50
+ "eval_runtime": 21.0874,
51
+ "eval_samples_per_second": 79.147,
52
+ "eval_steps_per_second": 2.513,
53
+ "step": 200
54
+ },
55
+ {
56
+ "epoch": 0.23932033026205576,
57
+ "grad_norm": 1.0729055090983661,
58
+ "learning_rate": 9.990951606304577e-05,
59
+ "loss": 1.8034,
60
+ "step": 250
61
+ },
62
+ {
63
+ "epoch": 0.2871843963144669,
64
+ "grad_norm": 0.9165676753698264,
65
+ "learning_rate": 9.985864308435186e-05,
66
+ "loss": 1.7682,
67
+ "step": 300
68
+ },
69
+ {
70
+ "epoch": 0.2871843963144669,
71
+ "eval_loss": 1.9485853910446167,
72
+ "eval_runtime": 21.0717,
73
+ "eval_samples_per_second": 79.206,
74
+ "eval_steps_per_second": 2.515,
75
+ "step": 300
76
+ },
77
+ {
78
+ "epoch": 0.3350484623668781,
79
+ "grad_norm": 1.1325681880153728,
80
+ "learning_rate": 9.979648869005052e-05,
81
+ "loss": 1.7404,
82
+ "step": 350
83
+ },
84
+ {
85
+ "epoch": 0.38291252841928924,
86
+ "grad_norm": 0.755313580182909,
87
+ "learning_rate": 9.972306708615532e-05,
88
+ "loss": 1.7019,
89
+ "step": 400
90
+ },
91
+ {
92
+ "epoch": 0.38291252841928924,
93
+ "eval_loss": 1.9063043594360352,
94
+ "eval_runtime": 21.0138,
95
+ "eval_samples_per_second": 79.424,
96
+ "eval_steps_per_second": 2.522,
97
+ "step": 400
98
+ },
99
+ {
100
+ "epoch": 0.43077659447170036,
101
+ "grad_norm": 1.1717374264056053,
102
+ "learning_rate": 9.963839505391421e-05,
103
+ "loss": 1.6788,
104
+ "step": 450
105
+ },
106
+ {
107
+ "epoch": 0.4786406605241115,
108
+ "grad_norm": 0.6787604600054761,
109
+ "learning_rate": 9.954249194597396e-05,
110
+ "loss": 1.6453,
111
+ "step": 500
112
+ },
113
+ {
114
+ "epoch": 0.4786406605241115,
115
+ "eval_loss": 1.8607771396636963,
116
+ "eval_runtime": 21.0412,
117
+ "eval_samples_per_second": 79.321,
118
+ "eval_steps_per_second": 2.519,
119
+ "step": 500
120
+ },
121
+ {
122
+ "epoch": 0.5265047265765227,
123
+ "grad_norm": 1.0494549836889278,
124
+ "learning_rate": 9.9435379681957e-05,
125
+ "loss": 1.6312,
126
+ "step": 550
127
+ },
128
+ {
129
+ "epoch": 0.5743687926289338,
130
+ "grad_norm": 0.9009235548667002,
131
+ "learning_rate": 9.931708274345138e-05,
132
+ "loss": 1.6216,
133
+ "step": 600
134
+ },
135
+ {
136
+ "epoch": 0.5743687926289338,
137
+ "eval_loss": 1.8505535125732422,
138
+ "eval_runtime": 21.0451,
139
+ "eval_samples_per_second": 79.306,
140
+ "eval_steps_per_second": 2.518,
141
+ "step": 600
142
+ },
143
+ {
144
+ "epoch": 0.622232858681345,
145
+ "grad_norm": 0.9461906039738908,
146
+ "learning_rate": 9.918762816841531e-05,
147
+ "loss": 1.6006,
148
+ "step": 650
149
+ },
150
+ {
151
+ "epoch": 0.6700969247337561,
152
+ "grad_norm": 0.6498644193748047,
153
+ "learning_rate": 9.904704554499731e-05,
154
+ "loss": 1.5799,
155
+ "step": 700
156
+ },
157
+ {
158
+ "epoch": 0.6700969247337561,
159
+ "eval_loss": 1.8395333290100098,
160
+ "eval_runtime": 21.0713,
161
+ "eval_samples_per_second": 79.207,
162
+ "eval_steps_per_second": 2.515,
163
+ "step": 700
164
+ },
165
+ {
166
+ "epoch": 0.7179609907861673,
167
+ "grad_norm": 0.9074833522097235,
168
+ "learning_rate": 9.889536700477361e-05,
169
+ "loss": 1.5536,
170
+ "step": 750
171
+ },
172
+ {
173
+ "epoch": 0.7658250568385785,
174
+ "grad_norm": 1.2230252736094536,
175
+ "learning_rate": 9.873262721540407e-05,
176
+ "loss": 1.5309,
177
+ "step": 800
178
+ },
179
+ {
180
+ "epoch": 0.7658250568385785,
181
+ "eval_loss": 1.8257369995117188,
182
+ "eval_runtime": 21.043,
183
+ "eval_samples_per_second": 79.314,
184
+ "eval_steps_per_second": 2.519,
185
+ "step": 800
186
+ },
187
+ {
188
+ "epoch": 0.8136891228909896,
189
+ "grad_norm": 0.6253647163206384,
190
+ "learning_rate": 9.855886337270864e-05,
191
+ "loss": 1.5119,
192
+ "step": 850
193
+ },
194
+ {
195
+ "epoch": 0.8615531889434007,
196
+ "grad_norm": 0.8410735177579716,
197
+ "learning_rate": 9.837411519216579e-05,
198
+ "loss": 1.5006,
199
+ "step": 900
200
+ },
201
+ {
202
+ "epoch": 0.8615531889434007,
203
+ "eval_loss": 1.8272072076797485,
204
+ "eval_runtime": 21.0343,
205
+ "eval_samples_per_second": 79.346,
206
+ "eval_steps_per_second": 2.52,
207
+ "step": 900
208
+ },
209
+ {
210
+ "epoch": 0.9094172549958119,
211
+ "grad_norm": 0.8123655725998327,
212
+ "learning_rate": 9.817842489983516e-05,
213
+ "loss": 1.4689,
214
+ "step": 950
215
+ },
216
+ {
217
+ "epoch": 0.957281321048223,
218
+ "grad_norm": 0.8342490687040726,
219
+ "learning_rate": 9.79718372227064e-05,
220
+ "loss": 1.4501,
221
+ "step": 1000
222
+ },
223
+ {
224
+ "epoch": 0.957281321048223,
225
+ "eval_loss": 1.8140053749084473,
226
+ "eval_runtime": 20.9975,
227
+ "eval_samples_per_second": 79.486,
228
+ "eval_steps_per_second": 2.524,
229
+ "step": 1000
230
+ },
231
+ {
232
+ "epoch": 1.0051453871006342,
233
+ "grad_norm": 1.1480754068243832,
234
+ "learning_rate": 9.775439937847636e-05,
235
+ "loss": 1.409,
236
+ "step": 1050
237
+ },
238
+ {
239
+ "epoch": 1.0530094531530454,
240
+ "grad_norm": 0.8492357134830527,
241
+ "learning_rate": 9.752616106475695e-05,
242
+ "loss": 1.3296,
243
+ "step": 1100
244
+ },
245
+ {
246
+ "epoch": 1.0530094531530454,
247
+ "eval_loss": 1.859767198562622,
248
+ "eval_runtime": 21.0033,
249
+ "eval_samples_per_second": 79.464,
250
+ "eval_steps_per_second": 2.523,
251
+ "step": 1100
252
+ },
253
+ {
254
+ "epoch": 1.1008735192054564,
255
+ "grad_norm": 0.968321964763954,
256
+ "learning_rate": 9.728717444771633e-05,
257
+ "loss": 1.3173,
258
+ "step": 1150
259
+ },
260
+ {
261
+ "epoch": 1.1487375852578676,
262
+ "grad_norm": 1.0176178029142426,
263
+ "learning_rate": 9.703749415015574e-05,
264
+ "loss": 1.279,
265
+ "step": 1200
266
+ },
267
+ {
268
+ "epoch": 1.1487375852578676,
269
+ "eval_loss": 1.8694957494735718,
270
+ "eval_runtime": 21.048,
271
+ "eval_samples_per_second": 79.295,
272
+ "eval_steps_per_second": 2.518,
273
+ "step": 1200
274
+ },
275
+ {
276
+ "epoch": 1.1966016513102788,
277
+ "grad_norm": 1.4236331269499436,
278
+ "learning_rate": 9.677717723902497e-05,
279
+ "loss": 1.2846,
280
+ "step": 1250
281
+ },
282
+ {
283
+ "epoch": 1.24446571736269,
284
+ "grad_norm": 0.8091348177738549,
285
+ "learning_rate": 9.650628321237907e-05,
286
+ "loss": 1.2543,
287
+ "step": 1300
288
+ },
289
+ {
290
+ "epoch": 1.24446571736269,
291
+ "eval_loss": 1.9360653162002563,
292
+ "eval_runtime": 21.0174,
293
+ "eval_samples_per_second": 79.41,
294
+ "eval_steps_per_second": 2.522,
295
+ "step": 1300
296
+ },
297
+ {
298
+ "epoch": 1.292329783415101,
299
+ "grad_norm": 0.7101275086446335,
300
+ "learning_rate": 9.622487398577953e-05,
301
+ "loss": 1.2405,
302
+ "step": 1350
303
+ },
304
+ {
305
+ "epoch": 1.3401938494675123,
306
+ "grad_norm": 0.6671838434267677,
307
+ "learning_rate": 9.59330138781428e-05,
308
+ "loss": 1.2367,
309
+ "step": 1400
310
+ },
311
+ {
312
+ "epoch": 1.3401938494675123,
313
+ "eval_loss": 1.9549767971038818,
314
+ "eval_runtime": 21.0544,
315
+ "eval_samples_per_second": 79.271,
316
+ "eval_steps_per_second": 2.517,
317
+ "step": 1400
318
+ },
319
+ {
320
+ "epoch": 1.3880579155199233,
321
+ "grad_norm": 0.6439294837331946,
322
+ "learning_rate": 9.563076959703957e-05,
323
+ "loss": 1.2101,
324
+ "step": 1450
325
+ },
326
+ {
327
+ "epoch": 1.4359219815723345,
328
+ "grad_norm": 0.7664976988303212,
329
+ "learning_rate": 9.531821022344813e-05,
330
+ "loss": 1.2183,
331
+ "step": 1500
332
+ },
333
+ {
334
+ "epoch": 1.4359219815723345,
335
+ "eval_loss": 1.9080383777618408,
336
+ "eval_runtime": 21.0279,
337
+ "eval_samples_per_second": 79.371,
338
+ "eval_steps_per_second": 2.52,
339
+ "step": 1500
340
+ },
341
+ {
342
+ "epoch": 1.4837860476247458,
343
+ "grad_norm": 0.7410488484186188,
344
+ "learning_rate": 9.499540719596515e-05,
345
+ "loss": 1.1824,
346
+ "step": 1550
347
+ },
348
+ {
349
+ "epoch": 1.531650113677157,
350
+ "grad_norm": 0.6607970750322004,
351
+ "learning_rate": 9.466243429447768e-05,
352
+ "loss": 1.1784,
353
+ "step": 1600
354
+ },
355
+ {
356
+ "epoch": 1.531650113677157,
357
+ "eval_loss": 2.0337448120117188,
358
+ "eval_runtime": 21.0701,
359
+ "eval_samples_per_second": 79.212,
360
+ "eval_steps_per_second": 2.515,
361
+ "step": 1600
362
+ },
363
+ {
364
+ "epoch": 1.579514179729568,
365
+ "grad_norm": 0.8404237223930839,
366
+ "learning_rate": 9.431936762330004e-05,
367
+ "loss": 1.1493,
368
+ "step": 1650
369
+ },
370
+ {
371
+ "epoch": 1.6273782457819792,
372
+ "grad_norm": 0.7694756450704843,
373
+ "learning_rate": 9.396628559377941e-05,
374
+ "loss": 1.148,
375
+ "step": 1700
376
+ },
377
+ {
378
+ "epoch": 1.6273782457819792,
379
+ "eval_loss": 2.067599058151245,
380
+ "eval_runtime": 21.1062,
381
+ "eval_samples_per_second": 79.076,
382
+ "eval_steps_per_second": 2.511,
383
+ "step": 1700
384
+ },
385
+ {
386
+ "epoch": 1.6752423118343902,
387
+ "grad_norm": 0.5992205912066282,
388
+ "learning_rate": 9.360326890637407e-05,
389
+ "loss": 1.1228,
390
+ "step": 1750
391
+ },
392
+ {
393
+ "epoch": 1.7231063778868014,
394
+ "grad_norm": 0.8664470255174751,
395
+ "learning_rate": 9.323040053220862e-05,
396
+ "loss": 1.1263,
397
+ "step": 1800
398
+ },
399
+ {
400
+ "epoch": 1.7231063778868014,
401
+ "eval_loss": 2.1029019355773926,
402
+ "eval_runtime": 21.0537,
403
+ "eval_samples_per_second": 79.273,
404
+ "eval_steps_per_second": 2.517,
405
+ "step": 1800
406
+ },
407
+ {
408
+ "epoch": 1.7709704439392127,
409
+ "grad_norm": 0.6459603286404826,
410
+ "learning_rate": 9.284776569410994e-05,
411
+ "loss": 1.1215,
412
+ "step": 1850
413
+ },
414
+ {
415
+ "epoch": 1.8188345099916239,
416
+ "grad_norm": 0.5908873970173206,
417
+ "learning_rate": 9.245545184712873e-05,
418
+ "loss": 1.0918,
419
+ "step": 1900
420
+ },
421
+ {
422
+ "epoch": 1.8188345099916239,
423
+ "eval_loss": 2.2022502422332764,
424
+ "eval_runtime": 21.0942,
425
+ "eval_samples_per_second": 79.121,
426
+ "eval_steps_per_second": 2.513,
427
+ "step": 1900
428
+ },
429
+ {
430
+ "epoch": 1.866698576044035,
431
+ "grad_norm": 0.8614612167467028,
432
+ "learning_rate": 9.205354865855082e-05,
433
+ "loss": 1.0904,
434
+ "step": 1950
435
+ },
436
+ {
437
+ "epoch": 1.914562642096446,
438
+ "grad_norm": 0.7165356146762782,
439
+ "learning_rate": 9.164214798740274e-05,
440
+ "loss": 1.0676,
441
+ "step": 2000
442
+ },
443
+ {
444
+ "epoch": 1.914562642096446,
445
+ "eval_loss": 2.255880117416382,
446
+ "eval_runtime": 21.0788,
447
+ "eval_samples_per_second": 79.179,
448
+ "eval_steps_per_second": 2.514,
449
+ "step": 2000
450
+ },
451
+ {
452
+ "epoch": 1.9624267081488571,
453
+ "grad_norm": 0.788521035581266,
454
+ "learning_rate": 9.122134386345641e-05,
455
+ "loss": 1.0561,
456
+ "step": 2050
457
+ },
458
+ {
459
+ "epoch": 2.0102907742012683,
460
+ "grad_norm": 0.7592948480598436,
461
+ "learning_rate": 9.079123246573775e-05,
462
+ "loss": 1.0499,
463
+ "step": 2100
464
+ },
465
+ {
466
+ "epoch": 2.0102907742012683,
467
+ "eval_loss": 2.354480504989624,
468
+ "eval_runtime": 21.0068,
469
+ "eval_samples_per_second": 79.451,
470
+ "eval_steps_per_second": 2.523,
471
+ "step": 2100
472
+ },
473
+ {
474
+ "epoch": 2.0581548402536796,
475
+ "grad_norm": 0.45289566958700733,
476
+ "learning_rate": 9.035191210054393e-05,
477
+ "loss": 0.9826,
478
+ "step": 2150
479
+ },
480
+ {
481
+ "epoch": 2.106018906306091,
482
+ "grad_norm": 0.6087678469648697,
483
+ "learning_rate": 8.990348317897447e-05,
484
+ "loss": 0.9745,
485
+ "step": 2200
486
+ },
487
+ {
488
+ "epoch": 2.106018906306091,
489
+ "eval_loss": 2.3817625045776367,
490
+ "eval_runtime": 21.0797,
491
+ "eval_samples_per_second": 79.176,
492
+ "eval_steps_per_second": 2.514,
493
+ "step": 2200
494
+ },
495
+ {
496
+ "epoch": 2.153882972358502,
497
+ "grad_norm": 0.5635460239865,
498
+ "learning_rate": 8.944604819398137e-05,
499
+ "loss": 0.9735,
500
+ "step": 2250
501
+ },
502
+ {
503
+ "epoch": 2.201747038410913,
504
+ "grad_norm": 0.4480848478738139,
505
+ "learning_rate": 8.897971169694319e-05,
506
+ "loss": 0.9535,
507
+ "step": 2300
508
+ },
509
+ {
510
+ "epoch": 2.201747038410913,
511
+ "eval_loss": 2.3746585845947266,
512
+ "eval_runtime": 21.0773,
513
+ "eval_samples_per_second": 79.185,
514
+ "eval_steps_per_second": 2.515,
515
+ "step": 2300
516
+ },
517
+ {
518
+ "epoch": 2.249611104463324,
519
+ "grad_norm": 0.7926613203890769,
520
+ "learning_rate": 8.850458027376885e-05,
521
+ "loss": 0.9542,
522
+ "step": 2350
523
+ },
524
+ {
525
+ "epoch": 2.2974751705157352,
526
+ "grad_norm": 0.5265416667258557,
527
+ "learning_rate": 8.802076252053634e-05,
528
+ "loss": 0.9582,
529
+ "step": 2400
530
+ },
531
+ {
532
+ "epoch": 2.2974751705157352,
533
+ "eval_loss": 2.3976855278015137,
534
+ "eval_runtime": 21.0616,
535
+ "eval_samples_per_second": 79.244,
536
+ "eval_steps_per_second": 2.516,
537
+ "step": 2400
538
+ },
539
+ {
540
+ "epoch": 2.3453392365681465,
541
+ "grad_norm": 0.5890939821852947,
542
+ "learning_rate": 8.752836901867192e-05,
543
+ "loss": 0.9604,
544
+ "step": 2450
545
+ },
546
+ {
547
+ "epoch": 2.3932033026205577,
548
+ "grad_norm": 1.004593418572418,
549
+ "learning_rate": 8.702751230967574e-05,
550
+ "loss": 0.9327,
551
+ "step": 2500
552
+ },
553
+ {
554
+ "epoch": 2.3932033026205577,
555
+ "eval_loss": 2.4603569507598877,
556
+ "eval_runtime": 21.1448,
557
+ "eval_samples_per_second": 78.932,
558
+ "eval_steps_per_second": 2.507,
559
+ "step": 2500
560
+ },
561
+ {
562
+ "epoch": 2.441067368672969,
563
+ "grad_norm": 0.5349468531081598,
564
+ "learning_rate": 8.651830686939928e-05,
565
+ "loss": 0.9552,
566
+ "step": 2550
567
+ },
568
+ {
569
+ "epoch": 2.48893143472538,
570
+ "grad_norm": 0.5512335365948596,
571
+ "learning_rate": 8.600086908188067e-05,
572
+ "loss": 0.9461,
573
+ "step": 2600
574
+ },
575
+ {
576
+ "epoch": 2.48893143472538,
577
+ "eval_loss": 2.484816074371338,
578
+ "eval_runtime": 21.0387,
579
+ "eval_samples_per_second": 79.33,
580
+ "eval_steps_per_second": 2.519,
581
+ "step": 2600
582
+ },
583
+ {
584
+ "epoch": 2.536795500777791,
585
+ "grad_norm": 2.4257517359045724,
586
+ "learning_rate": 8.547531721274413e-05,
587
+ "loss": 0.9345,
588
+ "step": 2650
589
+ },
590
+ {
591
+ "epoch": 2.584659566830202,
592
+ "grad_norm": 0.505849093857287,
593
+ "learning_rate": 8.494177138216894e-05,
594
+ "loss": 0.9487,
595
+ "step": 2700
596
+ },
597
+ {
598
+ "epoch": 2.584659566830202,
599
+ "eval_loss": 2.4580276012420654,
600
+ "eval_runtime": 21.0889,
601
+ "eval_samples_per_second": 79.141,
602
+ "eval_steps_per_second": 2.513,
603
+ "step": 2700
604
+ },
605
+ {
606
+ "epoch": 2.6325236328826134,
607
+ "grad_norm": 0.887374714400711,
608
+ "learning_rate": 8.440035353743496e-05,
609
+ "loss": 0.936,
610
+ "step": 2750
611
+ },
612
+ {
613
+ "epoch": 2.6803876989350246,
614
+ "grad_norm": 0.4720229831631241,
615
+ "learning_rate": 8.385118742505025e-05,
616
+ "loss": 0.919,
617
+ "step": 2800
618
+ },
619
+ {
620
+ "epoch": 2.6803876989350246,
621
+ "eval_loss": 2.530785322189331,
622
+ "eval_runtime": 21.0488,
623
+ "eval_samples_per_second": 79.292,
624
+ "eval_steps_per_second": 2.518,
625
+ "step": 2800
626
+ },
627
+ {
628
+ "epoch": 2.728251764987436,
629
+ "grad_norm": 0.5194208757023896,
630
+ "learning_rate": 8.329439856246754e-05,
631
+ "loss": 0.9192,
632
+ "step": 2850
633
+ },
634
+ {
635
+ "epoch": 2.7761158310398466,
636
+ "grad_norm": 0.6580393941466575,
637
+ "learning_rate": 8.273011420939601e-05,
638
+ "loss": 0.9236,
639
+ "step": 2900
640
+ },
641
+ {
642
+ "epoch": 2.7761158310398466,
643
+ "eval_loss": 2.5706753730773926,
644
+ "eval_runtime": 21.0715,
645
+ "eval_samples_per_second": 79.207,
646
+ "eval_steps_per_second": 2.515,
647
+ "step": 2900
648
+ },
649
+ {
650
+ "epoch": 2.823979897092258,
651
+ "grad_norm": 0.4499866190476793,
652
+ "learning_rate": 8.215846333871479e-05,
653
+ "loss": 0.9055,
654
+ "step": 2950
655
+ },
656
+ {
657
+ "epoch": 2.871843963144669,
658
+ "grad_norm": 0.47319436577853424,
659
+ "learning_rate": 8.157957660699487e-05,
660
+ "loss": 0.9173,
661
+ "step": 3000
662
+ },
663
+ {
664
+ "epoch": 2.871843963144669,
665
+ "eval_loss": 2.5460734367370605,
666
+ "eval_runtime": 21.0903,
667
+ "eval_samples_per_second": 79.136,
668
+ "eval_steps_per_second": 2.513,
669
+ "step": 3000
670
+ },
671
+ {
672
+ "epoch": 2.9197080291970803,
673
+ "grad_norm": 0.5406088894148916,
674
+ "learning_rate": 8.099358632463625e-05,
675
+ "loss": 0.8987,
676
+ "step": 3050
677
+ },
678
+ {
679
+ "epoch": 2.9675720952494915,
680
+ "grad_norm": 0.5415925821849361,
681
+ "learning_rate": 8.040062642562707e-05,
682
+ "loss": 0.9149,
683
+ "step": 3100
684
+ },
685
+ {
686
+ "epoch": 2.9675720952494915,
687
+ "eval_loss": 2.5993332862854004,
688
+ "eval_runtime": 21.0799,
689
+ "eval_samples_per_second": 79.175,
690
+ "eval_steps_per_second": 2.514,
691
+ "step": 3100
692
+ },
693
+ {
694
+ "epoch": 3.0154361613019027,
695
+ "grad_norm": 0.5895563971474631,
696
+ "learning_rate": 7.980083243693163e-05,
697
+ "loss": 0.8721,
698
+ "step": 3150
699
+ },
700
+ {
701
+ "epoch": 3.063300227354314,
702
+ "grad_norm": 0.5196758581908888,
703
+ "learning_rate": 7.919434144751435e-05,
704
+ "loss": 0.8001,
705
+ "step": 3200
706
+ },
707
+ {
708
+ "epoch": 3.063300227354314,
709
+ "eval_loss": 2.716139793395996,
710
+ "eval_runtime": 21.042,
711
+ "eval_samples_per_second": 79.318,
712
+ "eval_steps_per_second": 2.519,
713
+ "step": 3200
714
+ },
715
+ {
716
+ "epoch": 3.1111642934067247,
717
+ "grad_norm": 0.4270428798513727,
718
+ "learning_rate": 7.858129207700669e-05,
719
+ "loss": 0.8201,
720
+ "step": 3250
721
+ },
722
+ {
723
+ "epoch": 3.159028359459136,
724
+ "grad_norm": 0.3627950292769719,
725
+ "learning_rate": 7.796182444402422e-05,
726
+ "loss": 0.8155,
727
+ "step": 3300
728
+ },
729
+ {
730
+ "epoch": 3.159028359459136,
731
+ "eval_loss": 2.7008538246154785,
732
+ "eval_runtime": 21.1035,
733
+ "eval_samples_per_second": 79.087,
734
+ "eval_steps_per_second": 2.511,
735
+ "step": 3300
736
+ },
737
+ {
738
+ "epoch": 3.206892425511547,
739
+ "grad_norm": 0.4065490298404093,
740
+ "learning_rate": 7.733608013414119e-05,
741
+ "loss": 0.8111,
742
+ "step": 3350
743
+ },
744
+ {
745
+ "epoch": 3.2547564915639584,
746
+ "grad_norm": 0.42889313551956876,
747
+ "learning_rate": 7.670420216752956e-05,
748
+ "loss": 0.7987,
749
+ "step": 3400
750
+ },
751
+ {
752
+ "epoch": 3.2547564915639584,
753
+ "eval_loss": 2.6770336627960205,
754
+ "eval_runtime": 21.0695,
755
+ "eval_samples_per_second": 79.214,
756
+ "eval_steps_per_second": 2.515,
757
+ "step": 3400
758
+ },
759
+ {
760
+ "epoch": 3.3026205576163696,
761
+ "grad_norm": 0.4217567084634303,
762
+ "learning_rate": 7.606633496627045e-05,
763
+ "loss": 0.8145,
764
+ "step": 3450
765
+ },
766
+ {
767
+ "epoch": 3.3504846236687804,
768
+ "grad_norm": 1.6322652790106957,
769
+ "learning_rate": 7.542262432134494e-05,
770
+ "loss": 0.8173,
771
+ "step": 3500
772
+ },
773
+ {
774
+ "epoch": 3.3504846236687804,
775
+ "eval_loss": 2.7087204456329346,
776
+ "eval_runtime": 21.0624,
777
+ "eval_samples_per_second": 79.241,
778
+ "eval_steps_per_second": 2.516,
779
+ "step": 3500
780
+ },
781
+ {
782
+ "epoch": 3.3983486897211916,
783
+ "grad_norm": 0.36566860920317223,
784
+ "learning_rate": 7.477321735931209e-05,
785
+ "loss": 0.8091,
786
+ "step": 3550
787
+ },
788
+ {
789
+ "epoch": 3.446212755773603,
790
+ "grad_norm": 0.4809930959754767,
791
+ "learning_rate": 7.411826250868163e-05,
792
+ "loss": 0.813,
793
+ "step": 3600
794
+ },
795
+ {
796
+ "epoch": 3.446212755773603,
797
+ "eval_loss": 2.717087745666504,
798
+ "eval_runtime": 21.0919,
799
+ "eval_samples_per_second": 79.13,
800
+ "eval_steps_per_second": 2.513,
801
+ "step": 3600
802
+ },
803
+ {
804
+ "epoch": 3.4958717243029795,
805
+ "grad_norm": 0.5496017380355079,
806
+ "learning_rate": 7.345790946598922e-05,
807
+ "loss": 0.8004,
808
+ "step": 3650
809
+ },
810
+ {
811
+ "epoch": 3.5437357903553908,
812
+ "grad_norm": 0.4803109269616191,
813
+ "learning_rate": 7.279230916158176e-05,
814
+ "loss": 0.7928,
815
+ "step": 3700
816
+ },
817
+ {
818
+ "epoch": 3.5437357903553908,
819
+ "eval_loss": 2.7619569301605225,
820
+ "eval_runtime": 21.3609,
821
+ "eval_samples_per_second": 78.134,
822
+ "eval_steps_per_second": 2.481,
823
+ "step": 3700
824
+ },
825
+ {
826
+ "epoch": 3.591599856407802,
827
+ "grad_norm": 0.3252479820212815,
828
+ "learning_rate": 7.212161372512069e-05,
829
+ "loss": 0.7898,
830
+ "step": 3750
831
+ },
832
+ {
833
+ "epoch": 3.639463922460213,
834
+ "grad_norm": 0.9705997172962069,
835
+ "learning_rate": 7.144597645081128e-05,
836
+ "loss": 0.7991,
837
+ "step": 3800
838
+ },
839
+ {
840
+ "epoch": 3.639463922460213,
841
+ "eval_loss": 2.6866140365600586,
842
+ "eval_runtime": 21.343,
843
+ "eval_samples_per_second": 78.199,
844
+ "eval_steps_per_second": 2.483,
845
+ "step": 3800
846
+ },
847
+ {
848
+ "epoch": 3.687327988512624,
849
+ "grad_norm": 0.3273212997136576,
850
+ "learning_rate": 7.076555176236569e-05,
851
+ "loss": 0.7914,
852
+ "step": 3850
853
+ },
854
+ {
855
+ "epoch": 3.7351920545650352,
856
+ "grad_norm": 0.7433963628381278,
857
+ "learning_rate": 7.008049517770782e-05,
858
+ "loss": 0.7937,
859
+ "step": 3900
860
+ },
861
+ {
862
+ "epoch": 3.7351920545650352,
863
+ "eval_loss": 2.733802080154419,
864
+ "eval_runtime": 21.3212,
865
+ "eval_samples_per_second": 78.279,
866
+ "eval_steps_per_second": 2.486,
867
+ "step": 3900
868
+ },
869
+ {
870
+ "epoch": 3.7830561206174464,
871
+ "grad_norm": 0.3278726533320577,
872
+ "learning_rate": 6.939096327342817e-05,
873
+ "loss": 0.7962,
874
+ "step": 3950
875
+ },
876
+ {
877
+ "epoch": 3.8309201866698577,
878
+ "grad_norm": 0.403526762761248,
879
+ "learning_rate": 6.86971136489967e-05,
880
+ "loss": 0.7827,
881
+ "step": 4000
882
+ },
883
+ {
884
+ "epoch": 3.8309201866698577,
885
+ "eval_loss": 2.7678627967834473,
886
+ "eval_runtime": 21.3082,
887
+ "eval_samples_per_second": 78.327,
888
+ "eval_steps_per_second": 2.487,
889
+ "step": 4000
890
+ },
891
+ {
892
+ "epoch": 3.878784252722269,
893
+ "grad_norm": 0.3097142729991792,
894
+ "learning_rate": 6.79991048907418e-05,
895
+ "loss": 0.7986,
896
+ "step": 4050
897
+ },
898
+ {
899
+ "epoch": 3.9266483187746797,
900
+ "grad_norm": 0.40512187799915156,
901
+ "learning_rate": 6.729709653560387e-05,
902
+ "loss": 0.7865,
903
+ "step": 4100
904
+ },
905
+ {
906
+ "epoch": 3.9266483187746797,
907
+ "eval_loss": 2.776564359664917,
908
+ "eval_runtime": 21.3394,
909
+ "eval_samples_per_second": 78.212,
910
+ "eval_steps_per_second": 2.484,
911
+ "step": 4100
912
+ },
913
+ {
914
+ "epoch": 3.974512384827091,
915
+ "grad_norm": 0.33636006513301697,
916
+ "learning_rate": 6.659124903467149e-05,
917
+ "loss": 0.7812,
918
+ "step": 4150
919
+ },
920
+ {
921
+ "epoch": 4.022376450879502,
922
+ "grad_norm": 0.36839187380492466,
923
+ "learning_rate": 6.588172371650876e-05,
924
+ "loss": 0.7423,
925
+ "step": 4200
926
+ },
927
+ {
928
+ "epoch": 4.022376450879502,
929
+ "eval_loss": 2.83719539642334,
930
+ "eval_runtime": 21.3805,
931
+ "eval_samples_per_second": 78.062,
932
+ "eval_steps_per_second": 2.479,
933
+ "step": 4200
934
+ },
935
+ {
936
+ "epoch": 4.070240516931913,
937
+ "grad_norm": 0.304743292660407,
938
+ "learning_rate": 6.516868275028195e-05,
939
+ "loss": 0.7378,
940
+ "step": 4250
941
+ },
942
+ {
943
+ "epoch": 4.118104582984325,
944
+ "grad_norm": 0.5062863759898075,
945
+ "learning_rate": 6.445228910869418e-05,
946
+ "loss": 0.7342,
947
+ "step": 4300
948
+ },
949
+ {
950
+ "epoch": 4.118104582984325,
951
+ "eval_loss": 2.871426582336426,
952
+ "eval_runtime": 21.5168,
953
+ "eval_samples_per_second": 77.567,
954
+ "eval_steps_per_second": 2.463,
955
+ "step": 4300
956
+ },
957
+ {
958
+ "epoch": 4.165968649036736,
959
+ "grad_norm": 0.32802516560491085,
960
+ "learning_rate": 6.37327065307363e-05,
961
+ "loss": 0.7488,
962
+ "step": 4350
963
+ },
964
+ {
965
+ "epoch": 4.213832715089147,
966
+ "grad_norm": 0.4355037658190107,
967
+ "learning_rate": 6.301009948426275e-05,
968
+ "loss": 0.7304,
969
+ "step": 4400
970
+ },
971
+ {
972
+ "epoch": 4.213832715089147,
973
+ "eval_loss": 2.8587775230407715,
974
+ "eval_runtime": 21.3263,
975
+ "eval_samples_per_second": 78.26,
976
+ "eval_steps_per_second": 2.485,
977
+ "step": 4400
978
+ },
979
+ {
980
+ "epoch": 4.261696781141558,
981
+ "grad_norm": 0.31652373408670015,
982
+ "learning_rate": 6.228463312840077e-05,
983
+ "loss": 0.7381,
984
+ "step": 4450
985
+ },
986
+ {
987
+ "epoch": 4.3095608471939695,
988
+ "grad_norm": 0.31684272696520716,
989
+ "learning_rate": 6.155647327580152e-05,
990
+ "loss": 0.7248,
991
+ "step": 4500
992
+ },
993
+ {
994
+ "epoch": 4.3095608471939695,
995
+ "eval_loss": 2.8525378704071045,
996
+ "eval_runtime": 21.3409,
997
+ "eval_samples_per_second": 78.207,
998
+ "eval_steps_per_second": 2.483,
999
+ "step": 4500
1000
+ },
1001
+ {
1002
+ "epoch": 4.35742491324638,
1003
+ "grad_norm": 0.2982698939428768,
1004
+ "learning_rate": 6.082578635474208e-05,
1005
+ "loss": 0.7353,
1006
+ "step": 4550
1007
+ },
1008
+ {
1009
+ "epoch": 4.405288979298791,
1010
+ "grad_norm": 0.337753062517172,
1011
+ "learning_rate": 6.009273937108644e-05,
1012
+ "loss": 0.7345,
1013
+ "step": 4600
1014
+ },
1015
+ {
1016
+ "epoch": 4.405288979298791,
1017
+ "eval_loss": 2.892075538635254,
1018
+ "eval_runtime": 21.4253,
1019
+ "eval_samples_per_second": 77.898,
1020
+ "eval_steps_per_second": 2.474,
1021
+ "step": 4600
1022
+ },
1023
+ {
1024
+ "epoch": 4.453153045351202,
1025
+ "grad_norm": 0.3169770657574298,
1026
+ "learning_rate": 5.9357499870114764e-05,
1027
+ "loss": 0.7346,
1028
+ "step": 4650
1029
+ },
1030
+ {
1031
+ "epoch": 4.5010171114036135,
1032
+ "grad_norm": 0.29290427580511086,
1033
+ "learning_rate": 5.86202358982291e-05,
1034
+ "loss": 0.7267,
1035
+ "step": 4700
1036
+ },
1037
+ {
1038
+ "epoch": 4.5010171114036135,
1039
+ "eval_loss": 2.892781972885132,
1040
+ "eval_runtime": 21.3571,
1041
+ "eval_samples_per_second": 78.147,
1042
+ "eval_steps_per_second": 2.482,
1043
+ "step": 4700
1044
+ },
1045
+ {
1046
+ "epoch": 4.548881177456025,
1047
+ "grad_norm": 0.3367379688199369,
1048
+ "learning_rate": 5.788111596454479e-05,
1049
+ "loss": 0.7281,
1050
+ "step": 4750
1051
+ },
1052
+ {
1053
+ "epoch": 4.596745243508436,
1054
+ "grad_norm": 0.3112101102160335,
1055
+ "learning_rate": 5.7140309002375904e-05,
1056
+ "loss": 0.7426,
1057
+ "step": 4800
1058
+ },
1059
+ {
1060
+ "epoch": 4.596745243508436,
1061
+ "eval_loss": 2.9153106212615967,
1062
+ "eval_runtime": 21.3327,
1063
+ "eval_samples_per_second": 78.237,
1064
+ "eval_steps_per_second": 2.484,
1065
+ "step": 4800
1066
+ }
1067
+ ],
1068
+ "logging_steps": 50,
1069
+ "max_steps": 10440,
1070
+ "num_input_tokens_seen": 0,
1071
+ "num_train_epochs": 10,
1072
+ "save_steps": 400,
1073
+ "stateful_callbacks": {
1074
+ "TrainerControl": {
1075
+ "args": {
1076
+ "should_epoch_stop": false,
1077
+ "should_evaluate": false,
1078
+ "should_log": false,
1079
+ "should_save": true,
1080
+ "should_training_stop": false
1081
+ },
1082
+ "attributes": {}
1083
+ }
1084
+ },
1085
+ "total_flos": 4205401729925120.0,
1086
+ "train_batch_size": 4,
1087
+ "trial_name": null,
1088
+ "trial_params": null
1089
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:596d4d49db26de6f1de8e9adb8e4f0fa87a95138c272478132aa0baf5bfe1077
3
+ size 7224