We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c3ebff5 commit de136dfCopy full SHA for de136df
demo/gan/gan_conf.py
@@ -29,9 +29,9 @@
29
30
print('mode=%s' % mode)
31
# the dim of the noise (z) as the input of the generator network
32
-noise_dim = 10
+noise_dim = 10
33
# the dim of the hidden layer
34
-hidden_dim = 15
+hidden_dim = 10
35
# the dim of the generated sample
36
sample_dim = 2
37
demo/gan/gan_trainer.py
@@ -91,7 +91,7 @@ def load_mnist_data(imageFile):
91
data = data / 255.0 * 2.0 - 1.0
92
93
f.close()
94
- return data
+ return data.astype('float32')
95
96
def load_cifar_data(cifar_path):
97
batch_size = 10000
0 commit comments