Fix flowers data read in python3

test=develop
6 years ago · 8551e07abc
parent 2f5a7cc470
commit 8551e07abc
1 changed files with 3 additions and 3 deletions
--- a/python/paddle/dataset/flowers.py
+++ b/python/paddle/dataset/flowers.py
@ -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