[rllib] Improve conv_filters documentation (ray-project#2540)

* Improve conv_filters documentation * Update catalog.py * Update catalog.py
cerkovni · Aug 2, 2018 · 7a5f252 · 7a5f252
1 parent f7ec292
commit 7a5f252
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ray/rllib/models/catalog.py b/python/ray/rllib/models/catalog.py
@@ -21,7 +21,8 @@
 
 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