Transformers
PyTorch
English
bridgetower
Inference Endpoints
shaoyent's picture
Update config.json
52dfd71
raw
history blame
1.11 kB
{
"cache_dir":"/tmp",
"drop_rate":0.1,
"freeze_RoBERTa":false,
"freeze_ViT":false,
"freeze_layer_count_roberta":false,
"freeze_layer_count_vit":false,
"head_hidden_scale":2,
"hidden_size":768,
"image_size":288,
"input_text_embed_size":768,
"layer_norm_eps":1e-5,
"link_tower_shared":false,
"link_tower_type":"add",
"log_dir":"log_dir",
"max_text_len":50,
"mlp_ratio":4,
"model_type":"bridgetower",
"num_heads":12,
"num_layers":6,
"num_nodes":1,
"only_load_cross_modal_from_meter":false,
"patch_size":16,
"resolution_before":224,
"stop_gradient":false,
"task_head_layers":2,
"test_only":false,
"tokenizer":"roberta-base",
"unfreeze_RoBERTa_attention":false,
"unfreeze_RoBERTa_embeddings":false,
"unfreeze_RoBERTa_encoder":false,
"unfreeze_RoBERTa_layernorm":false,
"unfreeze_ViT_attention":false,
"unfreeze_ViT_layernorm":false,
"vit":"ViT-B/16",
"vit_layernorm_init_from_vit":false,
"vit_layernorm_shared":true,
"vit_remove_last":false,
"vocab_size":50265
}