End of training

README.md

@@ -33,8 +33,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 3e-05
-- train_batch_size: 32
-- eval_batch_size: 64
+- train_batch_size: 64
+- eval_batch_size: 128
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
@@ -50,4 +50,4 @@ The following hyperparameters were used during training:
 - Transformers 4.55.0
 - Pytorch 2.6.0+cu124
 - Datasets 4.0.0
-- Tokenizers 0.21.4
+- Tokenizers 0.21.2

config.json

@@ -9,8 +9,16 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
+  "id2label": {
+    "0": "safe",
+    "1": "vulnerable"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "safe": 0,
+    "vulnerable": 1
+  },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
   "model_type": "roberta",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:54037e9fb1e8fe118b8f05736047adb51a4a0cd7f17573b4e3baf726d2371945
+oid sha256:b45f447c0743a4ee275ba98d21c4ac85622ddf5dcde3b4b94da36789b9d25413
 size 498612824

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b534c46f49a0c984558ba066693d32a9622c174404dc9d6e4081052d9056772
+oid sha256:857ad044839a70d7339d35bd5f60d4bdbba61b83ab366708b8b430cc2906a6ec
 size 5368