[rllib] PPO Thread Limit (ray-project#1568)

rickyyx · Feb 27, 2018 · b79597d · b79597d
1 parent f43328f
commit b79597d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/ray/rllib/ppo/ppo.py b/python/ray/rllib/ppo/ppo.py
@@ -41,6 +41,8 @@
  "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,