"num_classes": 150 to 50
Browse files- config.json +2 -2
config.json
CHANGED
@@ -2431,13 +2431,13 @@
|
|
2431 |
},
|
2432 |
"layer_norm_eps": 1e-05,
|
2433 |
"mask_dim": 256,
|
2434 |
-
"mask_weight":
|
2435 |
"max_seq_len": 77,
|
2436 |
"model_type": "oneformer",
|
2437 |
"no_object_weight": 0.1,
|
2438 |
"norm": "GN",
|
2439 |
"num_attention_heads": 8,
|
2440 |
-
"num_classes":
|
2441 |
"num_hidden_layers": 10,
|
2442 |
"num_queries": 250,
|
2443 |
"output_attentions": true,
|
|
|
2431 |
},
|
2432 |
"layer_norm_eps": 1e-05,
|
2433 |
"mask_dim": 256,
|
2434 |
+
"mask_weight": 5.0,
|
2435 |
"max_seq_len": 77,
|
2436 |
"model_type": "oneformer",
|
2437 |
"no_object_weight": 0.1,
|
2438 |
"norm": "GN",
|
2439 |
"num_attention_heads": 8,
|
2440 |
+
"num_classes": 50,
|
2441 |
"num_hidden_layers": 10,
|
2442 |
"num_queries": 250,
|
2443 |
"output_attentions": true,
|