aboutsummaryrefslogtreecommitdiff
path: root/configs/bert_50M.json
diff options
context:
space:
mode:
Diffstat (limited to 'configs/bert_50M.json')
-rw-r--r--configs/bert_50M.json26
1 files changed, 26 insertions, 0 deletions
diff --git a/configs/bert_50M.json b/configs/bert_50M.json
new file mode 100644
index 0000000..9e2facb
--- /dev/null
+++ b/configs/bert_50M.json
@@ -0,0 +1,26 @@
+{
+ "_name_or_path": "bert-base-uncased",
+ "architectures": [
+ "BertForMaskedLM"
+ ],
+ "attention_probs_dropout_prob": 0.1,
+ "classifier_dropout": null,
+ "gradient_checkpointing": false,
+ "hidden_act": "gelu",
+ "hidden_dropout_prob": 0.1,
+ "hidden_size": 640,
+ "initializer_range": 0.02,
+ "intermediate_size": 2560,
+ "layer_norm_eps": 1e-12,
+ "max_position_embeddings": 128,
+ "model_type": "bert",
+ "num_attention_heads": 8,
+ "num_hidden_layers": 6,
+ "pad_token_id": 0,
+ "position_embedding_type": "absolute",
+ "torch_dtype": "float32",
+ "transformers_version": "4.46.2",
+ "type_vocab_size": 2,
+ "use_cache": true,
+ "vocab_size": 30522
+} \ No newline at end of file