remove cast in preprocess

55b7cb66 · Jin Dong · a2d0168e · 55b7cb66
Commit 55b7cb66 authored Jun 02, 2020 by Jin Dong
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