CoolFace
Modelpublic

nevernever69/dit-doclaynet-segmentation

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes15downloads
config.json97 linesDownload Raw Back to root
1{2  "_name_or_path": "/content/dit-base-doclaynet/checkpoint-500",3  "add_fpn": false,4  "architectures": [5    "BeitForSemanticSegmentation"6  ],7  "attention_probs_dropout_prob": 0.0,8  "auxiliary_channels": 256,9  "auxiliary_concat_input": false,10  "auxiliary_loss_weight": 0.4,11  "auxiliary_num_convs": 1,12  "drop_path_rate": 0.1,13  "hidden_act": "gelu",14  "hidden_dropout_prob": 0.0,15  "hidden_size": 768,16  "id2label": {17    "0": "LABEL_0",18    "1": "LABEL_1",19    "2": "LABEL_2",20    "3": "LABEL_3",21    "4": "LABEL_4",22    "5": "LABEL_5",23    "6": "LABEL_6",24    "7": "LABEL_7",25    "8": "LABEL_8",26    "9": "LABEL_9",27    "10": "LABEL_10"28  },29  "image_size": 224,30  "initializer_range": 0.02,31  "intermediate_size": 3072,32  "label2id": {33    "LABEL_0": 0,34    "LABEL_1": 1,35    "LABEL_10": 10,36    "LABEL_2": 2,37    "LABEL_3": 3,38    "LABEL_4": 4,39    "LABEL_5": 5,40    "LABEL_6": 6,41    "LABEL_7": 7,42    "LABEL_8": 8,43    "LABEL_9": 944  },45  "layer_norm_eps": 1e-12,46  "layer_scale_init_value": 0.1,47  "model_type": "beit",48  "num_attention_heads": 12,49  "num_channels": 3,50  "num_hidden_layers": 12,51  "out_features": [52    "stage3",53    "stage5",54    "stage7",55    "stage11"56  ],57  "out_indices": [58    3,59    5,60    7,61    1162  ],63  "patch_size": 16,64  "pool_scales": [65    1,66    2,67    3,68    669  ],70  "reshape_hidden_states": true,71  "semantic_loss_ignore_index": 255,72  "stage_names": [73    "stem",74    "stage1",75    "stage2",76    "stage3",77    "stage4",78    "stage5",79    "stage6",80    "stage7",81    "stage8",82    "stage9",83    "stage10",84    "stage11",85    "stage12"86  ],87  "torch_dtype": "float32",88  "transformers_version": "4.48.3",89  "use_absolute_position_embeddings": true,90  "use_auxiliary_head": true,91  "use_mask_token": true,92  "use_mean_pooling": true,93  "use_relative_position_bias": false,94  "use_shared_relative_position_bias": false,95  "vocab_size": 819296}97