Esempio n. 1
0
 def _thunk():
     env = gym.make(**env_base)
     env = NoopResetEnv(env, noop_max=30)
     env = MaxAndSkipEnv(env, skip=4)
     env = RewardCollector(env)
     env = EpisodicLifeEnv(env)
     env = ClipRewardEnv(env)
     env = WarpFrame(env)
     env = ScaledFloatFrame(env)
     env = TransposeImage(env)
     env = UnrealEnvBaseWrapper(env)
     return env
Esempio n. 2
0
 def wrap(env):
     env = RewardCollector(env)
     env = TransposeImage(env)
     env = ScaledFloatFrame(env)
     env = UnrealEnvBaseWrapper(env)
     return env