[rllib] Sync filters at end of iteration not start; hierarchical docs (#3769)

This commit is contained in:
Eric Liang
2019-01-15 16:25:25 -08:00
committed by Richard Liaw
parent 3918934dfd
commit 401e656b95
4 changed files with 39 additions and 6 deletions
+3 -1
View File
@@ -271,6 +271,8 @@ class Agent(Trainable):
ev.set_global_vars.remote(self.global_vars)
logger.debug("updated global vars: {}".format(self.global_vars))
result = Trainable.train(self)
if (self.config.get("observation_filter", "NoFilter") != "NoFilter"
and hasattr(self, "local_evaluator")):
FilterManager.synchronize(
@@ -280,12 +282,12 @@ class Agent(Trainable):
logger.debug("synchronized filters: {}".format(
self.local_evaluator.filters))
result = Trainable.train(self)
if self.config["callbacks"].get("on_train_result"):
self.config["callbacks"]["on_train_result"]({
"agent": self,
"result": result,
})
return result
@override(Trainable)