PEFT
Safetensors
NakshJain commited on
Commit
d380082
·
verified ·
1 Parent(s): c2edd82

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: state-spaces/mamba-790m-hf
3
+ library_name: peft
4
+ ---
5
+
6
+ # Model Card for Model ID
7
+
8
+ <!-- Provide a quick summary of what the model is/does. -->
9
+
10
+
11
+
12
+ ## Model Details
13
+
14
+ ### Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is. -->
17
+
18
+
19
+
20
+ - **Developed by:** [More Information Needed]
21
+ - **Funded by [optional]:** [More Information Needed]
22
+ - **Shared by [optional]:** [More Information Needed]
23
+ - **Model type:** [More Information Needed]
24
+ - **Language(s) (NLP):** [More Information Needed]
25
+ - **License:** [More Information Needed]
26
+ - **Finetuned from model [optional]:** [More Information Needed]
27
+
28
+ ### Model Sources [optional]
29
+
30
+ <!-- Provide the basic links for the model. -->
31
+
32
+ - **Repository:** [More Information Needed]
33
+ - **Paper [optional]:** [More Information Needed]
34
+ - **Demo [optional]:** [More Information Needed]
35
+
36
+ ## Uses
37
+
38
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
+
40
+ ### Direct Use
41
+
42
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
+
44
+ [More Information Needed]
45
+
46
+ ### Downstream Use [optional]
47
+
48
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Out-of-Scope Use
53
+
54
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
+
56
+ [More Information Needed]
57
+
58
+ ## Bias, Risks, and Limitations
59
+
60
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ### Recommendations
65
+
66
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
+
68
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
+
70
+ ## How to Get Started with the Model
71
+
72
+ Use the code below to get started with the model.
73
+
74
+ [More Information Needed]
75
+
76
+ ## Training Details
77
+
78
+ ### Training Data
79
+
80
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
+
82
+ [More Information Needed]
83
+
84
+ ### Training Procedure
85
+
86
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
+
88
+ #### Preprocessing [optional]
89
+
90
+ [More Information Needed]
91
+
92
+
93
+ #### Training Hyperparameters
94
+
95
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
+
97
+ #### Speeds, Sizes, Times [optional]
98
+
99
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
+
101
+ [More Information Needed]
102
+
103
+ ## Evaluation
104
+
105
+ <!-- This section describes the evaluation protocols and provides the results. -->
106
+
107
+ ### Testing Data, Factors & Metrics
108
+
109
+ #### Testing Data
110
+
111
+ <!-- This should link to a Dataset Card if possible. -->
112
+
113
+ [More Information Needed]
114
+
115
+ #### Factors
116
+
117
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Metrics
122
+
123
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
+
125
+ [More Information Needed]
126
+
127
+ ### Results
128
+
129
+ [More Information Needed]
130
+
131
+ #### Summary
132
+
133
+
134
+
135
+ ## Model Examination [optional]
136
+
137
+ <!-- Relevant interpretability work for the model goes here -->
138
+
139
+ [More Information Needed]
140
+
141
+ ## Environmental Impact
142
+
143
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
+
145
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
+
147
+ - **Hardware Type:** [More Information Needed]
148
+ - **Hours used:** [More Information Needed]
149
+ - **Cloud Provider:** [More Information Needed]
150
+ - **Compute Region:** [More Information Needed]
151
+ - **Carbon Emitted:** [More Information Needed]
152
+
153
+ ## Technical Specifications [optional]
154
+
155
+ ### Model Architecture and Objective
156
+
157
+ [More Information Needed]
158
+
159
+ ### Compute Infrastructure
160
+
161
+ [More Information Needed]
162
+
163
+ #### Hardware
164
+
165
+ [More Information Needed]
166
+
167
+ #### Software
168
+
169
+ [More Information Needed]
170
+
171
+ ## Citation [optional]
172
+
173
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
+
175
+ **BibTeX:**
176
+
177
+ [More Information Needed]
178
+
179
+ **APA:**
180
+
181
+ [More Information Needed]
182
+
183
+ ## Glossary [optional]
184
+
185
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
+
187
+ [More Information Needed]
188
+
189
+ ## More Information [optional]
190
+
191
+ [More Information Needed]
192
+
193
+ ## Model Card Authors [optional]
194
+
195
+ [More Information Needed]
196
+
197
+ ## Model Card Contact
198
+
199
+ [More Information Needed]
200
+ ### Framework versions
201
+
202
+ - PEFT 0.13.2
adapter_config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "state-spaces/mamba-790m-hf",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 32,
14
+ "lora_dropout": 0.05,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": null,
18
+ "peft_type": "LORA",
19
+ "r": 32,
20
+ "rank_pattern": {},
21
+ "revision": null,
22
+ "target_modules": [
23
+ "in_proj",
24
+ "x_proj",
25
+ "dt_proj",
26
+ "out_proj"
27
+ ],
28
+ "task_type": "CAUSAL_LM",
29
+ "use_dora": false,
30
+ "use_rslora": false
31
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2ab7e3187a5c2e57f5c0401a098a13dd2aa7dbb5cbe9768c98474a13e8ad3cc
3
+ size 114674136
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e6c42ec0de73015da6bf36b85eaf6ee21a03165056762375af40510a5a620db
3
+ size 94454091
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03a01c3a29763d0d67ef8eef7675c536f2121565a5034417c32ef93fb9ca024c
3
+ size 14645
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f380c4ac6a18bc848da342a46f42861ea54206294b1630bf78ce323046fcfaf7
3
+ size 1465
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|endoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<|endoftext|>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,214 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<|padding|>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "50254": {
23
+ "content": " ",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "50255": {
31
+ "content": " ",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "50256": {
39
+ "content": " ",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "50257": {
47
+ "content": " ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "50258": {
55
+ "content": " ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "50259": {
63
+ "content": " ",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "50260": {
71
+ "content": " ",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "50261": {
79
+ "content": " ",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "50262": {
87
+ "content": " ",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "50263": {
95
+ "content": " ",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "50264": {
103
+ "content": " ",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "50265": {
111
+ "content": " ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "50266": {
119
+ "content": " ",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ },
126
+ "50267": {
127
+ "content": " ",
128
+ "lstrip": false,
129
+ "normalized": true,
130
+ "rstrip": false,
131
+ "single_word": false,
132
+ "special": false
133
+ },
134
+ "50268": {
135
+ "content": " ",
136
+ "lstrip": false,
137
+ "normalized": true,
138
+ "rstrip": false,
139
+ "single_word": false,
140
+ "special": false
141
+ },
142
+ "50269": {
143
+ "content": " ",
144
+ "lstrip": false,
145
+ "normalized": true,
146
+ "rstrip": false,
147
+ "single_word": false,
148
+ "special": false
149
+ },
150
+ "50270": {
151
+ "content": " ",
152
+ "lstrip": false,
153
+ "normalized": true,
154
+ "rstrip": false,
155
+ "single_word": false,
156
+ "special": false
157
+ },
158
+ "50271": {
159
+ "content": " ",
160
+ "lstrip": false,
161
+ "normalized": true,
162
+ "rstrip": false,
163
+ "single_word": false,
164
+ "special": false
165
+ },
166
+ "50272": {
167
+ "content": " ",
168
+ "lstrip": false,
169
+ "normalized": true,
170
+ "rstrip": false,
171
+ "single_word": false,
172
+ "special": false
173
+ },
174
+ "50273": {
175
+ "content": " ",
176
+ "lstrip": false,
177
+ "normalized": true,
178
+ "rstrip": false,
179
+ "single_word": false,
180
+ "special": false
181
+ },
182
+ "50274": {
183
+ "content": " ",
184
+ "lstrip": false,
185
+ "normalized": true,
186
+ "rstrip": false,
187
+ "single_word": false,
188
+ "special": false
189
+ },
190
+ "50275": {
191
+ "content": " ",
192
+ "lstrip": false,
193
+ "normalized": true,
194
+ "rstrip": false,
195
+ "single_word": false,
196
+ "special": false
197
+ },
198
+ "50276": {
199
+ "content": " ",
200
+ "lstrip": false,
201
+ "normalized": true,
202
+ "rstrip": false,
203
+ "single_word": false,
204
+ "special": false
205
+ }
206
+ },
207
+ "bos_token": "<|endoftext|>",
208
+ "clean_up_tokenization_spaces": true,
209
+ "eos_token": "<|endoftext|>",
210
+ "model_max_length": 1000000000000000019884624838656,
211
+ "pad_token": "<|endoftext|>",
212
+ "tokenizer_class": "GPTNeoXTokenizer",
213
+ "unk_token": "<|endoftext|>"
214
+ }
trainer_state.json ADDED
@@ -0,0 +1,1492 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.184820294380188,
3
+ "best_model_checkpoint": "./mamba-790m-phase2/checkpoint-2850",
4
+ "epoch": 1.2324324324324325,
5
+ "eval_steps": 50,
6
+ "global_step": 2850,
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.008648648648648649,
13
+ "grad_norm": 0.13316184282302856,
14
+ "learning_rate": 2.158273381294964e-06,
15
+ "loss": 1.3731,
16
+ "step": 20
17
+ },
18
+ {
19
+ "epoch": 0.017297297297297298,
20
+ "grad_norm": 0.12288903445005417,
21
+ "learning_rate": 4.316546762589928e-06,
22
+ "loss": 1.5117,
23
+ "step": 40
24
+ },
25
+ {
26
+ "epoch": 0.021621621621621623,
27
+ "eval_loss": 1.5164753198623657,
28
+ "eval_runtime": 156.2707,
29
+ "eval_samples_per_second": 9.599,
30
+ "eval_steps_per_second": 2.4,
31
+ "step": 50
32
+ },
33
+ {
34
+ "epoch": 0.025945945945945945,
35
+ "grad_norm": 0.15546943247318268,
36
+ "learning_rate": 6.474820143884892e-06,
37
+ "loss": 1.5425,
38
+ "step": 60
39
+ },
40
+ {
41
+ "epoch": 0.034594594594594595,
42
+ "grad_norm": 0.13446089625358582,
43
+ "learning_rate": 8.633093525179856e-06,
44
+ "loss": 1.5552,
45
+ "step": 80
46
+ },
47
+ {
48
+ "epoch": 0.043243243243243246,
49
+ "grad_norm": 0.1574675589799881,
50
+ "learning_rate": 1.079136690647482e-05,
51
+ "loss": 1.4,
52
+ "step": 100
53
+ },
54
+ {
55
+ "epoch": 0.043243243243243246,
56
+ "eval_loss": 1.511157751083374,
57
+ "eval_runtime": 156.5135,
58
+ "eval_samples_per_second": 9.584,
59
+ "eval_steps_per_second": 2.396,
60
+ "step": 100
61
+ },
62
+ {
63
+ "epoch": 0.05189189189189189,
64
+ "grad_norm": 0.1704365462064743,
65
+ "learning_rate": 1.2949640287769784e-05,
66
+ "loss": 1.3642,
67
+ "step": 120
68
+ },
69
+ {
70
+ "epoch": 0.06054054054054054,
71
+ "grad_norm": 0.24820564687252045,
72
+ "learning_rate": 1.4999998160049665e-05,
73
+ "loss": 1.2921,
74
+ "step": 140
75
+ },
76
+ {
77
+ "epoch": 0.06486486486486487,
78
+ "eval_loss": 1.472154140472412,
79
+ "eval_runtime": 156.2248,
80
+ "eval_samples_per_second": 9.602,
81
+ "eval_steps_per_second": 2.4,
82
+ "step": 150
83
+ },
84
+ {
85
+ "epoch": 0.06918918918918919,
86
+ "grad_norm": 0.19965757429599762,
87
+ "learning_rate": 1.4999188596499784e-05,
88
+ "loss": 1.3836,
89
+ "step": 160
90
+ },
91
+ {
92
+ "epoch": 0.07783783783783783,
93
+ "grad_norm": 0.1957588493824005,
94
+ "learning_rate": 1.4996907255939673e-05,
95
+ "loss": 1.3328,
96
+ "step": 180
97
+ },
98
+ {
99
+ "epoch": 0.08648648648648649,
100
+ "grad_norm": 0.2037293016910553,
101
+ "learning_rate": 1.4993154586101044e-05,
102
+ "loss": 1.4367,
103
+ "step": 200
104
+ },
105
+ {
106
+ "epoch": 0.08648648648648649,
107
+ "eval_loss": 1.4281501770019531,
108
+ "eval_runtime": 156.5874,
109
+ "eval_samples_per_second": 9.579,
110
+ "eval_steps_per_second": 2.395,
111
+ "step": 200
112
+ },
113
+ {
114
+ "epoch": 0.09513513513513513,
115
+ "grad_norm": 0.2824411988258362,
116
+ "learning_rate": 1.4987931323476007e-05,
117
+ "loss": 1.4272,
118
+ "step": 220
119
+ },
120
+ {
121
+ "epoch": 0.10378378378378378,
122
+ "grad_norm": 0.30133140087127686,
123
+ "learning_rate": 1.4981238493172506e-05,
124
+ "loss": 1.2994,
125
+ "step": 240
126
+ },
127
+ {
128
+ "epoch": 0.10810810810810811,
129
+ "eval_loss": 1.3918442726135254,
130
+ "eval_runtime": 156.3152,
131
+ "eval_samples_per_second": 9.596,
132
+ "eval_steps_per_second": 2.399,
133
+ "step": 250
134
+ },
135
+ {
136
+ "epoch": 0.11243243243243244,
137
+ "grad_norm": 0.22853460907936096,
138
+ "learning_rate": 1.4973077408713158e-05,
139
+ "loss": 1.3668,
140
+ "step": 260
141
+ },
142
+ {
143
+ "epoch": 0.12108108108108108,
144
+ "grad_norm": 0.25242647528648376,
145
+ "learning_rate": 1.4963449671777435e-05,
146
+ "loss": 1.3376,
147
+ "step": 280
148
+ },
149
+ {
150
+ "epoch": 0.12972972972972974,
151
+ "grad_norm": 0.2533341348171234,
152
+ "learning_rate": 1.4952357171887349e-05,
153
+ "loss": 1.3953,
154
+ "step": 300
155
+ },
156
+ {
157
+ "epoch": 0.12972972972972974,
158
+ "eval_loss": 1.3653450012207031,
159
+ "eval_runtime": 156.5177,
160
+ "eval_samples_per_second": 9.584,
161
+ "eval_steps_per_second": 2.396,
162
+ "step": 300
163
+ },
164
+ {
165
+ "epoch": 0.13837837837837838,
166
+ "grad_norm": 0.4832008481025696,
167
+ "learning_rate": 1.4939802086036593e-05,
168
+ "loss": 1.4049,
169
+ "step": 320
170
+ },
171
+ {
172
+ "epoch": 0.14702702702702702,
173
+ "grad_norm": 0.27169403433799744,
174
+ "learning_rate": 1.4925786878263313e-05,
175
+ "loss": 1.4256,
176
+ "step": 340
177
+ },
178
+ {
179
+ "epoch": 0.15135135135135136,
180
+ "eval_loss": 1.3452147245407104,
181
+ "eval_runtime": 156.1759,
182
+ "eval_samples_per_second": 9.605,
183
+ "eval_steps_per_second": 2.401,
184
+ "step": 350
185
+ },
186
+ {
187
+ "epoch": 0.15567567567567567,
188
+ "grad_norm": 0.42071032524108887,
189
+ "learning_rate": 1.4910314299166496e-05,
190
+ "loss": 1.3099,
191
+ "step": 360
192
+ },
193
+ {
194
+ "epoch": 0.1643243243243243,
195
+ "grad_norm": 0.37401214241981506,
196
+ "learning_rate": 1.4893387385366164e-05,
197
+ "loss": 1.3174,
198
+ "step": 380
199
+ },
200
+ {
201
+ "epoch": 0.17297297297297298,
202
+ "grad_norm": 0.36374762654304504,
203
+ "learning_rate": 1.4875009458907393e-05,
204
+ "loss": 1.325,
205
+ "step": 400
206
+ },
207
+ {
208
+ "epoch": 0.17297297297297298,
209
+ "eval_loss": 1.3278440237045288,
210
+ "eval_runtime": 155.9619,
211
+ "eval_samples_per_second": 9.618,
212
+ "eval_steps_per_second": 2.404,
213
+ "step": 400
214
+ },
215
+ {
216
+ "epoch": 0.18162162162162163,
217
+ "grad_norm": 0.44585686922073364,
218
+ "learning_rate": 1.4855184126608352e-05,
219
+ "loss": 1.2749,
220
+ "step": 420
221
+ },
222
+ {
223
+ "epoch": 0.19027027027027027,
224
+ "grad_norm": 0.40485116839408875,
225
+ "learning_rate": 1.4833915279352421e-05,
226
+ "loss": 1.1299,
227
+ "step": 440
228
+ },
229
+ {
230
+ "epoch": 0.1945945945945946,
231
+ "eval_loss": 1.3141183853149414,
232
+ "eval_runtime": 156.3105,
233
+ "eval_samples_per_second": 9.596,
234
+ "eval_steps_per_second": 2.399,
235
+ "step": 450
236
+ },
237
+ {
238
+ "epoch": 0.1989189189189189,
239
+ "grad_norm": 0.4696143865585327,
240
+ "learning_rate": 1.4811207091324587e-05,
241
+ "loss": 1.1933,
242
+ "step": 460
243
+ },
244
+ {
245
+ "epoch": 0.20756756756756756,
246
+ "grad_norm": 0.6063587069511414,
247
+ "learning_rate": 1.478706401919221e-05,
248
+ "loss": 1.2983,
249
+ "step": 480
250
+ },
251
+ {
252
+ "epoch": 0.21621621621621623,
253
+ "grad_norm": 0.4177221953868866,
254
+ "learning_rate": 1.4761490801230385e-05,
255
+ "loss": 1.2339,
256
+ "step": 500
257
+ },
258
+ {
259
+ "epoch": 0.21621621621621623,
260
+ "eval_loss": 1.302538275718689,
261
+ "eval_runtime": 156.5384,
262
+ "eval_samples_per_second": 9.582,
263
+ "eval_steps_per_second": 2.396,
264
+ "step": 500
265
+ },
266
+ {
267
+ "epoch": 0.22486486486486487,
268
+ "grad_norm": 0.4580596089363098,
269
+ "learning_rate": 1.4734492456391998e-05,
270
+ "loss": 1.3529,
271
+ "step": 520
272
+ },
273
+ {
274
+ "epoch": 0.23351351351351352,
275
+ "grad_norm": 0.4583284556865692,
276
+ "learning_rate": 1.470607428332273e-05,
277
+ "loss": 1.3169,
278
+ "step": 540
279
+ },
280
+ {
281
+ "epoch": 0.23783783783783785,
282
+ "eval_loss": 1.2932534217834473,
283
+ "eval_runtime": 156.5697,
284
+ "eval_samples_per_second": 9.58,
285
+ "eval_steps_per_second": 2.395,
286
+ "step": 550
287
+ },
288
+ {
289
+ "epoch": 0.24216216216216216,
290
+ "grad_norm": 0.4312450885772705,
291
+ "learning_rate": 1.4676241859321143e-05,
292
+ "loss": 1.1324,
293
+ "step": 560
294
+ },
295
+ {
296
+ "epoch": 0.2508108108108108,
297
+ "grad_norm": 0.5200898051261902,
298
+ "learning_rate": 1.4645001039244101e-05,
299
+ "loss": 1.2754,
300
+ "step": 580
301
+ },
302
+ {
303
+ "epoch": 0.2594594594594595,
304
+ "grad_norm": 0.5838780403137207,
305
+ "learning_rate": 1.4612357954357695e-05,
306
+ "loss": 1.2903,
307
+ "step": 600
308
+ },
309
+ {
310
+ "epoch": 0.2594594594594595,
311
+ "eval_loss": 1.284968376159668,
312
+ "eval_runtime": 156.4193,
313
+ "eval_samples_per_second": 9.59,
314
+ "eval_steps_per_second": 2.397,
315
+ "step": 600
316
+ },
317
+ {
318
+ "epoch": 0.2681081081081081,
319
+ "grad_norm": 0.3660845458507538,
320
+ "learning_rate": 1.4578319011133939e-05,
321
+ "loss": 1.1267,
322
+ "step": 620
323
+ },
324
+ {
325
+ "epoch": 0.27675675675675676,
326
+ "grad_norm": 0.4065239429473877,
327
+ "learning_rate": 1.454289088999345e-05,
328
+ "loss": 1.2034,
329
+ "step": 640
330
+ },
331
+ {
332
+ "epoch": 0.2810810810810811,
333
+ "eval_loss": 1.2779653072357178,
334
+ "eval_runtime": 156.5245,
335
+ "eval_samples_per_second": 9.583,
336
+ "eval_steps_per_second": 2.396,
337
+ "step": 650
338
+ },
339
+ {
340
+ "epoch": 0.28540540540540543,
341
+ "grad_norm": 0.4448115825653076,
342
+ "learning_rate": 1.4506080543994352e-05,
343
+ "loss": 1.1839,
344
+ "step": 660
345
+ },
346
+ {
347
+ "epoch": 0.29405405405405405,
348
+ "grad_norm": 0.741157054901123,
349
+ "learning_rate": 1.4467895197467692e-05,
350
+ "loss": 1.2143,
351
+ "step": 680
352
+ },
353
+ {
354
+ "epoch": 0.3027027027027027,
355
+ "grad_norm": 0.45047101378440857,
356
+ "learning_rate": 1.4428342344599592e-05,
357
+ "loss": 1.2868,
358
+ "step": 700
359
+ },
360
+ {
361
+ "epoch": 0.3027027027027027,
362
+ "eval_loss": 1.2726409435272217,
363
+ "eval_runtime": 156.3492,
364
+ "eval_samples_per_second": 9.594,
365
+ "eval_steps_per_second": 2.398,
366
+ "step": 700
367
+ },
368
+ {
369
+ "epoch": 0.31135135135135134,
370
+ "grad_norm": 0.4149288535118103,
371
+ "learning_rate": 1.4387429747960472e-05,
372
+ "loss": 1.1808,
373
+ "step": 720
374
+ },
375
+ {
376
+ "epoch": 0.32,
377
+ "grad_norm": 0.5024465322494507,
378
+ "learning_rate": 1.4345165436981567e-05,
379
+ "loss": 1.1616,
380
+ "step": 740
381
+ },
382
+ {
383
+ "epoch": 0.32432432432432434,
384
+ "eval_loss": 1.2668416500091553,
385
+ "eval_runtime": 156.232,
386
+ "eval_samples_per_second": 9.601,
387
+ "eval_steps_per_second": 2.4,
388
+ "step": 750
389
+ },
390
+ {
391
+ "epoch": 0.3286486486486486,
392
+ "grad_norm": 0.5940514802932739,
393
+ "learning_rate": 1.430155770637909e-05,
394
+ "loss": 1.2955,
395
+ "step": 760
396
+ },
397
+ {
398
+ "epoch": 0.3372972972972973,
399
+ "grad_norm": 0.4798080623149872,
400
+ "learning_rate": 1.4256615114526333e-05,
401
+ "loss": 1.1858,
402
+ "step": 780
403
+ },
404
+ {
405
+ "epoch": 0.34594594594594597,
406
+ "grad_norm": 0.3747268319129944,
407
+ "learning_rate": 1.4210346481774007e-05,
408
+ "loss": 1.2592,
409
+ "step": 800
410
+ },
411
+ {
412
+ "epoch": 0.34594594594594597,
413
+ "eval_loss": 1.261027216911316,
414
+ "eval_runtime": 156.3223,
415
+ "eval_samples_per_second": 9.596,
416
+ "eval_steps_per_second": 2.399,
417
+ "step": 800
418
+ },
419
+ {
420
+ "epoch": 0.3545945945945946,
421
+ "grad_norm": 0.6627951860427856,
422
+ "learning_rate": 1.4162760888719174e-05,
423
+ "loss": 1.1585,
424
+ "step": 820
425
+ },
426
+ {
427
+ "epoch": 0.36324324324324325,
428
+ "grad_norm": 0.6270909309387207,
429
+ "learning_rate": 1.4113867674423123e-05,
430
+ "loss": 1.2351,
431
+ "step": 840
432
+ },
433
+ {
434
+ "epoch": 0.3675675675675676,
435
+ "eval_loss": 1.2562013864517212,
436
+ "eval_runtime": 156.3737,
437
+ "eval_samples_per_second": 9.592,
438
+ "eval_steps_per_second": 2.398,
439
+ "step": 850
440
+ },
441
+ {
442
+ "epoch": 0.37189189189189187,
443
+ "grad_norm": 0.5605448484420776,
444
+ "learning_rate": 1.4063676434578483e-05,
445
+ "loss": 1.2151,
446
+ "step": 860
447
+ },
448
+ {
449
+ "epoch": 0.38054054054054054,
450
+ "grad_norm": 0.4705483317375183,
451
+ "learning_rate": 1.4012197019626004e-05,
452
+ "loss": 1.1309,
453
+ "step": 880
454
+ },
455
+ {
456
+ "epoch": 0.3891891891891892,
457
+ "grad_norm": 0.5667669773101807,
458
+ "learning_rate": 1.3959439532821314e-05,
459
+ "loss": 1.2531,
460
+ "step": 900
461
+ },
462
+ {
463
+ "epoch": 0.3891891891891892,
464
+ "eval_loss": 1.2514050006866455,
465
+ "eval_runtime": 156.3146,
466
+ "eval_samples_per_second": 9.596,
467
+ "eval_steps_per_second": 2.399,
468
+ "step": 900
469
+ },
470
+ {
471
+ "epoch": 0.3978378378378378,
472
+ "grad_norm": 0.6878731846809387,
473
+ "learning_rate": 1.3905414328252085e-05,
474
+ "loss": 1.2473,
475
+ "step": 920
476
+ },
477
+ {
478
+ "epoch": 0.4064864864864865,
479
+ "grad_norm": 0.799188494682312,
480
+ "learning_rate": 1.3850132008805958e-05,
481
+ "loss": 1.1098,
482
+ "step": 940
483
+ },
484
+ {
485
+ "epoch": 0.41081081081081083,
486
+ "eval_loss": 1.2472807168960571,
487
+ "eval_runtime": 156.3138,
488
+ "eval_samples_per_second": 9.596,
489
+ "eval_steps_per_second": 2.399,
490
+ "step": 950
491
+ },
492
+ {
493
+ "epoch": 0.4151351351351351,
494
+ "grad_norm": 0.5942118763923645,
495
+ "learning_rate": 1.379360342408963e-05,
496
+ "loss": 1.1993,
497
+ "step": 960
498
+ },
499
+ {
500
+ "epoch": 0.4237837837837838,
501
+ "grad_norm": 0.44030797481536865,
502
+ "learning_rate": 1.3735839668299537e-05,
503
+ "loss": 1.2577,
504
+ "step": 980
505
+ },
506
+ {
507
+ "epoch": 0.43243243243243246,
508
+ "grad_norm": 0.5364145040512085,
509
+ "learning_rate": 1.3676852078044528e-05,
510
+ "loss": 1.209,
511
+ "step": 1000
512
+ },
513
+ {
514
+ "epoch": 0.43243243243243246,
515
+ "eval_loss": 1.2433239221572876,
516
+ "eval_runtime": 156.3683,
517
+ "eval_samples_per_second": 9.593,
518
+ "eval_steps_per_second": 2.398,
519
+ "step": 1000
520
+ },
521
+ {
522
+ "epoch": 0.4410810810810811,
523
+ "grad_norm": 0.7929672598838806,
524
+ "learning_rate": 1.3616652230120962e-05,
525
+ "loss": 1.2357,
526
+ "step": 1020
527
+ },
528
+ {
529
+ "epoch": 0.44972972972972974,
530
+ "grad_norm": 0.7581718564033508,
531
+ "learning_rate": 1.3555251939240659e-05,
532
+ "loss": 1.2092,
533
+ "step": 1040
534
+ },
535
+ {
536
+ "epoch": 0.4540540540540541,
537
+ "eval_loss": 1.239459753036499,
538
+ "eval_runtime": 156.2938,
539
+ "eval_samples_per_second": 9.597,
540
+ "eval_steps_per_second": 2.399,
541
+ "step": 1050
542
+ },
543
+ {
544
+ "epoch": 0.45837837837837836,
545
+ "grad_norm": 0.5872713327407837,
546
+ "learning_rate": 1.3492663255712187e-05,
547
+ "loss": 1.2605,
548
+ "step": 1060
549
+ },
550
+ {
551
+ "epoch": 0.46702702702702703,
552
+ "grad_norm": 0.5056009292602539,
553
+ "learning_rate": 1.3428898463075862e-05,
554
+ "loss": 1.2838,
555
+ "step": 1080
556
+ },
557
+ {
558
+ "epoch": 0.4756756756756757,
559
+ "grad_norm": 0.5987468957901001,
560
+ "learning_rate": 1.3363970075693032e-05,
561
+ "loss": 1.2619,
562
+ "step": 1100
563
+ },
564
+ {
565
+ "epoch": 0.4756756756756757,
566
+ "eval_loss": 1.2360332012176514,
567
+ "eval_runtime": 156.4955,
568
+ "eval_samples_per_second": 9.585,
569
+ "eval_steps_per_second": 2.396,
570
+ "step": 1100
571
+ },
572
+ {
573
+ "epoch": 0.4843243243243243,
574
+ "grad_norm": 0.6014208197593689,
575
+ "learning_rate": 1.3297890836290004e-05,
576
+ "loss": 1.2276,
577
+ "step": 1120
578
+ },
579
+ {
580
+ "epoch": 0.492972972972973,
581
+ "grad_norm": 0.7187291383743286,
582
+ "learning_rate": 1.3230673713457208e-05,
583
+ "loss": 1.1775,
584
+ "step": 1140
585
+ },
586
+ {
587
+ "epoch": 0.4972972972972973,
588
+ "eval_loss": 1.2334262132644653,
589
+ "eval_runtime": 156.3371,
590
+ "eval_samples_per_second": 9.595,
591
+ "eval_steps_per_second": 2.399,
592
+ "step": 1150
593
+ },
594
+ {
595
+ "epoch": 0.5016216216216216,
596
+ "grad_norm": 0.6566280722618103,
597
+ "learning_rate": 1.3162331899103977e-05,
598
+ "loss": 1.2047,
599
+ "step": 1160
600
+ },
601
+ {
602
+ "epoch": 0.5102702702702703,
603
+ "grad_norm": 0.8335544466972351,
604
+ "learning_rate": 1.3092878805869546e-05,
605
+ "loss": 1.1767,
606
+ "step": 1180
607
+ },
608
+ {
609
+ "epoch": 0.518918918918919,
610
+ "grad_norm": 0.6835708022117615,
611
+ "learning_rate": 1.3022328064490712e-05,
612
+ "loss": 1.1629,
613
+ "step": 1200
614
+ },
615
+ {
616
+ "epoch": 0.518918918918919,
617
+ "eval_loss": 1.2302730083465576,
618
+ "eval_runtime": 156.5356,
619
+ "eval_samples_per_second": 9.582,
620
+ "eval_steps_per_second": 2.396,
621
+ "step": 1200
622
+ },
623
+ {
624
+ "epoch": 0.5275675675675676,
625
+ "grad_norm": 0.5832236409187317,
626
+ "learning_rate": 1.2950693521126676e-05,
627
+ "loss": 1.1748,
628
+ "step": 1220
629
+ },
630
+ {
631
+ "epoch": 0.5362162162162162,
632
+ "grad_norm": 0.6758794188499451,
633
+ "learning_rate": 1.2877989234641634e-05,
634
+ "loss": 1.1371,
635
+ "step": 1240
636
+ },
637
+ {
638
+ "epoch": 0.5405405405405406,
639
+ "eval_loss": 1.2266525030136108,
640
+ "eval_runtime": 156.281,
641
+ "eval_samples_per_second": 9.598,
642
+ "eval_steps_per_second": 2.4,
643
+ "step": 1250
644
+ },
645
+ {
646
+ "epoch": 0.5448648648648649,
647
+ "grad_norm": 0.5459420680999756,
648
+ "learning_rate": 1.2804229473845609e-05,
649
+ "loss": 1.2151,
650
+ "step": 1260
651
+ },
652
+ {
653
+ "epoch": 0.5535135135135135,
654
+ "grad_norm": 0.5503604412078857,
655
+ "learning_rate": 1.2729428714694075e-05,
656
+ "loss": 1.1291,
657
+ "step": 1280
658
+ },
659
+ {
660
+ "epoch": 0.5621621621621622,
661
+ "grad_norm": 0.6145665645599365,
662
+ "learning_rate": 1.265360163744694e-05,
663
+ "loss": 1.1212,
664
+ "step": 1300
665
+ },
666
+ {
667
+ "epoch": 0.5621621621621622,
668
+ "eval_loss": 1.2243192195892334,
669
+ "eval_runtime": 156.3088,
670
+ "eval_samples_per_second": 9.596,
671
+ "eval_steps_per_second": 2.399,
672
+ "step": 1300
673
+ },
674
+ {
675
+ "epoch": 0.5708108108108109,
676
+ "grad_norm": 0.8270878791809082,
677
+ "learning_rate": 1.257676312378741e-05,
678
+ "loss": 1.0307,
679
+ "step": 1320
680
+ },
681
+ {
682
+ "epoch": 0.5794594594594594,
683
+ "grad_norm": 0.6336683034896851,
684
+ "learning_rate": 1.2498928253901358e-05,
685
+ "loss": 1.143,
686
+ "step": 1340
687
+ },
688
+ {
689
+ "epoch": 0.5837837837837838,
690
+ "eval_loss": 1.2215410470962524,
691
+ "eval_runtime": 156.2606,
692
+ "eval_samples_per_second": 9.599,
693
+ "eval_steps_per_second": 2.4,
694
+ "step": 1350
695
+ },
696
+ {
697
+ "epoch": 0.5881081081081081,
698
+ "grad_norm": 0.5482487678527832,
699
+ "learning_rate": 1.24201123035177e-05,
700
+ "loss": 1.3624,
701
+ "step": 1360
702
+ },
703
+ {
704
+ "epoch": 0.5967567567567568,
705
+ "grad_norm": 0.7318127751350403,
706
+ "learning_rate": 1.2340330740910413e-05,
707
+ "loss": 1.1077,
708
+ "step": 1380
709
+ },
710
+ {
711
+ "epoch": 0.6054054054054054,
712
+ "grad_norm": 0.5615875720977783,
713
+ "learning_rate": 1.2259599223862757e-05,
714
+ "loss": 1.232,
715
+ "step": 1400
716
+ },
717
+ {
718
+ "epoch": 0.6054054054054054,
719
+ "eval_loss": 1.2196455001831055,
720
+ "eval_runtime": 156.1978,
721
+ "eval_samples_per_second": 9.603,
722
+ "eval_steps_per_second": 2.401,
723
+ "step": 1400
724
+ },
725
+ {
726
+ "epoch": 0.614054054054054,
727
+ "grad_norm": 0.6447334885597229,
728
+ "learning_rate": 1.2177933596594313e-05,
729
+ "loss": 1.1058,
730
+ "step": 1420
731
+ },
732
+ {
733
+ "epoch": 0.6227027027027027,
734
+ "grad_norm": 0.8408645391464233,
735
+ "learning_rate": 1.2095349886651426e-05,
736
+ "loss": 1.302,
737
+ "step": 1440
738
+ },
739
+ {
740
+ "epoch": 0.6270270270270271,
741
+ "eval_loss": 1.2167232036590576,
742
+ "eval_runtime": 156.2131,
743
+ "eval_samples_per_second": 9.602,
744
+ "eval_steps_per_second": 2.401,
745
+ "step": 1450
746
+ },
747
+ {
748
+ "epoch": 0.6313513513513513,
749
+ "grad_norm": 0.4599859416484833,
750
+ "learning_rate": 1.2011864301761648e-05,
751
+ "loss": 1.1504,
752
+ "step": 1460
753
+ },
754
+ {
755
+ "epoch": 0.64,
756
+ "grad_norm": 0.9404271841049194,
757
+ "learning_rate": 1.1927493226652876e-05,
758
+ "loss": 1.2024,
759
+ "step": 1480
760
+ },
761
+ {
762
+ "epoch": 0.6486486486486487,
763
+ "grad_norm": 0.5044443011283875,
764
+ "learning_rate": 1.1842253219837676e-05,
765
+ "loss": 1.213,
766
+ "step": 1500
767
+ },
768
+ {
769
+ "epoch": 0.6486486486486487,
770
+ "eval_loss": 1.2149133682250977,
771
+ "eval_runtime": 156.3249,
772
+ "eval_samples_per_second": 9.595,
773
+ "eval_steps_per_second": 2.399,
774
+ "step": 1500
775
+ },
776
+ {
777
+ "epoch": 0.6572972972972972,
778
+ "grad_norm": 0.7133287191390991,
779
+ "learning_rate": 1.1756161010363576e-05,
780
+ "loss": 1.1682,
781
+ "step": 1520
782
+ },
783
+ {
784
+ "epoch": 0.6659459459459459,
785
+ "grad_norm": 0.7438573241233826,
786
+ "learning_rate": 1.1669233494529819e-05,
787
+ "loss": 1.0579,
788
+ "step": 1540
789
+ },
790
+ {
791
+ "epoch": 0.6702702702702703,
792
+ "eval_loss": 1.2130820751190186,
793
+ "eval_runtime": 156.4016,
794
+ "eval_samples_per_second": 9.591,
795
+ "eval_steps_per_second": 2.398,
796
+ "step": 1550
797
+ },
798
+ {
799
+ "epoch": 0.6745945945945946,
800
+ "grad_norm": 0.8739683032035828,
801
+ "learning_rate": 1.158148773257135e-05,
802
+ "loss": 1.0753,
803
+ "step": 1560
804
+ },
805
+ {
806
+ "epoch": 0.6832432432432433,
807
+ "grad_norm": 0.7708102464675903,
808
+ "learning_rate": 1.1492940945310608e-05,
809
+ "loss": 1.2175,
810
+ "step": 1580
811
+ },
812
+ {
813
+ "epoch": 0.6918918918918919,
814
+ "grad_norm": 0.549228847026825,
815
+ "learning_rate": 1.1403610510777778e-05,
816
+ "loss": 1.0838,
817
+ "step": 1600
818
+ },
819
+ {
820
+ "epoch": 0.6918918918918919,
821
+ "eval_loss": 1.2114548683166504,
822
+ "eval_runtime": 156.4625,
823
+ "eval_samples_per_second": 9.587,
824
+ "eval_steps_per_second": 2.397,
825
+ "step": 1600
826
+ },
827
+ {
828
+ "epoch": 0.7005405405405405,
829
+ "grad_norm": 0.9724882245063782,
830
+ "learning_rate": 1.1313513960800224e-05,
831
+ "loss": 1.0542,
832
+ "step": 1620
833
+ },
834
+ {
835
+ "epoch": 0.7091891891891892,
836
+ "grad_norm": 0.8888576030731201,
837
+ "learning_rate": 1.1222668977561725e-05,
838
+ "loss": 1.2136,
839
+ "step": 1640
840
+ },
841
+ {
842
+ "epoch": 0.7135135135135136,
843
+ "eval_loss": 1.2095521688461304,
844
+ "eval_runtime": 156.4366,
845
+ "eval_samples_per_second": 9.589,
846
+ "eval_steps_per_second": 2.397,
847
+ "step": 1650
848
+ },
849
+ {
850
+ "epoch": 0.7178378378378378,
851
+ "grad_norm": 1.1320910453796387,
852
+ "learning_rate": 1.1131093390132205e-05,
853
+ "loss": 1.1844,
854
+ "step": 1660
855
+ },
856
+ {
857
+ "epoch": 0.7264864864864865,
858
+ "grad_norm": 0.8856949210166931,
859
+ "learning_rate": 1.1038805170968615e-05,
860
+ "loss": 1.1142,
861
+ "step": 1680
862
+ },
863
+ {
864
+ "epoch": 0.7351351351351352,
865
+ "grad_norm": 1.0155227184295654,
866
+ "learning_rate": 1.0945822432387722e-05,
867
+ "loss": 1.1138,
868
+ "step": 1700
869
+ },
870
+ {
871
+ "epoch": 0.7351351351351352,
872
+ "eval_loss": 1.207614779472351,
873
+ "eval_runtime": 156.4751,
874
+ "eval_samples_per_second": 9.586,
875
+ "eval_steps_per_second": 2.397,
876
+ "step": 1700
877
+ },
878
+ {
879
+ "epoch": 0.7437837837837837,
880
+ "grad_norm": 0.8122560977935791,
881
+ "learning_rate": 1.0852163423011382e-05,
882
+ "loss": 1.1304,
883
+ "step": 1720
884
+ },
885
+ {
886
+ "epoch": 0.7524324324324324,
887
+ "grad_norm": 0.8628650903701782,
888
+ "learning_rate": 1.0757846524185135e-05,
889
+ "loss": 1.1426,
890
+ "step": 1740
891
+ },
892
+ {
893
+ "epoch": 0.7567567567567568,
894
+ "eval_loss": 1.2060997486114502,
895
+ "eval_runtime": 156.2117,
896
+ "eval_samples_per_second": 9.602,
897
+ "eval_steps_per_second": 2.401,
898
+ "step": 1750
899
+ },
900
+ {
901
+ "epoch": 0.7610810810810811,
902
+ "grad_norm": 1.243729591369629,
903
+ "learning_rate": 1.0662890246370682e-05,
904
+ "loss": 1.0881,
905
+ "step": 1760
906
+ },
907
+ {
908
+ "epoch": 0.7697297297297298,
909
+ "grad_norm": 0.8714478015899658,
910
+ "learning_rate": 1.0567313225513085e-05,
911
+ "loss": 1.2214,
912
+ "step": 1780
913
+ },
914
+ {
915
+ "epoch": 0.7783783783783784,
916
+ "grad_norm": 0.44382530450820923,
917
+ "learning_rate": 1.0471134219383299e-05,
918
+ "loss": 1.2328,
919
+ "step": 1800
920
+ },
921
+ {
922
+ "epoch": 0.7783783783783784,
923
+ "eval_loss": 1.2047566175460815,
924
+ "eval_runtime": 156.372,
925
+ "eval_samples_per_second": 9.593,
926
+ "eval_steps_per_second": 2.398,
927
+ "step": 1800
928
+ },
929
+ {
930
+ "epoch": 0.787027027027027,
931
+ "grad_norm": 1.0515871047973633,
932
+ "learning_rate": 1.0374372103896792e-05,
933
+ "loss": 1.1519,
934
+ "step": 1820
935
+ },
936
+ {
937
+ "epoch": 0.7956756756756757,
938
+ "grad_norm": 0.8955100178718567,
939
+ "learning_rate": 1.027704586940902e-05,
940
+ "loss": 1.0786,
941
+ "step": 1840
942
+ },
943
+ {
944
+ "epoch": 0.8,
945
+ "eval_loss": 1.2030872106552124,
946
+ "eval_runtime": 156.2817,
947
+ "eval_samples_per_second": 9.598,
948
+ "eval_steps_per_second": 2.4,
949
+ "step": 1850
950
+ },
951
+ {
952
+ "epoch": 0.8043243243243243,
953
+ "grad_norm": 0.6575765609741211,
954
+ "learning_rate": 1.0179174616988392e-05,
955
+ "loss": 1.2385,
956
+ "step": 1860
957
+ },
958
+ {
959
+ "epoch": 0.812972972972973,
960
+ "grad_norm": 0.5790345072746277,
961
+ "learning_rate": 1.008077755466754e-05,
962
+ "loss": 1.135,
963
+ "step": 1880
964
+ },
965
+ {
966
+ "epoch": 0.8216216216216217,
967
+ "grad_norm": 0.7308002710342407,
968
+ "learning_rate": 9.981873993673584e-06,
969
+ "loss": 1.2052,
970
+ "step": 1900
971
+ },
972
+ {
973
+ "epoch": 0.8216216216216217,
974
+ "eval_loss": 1.2018324136734009,
975
+ "eval_runtime": 156.3907,
976
+ "eval_samples_per_second": 9.591,
977
+ "eval_steps_per_second": 2.398,
978
+ "step": 1900
979
+ },
980
+ {
981
+ "epoch": 0.8302702702702702,
982
+ "grad_norm": 0.47168049216270447,
983
+ "learning_rate": 9.882483344638144e-06,
984
+ "loss": 1.1291,
985
+ "step": 1920
986
+ },
987
+ {
988
+ "epoch": 0.8389189189189189,
989
+ "grad_norm": 1.0765182971954346,
990
+ "learning_rate": 9.78262511378785e-06,
991
+ "loss": 1.2196,
992
+ "step": 1940
993
+ },
994
+ {
995
+ "epoch": 0.8432432432432433,
996
+ "eval_loss": 1.200095534324646,
997
+ "eval_runtime": 156.3628,
998
+ "eval_samples_per_second": 9.593,
999
+ "eval_steps_per_second": 2.398,
1000
+ "step": 1950
1001
+ },
1002
+ {
1003
+ "epoch": 0.8475675675675676,
1004
+ "grad_norm": 0.9777532815933228,
1005
+ "learning_rate": 9.682318899116075e-06,
1006
+ "loss": 1.2965,
1007
+ "step": 1960
1008
+ },
1009
+ {
1010
+ "epoch": 0.8562162162162162,
1011
+ "grad_norm": 0.6992659568786621,
1012
+ "learning_rate": 9.58158438653668e-06,
1013
+ "loss": 1.2705,
1014
+ "step": 1980
1015
+ },
1016
+ {
1017
+ "epoch": 0.8648648648648649,
1018
+ "grad_norm": 0.7154545783996582,
1019
+ "learning_rate": 9.480441346020491e-06,
1020
+ "loss": 1.1538,
1021
+ "step": 2000
1022
+ },
1023
+ {
1024
+ "epoch": 0.8648648648648649,
1025
+ "eval_loss": 1.1992037296295166,
1026
+ "eval_runtime": 156.5695,
1027
+ "eval_samples_per_second": 9.58,
1028
+ "eval_steps_per_second": 2.395,
1029
+ "step": 2000
1030
+ },
1031
+ {
1032
+ "epoch": 0.8735135135135135,
1033
+ "grad_norm": 0.9878578782081604,
1034
+ "learning_rate": 9.378909627715276e-06,
1035
+ "loss": 1.1575,
1036
+ "step": 2020
1037
+ },
1038
+ {
1039
+ "epoch": 0.8821621621621621,
1040
+ "grad_norm": 0.7160919308662415,
1041
+ "learning_rate": 9.277009158049991e-06,
1042
+ "loss": 1.2407,
1043
+ "step": 2040
1044
+ },
1045
+ {
1046
+ "epoch": 0.8864864864864865,
1047
+ "eval_loss": 1.1983826160430908,
1048
+ "eval_runtime": 156.5345,
1049
+ "eval_samples_per_second": 9.583,
1050
+ "eval_steps_per_second": 2.396,
1051
+ "step": 2050
1052
+ },
1053
+ {
1054
+ "epoch": 0.8908108108108108,
1055
+ "grad_norm": 0.8028165102005005,
1056
+ "learning_rate": 9.174759935824059e-06,
1057
+ "loss": 1.0845,
1058
+ "step": 2060
1059
+ },
1060
+ {
1061
+ "epoch": 0.8994594594594595,
1062
+ "grad_norm": 0.5290041565895081,
1063
+ "learning_rate": 9.072182028282447e-06,
1064
+ "loss": 1.1363,
1065
+ "step": 2080
1066
+ },
1067
+ {
1068
+ "epoch": 0.9081081081081082,
1069
+ "grad_norm": 0.8659726977348328,
1070
+ "learning_rate": 8.96929556717729e-06,
1071
+ "loss": 1.1844,
1072
+ "step": 2100
1073
+ },
1074
+ {
1075
+ "epoch": 0.9081081081081082,
1076
+ "eval_loss": 1.1970139741897583,
1077
+ "eval_runtime": 156.645,
1078
+ "eval_samples_per_second": 9.576,
1079
+ "eval_steps_per_second": 2.394,
1080
+ "step": 2100
1081
+ },
1082
+ {
1083
+ "epoch": 0.9167567567567567,
1084
+ "grad_norm": 0.7906278371810913,
1085
+ "learning_rate": 8.866120744816888e-06,
1086
+ "loss": 1.1765,
1087
+ "step": 2120
1088
+ },
1089
+ {
1090
+ "epoch": 0.9254054054054054,
1091
+ "grad_norm": 0.7997767925262451,
1092
+ "learning_rate": 8.762677810102787e-06,
1093
+ "loss": 1.0805,
1094
+ "step": 2140
1095
+ },
1096
+ {
1097
+ "epoch": 0.9297297297297298,
1098
+ "eval_loss": 1.195316195487976,
1099
+ "eval_runtime": 156.3354,
1100
+ "eval_samples_per_second": 9.595,
1101
+ "eval_steps_per_second": 2.399,
1102
+ "step": 2150
1103
+ },
1104
+ {
1105
+ "epoch": 0.9340540540540541,
1106
+ "grad_norm": 0.5412635803222656,
1107
+ "learning_rate": 8.658987064555783e-06,
1108
+ "loss": 1.0922,
1109
+ "step": 2160
1110
+ },
1111
+ {
1112
+ "epoch": 0.9427027027027027,
1113
+ "grad_norm": 0.8357748985290527,
1114
+ "learning_rate": 8.555068858331553e-06,
1115
+ "loss": 1.2273,
1116
+ "step": 2180
1117
+ },
1118
+ {
1119
+ "epoch": 0.9513513513513514,
1120
+ "grad_norm": 0.8305333852767944,
1121
+ "learning_rate": 8.450943586226805e-06,
1122
+ "loss": 1.2381,
1123
+ "step": 2200
1124
+ },
1125
+ {
1126
+ "epoch": 0.9513513513513514,
1127
+ "eval_loss": 1.1945637464523315,
1128
+ "eval_runtime": 157.8154,
1129
+ "eval_samples_per_second": 9.505,
1130
+ "eval_steps_per_second": 2.376,
1131
+ "step": 2200
1132
+ },
1133
+ {
1134
+ "epoch": 0.96,
1135
+ "grad_norm": 0.5756638646125793,
1136
+ "learning_rate": 8.346631683676616e-06,
1137
+ "loss": 1.0844,
1138
+ "step": 2220
1139
+ },
1140
+ {
1141
+ "epoch": 0.9686486486486486,
1142
+ "grad_norm": 0.8370802998542786,
1143
+ "learning_rate": 8.242153622743802e-06,
1144
+ "loss": 1.1966,
1145
+ "step": 2240
1146
+ },
1147
+ {
1148
+ "epoch": 0.972972972972973,
1149
+ "eval_loss": 1.193472981452942,
1150
+ "eval_runtime": 156.5861,
1151
+ "eval_samples_per_second": 9.579,
1152
+ "eval_steps_per_second": 2.395,
1153
+ "step": 2250
1154
+ },
1155
+ {
1156
+ "epoch": 0.9772972972972973,
1157
+ "grad_norm": 0.654332160949707,
1158
+ "learning_rate": 8.137529908101128e-06,
1159
+ "loss": 1.1768,
1160
+ "step": 2260
1161
+ },
1162
+ {
1163
+ "epoch": 0.985945945945946,
1164
+ "grad_norm": 0.8234332799911499,
1165
+ "learning_rate": 8.03278107300709e-06,
1166
+ "loss": 1.1339,
1167
+ "step": 2280
1168
+ },
1169
+ {
1170
+ "epoch": 0.9945945945945946,
1171
+ "grad_norm": 0.618282675743103,
1172
+ "learning_rate": 7.927927675276087e-06,
1173
+ "loss": 1.1226,
1174
+ "step": 2300
1175
+ },
1176
+ {
1177
+ "epoch": 0.9945945945945946,
1178
+ "eval_loss": 1.1929038763046265,
1179
+ "eval_runtime": 156.4657,
1180
+ "eval_samples_per_second": 9.587,
1181
+ "eval_steps_per_second": 2.397,
1182
+ "step": 2300
1183
+ },
1184
+ {
1185
+ "epoch": 1.0032432432432432,
1186
+ "grad_norm": 0.7228516340255737,
1187
+ "learning_rate": 7.822990293243793e-06,
1188
+ "loss": 1.1974,
1189
+ "step": 2320
1190
+ },
1191
+ {
1192
+ "epoch": 1.011891891891892,
1193
+ "grad_norm": 0.6766774654388428,
1194
+ "learning_rate": 7.71798952172849e-06,
1195
+ "loss": 1.01,
1196
+ "step": 2340
1197
+ },
1198
+ {
1199
+ "epoch": 1.0162162162162163,
1200
+ "eval_loss": 1.1917825937271118,
1201
+ "eval_runtime": 156.3134,
1202
+ "eval_samples_per_second": 9.596,
1203
+ "eval_steps_per_second": 2.399,
1204
+ "step": 2350
1205
+ },
1206
+ {
1207
+ "epoch": 1.0205405405405406,
1208
+ "grad_norm": 0.8917098045349121,
1209
+ "learning_rate": 7.612945967989164e-06,
1210
+ "loss": 1.0233,
1211
+ "step": 2360
1212
+ },
1213
+ {
1214
+ "epoch": 1.0291891891891891,
1215
+ "grad_norm": 0.7241379022598267,
1216
+ "learning_rate": 7.5078802476811515e-06,
1217
+ "loss": 1.2105,
1218
+ "step": 2380
1219
+ },
1220
+ {
1221
+ "epoch": 1.037837837837838,
1222
+ "grad_norm": 0.6121885776519775,
1223
+ "learning_rate": 7.402812980810174e-06,
1224
+ "loss": 1.1697,
1225
+ "step": 2400
1226
+ },
1227
+ {
1228
+ "epoch": 1.037837837837838,
1229
+ "eval_loss": 1.1907098293304443,
1230
+ "eval_runtime": 156.2571,
1231
+ "eval_samples_per_second": 9.6,
1232
+ "eval_steps_per_second": 2.4,
1233
+ "step": 2400
1234
+ },
1235
+ {
1236
+ "epoch": 1.0464864864864865,
1237
+ "grad_norm": 0.56922847032547,
1238
+ "learning_rate": 7.297764787685467e-06,
1239
+ "loss": 1.2328,
1240
+ "step": 2420
1241
+ },
1242
+ {
1243
+ "epoch": 1.0551351351351352,
1244
+ "grad_norm": 0.8866540789604187,
1245
+ "learning_rate": 7.192756284872893e-06,
1246
+ "loss": 1.1378,
1247
+ "step": 2440
1248
+ },
1249
+ {
1250
+ "epoch": 1.0594594594594595,
1251
+ "eval_loss": 1.190106987953186,
1252
+ "eval_runtime": 156.2921,
1253
+ "eval_samples_per_second": 9.597,
1254
+ "eval_steps_per_second": 2.399,
1255
+ "step": 2450
1256
+ },
1257
+ {
1258
+ "epoch": 1.0637837837837838,
1259
+ "grad_norm": 0.9187541604042053,
1260
+ "learning_rate": 7.087808081148768e-06,
1261
+ "loss": 1.027,
1262
+ "step": 2460
1263
+ },
1264
+ {
1265
+ "epoch": 1.0724324324324324,
1266
+ "grad_norm": 0.4501543939113617,
1267
+ "learning_rate": 6.982940773455215e-06,
1268
+ "loss": 1.2174,
1269
+ "step": 2480
1270
+ },
1271
+ {
1272
+ "epoch": 1.0810810810810811,
1273
+ "grad_norm": 0.670372486114502,
1274
+ "learning_rate": 6.878174942857852e-06,
1275
+ "loss": 0.9583,
1276
+ "step": 2500
1277
+ },
1278
+ {
1279
+ "epoch": 1.0810810810810811,
1280
+ "eval_loss": 1.1890244483947754,
1281
+ "eval_runtime": 156.3208,
1282
+ "eval_samples_per_second": 9.596,
1283
+ "eval_steps_per_second": 2.399,
1284
+ "step": 2500
1285
+ },
1286
+ {
1287
+ "epoch": 1.0897297297297297,
1288
+ "grad_norm": 0.5464470386505127,
1289
+ "learning_rate": 6.7735311505065885e-06,
1290
+ "loss": 1.117,
1291
+ "step": 2520
1292
+ },
1293
+ {
1294
+ "epoch": 1.0983783783783783,
1295
+ "grad_norm": 0.5962574481964111,
1296
+ "learning_rate": 6.669029933600332e-06,
1297
+ "loss": 1.1578,
1298
+ "step": 2540
1299
+ },
1300
+ {
1301
+ "epoch": 1.1027027027027028,
1302
+ "eval_loss": 1.1883645057678223,
1303
+ "eval_runtime": 156.3484,
1304
+ "eval_samples_per_second": 9.594,
1305
+ "eval_steps_per_second": 2.398,
1306
+ "step": 2550
1307
+ },
1308
+ {
1309
+ "epoch": 1.107027027027027,
1310
+ "grad_norm": 0.7612497210502625,
1311
+ "learning_rate": 6.564691801356392e-06,
1312
+ "loss": 1.1369,
1313
+ "step": 2560
1314
+ },
1315
+ {
1316
+ "epoch": 1.1156756756756756,
1317
+ "grad_norm": 1.0087133646011353,
1318
+ "learning_rate": 6.460537230985388e-06,
1319
+ "loss": 1.1793,
1320
+ "step": 2580
1321
+ },
1322
+ {
1323
+ "epoch": 1.1243243243243244,
1324
+ "grad_norm": 0.6867474913597107,
1325
+ "learning_rate": 6.356586663672409e-06,
1326
+ "loss": 1.1688,
1327
+ "step": 2600
1328
+ },
1329
+ {
1330
+ "epoch": 1.1243243243243244,
1331
+ "eval_loss": 1.1876789331436157,
1332
+ "eval_runtime": 156.4562,
1333
+ "eval_samples_per_second": 9.587,
1334
+ "eval_steps_per_second": 2.397,
1335
+ "step": 2600
1336
+ },
1337
+ {
1338
+ "epoch": 1.132972972972973,
1339
+ "grad_norm": 0.8074759840965271,
1340
+ "learning_rate": 6.252860500565295e-06,
1341
+ "loss": 1.1414,
1342
+ "step": 2620
1343
+ },
1344
+ {
1345
+ "epoch": 1.1416216216216215,
1346
+ "grad_norm": 1.0738656520843506,
1347
+ "learning_rate": 6.1493790987707146e-06,
1348
+ "loss": 1.2174,
1349
+ "step": 2640
1350
+ },
1351
+ {
1352
+ "epoch": 1.145945945945946,
1353
+ "eval_loss": 1.1870859861373901,
1354
+ "eval_runtime": 156.2871,
1355
+ "eval_samples_per_second": 9.598,
1356
+ "eval_steps_per_second": 2.399,
1357
+ "step": 2650
1358
+ },
1359
+ {
1360
+ "epoch": 1.1502702702702703,
1361
+ "grad_norm": 0.7977942824363708,
1362
+ "learning_rate": 6.0461627673589336e-06,
1363
+ "loss": 1.2674,
1364
+ "step": 2660
1365
+ },
1366
+ {
1367
+ "epoch": 1.1589189189189189,
1368
+ "grad_norm": 0.5102550983428955,
1369
+ "learning_rate": 5.943231763377995e-06,
1370
+ "loss": 1.1778,
1371
+ "step": 2680
1372
+ },
1373
+ {
1374
+ "epoch": 1.1675675675675676,
1375
+ "grad_norm": 0.7279404997825623,
1376
+ "learning_rate": 5.840606287878094e-06,
1377
+ "loss": 1.1715,
1378
+ "step": 2700
1379
+ },
1380
+ {
1381
+ "epoch": 1.1675675675675676,
1382
+ "eval_loss": 1.1867084503173828,
1383
+ "eval_runtime": 156.3684,
1384
+ "eval_samples_per_second": 9.593,
1385
+ "eval_steps_per_second": 2.398,
1386
+ "step": 2700
1387
+ },
1388
+ {
1389
+ "epoch": 1.1762162162162162,
1390
+ "grad_norm": 1.0068808794021606,
1391
+ "learning_rate": 5.738306481946973e-06,
1392
+ "loss": 1.1962,
1393
+ "step": 2720
1394
+ },
1395
+ {
1396
+ "epoch": 1.1848648648648648,
1397
+ "grad_norm": 1.274601936340332,
1398
+ "learning_rate": 5.636352422757062e-06,
1399
+ "loss": 1.1244,
1400
+ "step": 2740
1401
+ },
1402
+ {
1403
+ "epoch": 1.1891891891891893,
1404
+ "eval_loss": 1.185834288597107,
1405
+ "eval_runtime": 156.5394,
1406
+ "eval_samples_per_second": 9.582,
1407
+ "eval_steps_per_second": 2.396,
1408
+ "step": 2750
1409
+ },
1410
+ {
1411
+ "epoch": 1.1935135135135135,
1412
+ "grad_norm": 0.7775521278381348,
1413
+ "learning_rate": 5.534764119625167e-06,
1414
+ "loss": 1.1735,
1415
+ "step": 2760
1416
+ },
1417
+ {
1418
+ "epoch": 1.202162162162162,
1419
+ "grad_norm": 0.7912615537643433,
1420
+ "learning_rate": 5.433561510085475e-06,
1421
+ "loss": 1.1918,
1422
+ "step": 2780
1423
+ },
1424
+ {
1425
+ "epoch": 1.2108108108108109,
1426
+ "grad_norm": 0.7740307450294495,
1427
+ "learning_rate": 5.3327644559766674e-06,
1428
+ "loss": 1.0983,
1429
+ "step": 2800
1430
+ },
1431
+ {
1432
+ "epoch": 1.2108108108108109,
1433
+ "eval_loss": 1.185600757598877,
1434
+ "eval_runtime": 156.2111,
1435
+ "eval_samples_per_second": 9.602,
1436
+ "eval_steps_per_second": 2.401,
1437
+ "step": 2800
1438
+ },
1439
+ {
1440
+ "epoch": 1.2194594594594594,
1441
+ "grad_norm": 0.7494848370552063,
1442
+ "learning_rate": 5.232392739543848e-06,
1443
+ "loss": 1.0842,
1444
+ "step": 2820
1445
+ },
1446
+ {
1447
+ "epoch": 1.228108108108108,
1448
+ "grad_norm": 0.936766505241394,
1449
+ "learning_rate": 5.132466059556139e-06,
1450
+ "loss": 1.0828,
1451
+ "step": 2840
1452
+ },
1453
+ {
1454
+ "epoch": 1.2324324324324325,
1455
+ "eval_loss": 1.184820294380188,
1456
+ "eval_runtime": 156.2295,
1457
+ "eval_samples_per_second": 9.601,
1458
+ "eval_steps_per_second": 2.4,
1459
+ "step": 2850
1460
+ }
1461
+ ],
1462
+ "logging_steps": 20,
1463
+ "max_steps": 4624,
1464
+ "num_input_tokens_seen": 0,
1465
+ "num_train_epochs": 2,
1466
+ "save_steps": 50,
1467
+ "stateful_callbacks": {
1468
+ "EarlyStoppingCallback": {
1469
+ "args": {
1470
+ "early_stopping_patience": 2,
1471
+ "early_stopping_threshold": 0.0
1472
+ },
1473
+ "attributes": {
1474
+ "early_stopping_patience_counter": 0
1475
+ }
1476
+ },
1477
+ "TrainerControl": {
1478
+ "args": {
1479
+ "should_epoch_stop": false,
1480
+ "should_evaluate": false,
1481
+ "should_log": false,
1482
+ "should_save": true,
1483
+ "should_training_stop": false
1484
+ },
1485
+ "attributes": {}
1486
+ }
1487
+ },
1488
+ "total_flos": 2.08620239192064e+17,
1489
+ "train_batch_size": 4,
1490
+ "trial_name": null,
1491
+ "trial_params": null
1492
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebb27e7ef2130019447c79effafe19d46b3798be3ee10705a2da8a8a57e2ca29
3
+ size 5713