Python wrap_dqn示例

示例#1

文件： train.py 项目： behzadanksu/rl-attack

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

示例#2

文件： train.py 项目： elytopia/Info-Sec-Hammerer-and-Brunori

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

示例#3

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

示例#4

文件： enjoy-adv.py 项目： limin24kobe/cleverhans

def make_env(game_name):
    env = gym.make(game_name + "NoFrameskip-v4")
    env = SimpleMonitor(env)
    env = wrap_dqn(env)
    return env