hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[rllib] Restore TF soft placement config to fix multi-GPU optimizer (#2395)

Browse source
This commit is contained in:
Eric Liang 2018-07-13 10:34:37 +02:00 • committed by Richard Liaw
parent d6af50785e
commit 62f84d2f07
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/ray/rllib/agents/agent.py
View file

@ -46,6 +46,7 @@ COMMON_CONFIG = {
"gpu_options": {
"allow_growth": True,
},
"allow_soft_placement": True, # required by PPO multi-gpu
},
# Whether to LZ4 compress observations
"compress_observations": False,