[rllib] Fix incorrect documentation on how to use custom models #1405

This commit is contained in:
Eric Liang
2018-01-09 18:09:05 -08:00
committed by Richard Liaw
parent a59a9e20af
commit 5a2f85048d
+5 -3
View File
@@ -183,9 +183,11 @@ custom model and preprocessor classes are documented in the
ray.init()
alg = ppo.PPOAgent(env="CartPole-v0", config={
"custom_preprocessor": "my_prep",
"custom_model": "my_model",
"custom_options": {}, # extra options to pass to your classes
"model": {
"custom_preprocessor": "my_prep",
"custom_model": "my_model",
"custom_options": {}, # extra options to pass to your classes
},
})
Using RLlib with Ray.tune