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

PiperOrigin-RevId: 235457599
rajedan · Feb 25, 2019 · 8be2848 · 8be2848
1 parent 30db916
commit 8be2848
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tensorflow_examples/test_models/densenet/train.py b/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):