setup ViLA experiments for coffee

Author: NishanthJKumar

scripts/cluster_utils.py

@@ -134,7 +134,8 @@ def generate_run_configs(config_filename: str,
                                              train_refinement_estimator,
                                              start_seed, num_seeds)
                 else:
-                    for seed in range(start_seed, start_seed + num_seeds):
+                    # HACK! hardcoding seeds for now!
+                    for seed in [0, 2, 3, 4, 7]: #range(start_seed, start_seed + num_seeds):
                         yield SingleSeedRunConfig(experiment_id, approach, env,
                                                   run_args, run_flags, use_gpu,
                                                   use_mujoco,

scripts/configs/pred_invention_vlm.yaml

@@ -37,14 +37,14 @@ APPROACHES:
   #   FLAGS:
   #     offline_data_method: demo_with_vlm_imgs
   #     vlm_predicate_vision_api_generate_ground_atoms: True
-  # vila-with-fewshot:
-  #   NAME: "vlm_open_loop"
-  #   FLAGS:
-  #     vlm_open_loop_use_training_demos: True
-  # vila-pure:
-  #   NAME: "vlm_open_loop"
-  #   FLAGS:
-  #     vlm_open_loop_use_training_demos: False
+  vila-with-fewshot:
+    NAME: "vlm_open_loop"
+    FLAGS:
+      vlm_open_loop_use_training_demos: True
+  vila-pure:
+    NAME: "vlm_open_loop"
+    FLAGS:
+      vlm_open_loop_use_training_demos: False
 
 ENVS:
   # burger_no_move_more_stacks:
@@ -253,8 +253,9 @@ ENVS:
   #     grammar_search_early_termination_heuristic_thresh: 100
 
 ARGS:
-  []
+  # []
   # - "debug"
+  - "load_data"
 FLAGS: 
   vlm_model_name: gpt-4o
   num_test_tasks: 10