gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

fix atari stack overflow (#412)

Browse Source
This commit is contained in:
shadiendrawis
2019-10-06 18:14:21 +03:00
committed by GitHub
parent 0a712ecc94
commit 5ad5a58350
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/environments/gym_environment.py
View File

@@ -459,7 +459,7 @@ class GymEnvironment(Environment):
def _restart_environment_episode(self, force_environment_reset=False): def _restart_environment_episode(self, force_environment_reset=False):
# prevent reset of environment if there are ale lives left # prevent reset of environment if there are ale lives left
if (self.is_atari_env and self.env.unwrapped.ale.lives() > 0) \ if (self.is_atari_env and self.env.unwrapped.ale.lives() > 0) \
and not force_environment_reset: and not force_environment_reset and self.env.env._elapsed_steps < self.env.env._max_episode_steps:
self.step(self.action_space.default_action) self.step(self.action_space.default_action)
else: else:
self.state = self.env.reset() self.state = self.env.reset()