jordyvl commited on
Commit
b80b636
1 Parent(s): ba1bd97

Saving best model to hub

Browse files
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
config.json ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "EE_config": {
3
+ "alpha": 0.5,
4
+ "apply_ocr": true,
5
+ "batch_size": 1,
6
+ "benchmark_OCR": false,
7
+ "checkpoint": "",
8
+ "data_parallel": false,
9
+ "dataset": "rvl-cdip_single_10",
10
+ "device": "cuda",
11
+ "downsampling": 0,
12
+ "encoder_layer_strategy": "ramp",
13
+ "epochs": 3,
14
+ "eval_batch_size": 1,
15
+ "eval_start": false,
16
+ "exit_head_num_layers": 2,
17
+ "exit_threshold": -1,
18
+ "exits": [
19
+ "text_avg",
20
+ "vision_avg",
21
+ 1,
22
+ 4,
23
+ 8
24
+ ],
25
+ "forward_signature": [
26
+ "input_ids",
27
+ "attention_mask",
28
+ "bbox",
29
+ "pixel_values",
30
+ "labels",
31
+ "token_type_ids",
32
+ "position_ids",
33
+ "head_mask",
34
+ "inputs_embeds",
35
+ "output_attentions",
36
+ "output_hidden_states",
37
+ "return_dict",
38
+ "kwargs"
39
+ ],
40
+ "gamma": 0.5,
41
+ "get_raw_ocr_data": true,
42
+ "global_threshold": 0.9,
43
+ "gradient_accumulation_steps": 1,
44
+ "inference_strategy": "max_confidence",
45
+ "lowercase": false,
46
+ "lr": 2e-05,
47
+ "model": "EElayoutlmv3",
48
+ "model_weights": "microsoft/layoutlmv3-base",
49
+ "optimizer": "AdamW",
50
+ "seed": 42,
51
+ "temperature": 1,
52
+ "test_dataset": "jordyvl/rvl_cdip_100_examples_per_class",
53
+ "training_strategy": "one_stage_subgraphs",
54
+ "use_images": true,
55
+ "warmup_ratio": 0,
56
+ "weight_decay": 0
57
+ },
58
+ "_name_or_path": "microsoft/layoutlmv3-base",
59
+ "architectures": [
60
+ "LayoutLMv3EEForSequenceClassification"
61
+ ],
62
+ "attention_probs_dropout_prob": 0.1,
63
+ "bos_token_id": 0,
64
+ "classifier_dropout": null,
65
+ "coordinate_size": 128,
66
+ "eos_token_id": 2,
67
+ "exit_config": {
68
+ "encoder_layer_strategy": "ramp",
69
+ "exit_head_num_layers": 2,
70
+ "exits": [
71
+ "text_avg",
72
+ "vision_avg",
73
+ 1,
74
+ 4,
75
+ 8
76
+ ],
77
+ "global_threshold": 0.9,
78
+ "inference_strategy": "max_confidence",
79
+ "training_strategy": "one_stage_subgraphs"
80
+ },
81
+ "has_relative_attention_bias": true,
82
+ "has_spatial_attention_bias": true,
83
+ "hidden_act": "gelu",
84
+ "hidden_dropout_prob": 0.1,
85
+ "hidden_size": 768,
86
+ "id2label": {
87
+ "0": "letter",
88
+ "1": "form",
89
+ "2": "email",
90
+ "3": "handwritten",
91
+ "4": "advertisement",
92
+ "5": "scientific_report",
93
+ "6": "scientific_publication",
94
+ "7": "specification",
95
+ "8": "file_folder",
96
+ "9": "news_article",
97
+ "10": "budget",
98
+ "11": "invoice",
99
+ "12": "presentation",
100
+ "13": "questionnaire",
101
+ "14": "resume",
102
+ "15": "memo"
103
+ },
104
+ "initializer_range": 0.02,
105
+ "input_size": 224,
106
+ "intermediate_size": 3072,
107
+ "label2id": {
108
+ "advertisement": 4,
109
+ "budget": 10,
110
+ "email": 2,
111
+ "file_folder": 8,
112
+ "form": 1,
113
+ "handwritten": 3,
114
+ "invoice": 11,
115
+ "letter": 0,
116
+ "memo": 15,
117
+ "news_article": 9,
118
+ "presentation": 12,
119
+ "questionnaire": 13,
120
+ "resume": 14,
121
+ "scientific_publication": 6,
122
+ "scientific_report": 5,
123
+ "specification": 7
124
+ },
125
+ "layer_norm_eps": 1e-05,
126
+ "max_2d_position_embeddings": 1024,
127
+ "max_position_embeddings": 514,
128
+ "max_rel_2d_pos": 256,
129
+ "max_rel_pos": 128,
130
+ "model_type": "layoutlmv3",
131
+ "num_attention_heads": 12,
132
+ "num_channels": 3,
133
+ "num_hidden_layers": 12,
134
+ "pad_token_id": 1,
135
+ "patch_size": 16,
136
+ "rel_2d_pos_bins": 64,
137
+ "rel_pos_bins": 32,
138
+ "second_input_size": 112,
139
+ "shape_size": 128,
140
+ "text_embed": true,
141
+ "torch_dtype": "float32",
142
+ "transformers_version": "4.28.0.dev0",
143
+ "type_vocab_size": 1,
144
+ "visual_embed": true,
145
+ "vocab_size": 50265
146
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9afdee7b1453e4daae7550ae3308e41d8433b48320a62d11dfeb0b6ad9e78297
3
+ size 515861287
runs/May16_16-19-34_jordy-OMEN/1684246800.3235116/events.out.tfevents.1684246800.jordy-OMEN.76984.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a37719ad8689f5ebe0d666f4e05716f075ea12db2a432215df0ae7c211316f4
3
+ size 5994
runs/May16_16-19-34_jordy-OMEN/events.out.tfevents.1684246800.jordy-OMEN.76984.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35b44d45c01e40dff7645ecce5d65f7e5222910595b9366694b31273183db24a
3
+ size 9054
runs/May16_16-19-34_jordy-OMEN/events.out.tfevents.1684247071.jordy-OMEN.76984.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c4ae19e3d946a0e909046c222b9b6758c70ef07c4c8429ccb15a92072b7295e
3
+ size 658
special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "bos_token": {
4
+ "__type": "AddedToken",
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false
10
+ },
11
+ "cls_token": {
12
+ "__type": "AddedToken",
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "cls_token_box": [
20
+ 0,
21
+ 0,
22
+ 0,
23
+ 0
24
+ ],
25
+ "eos_token": {
26
+ "__type": "AddedToken",
27
+ "content": "</s>",
28
+ "lstrip": false,
29
+ "normalized": true,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ },
33
+ "errors": "replace",
34
+ "mask_token": {
35
+ "__type": "AddedToken",
36
+ "content": "<mask>",
37
+ "lstrip": true,
38
+ "normalized": true,
39
+ "rstrip": false,
40
+ "single_word": false
41
+ },
42
+ "model_max_length": 512,
43
+ "only_label_first_subword": true,
44
+ "pad_token": {
45
+ "__type": "AddedToken",
46
+ "content": "<pad>",
47
+ "lstrip": false,
48
+ "normalized": true,
49
+ "rstrip": false,
50
+ "single_word": false
51
+ },
52
+ "pad_token_box": [
53
+ 0,
54
+ 0,
55
+ 0,
56
+ 0
57
+ ],
58
+ "pad_token_label": -100,
59
+ "sep_token": {
60
+ "__type": "AddedToken",
61
+ "content": "</s>",
62
+ "lstrip": false,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false
66
+ },
67
+ "sep_token_box": [
68
+ 0,
69
+ 0,
70
+ 0,
71
+ 0
72
+ ],
73
+ "special_tokens_map_file": null,
74
+ "tokenizer_class": "LayoutLMv3Tokenizer",
75
+ "trim_offsets": true,
76
+ "unk_token": {
77
+ "__type": "AddedToken",
78
+ "content": "<unk>",
79
+ "lstrip": false,
80
+ "normalized": true,
81
+ "rstrip": false,
82
+ "single_word": false
83
+ }
84
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:622a9290dac610189eafdcbd3b66cc1794e08f11adcf3ff1951ec639798dbccb
3
+ size 3759
vocab.json ADDED
The diff for this file is too large to render. See raw diff