wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-12 12:05:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Expose original Atari game reward

Browse Source
This commit is contained in:
Shangtong Zhang
2017-12-23 20:52:17 -07:00
parent 69b7ea53cc
commit 964734755f
9 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
component/atari_wrapper.py
-5
View File
@@ -176,11 +176,6 @@ class ProcessFrame(gym.Wrapper):
def _reset(self):
return self.process_fn(self.env.reset())
class ClippedRewardsWrapper(gym.Wrapper):
def _step(self, action):
obs, reward, done, info = self.env.step(action)
return obs, np.sign(reward), done, info
class NormalizeFrame(gym.Wrapper):
def __init__(self, env=None):
super(NormalizeFrame, self).__init__(env)
component/task.py
+1 -2
View File
@@ -70,8 +70,7 @@ class PixelAtari(BasicTask):
env = MaxAndSkipEnv(env, skip=frame_skip)
if 'FIRE' in env.unwrapped.get_action_meanings():
env = FireResetEnv(env)
env = ProcessFrame(env, frame_size)
self.env = ClippedRewardsWrapper(env)
self.env = ProcessFrame(env, frame_size)
self.action_dim = self.env.action_space.n
def normalize_state(self, state):