Commit
•
c3715a4
1
Parent(s):
3c2de89
Add model
Browse files- config.json +3 -0
config.json
CHANGED
|
@@ -84,6 +84,7 @@
|
|
| 84 |
"use_auxiliary_loss": true,
|
| 85 |
"use_return_dict": true
|
| 86 |
},
|
|
|
|
| 87 |
"id2label": {
|
| 88 |
"0": "wall",
|
| 89 |
"1": "building",
|
|
@@ -389,6 +390,8 @@
|
|
| 389 |
"window ": 8
|
| 390 |
},
|
| 391 |
"model_type": "oneformer",
|
|
|
|
|
|
|
| 392 |
"text_encoder_config": {
|
| 393 |
"max_seq_len": 77,
|
| 394 |
"task_seq_len": 77,
|
|
|
|
| 84 |
"use_auxiliary_loss": true,
|
| 85 |
"use_return_dict": true
|
| 86 |
},
|
| 87 |
+
"hidden_size": 256,
|
| 88 |
"id2label": {
|
| 89 |
"0": "wall",
|
| 90 |
"1": "building",
|
|
|
|
| 390 |
"window ": 8
|
| 391 |
},
|
| 392 |
"model_type": "oneformer",
|
| 393 |
+
"num_attention_heads": 8,
|
| 394 |
+
"num_hidden_layers": 10,
|
| 395 |
"text_encoder_config": {
|
| 396 |
"max_seq_len": 77,
|
| 397 |
"task_seq_len": 77,
|