From 10cbcced7e4fb11407e34f6680d4058bea53f7a8 Mon Sep 17 00:00:00 2001
From: LorenzoCevolani <46997995+LorenzoCevolani@users.noreply.github.com>
Date: Sat, 27 Jul 2019 20:08:35 +0200
Subject: [PATCH] Correctly setting the input to Train (#3853)

In the ResNetTrainActor class, the data are now exactly build using the Train flag for the cifar_input script.
---
 examples/resnet/resnet_main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/resnet/resnet_main.py b/examples/resnet/resnet_main.py
index 3a6bb6a47..3d0089d77 100644
--- a/examples/resnet/resnet_main.py
+++ b/examples/resnet/resnet_main.py
@@ -107,7 +107,7 @@ class ResNetTrainActor(object):
         with tf.device("/gpu:0" if num_gpus > 0 else "/cpu:0"):
             # Build the model.
             images, labels = cifar_input.build_input(data, hps.batch_size,
-                                                     dataset, False)
+                                                     dataset, True)
             self.model = resnet_model.ResNet(hps, images, labels, "train")
             self.model.build_graph()
             config = tf.ConfigProto(allow_soft_placement=True)