nielsr HF staff commited on
Commit
6efcf58
1 Parent(s): 52ac6ea
Files changed (2) hide show
  1. config.json +22 -15
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -4,8 +4,14 @@
4
  ],
5
  "attention_probs_dropout_prob": 0.0,
6
  "auxiliary_loss_weight": 0.4,
7
- "channels": 256,
8
- "expand_channels": false,
 
 
 
 
 
 
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.0,
11
  "hidden_size": 1024,
@@ -318,27 +324,28 @@
318
  },
319
  "layer_norm_eps": 1e-12,
320
  "model_type": "dpt",
321
- "num_attention_heads": 16,
322
- "num_channels": 3,
323
- "num_hidden_layers": 24,
324
- "out_indices": [
325
- 5,
326
- 11,
327
- 17,
328
- 23
329
- ],
330
- "patch_size": 16,
331
- "post_process_channels": [
332
  256,
333
  512,
334
  1024,
335
  1024
336
  ],
 
 
 
 
337
  "qkv_bias": true,
338
  "readout_type": "project",
 
 
 
 
 
 
 
339
  "semantic_loss_ignore_index": 255,
340
  "torch_dtype": "float32",
341
- "transformers_version": "4.17.0.dev0",
342
  "use_auxiliary_head": true,
343
- "use_batch_norm": true
344
  }
 
4
  ],
5
  "attention_probs_dropout_prob": 0.0,
6
  "auxiliary_loss_weight": 0.4,
7
+ "backbone_out_indices": [
8
+ 5,
9
+ 11,
10
+ 17,
11
+ 23
12
+ ],
13
+ "fusion_hidden_size": 256,
14
+ "head_in_index": -1,
15
  "hidden_act": "gelu",
16
  "hidden_dropout_prob": 0.0,
17
  "hidden_size": 1024,
 
324
  },
325
  "layer_norm_eps": 1e-12,
326
  "model_type": "dpt",
327
+ "neck_hidden_sizes": [
 
 
 
 
 
 
 
 
 
 
328
  256,
329
  512,
330
  1024,
331
  1024
332
  ],
333
+ "num_attention_heads": 16,
334
+ "num_channels": 3,
335
+ "num_hidden_layers": 24,
336
+ "patch_size": 16,
337
  "qkv_bias": true,
338
  "readout_type": "project",
339
+ "reassemble_factors": [
340
+ 4,
341
+ 2,
342
+ 1,
343
+ 0.5
344
+ ],
345
+ "semantic_classifier_dropout": 0.1,
346
  "semantic_loss_ignore_index": 255,
347
  "torch_dtype": "float32",
348
+ "transformers_version": "4.18.0.dev0",
349
  "use_auxiliary_head": true,
350
+ "use_batch_norm_in_fusion_residual": true
351
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9592df53daf622c642aeadb6a0808bbe79befa811246dc10018eb8cf4f4ca27
3
- size 1371353129
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb9106382231b44f797e364a296a98976bf9435b47eecdab2c032b61e6ec7441
3
+ size 1371354793