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") #env = SimpleMonitor(env) env = wrappers.Monitor(env, './videos/' + str(time()) + '/') env = wrap_dqn(env) return env
def make_env(game_name): env = gym.make(game_name + "NoFrameskip-v4") env = SimpleMonitor(env) env = wrap_dqn(env) return env