2d34259281a6251cbdc67103ce8a1310010b8ceb,orbit/controller.py,Controller,evaluate,#Controller#,177

Before Change

    if eval_outputs:
      eval_outputs = tf.nest.map_structure(utils.get_value, eval_outputs)

    info = "step: {}        evaluation metric: {}".format(
        current_step, eval_outputs)
    _log_info(info)

    self.eval_summary_manager.write_summaries(eval_outputs)
    self.eval_summary_manager.flush()

After Change

    eval_output = tf.nest.map_structure(utils.get_value, eval_output or {})
    elapsed = time.time() - start

    _log(f" eval | step: {current_step: 6d} | "
         f"eval time: {elapsed: 6.1f} | "
         f"output: {_format_output(eval_output)}")

    self.eval_summary_manager.write_summaries(eval_output)
    self.eval_summary_manager.flush()

Instances