def make_env(game_name):
env = gym.make(game_name + "NoFrameskip-v4")
monitored_env = SimpleMonitor(
env
) # puts rewards and number of steps in info, before environment is wrapped
env = wrap_dqn(
monitored_env
) # applies a bunch of modification to simplify the observation space (downsample, make b/w)
return env, monitored_env
def make_env(game_name):
env = gym.make(game_name + "NoFrameskip-v4")
monitored_env = SimpleMonitor(env)
env = wrap_dqn(monitored_env)
return env, monitored_env
def make_env(game_name):
env = gym.make(game_name + "NoFrameskip-v4")
monitored_env = SimpleMonitor(env)
env = wrappers.Monitor(env, './videos/' + str(time()) + '/')
env = wrap_dqn(monitored_env)
return env, monitored_env