wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-13 04:26:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Remove need to pass around registry (#2250)

Browse Source 
* remove registry

* fix

* too many _

* fix

* cloudpickle

* Update registry.py

* yapf

* fix test

* fix kv check
This commit is contained in:
Eric Liang
2018-06-19 22:47:00 -07:00
committed by GitHub
parent 30684446a6
commit 30f7c08ca7
36 changed files with 202 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/examples/multiagent_mountaincar.py
+2 -2
View File
@@ -8,7 +8,7 @@ from gym.envs.registration import register
import ray
import ray.rllib.ppo as ppo
from ray.tune.registry import get_registry, register_env
from ray.tune.registry import register_env
env_name = "MultiAgentMountainCarEnv"
@@ -51,6 +51,6 @@ if __name__ == '__main__':
"multiagent_shared_model": False,
"multiagent_fcnet_hiddens": [[32, 32]] * 2}
config["model"].update({"custom_options": options})
alg = ppo.PPOAgent(env=env_name, registry=get_registry(), config=config)
alg = ppo.PPOAgent(env=env_name, config=config)
for i in range(1):
alg.train()
python/ray/rllib/examples/multiagent_pendulum.py
+2 -2
View File
@@ -8,7 +8,7 @@ from gym.envs.registration import register
import ray
import ray.rllib.ppo as ppo
from ray.tune.registry import get_registry, register_env
from ray.tune.registry import register_env
env_name = "MultiAgentPendulumEnv"
@@ -51,6 +51,6 @@ if __name__ == '__main__':
"multiagent_shared_model": True,
"multiagent_fcnet_hiddens": [[32, 32]] * 2}
config["model"].update({"custom_options": options})
alg = ppo.PPOAgent(env=env_name, registry=get_registry(), config=config)
alg = ppo.PPOAgent(env=env_name, config=config)
for i in range(1):
alg.train()