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

[rllib] ModelV2 support for pytorch (#5249)

Browse Source
This commit is contained in:
Eric Liang
2019-07-25 11:02:53 -07:00
committed by GitHub
parent 40395acadf
commit bf9199ad77
20 changed files with 222 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/model.py
+3 -1
View File
@@ -14,10 +14,12 @@ from ray.rllib.utils import try_import_tf
tf = try_import_tf()
@PublicAPI
# Deprecated: use TFModelV2 instead
class Model(object):
"""Defines an abstract network model for use with RLlib.
This class is deprecated: please use TFModelV2 instead.
Models convert input tensors to a number of output features. These features
can then be interpreted by ActionDistribution classes to determine
e.g. agent action values.