Update README.md
Browse files
README.md
CHANGED
@@ -48,8 +48,8 @@ Refer [here](https://github.com/arcleife/notebooks/blob/main/sentiment_finetunin
|
|
48 |
**Training hyperparameters**
|
49 |
|
50 |
- num_train_epochs = 5
|
51 |
-
- learning_rate =
|
52 |
-
- weight_decay =
|
53 |
- per_device_train_batch_size = 16
|
54 |
- per_device_eval_batch_size = 16
|
55 |
- fp16 = True
|
@@ -60,8 +60,8 @@ Refer [here](https://github.com/arcleife/notebooks/blob/main/sentiment_finetunin
|
|
60 |
|
61 |
|Epoch| Training Loss | Validation Loss | F1 | Recall | Precision |
|
62 |
|-----|---------------|-----------------|----------|----------|-----------|
|
63 |
-
|1 | No log | 0.
|
64 |
-
|2 | No log | 0.
|
65 |
-
|3 | No log | 0.
|
66 |
-
|4 | No log | 0.
|
67 |
-
|5 | No log | 0.
|
|
|
48 |
**Training hyperparameters**
|
49 |
|
50 |
- num_train_epochs = 5
|
51 |
+
- learning_rate = 5e-6
|
52 |
+
- weight_decay = 1e-1
|
53 |
- per_device_train_batch_size = 16
|
54 |
- per_device_eval_batch_size = 16
|
55 |
- fp16 = True
|
|
|
60 |
|
61 |
|Epoch| Training Loss | Validation Loss | F1 | Recall | Precision |
|
62 |
|-----|---------------|-----------------|----------|----------|-----------|
|
63 |
+
|1 | No log | 0.283834 | 0.908730 | 0.908730 | 0.908730 |
|
64 |
+
|2 | No log | 0.248232 | 0.930952 | 0.930952 | 0.930952 |
|
65 |
+
|3 | No log | 0.282172 | 0.930952 | 0.930952 | 0.930952 |
|
66 |
+
|4 | No log | 0.257302 | 0.936508 | 0.936508 | 0.936508 |
|
67 |
+
|5 | No log | 0.271212 | 0.939683 | 0.939683 | 0.939683 |
|