mirror of
https://github.com/vale981/ray
synced 2025-03-11 21:56:39 -04:00
fd0f967d2e
This reverts commit e4ceae19ef.
Reverts #25346
linux://python/ray/tests:test_client_library_integration never fail before this PR.
In the CI of the reverted PR, it also fails (https://buildkite.com/ray-project/ray-builders-pr/builds/34079#01812442-c541-4145-af22-2a012655c128). So high likely it's because of this PR.
And test output failure seems related as well (https://buildkite.com/ray-project/ray-builders-branch/builds/7923#018125c2-4812-4ead-a42f-7fddb344105b)
37 lines
997 B
YAML
37 lines
997 B
YAML
# This can reach 9k reward in 2 hours on a Titan XP GPU
|
|
# with 16 workers and 8 envs per worker.
|
|
halfcheetah-appo:
|
|
env: HalfCheetah-v2
|
|
run: APPO
|
|
stop:
|
|
time_total_s: 10800
|
|
config:
|
|
# Works for both torch and tf.
|
|
framework: tf
|
|
vtrace: True
|
|
gamma: 0.99
|
|
lambda: 0.95
|
|
rollout_fragment_length: 512
|
|
train_batch_size: 4096
|
|
num_workers: 16
|
|
num_gpus: 1
|
|
broadcast_interval: 1
|
|
max_sample_requests_in_flight_per_worker: 1
|
|
num_multi_gpu_tower_stacks: 1
|
|
num_envs_per_worker: 32
|
|
minibatch_buffer_size: 16
|
|
num_sgd_iter: 32
|
|
clip_param: 0.2
|
|
lr_schedule: [
|
|
[0, 0.0005],
|
|
[150000000, 0.000001],
|
|
]
|
|
vf_loss_coeff: 0.5
|
|
entropy_coeff: 0.01
|
|
grad_clip: 0.5
|
|
batch_mode: truncate_episodes
|
|
use_kl_loss: True
|
|
kl_coeff: 1.0
|
|
kl_target: 0.04
|
|
observation_filter: MeanStdFilter
|
|
|