noriyukipy commited on
Commit
565d251
1 Parent(s): 580d048

Update architecture to BertForPreTraining

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. config.json +1 -1
README.md CHANGED
@@ -1,6 +1,7 @@
1
  ---
2
  language: ja
3
  datasets: wikipedia
 
4
  widget:
5
  - text: "得意な科目は[MASK]です。"
6
  license: cc-by-sa-3.0
1
  ---
2
  language: ja
3
  datasets: wikipedia
4
+ pipeline_tag: fill-mask
5
  widget:
6
  - text: "得意な科目は[MASK]です。"
7
  license: cc-by-sa-3.0
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "release/bert-base-ja",
3
  "architectures": [
4
- "BertForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 2,
1
  {
2
  "_name_or_path": "release/bert-base-ja",
3
  "architectures": [
4
+ "BertForPreTraining"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 2,