pranavdarshan commited on
Commit
584e2f9
·
verified ·
1 Parent(s): da42b29

Upload 18 files

Browse files
biomedlm-qlora-medcasereasoning/checkpoint-3000/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stanford-crfm/BioMedLM
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:stanford-crfm/BioMedLM
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- 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. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ 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).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.18.1
biomedlm-qlora-medcasereasoning/checkpoint-3000/adapter_config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "stanford-crfm/BioMedLM",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "c_attn"
33
+ ],
34
+ "target_parameters": null,
35
+ "task_type": "CAUSAL_LM",
36
+ "trainable_token_indices": null,
37
+ "use_dora": false,
38
+ "use_qalora": false,
39
+ "use_rslora": false
40
+ }
biomedlm-qlora-medcasereasoning/checkpoint-3000/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02df11b51fca37b71ce21908a5e61f79cfd7445aedd546d699a75898f3b0285d
3
+ size 20979856
biomedlm-qlora-medcasereasoning/checkpoint-3000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a10996df22a4cda545dc00111fc9430909000dde03ef1ef8c00502bc8f52df7c
3
+ size 41998603
biomedlm-qlora-medcasereasoning/checkpoint-3000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e719f73cccf05e93acf6c47edff44520b9a3b40996453fd700d8e79a0308ccfc
3
+ size 14645
biomedlm-qlora-medcasereasoning/checkpoint-3000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c06279e7e4fb0cd495ddf269a4d51cb5a507293a8d79fec8d1c52be5ea020565
3
+ size 1383
biomedlm-qlora-medcasereasoning/checkpoint-3000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5c0de700bc1de664ad6f44870930a9f2b463662e272e9404ec3a1755a84ceab
3
+ size 1465
biomedlm-qlora-medcasereasoning/checkpoint-3000/trainer_state.json ADDED
@@ -0,0 +1,454 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.916590284142988,
6
+ "eval_steps": 500,
7
+ "global_step": 3000,
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.015276504735716468,
14
+ "grad_norm": 307.3346862792969,
15
+ "learning_rate": 0.00019700580507179958,
16
+ "loss": 4.2195,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.030553009471432937,
21
+ "grad_norm": 183.25218200683594,
22
+ "learning_rate": 0.0001939505041246563,
23
+ "loss": 3.8053,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.045829514207149404,
28
+ "grad_norm": 504.96185302734375,
29
+ "learning_rate": 0.000190895203177513,
30
+ "loss": 3.5541,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 0.061106018942865874,
35
+ "grad_norm": 432.70635986328125,
36
+ "learning_rate": 0.00018783990223036968,
37
+ "loss": 3.3857,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 0.07638252367858234,
42
+ "grad_norm": 737.7555541992188,
43
+ "learning_rate": 0.0001847846012832264,
44
+ "loss": 3.2653,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 0.09165902841429881,
49
+ "grad_norm": 143.675537109375,
50
+ "learning_rate": 0.00018172930033608312,
51
+ "loss": 3.1635,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 0.10693553315001528,
56
+ "grad_norm": 166.61778259277344,
57
+ "learning_rate": 0.00017867399938893984,
58
+ "loss": 3.0706,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 0.12221203788573175,
63
+ "grad_norm": 1532.4652099609375,
64
+ "learning_rate": 0.00017561869844179653,
65
+ "loss": 3.048,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 0.13748854262144822,
70
+ "grad_norm": 1346.9744873046875,
71
+ "learning_rate": 0.00017256339749465322,
72
+ "loss": 3.0503,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 0.15276504735716467,
77
+ "grad_norm": 335.35736083984375,
78
+ "learning_rate": 0.00016950809654750994,
79
+ "loss": 2.8301,
80
+ "step": 500
81
+ },
82
+ {
83
+ "epoch": 0.16804155209288116,
84
+ "grad_norm": 85.10761260986328,
85
+ "learning_rate": 0.00016645279560036663,
86
+ "loss": 2.7075,
87
+ "step": 550
88
+ },
89
+ {
90
+ "epoch": 0.18331805682859761,
91
+ "grad_norm": 127.73542785644531,
92
+ "learning_rate": 0.00016339749465322335,
93
+ "loss": 2.6585,
94
+ "step": 600
95
+ },
96
+ {
97
+ "epoch": 0.1985945615643141,
98
+ "grad_norm": 116.15327453613281,
99
+ "learning_rate": 0.00016034219370608007,
100
+ "loss": 2.5791,
101
+ "step": 650
102
+ },
103
+ {
104
+ "epoch": 0.21387106630003055,
105
+ "grad_norm": 159.0047607421875,
106
+ "learning_rate": 0.00015728689275893676,
107
+ "loss": 2.5435,
108
+ "step": 700
109
+ },
110
+ {
111
+ "epoch": 0.229147571035747,
112
+ "grad_norm": 265.98944091796875,
113
+ "learning_rate": 0.00015423159181179348,
114
+ "loss": 2.6597,
115
+ "step": 750
116
+ },
117
+ {
118
+ "epoch": 0.2444240757714635,
119
+ "grad_norm": 264.9404602050781,
120
+ "learning_rate": 0.00015117629086465017,
121
+ "loss": 2.6009,
122
+ "step": 800
123
+ },
124
+ {
125
+ "epoch": 0.25970058050718,
126
+ "grad_norm": 330.41644287109375,
127
+ "learning_rate": 0.0001481209899175069,
128
+ "loss": 2.5548,
129
+ "step": 850
130
+ },
131
+ {
132
+ "epoch": 0.27497708524289644,
133
+ "grad_norm": 108.10710906982422,
134
+ "learning_rate": 0.00014506568897036358,
135
+ "loss": 2.5311,
136
+ "step": 900
137
+ },
138
+ {
139
+ "epoch": 0.2902535899786129,
140
+ "grad_norm": 45.05527877807617,
141
+ "learning_rate": 0.00014201038802322028,
142
+ "loss": 2.554,
143
+ "step": 950
144
+ },
145
+ {
146
+ "epoch": 0.30553009471432935,
147
+ "grad_norm": 71.9140396118164,
148
+ "learning_rate": 0.000138955087076077,
149
+ "loss": 2.5609,
150
+ "step": 1000
151
+ },
152
+ {
153
+ "epoch": 0.3208065994500458,
154
+ "grad_norm": 79.13616943359375,
155
+ "learning_rate": 0.0001358997861289337,
156
+ "loss": 2.5838,
157
+ "step": 1050
158
+ },
159
+ {
160
+ "epoch": 0.3360831041857623,
161
+ "grad_norm": 1156.9407958984375,
162
+ "learning_rate": 0.00013284448518179043,
163
+ "loss": 2.6619,
164
+ "step": 1100
165
+ },
166
+ {
167
+ "epoch": 0.35135960892147877,
168
+ "grad_norm": 51.14873504638672,
169
+ "learning_rate": 0.00012978918423464712,
170
+ "loss": 2.5035,
171
+ "step": 1150
172
+ },
173
+ {
174
+ "epoch": 0.36663611365719523,
175
+ "grad_norm": 202.2974395751953,
176
+ "learning_rate": 0.00012673388328750382,
177
+ "loss": 2.5703,
178
+ "step": 1200
179
+ },
180
+ {
181
+ "epoch": 0.3819126183929117,
182
+ "grad_norm": 66.29734802246094,
183
+ "learning_rate": 0.00012367858234036053,
184
+ "loss": 2.4886,
185
+ "step": 1250
186
+ },
187
+ {
188
+ "epoch": 0.3971891231286282,
189
+ "grad_norm": 181.38006591796875,
190
+ "learning_rate": 0.00012062328139321724,
191
+ "loss": 2.4417,
192
+ "step": 1300
193
+ },
194
+ {
195
+ "epoch": 0.41246562786434465,
196
+ "grad_norm": 72.90442657470703,
197
+ "learning_rate": 0.00011756798044607393,
198
+ "loss": 2.3621,
199
+ "step": 1350
200
+ },
201
+ {
202
+ "epoch": 0.4277421326000611,
203
+ "grad_norm": 107.1488265991211,
204
+ "learning_rate": 0.00011451267949893065,
205
+ "loss": 2.3083,
206
+ "step": 1400
207
+ },
208
+ {
209
+ "epoch": 0.44301863733577757,
210
+ "grad_norm": 198.5157012939453,
211
+ "learning_rate": 0.00011145737855178736,
212
+ "loss": 2.2987,
213
+ "step": 1450
214
+ },
215
+ {
216
+ "epoch": 0.458295142071494,
217
+ "grad_norm": 167.29483032226562,
218
+ "learning_rate": 0.00010840207760464408,
219
+ "loss": 2.3515,
220
+ "step": 1500
221
+ },
222
+ {
223
+ "epoch": 0.47357164680721053,
224
+ "grad_norm": 332.89080810546875,
225
+ "learning_rate": 0.00010534677665750077,
226
+ "loss": 2.4297,
227
+ "step": 1550
228
+ },
229
+ {
230
+ "epoch": 0.488848151542927,
231
+ "grad_norm": 342.51971435546875,
232
+ "learning_rate": 0.00010229147571035746,
233
+ "loss": 2.8733,
234
+ "step": 1600
235
+ },
236
+ {
237
+ "epoch": 0.5041246562786434,
238
+ "grad_norm": 76.86676788330078,
239
+ "learning_rate": 9.923617476321419e-05,
240
+ "loss": 2.4265,
241
+ "step": 1650
242
+ },
243
+ {
244
+ "epoch": 0.51940116101436,
245
+ "grad_norm": 140.59532165527344,
246
+ "learning_rate": 9.618087381607088e-05,
247
+ "loss": 2.3204,
248
+ "step": 1700
249
+ },
250
+ {
251
+ "epoch": 0.5346776657500764,
252
+ "grad_norm": 112.0956802368164,
253
+ "learning_rate": 9.312557286892759e-05,
254
+ "loss": 2.3272,
255
+ "step": 1750
256
+ },
257
+ {
258
+ "epoch": 0.5499541704857929,
259
+ "grad_norm": 140.0964813232422,
260
+ "learning_rate": 9.00702719217843e-05,
261
+ "loss": 2.3092,
262
+ "step": 1800
263
+ },
264
+ {
265
+ "epoch": 0.5652306752215093,
266
+ "grad_norm": 105.89209747314453,
267
+ "learning_rate": 8.701497097464101e-05,
268
+ "loss": 2.2663,
269
+ "step": 1850
270
+ },
271
+ {
272
+ "epoch": 0.5805071799572258,
273
+ "grad_norm": 224.0403289794922,
274
+ "learning_rate": 8.395967002749772e-05,
275
+ "loss": 2.2835,
276
+ "step": 1900
277
+ },
278
+ {
279
+ "epoch": 0.5957836846929423,
280
+ "grad_norm": 124.36080169677734,
281
+ "learning_rate": 8.090436908035441e-05,
282
+ "loss": 2.3031,
283
+ "step": 1950
284
+ },
285
+ {
286
+ "epoch": 0.6110601894286587,
287
+ "grad_norm": 165.5836639404297,
288
+ "learning_rate": 7.784906813321113e-05,
289
+ "loss": 2.2918,
290
+ "step": 2000
291
+ },
292
+ {
293
+ "epoch": 0.6263366941643752,
294
+ "grad_norm": 237.20101928710938,
295
+ "learning_rate": 7.479376718606783e-05,
296
+ "loss": 2.4449,
297
+ "step": 2050
298
+ },
299
+ {
300
+ "epoch": 0.6416131989000916,
301
+ "grad_norm": 2535.52587890625,
302
+ "learning_rate": 7.173846623892454e-05,
303
+ "loss": 2.701,
304
+ "step": 2100
305
+ },
306
+ {
307
+ "epoch": 0.6568897036358081,
308
+ "grad_norm": 436.4781494140625,
309
+ "learning_rate": 6.868316529178125e-05,
310
+ "loss": 2.8249,
311
+ "step": 2150
312
+ },
313
+ {
314
+ "epoch": 0.6721662083715246,
315
+ "grad_norm": 615.870361328125,
316
+ "learning_rate": 6.562786434463795e-05,
317
+ "loss": 2.8493,
318
+ "step": 2200
319
+ },
320
+ {
321
+ "epoch": 0.687442713107241,
322
+ "grad_norm": 288.7181396484375,
323
+ "learning_rate": 6.257256339749466e-05,
324
+ "loss": 2.7684,
325
+ "step": 2250
326
+ },
327
+ {
328
+ "epoch": 0.7027192178429575,
329
+ "grad_norm": 137.13819885253906,
330
+ "learning_rate": 5.951726245035136e-05,
331
+ "loss": 2.7183,
332
+ "step": 2300
333
+ },
334
+ {
335
+ "epoch": 0.717995722578674,
336
+ "grad_norm": 102.33476257324219,
337
+ "learning_rate": 5.6461961503208073e-05,
338
+ "loss": 2.5867,
339
+ "step": 2350
340
+ },
341
+ {
342
+ "epoch": 0.7332722273143905,
343
+ "grad_norm": 51.76798629760742,
344
+ "learning_rate": 5.340666055606478e-05,
345
+ "loss": 2.6042,
346
+ "step": 2400
347
+ },
348
+ {
349
+ "epoch": 0.748548732050107,
350
+ "grad_norm": 56.010101318359375,
351
+ "learning_rate": 5.035135960892148e-05,
352
+ "loss": 2.5237,
353
+ "step": 2450
354
+ },
355
+ {
356
+ "epoch": 0.7638252367858234,
357
+ "grad_norm": 48.380126953125,
358
+ "learning_rate": 4.729605866177819e-05,
359
+ "loss": 2.4256,
360
+ "step": 2500
361
+ },
362
+ {
363
+ "epoch": 0.7791017415215399,
364
+ "grad_norm": 54.22650909423828,
365
+ "learning_rate": 4.4240757714634895e-05,
366
+ "loss": 2.3798,
367
+ "step": 2550
368
+ },
369
+ {
370
+ "epoch": 0.7943782462572564,
371
+ "grad_norm": 133.122802734375,
372
+ "learning_rate": 4.11854567674916e-05,
373
+ "loss": 2.1974,
374
+ "step": 2600
375
+ },
376
+ {
377
+ "epoch": 0.8096547509929728,
378
+ "grad_norm": 37.01502990722656,
379
+ "learning_rate": 3.8130155820348306e-05,
380
+ "loss": 2.17,
381
+ "step": 2650
382
+ },
383
+ {
384
+ "epoch": 0.8249312557286893,
385
+ "grad_norm": 111.2568588256836,
386
+ "learning_rate": 3.507485487320501e-05,
387
+ "loss": 2.1655,
388
+ "step": 2700
389
+ },
390
+ {
391
+ "epoch": 0.8402077604644057,
392
+ "grad_norm": 38.26518249511719,
393
+ "learning_rate": 3.2019553926061716e-05,
394
+ "loss": 2.2043,
395
+ "step": 2750
396
+ },
397
+ {
398
+ "epoch": 0.8554842652001222,
399
+ "grad_norm": 90.96832275390625,
400
+ "learning_rate": 2.8964252978918425e-05,
401
+ "loss": 2.1852,
402
+ "step": 2800
403
+ },
404
+ {
405
+ "epoch": 0.8707607699358387,
406
+ "grad_norm": 34.38174819946289,
407
+ "learning_rate": 2.590895203177513e-05,
408
+ "loss": 2.1714,
409
+ "step": 2850
410
+ },
411
+ {
412
+ "epoch": 0.8860372746715551,
413
+ "grad_norm": 50.069122314453125,
414
+ "learning_rate": 2.285365108463184e-05,
415
+ "loss": 2.1867,
416
+ "step": 2900
417
+ },
418
+ {
419
+ "epoch": 0.9013137794072716,
420
+ "grad_norm": 105.68548583984375,
421
+ "learning_rate": 1.9798350137488545e-05,
422
+ "loss": 2.1533,
423
+ "step": 2950
424
+ },
425
+ {
426
+ "epoch": 0.916590284142988,
427
+ "grad_norm": 174.91943359375,
428
+ "learning_rate": 1.674304919034525e-05,
429
+ "loss": 2.1456,
430
+ "step": 3000
431
+ }
432
+ ],
433
+ "logging_steps": 50,
434
+ "max_steps": 3273,
435
+ "num_input_tokens_seen": 0,
436
+ "num_train_epochs": 1,
437
+ "save_steps": 500,
438
+ "stateful_callbacks": {
439
+ "TrainerControl": {
440
+ "args": {
441
+ "should_epoch_stop": false,
442
+ "should_evaluate": false,
443
+ "should_log": false,
444
+ "should_save": true,
445
+ "should_training_stop": false
446
+ },
447
+ "attributes": {}
448
+ }
449
+ },
450
+ "total_flos": 1.8600802910208e+17,
451
+ "train_batch_size": 1,
452
+ "trial_name": null,
453
+ "trial_params": null
454
+ }
biomedlm-qlora-medcasereasoning/checkpoint-3000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fddde750cb92939d7aca03f1ec5d66f04926fedbc74d5756a294daffe8bab3d6
3
+ size 5841
biomedlm-qlora-medcasereasoning/checkpoint-3273/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stanford-crfm/BioMedLM
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:stanford-crfm/BioMedLM
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- 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. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ 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).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.18.1
biomedlm-qlora-medcasereasoning/checkpoint-3273/adapter_config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "stanford-crfm/BioMedLM",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 16,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "c_attn"
33
+ ],
34
+ "target_parameters": null,
35
+ "task_type": "CAUSAL_LM",
36
+ "trainable_token_indices": null,
37
+ "use_dora": false,
38
+ "use_qalora": false,
39
+ "use_rslora": false
40
+ }
biomedlm-qlora-medcasereasoning/checkpoint-3273/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbae80500dd9e929d3726973a6f1526be6e4540b4c6e893f957f6f0a604b01d5
3
+ size 20979856
biomedlm-qlora-medcasereasoning/checkpoint-3273/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1344e0fbb921e8d9ed68995c3e1327b59e232347a2ee8957bda17d29e59fc6e
3
+ size 41998603
biomedlm-qlora-medcasereasoning/checkpoint-3273/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2bf9f2ad016a93ceb5c63370a55da6a4c4bb96f21d09475b735912062bb64b0
3
+ size 14645
biomedlm-qlora-medcasereasoning/checkpoint-3273/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:afc7199df38e49c66f7e37e454675a061721008b120dd94077525fe67ccc6021
3
+ size 1383
biomedlm-qlora-medcasereasoning/checkpoint-3273/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70c113628ed6c1c86cd36ec99984096659af4058dad99d88506663b979e1b401
3
+ size 1465
biomedlm-qlora-medcasereasoning/checkpoint-3273/trainer_state.json ADDED
@@ -0,0 +1,489 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 3273,
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.015276504735716468,
14
+ "grad_norm": 307.3346862792969,
15
+ "learning_rate": 0.00019700580507179958,
16
+ "loss": 4.2195,
17
+ "step": 50
18
+ },
19
+ {
20
+ "epoch": 0.030553009471432937,
21
+ "grad_norm": 183.25218200683594,
22
+ "learning_rate": 0.0001939505041246563,
23
+ "loss": 3.8053,
24
+ "step": 100
25
+ },
26
+ {
27
+ "epoch": 0.045829514207149404,
28
+ "grad_norm": 504.96185302734375,
29
+ "learning_rate": 0.000190895203177513,
30
+ "loss": 3.5541,
31
+ "step": 150
32
+ },
33
+ {
34
+ "epoch": 0.061106018942865874,
35
+ "grad_norm": 432.70635986328125,
36
+ "learning_rate": 0.00018783990223036968,
37
+ "loss": 3.3857,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 0.07638252367858234,
42
+ "grad_norm": 737.7555541992188,
43
+ "learning_rate": 0.0001847846012832264,
44
+ "loss": 3.2653,
45
+ "step": 250
46
+ },
47
+ {
48
+ "epoch": 0.09165902841429881,
49
+ "grad_norm": 143.675537109375,
50
+ "learning_rate": 0.00018172930033608312,
51
+ "loss": 3.1635,
52
+ "step": 300
53
+ },
54
+ {
55
+ "epoch": 0.10693553315001528,
56
+ "grad_norm": 166.61778259277344,
57
+ "learning_rate": 0.00017867399938893984,
58
+ "loss": 3.0706,
59
+ "step": 350
60
+ },
61
+ {
62
+ "epoch": 0.12221203788573175,
63
+ "grad_norm": 1532.4652099609375,
64
+ "learning_rate": 0.00017561869844179653,
65
+ "loss": 3.048,
66
+ "step": 400
67
+ },
68
+ {
69
+ "epoch": 0.13748854262144822,
70
+ "grad_norm": 1346.9744873046875,
71
+ "learning_rate": 0.00017256339749465322,
72
+ "loss": 3.0503,
73
+ "step": 450
74
+ },
75
+ {
76
+ "epoch": 0.15276504735716467,
77
+ "grad_norm": 335.35736083984375,
78
+ "learning_rate": 0.00016950809654750994,
79
+ "loss": 2.8301,
80
+ "step": 500
81
+ },
82
+ {
83
+ "epoch": 0.16804155209288116,
84
+ "grad_norm": 85.10761260986328,
85
+ "learning_rate": 0.00016645279560036663,
86
+ "loss": 2.7075,
87
+ "step": 550
88
+ },
89
+ {
90
+ "epoch": 0.18331805682859761,
91
+ "grad_norm": 127.73542785644531,
92
+ "learning_rate": 0.00016339749465322335,
93
+ "loss": 2.6585,
94
+ "step": 600
95
+ },
96
+ {
97
+ "epoch": 0.1985945615643141,
98
+ "grad_norm": 116.15327453613281,
99
+ "learning_rate": 0.00016034219370608007,
100
+ "loss": 2.5791,
101
+ "step": 650
102
+ },
103
+ {
104
+ "epoch": 0.21387106630003055,
105
+ "grad_norm": 159.0047607421875,
106
+ "learning_rate": 0.00015728689275893676,
107
+ "loss": 2.5435,
108
+ "step": 700
109
+ },
110
+ {
111
+ "epoch": 0.229147571035747,
112
+ "grad_norm": 265.98944091796875,
113
+ "learning_rate": 0.00015423159181179348,
114
+ "loss": 2.6597,
115
+ "step": 750
116
+ },
117
+ {
118
+ "epoch": 0.2444240757714635,
119
+ "grad_norm": 264.9404602050781,
120
+ "learning_rate": 0.00015117629086465017,
121
+ "loss": 2.6009,
122
+ "step": 800
123
+ },
124
+ {
125
+ "epoch": 0.25970058050718,
126
+ "grad_norm": 330.41644287109375,
127
+ "learning_rate": 0.0001481209899175069,
128
+ "loss": 2.5548,
129
+ "step": 850
130
+ },
131
+ {
132
+ "epoch": 0.27497708524289644,
133
+ "grad_norm": 108.10710906982422,
134
+ "learning_rate": 0.00014506568897036358,
135
+ "loss": 2.5311,
136
+ "step": 900
137
+ },
138
+ {
139
+ "epoch": 0.2902535899786129,
140
+ "grad_norm": 45.05527877807617,
141
+ "learning_rate": 0.00014201038802322028,
142
+ "loss": 2.554,
143
+ "step": 950
144
+ },
145
+ {
146
+ "epoch": 0.30553009471432935,
147
+ "grad_norm": 71.9140396118164,
148
+ "learning_rate": 0.000138955087076077,
149
+ "loss": 2.5609,
150
+ "step": 1000
151
+ },
152
+ {
153
+ "epoch": 0.3208065994500458,
154
+ "grad_norm": 79.13616943359375,
155
+ "learning_rate": 0.0001358997861289337,
156
+ "loss": 2.5838,
157
+ "step": 1050
158
+ },
159
+ {
160
+ "epoch": 0.3360831041857623,
161
+ "grad_norm": 1156.9407958984375,
162
+ "learning_rate": 0.00013284448518179043,
163
+ "loss": 2.6619,
164
+ "step": 1100
165
+ },
166
+ {
167
+ "epoch": 0.35135960892147877,
168
+ "grad_norm": 51.14873504638672,
169
+ "learning_rate": 0.00012978918423464712,
170
+ "loss": 2.5035,
171
+ "step": 1150
172
+ },
173
+ {
174
+ "epoch": 0.36663611365719523,
175
+ "grad_norm": 202.2974395751953,
176
+ "learning_rate": 0.00012673388328750382,
177
+ "loss": 2.5703,
178
+ "step": 1200
179
+ },
180
+ {
181
+ "epoch": 0.3819126183929117,
182
+ "grad_norm": 66.29734802246094,
183
+ "learning_rate": 0.00012367858234036053,
184
+ "loss": 2.4886,
185
+ "step": 1250
186
+ },
187
+ {
188
+ "epoch": 0.3971891231286282,
189
+ "grad_norm": 181.38006591796875,
190
+ "learning_rate": 0.00012062328139321724,
191
+ "loss": 2.4417,
192
+ "step": 1300
193
+ },
194
+ {
195
+ "epoch": 0.41246562786434465,
196
+ "grad_norm": 72.90442657470703,
197
+ "learning_rate": 0.00011756798044607393,
198
+ "loss": 2.3621,
199
+ "step": 1350
200
+ },
201
+ {
202
+ "epoch": 0.4277421326000611,
203
+ "grad_norm": 107.1488265991211,
204
+ "learning_rate": 0.00011451267949893065,
205
+ "loss": 2.3083,
206
+ "step": 1400
207
+ },
208
+ {
209
+ "epoch": 0.44301863733577757,
210
+ "grad_norm": 198.5157012939453,
211
+ "learning_rate": 0.00011145737855178736,
212
+ "loss": 2.2987,
213
+ "step": 1450
214
+ },
215
+ {
216
+ "epoch": 0.458295142071494,
217
+ "grad_norm": 167.29483032226562,
218
+ "learning_rate": 0.00010840207760464408,
219
+ "loss": 2.3515,
220
+ "step": 1500
221
+ },
222
+ {
223
+ "epoch": 0.47357164680721053,
224
+ "grad_norm": 332.89080810546875,
225
+ "learning_rate": 0.00010534677665750077,
226
+ "loss": 2.4297,
227
+ "step": 1550
228
+ },
229
+ {
230
+ "epoch": 0.488848151542927,
231
+ "grad_norm": 342.51971435546875,
232
+ "learning_rate": 0.00010229147571035746,
233
+ "loss": 2.8733,
234
+ "step": 1600
235
+ },
236
+ {
237
+ "epoch": 0.5041246562786434,
238
+ "grad_norm": 76.86676788330078,
239
+ "learning_rate": 9.923617476321419e-05,
240
+ "loss": 2.4265,
241
+ "step": 1650
242
+ },
243
+ {
244
+ "epoch": 0.51940116101436,
245
+ "grad_norm": 140.59532165527344,
246
+ "learning_rate": 9.618087381607088e-05,
247
+ "loss": 2.3204,
248
+ "step": 1700
249
+ },
250
+ {
251
+ "epoch": 0.5346776657500764,
252
+ "grad_norm": 112.0956802368164,
253
+ "learning_rate": 9.312557286892759e-05,
254
+ "loss": 2.3272,
255
+ "step": 1750
256
+ },
257
+ {
258
+ "epoch": 0.5499541704857929,
259
+ "grad_norm": 140.0964813232422,
260
+ "learning_rate": 9.00702719217843e-05,
261
+ "loss": 2.3092,
262
+ "step": 1800
263
+ },
264
+ {
265
+ "epoch": 0.5652306752215093,
266
+ "grad_norm": 105.89209747314453,
267
+ "learning_rate": 8.701497097464101e-05,
268
+ "loss": 2.2663,
269
+ "step": 1850
270
+ },
271
+ {
272
+ "epoch": 0.5805071799572258,
273
+ "grad_norm": 224.0403289794922,
274
+ "learning_rate": 8.395967002749772e-05,
275
+ "loss": 2.2835,
276
+ "step": 1900
277
+ },
278
+ {
279
+ "epoch": 0.5957836846929423,
280
+ "grad_norm": 124.36080169677734,
281
+ "learning_rate": 8.090436908035441e-05,
282
+ "loss": 2.3031,
283
+ "step": 1950
284
+ },
285
+ {
286
+ "epoch": 0.6110601894286587,
287
+ "grad_norm": 165.5836639404297,
288
+ "learning_rate": 7.784906813321113e-05,
289
+ "loss": 2.2918,
290
+ "step": 2000
291
+ },
292
+ {
293
+ "epoch": 0.6263366941643752,
294
+ "grad_norm": 237.20101928710938,
295
+ "learning_rate": 7.479376718606783e-05,
296
+ "loss": 2.4449,
297
+ "step": 2050
298
+ },
299
+ {
300
+ "epoch": 0.6416131989000916,
301
+ "grad_norm": 2535.52587890625,
302
+ "learning_rate": 7.173846623892454e-05,
303
+ "loss": 2.701,
304
+ "step": 2100
305
+ },
306
+ {
307
+ "epoch": 0.6568897036358081,
308
+ "grad_norm": 436.4781494140625,
309
+ "learning_rate": 6.868316529178125e-05,
310
+ "loss": 2.8249,
311
+ "step": 2150
312
+ },
313
+ {
314
+ "epoch": 0.6721662083715246,
315
+ "grad_norm": 615.870361328125,
316
+ "learning_rate": 6.562786434463795e-05,
317
+ "loss": 2.8493,
318
+ "step": 2200
319
+ },
320
+ {
321
+ "epoch": 0.687442713107241,
322
+ "grad_norm": 288.7181396484375,
323
+ "learning_rate": 6.257256339749466e-05,
324
+ "loss": 2.7684,
325
+ "step": 2250
326
+ },
327
+ {
328
+ "epoch": 0.7027192178429575,
329
+ "grad_norm": 137.13819885253906,
330
+ "learning_rate": 5.951726245035136e-05,
331
+ "loss": 2.7183,
332
+ "step": 2300
333
+ },
334
+ {
335
+ "epoch": 0.717995722578674,
336
+ "grad_norm": 102.33476257324219,
337
+ "learning_rate": 5.6461961503208073e-05,
338
+ "loss": 2.5867,
339
+ "step": 2350
340
+ },
341
+ {
342
+ "epoch": 0.7332722273143905,
343
+ "grad_norm": 51.76798629760742,
344
+ "learning_rate": 5.340666055606478e-05,
345
+ "loss": 2.6042,
346
+ "step": 2400
347
+ },
348
+ {
349
+ "epoch": 0.748548732050107,
350
+ "grad_norm": 56.010101318359375,
351
+ "learning_rate": 5.035135960892148e-05,
352
+ "loss": 2.5237,
353
+ "step": 2450
354
+ },
355
+ {
356
+ "epoch": 0.7638252367858234,
357
+ "grad_norm": 48.380126953125,
358
+ "learning_rate": 4.729605866177819e-05,
359
+ "loss": 2.4256,
360
+ "step": 2500
361
+ },
362
+ {
363
+ "epoch": 0.7791017415215399,
364
+ "grad_norm": 54.22650909423828,
365
+ "learning_rate": 4.4240757714634895e-05,
366
+ "loss": 2.3798,
367
+ "step": 2550
368
+ },
369
+ {
370
+ "epoch": 0.7943782462572564,
371
+ "grad_norm": 133.122802734375,
372
+ "learning_rate": 4.11854567674916e-05,
373
+ "loss": 2.1974,
374
+ "step": 2600
375
+ },
376
+ {
377
+ "epoch": 0.8096547509929728,
378
+ "grad_norm": 37.01502990722656,
379
+ "learning_rate": 3.8130155820348306e-05,
380
+ "loss": 2.17,
381
+ "step": 2650
382
+ },
383
+ {
384
+ "epoch": 0.8249312557286893,
385
+ "grad_norm": 111.2568588256836,
386
+ "learning_rate": 3.507485487320501e-05,
387
+ "loss": 2.1655,
388
+ "step": 2700
389
+ },
390
+ {
391
+ "epoch": 0.8402077604644057,
392
+ "grad_norm": 38.26518249511719,
393
+ "learning_rate": 3.2019553926061716e-05,
394
+ "loss": 2.2043,
395
+ "step": 2750
396
+ },
397
+ {
398
+ "epoch": 0.8554842652001222,
399
+ "grad_norm": 90.96832275390625,
400
+ "learning_rate": 2.8964252978918425e-05,
401
+ "loss": 2.1852,
402
+ "step": 2800
403
+ },
404
+ {
405
+ "epoch": 0.8707607699358387,
406
+ "grad_norm": 34.38174819946289,
407
+ "learning_rate": 2.590895203177513e-05,
408
+ "loss": 2.1714,
409
+ "step": 2850
410
+ },
411
+ {
412
+ "epoch": 0.8860372746715551,
413
+ "grad_norm": 50.069122314453125,
414
+ "learning_rate": 2.285365108463184e-05,
415
+ "loss": 2.1867,
416
+ "step": 2900
417
+ },
418
+ {
419
+ "epoch": 0.9013137794072716,
420
+ "grad_norm": 105.68548583984375,
421
+ "learning_rate": 1.9798350137488545e-05,
422
+ "loss": 2.1533,
423
+ "step": 2950
424
+ },
425
+ {
426
+ "epoch": 0.916590284142988,
427
+ "grad_norm": 174.91943359375,
428
+ "learning_rate": 1.674304919034525e-05,
429
+ "loss": 2.1456,
430
+ "step": 3000
431
+ },
432
+ {
433
+ "epoch": 0.9318667888787046,
434
+ "grad_norm": 39.45339584350586,
435
+ "learning_rate": 1.3687748243201956e-05,
436
+ "loss": 2.1246,
437
+ "step": 3050
438
+ },
439
+ {
440
+ "epoch": 0.9471432936144211,
441
+ "grad_norm": 47.88697052001953,
442
+ "learning_rate": 1.0632447296058663e-05,
443
+ "loss": 2.1387,
444
+ "step": 3100
445
+ },
446
+ {
447
+ "epoch": 0.9624197983501375,
448
+ "grad_norm": 25.454118728637695,
449
+ "learning_rate": 7.577146348915369e-06,
450
+ "loss": 2.1273,
451
+ "step": 3150
452
+ },
453
+ {
454
+ "epoch": 0.977696303085854,
455
+ "grad_norm": 125.96427154541016,
456
+ "learning_rate": 4.521845401772075e-06,
457
+ "loss": 2.1116,
458
+ "step": 3200
459
+ },
460
+ {
461
+ "epoch": 0.9929728078215704,
462
+ "grad_norm": 70.28050994873047,
463
+ "learning_rate": 1.466544454628781e-06,
464
+ "loss": 2.1406,
465
+ "step": 3250
466
+ }
467
+ ],
468
+ "logging_steps": 50,
469
+ "max_steps": 3273,
470
+ "num_input_tokens_seen": 0,
471
+ "num_train_epochs": 1,
472
+ "save_steps": 500,
473
+ "stateful_callbacks": {
474
+ "TrainerControl": {
475
+ "args": {
476
+ "should_epoch_stop": false,
477
+ "should_evaluate": false,
478
+ "should_log": false,
479
+ "should_save": true,
480
+ "should_training_stop": true
481
+ },
482
+ "attributes": {}
483
+ }
484
+ },
485
+ "total_flos": 2.0293475975036928e+17,
486
+ "train_batch_size": 1,
487
+ "trial_name": null,
488
+ "trial_params": null
489
+ }
biomedlm-qlora-medcasereasoning/checkpoint-3273/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fddde750cb92939d7aca03f1ec5d66f04926fedbc74d5756a294daffe8bab3d6
3
+ size 5841