Monica95 commited on
Commit
1b2e4f4
·
1 Parent(s): 9f38db2

Training in progress, step 500

Browse files
Files changed (3) hide show
  1. config.json +17 -17
  2. pytorch_model.bin +2 -2
  3. training_args.bin +1 -1
config.json CHANGED
@@ -1,33 +1,33 @@
1
  {
2
- "_name_or_path": "sentence-transformers/all-mpnet-base-v2",
 
3
  "architectures": [
4
- "MPNetForSequenceClassification"
5
  ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "bos_token_id": 0,
8
- "eos_token_id": 2,
9
- "hidden_act": "gelu",
10
- "hidden_dropout_prob": 0.1,
11
- "hidden_size": 768,
12
  "id2label": {
13
  "0": "Depressive",
14
  "1": "Non-Depressive"
15
  },
16
  "initializer_range": 0.02,
17
- "intermediate_size": 3072,
18
  "label2id": {
19
  "Depressive": 0,
20
  "Non-Depressive": 1
21
  },
22
- "layer_norm_eps": 1e-05,
23
- "max_position_embeddings": 514,
24
- "model_type": "mpnet",
25
- "num_attention_heads": 12,
26
- "num_hidden_layers": 12,
27
- "pad_token_id": 1,
28
  "problem_type": "single_label_classification",
29
- "relative_attention_num_buckets": 32,
 
 
 
30
  "torch_dtype": "float32",
31
  "transformers_version": "4.34.1",
32
- "vocab_size": 30527
33
  }
 
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": "Depressive",
13
  "1": "Non-Depressive"
14
  },
15
  "initializer_range": 0.02,
 
16
  "label2id": {
17
  "Depressive": 0,
18
  "Non-Depressive": 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.34.1",
32
+ "vocab_size": 30522
33
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c42b8a4c5d2dfcbce6e3b63396806f4cb2b702f83f91402452392f296e04e5fb
3
- size 438017646
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:494f2419badc276c9d445801ffb25da9945f1326b35778557808a33e6dd111dc
3
+ size 267854570
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c90d52e4fc5ae7998de9c6cc4dde02f5d6a5ab571e9483b84fe3282449a4f447
3
  size 4472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9c7d924972eb3f899d51a0c563bb5d9813502467507c74ba801c6e3ce76045f
3
  size 4472