BroLaurens
commited on
Commit
•
5c77061
1
Parent(s):
0296e8f
Update README.md
Browse files
README.md
CHANGED
@@ -62,6 +62,7 @@ predicted_labels = [[int2str[x] for x in t] for t in predictions.tolist()]
|
|
62 |
|
63 |
The model was trained using the following hyperparameters:
|
64 |
|
|
|
65 |
base_model: distilbert/distilbert-base-uncased
|
66 |
learning_rate: 2e-5
|
67 |
batch_size: 32
|
@@ -70,4 +71,5 @@ optimizer: adamw
|
|
70 |
adam_beta1: 0.9
|
71 |
adam_beta2: 0.999
|
72 |
adam_epsilon: 1e-08
|
73 |
-
loss_function: cross entropy loss
|
|
|
|
62 |
|
63 |
The model was trained using the following hyperparameters:
|
64 |
|
65 |
+
```
|
66 |
base_model: distilbert/distilbert-base-uncased
|
67 |
learning_rate: 2e-5
|
68 |
batch_size: 32
|
|
|
71 |
adam_beta1: 0.9
|
72 |
adam_beta2: 0.999
|
73 |
adam_epsilon: 1e-08
|
74 |
+
loss_function: cross entropy loss
|
75 |
+
```
|