| { |
| "model_name": "WIP2", |
| "base_model": "meta-llama/Llama-Prompt-Guard-2-22M", |
| "fine_tuning_method": "LoRA", |
| "dataset": "qualifire/prompt-injections-benchmark", |
| "num_labels": 2, |
| "lora_rank": 16, |
| "lora_alpha": 32, |
| "lora_dropout": 0.1, |
| "max_length": 512, |
| "final_metrics": { |
| "eval_loss": 0.2866095304489136, |
| "eval_accuracy": 0.9197080291970803, |
| "eval_f1": 0.8929068150208623, |
| "eval_precision": 0.9525222551928784, |
| "eval_recall": 0.8403141361256544, |
| "eval_roc_auc": 0.9797109076555935, |
| "eval_runtime": 18.0185, |
| "eval_samples_per_second": 53.223, |
| "eval_steps_per_second": 6.66, |
| "epoch": 5.0 |
| }, |
| "hf_ready": true, |
| "merge_lora": true |
| } |
