Upload BertForSequenceClassification

config.json

@@ -5,25 +5,25 @@
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
-  "directionality": "bidi",
+  "gradient_checkpointing": false,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
     "0": "ELECTRICAL SYSTEM",
-    "1": "OTHER",
+    "1": "STRUCTURE",
     "2": "SERVICE BRAKES",
-    "3": "AIR BAGS",
-    "4": "STRUCTURE"
+    "3": "OTHER",
+    "4": "AIR BAGS"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "AIR BAGS": 3,
+    "AIR BAGS": 4,
     "ELECTRICAL SYSTEM": 0,
-    "OTHER": 1,
+    "OTHER": 3,
     "SERVICE BRAKES": 2,
-    "STRUCTURE": 4
+    "STRUCTURE": 1
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
@@ -31,16 +31,11 @@
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 0,
-  "pooler_fc_size": 768,
-  "pooler_num_attention_heads": 12,
-  "pooler_num_fc_layers": 3,
-  "pooler_size_per_head": 128,
-  "pooler_type": "first_token_transform",
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.48.1",
   "type_vocab_size": 2,
   "use_cache": true,
-  "vocab_size": 119547
+  "vocab_size": 30522
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1484957640ab2427950217224da862386d68f5271696fee43042d1b46546de3b
-size 711452684
+oid sha256:9f5dc85ec1018ce10be5535821be123892b3ff1a089277c5561366900ca32c0b
+size 437967876