Skip to content

Commit

Permalink
Pass iteration_number to metrics.log_metric as keyword argument. Fix …
Browse files Browse the repository at this point in the history
…bug where it was being passed to the timestamp positional argument. (#62)

Overriding CI fail due to urgency and clear fix.
  • Loading branch information
owen-t authored and ddavydenko committed Nov 30, 2019
1 parent d28dc42 commit 41fdc80
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion smdebug/core/hook.py
Expand Up @@ -569,7 +569,8 @@ def _write_scalars(self):
write_tb = scalar_obj.write_tb
write_event = scalar_obj.write_event
if self.metrics_writer and sm_metric:
self.metrics_writer.log_metric(scalar_name, scalar_val, self.mode_steps[self.mode])
self.metrics_writer.log_metric(scalar_name, scalar_val,
iteration_number=self.mode_steps[self.mode])
if write_tb:
tb_writer = self._maybe_get_tb_writer()
if tb_writer:
Expand Down

0 comments on commit 41fdc80

Please sign in to comment.