ONNX Model Zoo

palash247/results

By palash247 · ↓ 3 · ♥ 0 · Updated 2024-10-11

What's in this repo

This page lists the files in palash247/results, including model.onnx.

  • 1 ONNX models (.onnx files)
  • 17 files, including 10 .json, 2 .model, 1 .md, 1 .safetensors, 1 .onnx, 1 .bin
  • Last updated 2024-10-11

ONNX files

File Size Download
onnx/model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

results

This model is a fine-tuned version of protectai/deberta-v3-base-prompt-injection-v2 on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 1.5060

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 5

Training results

Training Loss Epoch Step Validation Loss
1.6064 1.0 54 1.6097
1.62 2.0 108 1.6100
1.6059 3.0 162 1.6076
1.5734 4.0 216 1.5249
1.5444 5.0 270 1.5060

Framework versions

  • Transformers 4.44.2
  • Pytorch 2.4.1+cu121
  • Datasets 3.0.1
  • Tokenizers 0.19.1

Model info

text-classification transformers
transformers onnx safetensors deberta-v2 text-classification generated_from_trainer base_model:protectai/deberta-v3-base-prompt-injection-v2 base_model:quantized:protectai/deberta-v3-base-prompt-injection-v2 license:apache-2.0 text-embeddings-inference endpoints_compatible region:us
↓ 3 ♥ 0

Created: 2024-10-11

Updated: 2024-10-11

View on Hugging Face

Files (17)

.gitattributes
README.md
added_tokens.json
config.json
model.safetensors
onnx/added_tokens.json
onnx/config.json
onnx/model.onnx ONNX
onnx/special_tokens_map.json
onnx/spm.model
onnx/tokenizer.json
onnx/tokenizer_config.json
special_tokens_map.json
spm.model
tokenizer.json
tokenizer_config.json
training_args.bin

Related models