Merge pull request #8565 from djdongjin:label_cast

PiperOrigin-RevId: 315024646

Merge pull request #8565 from djdongjin:label_cast
PiperOrigin-RevId: 315024646
fc846697 · A. Unique TensorFlower · b4df77a0 · 55b7cb66 · fc846697
Commit fc846697 authored Jun 05, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

official/vision/image_classification/dataset_factory.py official/vision/image_classification/dataset_factory.py +0 -1

No files found.
--- a/official/vision/image_classification/dataset_factory.py
+++ b/official/vision/image_classification/dataset_factory.py
@@ -477,7 +477,6 @@ class DatasetBuilder:
    parsed = tf.io.parse_single_example(record, keys_to_features)

    label = tf.reshape(parsed['image/class/label'], shape=[1])
-    label = tf.cast(label, dtype=tf.int32)

    # Subtract one so that labels are in [0, 1000)
    label -= 1