gguichard commited on
Commit
ec1c247
1 Parent(s): 8da7758

Training in progress, epoch 13, checkpoint

Browse files
checkpoint-331968/README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: FacebookAI/roberta-large
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.9.0
checkpoint-331968/adapter_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "FacebookAI/roberta-large",
5
+ "bias": "all",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "loftq_config": {},
12
+ "lora_alpha": 16,
13
+ "lora_dropout": 0.1,
14
+ "megatron_config": null,
15
+ "megatron_core": "megatron.core",
16
+ "modules_to_save": null,
17
+ "peft_type": "LORA",
18
+ "r": 64,
19
+ "rank_pattern": {},
20
+ "revision": null,
21
+ "target_modules": [
22
+ "query",
23
+ "dense",
24
+ "value",
25
+ "key"
26
+ ],
27
+ "task_type": "FEATURE_EXTRACTION",
28
+ "use_dora": false,
29
+ "use_rslora": false
30
+ }
checkpoint-331968/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f02412422cce20124a39e416314ea6dcee6483de1fbe6a1f7bbf82c222b1c1c
3
+ size 114926144
checkpoint-331968/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-331968/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cb9df86c7741c92ba06e377de89b399aca0495941d6af2e935f34f487c0121a
3
+ size 229067189
checkpoint-331968/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5047c6a5dea6ee65091482e532257a6f04e631cf60718b16bd72de84e2031b71
3
+ size 14244
checkpoint-331968/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a996d77e4d60fc4a1b5f3a109369c86a8145030ed20caa372c99a666bca558bc
3
+ size 1064
checkpoint-331968/special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
checkpoint-331968/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-331968/tokenizer_config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": true,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "mask_token": "<mask>",
51
+ "model_max_length": 512,
52
+ "pad_token": "<pad>",
53
+ "sep_token": "</s>",
54
+ "tokenizer_class": "RobertaTokenizer",
55
+ "trim_offsets": true,
56
+ "unk_token": "<unk>"
57
+ }
checkpoint-331968/trainer_state.json ADDED
@@ -0,0 +1,237 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.002463332610204816,
3
+ "best_model_checkpoint": "wsd_embedding/checkpoint-331968",
4
+ "epoch": 13.0,
5
+ "eval_steps": 500,
6
+ "global_step": 331968,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "grad_norm": 0.013213178142905235,
14
+ "learning_rate": 9.800046992481203e-05,
15
+ "loss": 0.0078,
16
+ "step": 25536
17
+ },
18
+ {
19
+ "epoch": 1.0,
20
+ "eval_loss": 0.004399613011628389,
21
+ "eval_runtime": 113.9442,
22
+ "eval_samples_per_second": 94.362,
23
+ "eval_steps_per_second": 11.795,
24
+ "step": 25536
25
+ },
26
+ {
27
+ "epoch": 2.0,
28
+ "grad_norm": 0.008942891843616962,
29
+ "learning_rate": 9.600117481203008e-05,
30
+ "loss": 0.0056,
31
+ "step": 51072
32
+ },
33
+ {
34
+ "epoch": 2.0,
35
+ "eval_loss": 0.0036812573671340942,
36
+ "eval_runtime": 113.8113,
37
+ "eval_samples_per_second": 94.472,
38
+ "eval_steps_per_second": 11.809,
39
+ "step": 51072
40
+ },
41
+ {
42
+ "epoch": 3.0,
43
+ "grad_norm": 0.019620591774582863,
44
+ "learning_rate": 9.400180137844612e-05,
45
+ "loss": 0.0049,
46
+ "step": 76608
47
+ },
48
+ {
49
+ "epoch": 3.0,
50
+ "eval_loss": 0.003332871478050947,
51
+ "eval_runtime": 113.7743,
52
+ "eval_samples_per_second": 94.503,
53
+ "eval_steps_per_second": 11.813,
54
+ "step": 76608
55
+ },
56
+ {
57
+ "epoch": 4.0,
58
+ "grad_norm": 0.009495776146650314,
59
+ "learning_rate": 9.200250626566417e-05,
60
+ "loss": 0.0045,
61
+ "step": 102144
62
+ },
63
+ {
64
+ "epoch": 4.0,
65
+ "eval_loss": 0.0030836393125355244,
66
+ "eval_runtime": 113.745,
67
+ "eval_samples_per_second": 94.527,
68
+ "eval_steps_per_second": 11.816,
69
+ "step": 102144
70
+ },
71
+ {
72
+ "epoch": 5.0,
73
+ "grad_norm": 0.007638801820576191,
74
+ "learning_rate": 9.00031328320802e-05,
75
+ "loss": 0.0043,
76
+ "step": 127680
77
+ },
78
+ {
79
+ "epoch": 5.0,
80
+ "eval_loss": 0.002948065521195531,
81
+ "eval_runtime": 113.6579,
82
+ "eval_samples_per_second": 94.6,
83
+ "eval_steps_per_second": 11.825,
84
+ "step": 127680
85
+ },
86
+ {
87
+ "epoch": 6.0,
88
+ "grad_norm": 0.010788742452859879,
89
+ "learning_rate": 8.800407268170426e-05,
90
+ "loss": 0.0041,
91
+ "step": 153216
92
+ },
93
+ {
94
+ "epoch": 6.0,
95
+ "eval_loss": 0.0028368972707539797,
96
+ "eval_runtime": 114.0546,
97
+ "eval_samples_per_second": 94.271,
98
+ "eval_steps_per_second": 11.784,
99
+ "step": 153216
100
+ },
101
+ {
102
+ "epoch": 7.0,
103
+ "grad_norm": 0.017284151166677475,
104
+ "learning_rate": 8.600477756892231e-05,
105
+ "loss": 0.0039,
106
+ "step": 178752
107
+ },
108
+ {
109
+ "epoch": 7.0,
110
+ "eval_loss": 0.0027765275444835424,
111
+ "eval_runtime": 113.8381,
112
+ "eval_samples_per_second": 94.45,
113
+ "eval_steps_per_second": 11.806,
114
+ "step": 178752
115
+ },
116
+ {
117
+ "epoch": 8.0,
118
+ "grad_norm": 0.008209532126784325,
119
+ "learning_rate": 8.400540413533835e-05,
120
+ "loss": 0.0038,
121
+ "step": 204288
122
+ },
123
+ {
124
+ "epoch": 8.0,
125
+ "eval_loss": 0.002683098893612623,
126
+ "eval_runtime": 113.811,
127
+ "eval_samples_per_second": 94.472,
128
+ "eval_steps_per_second": 11.809,
129
+ "step": 204288
130
+ },
131
+ {
132
+ "epoch": 9.0,
133
+ "grad_norm": 0.01161679532378912,
134
+ "learning_rate": 8.20062656641604e-05,
135
+ "loss": 0.0037,
136
+ "step": 229824
137
+ },
138
+ {
139
+ "epoch": 9.0,
140
+ "eval_loss": 0.002616806188598275,
141
+ "eval_runtime": 113.6336,
142
+ "eval_samples_per_second": 94.62,
143
+ "eval_steps_per_second": 11.827,
144
+ "step": 229824
145
+ },
146
+ {
147
+ "epoch": 10.0,
148
+ "grad_norm": 0.005629376508295536,
149
+ "learning_rate": 8.000689223057644e-05,
150
+ "loss": 0.0036,
151
+ "step": 255360
152
+ },
153
+ {
154
+ "epoch": 10.0,
155
+ "eval_loss": 0.0025433956179767847,
156
+ "eval_runtime": 113.7133,
157
+ "eval_samples_per_second": 94.554,
158
+ "eval_steps_per_second": 11.819,
159
+ "step": 255360
160
+ },
161
+ {
162
+ "epoch": 11.0,
163
+ "grad_norm": 0.009585434570908546,
164
+ "learning_rate": 7.800767543859649e-05,
165
+ "loss": 0.0035,
166
+ "step": 280896
167
+ },
168
+ {
169
+ "epoch": 11.0,
170
+ "eval_loss": 0.002532488200813532,
171
+ "eval_runtime": 113.8464,
172
+ "eval_samples_per_second": 94.443,
173
+ "eval_steps_per_second": 11.805,
174
+ "step": 280896
175
+ },
176
+ {
177
+ "epoch": 12.0,
178
+ "grad_norm": 0.007825367152690887,
179
+ "learning_rate": 7.600838032581454e-05,
180
+ "loss": 0.0035,
181
+ "step": 306432
182
+ },
183
+ {
184
+ "epoch": 12.0,
185
+ "eval_loss": 0.002520823385566473,
186
+ "eval_runtime": 113.5516,
187
+ "eval_samples_per_second": 94.688,
188
+ "eval_steps_per_second": 11.836,
189
+ "step": 306432
190
+ },
191
+ {
192
+ "epoch": 13.0,
193
+ "grad_norm": 0.006005879025906324,
194
+ "learning_rate": 7.400916353383459e-05,
195
+ "loss": 0.0034,
196
+ "step": 331968
197
+ },
198
+ {
199
+ "epoch": 13.0,
200
+ "eval_loss": 0.002463332610204816,
201
+ "eval_runtime": 113.7145,
202
+ "eval_samples_per_second": 94.553,
203
+ "eval_steps_per_second": 11.819,
204
+ "step": 331968
205
+ }
206
+ ],
207
+ "logging_steps": 500,
208
+ "max_steps": 1276800,
209
+ "num_input_tokens_seen": 0,
210
+ "num_train_epochs": 50,
211
+ "save_steps": 500,
212
+ "stateful_callbacks": {
213
+ "EarlyStoppingCallback": {
214
+ "args": {
215
+ "early_stopping_patience": 5,
216
+ "early_stopping_threshold": 0.0
217
+ },
218
+ "attributes": {
219
+ "early_stopping_patience_counter": 0
220
+ }
221
+ },
222
+ "TrainerControl": {
223
+ "args": {
224
+ "should_epoch_stop": false,
225
+ "should_evaluate": false,
226
+ "should_log": false,
227
+ "should_save": true,
228
+ "should_training_stop": false
229
+ },
230
+ "attributes": {}
231
+ }
232
+ },
233
+ "total_flos": 6.767405621850931e+17,
234
+ "train_batch_size": 8,
235
+ "trial_name": null,
236
+ "trial_params": null
237
+ }
checkpoint-331968/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7b39c5468d1d22a1d286c10e6afc9d4344aa64c4e0b83454357eee282e06a94
3
+ size 5176
checkpoint-331968/vocab.json ADDED
The diff for this file is too large to render. See raw diff