madzzey commited on
Commit
462f590
·
1 Parent(s): 8f8f500

Upload chemistry model checkpoint

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ *.pth. filter=lfs diff=lfs merge=lfs -text
README copy.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-2-7b-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.15.0
adapter_config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "meta-llama/Llama-2-7b-hf",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.05,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "r": 16,
24
+ "rank_pattern": {},
25
+ "revision": null,
26
+ "target_modules": [
27
+ "o_proj",
28
+ "q_proj",
29
+ "v_proj",
30
+ "gate_proj",
31
+ "k_proj",
32
+ "down_proj",
33
+ "up_proj"
34
+ ],
35
+ "task_type": "CAUSAL_LM",
36
+ "trainable_token_indices": null,
37
+ "use_dora": false,
38
+ "use_rslora": false
39
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4daaa601eae7abdb563e384b10a1c40c22bcab196a243559818f58a2c293d31
3
+ size 159967880
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb0d4375a5b3ffbb399b4d2db1fff9648f1edaaeb7ac169565a5e37f94565c8e
3
+ size 81735892
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0c6da79ebd6492284173d69c7a4146ecfedcd530420666bb236d2c2b442a0e5
3
+ size 14244
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8fc201b65c94478c92cbf45c1cc82951745f0055b53a4a3eac2622fb173fe69
3
+ size 988
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3dc437fc927c14c0b428c980cdf10792e579474303182db600630adefd5ad3f
3
+ size 1064
special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "</s>",
17
+ "unk_token": {
18
+ "content": "<unk>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
tokenizer_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<s>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "2": {
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": true
29
+ }
30
+ },
31
+ "bos_token": "<s>",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "extra_special_tokens": {},
35
+ "legacy": false,
36
+ "model_max_length": 1024,
37
+ "pad_token": "</s>",
38
+ "padding_side": "right",
39
+ "sp_model_kwargs": {},
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": false
43
+ }
trainer_state.json ADDED
@@ -0,0 +1,783 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 2500,
3
+ "best_metric": 0.3031768798828125,
4
+ "best_model_checkpoint": "./chemistry-qlora-optimized/checkpoint-2500",
5
+ "epoch": 0.29867240117676924,
6
+ "eval_steps": 500,
7
+ "global_step": 2500,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.002986724011767693,
14
+ "grad_norm": 0.3601633906364441,
15
+ "learning_rate": 2.485089463220676e-05,
16
+ "loss": 0.9242,
17
+ "step": 25
18
+ },
19
+ {
20
+ "epoch": 0.005973448023535386,
21
+ "grad_norm": 0.20105794072151184,
22
+ "learning_rate": 4.970178926441352e-05,
23
+ "loss": 0.5975,
24
+ "step": 50
25
+ },
26
+ {
27
+ "epoch": 0.008960172035303077,
28
+ "grad_norm": 0.0951744094491005,
29
+ "learning_rate": 7.455268389662027e-05,
30
+ "loss": 0.4182,
31
+ "step": 75
32
+ },
33
+ {
34
+ "epoch": 0.011946896047070771,
35
+ "grad_norm": 0.075951486825943,
36
+ "learning_rate": 9.940357852882704e-05,
37
+ "loss": 0.3922,
38
+ "step": 100
39
+ },
40
+ {
41
+ "epoch": 0.014933620058838463,
42
+ "grad_norm": 0.07564502209424973,
43
+ "learning_rate": 0.0001242544731610338,
44
+ "loss": 0.3782,
45
+ "step": 125
46
+ },
47
+ {
48
+ "epoch": 0.017920344070606154,
49
+ "grad_norm": 0.09620492905378342,
50
+ "learning_rate": 0.00014910536779324054,
51
+ "loss": 0.3748,
52
+ "step": 150
53
+ },
54
+ {
55
+ "epoch": 0.020907068082373848,
56
+ "grad_norm": 0.07839681208133698,
57
+ "learning_rate": 0.00017395626242544733,
58
+ "loss": 0.3705,
59
+ "step": 175
60
+ },
61
+ {
62
+ "epoch": 0.023893792094141542,
63
+ "grad_norm": 0.08541898429393768,
64
+ "learning_rate": 0.00019880715705765408,
65
+ "loss": 0.3669,
66
+ "step": 200
67
+ },
68
+ {
69
+ "epoch": 0.026880516105909233,
70
+ "grad_norm": 0.07105179876089096,
71
+ "learning_rate": 0.00022365805168986084,
72
+ "loss": 0.36,
73
+ "step": 225
74
+ },
75
+ {
76
+ "epoch": 0.029867240117676927,
77
+ "grad_norm": 0.07087098807096481,
78
+ "learning_rate": 0.0002485089463220676,
79
+ "loss": 0.3606,
80
+ "step": 250
81
+ },
82
+ {
83
+ "epoch": 0.03285396412944462,
84
+ "grad_norm": 0.07232872396707535,
85
+ "learning_rate": 0.00027335984095427435,
86
+ "loss": 0.3572,
87
+ "step": 275
88
+ },
89
+ {
90
+ "epoch": 0.03584068814121231,
91
+ "grad_norm": 0.0895916074514389,
92
+ "learning_rate": 0.0002982107355864811,
93
+ "loss": 0.3537,
94
+ "step": 300
95
+ },
96
+ {
97
+ "epoch": 0.03882741215298,
98
+ "grad_norm": 0.06402889639139175,
99
+ "learning_rate": 0.0003230616302186879,
100
+ "loss": 0.3618,
101
+ "step": 325
102
+ },
103
+ {
104
+ "epoch": 0.041814136164747696,
105
+ "grad_norm": 0.055026255548000336,
106
+ "learning_rate": 0.00034791252485089465,
107
+ "loss": 0.3504,
108
+ "step": 350
109
+ },
110
+ {
111
+ "epoch": 0.04480086017651539,
112
+ "grad_norm": 0.06303983181715012,
113
+ "learning_rate": 0.0003727634194831014,
114
+ "loss": 0.3526,
115
+ "step": 375
116
+ },
117
+ {
118
+ "epoch": 0.047787584188283085,
119
+ "grad_norm": 0.06284493952989578,
120
+ "learning_rate": 0.00039761431411530816,
121
+ "loss": 0.3467,
122
+ "step": 400
123
+ },
124
+ {
125
+ "epoch": 0.05077430820005077,
126
+ "grad_norm": 0.05819180980324745,
127
+ "learning_rate": 0.00042246520874751495,
128
+ "loss": 0.3506,
129
+ "step": 425
130
+ },
131
+ {
132
+ "epoch": 0.053761032211818466,
133
+ "grad_norm": 0.08199992030858994,
134
+ "learning_rate": 0.0004473161033797217,
135
+ "loss": 0.3453,
136
+ "step": 450
137
+ },
138
+ {
139
+ "epoch": 0.05674775622358616,
140
+ "grad_norm": 0.06815178692340851,
141
+ "learning_rate": 0.0004721669980119284,
142
+ "loss": 0.3427,
143
+ "step": 475
144
+ },
145
+ {
146
+ "epoch": 0.059734480235353854,
147
+ "grad_norm": 0.050426531583070755,
148
+ "learning_rate": 0.0004970178926441352,
149
+ "loss": 0.3441,
150
+ "step": 500
151
+ },
152
+ {
153
+ "epoch": 0.059734480235353854,
154
+ "eval_loss": 0.3422516882419586,
155
+ "eval_runtime": 80020.5739,
156
+ "eval_samples_per_second": 3.347,
157
+ "eval_steps_per_second": 0.418,
158
+ "step": 500
159
+ },
160
+ {
161
+ "epoch": 0.06272120424712155,
162
+ "grad_norm": 0.08422357589006424,
163
+ "learning_rate": 0.0004993225349510377,
164
+ "loss": 0.3449,
165
+ "step": 525
166
+ },
167
+ {
168
+ "epoch": 0.06570792825888924,
169
+ "grad_norm": 0.09289127588272095,
170
+ "learning_rate": 0.0004985526883044898,
171
+ "loss": 0.338,
172
+ "step": 550
173
+ },
174
+ {
175
+ "epoch": 0.06869465227065694,
176
+ "grad_norm": 0.13212044537067413,
177
+ "learning_rate": 0.0004977828416579417,
178
+ "loss": 0.3402,
179
+ "step": 575
180
+ },
181
+ {
182
+ "epoch": 0.07168137628242462,
183
+ "grad_norm": 0.0823817029595375,
184
+ "learning_rate": 0.0004970129950113937,
185
+ "loss": 0.3372,
186
+ "step": 600
187
+ },
188
+ {
189
+ "epoch": 0.07466810029419231,
190
+ "grad_norm": 0.05125906690955162,
191
+ "learning_rate": 0.0004962431483648457,
192
+ "loss": 0.336,
193
+ "step": 625
194
+ },
195
+ {
196
+ "epoch": 0.07765482430596,
197
+ "grad_norm": 0.267816424369812,
198
+ "learning_rate": 0.0004954733017182978,
199
+ "loss": 0.34,
200
+ "step": 650
201
+ },
202
+ {
203
+ "epoch": 0.0806415483177277,
204
+ "grad_norm": 0.057936109602451324,
205
+ "learning_rate": 0.0004947034550717497,
206
+ "loss": 0.3364,
207
+ "step": 675
208
+ },
209
+ {
210
+ "epoch": 0.08362827232949539,
211
+ "grad_norm": 0.050852783024311066,
212
+ "learning_rate": 0.0004939336084252017,
213
+ "loss": 0.3353,
214
+ "step": 700
215
+ },
216
+ {
217
+ "epoch": 0.08661499634126309,
218
+ "grad_norm": 0.04943128302693367,
219
+ "learning_rate": 0.0004931637617786536,
220
+ "loss": 0.3284,
221
+ "step": 725
222
+ },
223
+ {
224
+ "epoch": 0.08960172035303078,
225
+ "grad_norm": 0.04955856129527092,
226
+ "learning_rate": 0.0004923939151321057,
227
+ "loss": 0.3285,
228
+ "step": 750
229
+ },
230
+ {
231
+ "epoch": 0.09258844436479848,
232
+ "grad_norm": 0.04910452291369438,
233
+ "learning_rate": 0.0004916240684855577,
234
+ "loss": 0.3291,
235
+ "step": 775
236
+ },
237
+ {
238
+ "epoch": 0.09557516837656617,
239
+ "grad_norm": 0.05659394711256027,
240
+ "learning_rate": 0.0004908542218390097,
241
+ "loss": 0.3283,
242
+ "step": 800
243
+ },
244
+ {
245
+ "epoch": 0.09856189238833385,
246
+ "grad_norm": 0.05103044584393501,
247
+ "learning_rate": 0.0004900843751924616,
248
+ "loss": 0.329,
249
+ "step": 825
250
+ },
251
+ {
252
+ "epoch": 0.10154861640010154,
253
+ "grad_norm": 0.07405893504619598,
254
+ "learning_rate": 0.0004893145285459137,
255
+ "loss": 0.3266,
256
+ "step": 850
257
+ },
258
+ {
259
+ "epoch": 0.10453534041186924,
260
+ "grad_norm": 0.04762987419962883,
261
+ "learning_rate": 0.0004885446818993657,
262
+ "loss": 0.3249,
263
+ "step": 875
264
+ },
265
+ {
266
+ "epoch": 0.10752206442363693,
267
+ "grad_norm": 0.050489380955696106,
268
+ "learning_rate": 0.0004877748352528176,
269
+ "loss": 0.3244,
270
+ "step": 900
271
+ },
272
+ {
273
+ "epoch": 0.11050878843540463,
274
+ "grad_norm": 0.049161575734615326,
275
+ "learning_rate": 0.00048700498860626963,
276
+ "loss": 0.3275,
277
+ "step": 925
278
+ },
279
+ {
280
+ "epoch": 0.11349551244717232,
281
+ "grad_norm": 0.0867818295955658,
282
+ "learning_rate": 0.0004862351419597216,
283
+ "loss": 0.3207,
284
+ "step": 950
285
+ },
286
+ {
287
+ "epoch": 0.11648223645894001,
288
+ "grad_norm": 0.049138959497213364,
289
+ "learning_rate": 0.00048546529531317363,
290
+ "loss": 0.3203,
291
+ "step": 975
292
+ },
293
+ {
294
+ "epoch": 0.11946896047070771,
295
+ "grad_norm": 0.049613844603300095,
296
+ "learning_rate": 0.0004846954486666256,
297
+ "loss": 0.321,
298
+ "step": 1000
299
+ },
300
+ {
301
+ "epoch": 0.11946896047070771,
302
+ "eval_loss": 0.32202616333961487,
303
+ "eval_runtime": 80142.6561,
304
+ "eval_samples_per_second": 3.342,
305
+ "eval_steps_per_second": 0.418,
306
+ "step": 1000
307
+ },
308
+ {
309
+ "epoch": 0.1224556844824754,
310
+ "grad_norm": 0.05376691371202469,
311
+ "learning_rate": 0.00048392560202007763,
312
+ "loss": 0.3184,
313
+ "step": 1025
314
+ },
315
+ {
316
+ "epoch": 0.1254424084942431,
317
+ "grad_norm": 0.048957519233226776,
318
+ "learning_rate": 0.0004831557553735296,
319
+ "loss": 0.3193,
320
+ "step": 1050
321
+ },
322
+ {
323
+ "epoch": 0.1284291325060108,
324
+ "grad_norm": 0.0559055432677269,
325
+ "learning_rate": 0.00048238590872698157,
326
+ "loss": 0.3261,
327
+ "step": 1075
328
+ },
329
+ {
330
+ "epoch": 0.13141585651777848,
331
+ "grad_norm": 0.06237791106104851,
332
+ "learning_rate": 0.00048161606208043354,
333
+ "loss": 0.3235,
334
+ "step": 1100
335
+ },
336
+ {
337
+ "epoch": 0.13440258052954618,
338
+ "grad_norm": 0.05047965049743652,
339
+ "learning_rate": 0.00048084621543388557,
340
+ "loss": 0.3232,
341
+ "step": 1125
342
+ },
343
+ {
344
+ "epoch": 0.13738930454131387,
345
+ "grad_norm": 0.05053680017590523,
346
+ "learning_rate": 0.00048007636878733754,
347
+ "loss": 0.3166,
348
+ "step": 1150
349
+ },
350
+ {
351
+ "epoch": 0.14037602855308154,
352
+ "grad_norm": 0.1490963250398636,
353
+ "learning_rate": 0.00047930652214078957,
354
+ "loss": 0.3196,
355
+ "step": 1175
356
+ },
357
+ {
358
+ "epoch": 0.14336275256484923,
359
+ "grad_norm": 0.06572224199771881,
360
+ "learning_rate": 0.00047853667549424154,
361
+ "loss": 0.3162,
362
+ "step": 1200
363
+ },
364
+ {
365
+ "epoch": 0.14634947657661693,
366
+ "grad_norm": 0.04761017858982086,
367
+ "learning_rate": 0.00047776682884769357,
368
+ "loss": 0.3188,
369
+ "step": 1225
370
+ },
371
+ {
372
+ "epoch": 0.14933620058838462,
373
+ "grad_norm": 0.05511182174086571,
374
+ "learning_rate": 0.00047699698220114554,
375
+ "loss": 0.3211,
376
+ "step": 1250
377
+ },
378
+ {
379
+ "epoch": 0.15232292460015232,
380
+ "grad_norm": 0.05186431482434273,
381
+ "learning_rate": 0.0004762271355545975,
382
+ "loss": 0.3166,
383
+ "step": 1275
384
+ },
385
+ {
386
+ "epoch": 0.15530964861192,
387
+ "grad_norm": 0.05169636383652687,
388
+ "learning_rate": 0.0004754572889080495,
389
+ "loss": 0.3161,
390
+ "step": 1300
391
+ },
392
+ {
393
+ "epoch": 0.1582963726236877,
394
+ "grad_norm": 0.05161893740296364,
395
+ "learning_rate": 0.0004746874422615015,
396
+ "loss": 0.3106,
397
+ "step": 1325
398
+ },
399
+ {
400
+ "epoch": 0.1612830966354554,
401
+ "grad_norm": 0.047998517751693726,
402
+ "learning_rate": 0.0004739175956149535,
403
+ "loss": 0.3092,
404
+ "step": 1350
405
+ },
406
+ {
407
+ "epoch": 0.1642698206472231,
408
+ "grad_norm": 0.05086188018321991,
409
+ "learning_rate": 0.0004731477489684055,
410
+ "loss": 0.314,
411
+ "step": 1375
412
+ },
413
+ {
414
+ "epoch": 0.16725654465899079,
415
+ "grad_norm": 0.5123017430305481,
416
+ "learning_rate": 0.0004723779023218575,
417
+ "loss": 0.3138,
418
+ "step": 1400
419
+ },
420
+ {
421
+ "epoch": 0.17024326867075848,
422
+ "grad_norm": 0.513543426990509,
423
+ "learning_rate": 0.0004716080556753095,
424
+ "loss": 0.3477,
425
+ "step": 1425
426
+ },
427
+ {
428
+ "epoch": 0.17322999268252617,
429
+ "grad_norm": 0.06812719255685806,
430
+ "learning_rate": 0.00047083820902876143,
431
+ "loss": 0.3183,
432
+ "step": 1450
433
+ },
434
+ {
435
+ "epoch": 0.17621671669429387,
436
+ "grad_norm": 0.053058214485645294,
437
+ "learning_rate": 0.00047006836238221346,
438
+ "loss": 0.3132,
439
+ "step": 1475
440
+ },
441
+ {
442
+ "epoch": 0.17920344070606156,
443
+ "grad_norm": 0.05340981110930443,
444
+ "learning_rate": 0.00046929851573566543,
445
+ "loss": 0.3167,
446
+ "step": 1500
447
+ },
448
+ {
449
+ "epoch": 0.17920344070606156,
450
+ "eval_loss": 0.31384027004241943,
451
+ "eval_runtime": 79940.0478,
452
+ "eval_samples_per_second": 3.351,
453
+ "eval_steps_per_second": 0.419,
454
+ "step": 1500
455
+ },
456
+ {
457
+ "epoch": 0.18219016471782926,
458
+ "grad_norm": 0.06443135440349579,
459
+ "learning_rate": 0.00046852866908911746,
460
+ "loss": 0.3088,
461
+ "step": 1525
462
+ },
463
+ {
464
+ "epoch": 0.18517688872959695,
465
+ "grad_norm": 0.05341175198554993,
466
+ "learning_rate": 0.00046775882244256943,
467
+ "loss": 0.3131,
468
+ "step": 1550
469
+ },
470
+ {
471
+ "epoch": 0.18816361274136464,
472
+ "grad_norm": 0.05056153982877731,
473
+ "learning_rate": 0.00046698897579602145,
474
+ "loss": 0.3101,
475
+ "step": 1575
476
+ },
477
+ {
478
+ "epoch": 0.19115033675313234,
479
+ "grad_norm": 0.05725909024477005,
480
+ "learning_rate": 0.0004662191291494734,
481
+ "loss": 0.3086,
482
+ "step": 1600
483
+ },
484
+ {
485
+ "epoch": 0.19413706076490003,
486
+ "grad_norm": 0.473294198513031,
487
+ "learning_rate": 0.0004654800763687874,
488
+ "loss": 0.3339,
489
+ "step": 1625
490
+ },
491
+ {
492
+ "epoch": 0.1971237847766677,
493
+ "grad_norm": 0.5311633944511414,
494
+ "learning_rate": 0.00046471022972223935,
495
+ "loss": 0.3326,
496
+ "step": 1650
497
+ },
498
+ {
499
+ "epoch": 0.2001105087884354,
500
+ "grad_norm": 0.05935530364513397,
501
+ "learning_rate": 0.0004639403830756913,
502
+ "loss": 0.3207,
503
+ "step": 1675
504
+ },
505
+ {
506
+ "epoch": 0.2030972328002031,
507
+ "grad_norm": 0.0760846734046936,
508
+ "learning_rate": 0.0004631705364291433,
509
+ "loss": 0.3103,
510
+ "step": 1700
511
+ },
512
+ {
513
+ "epoch": 0.20608395681197078,
514
+ "grad_norm": 0.05102820321917534,
515
+ "learning_rate": 0.0004624006897825953,
516
+ "loss": 0.3103,
517
+ "step": 1725
518
+ },
519
+ {
520
+ "epoch": 0.20907068082373848,
521
+ "grad_norm": 0.05322102829813957,
522
+ "learning_rate": 0.0004616308431360473,
523
+ "loss": 0.3107,
524
+ "step": 1750
525
+ },
526
+ {
527
+ "epoch": 0.21205740483550617,
528
+ "grad_norm": 0.05677218735218048,
529
+ "learning_rate": 0.0004608609964894993,
530
+ "loss": 0.3107,
531
+ "step": 1775
532
+ },
533
+ {
534
+ "epoch": 0.21504412884727386,
535
+ "grad_norm": 0.05003349855542183,
536
+ "learning_rate": 0.0004600911498429513,
537
+ "loss": 0.307,
538
+ "step": 1800
539
+ },
540
+ {
541
+ "epoch": 0.21803085285904156,
542
+ "grad_norm": 0.05029498040676117,
543
+ "learning_rate": 0.0004593213031964033,
544
+ "loss": 0.3054,
545
+ "step": 1825
546
+ },
547
+ {
548
+ "epoch": 0.22101757687080925,
549
+ "grad_norm": 0.05433105677366257,
550
+ "learning_rate": 0.0004585514565498553,
551
+ "loss": 0.3108,
552
+ "step": 1850
553
+ },
554
+ {
555
+ "epoch": 0.22400430088257695,
556
+ "grad_norm": 0.06716553121805191,
557
+ "learning_rate": 0.00045778160990330726,
558
+ "loss": 0.3089,
559
+ "step": 1875
560
+ },
561
+ {
562
+ "epoch": 0.22699102489434464,
563
+ "grad_norm": 0.06449099630117416,
564
+ "learning_rate": 0.00045701176325675923,
565
+ "loss": 0.3081,
566
+ "step": 1900
567
+ },
568
+ {
569
+ "epoch": 0.22997774890611233,
570
+ "grad_norm": 0.060889799147844315,
571
+ "learning_rate": 0.00045624191661021126,
572
+ "loss": 0.3085,
573
+ "step": 1925
574
+ },
575
+ {
576
+ "epoch": 0.23296447291788003,
577
+ "grad_norm": 0.13389873504638672,
578
+ "learning_rate": 0.00045547206996366323,
579
+ "loss": 0.3048,
580
+ "step": 1950
581
+ },
582
+ {
583
+ "epoch": 0.23595119692964772,
584
+ "grad_norm": 0.05601953715085983,
585
+ "learning_rate": 0.00045470222331711526,
586
+ "loss": 0.3063,
587
+ "step": 1975
588
+ },
589
+ {
590
+ "epoch": 0.23893792094141542,
591
+ "grad_norm": 0.05448105186223984,
592
+ "learning_rate": 0.00045393237667056723,
593
+ "loss": 0.3033,
594
+ "step": 2000
595
+ },
596
+ {
597
+ "epoch": 0.23893792094141542,
598
+ "eval_loss": 0.3051091432571411,
599
+ "eval_runtime": 80198.5724,
600
+ "eval_samples_per_second": 3.34,
601
+ "eval_steps_per_second": 0.417,
602
+ "step": 2000
603
+ },
604
+ {
605
+ "epoch": 0.2419246449531831,
606
+ "grad_norm": 0.0593348927795887,
607
+ "learning_rate": 0.00045316253002401926,
608
+ "loss": 0.3033,
609
+ "step": 2025
610
+ },
611
+ {
612
+ "epoch": 0.2449113689649508,
613
+ "grad_norm": 0.0662638247013092,
614
+ "learning_rate": 0.0004523926833774712,
615
+ "loss": 0.3028,
616
+ "step": 2050
617
+ },
618
+ {
619
+ "epoch": 0.2478980929767185,
620
+ "grad_norm": 0.08647308498620987,
621
+ "learning_rate": 0.0004516228367309232,
622
+ "loss": 0.3015,
623
+ "step": 2075
624
+ },
625
+ {
626
+ "epoch": 0.2508848169884862,
627
+ "grad_norm": 0.06536151468753815,
628
+ "learning_rate": 0.0004508529900843752,
629
+ "loss": 0.3031,
630
+ "step": 2100
631
+ },
632
+ {
633
+ "epoch": 0.2538715410002539,
634
+ "grad_norm": 0.05942423641681671,
635
+ "learning_rate": 0.0004500831434378272,
636
+ "loss": 0.3037,
637
+ "step": 2125
638
+ },
639
+ {
640
+ "epoch": 0.2568582650120216,
641
+ "grad_norm": 0.05728958174586296,
642
+ "learning_rate": 0.0004493132967912792,
643
+ "loss": 0.3041,
644
+ "step": 2150
645
+ },
646
+ {
647
+ "epoch": 0.2598449890237893,
648
+ "grad_norm": 0.07348435372114182,
649
+ "learning_rate": 0.0004485434501447312,
650
+ "loss": 0.3054,
651
+ "step": 2175
652
+ },
653
+ {
654
+ "epoch": 0.26283171303555697,
655
+ "grad_norm": 0.07285642623901367,
656
+ "learning_rate": 0.0004477736034981832,
657
+ "loss": 0.3012,
658
+ "step": 2200
659
+ },
660
+ {
661
+ "epoch": 0.26581843704732466,
662
+ "grad_norm": 0.06905517727136612,
663
+ "learning_rate": 0.0004470037568516352,
664
+ "loss": 0.3002,
665
+ "step": 2225
666
+ },
667
+ {
668
+ "epoch": 0.26880516105909236,
669
+ "grad_norm": 0.047947004437446594,
670
+ "learning_rate": 0.0004462339102050871,
671
+ "loss": 0.3013,
672
+ "step": 2250
673
+ },
674
+ {
675
+ "epoch": 0.27179188507086005,
676
+ "grad_norm": 0.12290674448013306,
677
+ "learning_rate": 0.00044546406355853914,
678
+ "loss": 0.3012,
679
+ "step": 2275
680
+ },
681
+ {
682
+ "epoch": 0.27477860908262774,
683
+ "grad_norm": 0.05509233847260475,
684
+ "learning_rate": 0.0004446942169119911,
685
+ "loss": 0.2998,
686
+ "step": 2300
687
+ },
688
+ {
689
+ "epoch": 0.27776533309439544,
690
+ "grad_norm": 0.06934493780136108,
691
+ "learning_rate": 0.00044392437026544314,
692
+ "loss": 0.3014,
693
+ "step": 2325
694
+ },
695
+ {
696
+ "epoch": 0.2807520571061631,
697
+ "grad_norm": 0.1120632141828537,
698
+ "learning_rate": 0.0004431545236188951,
699
+ "loss": 0.299,
700
+ "step": 2350
701
+ },
702
+ {
703
+ "epoch": 0.28373878111793077,
704
+ "grad_norm": 0.0867624506354332,
705
+ "learning_rate": 0.00044238467697234714,
706
+ "loss": 0.2983,
707
+ "step": 2375
708
+ },
709
+ {
710
+ "epoch": 0.28672550512969847,
711
+ "grad_norm": 0.07918841391801834,
712
+ "learning_rate": 0.0004416148303257991,
713
+ "loss": 0.3006,
714
+ "step": 2400
715
+ },
716
+ {
717
+ "epoch": 0.28971222914146616,
718
+ "grad_norm": 0.08205003291368484,
719
+ "learning_rate": 0.0004408449836792511,
720
+ "loss": 0.3008,
721
+ "step": 2425
722
+ },
723
+ {
724
+ "epoch": 0.29269895315323385,
725
+ "grad_norm": 0.05923304334282875,
726
+ "learning_rate": 0.00044007513703270306,
727
+ "loss": 0.2984,
728
+ "step": 2450
729
+ },
730
+ {
731
+ "epoch": 0.29568567716500155,
732
+ "grad_norm": 0.06517321616411209,
733
+ "learning_rate": 0.0004393052903861551,
734
+ "loss": 0.3024,
735
+ "step": 2475
736
+ },
737
+ {
738
+ "epoch": 0.29867240117676924,
739
+ "grad_norm": 0.09344199299812317,
740
+ "learning_rate": 0.00043853544373960706,
741
+ "loss": 0.2991,
742
+ "step": 2500
743
+ },
744
+ {
745
+ "epoch": 0.29867240117676924,
746
+ "eval_loss": 0.3031768798828125,
747
+ "eval_runtime": 80460.0276,
748
+ "eval_samples_per_second": 3.329,
749
+ "eval_steps_per_second": 0.416,
750
+ "step": 2500
751
+ }
752
+ ],
753
+ "logging_steps": 25,
754
+ "max_steps": 16740,
755
+ "num_input_tokens_seen": 0,
756
+ "num_train_epochs": 2,
757
+ "save_steps": 500,
758
+ "stateful_callbacks": {
759
+ "EarlyStoppingCallback": {
760
+ "args": {
761
+ "early_stopping_patience": 3,
762
+ "early_stopping_threshold": 0.01
763
+ },
764
+ "attributes": {
765
+ "early_stopping_patience_counter": 3
766
+ }
767
+ },
768
+ "TrainerControl": {
769
+ "args": {
770
+ "should_epoch_stop": false,
771
+ "should_evaluate": false,
772
+ "should_log": false,
773
+ "should_save": true,
774
+ "should_training_stop": true
775
+ },
776
+ "attributes": {}
777
+ }
778
+ },
779
+ "total_flos": 1.306916403806208e+19,
780
+ "train_batch_size": 4,
781
+ "trial_name": null,
782
+ "trial_params": null
783
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b6d90e7adf6ae1f16366d9b55bf256ac171ff27dff50836886b107ea7802d03
3
+ size 5368