changed initialization method for demo/gan

demo/gan/gan_conf_image.py

@@ -117,10 +117,12 @@ def generator(noise):
     """
     generator generates a sample given noise
     """
-    param_attr = ParamAttr(is_static=is_discriminator_training)
+    param_attr = ParamAttr(is_static=is_discriminator_training,
+                           initial_mean=0.0,
+                           initial_std=0.02)
     bias_attr = ParamAttr(is_static=is_discriminator_training,
-                           initial_mean=1.0,
-                           initial_std=0)
+                           initial_mean=0.0,
+                           initial_std=0.0)
     
     param_attr_bn=ParamAttr(is_static=is_discriminator_training,
                            initial_mean=1.0,

demo/gan/gan_trainer_image.py

@@ -197,7 +197,7 @@ def main():
     
     curr_train = "dis"
     curr_strike = 0
-    MAX_strike = 100
+    MAX_strike = 10
      
     for train_pass in xrange(100):
         dis_trainer.startTrainPass()