Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
text-embeddings-inference
Instructions to use mariadg/AttackVectorClassifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mariadg/AttackVectorClassifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mariadg/AttackVectorClassifier")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mariadg/AttackVectorClassifier") model = AutoModelForSequenceClassification.from_pretrained("mariadg/AttackVectorClassifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress, epoch 1, checkpoint
Browse files- last-checkpoint/config.json +47 -0
- last-checkpoint/model.safetensors +3 -0
- last-checkpoint/optimizer.pt +3 -0
- last-checkpoint/rng_state.pth +3 -0
- last-checkpoint/scheduler.pt +3 -0
- last-checkpoint/tokenizer.json +0 -0
- last-checkpoint/tokenizer_config.json +24 -0
- last-checkpoint/trainer_state.json +75 -0
- last-checkpoint/training_args.bin +3 -0
last-checkpoint/config.json
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"DebertaV2ForSequenceClassification"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 1,
|
| 7 |
+
"dtype": "float16",
|
| 8 |
+
"eos_token_id": 2,
|
| 9 |
+
"hidden_act": "gelu",
|
| 10 |
+
"hidden_dropout_prob": 0.1,
|
| 11 |
+
"hidden_size": 768,
|
| 12 |
+
"id2label": {
|
| 13 |
+
"0": "no_framing",
|
| 14 |
+
"1": "has_framing"
|
| 15 |
+
},
|
| 16 |
+
"initializer_range": 0.02,
|
| 17 |
+
"intermediate_size": 3072,
|
| 18 |
+
"label2id": {
|
| 19 |
+
"has_framing": 1,
|
| 20 |
+
"no_framing": 0
|
| 21 |
+
},
|
| 22 |
+
"layer_norm_eps": 1e-07,
|
| 23 |
+
"legacy": true,
|
| 24 |
+
"max_position_embeddings": 512,
|
| 25 |
+
"max_relative_positions": -1,
|
| 26 |
+
"model_type": "deberta-v2",
|
| 27 |
+
"norm_rel_ebd": "layer_norm",
|
| 28 |
+
"num_attention_heads": 12,
|
| 29 |
+
"num_hidden_layers": 12,
|
| 30 |
+
"pad_token_id": 0,
|
| 31 |
+
"pooler_dropout": 0.0,
|
| 32 |
+
"pooler_hidden_act": "gelu",
|
| 33 |
+
"pooler_hidden_size": 768,
|
| 34 |
+
"pos_att_type": [
|
| 35 |
+
"p2c",
|
| 36 |
+
"c2p"
|
| 37 |
+
],
|
| 38 |
+
"position_biased_input": false,
|
| 39 |
+
"position_buckets": 256,
|
| 40 |
+
"relative_attention": true,
|
| 41 |
+
"share_att_key": true,
|
| 42 |
+
"tie_word_embeddings": true,
|
| 43 |
+
"transformers_version": "5.13.1",
|
| 44 |
+
"type_vocab_size": 0,
|
| 45 |
+
"use_cache": false,
|
| 46 |
+
"vocab_size": 128100
|
| 47 |
+
}
|
last-checkpoint/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b8382937490df6137c1fd1d38ba0f2b1b42eb923ae848f769240e5dc21ea8533
|
| 3 |
+
size 368871908
|
last-checkpoint/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d8ca6d770dc2d0d444d05996b7c6eeaa860981ba26d43183f8a9919dcef2e8b3
|
| 3 |
+
size 737866955
|
last-checkpoint/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b7e633624ad002c88db2ec0b3f763a06ed8e3232a80fc4851f0a47730b62f4de
|
| 3 |
+
size 14645
|
last-checkpoint/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:749320dbe724bd79cb64e03764e4ce7cd007dd9fa3dcc3514dae8ac78750eb22
|
| 3 |
+
size 1465
|
last-checkpoint/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
last-checkpoint/tokenizer_config.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": true,
|
| 3 |
+
"backend": "tokenizers",
|
| 4 |
+
"bos_token": "[CLS]",
|
| 5 |
+
"cls_token": "[CLS]",
|
| 6 |
+
"do_lower_case": false,
|
| 7 |
+
"eos_token": "[SEP]",
|
| 8 |
+
"extra_special_tokens": [
|
| 9 |
+
"[PAD]",
|
| 10 |
+
"[CLS]",
|
| 11 |
+
"[SEP]"
|
| 12 |
+
],
|
| 13 |
+
"is_local": false,
|
| 14 |
+
"local_files_only": false,
|
| 15 |
+
"mask_token": "[MASK]",
|
| 16 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 17 |
+
"pad_token": "[PAD]",
|
| 18 |
+
"sep_token": "[SEP]",
|
| 19 |
+
"split_by_punct": false,
|
| 20 |
+
"tokenizer_class": "DebertaV2Tokenizer",
|
| 21 |
+
"unk_id": 3,
|
| 22 |
+
"unk_token": "[UNK]",
|
| 23 |
+
"vocab_type": "spm"
|
| 24 |
+
}
|
last-checkpoint/trainer_state.json
ADDED
|
@@ -0,0 +1,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": 2393,
|
| 3 |
+
"best_metric": 0.0,
|
| 4 |
+
"best_model_checkpoint": "./results_deberta_framing/checkpoint-2393",
|
| 5 |
+
"epoch": 1.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 2393,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 0.20894274968658588,
|
| 14 |
+
"grad_norm": 7.546875,
|
| 15 |
+
"learning_rate": 1.8609834238751915e-05,
|
| 16 |
+
"loss": 0.7753585815429688,
|
| 17 |
+
"step": 500
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.41788549937317176,
|
| 21 |
+
"grad_norm": 2.30859375,
|
| 22 |
+
"learning_rate": 1.721688257417468e-05,
|
| 23 |
+
"loss": 0.6949208984375,
|
| 24 |
+
"step": 1000
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.6268282490597576,
|
| 28 |
+
"grad_norm": 0.0870361328125,
|
| 29 |
+
"learning_rate": 1.5823930909597438e-05,
|
| 30 |
+
"loss": 0.69367626953125,
|
| 31 |
+
"step": 1500
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.8357709987463435,
|
| 35 |
+
"grad_norm": NaN,
|
| 36 |
+
"learning_rate": 1.4430979245020198e-05,
|
| 37 |
+
"loss": 0.6934404907226562,
|
| 38 |
+
"step": 2000
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 1.0,
|
| 42 |
+
"eval_accuracy": 0.4597220771079302,
|
| 43 |
+
"eval_f1": 0.0,
|
| 44 |
+
"eval_loss": NaN,
|
| 45 |
+
"eval_precision": 0.0,
|
| 46 |
+
"eval_recall": 0.0,
|
| 47 |
+
"eval_roc_auc": 0.0,
|
| 48 |
+
"eval_runtime": 329.3943,
|
| 49 |
+
"eval_samples_per_second": 29.056,
|
| 50 |
+
"eval_steps_per_second": 1.818,
|
| 51 |
+
"step": 2393
|
| 52 |
+
}
|
| 53 |
+
],
|
| 54 |
+
"logging_steps": 500,
|
| 55 |
+
"max_steps": 7179,
|
| 56 |
+
"num_input_tokens_seen": 0,
|
| 57 |
+
"num_train_epochs": 3,
|
| 58 |
+
"save_steps": 500,
|
| 59 |
+
"stateful_callbacks": {
|
| 60 |
+
"TrainerControl": {
|
| 61 |
+
"args": {
|
| 62 |
+
"should_epoch_stop": false,
|
| 63 |
+
"should_evaluate": false,
|
| 64 |
+
"should_log": false,
|
| 65 |
+
"should_save": true,
|
| 66 |
+
"should_training_stop": false
|
| 67 |
+
},
|
| 68 |
+
"attributes": {}
|
| 69 |
+
}
|
| 70 |
+
},
|
| 71 |
+
"total_flos": 6497245252899744.0,
|
| 72 |
+
"train_batch_size": 16,
|
| 73 |
+
"trial_name": null,
|
| 74 |
+
"trial_params": null
|
| 75 |
+
}
|
last-checkpoint/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:268a3feca8af97dcf2a4a454060601cea918cf013dc64b3dcf7b37eb81ba5579
|
| 3 |
+
size 5201
|