Exemplos de set_global_seeds em Python

Exemplo n.º 1

Arquivo: gym.py Projeto: YuhangSong/pytorch-dqn

def get_ram_env(env, seed):
    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = '/tmp/gym-results'
    env = wrappers.Monitor(env, expt_dir, force=True)
    env = wrap_deepmind_ram(env)

    return env

Exemplo n.º 2

Arquivo: gym.py Projeto: kazizzad/GATS

def get_env_by_id(env_id, seed, skip_f=4, clip_max=False):
    env = gym.make(env_id)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'tmp/gym-results'
    env = wrap_deepmind_nomaxclip(env, skip_f)

    return env

Exemplo n.º 3

Arquivo: gym.py Projeto: devloop0/aa228-stargunner

def get_env(task, seed, monitor=True):
    env = gym.make(task)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = "tmp/gym-results"
    if monitor:
        env = wrappers.Monitor(env, expt_dir, force=True)
    env = wrap_deepmind(env)

    return env

Exemplo n.º 4

Arquivo: gym.py Projeto: YuhangSong/pytorch-dqn

def get_env(env_id, seed):

    env = gym.make(env_id)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'tmp/gym-results'
    env = wrappers.Monitor(env, expt_dir, force=True)
    env = wrap_deepmind(env)

    return env

Exemplo n.º 5

Arquivo: gym.py Projeto: viralparekh/cvitmlss17

def get_env(task, seed):
    env_id = task.env_id

    env = gym.make(env_id)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'results/pong'
    env = wrappers.Monitor(env, expt_dir, force=True)
    env = wrap_deepmind(env)

    return env

Exemplo n.º 6

def get_env(seed):
    #     env_id = task.env_id

    env = gym.make('MsPacman-v0')

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'tmp/gym-results'
    env = wrappers.Monitor(env, expt_dir, force=True)
    env = wrap_deepmind(env)

    return env

Exemplo n.º 7

Arquivo: my_gym.py Projeto: SSARCandy/pytorch-dqn

def get_ram_env(env, seed):
    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = '/tmp/gym-results'
    env = wrappers.Monitor(
        env,
        expt_dir,
        force=True,
        video_callable=lambda episode_id: episode_id % 100 == 0)
    env = wrap_deepmind_ram(env)

    return env

Exemplo n.º 8

Arquivo: gym.py Projeto: YSLIU627/ML-project-2020

def get_env(env, seed):
    # env_id = task.env_id

    # env = gym.make(env_id)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'tmp/gym-results'
    env = wrappers.Monitor(env, expt_dir, force=True)
    print(env.observation_space)
    print(env.spec.id)
    env = wrap_deepmind(env)
    print("get_env")
    return env

Exemplo n.º 9

Arquivo: my_gym.py Projeto: SSARCandy/pytorch-dqn

def get_env(task, seed):
    env_id = task.env_id

    env = gym.make(env_id)

    set_global_seeds(seed)
    env.seed(seed)

    expt_dir = 'tmp/gym-results'
    env = wrappers.Monitor(
        env,
        expt_dir,
        force=True,
        video_callable=lambda episode_id: episode_id % 100 == 0)
    env = wrap_deepmind(env)

    return env