make changes necessary to prevent crashing on supercloud expers (#82)

Co-authored-by: Willie McClinton <[email protected]>
Learning-and-Intelligent-Systems · Nov 2, 2023 · c4e38c0 · c4e38c0
1 parent f278494
commit c4e38c0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/predicators/main.py b/predicators/main.py
@@ -123,7 +123,7 @@ def main() -> None:
     # Run the full pipeline.
     _run_pipeline(env, approach, stripped_train_tasks, offline_dataset)
     script_time = time.perf_counter() - script_start
-    if CFG.env == "behavior":  # pragma: no cover
+    if CFG.env == "behavior" and CFG.behavior_save_video:  # pragma: no cover
         assert isinstance(env, BehaviorEnv)
         task_name = str(CFG.behavior_task_list)[2:-2]
         env.igibson_behavior_env.simulator.viewer.make_video(

diff --git a/scripts/configs/behavior_pick_place_evaluation.yaml b/scripts/configs/behavior_pick_place_evaluation.yaml
@@ -84,7 +84,7 @@ FLAGS:  # general flags
   behavior_override_learned_samplers: True
   behavior_ignore_discover_failures: True
   num_train_tasks: 10
-  num_test_tasks: 0
-START_SEED: 463
-NUM_SEEDS: 2
+  num_test_tasks: 10
+START_SEED: 456
+NUM_SEEDS: 10
 USE_GPU: False