Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ result = classifier(jailbreak)
|
|
| 39 |
|
| 40 |
## Training Details
|
| 41 |
|
| 42 |
-
Trained for 1 Hour on an A100 with the following parameters
|
| 43 |
|
| 44 |
| Parameter | Value |
|
| 45 |
|-----------|-------|
|
|
@@ -51,12 +51,6 @@ Trained for 1 Hour on an A100 with the following parameters
|
|
| 51 |
| warmup_ratio | 0.1 |
|
| 52 |
| weight_decay | 0.01 |
|
| 53 |
| fp16 | True |
|
| 54 |
-
| eval_strategy | "steps" |
|
| 55 |
-
| eval_steps | 500 |
|
| 56 |
-
| save_strategy | "steps" |
|
| 57 |
-
| save_steps | 500 |
|
| 58 |
-
| logging_steps | 100 |
|
| 59 |
-
| load_best_model_at_end | True |
|
| 60 |
| metric_for_best_model | "eval_loss" |
|
| 61 |
|
| 62 |
Resulting in the following training metrics:
|
|
|
|
| 39 |
|
| 40 |
## Training Details
|
| 41 |
|
| 42 |
+
Trained for 1 Hour on an A100 with the following parameters via [transformers](https://huggingface.co/docs/transformers/en/index)
|
| 43 |
|
| 44 |
| Parameter | Value |
|
| 45 |
|-----------|-------|
|
|
|
|
| 51 |
| warmup_ratio | 0.1 |
|
| 52 |
| weight_decay | 0.01 |
|
| 53 |
| fp16 | True |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
| metric_for_best_model | "eval_loss" |
|
| 55 |
|
| 56 |
Resulting in the following training metrics:
|