add588d68888e8ba869ffce17d214c48e41ca019,official/utils/logs/logger.py,BenchmarkFileLogger,log_metric,#BenchmarkFileLogger#Any#Any#Any#Any#Any#,166

Before Change

    
    metric = _process_metric_to_json(name, value, unit, global_step, extras)
    if metric:
      with tf.gfile.GFile(
          os.path.join(self._logging_dir, METRIC_LOG_FILE_NAME), "a") as f:
        try:
          json.dump(metric, f)
          f.write("\n")
        except (TypeError, ValueError) as e:
          tf.logging.warning("Failed to dump metric to log file: "
                             "name %s, value %s, error %s", name, value, e)

  def log_run_info(self, model_name, dataset_name, run_params, test_id=None):
    Collect most of the TF runtime information for the local env.

    The schema of the run info follows official/benchmark/datastore/schema.

After Change

      try:
        json.dump(metric, self._metric_file_handler)
        self._metric_file_handler.write("\n")
        self._metric_file_handler.flush()
      except (TypeError, ValueError) as e:
        tf.logging.warning("Failed to dump metric to log file: "
                           "name %s, value %s, error %s", name, value, e)

Instances