Update README.md
Browse files
README.md
CHANGED
@@ -3,12 +3,19 @@ This model is a fine-tuned version of bert-base-uncased on an NLI dataset. It ac
|
|
3 |
{'precision': 0.9690210656753407} {'recall': 0.9722337339411521} {'f1': 0.9706247414149772} {'accuracy': 0.9535340314136126}
|
4 |
|
5 |
Training hyperparameters:
|
|
|
6 |
learning_rate: 2e-5
|
|
|
7 |
train_batch_size: 8
|
|
|
8 |
eval_batch_size: 8
|
|
|
9 |
optimizer: AdamW with betas=(0.9,0.999) and epsilon=1e-08
|
|
|
10 |
weight_decay= 0.01
|
|
|
11 |
lr_scheduler_type: linear
|
|
|
12 |
num_epochs: 3
|
13 |
|
14 |
|
|
|
3 |
{'precision': 0.9690210656753407} {'recall': 0.9722337339411521} {'f1': 0.9706247414149772} {'accuracy': 0.9535340314136126}
|
4 |
|
5 |
Training hyperparameters:
|
6 |
+
|
7 |
learning_rate: 2e-5
|
8 |
+
|
9 |
train_batch_size: 8
|
10 |
+
|
11 |
eval_batch_size: 8
|
12 |
+
|
13 |
optimizer: AdamW with betas=(0.9,0.999) and epsilon=1e-08
|
14 |
+
|
15 |
weight_decay= 0.01
|
16 |
+
|
17 |
lr_scheduler_type: linear
|
18 |
+
|
19 |
num_epochs: 3
|
20 |
|
21 |
|