STU-Injection / training_info.json
ButterM40's picture
Upload folder using huggingface_hub
4b793ee verified
{
"model_name": "microsoft/deberta-v3-base",
"datasets_used": [
"xTRam1/safe-guard-prompt-injection",
"deepset/prompt-injections",
"jayavibhav/prompt-injection-safety"
],
"training_samples": 52903,
"validation_samples": 5879,
"final_metrics": {
"eval_loss": 0.027365634217858315,
"eval_accuracy": 0.995917673073652,
"eval_precision": 0.9976027397260274,
"eval_recall": 0.99419795221843,
"eval_f1": 0.9958974358974358,
"eval_runtime": 26.8342,
"eval_samples_per_second": 219.086,
"eval_steps_per_second": 27.39,
"epoch": 3.0
},
"training_args": {
"learning_rate": 3e-05,
"batch_size": 8,
"num_epochs": 3,
"weight_decay": 0.01
}
}