test=develop, fix mnist will segment fault (#18083)

6 years ago · ff59866502
parent d9270af931
commit ff59866502
1 changed files with 4 additions and 1 deletions
--- a/python/paddle/dataset/mnist.py
+++ b/python/paddle/dataset/mnist.py
@ -78,7 +78,10 @@ def reader_creator(image_filename, label_filename, buffer_size):
                        buffer_size, rows * cols)).astype('float32')
                    offset_img += struct.calcsize(fmt_images)

-                    images = images / 255.0 * 2.0 - 1.0
+                    images = images / 255.0
+                    images = images * 2.0
+                    images = images - 1.0
+
                    for i in range(buffer_size):
                        yield images[i, :], int(labels[i])