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

[rllib] Refactor pytorch custom model support (#3634)

Browse Source
This commit is contained in:
Eric Liang
2019-01-03 13:48:33 +08:00
committed by GitHub
parent b6bcd18d65
commit 47d36d7bd6
19 changed files with 402 additions and 240 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/models/model.py
+5 -2
View File
@@ -160,11 +160,14 @@ class Model(object):
self._num_outputs, shape))
def _restore_original_dimensions(input_dict, obs_space):
def _restore_original_dimensions(input_dict, obs_space, tensorlib=tf):
if hasattr(obs_space, "original_space"):
return dict(
input_dict,
obs=_unpack_obs(input_dict["obs"], obs_space.original_space))
obs=_unpack_obs(
input_dict["obs"],
obs_space.original_space,
tensorlib=tensorlib))
return input_dict