Update README.md
Browse files
README.md
CHANGED
|
@@ -29,14 +29,14 @@ Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jac
|
|
| 29 |
|
| 30 |
#### Training Hyperparameters
|
| 31 |
|
| 32 |
-
|
| 33 |
- learning_rate = 5e-5
|
| 34 |
- train_batch_size = 8
|
| 35 |
- eval_batch_size = 8
|
| 36 |
- lr_scheduler_type = linear
|
| 37 |
- num_train_epochs = 5.0
|
| 38 |
|
| 39 |
-
|
| 40 |
- learning_rate = 1e-5
|
| 41 |
- train_batch_size = 8
|
| 42 |
- eval_batch_size = 8
|
|
|
|
| 29 |
|
| 30 |
#### Training Hyperparameters
|
| 31 |
|
| 32 |
+
Second fine-tuning hyper-parameters(on train(0.8) and val(0.2))
|
| 33 |
- learning_rate = 5e-5
|
| 34 |
- train_batch_size = 8
|
| 35 |
- eval_batch_size = 8
|
| 36 |
- lr_scheduler_type = linear
|
| 37 |
- num_train_epochs = 5.0
|
| 38 |
|
| 39 |
+
Fecond fine-tuning hyper-parameters(on train and test)
|
| 40 |
- learning_rate = 1e-5
|
| 41 |
- train_batch_size = 8
|
| 42 |
- eval_batch_size = 8
|