khalidddd commited on
Commit
d9ac21b
1 Parent(s): 7a30256

"num_classes": 150 to 50

Browse files
Files changed (1) hide show
  1. 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": 50.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": 150,
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,