提交 8551e07a 编写于 作者: M minqiyang

Fix flowers data read in python3

test=develop
上级 2f5a7cc4
......@@ -126,9 +126,9 @@ def reader_creator(data_file,
batch = pickle.load(f)
else:
batch = pickle.load(f, encoding='bytes')
data = batch['data']
labels = batch['label']
for sample, label in zip(data, batch['label']):
data = batch[six.b('data')]
labels = batch[six.b('label')]
for sample, label in zip(data, batch[six.b('label')]):
yield sample, int(label) - 1
if not cycle:
break
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册