praeclarumjj3
commited on
Commit
•
f066502
1
Parent(s):
e2f2821
Add model
Browse files- config.json +45 -56
config.json
CHANGED
@@ -2145,51 +2145,19 @@
|
|
2145 |
"typical_p": 1.0,
|
2146 |
"use_bfloat16": false
|
2147 |
},
|
2148 |
-
"
|
2149 |
-
|
2150 |
-
|
2151 |
-
|
2152 |
-
|
2153 |
-
|
2154 |
-
|
2155 |
-
|
2156 |
-
|
2157 |
-
|
2158 |
-
|
2159 |
-
|
2160 |
-
|
2161 |
-
"pre_norm": false,
|
2162 |
-
"query_dec_layers": 2,
|
2163 |
-
"use_task_norm": true
|
2164 |
-
},
|
2165 |
-
"general_config": {
|
2166 |
-
"class_weight": 2.0,
|
2167 |
-
"contrastive_temperature": 0.07,
|
2168 |
-
"contrastive_weight": 0.5,
|
2169 |
-
"deep_supervision": true,
|
2170 |
-
"dice_weight": 5.0,
|
2171 |
-
"ignore_value": 255,
|
2172 |
-
"importance_sample_ratio": 0.75,
|
2173 |
-
"init_std": 0.02,
|
2174 |
-
"init_xavier_std": 1.0,
|
2175 |
-
"is_train": false,
|
2176 |
-
"layer_norm_eps": 1e-05,
|
2177 |
-
"mask_weight": 5.0,
|
2178 |
-
"no_object_weight": 0.1,
|
2179 |
-
"num_classes": 19,
|
2180 |
-
"num_queries": 250,
|
2181 |
-
"output_auxiliary_logits": true,
|
2182 |
-
"oversample_ratio": 3.0,
|
2183 |
-
"strides": [
|
2184 |
-
4,
|
2185 |
-
8,
|
2186 |
-
16,
|
2187 |
-
32
|
2188 |
-
],
|
2189 |
-
"train_num_points": 12544,
|
2190 |
-
"use_auxiliary_loss": true
|
2191 |
-
},
|
2192 |
-
"hidden_size": 256,
|
2193 |
"id2label": {
|
2194 |
"0": "road",
|
2195 |
"1": "sidewalk",
|
@@ -2211,8 +2179,11 @@
|
|
2211 |
"17": "motorcycle",
|
2212 |
"18": "bicycle"
|
2213 |
},
|
|
|
|
|
2214 |
"init_std": 0.02,
|
2215 |
"init_xavier_std": 1.0,
|
|
|
2216 |
"label2id": {
|
2217 |
"bicycle": 18,
|
2218 |
"building": 2,
|
@@ -2234,21 +2205,39 @@
|
|
2234 |
"vegetation": 8,
|
2235 |
"wall": 3
|
2236 |
},
|
|
|
|
|
|
|
|
|
2237 |
"model_type": "oneformer",
|
|
|
|
|
2238 |
"num_attention_heads": 8,
|
|
|
2239 |
"num_hidden_layers": 10,
|
|
|
2240 |
"output_attentions": true,
|
|
|
2241 |
"output_hidden_states": true,
|
2242 |
-
"
|
2243 |
-
|
2244 |
-
|
2245 |
-
|
2246 |
-
|
2247 |
-
|
2248 |
-
|
2249 |
-
|
2250 |
-
|
2251 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
2252 |
"torch_dtype": "float32",
|
2253 |
-
"
|
|
|
|
|
|
|
2254 |
}
|
|
|
2145 |
"typical_p": 1.0,
|
2146 |
"use_bfloat16": false
|
2147 |
},
|
2148 |
+
"class_weight": 2.0,
|
2149 |
+
"common_stride": 4,
|
2150 |
+
"contrastive_temperature": 0.07,
|
2151 |
+
"contrastive_weight": 0.5,
|
2152 |
+
"conv_dim": 256,
|
2153 |
+
"decoder_layers": 10,
|
2154 |
+
"dice_weight": 5.0,
|
2155 |
+
"dim_feedforward": 2048,
|
2156 |
+
"dropout": 0.1,
|
2157 |
+
"encoder_feedforward_dim": 1024,
|
2158 |
+
"encoder_layers": 6,
|
2159 |
+
"enforce_input_proj": false,
|
2160 |
+
"hidden_dim": 256,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2161 |
"id2label": {
|
2162 |
"0": "road",
|
2163 |
"1": "sidewalk",
|
|
|
2179 |
"17": "motorcycle",
|
2180 |
"18": "bicycle"
|
2181 |
},
|
2182 |
+
"ignore_value": 255,
|
2183 |
+
"importance_sample_ratio": 0.75,
|
2184 |
"init_std": 0.02,
|
2185 |
"init_xavier_std": 1.0,
|
2186 |
+
"is_training": false,
|
2187 |
"label2id": {
|
2188 |
"bicycle": 18,
|
2189 |
"building": 2,
|
|
|
2205 |
"vegetation": 8,
|
2206 |
"wall": 3
|
2207 |
},
|
2208 |
+
"layer_norm_eps": 1e-05,
|
2209 |
+
"mask_dim": 256,
|
2210 |
+
"mask_weight": 5.0,
|
2211 |
+
"max_seq_len": 77,
|
2212 |
"model_type": "oneformer",
|
2213 |
+
"no_object_weight": 0.1,
|
2214 |
+
"norm": "GN",
|
2215 |
"num_attention_heads": 8,
|
2216 |
+
"num_classes": 19,
|
2217 |
"num_hidden_layers": 10,
|
2218 |
+
"num_queries": 250,
|
2219 |
"output_attentions": true,
|
2220 |
+
"output_auxiliary_logits": true,
|
2221 |
"output_hidden_states": true,
|
2222 |
+
"oversample_ratio": 3.0,
|
2223 |
+
"pre_norm": false,
|
2224 |
+
"query_dec_layers": 2,
|
2225 |
+
"strides": [
|
2226 |
+
4,
|
2227 |
+
8,
|
2228 |
+
16,
|
2229 |
+
32
|
2230 |
+
],
|
2231 |
+
"task_seq_len": 77,
|
2232 |
+
"text_encoder_context_length": 77,
|
2233 |
+
"text_encoder_n_ctx": 16,
|
2234 |
+
"text_encoder_num_layers": 6,
|
2235 |
+
"text_encoder_proj_layers": 2,
|
2236 |
+
"text_encoder_vocab_size": 49408,
|
2237 |
+
"text_encoder_width": 256,
|
2238 |
"torch_dtype": "float32",
|
2239 |
+
"train_num_points": 12544,
|
2240 |
+
"transformers_version": null,
|
2241 |
+
"use_auxiliary_loss": true,
|
2242 |
+
"use_task_norm": true
|
2243 |
}
|