dpt-large-redesign / config.json
nielsr's picture
nielsr HF staff
Add model
645d531
{
"architectures": [
"DPTForDepthEstimation"
],
"attention_probs_dropout_prob": 0.0,
"auxiliary_loss_weight": 0.4,
"channels": 256,
"expand_channels": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"image_size": 384,
"initializer_range": 0.02,
"intermediate_size": 4096,
"layer_norm_eps": 1e-12,
"model_type": "dpt",
"num_attention_heads": 16,
"num_channels": 3,
"num_hidden_layers": 24,
"out_indices": [
5,
11,
17,
23
],
"patch_size": 16,
"post_process_channels": [
256,
512,
1024,
1024
],
"qkv_bias": true,
"readout_type": "project",
"semantic_loss_ignore_index": 255,
"torch_dtype": "float32",
"transformers_version": "4.17.0.dev0",
"use_auxiliary_head": true,
"use_batch_norm": false
}