0789cc08dd92a43422469fc4973ecc972d032421,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

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

Instances