[rllib] Include config dicts in the sphinx docs (#3064)

This commit is contained in:
Eric Liang
2018-10-16 15:55:11 -07:00
committed by GitHub
parent 64e5eb305e
commit a9e454f6fd
26 changed files with 236 additions and 137 deletions
+3 -7
View File
@@ -8,6 +8,7 @@ from ray.rllib.utils import merge_dicts
from ray.rllib.optimizers import SyncSamplesOptimizer, LocalMultiGPUOptimizer
from ray.tune.trial import Resources
# __sphinx_doc_begin__
DEFAULT_CONFIG = with_common_config({
# If true, use the Generalized Advantage Estimator (GAE)
# with a value function, see https://arxiv.org/pdf/1506.02438.pdf.
@@ -53,15 +54,10 @@ DEFAULT_CONFIG = with_common_config({
"observation_filter": "MeanStdFilter",
# Use the sync samples optimizer instead of the multi-gpu one
"simple_optimizer": False,
# Override model config
"model": {
# Whether to use LSTM model
"use_lstm": False,
# Max seq length for LSTM training.
"max_seq_len": 20,
},
})
# __sphinx_doc_end__
class PPOAgent(Agent):
"""Multi-GPU optimized implementation of PPO in TensorFlow."""