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 |
|