liljacharlotte commited on
Commit
bc92a59
·
verified ·
1 Parent(s): 0419c84

Upload NorbertForSequenceClassification

Browse files
Files changed (3) hide show
  1. config.json +83 -0
  2. configuration_norbert.py +34 -0
  3. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/cluster/work/projects/ec30/liljacs_/models/joint/norpac_titles_42",
3
+ "architectures": [
4
+ "NorbertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "auto_map": {
8
+ "AutoConfig": "configuration_norbert.NorbertConfig",
9
+ "AutoModel": "modeling_norbert.NorbertModel",
10
+ "AutoModelForMaskedLM": "modeling_norbert.NorbertForMaskedLM",
11
+ "AutoModelForMultipleChoice": "modeling_norbert.NorbertForMultipleChoice",
12
+ "AutoModelForQuestionAnswering": "modeling_norbert.NorbertForQuestionAnswering",
13
+ "AutoModelForSequenceClassification": "modeling_norbert.NorbertForSequenceClassification",
14
+ "AutoModelForTokenClassification": "modeling_norbert.NorbertForTokenClassification"
15
+ },
16
+ "hidden_dropout_prob": 0.1,
17
+ "hidden_size": 1024,
18
+ "id2label": {
19
+ "0": "LABEL_0",
20
+ "1": "LABEL_1",
21
+ "2": "LABEL_2",
22
+ "3": "LABEL_3",
23
+ "4": "LABEL_4",
24
+ "5": "LABEL_5",
25
+ "6": "LABEL_6",
26
+ "7": "LABEL_7",
27
+ "8": "LABEL_8",
28
+ "9": "LABEL_9",
29
+ "10": "LABEL_10",
30
+ "11": "LABEL_11",
31
+ "12": "LABEL_12",
32
+ "13": "LABEL_13",
33
+ "14": "LABEL_14",
34
+ "15": "LABEL_15",
35
+ "16": "LABEL_16",
36
+ "17": "LABEL_17",
37
+ "18": "LABEL_18",
38
+ "19": "LABEL_19",
39
+ "20": "LABEL_20",
40
+ "21": "LABEL_21",
41
+ "22": "LABEL_22",
42
+ "23": "LABEL_23",
43
+ "24": "LABEL_24"
44
+ },
45
+ "intermediate_size": 2730,
46
+ "label2id": {
47
+ "LABEL_0": 0,
48
+ "LABEL_1": 1,
49
+ "LABEL_10": 10,
50
+ "LABEL_11": 11,
51
+ "LABEL_12": 12,
52
+ "LABEL_13": 13,
53
+ "LABEL_14": 14,
54
+ "LABEL_15": 15,
55
+ "LABEL_16": 16,
56
+ "LABEL_17": 17,
57
+ "LABEL_18": 18,
58
+ "LABEL_19": 19,
59
+ "LABEL_2": 2,
60
+ "LABEL_20": 20,
61
+ "LABEL_21": 21,
62
+ "LABEL_22": 22,
63
+ "LABEL_23": 23,
64
+ "LABEL_24": 24,
65
+ "LABEL_3": 3,
66
+ "LABEL_4": 4,
67
+ "LABEL_5": 5,
68
+ "LABEL_6": 6,
69
+ "LABEL_7": 7,
70
+ "LABEL_8": 8,
71
+ "LABEL_9": 9
72
+ },
73
+ "layer_norm_eps": 1e-07,
74
+ "max_position_embeddings": 512,
75
+ "num_attention_heads": 16,
76
+ "num_hidden_layers": 24,
77
+ "output_all_encoded_layers": true,
78
+ "position_bucket_size": 32,
79
+ "problem_type": "multi_label_classification",
80
+ "torch_dtype": "float32",
81
+ "transformers_version": "4.47.1",
82
+ "vocab_size": 50000
83
+ }
configuration_norbert.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers.configuration_utils import PretrainedConfig
2
+
3
+
4
+ class NorbertConfig(PretrainedConfig):
5
+ """Configuration class to store the configuration of a `NorbertModel`.
6
+ """
7
+ def __init__(
8
+ self,
9
+ vocab_size=50000,
10
+ attention_probs_dropout_prob=0.1,
11
+ hidden_dropout_prob=0.1,
12
+ hidden_size=768,
13
+ intermediate_size=2048,
14
+ max_position_embeddings=512,
15
+ position_bucket_size=32,
16
+ num_attention_heads=12,
17
+ num_hidden_layers=12,
18
+ layer_norm_eps=1.0e-7,
19
+ output_all_encoded_layers=True,
20
+ **kwargs,
21
+ ):
22
+ super().__init__(**kwargs)
23
+
24
+ self.vocab_size = vocab_size
25
+ self.hidden_size = hidden_size
26
+ self.num_hidden_layers = num_hidden_layers
27
+ self.num_attention_heads = num_attention_heads
28
+ self.intermediate_size = intermediate_size
29
+ self.hidden_dropout_prob = hidden_dropout_prob
30
+ self.attention_probs_dropout_prob = attention_probs_dropout_prob
31
+ self.max_position_embeddings = max_position_embeddings
32
+ self.output_all_encoded_layers = output_all_encoded_layers
33
+ self.position_bucket_size = position_bucket_size
34
+ self.layer_norm_eps = layer_norm_eps
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:699dc7b0b83e8d081856dcf814651024605ec1e9db71f7332b72a3f21bf3c0ce
3
+ size 1417749580