Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
|
@@ -7,6 +7,7 @@
|
|
| 7 |
"architectures": [
|
| 8 |
"BartModel"
|
| 9 |
],
|
|
|
|
| 10 |
"attention_dropout": 0.0,
|
| 11 |
"bos_token_id": 0,
|
| 12 |
"classif_dropout": 0.0,
|
|
@@ -105,12 +106,11 @@
|
|
| 105 |
},
|
| 106 |
"max_position_embeddings": 1024,
|
| 107 |
"model_type": "bart",
|
| 108 |
-
"
|
| 109 |
"normalize_before": false,
|
| 110 |
"num_hidden_layers": 12,
|
| 111 |
"output_past": false,
|
| 112 |
"pad_token_id": 1,
|
| 113 |
-
"problem_type": "multi_label_classification",
|
| 114 |
"scale_embedding": false,
|
| 115 |
"torch_dtype": "float32",
|
| 116 |
"transformers_version": "4.26.0",
|
|
|
|
| 7 |
"architectures": [
|
| 8 |
"BartModel"
|
| 9 |
],
|
| 10 |
+
"multi_target_strategy": "multi-output",
|
| 11 |
"attention_dropout": 0.0,
|
| 12 |
"bos_token_id": 0,
|
| 13 |
"classif_dropout": 0.0,
|
|
|
|
| 106 |
},
|
| 107 |
"max_position_embeddings": 1024,
|
| 108 |
"model_type": "bart",
|
| 109 |
+
"problem_type": "multi_label_classification",
|
| 110 |
"normalize_before": false,
|
| 111 |
"num_hidden_layers": 12,
|
| 112 |
"output_past": false,
|
| 113 |
"pad_token_id": 1,
|
|
|
|
| 114 |
"scale_embedding": false,
|
| 115 |
"torch_dtype": "float32",
|
| 116 |
"transformers_version": "4.26.0",
|