wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-15 11:25:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Improve conv_filters documentation (#2540)

Browse Source 
* Improve conv_filters documentation

* Update catalog.py

* Update catalog.py
This commit is contained in:
Peter Schafhalter
2018-08-02 14:29:40 -07:00
committed by Eric Liang
parent f7ec292360
commit 7a5f25248e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/catalog.py
+2 -1
View File
@@ -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