Skip to content
Snippets Groups Projects
Commit 31396ec1 authored by gmollard's avatar gmollard
Browse files

added hardware parameters

parent 9897ed52
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment