mirror of
https://github.com/wassname/ray.git
synced 2026-08-01 12:51:09 +08:00
[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:
committed by
Philipp Moritz
parent
df65e87fc7
commit
c30fdb4ab0
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user