Add configuration to control episode iterator by episode ID. (#1935)

facebookresearch · May 2, 2024 · 9b29006 · 9b29006
1 parent 42ea4dc
commit 9b29006
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/hitl/rearrange_v2/config/rearrange_v2.yaml b/examples/hitl/rearrange_v2/config/rearrange_v2.yaml
@@ -10,9 +10,12 @@ habitat:
   # various config args to ensure the episode never ends
   environment:
     max_episode_steps: 0
+    # Ensure that the episode iterator can be controlled by specifying episode IDs.
     iterator_options:
-      # For the demo, we want to showcase the episodes in the specified order
+      max_scene_repeat_steps: -1
+      max_scene_repeat_episodes: -1
       shuffle: False
+      group_by_scene: False
   task:
     measurements:
       rearrange_cooperate_reward: