Python set_global_seedsの例

コード例 #1

ファイル: gym.py プロジェクト: 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

コード例 #2

ファイル: gym.py プロジェクト: 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

コード例 #3

ファイル: gym.py プロジェクト: 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

コード例 #4

ファイル: gym.py プロジェクト: 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

コード例 #5

ファイル: gym.py プロジェクト: 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

コード例 #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

コード例 #7

ファイル: my_gym.py プロジェクト: 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

コード例 #8

ファイル: gym.py プロジェクト: 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

コード例 #9

ファイル: my_gym.py プロジェクト: 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