58086fa0a4fdd7c72a9ca51746a754fbd6644b91,models/common/distributed_executor.py,DistributedExecutor,_run_evaluation,#DistributedExecutor#Any#Any#Any#Any#Any#Any#,358

Before Change

    for _ in range(eval_steps):
      test_step(strategy, model, metric, test_iterator)

    metric_result = metric.result().numpy().astype(float)
    logging.info("Step: [%d] Validation metric = %f", current_training_step,
                 metric_result)
    return metric_result

After Change

          "Both test_iterator (%s) and metrics (%s) must not be None.",
          test_iterator, metric)
      return None
    logging.info("Running evaluation after step: %s.", current_training_step)
    test_step = self._create_test_step()
    for _ in range(eval_steps):
      test_step(strategy, model, metric, test_iterator)

Instances