[rllib] on_train_result results do not get logged (#3865)

This commit is contained in:
Eric Liang
2019-02-01 20:32:07 -08:00
committed by Richard Liaw
parent e0f82fd260
commit 0f81bc9a33
2 changed files with 17 additions and 3 deletions
+10 -1
View File
@@ -202,7 +202,7 @@ class Trainable(object):
timesteps_since_restore=self._timesteps_since_restore,
iterations_since_restore=self._iterations_since_restore)
self._result_logger.on_result(result)
self._log_result(result)
return result
@@ -415,6 +415,15 @@ class Trainable(object):
"""
pass
def _log_result(self, result):
"""Subclasses can optionally override this to customize logging.
Args:
result (dict): Training result returned by _train().
"""
self._result_logger.on_result(result)
def _stop(self):
"""Subclasses should override this for any cleanup on stop."""
pass