mirror of
https://github.com/vale981/ray
synced 2025-03-12 14:16:39 -04:00
24976ef23a
Why are these changes needed? Also: Add validation to make sure multi-gpu and micro-batch is not used together. Update A2C learning test to hit the microbatching branch. Minor comment updates.
14 lines
315 B
YAML
14 lines
315 B
YAML
cartpole-a2c-fake-gpus:
|
|
env: CartPole-v0
|
|
run: A2C
|
|
stop:
|
|
episode_reward_mean: 150
|
|
training_iteration: 200
|
|
config:
|
|
# Works for both torch and tf.
|
|
framework: tf
|
|
num_workers: 0
|
|
lr: 0.001
|
|
# Fake 2 GPUs.
|
|
num_gpus: 2
|
|
_fake_gpus: true
|