wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-10 11:40:58 +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/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):