8d77bc5f267a49ed890222039f9ee058cca7f22f,keras/callbacks.py,TensorBoard,_stop_trace,#TensorBoard#Any#,2398

Before Change

      with tf.summary.record_if(True):
        // TODO(b/126388999): Remove step info in the summary name.
        tf.summary.trace_export(name="batch_%d" % batch, step=batch)
    tf.profiler.experimental.stop()
    self._is_tracing = False

  def _collect_learning_rate(self, logs):
    lr_schedule = getattr(self.model.optimizer, "lr", None)

After Change

      with tf.summary.record_if(True):
        // TODO(b/126388999): Remove step info in the summary name.
        tf.summary.trace_export(name="batch_%d" % batch, step=batch)
    self._stop_profiler()
    self._is_tracing = False

  def _collect_learning_rate(self, logs):
    lr_schedule = getattr(self.model.optimizer, "lr", None)

Instances