row56 commited on
Commit
5d2a35c
·
verified ·
1 Parent(s): caf29aa

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -4
config.json CHANGED
@@ -1,11 +1,13 @@
1
  {
2
- "_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",
3
  "architectures": ["ProtoForMultiLabelClassification"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "classifier_dropout": null,
6
  "hidden_act": "gelu",
7
  "hidden_dropout_prob": 0.1,
8
- "hidden_size": 768,
 
 
9
  "initializer_range": 0.02,
10
  "intermediate_size": 3072,
11
  "layer_norm_eps": 1e-12,
@@ -18,5 +20,5 @@
18
  "transformers_version": "4.45.2",
19
  "type_vocab_size": 1,
20
  "use_cache": true,
21
- "vocab_size": 250002
22
- }
 
1
  {
2
+ "pretrained_model_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext",
3
  "architectures": ["ProtoForMultiLabelClassification"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "classifier_dropout": null,
6
  "hidden_act": "gelu",
7
  "hidden_dropout_prob": 0.1,
8
+ "num_labels": 519,
9
+ "reduce_hidden_size": 256,
10
+ "hidden_size": 256,
11
  "initializer_range": 0.02,
12
  "intermediate_size": 3072,
13
  "layer_norm_eps": 1e-12,
 
20
  "transformers_version": "4.45.2",
21
  "type_vocab_size": 1,
22
  "use_cache": true,
23
+ "vocab_size": 250002,
24
+ }