sreejith8100
commited on
Commit
•
31b8128
1
Parent(s):
3b3a72c
Training in progress, epoch 1
Browse files- config.json +18 -126
- pytorch_model.bin +2 -2
- training_args.bin +1 -1
config.json
CHANGED
@@ -1,152 +1,44 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "microsoft/
|
3 |
"architectures": [
|
4 |
-
"
|
5 |
],
|
6 |
"attention_probs_dropout_prob": 0.1,
|
7 |
-
"convert_sync_batchnorm": true,
|
8 |
-
"coordinate_size": 128,
|
9 |
-
"detectron2_config_args": {
|
10 |
-
"MODEL.ANCHOR_GENERATOR.SIZES": [
|
11 |
-
[
|
12 |
-
32
|
13 |
-
],
|
14 |
-
[
|
15 |
-
64
|
16 |
-
],
|
17 |
-
[
|
18 |
-
128
|
19 |
-
],
|
20 |
-
[
|
21 |
-
256
|
22 |
-
],
|
23 |
-
[
|
24 |
-
512
|
25 |
-
]
|
26 |
-
],
|
27 |
-
"MODEL.BACKBONE.NAME": "build_resnet_fpn_backbone",
|
28 |
-
"MODEL.FPN.IN_FEATURES": [
|
29 |
-
"res2",
|
30 |
-
"res3",
|
31 |
-
"res4",
|
32 |
-
"res5"
|
33 |
-
],
|
34 |
-
"MODEL.MASK_ON": true,
|
35 |
-
"MODEL.PIXEL_STD": [
|
36 |
-
57.375,
|
37 |
-
57.12,
|
38 |
-
58.395
|
39 |
-
],
|
40 |
-
"MODEL.POST_NMS_TOPK_TEST": 1000,
|
41 |
-
"MODEL.RESNETS.ASPECT_RATIOS": [
|
42 |
-
[
|
43 |
-
0.5,
|
44 |
-
1.0,
|
45 |
-
2.0
|
46 |
-
]
|
47 |
-
],
|
48 |
-
"MODEL.RESNETS.DEPTH": 101,
|
49 |
-
"MODEL.RESNETS.NUM_GROUPS": 32,
|
50 |
-
"MODEL.RESNETS.OUT_FEATURES": [
|
51 |
-
"res2",
|
52 |
-
"res3",
|
53 |
-
"res4",
|
54 |
-
"res5"
|
55 |
-
],
|
56 |
-
"MODEL.RESNETS.SIZES": [
|
57 |
-
[
|
58 |
-
32
|
59 |
-
],
|
60 |
-
[
|
61 |
-
64
|
62 |
-
],
|
63 |
-
[
|
64 |
-
128
|
65 |
-
],
|
66 |
-
[
|
67 |
-
256
|
68 |
-
],
|
69 |
-
[
|
70 |
-
512
|
71 |
-
]
|
72 |
-
],
|
73 |
-
"MODEL.RESNETS.STRIDE_IN_1X1": false,
|
74 |
-
"MODEL.RESNETS.WIDTH_PER_GROUP": 8,
|
75 |
-
"MODEL.ROI_BOX_HEAD.NAME": "FastRCNNConvFCHead",
|
76 |
-
"MODEL.ROI_BOX_HEAD.NUM_FC": 2,
|
77 |
-
"MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION": 14,
|
78 |
-
"MODEL.ROI_HEADS.IN_FEATURES": [
|
79 |
-
"p2",
|
80 |
-
"p3",
|
81 |
-
"p4",
|
82 |
-
"p5"
|
83 |
-
],
|
84 |
-
"MODEL.ROI_HEADS.NAME": "StandardROIHeads",
|
85 |
-
"MODEL.ROI_HEADS.NUM_CLASSES": 5,
|
86 |
-
"MODEL.ROI_MASK_HEAD.NAME": "MaskRCNNConvUpsampleHead",
|
87 |
-
"MODEL.ROI_MASK_HEAD.NUM_CONV": 4,
|
88 |
-
"MODEL.ROI_MASK_HEAD.POOLER_RESOLUTION": 7,
|
89 |
-
"MODEL.RPN.IN_FEATURES": [
|
90 |
-
"p2",
|
91 |
-
"p3",
|
92 |
-
"p4",
|
93 |
-
"p5",
|
94 |
-
"p6"
|
95 |
-
],
|
96 |
-
"MODEL.RPN.POST_NMS_TOPK_TRAIN": 1000,
|
97 |
-
"MODEL.RPN.PRE_NMS_TOPK_TEST": 1000,
|
98 |
-
"MODEL.RPN.PRE_NMS_TOPK_TRAIN": 2000
|
99 |
-
},
|
100 |
-
"fast_qkv": true,
|
101 |
-
"gradient_checkpointing": false,
|
102 |
-
"has_relative_attention_bias": true,
|
103 |
-
"has_spatial_attention_bias": true,
|
104 |
-
"has_visual_segment_embedding": false,
|
105 |
"hidden_act": "gelu",
|
106 |
"hidden_dropout_prob": 0.1,
|
107 |
"hidden_size": 768,
|
108 |
"id2label": {
|
109 |
"0": "O",
|
110 |
-
"1": "
|
111 |
-
"2": "
|
112 |
-
"3": "
|
113 |
-
"4": "
|
114 |
-
"5": "
|
115 |
-
"6": "
|
116 |
-
"7": "Random Number"
|
117 |
},
|
118 |
-
"image_feature_pool_shape": [
|
119 |
-
7,
|
120 |
-
7,
|
121 |
-
256
|
122 |
-
],
|
123 |
"initializer_range": 0.02,
|
124 |
"intermediate_size": 3072,
|
125 |
"label2id": {
|
126 |
-
"
|
127 |
-
"
|
128 |
-
"
|
129 |
-
"
|
130 |
-
"
|
131 |
-
"
|
132 |
-
"
|
133 |
-
"Title": 6
|
134 |
},
|
135 |
"layer_norm_eps": 1e-12,
|
136 |
"max_2d_position_embeddings": 1024,
|
137 |
"max_position_embeddings": 512,
|
138 |
-
"
|
139 |
-
"max_rel_pos": 128,
|
140 |
-
"model_type": "layoutlmv2",
|
141 |
"num_attention_heads": 12,
|
142 |
"num_hidden_layers": 12,
|
143 |
"output_past": true,
|
144 |
"pad_token_id": 0,
|
145 |
-
"
|
146 |
-
"rel_pos_bins": 32,
|
147 |
-
"shape_size": 128,
|
148 |
"torch_dtype": "float32",
|
149 |
"transformers_version": "4.34.0",
|
150 |
"type_vocab_size": 2,
|
|
|
151 |
"vocab_size": 30522
|
152 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "microsoft/layoutlm-base-uncased",
|
3 |
"architectures": [
|
4 |
+
"LayoutLMForTokenClassification"
|
5 |
],
|
6 |
"attention_probs_dropout_prob": 0.1,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
"hidden_act": "gelu",
|
8 |
"hidden_dropout_prob": 0.1,
|
9 |
"hidden_size": 768,
|
10 |
"id2label": {
|
11 |
"0": "O",
|
12 |
+
"1": "B-HEADER",
|
13 |
+
"2": "I-HEADER",
|
14 |
+
"3": "B-QUESTION",
|
15 |
+
"4": "I-QUESTION",
|
16 |
+
"5": "B-ANSWER",
|
17 |
+
"6": "I-ANSWER"
|
|
|
18 |
},
|
|
|
|
|
|
|
|
|
|
|
19 |
"initializer_range": 0.02,
|
20 |
"intermediate_size": 3072,
|
21 |
"label2id": {
|
22 |
+
"B-ANSWER": 5,
|
23 |
+
"B-HEADER": 1,
|
24 |
+
"B-QUESTION": 3,
|
25 |
+
"I-ANSWER": 6,
|
26 |
+
"I-HEADER": 2,
|
27 |
+
"I-QUESTION": 4,
|
28 |
+
"O": 0
|
|
|
29 |
},
|
30 |
"layer_norm_eps": 1e-12,
|
31 |
"max_2d_position_embeddings": 1024,
|
32 |
"max_position_embeddings": 512,
|
33 |
+
"model_type": "layoutlm",
|
|
|
|
|
34 |
"num_attention_heads": 12,
|
35 |
"num_hidden_layers": 12,
|
36 |
"output_past": true,
|
37 |
"pad_token_id": 0,
|
38 |
+
"position_embedding_type": "absolute",
|
|
|
|
|
39 |
"torch_dtype": "float32",
|
40 |
"transformers_version": "4.34.0",
|
41 |
"type_vocab_size": 2,
|
42 |
+
"use_cache": true,
|
43 |
"vocab_size": 30522
|
44 |
}
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b06e18d9ffba6f80efeb1ed4c2e70cf46b57894b3ca94dcbbfb2c2606716204b
|
3 |
+
size 450603969
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4091
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ac3a78eed4e730cd6ba4d0a047f23423e4246c3ea0ad20cd278f1755aa7b3ea5
|
3 |
size 4091
|