junaid1993
/

Belief_BERT_MultiClass

Text Classification

Model card Files Files and versions

Metrics Training metrics Community

junaid1993 commited on May 7, 2024

Commit

b072767

·

verified ·

1 Parent(s): d15dc16

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -24,7 +24,7 @@
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
-  "n_layers": 6,
   "pad_token_id": 0,
   "problem_type": "single_label_classification",
   "qa_dropout": 0.1,

   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
+  "n_layers": 4,
   "pad_token_id": 0,
   "problem_type": "single_label_classification",
   "qa_dropout": 0.1,