From 7a5f25248ec181f88b12fa572de512de31aacb1b Mon Sep 17 00:00:00 2001 From: Peter Schafhalter Date: Thu, 2 Aug 2018 14:29:40 -0700 Subject: [PATCH] [rllib] Improve conv_filters documentation (#2540) * Improve conv_filters documentation * Update catalog.py * Update catalog.py --- python/ray/rllib/models/catalog.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ray/rllib/models/catalog.py b/python/ray/rllib/models/catalog.py index b611d2e31..71e181d10 100644 --- a/python/ray/rllib/models/catalog.py +++ b/python/ray/rllib/models/catalog.py @@ -21,7 +21,8 @@ from ray.rllib.models.multiagentfcnet import MultiAgentFullyConnectedNetwork MODEL_CONFIGS = [ # === Built-in options === - "conv_filters", # Filter configuration + # Filter config. List of [out_channels, kernel, stride] for each filter + "conv_filters", "conv_activation", # Nonlinearity for built-in convnet "fcnet_activation", # Nonlinearity for fully connected net (tanh, relu) "fcnet_hiddens", # Number of hidden layers for fully connected net