Spaces:
Running
on
Zero
Running
on
Zero
File size: 745 Bytes
2d9a728 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
VisionEncoders = dict()
TextEncoders = dict()
TextEncoders["bert"] = dict(
name="bert_base",
pretrained="bert-base-uncased",
config="configs/config_bert.json",
d_model=768,
fusion_layer=9,
)
TextEncoders["bert_large"] = dict(
name="bert_large",
pretrained="bert-large-uncased",
config="configs/config_bert_large.json",
d_model=1024,
fusion_layer=19,
)
TextEncoders["med_bert"] = dict(
name="med_bert_base",
pretrained="bert-base-uncased",
config="configs/med_config.json",
d_model=768,
)
TextEncoders["med_bert_large"] = dict(
name="med_bert_large",
pretrained="bert-base-uncased", # not a bug, it just follows BLIP.
config="configs/med_large_config.json",
d_model=768
) |