[tune] Tensorboard logger incorrectly reports training iteration as cur timestep value

This commit is contained in:
Eric Liang
2017-12-16 23:30:15 -08:00
committed by GitHub
parent d21ea0ca45
commit bab44837e0
+1 -1
View File
@@ -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()