dimpo
/

bert-pretrained-wikitext-2-raw-v1

Generated from Trainer

Model card Files Files and versions

dimpo commited on Sep 14, 2023

Commit

6f4d079

·

1 Parent(s): e9fcf93

Update config.json

Change the architecture to "BertForMaskedLM".

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "BertForPreTraining"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,

 {
   "architectures": [
+    "BertForMaskedLM"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,