From 31396ec1a8821bcfe88305f70428361f114c12bc Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Tue, 14 May 2019 13:53:03 +0200
Subject: [PATCH] added hardware parameters
---
grid_search_train.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/grid_search_train.py b/grid_search_train.py
index 16b0a42..919c012 100644
--- a/grid_search_train.py
+++ b/grid_search_train.py
@@ -78,6 +78,14 @@ def train(config, reporter):
"policies_to_train": list(policy_graphs.keys())}
agent_config["horizon"] = config['horizon']
+ agent_config["num_workers"] = 0
+ agent_config["num_cpus_per_worker"] = 10
+ agent_config["num_gpus"] = 0.5
+ agent_config["num_gpus_per_worker"] = 0.5
+ agent_config["num_cpus_for_driver"] = 1
+ agent_config["num_envs_per_worker"] = 10
+ agent_config["batch_mode"] = "complete_episodes"
+
ppo_trainer = PPOAgent(env=env_name, config=agent_config)
for i in range(100000 + 2):
--
GitLab