ZarahShibli commited on
Commit
b4b0cdc
1 Parent(s): 6b5c45a

Upload config

Browse files
Files changed (1) hide show
  1. config.json +2 -20
config.json CHANGED
@@ -1,32 +1,15 @@
1
  {
2
- "_name_or_path": "/python-code-comment-classification",
3
- "architecture": "BERTClass",
4
  "architectures": [
5
- "BertModel"
6
  ],
7
  "attention_probs_dropout_prob": 0.1,
8
- "author": "Zarah",
9
  "classifier_dropout": null,
10
- "description": "BERT model fine-tuned for code comment classification",
11
  "hidden_act": "gelu",
12
  "hidden_dropout_prob": 0.1,
13
  "hidden_size": 768,
14
- "id2label": {
15
- "0": "LABEL_0",
16
- "1": "LABEL_1",
17
- "2": "LABEL_2",
18
- "3": "LABEL_3",
19
- "4": "LABEL_4"
20
- },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
- "label2id": {
24
- "LABEL_0": 0,
25
- "LABEL_1": 1,
26
- "LABEL_2": 2,
27
- "LABEL_3": 3,
28
- "LABEL_4": 4
29
- },
30
  "layer_norm_eps": 1e-12,
31
  "max_position_embeddings": 512,
32
  "model_type": "bert",
@@ -34,7 +17,6 @@
34
  "num_hidden_layers": 12,
35
  "pad_token_id": 0,
36
  "position_embedding_type": "absolute",
37
- "torch_dtype": "float32",
38
  "transformers_version": "4.36.0",
39
  "type_vocab_size": 2,
40
  "use_cache": true,
 
1
  {
 
 
2
  "architectures": [
3
+ "BertForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
6
  "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
 
 
 
 
 
 
 
11
  "initializer_range": 0.02,
12
  "intermediate_size": 3072,
 
 
 
 
 
 
 
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
15
  "model_type": "bert",
 
17
  "num_hidden_layers": 12,
18
  "pad_token_id": 0,
19
  "position_embedding_type": "absolute",
 
20
  "transformers_version": "4.36.0",
21
  "type_vocab_size": 2,
22
  "use_cache": true,