Update config.json

config.json

@@ -1,31 +1,27 @@
 {
- 
- "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_act":"gelu",
  "hidden_size":768,
  "image_size":288,
  "input_text_embed_size":768,
+ "input_image_embed_size":768,
+ "is_encoder_decoder":false,
  "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,
+ "num_attention_heads":12,
+ "num_hidden_layers":6,
  "resolution_before":224,
  "stop_gradient":false,
  "task_head_layers":2,
- "test_only":false,
+ "tie_word_embeddings":false,
  "tokenizer":"roberta-base",
  "unfreeze_RoBERTa_attention":false,
  "unfreeze_RoBERTa_embeddings":false,
@@ -33,9 +29,13 @@
  "unfreeze_RoBERTa_layernorm":false,
  "unfreeze_ViT_attention":false,
  "unfreeze_ViT_layernorm":false,
- "vit":"ViT-B/16",
+ "vit_embed_dim":512,
+ "vit_layers":12,
  "vit_layernorm_init_from_vit":false,
  "vit_layernorm_shared":true,
+ "vit_patch_size":16,
  "vit_remove_last":false,
+ "vit_transformer_width":512,
+ "vit_width":768,
  "vocab_size":50265
 }