Mardiyyah commited on
Commit
937a9d1
·
verified ·
1 Parent(s): fee4011

Model save

Browse files
Files changed (7) hide show
  1. README.md +25 -25
  2. all_results.json +12 -12
  3. config.json +12 -12
  4. eval_results.json +12 -12
  5. model.safetensors +1 -1
  6. train_results.json +4 -4
  7. trainer_state.json +206 -206
README.md CHANGED
@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext](https://huggingface.co/microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.4408
25
- - Precision: 0.5732
26
- - Recall: 0.5514
27
- - F1: 0.5621
28
- - Accuracy: 0.9297
29
 
30
  ## Model description
31
 
@@ -58,26 +58,26 @@ The following hyperparameters were used during training:
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
- | 1.0965 | 1.0 | 115 | 0.4152 | 0.2414 | 0.1725 | 0.2012 | 0.8701 |
62
- | 0.2292 | 2.0 | 230 | 0.2860 | 0.4389 | 0.4920 | 0.4639 | 0.9119 |
63
- | 0.1037 | 3.0 | 345 | 0.2779 | 0.4939 | 0.5201 | 0.5067 | 0.9200 |
64
- | 0.0684 | 4.0 | 460 | 0.2971 | 0.5515 | 0.5217 | 0.5362 | 0.9274 |
65
- | 0.0463 | 5.0 | 575 | 0.3129 | 0.5618 | 0.5010 | 0.5297 | 0.9276 |
66
- | 0.0319 | 6.0 | 690 | 0.3199 | 0.5849 | 0.5440 | 0.5637 | 0.9311 |
67
- | 0.0257 | 7.0 | 805 | 0.3769 | 0.5466 | 0.5068 | 0.5260 | 0.9257 |
68
- | 0.0223 | 8.0 | 920 | 0.3569 | 0.5240 | 0.5510 | 0.5372 | 0.9242 |
69
- | 0.0176 | 9.0 | 1035 | 0.3772 | 0.5599 | 0.5393 | 0.5494 | 0.9250 |
70
- | 0.0129 | 10.0 | 1150 | 0.3583 | 0.5622 | 0.5479 | 0.5550 | 0.9298 |
71
- | 0.0112 | 11.0 | 1265 | 0.3943 | 0.5513 | 0.5194 | 0.5348 | 0.9262 |
72
- | 0.0096 | 12.0 | 1380 | 0.4049 | 0.5701 | 0.5327 | 0.5507 | 0.9285 |
73
- | 0.0078 | 13.0 | 1495 | 0.4228 | 0.5410 | 0.5420 | 0.5415 | 0.9271 |
74
- | 0.0068 | 14.0 | 1610 | 0.4158 | 0.5503 | 0.5327 | 0.5413 | 0.9264 |
75
- | 0.0057 | 15.0 | 1725 | 0.4150 | 0.5837 | 0.5428 | 0.5625 | 0.9306 |
76
- | 0.0051 | 16.0 | 1840 | 0.4277 | 0.5696 | 0.5491 | 0.5591 | 0.9295 |
77
- | 0.0048 | 17.0 | 1955 | 0.4313 | 0.5614 | 0.5487 | 0.5550 | 0.9296 |
78
- | 0.004 | 18.0 | 2070 | 0.4391 | 0.5724 | 0.5522 | 0.5621 | 0.9297 |
79
- | 0.0036 | 19.0 | 2185 | 0.4471 | 0.5718 | 0.5467 | 0.5590 | 0.9293 |
80
- | 0.0035 | 20.0 | 2300 | 0.4408 | 0.5732 | 0.5514 | 0.5621 | 0.9297 |
81
 
82
 
83
  ### Framework versions
 
21
 
22
  This model is a fine-tuned version of [microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext](https://huggingface.co/microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.4573
25
+ - Precision: 0.5991
26
+ - Recall: 0.5668
27
+ - F1: 0.5825
28
+ - Accuracy: 0.9301
29
 
30
  ## Model description
31
 
 
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
+ | 1.0711 | 1.0 | 115 | 0.4380 | 0.2039 | 0.1800 | 0.1912 | 0.8599 |
62
+ | 0.2579 | 2.0 | 230 | 0.2830 | 0.4117 | 0.3991 | 0.4053 | 0.9112 |
63
+ | 0.1083 | 3.0 | 345 | 0.2841 | 0.5199 | 0.4894 | 0.5042 | 0.9222 |
64
+ | 0.0707 | 4.0 | 460 | 0.2899 | 0.5291 | 0.5481 | 0.5384 | 0.9240 |
65
+ | 0.0472 | 5.0 | 575 | 0.2954 | 0.5816 | 0.5354 | 0.5576 | 0.9297 |
66
+ | 0.0336 | 6.0 | 690 | 0.3135 | 0.5902 | 0.5354 | 0.5615 | 0.9296 |
67
+ | 0.0232 | 7.0 | 805 | 0.3596 | 0.6125 | 0.5419 | 0.5750 | 0.9299 |
68
+ | 0.0178 | 8.0 | 920 | 0.3412 | 0.5675 | 0.5903 | 0.5787 | 0.9285 |
69
+ | 0.0181 | 9.0 | 1035 | 0.3621 | 0.6100 | 0.5470 | 0.5768 | 0.9296 |
70
+ | 0.0132 | 10.0 | 1150 | 0.3638 | 0.5904 | 0.5733 | 0.5817 | 0.9302 |
71
+ | 0.0106 | 11.0 | 1265 | 0.4134 | 0.6078 | 0.5556 | 0.5805 | 0.9303 |
72
+ | 0.0088 | 12.0 | 1380 | 0.4197 | 0.6023 | 0.5579 | 0.5793 | 0.9297 |
73
+ | 0.0071 | 13.0 | 1495 | 0.4162 | 0.5947 | 0.5712 | 0.5828 | 0.9301 |
74
+ | 0.0067 | 14.0 | 1610 | 0.4196 | 0.6077 | 0.5518 | 0.5784 | 0.9305 |
75
+ | 0.005 | 15.0 | 1725 | 0.4391 | 0.5892 | 0.5661 | 0.5774 | 0.9295 |
76
+ | 0.0047 | 16.0 | 1840 | 0.4424 | 0.5872 | 0.5682 | 0.5775 | 0.9295 |
77
+ | 0.0042 | 17.0 | 1955 | 0.4632 | 0.5939 | 0.5539 | 0.5732 | 0.9293 |
78
+ | 0.0037 | 18.0 | 2070 | 0.4541 | 0.5971 | 0.5692 | 0.5828 | 0.9300 |
79
+ | 0.0041 | 19.0 | 2185 | 0.4611 | 0.6069 | 0.5631 | 0.5842 | 0.9307 |
80
+ | 0.0034 | 20.0 | 2300 | 0.4573 | 0.5991 | 0.5668 | 0.5825 | 0.9301 |
81
 
82
 
83
  ### Framework versions
all_results.json CHANGED
@@ -1,17 +1,17 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_accuracy": 0.9306721223816056,
4
- "eval_f1": 0.6088867954152929,
5
- "eval_loss": 0.38036319613456726,
6
- "eval_precision": 0.6474123539232054,
7
- "eval_recall": 0.5746887966804979,
8
- "eval_runtime": 4.4327,
9
  "eval_samples": 952,
10
- "eval_samples_per_second": 214.769,
11
- "eval_steps_per_second": 13.536,
12
  "total_flos": 1090015519875414.0,
13
- "train_loss": 0.09238651011301123,
14
- "train_runtime": 503.6467,
15
- "train_samples_per_second": 72.948,
16
- "train_steps_per_second": 4.567
17
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_accuracy": 0.9311370980866249,
4
+ "eval_f1": 0.5635515913237381,
5
+ "eval_loss": 0.32190775871276855,
6
+ "eval_precision": 0.585016835016835,
7
+ "eval_recall": 0.543605788032851,
8
+ "eval_runtime": 4.6723,
9
  "eval_samples": 952,
10
+ "eval_samples_per_second": 203.754,
11
+ "eval_steps_per_second": 12.842,
12
  "total_flos": 1090015519875414.0,
13
+ "train_loss": 0.08582915891771731,
14
+ "train_runtime": 509.1667,
15
+ "train_samples_per_second": 72.157,
16
+ "train_steps_per_second": 4.517
17
  }
config.json CHANGED
@@ -9,24 +9,24 @@
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "O",
13
  "1": "B-CellType",
14
- "2": "I-Tissue",
15
- "3": "B-Tissue",
16
- "4": "B-CellLine",
17
- "5": "I-CellLine",
18
- "6": "I-CellType"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "B-CellLine": 4,
24
  "B-CellType": 1,
25
- "B-Tissue": 3,
26
- "I-CellLine": 5,
27
- "I-CellType": 6,
28
- "I-Tissue": 2,
29
- "O": 0
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "max_position_embeddings": 512,
 
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "B-CellLine",
13
  "1": "B-CellType",
14
+ "2": "B-Tissue",
15
+ "3": "I-CellLine",
16
+ "4": "O",
17
+ "5": "I-CellType",
18
+ "6": "I-Tissue"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "B-CellLine": 0,
24
  "B-CellType": 1,
25
+ "B-Tissue": 2,
26
+ "I-CellLine": 3,
27
+ "I-CellType": 5,
28
+ "I-Tissue": 6,
29
+ "O": 4
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "max_position_embeddings": 512,
eval_results.json CHANGED
@@ -1,17 +1,17 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_accuracy": 0.9306721223816056,
4
- "eval_f1": 0.6088867954152929,
5
- "eval_loss": 0.38036319613456726,
6
- "eval_precision": 0.6474123539232054,
7
- "eval_recall": 0.5746887966804979,
8
- "eval_runtime": 4.4327,
9
  "eval_samples": 952,
10
- "eval_samples_per_second": 214.769,
11
- "eval_steps_per_second": 13.536,
12
  "total_flos": 1090015519875414.0,
13
- "train_loss": 0.09238651011301123,
14
- "train_runtime": 503.6467,
15
- "train_samples_per_second": 72.948,
16
- "train_steps_per_second": 4.567
17
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_accuracy": 0.9311370980866249,
4
+ "eval_f1": 0.5635515913237381,
5
+ "eval_loss": 0.32190775871276855,
6
+ "eval_precision": 0.585016835016835,
7
+ "eval_recall": 0.543605788032851,
8
+ "eval_runtime": 4.6723,
9
  "eval_samples": 952,
10
+ "eval_samples_per_second": 203.754,
11
+ "eval_steps_per_second": 12.842,
12
  "total_flos": 1090015519875414.0,
13
+ "train_loss": 0.08582915891771731,
14
+ "train_runtime": 509.1667,
15
+ "train_samples_per_second": 72.157,
16
+ "train_steps_per_second": 4.517
17
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a268e7c25a12787aebe8fb71978a422e5aabc01d313c7c2b2a90fcc54a70c90b
3
  size 435611468
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8681f37519c7f7232f0f8d65a1870739767260dc6529a286548ff77034525cf
3
  size 435611468
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 20.0,
3
  "total_flos": 1090015519875414.0,
4
- "train_loss": 0.09238651011301123,
5
- "train_runtime": 503.6467,
6
- "train_samples_per_second": 72.948,
7
- "train_steps_per_second": 4.567
8
  }
 
1
  {
2
  "epoch": 20.0,
3
  "total_flos": 1090015519875414.0,
4
+ "train_loss": 0.08582915891771731,
5
+ "train_runtime": 509.1667,
6
+ "train_samples_per_second": 72.157,
7
+ "train_steps_per_second": 4.517
8
  }
trainer_state.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "best_metric": 0.6088867954152929,
3
- "best_model_checkpoint": "/nfs/production/literature/amina-mardiyyah/new_data/OT-Entity-Extraction-Pipeline/model_outputs/hf/CeLLaTe-AL-Test_1.0/base/BiomedNLP_BiomedBERT_base/checkpoint-1035",
4
  "epoch": 20.0,
5
  "eval_steps": 500,
6
  "global_step": 2300,
@@ -10,392 +10,392 @@
10
  "log_history": [
11
  {
12
  "epoch": 1.0,
13
- "grad_norm": 5.316380977630615,
14
  "learning_rate": 9.652173913043478e-06,
15
- "loss": 1.2128,
16
  "step": 115
17
  },
18
  {
19
  "epoch": 1.0,
20
- "eval_accuracy": 0.8566014925720131,
21
- "eval_f1": 0.06772100567721005,
22
- "eval_loss": 0.48805075883865356,
23
- "eval_precision": 0.10718870346598203,
24
- "eval_recall": 0.04949614700652045,
25
- "eval_runtime": 4.4661,
26
- "eval_samples_per_second": 213.16,
27
- "eval_steps_per_second": 13.434,
28
  "step": 115
29
  },
30
  {
31
  "epoch": 2.0,
32
- "grad_norm": 7.310900688171387,
33
  "learning_rate": 1.965217391304348e-05,
34
- "loss": 0.2669,
35
  "step": 230
36
  },
37
  {
38
  "epoch": 2.0,
39
- "eval_accuracy": 0.9064003905795922,
40
- "eval_f1": 0.42481833426495247,
41
- "eval_loss": 0.31054508686065674,
42
- "eval_precision": 0.40190375462718136,
43
- "eval_recall": 0.45050385299347956,
44
- "eval_runtime": 4.5249,
45
- "eval_samples_per_second": 210.392,
46
- "eval_steps_per_second": 13.26,
47
  "step": 230
48
  },
49
  {
50
  "epoch": 3.0,
51
- "grad_norm": 1.7260887622833252,
52
  "learning_rate": 1.892753623188406e-05,
53
- "loss": 0.1124,
54
  "step": 345
55
  },
56
  {
57
  "epoch": 3.0,
58
- "eval_accuracy": 0.9216283449189779,
59
- "eval_f1": 0.5345465761875386,
60
- "eval_loss": 0.28254440426826477,
61
- "eval_precision": 0.5572347266881029,
62
- "eval_recall": 0.5136336692353289,
63
- "eval_runtime": 4.5035,
64
- "eval_samples_per_second": 211.39,
65
- "eval_steps_per_second": 13.323,
66
  "step": 345
67
  },
68
  {
69
  "epoch": 4.0,
70
- "grad_norm": 3.026144504547119,
71
  "learning_rate": 1.781642512077295e-05,
72
- "loss": 0.0636,
73
  "step": 460
74
  },
75
  {
76
  "epoch": 4.0,
77
- "eval_accuracy": 0.9237207355915653,
78
- "eval_f1": 0.5617733684700047,
79
- "eval_loss": 0.30061712861061096,
80
- "eval_precision": 0.5888852778680533,
81
- "eval_recall": 0.5370480142264374,
82
- "eval_runtime": 4.502,
83
- "eval_samples_per_second": 211.462,
84
- "eval_steps_per_second": 13.327,
85
  "step": 460
86
  },
87
  {
88
  "epoch": 5.0,
89
- "grad_norm": 2.986138105392456,
90
  "learning_rate": 1.6705314009661836e-05,
91
- "loss": 0.0479,
92
  "step": 575
93
  },
94
  {
95
  "epoch": 5.0,
96
- "eval_accuracy": 0.9304163857438449,
97
- "eval_f1": 0.5881973556917123,
98
- "eval_loss": 0.3030494153499603,
99
- "eval_precision": 0.6449787835926449,
100
- "eval_recall": 0.5406046235921754,
101
- "eval_runtime": 4.4799,
102
- "eval_samples_per_second": 212.503,
103
- "eval_steps_per_second": 13.393,
104
  "step": 575
105
  },
106
  {
107
  "epoch": 6.0,
108
- "grad_norm": 3.3358774185180664,
109
  "learning_rate": 1.5594202898550727e-05,
110
- "loss": 0.0309,
111
  "step": 690
112
  },
113
  {
114
  "epoch": 6.0,
115
- "eval_accuracy": 0.9293469416223002,
116
- "eval_f1": 0.5916653462208842,
117
- "eval_loss": 0.33140063285827637,
118
- "eval_precision": 0.6356826693905345,
119
- "eval_recall": 0.55334914048607,
120
- "eval_runtime": 4.7277,
121
- "eval_samples_per_second": 201.366,
122
- "eval_steps_per_second": 12.691,
123
  "step": 690
124
  },
125
  {
126
  "epoch": 7.0,
127
- "grad_norm": 0.1228083148598671,
128
  "learning_rate": 1.4483091787439615e-05,
129
- "loss": 0.0211,
130
  "step": 805
131
  },
132
  {
133
  "epoch": 7.0,
134
- "eval_accuracy": 0.9299049124683235,
135
- "eval_f1": 0.587739822180627,
136
- "eval_loss": 0.3424608111381531,
137
- "eval_precision": 0.6203490286466908,
138
- "eval_recall": 0.5583876704208655,
139
- "eval_runtime": 4.5844,
140
- "eval_samples_per_second": 207.659,
141
- "eval_steps_per_second": 13.088,
142
  "step": 805
143
  },
144
  {
145
  "epoch": 8.0,
146
- "grad_norm": 1.2195112705230713,
147
  "learning_rate": 1.3371980676328504e-05,
148
- "loss": 0.0185,
149
  "step": 920
150
  },
151
  {
152
  "epoch": 8.0,
153
- "eval_accuracy": 0.9258596238346546,
154
- "eval_f1": 0.5938668661181751,
155
- "eval_loss": 0.3702571392059326,
156
- "eval_precision": 0.59951676230746,
157
- "eval_recall": 0.5883224659158269,
158
- "eval_runtime": 4.48,
159
- "eval_samples_per_second": 212.501,
160
- "eval_steps_per_second": 13.393,
161
  "step": 920
162
  },
163
  {
164
  "epoch": 9.0,
165
- "grad_norm": 1.2991191148757935,
166
  "learning_rate": 1.2260869565217393e-05,
167
- "loss": 0.0136,
168
  "step": 1035
169
  },
170
  {
171
  "epoch": 9.0,
172
- "eval_accuracy": 0.9306721223816056,
173
- "eval_f1": 0.6088867954152929,
174
- "eval_loss": 0.3786067068576813,
175
- "eval_precision": 0.6474123539232054,
176
- "eval_recall": 0.5746887966804979,
177
- "eval_runtime": 4.5242,
178
- "eval_samples_per_second": 210.426,
179
- "eval_steps_per_second": 13.262,
180
  "step": 1035
181
  },
182
  {
183
  "epoch": 10.0,
184
- "grad_norm": 0.06892040371894836,
185
  "learning_rate": 1.114975845410628e-05,
186
- "loss": 0.0109,
187
  "step": 1150
188
  },
189
  {
190
  "epoch": 10.0,
191
- "eval_accuracy": 0.9306256248111037,
192
- "eval_f1": 0.6087756980668917,
193
- "eval_loss": 0.36435195803642273,
194
- "eval_precision": 0.6310432569974554,
195
- "eval_recall": 0.5880260818020154,
196
- "eval_runtime": 4.539,
197
- "eval_samples_per_second": 209.737,
198
- "eval_steps_per_second": 13.219,
199
  "step": 1150
200
  },
201
  {
202
  "epoch": 11.0,
203
- "grad_norm": 2.4444758892059326,
204
  "learning_rate": 1.003864734299517e-05,
205
- "loss": 0.0088,
206
  "step": 1265
207
  },
208
  {
209
  "epoch": 11.0,
210
- "eval_accuracy": 0.9302071466765861,
211
- "eval_f1": 0.6059678653404744,
212
- "eval_loss": 0.38957855105400085,
213
- "eval_precision": 0.6263840556785827,
214
- "eval_recall": 0.5868405453467694,
215
- "eval_runtime": 4.5063,
216
- "eval_samples_per_second": 211.261,
217
- "eval_steps_per_second": 13.315,
218
  "step": 1265
219
  },
220
  {
221
  "epoch": 12.0,
222
- "grad_norm": 1.3031020164489746,
223
  "learning_rate": 8.927536231884059e-06,
224
- "loss": 0.0074,
225
  "step": 1380
226
  },
227
  {
228
  "epoch": 12.0,
229
- "eval_accuracy": 0.9306488735963546,
230
- "eval_f1": 0.6029277218664226,
231
- "eval_loss": 0.4022594094276428,
232
- "eval_precision": 0.6209170854271356,
233
- "eval_recall": 0.5859513930053349,
234
- "eval_runtime": 4.6831,
235
- "eval_samples_per_second": 203.283,
236
- "eval_steps_per_second": 12.812,
237
  "step": 1380
238
  },
239
  {
240
  "epoch": 13.0,
241
- "grad_norm": 0.01429770328104496,
242
  "learning_rate": 7.816425120772948e-06,
243
- "loss": 0.0061,
244
  "step": 1495
245
  },
246
  {
247
  "epoch": 13.0,
248
- "eval_accuracy": 0.9264640922511799,
249
- "eval_f1": 0.5745501285347044,
250
- "eval_loss": 0.4580424129962921,
251
- "eval_precision": 0.6273684210526316,
252
- "eval_recall": 0.5299347954949615,
253
- "eval_runtime": 4.4923,
254
- "eval_samples_per_second": 211.919,
255
- "eval_steps_per_second": 13.356,
256
  "step": 1495
257
  },
258
  {
259
  "epoch": 14.0,
260
- "grad_norm": 0.15813925862312317,
261
  "learning_rate": 6.705314009661836e-06,
262
- "loss": 0.0054,
263
  "step": 1610
264
  },
265
  {
266
  "epoch": 14.0,
267
- "eval_accuracy": 0.9284634877827633,
268
- "eval_f1": 0.5897002305918525,
269
- "eval_loss": 0.4323696494102478,
270
- "eval_precision": 0.6125838390290642,
271
- "eval_recall": 0.5684647302904564,
272
- "eval_runtime": 4.525,
273
- "eval_samples_per_second": 210.386,
274
- "eval_steps_per_second": 13.26,
275
  "step": 1610
276
  },
277
  {
278
  "epoch": 15.0,
279
- "grad_norm": 0.03321361541748047,
280
  "learning_rate": 5.594202898550725e-06,
281
- "loss": 0.0046,
282
  "step": 1725
283
  },
284
  {
285
  "epoch": 15.0,
286
- "eval_accuracy": 0.9308581126636133,
287
- "eval_f1": 0.606097192876547,
288
- "eval_loss": 0.4126320481300354,
289
- "eval_precision": 0.6174661746617466,
290
- "eval_recall": 0.5951393005334914,
291
- "eval_runtime": 4.4658,
292
- "eval_samples_per_second": 213.173,
293
- "eval_steps_per_second": 13.435,
294
  "step": 1725
295
  },
296
  {
297
  "epoch": 16.0,
298
- "grad_norm": 1.5817041397094727,
299
  "learning_rate": 4.483091787439614e-06,
300
- "loss": 0.0039,
301
  "step": 1840
302
  },
303
  {
304
  "epoch": 16.0,
305
- "eval_accuracy": 0.9288354683467789,
306
- "eval_f1": 0.6030120481927711,
307
- "eval_loss": 0.43053802847862244,
308
- "eval_precision": 0.6129822412737294,
309
- "eval_recall": 0.5933609958506224,
310
- "eval_runtime": 4.5,
311
- "eval_samples_per_second": 211.555,
312
- "eval_steps_per_second": 13.333,
313
  "step": 1840
314
  },
315
  {
316
  "epoch": 17.0,
317
- "grad_norm": 0.15061946213245392,
318
  "learning_rate": 3.371980676328503e-06,
319
- "loss": 0.0038,
320
  "step": 1955
321
  },
322
  {
323
  "epoch": 17.0,
324
- "eval_accuracy": 0.9294631855485551,
325
- "eval_f1": 0.5975293579380816,
326
- "eval_loss": 0.4329366087913513,
327
- "eval_precision": 0.6154571159283695,
328
- "eval_recall": 0.580616478956728,
329
- "eval_runtime": 4.5338,
330
- "eval_samples_per_second": 209.98,
331
- "eval_steps_per_second": 13.234,
332
  "step": 1955
333
  },
334
  {
335
  "epoch": 18.0,
336
- "grad_norm": 0.04291839525103569,
337
  "learning_rate": 2.2608695652173913e-06,
338
- "loss": 0.003,
339
  "step": 2070
340
  },
341
  {
342
  "epoch": 18.0,
343
- "eval_accuracy": 0.9300676539650803,
344
- "eval_f1": 0.6012176560121766,
345
- "eval_loss": 0.43739748001098633,
346
- "eval_precision": 0.6179599499374218,
347
- "eval_recall": 0.5853586247777119,
348
- "eval_runtime": 4.6599,
349
- "eval_samples_per_second": 204.295,
350
- "eval_steps_per_second": 12.876,
351
  "step": 2070
352
  },
353
  {
354
  "epoch": 19.0,
355
- "grad_norm": 0.073442243039608,
356
  "learning_rate": 1.1497584541062802e-06,
357
- "loss": 0.0031,
358
  "step": 2185
359
  },
360
  {
361
  "epoch": 19.0,
362
- "eval_accuracy": 0.930369888173343,
363
- "eval_f1": 0.5989502933004013,
364
- "eval_loss": 0.452503502368927,
365
- "eval_precision": 0.625,
366
- "eval_recall": 0.5749851807943094,
367
- "eval_runtime": 4.4956,
368
- "eval_samples_per_second": 211.761,
369
- "eval_steps_per_second": 13.346,
370
  "step": 2185
371
  },
372
  {
373
  "epoch": 20.0,
374
- "grad_norm": 0.008639683946967125,
375
  "learning_rate": 3.864734299516908e-08,
376
- "loss": 0.0028,
377
  "step": 2300
378
  },
379
  {
380
  "epoch": 20.0,
381
- "eval_accuracy": 0.9302071466765861,
382
- "eval_f1": 0.5992935033021043,
383
- "eval_loss": 0.4482789933681488,
384
- "eval_precision": 0.621931781957284,
385
- "eval_recall": 0.578245406046236,
386
- "eval_runtime": 4.5482,
387
- "eval_samples_per_second": 209.315,
388
- "eval_steps_per_second": 13.192,
389
  "step": 2300
390
  },
391
  {
392
  "epoch": 20.0,
393
  "step": 2300,
394
  "total_flos": 1090015519875414.0,
395
- "train_loss": 0.09238651011301123,
396
- "train_runtime": 503.6467,
397
- "train_samples_per_second": 72.948,
398
- "train_steps_per_second": 4.567
399
  }
400
  ],
401
  "logging_steps": 500,
 
1
  {
2
+ "best_metric": 0.5637284701114489,
3
+ "best_model_checkpoint": "/nfs/production/literature/amina-mardiyyah/new_data/OT-Entity-Extraction-Pipeline/model_outputs/hf/CeLLaTe-AL-Test_1.0/base/BiomedNLP_BiomedBERT_base/checkpoint-690",
4
  "epoch": 20.0,
5
  "eval_steps": 500,
6
  "global_step": 2300,
 
10
  "log_history": [
11
  {
12
  "epoch": 1.0,
13
+ "grad_norm": 3.141930341720581,
14
  "learning_rate": 9.652173913043478e-06,
15
+ "loss": 1.0965,
16
  "step": 115
17
  },
18
  {
19
  "epoch": 1.0,
20
+ "eval_accuracy": 0.8700857880175761,
21
+ "eval_f1": 0.2011861313868613,
22
+ "eval_loss": 0.4151777923107147,
23
+ "eval_precision": 0.2413793103448276,
24
+ "eval_recall": 0.1724677356276887,
25
+ "eval_runtime": 4.6552,
26
+ "eval_samples_per_second": 204.501,
27
+ "eval_steps_per_second": 12.889,
28
  "step": 115
29
  },
30
  {
31
  "epoch": 2.0,
32
+ "grad_norm": 5.598133563995361,
33
  "learning_rate": 1.965217391304348e-05,
34
+ "loss": 0.2292,
35
  "step": 230
36
  },
37
  {
38
  "epoch": 2.0,
39
+ "eval_accuracy": 0.9119103526840723,
40
+ "eval_f1": 0.46394984326018807,
41
+ "eval_loss": 0.2859836220741272,
42
+ "eval_precision": 0.43893928820655964,
43
+ "eval_recall": 0.4919827923347673,
44
+ "eval_runtime": 4.7635,
45
+ "eval_samples_per_second": 199.851,
46
+ "eval_steps_per_second": 12.596,
47
  "step": 230
48
  },
49
  {
50
  "epoch": 3.0,
51
+ "grad_norm": 1.903611183166504,
52
  "learning_rate": 1.892753623188406e-05,
53
+ "loss": 0.1037,
54
  "step": 345
55
  },
56
  {
57
  "epoch": 3.0,
58
+ "eval_accuracy": 0.9199544323809081,
59
+ "eval_f1": 0.5066666666666667,
60
+ "eval_loss": 0.2779094874858856,
61
+ "eval_precision": 0.49387300408466395,
62
+ "eval_recall": 0.5201407899882675,
63
+ "eval_runtime": 4.7455,
64
+ "eval_samples_per_second": 200.612,
65
+ "eval_steps_per_second": 12.644,
66
  "step": 345
67
  },
68
  {
69
  "epoch": 4.0,
70
+ "grad_norm": 1.9366143941879272,
71
  "learning_rate": 1.781642512077295e-05,
72
+ "loss": 0.0684,
73
  "step": 460
74
  },
75
  {
76
  "epoch": 4.0,
77
+ "eval_accuracy": 0.9274405412317206,
78
+ "eval_f1": 0.5361736334405145,
79
+ "eval_loss": 0.29712674021720886,
80
+ "eval_precision": 0.5514675485737909,
81
+ "eval_recall": 0.5217051231912397,
82
+ "eval_runtime": 4.733,
83
+ "eval_samples_per_second": 201.143,
84
+ "eval_steps_per_second": 12.677,
85
  "step": 460
86
  },
87
  {
88
  "epoch": 5.0,
89
+ "grad_norm": 1.2101763486862183,
90
  "learning_rate": 1.6705314009661836e-05,
91
+ "loss": 0.0463,
92
  "step": 575
93
  },
94
  {
95
  "epoch": 5.0,
96
+ "eval_accuracy": 0.9276265315137284,
97
+ "eval_f1": 0.5296671490593343,
98
+ "eval_loss": 0.3128739893436432,
99
+ "eval_precision": 0.5618421052631579,
100
+ "eval_recall": 0.5009777082518576,
101
+ "eval_runtime": 4.7226,
102
+ "eval_samples_per_second": 201.582,
103
+ "eval_steps_per_second": 12.705,
104
  "step": 575
105
  },
106
  {
107
  "epoch": 6.0,
108
+ "grad_norm": 2.3417909145355225,
109
  "learning_rate": 1.5594202898550727e-05,
110
+ "loss": 0.0319,
111
  "step": 690
112
  },
113
  {
114
  "epoch": 6.0,
115
+ "eval_accuracy": 0.931113849301374,
116
+ "eval_f1": 0.5637284701114489,
117
+ "eval_loss": 0.3199135661125183,
118
+ "eval_precision": 0.5849453322119428,
119
+ "eval_recall": 0.5439968713335941,
120
+ "eval_runtime": 5.02,
121
+ "eval_samples_per_second": 189.643,
122
+ "eval_steps_per_second": 11.952,
123
  "step": 690
124
  },
125
  {
126
  "epoch": 7.0,
127
+ "grad_norm": 0.4602757394313812,
128
  "learning_rate": 1.4483091787439615e-05,
129
+ "loss": 0.0257,
130
  "step": 805
131
  },
132
  {
133
  "epoch": 7.0,
134
+ "eval_accuracy": 0.9257201311231488,
135
+ "eval_f1": 0.525974025974026,
136
+ "eval_loss": 0.37688568234443665,
137
+ "eval_precision": 0.5466048080978491,
138
+ "eval_recall": 0.5068439577630035,
139
+ "eval_runtime": 4.7781,
140
+ "eval_samples_per_second": 199.241,
141
+ "eval_steps_per_second": 12.557,
142
  "step": 805
143
  },
144
  {
145
  "epoch": 8.0,
146
+ "grad_norm": 1.866155743598938,
147
  "learning_rate": 1.3371980676328504e-05,
148
+ "loss": 0.0223,
149
  "step": 920
150
  },
151
  {
152
  "epoch": 8.0,
153
+ "eval_accuracy": 0.9242089600818357,
154
+ "eval_f1": 0.5371711780404118,
155
+ "eval_loss": 0.356906533241272,
156
+ "eval_precision": 0.523986612123466,
157
+ "eval_recall": 0.5510363707469691,
158
+ "eval_runtime": 4.6714,
159
+ "eval_samples_per_second": 203.793,
160
+ "eval_steps_per_second": 12.844,
161
  "step": 920
162
  },
163
  {
164
  "epoch": 9.0,
165
+ "grad_norm": 1.9551316499710083,
166
  "learning_rate": 1.2260869565217393e-05,
167
+ "loss": 0.0176,
168
  "step": 1035
169
  },
170
  {
171
  "epoch": 9.0,
172
+ "eval_accuracy": 0.9249529212098668,
173
+ "eval_f1": 0.549402390438247,
174
+ "eval_loss": 0.3771890699863434,
175
+ "eval_precision": 0.559886317498985,
176
+ "eval_recall": 0.5393038717246773,
177
+ "eval_runtime": 4.7822,
178
+ "eval_samples_per_second": 199.072,
179
+ "eval_steps_per_second": 12.547,
180
  "step": 1035
181
  },
182
  {
183
  "epoch": 10.0,
184
+ "grad_norm": 0.21579106152057648,
185
  "learning_rate": 1.114975845410628e-05,
186
+ "loss": 0.0129,
187
  "step": 1150
188
  },
189
  {
190
  "epoch": 10.0,
191
+ "eval_accuracy": 0.9298119173273196,
192
+ "eval_f1": 0.5549613784907903,
193
+ "eval_loss": 0.3583073616027832,
194
+ "eval_precision": 0.562199036918138,
195
+ "eval_recall": 0.5479077043410246,
196
+ "eval_runtime": 4.7887,
197
+ "eval_samples_per_second": 198.801,
198
+ "eval_steps_per_second": 12.529,
199
  "step": 1150
200
  },
201
  {
202
  "epoch": 11.0,
203
+ "grad_norm": 0.4364924728870392,
204
  "learning_rate": 1.003864734299517e-05,
205
+ "loss": 0.0112,
206
  "step": 1265
207
  },
208
  {
209
  "epoch": 11.0,
210
+ "eval_accuracy": 0.9262316043986701,
211
+ "eval_f1": 0.5348368908578334,
212
+ "eval_loss": 0.39430442452430725,
213
+ "eval_precision": 0.5512660855126609,
214
+ "eval_recall": 0.5193586233867814,
215
+ "eval_runtime": 4.6852,
216
+ "eval_samples_per_second": 203.191,
217
+ "eval_steps_per_second": 12.806,
218
  "step": 1265
219
  },
220
  {
221
  "epoch": 12.0,
222
+ "grad_norm": 1.8533910512924194,
223
  "learning_rate": 8.927536231884059e-06,
224
+ "loss": 0.0096,
225
  "step": 1380
226
  },
227
  {
228
  "epoch": 12.0,
229
+ "eval_accuracy": 0.9285099853532653,
230
+ "eval_f1": 0.5507480792559645,
231
+ "eval_loss": 0.4048748314380646,
232
+ "eval_precision": 0.5701130179991628,
233
+ "eval_recall": 0.5326554556120454,
234
+ "eval_runtime": 4.9847,
235
+ "eval_samples_per_second": 190.986,
236
+ "eval_steps_per_second": 12.037,
237
  "step": 1380
238
  },
239
  {
240
  "epoch": 13.0,
241
+ "grad_norm": 0.022831812500953674,
242
  "learning_rate": 7.816425120772948e-06,
243
+ "loss": 0.0078,
244
  "step": 1495
245
  },
246
  {
247
  "epoch": 13.0,
248
+ "eval_accuracy": 0.927138307023458,
249
+ "eval_f1": 0.5415120140652471,
250
+ "eval_loss": 0.42278555035591125,
251
+ "eval_precision": 0.5409836065573771,
252
+ "eval_recall": 0.5420414548298788,
253
+ "eval_runtime": 4.7419,
254
+ "eval_samples_per_second": 200.765,
255
+ "eval_steps_per_second": 12.653,
256
  "step": 1495
257
  },
258
  {
259
  "epoch": 14.0,
260
+ "grad_norm": 2.755345582962036,
261
  "learning_rate": 6.705314009661836e-06,
262
+ "loss": 0.0068,
263
  "step": 1610
264
  },
265
  {
266
  "epoch": 14.0,
267
+ "eval_accuracy": 0.9264408434659289,
268
+ "eval_f1": 0.541335453100159,
269
+ "eval_loss": 0.41575905680656433,
270
+ "eval_precision": 0.5503030303030303,
271
+ "eval_recall": 0.5326554556120454,
272
+ "eval_runtime": 4.73,
273
+ "eval_samples_per_second": 201.268,
274
+ "eval_steps_per_second": 12.685,
275
  "step": 1610
276
  },
277
  {
278
  "epoch": 15.0,
279
+ "grad_norm": 0.009739220142364502,
280
  "learning_rate": 5.594202898550725e-06,
281
+ "loss": 0.0057,
282
  "step": 1725
283
  },
284
  {
285
  "epoch": 15.0,
286
+ "eval_accuracy": 0.9306256248111037,
287
+ "eval_f1": 0.5625126646403241,
288
+ "eval_loss": 0.41495904326438904,
289
+ "eval_precision": 0.5836837678721615,
290
+ "eval_recall": 0.5428236214313649,
291
+ "eval_runtime": 4.7302,
292
+ "eval_samples_per_second": 201.262,
293
+ "eval_steps_per_second": 12.685,
294
  "step": 1725
295
  },
296
  {
297
  "epoch": 16.0,
298
+ "grad_norm": 0.20581257343292236,
299
  "learning_rate": 4.483091787439614e-06,
300
+ "loss": 0.0051,
301
  "step": 1840
302
  },
303
  {
304
  "epoch": 16.0,
305
+ "eval_accuracy": 0.929509683119057,
306
+ "eval_f1": 0.5591397849462366,
307
+ "eval_loss": 0.427743524312973,
308
+ "eval_precision": 0.5695740365111562,
309
+ "eval_recall": 0.5490809542432538,
310
+ "eval_runtime": 4.7523,
311
+ "eval_samples_per_second": 200.322,
312
+ "eval_steps_per_second": 12.625,
313
  "step": 1840
314
  },
315
  {
316
  "epoch": 17.0,
317
+ "grad_norm": 1.4635635614395142,
318
  "learning_rate": 3.371980676328503e-06,
319
+ "loss": 0.0048,
320
  "step": 1955
321
  },
322
  {
323
  "epoch": 17.0,
324
+ "eval_accuracy": 0.9296026782600609,
325
+ "eval_f1": 0.5549841772151899,
326
+ "eval_loss": 0.4313228130340576,
327
+ "eval_precision": 0.5614245698279312,
328
+ "eval_recall": 0.5486898709425108,
329
+ "eval_runtime": 4.9134,
330
+ "eval_samples_per_second": 193.757,
331
+ "eval_steps_per_second": 12.212,
332
  "step": 1955
333
  },
334
  {
335
  "epoch": 18.0,
336
+ "grad_norm": 0.020520634949207306,
337
  "learning_rate": 2.2608695652173913e-06,
338
+ "loss": 0.004,
339
  "step": 2070
340
  },
341
  {
342
  "epoch": 18.0,
343
+ "eval_accuracy": 0.9297421709715668,
344
+ "eval_f1": 0.5621019108280255,
345
+ "eval_loss": 0.4391424357891083,
346
+ "eval_precision": 0.5723550871503851,
347
+ "eval_recall": 0.5522096206491983,
348
+ "eval_runtime": 4.7507,
349
+ "eval_samples_per_second": 200.39,
350
+ "eval_steps_per_second": 12.63,
351
  "step": 2070
352
  },
353
  {
354
  "epoch": 19.0,
355
+ "grad_norm": 1.6688083410263062,
356
  "learning_rate": 1.1497584541062802e-06,
357
+ "loss": 0.0036,
358
  "step": 2185
359
  },
360
  {
361
  "epoch": 19.0,
362
+ "eval_accuracy": 0.9293236928370493,
363
+ "eval_f1": 0.5589764094362255,
364
+ "eval_loss": 0.44707345962524414,
365
+ "eval_precision": 0.5717791411042945,
366
+ "eval_recall": 0.5467344544387954,
367
+ "eval_runtime": 4.7037,
368
+ "eval_samples_per_second": 202.393,
369
+ "eval_steps_per_second": 12.756,
370
  "step": 2185
371
  },
372
  {
373
  "epoch": 20.0,
374
+ "grad_norm": 0.006341221742331982,
375
  "learning_rate": 3.864734299516908e-08,
376
+ "loss": 0.0035,
377
  "step": 2300
378
  },
379
  {
380
  "epoch": 20.0,
381
+ "eval_accuracy": 0.9296956734010648,
382
+ "eval_f1": 0.5620888977476579,
383
+ "eval_loss": 0.4408387243747711,
384
+ "eval_precision": 0.573170731707317,
385
+ "eval_recall": 0.5514274540477122,
386
+ "eval_runtime": 4.7859,
387
+ "eval_samples_per_second": 198.919,
388
+ "eval_steps_per_second": 12.537,
389
  "step": 2300
390
  },
391
  {
392
  "epoch": 20.0,
393
  "step": 2300,
394
  "total_flos": 1090015519875414.0,
395
+ "train_loss": 0.08582915891771731,
396
+ "train_runtime": 509.1667,
397
+ "train_samples_per_second": 72.157,
398
+ "train_steps_per_second": 4.517
399
  }
400
  ],
401
  "logging_steps": 500,