Edwin Dn commited on
Commit
426ed9d
·
verified ·
1 Parent(s): 58fa233

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +21 -19
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,33 +1,35 @@
1
  {
2
- "_name_or_path": "distilbert-base-uncased",
3
- "activation": "gelu",
4
  "architectures": [
5
- "DistilBertForSequenceClassification"
6
  ],
7
- "attention_dropout": 0.1,
8
- "dim": 768,
9
- "dropout": 0.1,
10
- "hidden_dim": 3072,
 
 
11
  "id2label": {
12
- "0": "NO",
13
- "1": "YES"
14
  },
15
  "initializer_range": 0.02,
 
16
  "label2id": {
17
- "NO": 0,
18
- "YES": 1
19
  },
 
20
  "max_position_embeddings": 512,
21
- "model_type": "distilbert",
22
- "n_heads": 12,
23
- "n_layers": 6,
24
  "pad_token_id": 0,
 
25
  "problem_type": "single_label_classification",
26
- "qa_dropout": 0.1,
27
- "seq_classif_dropout": 0.2,
28
- "sinusoidal_pos_embds": false,
29
- "tie_weights_": true,
30
  "torch_dtype": "float32",
31
- "transformers_version": "4.44.2",
 
 
32
  "vocab_size": 30522
33
  }
 
1
  {
2
+ "_name_or_path": "luzalbaposse/HelloBERT",
 
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": "greeting",
14
+ "1": "not_greeting"
15
  },
16
  "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
  "label2id": {
19
+ "greeting": 0,
20
+ "not_greeting": 1
21
  },
22
+ "layer_norm_eps": 1e-12,
23
  "max_position_embeddings": 512,
24
+ "model_type": "bert",
25
+ "num_attention_heads": 12,
26
+ "num_hidden_layers": 12,
27
  "pad_token_id": 0,
28
+ "position_embedding_type": "absolute",
29
  "problem_type": "single_label_classification",
 
 
 
 
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.46.3",
32
+ "type_vocab_size": 2,
33
+ "use_cache": true,
34
  "vocab_size": 30522
35
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d5445955d2f786728ae0057cfac5bd46004f6d46dda069c87bf6065dbfdbe1c1
3
- size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a0e1445aefd760aa04bca11b6486404ceb7d84520346868396147158f2b87f8
3
+ size 437958648