system HF staff commited on
Commit
b5ca75f
1 Parent(s): ad404c2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "BertForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": null,
@@ -30,7 +30,7 @@
30
  "num_attention_heads": 12,
31
  "num_beams": 1,
32
  "num_hidden_layers": 12,
33
- "num_labels": 3,
34
  "num_return_sequences": 1,
35
  "output_attentions": false,
36
  "output_hidden_states": false,
1
  {
2
  "architectures": [
3
+ "BertForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": null,
30
  "num_attention_heads": 12,
31
  "num_beams": 1,
32
  "num_hidden_layers": 12,
33
+ "num_labels": 2,
34
  "num_return_sequences": 1,
35
  "output_attentions": false,
36
  "output_hidden_states": false,