Add button press curriculum

Kinds-of-Intelligence-CFI · Aug 28, 2023 · b717bfb · b717bfb
1 parent 01ae144
commit b717bfb
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/configs/paper/buttonPressTask/buttonPressGreenCurriculum.yml b/configs/paper/buttonPressTask/buttonPressGreenCurriculum.yml
@@ -0,0 +1,53 @@
+!ArenaConfig
+arenas:
+  0: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 2, y: 0, z: -1}
+      rotations: [90]
+    - !Item
+      name: Pillar-Button
+      positions:
+      - !Vector3 {x: 20, y: 0, z: 20}
+      moveDurations: [0.1]
+      resetDurations: [1.0]
+      rewardNames: ["GoodGoal", "BadGoal", "GoodGoalMulti"]
+      rewardWeights: [1, 0, 0]
+      spawnProbability: 1.0
+      rewardSpawnPos: !Vector3 {x: 30, y: 0, z: 35}
+  1: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 2, y: 0, z: -1}
+      rotations: [90]
+    - !Item
+      name: Pillar-Button
+      positions:
+      - !Vector3 {x: 5, y: 0, z: 20}
+      moveDurations: [0.1]
+      resetDurations: [1.0]
+      rewardNames: ["GoodGoal", "BadGoal", "GoodGoalMulti"]
+      rewardWeights: [1, 0, 0]
+      spawnProbability: 1.0
+      rewardSpawnPos: !Vector3 {x: 30, y: 0, z: 35}
+  2: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 2, y: 0, z: -1}
+      rotations: [90]
+    - !Item
+      name: GoodGoal
+      positions:
+      - !Vector3 {x: 5, y: 0, z: -1}