mirror of
https://github.com/wassname/ray.git
synced 2026-07-01 04:50:40 +08:00
[tune] Tensorboard logger incorrectly reports training iteration as cur timestep value
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user