Compare commits

...

1 Commits

Author SHA1 Message Date
Peter Zhokhov
3e13bb8dc8 short-circuit framestack wrapper with size 1 2019-04-05 15:13:59 -07:00

View File

@@ -215,6 +215,7 @@ def wrap_deepmind_retro(env, scale=True, frame_stack=4):
"""
env = WarpFrame(env)
env = ClipRewardEnv(env)
if frame_stack > 1:
env = FrameStack(env, frame_stack)
if scale:
env = ScaledFloatFrame(env)