From b79597dc0063c714d7fa05abb5726e13368b6541 Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Mon, 26 Feb 2018 22:22:05 -0800 Subject: [PATCH] [rllib] PPO Thread Limit (#1568) --- python/ray/rllib/ppo/ppo.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/ray/rllib/ppo/ppo.py b/python/ray/rllib/ppo/ppo.py index 8228764ec..5811b2e60 100644 --- a/python/ray/rllib/ppo/ppo.py +++ b/python/ray/rllib/ppo/ppo.py @@ -41,6 +41,8 @@ DEFAULT_CONFIG = { "device_count": {"CPU": 4}, "log_device_placement": False, "allow_soft_placement": True, + "intra_op_parallelism_threads": 1, + "inter_op_parallelism_threads": 2, }, # Batch size for policy evaluations for rollouts "rollout_batchsize": 1,