Update config.js
Browse files
config.js
CHANGED
|
@@ -1,11 +1,6 @@
|
|
| 1 |
-
{
|
| 2 |
-
"
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
"
|
| 6 |
-
"image_encoder": "swin_base_patch4_window7_224",
|
| 7 |
-
"text_encoder": "roberta-base",
|
| 8 |
-
"embedding_dim": 512,
|
| 9 |
-
"t_init": 0.07,
|
| 10 |
-
"b_init": 0.0
|
| 11 |
}
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "lvl",
|
| 3 |
+
"image_backbone": "swin_base_patch4_window7_224.ms_in22k",
|
| 4 |
+
"text_backbone": "roberta-base",
|
| 5 |
+
"embedding_dim": 512
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
}
|