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
|