dodox/mujoco_maze
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Four Rooms] Change the start position

Browse Source
This commit is contained in:
kngwyu 2020-09-07 17:44:57 +09:00
parent b8850f670b
commit 3c3a145de2
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mujoco_maze/maze_task.py
View File

@ -200,7 +200,7 @@ class GoalReward4Rooms(MazeTask):
def __init__(self, scale: float) -> None: def __init__(self, scale: float) -> None:
super().__init__(scale) super().__init__(scale)
self.goals = [MazeGoal(np.array([6.0 * scale, 6.0 * scale]))] self.goals = [MazeGoal(np.array([6.0 * scale, -6.0 * scale]))]
def reward(self, obs: np.ndarray) -> float: def reward(self, obs: np.ndarray) -> float:
for goal in self.goals: for goal in self.goals:
@ -213,13 +213,13 @@ class GoalReward4Rooms(MazeTask):
E, B, R = MazeCell.EMPTY, MazeCell.BLOCK, MazeCell.ROBOT E, B, R = MazeCell.EMPTY, MazeCell.BLOCK, MazeCell.ROBOT
return [ return [
[B, B, B, B, B, B, B, B, B], [B, B, B, B, B, B, B, B, B],
[B, R, E, E, B, E, E, E, B], [B, E, E, E, B, E, E, E, B],
[B, E, E, E, E, E, E, E, B], [B, E, E, E, E, E, E, E, B],
[B, E, E, E, B, E, E, E, B], [B, E, E, E, B, E, E, E, B],
[B, B, E, B, B, B, E, B, B], [B, B, E, B, B, B, E, B, B],
[B, E, E, E, B, E, E, E, B], [B, E, E, E, B, E, E, E, B],
[B, E, E, E, E, E, E, E, B], [B, E, E, E, E, E, E, E, B],
[B, E, E, E, B, E, E, E, B], [B, R, E, E, B, E, E, E, B],
[B, B, B, B, B, B, B, B, B], [B, B, B, B, B, B, B, B, B],
] ]
@ -232,7 +232,7 @@ class SubGoal4Rooms(GoalReward4Rooms):
def __init__(self, scale: float) -> None: def __init__(self, scale: float) -> None:
super().__init__(scale) super().__init__(scale)
self.goals += [ self.goals += [
MazeGoal(np.array([0.0 * scale, 6.0 * scale]), 0.5, GREEN), MazeGoal(np.array([0.0 * scale, -6.0 * scale]), 0.5, GREEN),
MazeGoal(np.array([6.0 * scale, 0.0 * scale]), 0.5, GREEN), MazeGoal(np.array([6.0 * scale, 0.0 * scale]), 0.5, GREEN),
] ]

BIN
screenshots/Point4Rooms.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 79 KiB

After

Width:  |  Height:  |  Size: 93 KiB