"tests/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "8c644744b851eb5b454b1acf93a135ab22ffbd80"
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,12 +320,9 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator):
# `self.validation_loss` metric was not updated, because the validation
# loss was not returned from the task's `validation_step` method.
logging.info("The task did not report validation loss.")
# 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)
logs.update(metrics)
if aggregated_logs:
metrics = self.task.reduce_aggregated_logs(aggregated_logs)
logs.update(metrics)
if self._checkpoint_exporter:
self._checkpoint_exporter.maybe_export_checkpoint(
......
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