annamalai-s commited on
Commit
383c7e4
1 Parent(s): 3d6bbc4

Upload LayoutLMv2ForTokenClassification

Browse files
Files changed (2) hide show
  1. config.json +14 -14
  2. model.safetensors +1 -1
config.json CHANGED
@@ -106,13 +106,13 @@
106
  "hidden_dropout_prob": 0.1,
107
  "hidden_size": 768,
108
  "id2label": {
109
- "0": "LABEL_0",
110
- "1": "LABEL_1",
111
- "2": "LABEL_2",
112
- "3": "LABEL_3",
113
- "4": "LABEL_4",
114
- "5": "LABEL_5",
115
- "6": "LABEL_6"
116
  },
117
  "image_feature_pool_shape": [
118
  7,
@@ -122,13 +122,13 @@
122
  "initializer_range": 0.02,
123
  "intermediate_size": 3072,
124
  "label2id": {
125
- "LABEL_0": 0,
126
- "LABEL_1": 1,
127
- "LABEL_2": 2,
128
- "LABEL_3": 3,
129
- "LABEL_4": 4,
130
- "LABEL_5": 5,
131
- "LABEL_6": 6
132
  },
133
  "layer_norm_eps": 1e-12,
134
  "max_2d_position_embeddings": 1024,
 
106
  "hidden_dropout_prob": 0.1,
107
  "hidden_size": 768,
108
  "id2label": {
109
+ "0": "address",
110
+ "1": "email",
111
+ "2": "number",
112
+ "3": "organization",
113
+ "4": "person",
114
+ "5": "role",
115
+ "6": "website"
116
  },
117
  "image_feature_pool_shape": [
118
  7,
 
122
  "initializer_range": 0.02,
123
  "intermediate_size": 3072,
124
  "label2id": {
125
+ "address": 0,
126
+ "email": 1,
127
+ "number": 2,
128
+ "organization": 3,
129
+ "person": 4,
130
+ "role": 5,
131
+ "website": 6
132
  },
133
  "layer_norm_eps": 1e-12,
134
  "max_2d_position_embeddings": 1024,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6a971cf407938829488b652208e705a55f8735ce3bdf74fc8a7d284b22ce871
3
  size 802065940
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b55a0a202f5cb6706eb3c7e9c35af7a17c35301d37be63a3abf884e49df2ab6
3
  size 802065940