From bab44837e07b022345e13d5bb2330806317d61d6 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Sat, 16 Dec 2017 23:30:15 -0800 Subject: [PATCH] [tune] Tensorboard logger incorrectly reports training iteration as cur timestep value --- python/ray/tune/logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/tune/logger.py b/python/ray/tune/logger.py index 6faefc79d..d08f67470 100644 --- a/python/ray/tune/logger.py +++ b/python/ray/tune/logger.py @@ -115,7 +115,7 @@ class _TFLogger(Logger): tag="ray/tune/{}".format(attr), simple_value=getattr(result, attr))) train_stats = tf.Summary(value=values) - self._file_writer.add_summary(train_stats, result.training_iteration) + self._file_writer.add_summary(train_stats, result.timesteps_total) def close(self): self._file_writer.close()