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 |
+
```
|