.result() returns a tensor where .numpy() will return a value.

PiperOrigin-RevId: 235457599

Author: yashk2810

tensorflow_examples/test_models/densenet/train.py

@@ -216,8 +216,10 @@ def custom_loop(self, train_dataset, test_dataset, model):
         self.test_loss_metric.reset_states()
         self.test_acc_metric.reset_states()
 
-    return (self.train_loss_metric.result(), self.train_acc_metric.result(),
-            self.test_loss_metric.result(), self.test_acc_metric.result())
+    return (self.train_loss_metric.result().numpy(),
+            self.train_acc_metric.result().numpy(),
+            self.test_loss_metric.result().numpy(),
+            self.test_acc_metric.result().numpy())
 
 
 def run_main(argv):