From: Asim Shankar <ashankar@google.com>
Date: Mon, 26 Feb 2018 19:10:20 +0000 (-0800)
Subject: eager/examples/resnet50: Fix breakage.
X-Git-Tag: upstream/v1.7.0~117
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=2513479d7b39235f9504ede2bf6f61cb78aae923;p=platform%2Fupstream%2Ftensorflow.git

eager/examples/resnet50: Fix breakage.

PiperOrigin-RevId: 187050075
---

diff --git a/tensorflow/contrib/eager/python/examples/resnet50/resnet50_test.py b/tensorflow/contrib/eager/python/examples/resnet50/resnet50_test.py
index c106ab0..65dcc53 100644
--- a/tensorflow/contrib/eager/python/examples/resnet50/resnet50_test.py
+++ b/tensorflow/contrib/eager/python/examples/resnet50/resnet50_test.py
@@ -194,11 +194,11 @@ class ResNet50Benchmarks(tf.test.Benchmark):
     with tf.device(device):
       images, _ = random_batch(batch_size)
       for _ in xrange(num_burn):
-        model(images).cpu()
+        model(images, training=False).cpu()
       gc.collect()
       start = time.time()
       for _ in xrange(num_iters):
-        model(images).cpu()
+        model(images, training=False).cpu()
       self._report(label, start, num_iters, device, batch_size, data_format)
 
   def benchmark_eager_apply(self):