From 0df8af993b7e54aebd13f452ec2909af5c5ab83b Mon Sep 17 00:00:00 2001 From: wangyang59 Date: Tue, 8 Nov 2016 09:50:57 -0800 Subject: [PATCH] gan working on mnist --- demo/gan/gan_conf_image.py | 6 +++--- demo/gan/gan_trainer_image.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/demo/gan/gan_conf_image.py b/demo/gan/gan_conf_image.py index 00711730d56..83fa34fabcf 100644 --- a/demo/gan/gan_conf_image.py +++ b/demo/gan/gan_conf_image.py @@ -36,9 +36,9 @@ s2, s4 = int(sample_dim/2), int(sample_dim/4), s8, s16 = int(sample_dim/8), int(sample_dim/16) settings( - batch_size=100, - learning_rate=1e-4, - learning_method=AdamOptimizer() + batch_size=128, + learning_rate=2e-4, + learning_method=AdamOptimizer(beta1=0.5) ) def conv_bn(input, channels, imgSize, num_filters, output_x, stride, name, diff --git a/demo/gan/gan_trainer_image.py b/demo/gan/gan_trainer_image.py index 8f1e17b9c74..51bbbe8f1f4 100644 --- a/demo/gan/gan_trainer_image.py +++ b/demo/gan/gan_trainer_image.py @@ -88,7 +88,7 @@ def load_mnist_data(imageFile): for i in range(n): pixels = [] for j in range(28 * 28): - pixels.append(float(ord(f.read(1))) / 255.0) + pixels.append(float(ord(f.read(1))) / 255.0 * 2.0 - 1.0) data[i, :] = pixels f.close() -- GitLab