File size: 745 Bytes
a0fcf71 | 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 | {
"library_name": "neurovlm",
"model_type": "neurovlm_projection_heads",
"projection_heads": {
"proj_head_image_infonce": {
"class": "neurovlm.models.ProjHead",
"file": "proj_head_image_infonce.safetensors",
"hidden_dim": 384,
"latent_in_dim": 384,
"latent_out_dim": 384
},
"proj_head_text_infonce": {
"class": "neurovlm.models.ProjHead",
"file": "proj_head_text_infonce.safetensors",
"hidden_dim": 512,
"latent_in_dim": 768,
"latent_out_dim": 384
},
"proj_head_text_mse": {
"class": "neurovlm.models.ProjHead",
"file": "proj_head_text_mse.safetensors",
"hidden_dim": 512,
"latent_in_dim": 768,
"latent_out_dim": 384
}
}
}
|