| { | |
| "model_name": "meta-llama/Llama-Prompt-Guard-2-22M", | |
| "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 | |
| } |