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] Issue 7046 cannot restore keras model from h5 file. (#7482)

Browse Source
This commit is contained in:
Sven Mika
2020-03-23 20:19:30 +01:00
committed by GitHub
parent ee8c9ff732
commit 1138f2ebed
15 changed files with 364 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/trainable.py
+4 -1
View File
@@ -473,13 +473,16 @@ class Trainable:
export model to local directory.
Args:
export_formats (list): List of formats that should be exported.
export_formats (Union[list,str]): Format or list of (str) formats
that should be exported.
export_dir (str): Optional dir to place the exported model.
Defaults to self.logdir.
Returns:
A dict that maps ExportFormats to successfully exported models.
"""
if isinstance(export_formats, str):
export_formats = [export_formats]
export_dir = export_dir or self.logdir
return self._export_model(export_formats, export_dir)