Update config.json
Browse files- config.json +7 -3
config.json
CHANGED
|
@@ -1,6 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
-
"architectures":
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
"attention_dropout": 0.0,
|
| 5 |
"bos_token_id": 0,
|
| 6 |
"classif_dropout": 0.0,
|
|
@@ -33,11 +37,11 @@
|
|
| 33 |
"model_type": "bart",
|
| 34 |
"num_beams": 1,
|
| 35 |
"num_hidden_layers": 12,
|
| 36 |
-
"num_labels":
|
| 37 |
"num_return_sequences": 1,
|
| 38 |
"output_attentions": false,
|
| 39 |
"output_hidden_states": false,
|
| 40 |
-
"output_past":
|
| 41 |
"pad_token_id": 1,
|
| 42 |
"pruned_heads": {},
|
| 43 |
"repetition_penalty": 1.0,
|
|
|
|
| 1 |
{
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
+
"architectures": [
|
| 4 |
+
"BartModel",
|
| 5 |
+
"BartForMaskedLM",
|
| 6 |
+
"BartForSequenceClassification"
|
| 7 |
+
],
|
| 8 |
"attention_dropout": 0.0,
|
| 9 |
"bos_token_id": 0,
|
| 10 |
"classif_dropout": 0.0,
|
|
|
|
| 37 |
"model_type": "bart",
|
| 38 |
"num_beams": 1,
|
| 39 |
"num_hidden_layers": 12,
|
| 40 |
+
"num_labels": 2,
|
| 41 |
"num_return_sequences": 1,
|
| 42 |
"output_attentions": false,
|
| 43 |
"output_hidden_states": false,
|
| 44 |
+
"output_past": false,
|
| 45 |
"pad_token_id": 1,
|
| 46 |
"pruned_heads": {},
|
| 47 |
"repetition_penalty": 1.0,
|