Commit 830551b8 authored by Rajagopal Ananthanarayanan's avatar Rajagopal Ananthanarayanan Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 355082063
parent df0179e0
...@@ -320,10 +320,7 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator): ...@@ -320,10 +320,7 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator):
# `self.validation_loss` metric was not updated, because the validation # `self.validation_loss` metric was not updated, because the validation
# loss was not returned from the task's `validation_step` method. # loss was not returned from the task's `validation_step` method.
logging.info("The task did not report validation loss.") logging.info("The task did not report validation loss.")
if aggregated_logs:
# Merges additional metrics from `reduce_aggregated_logs` method.
# By default, the method in `base_task.Task` returns an empty dict, while
# the subclass may override it to return metrics computed on host.
metrics = self.task.reduce_aggregated_logs(aggregated_logs) metrics = self.task.reduce_aggregated_logs(aggregated_logs)
logs.update(metrics) logs.update(metrics)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment