[rllib] PPO Thread Limit (#1568)

2026-06-28 19:01:10 +08:00 · 2018-02-26 22:22:05 -08:00
parent f43328f332
commit b79597dc00
1 changed files with 2 additions and 0 deletions
@@ -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,