hadifar commited on
Commit
5a139b2
·
verified ·
1 Parent(s): 5c3dd26

Upload DistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +7 -1
  2. model.safetensors +2 -2
config.json CHANGED
@@ -8,13 +8,19 @@
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
 
 
 
11
  "initializer_range": 0.02,
 
 
 
12
  "max_position_embeddings": 512,
13
  "model_type": "distilbert",
14
  "n_heads": 12,
15
  "n_layers": 6,
16
  "pad_token_id": 0,
17
- "problem_type": "single_label_classification",
18
  "qa_dropout": 0.1,
19
  "seq_classif_dropout": 0.2,
20
  "sinusoidal_pos_embds": false,
 
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "LABEL_0"
13
+ },
14
  "initializer_range": 0.02,
15
+ "label2id": {
16
+ "LABEL_0": 0
17
+ },
18
  "max_position_embeddings": 512,
19
  "model_type": "distilbert",
20
  "n_heads": 12,
21
  "n_layers": 6,
22
  "pad_token_id": 0,
23
+ "problem_type": "regression",
24
  "qa_dropout": 0.1,
25
  "seq_classif_dropout": 0.2,
26
  "sinusoidal_pos_embds": false,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:02a1a547da64a2483f49bcbb758f2da590f7b70c55ed79f52ad01894c744c69d
3
- size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e40eb502b13fd86c9c9ba58e9b20c8659962471f66c007b8e68db5d4454d4fcd
3
+ size 267829484