emanuelaboros commited on
Commit
f7a403e
·
1 Parent(s): 665dc46

update model small

Browse files
Files changed (4) hide show
  1. config.json +4 -119
  2. label_map.json +1 -1
  3. pytorch_model.bin +2 -2
  4. training_args.bin +1 -1
config.json CHANGED
@@ -1,138 +1,23 @@
1
  {
2
- "_name_or_path": "experiments/model_dbmdz_bert_medium_historic_multilingual_cased_max_sequence_length_512_epochs_5_run_multitask.baseline.False2025/",
3
  "architectures": [
4
- "ExtendedMultitaskTimeModelForTokenClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
- "auto_map": {
8
- "AutoConfig": "configuration_stacked.ImpressoConfig",
9
- "AutoModelForTokenClassification": "modeling_stacked.ExtendedMultitaskTimeModelForTokenClassification"
10
- },
11
  "classifier_dropout": null,
12
- "custom_pipelines": {
13
- "generic-ner": {
14
- "impl": "generic_ner.ExtendedMultitaskTimeModelForTokenClassificationPipeline",
15
- "pt": "AutoModelForTokenClassification"
16
- }
17
- },
18
  "hidden_act": "gelu",
19
  "hidden_dropout_prob": 0.1,
20
  "hidden_size": 512,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 2048,
23
- "label_map": {
24
- "NE-COARSE-LIT": {
25
- "I-pers": 0,
26
- "I-prod": 1,
27
- "B-prod": 2,
28
- "B-loc": 3,
29
- "I-time": 4,
30
- "B-pers": 5,
31
- "B-org": 6,
32
- "B-time": 7,
33
- "I-loc": 8,
34
- "O": 9,
35
- "I-org": 10
36
- },
37
- "NE-FINE-COMP": {
38
- "I-comp.title": 0,
39
- "B-comp.title": 1,
40
- "I-comp.function": 2,
41
- "I-comp.name": 3,
42
- "B-comp.function": 4,
43
- "O": 5,
44
- "B-comp.name": 6
45
- }
46
- },
47
  "layer_norm_eps": 1e-12,
48
  "max_position_embeddings": 512,
49
- "model_type": "stacked_bert",
50
  "num_attention_heads": 8,
51
  "num_hidden_layers": 8,
52
  "pad_token_id": 0,
53
  "position_embedding_type": "absolute",
54
- "pretrained_config": {
55
- "_name_or_path": "dbmdz/bert-medium-historic-multilingual-cased",
56
- "add_cross_attention": false,
57
- "architectures": [
58
- "BertForMaskedLM"
59
- ],
60
- "attention_probs_dropout_prob": 0.1,
61
- "bad_words_ids": null,
62
- "begin_suppress_tokens": null,
63
- "bos_token_id": null,
64
- "chunk_size_feed_forward": 0,
65
- "classifier_dropout": null,
66
- "cross_attention_hidden_size": null,
67
- "decoder_start_token_id": null,
68
- "diversity_penalty": 0.0,
69
- "do_sample": false,
70
- "early_stopping": false,
71
- "encoder_no_repeat_ngram_size": 0,
72
- "eos_token_id": null,
73
- "exponential_decay_length_penalty": null,
74
- "finetuning_task": null,
75
- "forced_bos_token_id": null,
76
- "forced_eos_token_id": null,
77
- "hidden_act": "gelu",
78
- "hidden_dropout_prob": 0.1,
79
- "hidden_size": 512,
80
- "id2label": {
81
- "0": "LABEL_0",
82
- "1": "LABEL_1"
83
- },
84
- "initializer_range": 0.02,
85
- "intermediate_size": 2048,
86
- "is_decoder": false,
87
- "is_encoder_decoder": false,
88
- "label2id": {
89
- "LABEL_0": 0,
90
- "LABEL_1": 1
91
- },
92
- "layer_norm_eps": 1e-12,
93
- "length_penalty": 1.0,
94
- "max_length": 20,
95
- "max_position_embeddings": 512,
96
- "min_length": 0,
97
- "model_type": "bert",
98
- "no_repeat_ngram_size": 0,
99
- "num_attention_heads": 8,
100
- "num_beam_groups": 1,
101
- "num_beams": 1,
102
- "num_hidden_layers": 8,
103
- "num_return_sequences": 1,
104
- "output_attentions": false,
105
- "output_hidden_states": false,
106
- "output_scores": false,
107
- "pad_token_id": 0,
108
- "position_embedding_type": "absolute",
109
- "prefix": null,
110
- "problem_type": null,
111
- "pruned_heads": {},
112
- "remove_invalid_values": false,
113
- "repetition_penalty": 1.0,
114
- "return_dict": true,
115
- "return_dict_in_generate": false,
116
- "sep_token_id": null,
117
- "suppress_tokens": null,
118
- "task_specific_params": null,
119
- "temperature": 1.0,
120
- "tf_legacy_loss": false,
121
- "tie_encoder_decoder": false,
122
- "tie_word_embeddings": true,
123
- "tokenizer_class": null,
124
- "top_k": 50,
125
- "top_p": 1.0,
126
- "torch_dtype": null,
127
- "torchscript": false,
128
- "type_vocab_size": 2,
129
- "typical_p": 1.0,
130
- "use_bfloat16": false,
131
- "use_cache": true,
132
- "vocab_size": 32000
133
- },
134
- "torch_dtype": "float32",
135
- "transformers_version": "4.40.0.dev0",
136
  "type_vocab_size": 2,
137
  "use_cache": true,
138
  "vocab_size": 32000
 
1
  {
 
2
  "architectures": [
3
+ "MultitaskTimeModelForTokenClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
 
 
 
6
  "classifier_dropout": null,
7
+ "dtype": "float32",
 
 
 
 
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 512,
11
  "initializer_range": 0.02,
12
  "intermediate_size": 2048,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
15
+ "model_type": "bert",
16
  "num_attention_heads": 8,
17
  "num_hidden_layers": 8,
18
  "pad_token_id": 0,
19
  "position_embedding_type": "absolute",
20
+ "transformers_version": "4.57.0.dev0",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  "type_vocab_size": 2,
22
  "use_cache": true,
23
  "vocab_size": 32000
label_map.json CHANGED
@@ -1 +1 @@
1
- {"NE-COARSE-LIT": {"I-pers": 0, "I-prod": 1, "B-prod": 2, "B-loc": 3, "I-time": 4, "B-pers": 5, "B-org": 6, "B-time": 7, "I-loc": 8, "O": 9, "I-org": 10}, "NE-FINE-COMP": {"I-comp.title": 0, "B-comp.title": 1, "I-comp.function": 2, "I-comp.name": 3, "B-comp.function": 4, "O": 5, "B-comp.name": 6}}
 
1
+ {"NE-COARSE-LIT": {"B-loc": 0, "B-org": 1, "I-pers": 2, "B-pers": 3, "I-loc": 4, "I-org": 5, "O": 6}, "NE-FINE-COMP": {"B-comp.title": 0, "I-comp.name": 1, "I-comp.title": 2, "I-comp.function": 3, "B-comp.name": 4, "B-comp.function": 5, "O": 6}}
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4693904ab70f0ef7c0249db6c23b0b1e3b2760629d8dafa9010f5ec9feb7de39
3
- size 168604214
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d85dc101a70451930bedc505f6ba1b231f42dc8b84795c5515f43e5d715be0a2
3
+ size 168596022
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cc92dca5d693d80c40bfa708d0ee9551d1f85b832c57710b3edfc72dc86707e1
3
  size 2104
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0eae1fb01475a2434e6ce45356e764e9d567efedff16ac7ee77c63b9a6e4e2c
3
  size 2104