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

[rllib] Fix PyTorch initialization (#1961)

Browse Source 
* Fix typo

* Fix A3C PyTorch agent initialization

`registry` needs to be passed as an argument or else the `super` init will
fail.
This commit is contained in:
Alok Singh
2018-05-01 18:39:01 -07:00
committed by Philipp Moritz
parent b55f4a7f04
commit 06a0898af7
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/a3c/shared_torch_policy.py
+2 -2
View File
@@ -17,9 +17,9 @@ class SharedTorchPolicy(TorchPolicy):
other_output = ["vf_preds"]
is_recurrent = False
def __init__(self, ob_space, ac_space, config, **kwargs):
def __init__(self, registry, ob_space, ac_space, config, **kwargs):
super(SharedTorchPolicy, self).__init__(
ob_space, ac_space, config, **kwargs)
registry, ob_space, ac_space, config, **kwargs)
def _setup_graph(self, ob_space, ac_space):
_, self.logit_dim = ModelCatalog.get_action_dist(ac_space)