oranne55 commited on
Commit
68d73db
1 Parent(s): e8a03fc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -29,14 +29,14 @@ Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jac
29
 
30
  #### Training Hyperparameters
31
 
32
- 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
- second fine-tuning hyper-parameters(on train and test)
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