diff options
Diffstat (limited to 'configs/bert_6M.json')
-rw-r--r-- | configs/bert_6M.json | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/configs/bert_6M.json b/configs/bert_6M.json new file mode 100644 index 0000000..b093bc5 --- /dev/null +++ b/configs/bert_6M.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": 160, + "initializer_range": 0.02, + "intermediate_size": 640, + "layer_norm_eps": 1e-12, + "max_position_embeddings": 128, + "model_type": "bert", + "num_attention_heads": 4, + "num_hidden_layers": 4, + "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 |