mirror of
https://github.com/vale981/ray
synced 2025-03-10 13:26:39 -04:00
18 lines
485 B
YAML
18 lines
485 B
YAML
|
memory-leak-test-impala:
|
||
|
|
stop:
|
||
|
|
timesteps_total: 150000
|
||
|
|
env:
|
||
|
|
ray.rllib.examples.env.random_env.RandomLargeObsSpaceEnv
|
||
|
|
run: IMPALA
|
||
|
|
config:
|
||
|
|
# Works for both torch and tf.
|
||
|
|
framework: tf
|
||
|
|
# Switch off np.random, which is known to have memory leaks.
|
||
|
|
env_config:
|
||
|
|
config:
|
||
|
|
static_samples: true
|
||
|
|
num_gpus: 0
|
||
|
|
num_workers: 3
|
||
|
|
num_aggregation_workers: 1
|
||
|
|
num_envs_per_worker: 5
|