Commit 0c8dbe54 authored by Marianne Linhares Monteiro's avatar Marianne Linhares Monteiro Committed by GitHub
Browse files

Removing cast, since it was fixed in another PR

parent 4359c6d4
......@@ -69,8 +69,7 @@ class Cifar10DataSet(object):
image = tf.transpose(tf.reshape(image, [DEPTH, HEIGHT, WIDTH]), [1, 2, 0])
label = tf.cast(features['label'], tf.int32)
# Custom preprocessing .
image = tf.cast(image, tf.float32)
# Custom preprocessing.
image = self.preprocess(image)
return image, label
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment