Update configs/kosmos_ph_oxe-pretrain.json
Browse files
configs/kosmos_ph_oxe-pretrain.json
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
{
|
2 |
"robovlm_name": "RoboKosMos",
|
3 |
"parent": null,
|
4 |
-
"task_name": "calvin_finetune",
|
5 |
"model": "kosmos",
|
6 |
"model_url": "https://huggingface.co/microsoft/kosmos-2-patch14-224",
|
7 |
"seq_len": 1,
|
@@ -133,7 +132,24 @@
|
|
133 |
"shuffle_buffer_size": 10000,
|
134 |
"train": false
|
135 |
},
|
136 |
-
"
|
137 |
-
"
|
138 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
139 |
}
|
|
|
1 |
{
|
2 |
"robovlm_name": "RoboKosMos",
|
3 |
"parent": null,
|
|
|
4 |
"model": "kosmos",
|
5 |
"model_url": "https://huggingface.co/microsoft/kosmos-2-patch14-224",
|
6 |
"seq_len": 1,
|
|
|
132 |
"shuffle_buffer_size": 10000,
|
133 |
"train": false
|
134 |
},
|
135 |
+
"raw_config_path": "configs/kosmos/oxe_new/finetune_kosmos_cont-lstm-post_full-ft_text_vision_wd=0_hist=16_act=10_use-hand_aug-shift_act-norm_lr-2e-5_oxe.json",
|
136 |
+
"config": "configs/kosmos/oxe_new/finetune_kosmos_cont-lstm-post_full-ft_text_vision_wd=0_hist=16_act=10_use-hand_aug-shift_act-norm_lr-2e-5_oxe.json",
|
137 |
+
"gpus": 8,
|
138 |
+
"num_nodes": 4,
|
139 |
+
"log_dir": "runs/logs/oxe_pretrain/kosmos/calvin_finetune/2024-11-21/01-16",
|
140 |
+
"output_dir": "runs/checkpoints/oxe_pretrain/kosmos/calvin_finetune/2024-11-21/01-16",
|
141 |
+
"data_dir": null,
|
142 |
+
"annotation_file": null,
|
143 |
+
"data_subfolder": null,
|
144 |
+
"task_num": null,
|
145 |
+
"exp_name": "01-16",
|
146 |
+
"use_multi_modal_emb": false,
|
147 |
+
"no_video_pretrained_model": false,
|
148 |
+
"finetune": false,
|
149 |
+
"llm": {
|
150 |
+
"type": null,
|
151 |
+
"n_embd": null,
|
152 |
+
"n_layer": null,
|
153 |
+
"n_head": null
|
154 |
+
}
|
155 |
}
|