remove calls to metrics.result() inside the train_step

PiperOrigin-RevId: 364096705

official/nlp/tasks/translation.py

@@ -302,7 +302,6 @@ class TranslationTask(base_task.Task):
     logs = {self.loss: loss}
     if metrics:
       self.process_metrics(metrics, inputs["targets"], outputs)
-      logs.update({m.name: m.result() for m in metrics})
     return logs
 
   def validation_step(self, inputs, model: tf.keras.Model, metrics=None):

official/vision/beta/tasks/image_classification.py

@@ -199,7 +199,6 @@ class ImageClassificationTask(base_task.Task):
     logs = {self.loss: loss}
     if metrics:
       self.process_metrics(metrics, labels, outputs)
-      logs.update({m.name: m.result() for m in metrics})
     elif model.compiled_metrics:
       self.process_compiled_metrics(model.compiled_metrics, labels, outputs)
       logs.update({m.name: m.result() for m in model.metrics})
@@ -228,7 +227,6 @@ class ImageClassificationTask(base_task.Task):
     logs = {self.loss: loss}
     if metrics:
       self.process_metrics(metrics, labels, outputs)
-      logs.update({m.name: m.result() for m in metrics})
     elif model.compiled_metrics:
       self.process_compiled_metrics(model.compiled_metrics, labels, outputs)
       logs.update({m.name: m.result() for m in model.metrics})

official/vision/beta/tasks/maskrcnn.py

@@ -291,7 +291,6 @@ class MaskRCNNTask(base_task.Task):
     if metrics:
       for m in metrics:
         m.update_state(losses[m.name])
-        logs.update({m.name: m.result()})
 
     return logs