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

Fix 2Rooms Reward

Browse Source
This commit is contained in:
kngwyu 2020-06-30 13:17:11 +09:00
parent 142b42e34f
commit f23b39067a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mujoco_maze/maze_task.py
View File

@ -150,7 +150,10 @@ class SingleGoalSparse2Rooms(MazeTask):
self.goals = [MazeGoal(np.array([0.0, 4.0 * scale]))] self.goals = [MazeGoal(np.array([0.0, 4.0 * scale]))]
def reward(self, obs: np.ndarray) -> float: def reward(self, obs: np.ndarray) -> float:
return 1.0 if self.termination(obs) else -0.0001 for goal in self.goals:
if goal.neighbor(obs):
return goal.reward_scale
return -0.0001
@staticmethod @staticmethod
def create_maze() -> List[List[MazeCell]]: def create_maze() -> List[List[MazeCell]]: