提交 de136df8 编写于 作者: W wangyang59

fixed small bug in demo/gan/mnist and is stable version

上级 c3ebff5e
...@@ -29,9 +29,9 @@ is_discriminator = mode == "discriminator" ...@@ -29,9 +29,9 @@ is_discriminator = mode == "discriminator"
print('mode=%s' % mode) print('mode=%s' % mode)
# the dim of the noise (z) as the input of the generator network # the dim of the noise (z) as the input of the generator network
noise_dim = 10 noise_dim = 10
# the dim of the hidden layer # the dim of the hidden layer
hidden_dim = 15 hidden_dim = 10
# the dim of the generated sample # the dim of the generated sample
sample_dim = 2 sample_dim = 2
......
...@@ -91,7 +91,7 @@ def load_mnist_data(imageFile): ...@@ -91,7 +91,7 @@ def load_mnist_data(imageFile):
data = data / 255.0 * 2.0 - 1.0 data = data / 255.0 * 2.0 - 1.0
f.close() f.close()
return data return data.astype('float32')
def load_cifar_data(cifar_path): def load_cifar_data(cifar_path):
batch_size = 10000 batch_size = 10000
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册