Add autoshaping button press curriculum

Kinds-of-Intelligence-CFI · Sep 6, 2023 · 5a12a87 · 5a12a87
1 parent b717bfb
commit 5a12a87
Showing 1 changed file with 74 additions and 8 deletions.
diff --git a/configs/paper/buttonPressTask/buttonPressGreenCurriculum.yml b/configs/paper/buttonPressTask/buttonPressGreenCurriculum.yml
@@ -1,6 +1,51 @@
 !ArenaConfig
 arenas:
   0: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 15, y: 0, z: 20}
+      rotations: [90]
+    - !Item
+      name: GoodGoal
+      positions:
+      - !Vector3 {x: 20, y: 0, z: 20}
+      sizes:
+      - !Vector3 {x: 1, y: 1, z: 1}
+  1: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 2, y: 0, z: 20}
+      rotations: [90]
+    - !Item
+      name: GoodGoal
+      positions:
+      - !Vector3 {x: 20, y: 0, z: 20}
+      sizes:
+      - !Vector3 {x: 1, y: 1, z: 1}
+  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: 20, y: 0, z: 20}
+      sizes:
+      - !Vector3 {x: 1, y: 1, z: 1}
+  3: !Arena
     pass_mark: 0
     t: 500
     items:
@@ -9,6 +54,21 @@ arenas:
       positions:
       - !Vector3 {x: 2, y: 0, z: -1}
       rotations: [90]
+    - !Item
+      name: GoodGoal
+      positions:
+      - !Vector3 {x: 20, y: 0, z: -1}
+      sizes:
+      - !Vector3 {x: 1, y: 1, z: 1}
+  4: !Arena
+    pass_mark: 0
+    t: 500
+    items:
+    - !Item
+      name: Agent
+      positions:
+      - !Vector3 {x: 15, y: 0, z: 20}
+      rotations: [90]
     - !Item
       name: Pillar-Button
       positions:
@@ -18,27 +78,27 @@ arenas:
       rewardNames: ["GoodGoal", "BadGoal", "GoodGoalMulti"]
       rewardWeights: [1, 0, 0]
       spawnProbability: 1.0
-      rewardSpawnPos: !Vector3 {x: 30, y: 0, z: 35}
-  1: !Arena
+      rewardSpawnPos: !Vector3 {x: 20, y: 0, z: 35}
+  5: !Arena
     pass_mark: 0
     t: 500
     items:
     - !Item
       name: Agent
       positions:
-      - !Vector3 {x: 2, y: 0, z: -1}
+      - !Vector3 {x: 2, y: 0, z: 20}
       rotations: [90]
     - !Item
       name: Pillar-Button
       positions:
-      - !Vector3 {x: 5, y: 0, z: 20}
+      - !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}
-  2: !Arena
+      rewardSpawnPos: !Vector3 {x: 20, y: 0, z: 35}
+  6: !Arena
     pass_mark: 0
     t: 500
     items:
@@ -48,6 +108,12 @@ arenas:
       - !Vector3 {x: 2, y: 0, z: -1}
       rotations: [90]
     - !Item
-      name: GoodGoal
+      name: Pillar-Button
       positions:
-      - !Vector3 {x: 5, y: 0, z: -1}
+      - !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: 20, y: 0, z: 35}