wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-06 13:31:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/visionnet.py
+2 -2
View File
@@ -17,7 +17,7 @@ class VisionNetwork(Model):
if not filters:
filters = get_filter_config(options)
activation = get_activation_fn(options.get("conv_activation", "relu"))
activation = get_activation_fn(options.get("conv_activation"))
with tf.name_scope("vision_net"):
for i, (out_size, kernel, stride) in enumerate(filters[:-1], 1):
@@ -57,7 +57,7 @@ def get_filter_config(options):
[32, [4, 4], 2],
[256, [11, 11], 1],
]
dim = options.get("dim", 84)
dim = options.get("dim")
if dim == 84:
return filters_84x84
elif dim == 42: