henrytang785 commited on
Commit
3a74bc1
·
verified ·
1 Parent(s): 597ef3f

Add config for P

Browse files
Files changed (1) hide show
  1. remember-P/config.json +23 -0
remember-P/config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_cfg": {
3
+ "embed_dim": 512,
4
+ "vision_cfg": {
5
+ "timm_model_name": "vit_base_patch16_224",
6
+ "timm_model_pretrained": false,
7
+ "timm_pool": "",
8
+ "timm_proj": "linear",
9
+ "image_size": 224
10
+ },
11
+ "text_cfg": {
12
+ "hf_model_name": "microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract",
13
+ "hf_tokenizer_name": "microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract",
14
+ "hf_proj_type": "mlp",
15
+ "hf_pooler_type": "cls_last_hidden_state_pooler",
16
+ "context_length": 256
17
+ }
18
+ },
19
+ "preprocess_cfg": {
20
+ "mean": [0.48145466, 0.4578275, 0.40821073],
21
+ "std": [0.26862954, 0.26130258, 0.27577711]
22
+ }
23
+ }