message: Fixes Github Step number of batch_steps_per_second
diff: @@ -2445,7 +2445,8 @@ class TensorBoard(Callback, version_utils.TensorBoardVersionSelector):
  if self.write_steps_per_second:
  batch_run_time = time.time() - self._batch_start_time
  self._train_accumulated_time += batch_run_time
  - summary_ops_v2.scalar('batch_steps_per_second', 1. / batch_run_time)
  + summary_ops_v2.scalar('batch_steps_per_second', 1. / batch_run_time,
  + step=self._train_step)
  if not self._should_trace:
  return
  