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

[rllib] Make the free_logstd param generic (#863)

Browse Source 
* make free log std param generic

* fixes

* fixes
This commit is contained in:
Eric Liang
2017-08-24 12:43:51 -07:00
committed by Philipp Moritz
parent 46641a642f
commit 617bc4d239
6 changed files with 30 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/catalog.py
+2 -5
View File
@@ -48,7 +48,7 @@ class ModelCatalog(object):
"Unsupported args: {} {}".format(action_space, dist_type))
@staticmethod
def get_model(inputs, num_outputs, options=None):
def get_model(inputs, num_outputs, options=dict()):
"""Returns a suitable model conforming to given input and output specs.
Args:
@@ -60,9 +60,6 @@ class ModelCatalog(object):
model (Model): Neural network model.
"""
if options is None:
options = {}
obs_rank = len(inputs.get_shape()) - 1
if obs_rank > 1:
@@ -71,7 +68,7 @@ class ModelCatalog(object):
return FullyConnectedNetwork(inputs, num_outputs, options)
@staticmethod
def ConvolutionalNetwork(inputs, num_outputs, options=None):
def ConvolutionalNetwork(inputs, num_outputs, options=dict()):
return ConvolutionalNetwork(inputs, num_outputs, options)
@staticmethod