wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-09 12:20:09 +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/tune/registry.py
+1 -1
View File
@@ -23,7 +23,7 @@ def register_trainable(name, trainable):
Args:
name (str): Name to register.
trainable (obj): Function or tune.Trainable clsas. Functions must
trainable (obj): Function or tune.Trainable class. Functions must
take (config, status_reporter) as arguments and will be
automatically converted into a class during registration.
"""