changed initialization method for demo/gan

9 years ago · bd8613acf8
parent 70b78b74ff
commit bd8613acf8
2 changed files with 6 additions and 4 deletions
--- a/demo/gan/gan_conf_image.py
+++ b/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,
--- a/demo/gan/gan_trainer_image.py
+++ b/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()