ray/rllib/tuned_examples/dqn/memory-leak-test-dqn.yaml
2022-05-17 13:43:49 +02:00

13 lines
424 B
YAML

memory-leak-test-dqn:
env:
ray.rllib.examples.env.random_env.RandomLargeObsSpaceEnv
run: DQN
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
replay_buffer_config:
capacity: 500 # use small buffer to catch memory leaks