Update config.json

config.json

@@ -1,9 +1,5 @@
 {
     "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,
@@ -21,26 +17,19 @@
     "resolution_before":224,
     "stop_gradient":false,
     "tie_word_embeddings":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_embed_dim":512,
-    "vit_layers":12,
+    "vit_num_hidden_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,
-    "classifier_dropout": null,
+    "vit_intermediate_size":512,
+    "vit_hidden_size":768,
     "vocab_size":50265,
     "text_config_dict": null,
     "text_config":{
       "architectures": ["BridgeTowerTextModel"],
+      "classifier_dropout": null,
       "vocab_size": 50265,
       "hidden_size": 768,
       "num_hidden_layers": 12,