Update README.md
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ Due to being active vulnerabilities under review, our dataset of malicious promp
|
|
22 |
## Training Details
|
23 |
|
24 |
The model was trained using RobertaForSequenceClassification.from_pretrained.
|
25 |
-
In this process, both the model and tokenizer pertinent to the RoBERTa-base were employed and trained for 10 epochs (learning rate
|
26 |
|
27 |
## Inference
|
28 |
|
|
|
22 |
## Training Details
|
23 |
|
24 |
The model was trained using RobertaForSequenceClassification.from_pretrained.
|
25 |
+
In this process, both the model and tokenizer pertinent to the RoBERTa-base were employed and trained for 10 epochs (learning rate 2e-5 and AdamW Optimizer).
|
26 |
|
27 |
## Inference
|
28 |
|