ray/rllib/tuned_examples/impala/atari-impala-large.yaml

# Runs on a g3.16xl node with 5 m5.24xl workers
# Takes roughly 10 minutes.
atari-impala:
    env:
        grid_search:
            - BreakoutNoFrameskip-v4
            - BeamRiderNoFrameskip-v4
            - QbertNoFrameskip-v4
            - SpaceInvadersNoFrameskip-v4
    run: IMPALA
    stop:
        timesteps_total: 3000000
    config:
        rollout_fragment_length: 50
        train_batch_size: 500
        num_workers: 128
        num_envs_per_worker: 5
        clip_rewards: True
        lr_schedule: [
            [0, 0.0005],
            [20000000, 0.000000000001],
        ]
Application Stress Tests (#3612) 2019-01-16 02:05:16 -08:00			`# Runs on a g3.16xl node with 5 m5.24xl workers`
			`# Takes roughly 10 minutes.`
			`atari-impala:`
			`env:`
			`grid_search:`
			`- BreakoutNoFrameskip-v4`
			`- BeamRiderNoFrameskip-v4`
			`- QbertNoFrameskip-v4`
			`- SpaceInvadersNoFrameskip-v4`
			`run: IMPALA`
			`stop:`
			`timesteps_total: 3000000`
			`config:`
[rllib] Rename sample_batch_size => rollout_fragment_length (#7503) * bulk rename * deprecation warn * update doc * update fig * line length * rename * make pytest comptaible * fix test * fi sys * rename * wip * fix more * lint * update svg * comments * lint * fix use of batch steps 2020-03-14 12:05:04 -07:00			`rollout_fragment_length: 50`
Application Stress Tests (#3612) 2019-01-16 02:05:16 -08:00			`train_batch_size: 500`
			`num_workers: 128`
			`num_envs_per_worker: 5`
			`clip_rewards: True`
			`lr_schedule: [`
			`[0, 0.0005],`
			`[20000000, 0.000000000001],`
			`]`