"tests/vscode:/vscode.git/clone" did not exist on "8bf97719fad546b6d49be951010e0aef87320455"
Unverified Commit 39b7dcf9 authored by Yuxin Wu's avatar Yuxin Wu Committed by GitHub
Browse files

Update build_imagenet_data.py

parent 7e4c66b2
......@@ -171,7 +171,7 @@ def _float_feature(value):
def _bytes_feature(value):
"""Wrapper for inserting bytes features into Example proto."""
if isinstance(value, six.string_types):
if six.PY3 and isinstance(value, six.text_type):
value = six.binary_type(value, encoding='utf-8')
return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))
......
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