[rllib] Code for Supporting Shared Models (#775)

* Code for Supporting Shared Models

* Running (with vnet modification) - needs to be tested for performance

* Small fix for jenkins

* Linting

* linting

* Summaries

* Small refactoring + generalized to more domains

* Addressing changes

* Addressing changes

* Update envs.py

* Addressing changes

* convnet

* final touches

* Merge - new model

* final linting

* Changing iterations back

* Policy option removed, fixed small things

* Nits

* nit

* Linting

* Linting
This commit is contained in:
Richard Liaw
2017-08-03 19:29:01 -07:00
committed by Philipp Moritz
parent df65e87fc7
commit c30fdb4ab0
10 changed files with 197 additions and 29 deletions
+5
View File
@@ -8,6 +8,7 @@ from ray.rllib.models.action_dist import (
Categorical, Deterministic, DiagGaussian)
from ray.rllib.models.fcnet import FullyConnectedNetwork
from ray.rllib.models.visionnet import VisionNetwork
from ray.rllib.models.convnet import ConvolutionalNetwork
class ModelCatalog(object):
@@ -67,6 +68,10 @@ class ModelCatalog(object):
return FullyConnectedNetwork(inputs, num_outputs, options)
@staticmethod
def ConvolutionalNetwork(inputs, num_outputs, options=None):
return ConvolutionalNetwork(inputs, num_outputs, options)
@staticmethod
def get_preprocessor(env_name):
"""Returns a suitable processor for the given environment.