Commit 3f9fa67f authored by Max Lapan's avatar Max Lapan Committed by Yuxin Wu

Use monitors in DQN and A3C (#166)

parent 5cbf81c2
......@@ -112,5 +112,5 @@ class Evaluator(Triggerable):
t = time.time() - t
if t > 10 * 60: # eval takes too long
self.eval_episode = int(self.eval_episode * 0.94)
self.trainer.add_scalar_summary('mean_score', mean)
self.trainer.add_scalar_summary('max_score', max)
self.trainer.monitors.put('mean_score', mean)
self.trainer.monitors.put('max_score', max)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment