search

ethanhe42 / GAN-MNIST

Generative Adversarial Network for MNIST with tensorflow
https://yihui-he.github.io/blog/gan-on-mnist-with-tensorflow
MIT License
185 stars 70 forks source link

Problem with load.py #5

Open Karmans1 opened 6 years ago

Karmans1 commented 6 years ago

I apologize if its a basic question, I'm new to Tensorflow. When I run train.py I get the following error while loading the MNIST dataset:

Traceback (most recent call last): File "/home/m132/Desktop/Python/Learning GANs/GAN-MNIST-DC/train.py", line 19, in trX, vaX, teX, trY, vaY, teY = mnist_with_valid_set() File "/home/m132/Desktop/Python/Learning GANs/GAN-MNIST-DC/Load_Data.py", line 31, in mnist_with_valid_set trX, teX, trY, teY = mnist() File "/home/m132/Desktop/Python/Learning GANs/GAN-MNIST-DC/Load_Data.py", line 11, in mnist trX = loaded[16:].reshape((60000,28*28)).astype(float) ValueError: cannot reshape array of size 9912406 into shape (60000,784)

Any help would be appreciated.

wangtianrui commented 6 years ago

you can use input_data.py (demo) to load