eperezs commited on
Commit
dae2772
1 Parent(s): 4c649b3

Upload RobertaForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +14 -12
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,14 +1,16 @@
1
  {
2
- "_name_or_path": "mental/mental-bert-base-uncased",
3
  "architectures": [
4
- "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
 
7
  "classifier_dropout": null,
 
8
  "gradient_checkpointing": false,
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
- "hidden_size": 768,
12
  "id2label": {
13
  "0": "Feeling-bad-about-yourself-or-that-you-are-a-failure-or-have-let-yourself-or-your-family-down",
14
  "1": "Feeling-down-depressed-or-hopeless",
@@ -21,7 +23,7 @@
21
  "8": "Trouble-falling-or-staying-asleep-or-sleeping-too-much"
22
  },
23
  "initializer_range": 0.02,
24
- "intermediate_size": 3072,
25
  "label2id": {
26
  "Feeling-bad-about-yourself-or-that-you-are-a-failure-or-have-let-yourself-or-your-family-down": 0,
27
  "Feeling-down-depressed-or-hopeless": 1,
@@ -33,17 +35,17 @@
33
  "Trouble-concentrating-on-things-such-as-reading-the-newspaper-or-watching-television": 7,
34
  "Trouble-falling-or-staying-asleep-or-sleeping-too-much": 8
35
  },
36
- "layer_norm_eps": 1e-12,
37
- "max_position_embeddings": 512,
38
- "model_type": "bert",
39
- "num_attention_heads": 12,
40
- "num_hidden_layers": 12,
41
- "pad_token_id": 0,
42
  "position_embedding_type": "absolute",
43
  "problem_type": "multi_label_classification",
44
  "torch_dtype": "float32",
45
  "transformers_version": "4.41.2",
46
- "type_vocab_size": 2,
47
  "use_cache": true,
48
- "vocab_size": 30522
49
  }
 
1
  {
2
+ "_name_or_path": "AIMH/mental-roberta-large",
3
  "architectures": [
4
+ "RobertaForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
  "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
  "gradient_checkpointing": false,
11
  "hidden_act": "gelu",
12
  "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 1024,
14
  "id2label": {
15
  "0": "Feeling-bad-about-yourself-or-that-you-are-a-failure-or-have-let-yourself-or-your-family-down",
16
  "1": "Feeling-down-depressed-or-hopeless",
 
23
  "8": "Trouble-falling-or-staying-asleep-or-sleeping-too-much"
24
  },
25
  "initializer_range": 0.02,
26
+ "intermediate_size": 4096,
27
  "label2id": {
28
  "Feeling-bad-about-yourself-or-that-you-are-a-failure-or-have-let-yourself-or-your-family-down": 0,
29
  "Feeling-down-depressed-or-hopeless": 1,
 
35
  "Trouble-concentrating-on-things-such-as-reading-the-newspaper-or-watching-television": 7,
36
  "Trouble-falling-or-staying-asleep-or-sleeping-too-much": 8
37
  },
38
+ "layer_norm_eps": 1e-05,
39
+ "max_position_embeddings": 514,
40
+ "model_type": "roberta",
41
+ "num_attention_heads": 16,
42
+ "num_hidden_layers": 24,
43
+ "pad_token_id": 1,
44
  "position_embedding_type": "absolute",
45
  "problem_type": "multi_label_classification",
46
  "torch_dtype": "float32",
47
  "transformers_version": "4.41.2",
48
+ "type_vocab_size": 1,
49
  "use_cache": true,
50
+ "vocab_size": 50265
51
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d28f6c43d1340129e333266571bac139f4834f327d1042e37692510ec25fdbea
3
- size 437980180
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0a52d040e5bb0e06d8827251816d9cfe02bfdcd16bd1f499a62a643b267d056
3
+ size 1421524116