andreas122001 commited on
Commit
be6123a
1 Parent(s): 37e2605

Upload 12 files

Browse files
.gitattributes CHANGED
@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bigscience/bloomz-3b",
3
+ "apply_residual_connection_post_layernorm": false,
4
+ "architectures": [
5
+ "BloomForSequenceClassification"
6
+ ],
7
+ "attention_dropout": 0.0,
8
+ "attention_softmax_in_fp32": true,
9
+ "bias_dropout_fusion": true,
10
+ "bos_token_id": 1,
11
+ "eos_token_id": 2,
12
+ "hidden_dropout": 0.0,
13
+ "hidden_size": 2560,
14
+ "id2label": {
15
+ "0": "human-produced",
16
+ "1": "machine-generated"
17
+ },
18
+ "initializer_range": 0.02,
19
+ "label2id": {
20
+ "human-produced": 0,
21
+ "machine-generated": 1
22
+ },
23
+ "layer_norm_epsilon": 1e-05,
24
+ "masked_softmax_fusion": true,
25
+ "model_type": "bloom",
26
+ "n_head": 32,
27
+ "n_inner": null,
28
+ "n_layer": 30,
29
+ "offset_alibi": 100,
30
+ "pad_token_id": 3,
31
+ "pretraining_tp": 4,
32
+ "problem_type": "single_label_classification",
33
+ "seq_length": 2048,
34
+ "skip_bias_add": true,
35
+ "skip_bias_add_qkv": false,
36
+ "slow_but_exact": false,
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.26.1",
39
+ "unk_token_id": 0,
40
+ "use_cache": true,
41
+ "vocab_size": 250880
42
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e25972de8180fc10cad9e151af9cf748eb5e70bce391ee42a5beee72d894aac
3
+ size 24020817451
pytorch_model-00001-of-00002.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9c8f97329473aa11ca607af7aadd2dde0f2e8670980f786a51cd176f2762840
3
+ size 9912311464
pytorch_model-00002-of-00002.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8d463e9eeaf58b97d6a48cb3dbcb3169370ab28f131ea12cc27655cd9c16021
3
+ size 2098070557
pytorch_model.bin.index.json ADDED
@@ -0,0 +1,373 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 12010250240
4
+ },
5
+ "weight_map": {
6
+ "score.weight": "pytorch_model-00002-of-00002.bin",
7
+ "transformer.h.0.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
8
+ "transformer.h.0.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
9
+ "transformer.h.0.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
10
+ "transformer.h.0.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
11
+ "transformer.h.0.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
12
+ "transformer.h.0.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
13
+ "transformer.h.0.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
14
+ "transformer.h.0.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
15
+ "transformer.h.0.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
16
+ "transformer.h.0.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
17
+ "transformer.h.0.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
18
+ "transformer.h.0.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
19
+ "transformer.h.1.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
20
+ "transformer.h.1.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
21
+ "transformer.h.1.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
22
+ "transformer.h.1.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
23
+ "transformer.h.1.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
24
+ "transformer.h.1.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
25
+ "transformer.h.1.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
26
+ "transformer.h.1.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
27
+ "transformer.h.1.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
28
+ "transformer.h.1.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
29
+ "transformer.h.1.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
30
+ "transformer.h.1.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
31
+ "transformer.h.10.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
32
+ "transformer.h.10.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
33
+ "transformer.h.10.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
34
+ "transformer.h.10.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
35
+ "transformer.h.10.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
36
+ "transformer.h.10.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
37
+ "transformer.h.10.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
38
+ "transformer.h.10.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
39
+ "transformer.h.10.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
40
+ "transformer.h.10.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
41
+ "transformer.h.10.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
42
+ "transformer.h.10.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
43
+ "transformer.h.11.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
44
+ "transformer.h.11.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
45
+ "transformer.h.11.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
46
+ "transformer.h.11.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
47
+ "transformer.h.11.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
48
+ "transformer.h.11.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
49
+ "transformer.h.11.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
50
+ "transformer.h.11.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
51
+ "transformer.h.11.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
52
+ "transformer.h.11.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
53
+ "transformer.h.11.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
54
+ "transformer.h.11.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
55
+ "transformer.h.12.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
56
+ "transformer.h.12.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
57
+ "transformer.h.12.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
58
+ "transformer.h.12.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
59
+ "transformer.h.12.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
60
+ "transformer.h.12.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
61
+ "transformer.h.12.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
62
+ "transformer.h.12.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
63
+ "transformer.h.12.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
64
+ "transformer.h.12.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
65
+ "transformer.h.12.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
66
+ "transformer.h.12.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
67
+ "transformer.h.13.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
68
+ "transformer.h.13.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
69
+ "transformer.h.13.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
70
+ "transformer.h.13.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
71
+ "transformer.h.13.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
72
+ "transformer.h.13.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
73
+ "transformer.h.13.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
74
+ "transformer.h.13.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
75
+ "transformer.h.13.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
76
+ "transformer.h.13.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
77
+ "transformer.h.13.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
78
+ "transformer.h.13.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
79
+ "transformer.h.14.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
80
+ "transformer.h.14.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
81
+ "transformer.h.14.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
82
+ "transformer.h.14.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
83
+ "transformer.h.14.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
84
+ "transformer.h.14.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
85
+ "transformer.h.14.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
86
+ "transformer.h.14.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
87
+ "transformer.h.14.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
88
+ "transformer.h.14.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
89
+ "transformer.h.14.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
90
+ "transformer.h.14.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
91
+ "transformer.h.15.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
92
+ "transformer.h.15.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
93
+ "transformer.h.15.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
94
+ "transformer.h.15.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
95
+ "transformer.h.15.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
96
+ "transformer.h.15.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
97
+ "transformer.h.15.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
98
+ "transformer.h.15.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
99
+ "transformer.h.15.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
100
+ "transformer.h.15.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
101
+ "transformer.h.15.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
102
+ "transformer.h.15.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
103
+ "transformer.h.16.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
104
+ "transformer.h.16.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
105
+ "transformer.h.16.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
106
+ "transformer.h.16.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
107
+ "transformer.h.16.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
108
+ "transformer.h.16.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
109
+ "transformer.h.16.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
110
+ "transformer.h.16.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
111
+ "transformer.h.16.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
112
+ "transformer.h.16.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
113
+ "transformer.h.16.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
114
+ "transformer.h.16.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
115
+ "transformer.h.17.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
116
+ "transformer.h.17.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
117
+ "transformer.h.17.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
118
+ "transformer.h.17.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
119
+ "transformer.h.17.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
120
+ "transformer.h.17.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
121
+ "transformer.h.17.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
122
+ "transformer.h.17.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
123
+ "transformer.h.17.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
124
+ "transformer.h.17.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
125
+ "transformer.h.17.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
126
+ "transformer.h.17.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
127
+ "transformer.h.18.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
128
+ "transformer.h.18.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
129
+ "transformer.h.18.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
130
+ "transformer.h.18.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
131
+ "transformer.h.18.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
132
+ "transformer.h.18.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
133
+ "transformer.h.18.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
134
+ "transformer.h.18.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
135
+ "transformer.h.18.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
136
+ "transformer.h.18.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
137
+ "transformer.h.18.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
138
+ "transformer.h.18.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
139
+ "transformer.h.19.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
140
+ "transformer.h.19.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
141
+ "transformer.h.19.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
142
+ "transformer.h.19.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
143
+ "transformer.h.19.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
144
+ "transformer.h.19.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
145
+ "transformer.h.19.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
146
+ "transformer.h.19.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
147
+ "transformer.h.19.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
148
+ "transformer.h.19.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
149
+ "transformer.h.19.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
150
+ "transformer.h.19.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
151
+ "transformer.h.2.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
152
+ "transformer.h.2.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
153
+ "transformer.h.2.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
154
+ "transformer.h.2.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
155
+ "transformer.h.2.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
156
+ "transformer.h.2.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
157
+ "transformer.h.2.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
158
+ "transformer.h.2.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
159
+ "transformer.h.2.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
160
+ "transformer.h.2.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
161
+ "transformer.h.2.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
162
+ "transformer.h.2.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
163
+ "transformer.h.20.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
164
+ "transformer.h.20.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
165
+ "transformer.h.20.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
166
+ "transformer.h.20.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
167
+ "transformer.h.20.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
168
+ "transformer.h.20.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
169
+ "transformer.h.20.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
170
+ "transformer.h.20.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
171
+ "transformer.h.20.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
172
+ "transformer.h.20.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
173
+ "transformer.h.20.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
174
+ "transformer.h.20.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
175
+ "transformer.h.21.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
176
+ "transformer.h.21.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
177
+ "transformer.h.21.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
178
+ "transformer.h.21.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
179
+ "transformer.h.21.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
180
+ "transformer.h.21.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
181
+ "transformer.h.21.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
182
+ "transformer.h.21.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
183
+ "transformer.h.21.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
184
+ "transformer.h.21.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
185
+ "transformer.h.21.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
186
+ "transformer.h.21.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
187
+ "transformer.h.22.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
188
+ "transformer.h.22.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
189
+ "transformer.h.22.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
190
+ "transformer.h.22.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
191
+ "transformer.h.22.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
192
+ "transformer.h.22.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
193
+ "transformer.h.22.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
194
+ "transformer.h.22.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
195
+ "transformer.h.22.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
196
+ "transformer.h.22.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
197
+ "transformer.h.22.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
198
+ "transformer.h.22.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
199
+ "transformer.h.23.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
200
+ "transformer.h.23.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
201
+ "transformer.h.23.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
202
+ "transformer.h.23.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
203
+ "transformer.h.23.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
204
+ "transformer.h.23.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
205
+ "transformer.h.23.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
206
+ "transformer.h.23.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
207
+ "transformer.h.23.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
208
+ "transformer.h.23.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
209
+ "transformer.h.23.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
210
+ "transformer.h.23.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
211
+ "transformer.h.24.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
212
+ "transformer.h.24.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
213
+ "transformer.h.24.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
214
+ "transformer.h.24.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
215
+ "transformer.h.24.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
216
+ "transformer.h.24.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
217
+ "transformer.h.24.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
218
+ "transformer.h.24.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
219
+ "transformer.h.24.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
220
+ "transformer.h.24.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
221
+ "transformer.h.24.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
222
+ "transformer.h.24.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
223
+ "transformer.h.25.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
224
+ "transformer.h.25.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
225
+ "transformer.h.25.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
226
+ "transformer.h.25.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
227
+ "transformer.h.25.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
228
+ "transformer.h.25.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
229
+ "transformer.h.25.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
230
+ "transformer.h.25.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
231
+ "transformer.h.25.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
232
+ "transformer.h.25.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
233
+ "transformer.h.25.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
234
+ "transformer.h.25.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
235
+ "transformer.h.26.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
236
+ "transformer.h.26.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
237
+ "transformer.h.26.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
238
+ "transformer.h.26.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
239
+ "transformer.h.26.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
240
+ "transformer.h.26.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
241
+ "transformer.h.26.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
242
+ "transformer.h.26.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
243
+ "transformer.h.26.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
244
+ "transformer.h.26.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
245
+ "transformer.h.26.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
246
+ "transformer.h.26.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
247
+ "transformer.h.27.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
248
+ "transformer.h.27.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
249
+ "transformer.h.27.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
250
+ "transformer.h.27.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
251
+ "transformer.h.27.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
252
+ "transformer.h.27.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
253
+ "transformer.h.27.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
254
+ "transformer.h.27.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
255
+ "transformer.h.27.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
256
+ "transformer.h.27.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
257
+ "transformer.h.27.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
258
+ "transformer.h.27.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
259
+ "transformer.h.28.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
260
+ "transformer.h.28.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
261
+ "transformer.h.28.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
262
+ "transformer.h.28.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
263
+ "transformer.h.28.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
264
+ "transformer.h.28.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
265
+ "transformer.h.28.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
266
+ "transformer.h.28.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
267
+ "transformer.h.28.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
268
+ "transformer.h.28.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
269
+ "transformer.h.28.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
270
+ "transformer.h.28.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
271
+ "transformer.h.29.input_layernorm.bias": "pytorch_model-00002-of-00002.bin",
272
+ "transformer.h.29.input_layernorm.weight": "pytorch_model-00002-of-00002.bin",
273
+ "transformer.h.29.mlp.dense_4h_to_h.bias": "pytorch_model-00002-of-00002.bin",
274
+ "transformer.h.29.mlp.dense_4h_to_h.weight": "pytorch_model-00002-of-00002.bin",
275
+ "transformer.h.29.mlp.dense_h_to_4h.bias": "pytorch_model-00002-of-00002.bin",
276
+ "transformer.h.29.mlp.dense_h_to_4h.weight": "pytorch_model-00002-of-00002.bin",
277
+ "transformer.h.29.post_attention_layernorm.bias": "pytorch_model-00002-of-00002.bin",
278
+ "transformer.h.29.post_attention_layernorm.weight": "pytorch_model-00002-of-00002.bin",
279
+ "transformer.h.29.self_attention.dense.bias": "pytorch_model-00002-of-00002.bin",
280
+ "transformer.h.29.self_attention.dense.weight": "pytorch_model-00002-of-00002.bin",
281
+ "transformer.h.29.self_attention.query_key_value.bias": "pytorch_model-00002-of-00002.bin",
282
+ "transformer.h.29.self_attention.query_key_value.weight": "pytorch_model-00002-of-00002.bin",
283
+ "transformer.h.3.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
284
+ "transformer.h.3.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
285
+ "transformer.h.3.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
286
+ "transformer.h.3.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
287
+ "transformer.h.3.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
288
+ "transformer.h.3.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
289
+ "transformer.h.3.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
290
+ "transformer.h.3.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
291
+ "transformer.h.3.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
292
+ "transformer.h.3.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
293
+ "transformer.h.3.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
294
+ "transformer.h.3.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
295
+ "transformer.h.4.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
296
+ "transformer.h.4.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
297
+ "transformer.h.4.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
298
+ "transformer.h.4.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
299
+ "transformer.h.4.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
300
+ "transformer.h.4.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
301
+ "transformer.h.4.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
302
+ "transformer.h.4.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
303
+ "transformer.h.4.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
304
+ "transformer.h.4.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
305
+ "transformer.h.4.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
306
+ "transformer.h.4.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
307
+ "transformer.h.5.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
308
+ "transformer.h.5.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
309
+ "transformer.h.5.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
310
+ "transformer.h.5.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
311
+ "transformer.h.5.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
312
+ "transformer.h.5.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
313
+ "transformer.h.5.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
314
+ "transformer.h.5.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
315
+ "transformer.h.5.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
316
+ "transformer.h.5.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
317
+ "transformer.h.5.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
318
+ "transformer.h.5.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
319
+ "transformer.h.6.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
320
+ "transformer.h.6.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
321
+ "transformer.h.6.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
322
+ "transformer.h.6.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
323
+ "transformer.h.6.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
324
+ "transformer.h.6.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
325
+ "transformer.h.6.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
326
+ "transformer.h.6.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
327
+ "transformer.h.6.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
328
+ "transformer.h.6.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
329
+ "transformer.h.6.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
330
+ "transformer.h.6.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
331
+ "transformer.h.7.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
332
+ "transformer.h.7.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
333
+ "transformer.h.7.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
334
+ "transformer.h.7.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
335
+ "transformer.h.7.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
336
+ "transformer.h.7.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
337
+ "transformer.h.7.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
338
+ "transformer.h.7.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
339
+ "transformer.h.7.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
340
+ "transformer.h.7.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
341
+ "transformer.h.7.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
342
+ "transformer.h.7.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
343
+ "transformer.h.8.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
344
+ "transformer.h.8.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
345
+ "transformer.h.8.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
346
+ "transformer.h.8.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
347
+ "transformer.h.8.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
348
+ "transformer.h.8.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
349
+ "transformer.h.8.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
350
+ "transformer.h.8.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
351
+ "transformer.h.8.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
352
+ "transformer.h.8.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
353
+ "transformer.h.8.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
354
+ "transformer.h.8.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
355
+ "transformer.h.9.input_layernorm.bias": "pytorch_model-00001-of-00002.bin",
356
+ "transformer.h.9.input_layernorm.weight": "pytorch_model-00001-of-00002.bin",
357
+ "transformer.h.9.mlp.dense_4h_to_h.bias": "pytorch_model-00001-of-00002.bin",
358
+ "transformer.h.9.mlp.dense_4h_to_h.weight": "pytorch_model-00001-of-00002.bin",
359
+ "transformer.h.9.mlp.dense_h_to_4h.bias": "pytorch_model-00001-of-00002.bin",
360
+ "transformer.h.9.mlp.dense_h_to_4h.weight": "pytorch_model-00001-of-00002.bin",
361
+ "transformer.h.9.post_attention_layernorm.bias": "pytorch_model-00001-of-00002.bin",
362
+ "transformer.h.9.post_attention_layernorm.weight": "pytorch_model-00001-of-00002.bin",
363
+ "transformer.h.9.self_attention.dense.bias": "pytorch_model-00001-of-00002.bin",
364
+ "transformer.h.9.self_attention.dense.weight": "pytorch_model-00001-of-00002.bin",
365
+ "transformer.h.9.self_attention.query_key_value.bias": "pytorch_model-00001-of-00002.bin",
366
+ "transformer.h.9.self_attention.query_key_value.weight": "pytorch_model-00001-of-00002.bin",
367
+ "transformer.ln_f.bias": "pytorch_model-00002-of-00002.bin",
368
+ "transformer.ln_f.weight": "pytorch_model-00002-of-00002.bin",
369
+ "transformer.word_embeddings.weight": "pytorch_model-00001-of-00002.bin",
370
+ "transformer.word_embeddings_layernorm.bias": "pytorch_model-00001-of-00002.bin",
371
+ "transformer.word_embeddings_layernorm.weight": "pytorch_model-00001-of-00002.bin"
372
+ }
373
+ }
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d944d6907131fce4941c1b98753b6c6304e9268438faa750202a7ed282544a4
3
+ size 14575
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19c575370b6179b490f435df9c74d0ea7aabd16f264cf932690c39af9fca6ebb
3
+ size 627
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "pad_token": "<pad>",
5
+ "unk_token": "<unk>"
6
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3955d17bdf8285663734ab4418c7a0d6397bf2fa7d7a2f8d820f23b3130cd48
3
+ size 14500708
tokenizer_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "model_max_length": 1000000000000000019884624838656,
6
+ "name_or_path": "bigscience/bloomz-3b",
7
+ "pad_token": "<pad>",
8
+ "special_tokens_map_file": null,
9
+ "tokenizer_class": "BloomTokenizer",
10
+ "unk_token": "<unk>"
11
+ }
trainer_state.json ADDED
@@ -0,0 +1,922 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.0,
5
+ "global_step": 1750,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 4.9971428571428576e-05,
13
+ "loss": 8.3826,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.02,
18
+ "learning_rate": 4.9e-05,
19
+ "loss": 3.6284,
20
+ "step": 35
21
+ },
22
+ {
23
+ "epoch": 0.02,
24
+ "eval_accuracy": 0.656,
25
+ "eval_f1": 0.47560975609756095,
26
+ "eval_loss": 1.8382066488265991,
27
+ "eval_precision": 1.0,
28
+ "eval_recall": 0.312,
29
+ "eval_runtime": 462.8972,
30
+ "eval_samples_per_second": 6.481,
31
+ "eval_steps_per_second": 0.81,
32
+ "step": 35
33
+ },
34
+ {
35
+ "epoch": 0.04,
36
+ "learning_rate": 4.8e-05,
37
+ "loss": 0.8755,
38
+ "step": 70
39
+ },
40
+ {
41
+ "epoch": 0.04,
42
+ "eval_accuracy": 0.891,
43
+ "eval_f1": 0.8799118619170032,
44
+ "eval_loss": 0.9740824103355408,
45
+ "eval_precision": 0.9795584627964022,
46
+ "eval_recall": 0.7986666666666666,
47
+ "eval_runtime": 462.3762,
48
+ "eval_samples_per_second": 6.488,
49
+ "eval_steps_per_second": 0.811,
50
+ "step": 70
51
+ },
52
+ {
53
+ "epoch": 0.06,
54
+ "learning_rate": 4.7e-05,
55
+ "loss": 0.5529,
56
+ "step": 105
57
+ },
58
+ {
59
+ "epoch": 0.06,
60
+ "eval_accuracy": 0.85,
61
+ "eval_f1": 0.823943661971831,
62
+ "eval_loss": 0.8984822630882263,
63
+ "eval_precision": 0.9971590909090909,
64
+ "eval_recall": 0.702,
65
+ "eval_runtime": 463.7211,
66
+ "eval_samples_per_second": 6.469,
67
+ "eval_steps_per_second": 0.809,
68
+ "step": 105
69
+ },
70
+ {
71
+ "epoch": 0.08,
72
+ "learning_rate": 4.600000000000001e-05,
73
+ "loss": 0.4281,
74
+ "step": 140
75
+ },
76
+ {
77
+ "epoch": 0.08,
78
+ "eval_accuracy": 0.948,
79
+ "eval_f1": 0.948,
80
+ "eval_loss": 0.28269827365875244,
81
+ "eval_precision": 0.948,
82
+ "eval_recall": 0.948,
83
+ "eval_runtime": 462.5168,
84
+ "eval_samples_per_second": 6.486,
85
+ "eval_steps_per_second": 0.811,
86
+ "step": 140
87
+ },
88
+ {
89
+ "epoch": 0.1,
90
+ "learning_rate": 4.5e-05,
91
+ "loss": 0.4027,
92
+ "step": 175
93
+ },
94
+ {
95
+ "epoch": 0.1,
96
+ "eval_accuracy": 0.8686666666666667,
97
+ "eval_f1": 0.8490421455938698,
98
+ "eval_loss": 1.2850149869918823,
99
+ "eval_precision": 0.9981981981981982,
100
+ "eval_recall": 0.7386666666666667,
101
+ "eval_runtime": 462.9807,
102
+ "eval_samples_per_second": 6.48,
103
+ "eval_steps_per_second": 0.81,
104
+ "step": 175
105
+ },
106
+ {
107
+ "epoch": 0.12,
108
+ "learning_rate": 4.4000000000000006e-05,
109
+ "loss": 0.4487,
110
+ "step": 210
111
+ },
112
+ {
113
+ "epoch": 0.12,
114
+ "eval_accuracy": 0.9423333333333334,
115
+ "eval_f1": 0.9391059486096445,
116
+ "eval_loss": 0.1546011120080948,
117
+ "eval_precision": 0.9947800149142431,
118
+ "eval_recall": 0.8893333333333333,
119
+ "eval_runtime": 462.5947,
120
+ "eval_samples_per_second": 6.485,
121
+ "eval_steps_per_second": 0.811,
122
+ "step": 210
123
+ },
124
+ {
125
+ "epoch": 0.14,
126
+ "learning_rate": 4.3e-05,
127
+ "loss": 0.4095,
128
+ "step": 245
129
+ },
130
+ {
131
+ "epoch": 0.14,
132
+ "eval_accuracy": 0.8943333333333333,
133
+ "eval_f1": 0.8818486768542676,
134
+ "eval_loss": 0.8063055276870728,
135
+ "eval_precision": 1.0,
136
+ "eval_recall": 0.7886666666666666,
137
+ "eval_runtime": 463.8636,
138
+ "eval_samples_per_second": 6.467,
139
+ "eval_steps_per_second": 0.808,
140
+ "step": 245
141
+ },
142
+ {
143
+ "epoch": 0.16,
144
+ "learning_rate": 4.2e-05,
145
+ "loss": 0.2566,
146
+ "step": 280
147
+ },
148
+ {
149
+ "epoch": 0.16,
150
+ "eval_accuracy": 0.9623333333333334,
151
+ "eval_f1": 0.9620932572962093,
152
+ "eval_loss": 0.37129560112953186,
153
+ "eval_precision": 0.9682646860229575,
154
+ "eval_recall": 0.956,
155
+ "eval_runtime": 462.5562,
156
+ "eval_samples_per_second": 6.486,
157
+ "eval_steps_per_second": 0.811,
158
+ "step": 280
159
+ },
160
+ {
161
+ "epoch": 0.18,
162
+ "learning_rate": 4.1e-05,
163
+ "loss": 0.2851,
164
+ "step": 315
165
+ },
166
+ {
167
+ "epoch": 0.18,
168
+ "eval_accuracy": 0.9386666666666666,
169
+ "eval_f1": 0.9357990230286113,
170
+ "eval_loss": 0.3722746670246124,
171
+ "eval_precision": 0.9816983894582724,
172
+ "eval_recall": 0.894,
173
+ "eval_runtime": 462.4018,
174
+ "eval_samples_per_second": 6.488,
175
+ "eval_steps_per_second": 0.811,
176
+ "step": 315
177
+ },
178
+ {
179
+ "epoch": 0.2,
180
+ "learning_rate": 4e-05,
181
+ "loss": 0.4674,
182
+ "step": 350
183
+ },
184
+ {
185
+ "epoch": 0.2,
186
+ "eval_accuracy": 0.8966666666666666,
187
+ "eval_f1": 0.8849294729027467,
188
+ "eval_loss": 0.9987091422080994,
189
+ "eval_precision": 0.998324958123953,
190
+ "eval_recall": 0.7946666666666666,
191
+ "eval_runtime": 462.5055,
192
+ "eval_samples_per_second": 6.486,
193
+ "eval_steps_per_second": 0.811,
194
+ "step": 350
195
+ },
196
+ {
197
+ "epoch": 0.22,
198
+ "learning_rate": 3.9000000000000006e-05,
199
+ "loss": 0.1749,
200
+ "step": 385
201
+ },
202
+ {
203
+ "epoch": 0.22,
204
+ "eval_accuracy": 0.9473333333333334,
205
+ "eval_f1": 0.9446002805049089,
206
+ "eval_loss": 0.4473351240158081,
207
+ "eval_precision": 0.996301775147929,
208
+ "eval_recall": 0.898,
209
+ "eval_runtime": 462.5402,
210
+ "eval_samples_per_second": 6.486,
211
+ "eval_steps_per_second": 0.811,
212
+ "step": 385
213
+ },
214
+ {
215
+ "epoch": 0.24,
216
+ "learning_rate": 3.8e-05,
217
+ "loss": 0.3134,
218
+ "step": 420
219
+ },
220
+ {
221
+ "epoch": 0.24,
222
+ "eval_accuracy": 0.9313333333333333,
223
+ "eval_f1": 0.9263759828448893,
224
+ "eval_loss": 0.8257947564125061,
225
+ "eval_precision": 0.9984591679506933,
226
+ "eval_recall": 0.864,
227
+ "eval_runtime": 462.4771,
228
+ "eval_samples_per_second": 6.487,
229
+ "eval_steps_per_second": 0.811,
230
+ "step": 420
231
+ },
232
+ {
233
+ "epoch": 0.26,
234
+ "learning_rate": 3.7e-05,
235
+ "loss": 0.4379,
236
+ "step": 455
237
+ },
238
+ {
239
+ "epoch": 0.26,
240
+ "eval_accuracy": 0.9496666666666667,
241
+ "eval_f1": 0.9478050466643623,
242
+ "eval_loss": 0.34615999460220337,
243
+ "eval_precision": 0.9842067480258435,
244
+ "eval_recall": 0.914,
245
+ "eval_runtime": 462.467,
246
+ "eval_samples_per_second": 6.487,
247
+ "eval_steps_per_second": 0.811,
248
+ "step": 455
249
+ },
250
+ {
251
+ "epoch": 0.28,
252
+ "learning_rate": 3.6e-05,
253
+ "loss": 0.1028,
254
+ "step": 490
255
+ },
256
+ {
257
+ "epoch": 0.28,
258
+ "eval_accuracy": 0.9523333333333334,
259
+ "eval_f1": 0.9523174391463821,
260
+ "eval_loss": 0.3040553629398346,
261
+ "eval_precision": 0.95263509006004,
262
+ "eval_recall": 0.952,
263
+ "eval_runtime": 462.402,
264
+ "eval_samples_per_second": 6.488,
265
+ "eval_steps_per_second": 0.811,
266
+ "step": 490
267
+ },
268
+ {
269
+ "epoch": 0.3,
270
+ "learning_rate": 3.5e-05,
271
+ "loss": 0.1392,
272
+ "step": 525
273
+ },
274
+ {
275
+ "epoch": 0.3,
276
+ "eval_accuracy": 0.9576666666666667,
277
+ "eval_f1": 0.9566996249573816,
278
+ "eval_loss": 0.3112805485725403,
279
+ "eval_precision": 0.9790648988136776,
280
+ "eval_recall": 0.9353333333333333,
281
+ "eval_runtime": 463.4665,
282
+ "eval_samples_per_second": 6.473,
283
+ "eval_steps_per_second": 0.809,
284
+ "step": 525
285
+ },
286
+ {
287
+ "epoch": 0.32,
288
+ "learning_rate": 3.4000000000000007e-05,
289
+ "loss": 0.0944,
290
+ "step": 560
291
+ },
292
+ {
293
+ "epoch": 0.32,
294
+ "eval_accuracy": 0.9196666666666666,
295
+ "eval_f1": 0.9127759681505611,
296
+ "eval_loss": 0.9487587213516235,
297
+ "eval_precision": 0.9984164687252574,
298
+ "eval_recall": 0.8406666666666667,
299
+ "eval_runtime": 463.9336,
300
+ "eval_samples_per_second": 6.466,
301
+ "eval_steps_per_second": 0.808,
302
+ "step": 560
303
+ },
304
+ {
305
+ "epoch": 0.34,
306
+ "learning_rate": 3.3e-05,
307
+ "loss": 0.1758,
308
+ "step": 595
309
+ },
310
+ {
311
+ "epoch": 0.34,
312
+ "eval_accuracy": 0.935,
313
+ "eval_f1": 0.9306296691568836,
314
+ "eval_loss": 0.4119901657104492,
315
+ "eval_precision": 0.9977116704805492,
316
+ "eval_recall": 0.872,
317
+ "eval_runtime": 462.719,
318
+ "eval_samples_per_second": 6.483,
319
+ "eval_steps_per_second": 0.81,
320
+ "step": 595
321
+ },
322
+ {
323
+ "epoch": 0.36,
324
+ "learning_rate": 3.2000000000000005e-05,
325
+ "loss": 0.1123,
326
+ "step": 630
327
+ },
328
+ {
329
+ "epoch": 0.36,
330
+ "eval_accuracy": 0.9673333333333334,
331
+ "eval_f1": 0.9671361502347419,
332
+ "eval_loss": 0.18079321086406708,
333
+ "eval_precision": 0.9730094466936572,
334
+ "eval_recall": 0.9613333333333334,
335
+ "eval_runtime": 462.4961,
336
+ "eval_samples_per_second": 6.487,
337
+ "eval_steps_per_second": 0.811,
338
+ "step": 630
339
+ },
340
+ {
341
+ "epoch": 0.38,
342
+ "learning_rate": 3.1e-05,
343
+ "loss": 0.187,
344
+ "step": 665
345
+ },
346
+ {
347
+ "epoch": 0.38,
348
+ "eval_accuracy": 0.9563333333333334,
349
+ "eval_f1": 0.954936360509116,
350
+ "eval_loss": 0.3595654368400574,
351
+ "eval_precision": 0.9864960909737029,
352
+ "eval_recall": 0.9253333333333333,
353
+ "eval_runtime": 463.4737,
354
+ "eval_samples_per_second": 6.473,
355
+ "eval_steps_per_second": 0.809,
356
+ "step": 665
357
+ },
358
+ {
359
+ "epoch": 0.4,
360
+ "learning_rate": 3e-05,
361
+ "loss": 0.1876,
362
+ "step": 700
363
+ },
364
+ {
365
+ "epoch": 0.4,
366
+ "eval_accuracy": 0.962,
367
+ "eval_f1": 0.9621262458471761,
368
+ "eval_loss": 0.26878979802131653,
369
+ "eval_precision": 0.9589403973509933,
370
+ "eval_recall": 0.9653333333333334,
371
+ "eval_runtime": 462.4658,
372
+ "eval_samples_per_second": 6.487,
373
+ "eval_steps_per_second": 0.811,
374
+ "step": 700
375
+ },
376
+ {
377
+ "epoch": 0.42,
378
+ "learning_rate": 2.9e-05,
379
+ "loss": 0.1762,
380
+ "step": 735
381
+ },
382
+ {
383
+ "epoch": 0.42,
384
+ "eval_accuracy": 0.9513333333333334,
385
+ "eval_f1": 0.949375866851595,
386
+ "eval_loss": 0.30938494205474854,
387
+ "eval_precision": 0.9891618497109826,
388
+ "eval_recall": 0.9126666666666666,
389
+ "eval_runtime": 462.6352,
390
+ "eval_samples_per_second": 6.485,
391
+ "eval_steps_per_second": 0.811,
392
+ "step": 735
393
+ },
394
+ {
395
+ "epoch": 0.44,
396
+ "learning_rate": 2.8000000000000003e-05,
397
+ "loss": 0.0164,
398
+ "step": 770
399
+ },
400
+ {
401
+ "epoch": 0.44,
402
+ "eval_accuracy": 0.9543333333333334,
403
+ "eval_f1": 0.9526115530958146,
404
+ "eval_loss": 0.3230161964893341,
405
+ "eval_precision": 0.9899352983465133,
406
+ "eval_recall": 0.918,
407
+ "eval_runtime": 464.4305,
408
+ "eval_samples_per_second": 6.46,
409
+ "eval_steps_per_second": 0.807,
410
+ "step": 770
411
+ },
412
+ {
413
+ "epoch": 0.46,
414
+ "learning_rate": 2.7000000000000002e-05,
415
+ "loss": 0.0903,
416
+ "step": 805
417
+ },
418
+ {
419
+ "epoch": 0.46,
420
+ "eval_accuracy": 0.9593333333333334,
421
+ "eval_f1": 0.9585597826086957,
422
+ "eval_loss": 0.33148592710494995,
423
+ "eval_precision": 0.9771468144044322,
424
+ "eval_recall": 0.9406666666666667,
425
+ "eval_runtime": 462.7729,
426
+ "eval_samples_per_second": 6.483,
427
+ "eval_steps_per_second": 0.81,
428
+ "step": 805
429
+ },
430
+ {
431
+ "epoch": 0.48,
432
+ "learning_rate": 2.6000000000000002e-05,
433
+ "loss": 0.2401,
434
+ "step": 840
435
+ },
436
+ {
437
+ "epoch": 0.48,
438
+ "eval_accuracy": 0.9326666666666666,
439
+ "eval_f1": 0.9285208775654635,
440
+ "eval_loss": 0.5261781215667725,
441
+ "eval_precision": 0.9894419306184012,
442
+ "eval_recall": 0.8746666666666667,
443
+ "eval_runtime": 462.0723,
444
+ "eval_samples_per_second": 6.492,
445
+ "eval_steps_per_second": 0.812,
446
+ "step": 840
447
+ },
448
+ {
449
+ "epoch": 0.5,
450
+ "learning_rate": 2.5e-05,
451
+ "loss": 0.1308,
452
+ "step": 875
453
+ },
454
+ {
455
+ "epoch": 0.5,
456
+ "eval_accuracy": 0.922,
457
+ "eval_f1": 0.9155844155844155,
458
+ "eval_loss": 0.5081947445869446,
459
+ "eval_precision": 0.9976415094339622,
460
+ "eval_recall": 0.846,
461
+ "eval_runtime": 462.3765,
462
+ "eval_samples_per_second": 6.488,
463
+ "eval_steps_per_second": 0.811,
464
+ "step": 875
465
+ },
466
+ {
467
+ "epoch": 0.52,
468
+ "learning_rate": 2.4e-05,
469
+ "loss": 0.1004,
470
+ "step": 910
471
+ },
472
+ {
473
+ "epoch": 0.52,
474
+ "eval_accuracy": 0.9353333333333333,
475
+ "eval_f1": 0.9321203638908327,
476
+ "eval_loss": 0.6189997792243958,
477
+ "eval_precision": 0.9808541973490427,
478
+ "eval_recall": 0.888,
479
+ "eval_runtime": 462.3735,
480
+ "eval_samples_per_second": 6.488,
481
+ "eval_steps_per_second": 0.811,
482
+ "step": 910
483
+ },
484
+ {
485
+ "epoch": 0.54,
486
+ "learning_rate": 2.3000000000000003e-05,
487
+ "loss": 0.0982,
488
+ "step": 945
489
+ },
490
+ {
491
+ "epoch": 0.54,
492
+ "eval_accuracy": 0.9496666666666667,
493
+ "eval_f1": 0.9480921278789962,
494
+ "eval_loss": 0.5868619680404663,
495
+ "eval_precision": 0.978708303761533,
496
+ "eval_recall": 0.9193333333333333,
497
+ "eval_runtime": 462.6069,
498
+ "eval_samples_per_second": 6.485,
499
+ "eval_steps_per_second": 0.811,
500
+ "step": 945
501
+ },
502
+ {
503
+ "epoch": 0.56,
504
+ "learning_rate": 2.2000000000000003e-05,
505
+ "loss": 0.1627,
506
+ "step": 980
507
+ },
508
+ {
509
+ "epoch": 0.56,
510
+ "eval_accuracy": 0.9503333333333334,
511
+ "eval_f1": 0.9483177245924385,
512
+ "eval_loss": 0.5452014803886414,
513
+ "eval_precision": 0.9884309472161966,
514
+ "eval_recall": 0.9113333333333333,
515
+ "eval_runtime": 462.2788,
516
+ "eval_samples_per_second": 6.49,
517
+ "eval_steps_per_second": 0.811,
518
+ "step": 980
519
+ },
520
+ {
521
+ "epoch": 0.58,
522
+ "learning_rate": 2.1e-05,
523
+ "loss": 0.2264,
524
+ "step": 1015
525
+ },
526
+ {
527
+ "epoch": 0.58,
528
+ "eval_accuracy": 0.962,
529
+ "eval_f1": 0.9608785175017158,
530
+ "eval_loss": 0.36828112602233887,
531
+ "eval_precision": 0.9900990099009901,
532
+ "eval_recall": 0.9333333333333333,
533
+ "eval_runtime": 462.546,
534
+ "eval_samples_per_second": 6.486,
535
+ "eval_steps_per_second": 0.811,
536
+ "step": 1015
537
+ },
538
+ {
539
+ "epoch": 0.6,
540
+ "learning_rate": 2e-05,
541
+ "loss": 0.2683,
542
+ "step": 1050
543
+ },
544
+ {
545
+ "epoch": 0.6,
546
+ "eval_accuracy": 0.959,
547
+ "eval_f1": 0.9577464788732394,
548
+ "eval_loss": 0.2027631402015686,
549
+ "eval_precision": 0.9879518072289156,
550
+ "eval_recall": 0.9293333333333333,
551
+ "eval_runtime": 462.4537,
552
+ "eval_samples_per_second": 6.487,
553
+ "eval_steps_per_second": 0.811,
554
+ "step": 1050
555
+ },
556
+ {
557
+ "epoch": 0.62,
558
+ "learning_rate": 1.9e-05,
559
+ "loss": 0.088,
560
+ "step": 1085
561
+ },
562
+ {
563
+ "epoch": 0.62,
564
+ "eval_accuracy": 0.937,
565
+ "eval_f1": 0.9328596802841918,
566
+ "eval_loss": 0.5130247473716736,
567
+ "eval_precision": 0.9984790874524715,
568
+ "eval_recall": 0.8753333333333333,
569
+ "eval_runtime": 462.3506,
570
+ "eval_samples_per_second": 6.489,
571
+ "eval_steps_per_second": 0.811,
572
+ "step": 1085
573
+ },
574
+ {
575
+ "epoch": 0.64,
576
+ "learning_rate": 1.8e-05,
577
+ "loss": 0.214,
578
+ "step": 1120
579
+ },
580
+ {
581
+ "epoch": 0.64,
582
+ "eval_accuracy": 0.9443333333333334,
583
+ "eval_f1": 0.9413829413829414,
584
+ "eval_loss": 0.38535651564598083,
585
+ "eval_precision": 0.994069681245367,
586
+ "eval_recall": 0.894,
587
+ "eval_runtime": 462.2352,
588
+ "eval_samples_per_second": 6.49,
589
+ "eval_steps_per_second": 0.811,
590
+ "step": 1120
591
+ },
592
+ {
593
+ "epoch": 0.66,
594
+ "learning_rate": 1.7000000000000003e-05,
595
+ "loss": 0.0811,
596
+ "step": 1155
597
+ },
598
+ {
599
+ "epoch": 0.66,
600
+ "eval_accuracy": 0.9563333333333334,
601
+ "eval_f1": 0.954936360509116,
602
+ "eval_loss": 0.2761794328689575,
603
+ "eval_precision": 0.9864960909737029,
604
+ "eval_recall": 0.9253333333333333,
605
+ "eval_runtime": 463.2198,
606
+ "eval_samples_per_second": 6.476,
607
+ "eval_steps_per_second": 0.81,
608
+ "step": 1155
609
+ },
610
+ {
611
+ "epoch": 0.68,
612
+ "learning_rate": 1.6000000000000003e-05,
613
+ "loss": 0.0716,
614
+ "step": 1190
615
+ },
616
+ {
617
+ "epoch": 0.68,
618
+ "eval_accuracy": 0.9603333333333334,
619
+ "eval_f1": 0.9593717992488903,
620
+ "eval_loss": 0.3206270635128021,
621
+ "eval_precision": 0.9832050384884534,
622
+ "eval_recall": 0.9366666666666666,
623
+ "eval_runtime": 461.5741,
624
+ "eval_samples_per_second": 6.499,
625
+ "eval_steps_per_second": 0.812,
626
+ "step": 1190
627
+ },
628
+ {
629
+ "epoch": 0.7,
630
+ "learning_rate": 1.5e-05,
631
+ "loss": 0.0843,
632
+ "step": 1225
633
+ },
634
+ {
635
+ "epoch": 0.7,
636
+ "eval_accuracy": 0.9546666666666667,
637
+ "eval_f1": 0.953103448275862,
638
+ "eval_loss": 0.38527336716651917,
639
+ "eval_precision": 0.9871428571428571,
640
+ "eval_recall": 0.9213333333333333,
641
+ "eval_runtime": 462.2164,
642
+ "eval_samples_per_second": 6.49,
643
+ "eval_steps_per_second": 0.811,
644
+ "step": 1225
645
+ },
646
+ {
647
+ "epoch": 0.72,
648
+ "learning_rate": 1.4000000000000001e-05,
649
+ "loss": 0.0744,
650
+ "step": 1260
651
+ },
652
+ {
653
+ "epoch": 0.72,
654
+ "eval_accuracy": 0.9556666666666667,
655
+ "eval_f1": 0.9542168674698795,
656
+ "eval_loss": 0.4054282307624817,
657
+ "eval_precision": 0.9864768683274021,
658
+ "eval_recall": 0.924,
659
+ "eval_runtime": 461.3748,
660
+ "eval_samples_per_second": 6.502,
661
+ "eval_steps_per_second": 0.813,
662
+ "step": 1260
663
+ },
664
+ {
665
+ "epoch": 0.74,
666
+ "learning_rate": 1.3000000000000001e-05,
667
+ "loss": 0.0748,
668
+ "step": 1295
669
+ },
670
+ {
671
+ "epoch": 0.74,
672
+ "eval_accuracy": 0.9476666666666667,
673
+ "eval_f1": 0.9450472523626182,
674
+ "eval_loss": 0.4932408928871155,
675
+ "eval_precision": 0.9948415622697127,
676
+ "eval_recall": 0.9,
677
+ "eval_runtime": 462.1327,
678
+ "eval_samples_per_second": 6.492,
679
+ "eval_steps_per_second": 0.811,
680
+ "step": 1295
681
+ },
682
+ {
683
+ "epoch": 0.76,
684
+ "learning_rate": 1.2e-05,
685
+ "loss": 0.0181,
686
+ "step": 1330
687
+ },
688
+ {
689
+ "epoch": 0.76,
690
+ "eval_accuracy": 0.95,
691
+ "eval_f1": 0.9479889042995839,
692
+ "eval_loss": 0.4586262106895447,
693
+ "eval_precision": 0.9877167630057804,
694
+ "eval_recall": 0.9113333333333333,
695
+ "eval_runtime": 462.3672,
696
+ "eval_samples_per_second": 6.488,
697
+ "eval_steps_per_second": 0.811,
698
+ "step": 1330
699
+ },
700
+ {
701
+ "epoch": 0.78,
702
+ "learning_rate": 1.1000000000000001e-05,
703
+ "loss": 0.098,
704
+ "step": 1365
705
+ },
706
+ {
707
+ "epoch": 0.78,
708
+ "eval_accuracy": 0.9503333333333334,
709
+ "eval_f1": 0.9479566887879846,
710
+ "eval_loss": 0.4024898409843445,
711
+ "eval_precision": 0.9955979457079971,
712
+ "eval_recall": 0.9046666666666666,
713
+ "eval_runtime": 461.9999,
714
+ "eval_samples_per_second": 6.494,
715
+ "eval_steps_per_second": 0.812,
716
+ "step": 1365
717
+ },
718
+ {
719
+ "epoch": 0.8,
720
+ "learning_rate": 1e-05,
721
+ "loss": 0.0752,
722
+ "step": 1400
723
+ },
724
+ {
725
+ "epoch": 0.8,
726
+ "eval_accuracy": 0.955,
727
+ "eval_f1": 0.9534322180062091,
728
+ "eval_loss": 0.2519637942314148,
729
+ "eval_precision": 0.9878484631879915,
730
+ "eval_recall": 0.9213333333333333,
731
+ "eval_runtime": 462.3269,
732
+ "eval_samples_per_second": 6.489,
733
+ "eval_steps_per_second": 0.811,
734
+ "step": 1400
735
+ },
736
+ {
737
+ "epoch": 0.82,
738
+ "learning_rate": 9e-06,
739
+ "loss": 0.0575,
740
+ "step": 1435
741
+ },
742
+ {
743
+ "epoch": 0.82,
744
+ "eval_accuracy": 0.959,
745
+ "eval_f1": 0.958120531154239,
746
+ "eval_loss": 0.24917536973953247,
747
+ "eval_precision": 0.9791231732776617,
748
+ "eval_recall": 0.938,
749
+ "eval_runtime": 462.2475,
750
+ "eval_samples_per_second": 6.49,
751
+ "eval_steps_per_second": 0.811,
752
+ "step": 1435
753
+ },
754
+ {
755
+ "epoch": 0.84,
756
+ "learning_rate": 8.000000000000001e-06,
757
+ "loss": 0.0691,
758
+ "step": 1470
759
+ },
760
+ {
761
+ "epoch": 0.84,
762
+ "eval_accuracy": 0.9576666666666667,
763
+ "eval_f1": 0.9564322469982849,
764
+ "eval_loss": 0.3299550414085388,
765
+ "eval_precision": 0.9851590106007068,
766
+ "eval_recall": 0.9293333333333333,
767
+ "eval_runtime": 463.7914,
768
+ "eval_samples_per_second": 6.468,
769
+ "eval_steps_per_second": 0.809,
770
+ "step": 1470
771
+ },
772
+ {
773
+ "epoch": 0.86,
774
+ "learning_rate": 7.000000000000001e-06,
775
+ "loss": 0.0617,
776
+ "step": 1505
777
+ },
778
+ {
779
+ "epoch": 0.86,
780
+ "eval_accuracy": 0.957,
781
+ "eval_f1": 0.9556853315012023,
782
+ "eval_loss": 0.33974531292915344,
783
+ "eval_precision": 0.9858256555634302,
784
+ "eval_recall": 0.9273333333333333,
785
+ "eval_runtime": 462.1154,
786
+ "eval_samples_per_second": 6.492,
787
+ "eval_steps_per_second": 0.811,
788
+ "step": 1505
789
+ },
790
+ {
791
+ "epoch": 0.88,
792
+ "learning_rate": 6e-06,
793
+ "loss": 0.0392,
794
+ "step": 1540
795
+ },
796
+ {
797
+ "epoch": 0.88,
798
+ "eval_accuracy": 0.9573333333333334,
799
+ "eval_f1": 0.9561042524005487,
800
+ "eval_loss": 0.3205489218235016,
801
+ "eval_precision": 0.9844632768361582,
802
+ "eval_recall": 0.9293333333333333,
803
+ "eval_runtime": 463.3482,
804
+ "eval_samples_per_second": 6.475,
805
+ "eval_steps_per_second": 0.809,
806
+ "step": 1540
807
+ },
808
+ {
809
+ "epoch": 0.9,
810
+ "learning_rate": 5e-06,
811
+ "loss": 0.0386,
812
+ "step": 1575
813
+ },
814
+ {
815
+ "epoch": 0.9,
816
+ "eval_accuracy": 0.958,
817
+ "eval_f1": 0.9571428571428571,
818
+ "eval_loss": 0.27690985798835754,
819
+ "eval_precision": 0.9770833333333333,
820
+ "eval_recall": 0.938,
821
+ "eval_runtime": 462.2334,
822
+ "eval_samples_per_second": 6.49,
823
+ "eval_steps_per_second": 0.811,
824
+ "step": 1575
825
+ },
826
+ {
827
+ "epoch": 0.92,
828
+ "learning_rate": 4.000000000000001e-06,
829
+ "loss": 0.0334,
830
+ "step": 1610
831
+ },
832
+ {
833
+ "epoch": 0.92,
834
+ "eval_accuracy": 0.9573333333333334,
835
+ "eval_f1": 0.9562243502051984,
836
+ "eval_loss": 0.30371883511543274,
837
+ "eval_precision": 0.9817415730337079,
838
+ "eval_recall": 0.932,
839
+ "eval_runtime": 462.4404,
840
+ "eval_samples_per_second": 6.487,
841
+ "eval_steps_per_second": 0.811,
842
+ "step": 1610
843
+ },
844
+ {
845
+ "epoch": 0.94,
846
+ "learning_rate": 3e-06,
847
+ "loss": 0.1167,
848
+ "step": 1645
849
+ },
850
+ {
851
+ "epoch": 0.94,
852
+ "eval_accuracy": 0.9593333333333334,
853
+ "eval_f1": 0.9583333333333333,
854
+ "eval_loss": 0.3056192100048065,
855
+ "eval_precision": 0.9824929971988795,
856
+ "eval_recall": 0.9353333333333333,
857
+ "eval_runtime": 461.9141,
858
+ "eval_samples_per_second": 6.495,
859
+ "eval_steps_per_second": 0.812,
860
+ "step": 1645
861
+ },
862
+ {
863
+ "epoch": 0.96,
864
+ "learning_rate": 2.0000000000000003e-06,
865
+ "loss": 0.0505,
866
+ "step": 1680
867
+ },
868
+ {
869
+ "epoch": 0.96,
870
+ "eval_accuracy": 0.953,
871
+ "eval_f1": 0.9510586601874348,
872
+ "eval_loss": 0.42186784744262695,
873
+ "eval_precision": 0.9920347574221579,
874
+ "eval_recall": 0.9133333333333333,
875
+ "eval_runtime": 462.3061,
876
+ "eval_samples_per_second": 6.489,
877
+ "eval_steps_per_second": 0.811,
878
+ "step": 1680
879
+ },
880
+ {
881
+ "epoch": 0.98,
882
+ "learning_rate": 1.0000000000000002e-06,
883
+ "loss": 0.0468,
884
+ "step": 1715
885
+ },
886
+ {
887
+ "epoch": 0.98,
888
+ "eval_accuracy": 0.9576666666666667,
889
+ "eval_f1": 0.9562822719449227,
890
+ "eval_loss": 0.3499450385570526,
891
+ "eval_precision": 0.9886120996441281,
892
+ "eval_recall": 0.926,
893
+ "eval_runtime": 462.275,
894
+ "eval_samples_per_second": 6.49,
895
+ "eval_steps_per_second": 0.811,
896
+ "step": 1715
897
+ },
898
+ {
899
+ "epoch": 1.0,
900
+ "learning_rate": 0.0,
901
+ "loss": 0.0429,
902
+ "step": 1750
903
+ },
904
+ {
905
+ "epoch": 1.0,
906
+ "eval_accuracy": 0.9576666666666667,
907
+ "eval_f1": 0.9562822719449227,
908
+ "eval_loss": 0.34879612922668457,
909
+ "eval_precision": 0.9886120996441281,
910
+ "eval_recall": 0.926,
911
+ "eval_runtime": 462.1636,
912
+ "eval_samples_per_second": 6.491,
913
+ "eval_steps_per_second": 0.811,
914
+ "step": 1750
915
+ }
916
+ ],
917
+ "max_steps": 1750,
918
+ "num_train_epochs": 1,
919
+ "total_flos": 1.0151220215808e+17,
920
+ "trial_name": null,
921
+ "trial_params": null
922
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2d8678b7c6bfe9615db5aca300ebb7a40d9d87b308c181a8bff98588aba6c7b
3
+ size 3451