Deep GAN

ragavvenkatesan · Mar 4, 2017 · 9652a17 · 9652a17
1 parent 9fedff4
commit 9652a17
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pantry/tutorials/gan.py b/pantry/tutorials/gan.py
@@ -266,6 +266,9 @@ def deep_gan (dataset, verbose = 1 ):
 
     Returns:
         net: A Network object.
+
+    Notes:
+        This is not setup properly therefore does not learn at the moment.
     """
     if verbose >=2:
         print (".. Creating a GAN network")
@@ -330,15 +333,15 @@ def deep_gan (dataset, verbose = 1 ):
                     origin = "z",
                     id = "G1",
                     num_neurons = 128,
-                    activation = 'tanh',
+                    activation = 'relu',
                     verbose = verbose
                     ) 
 
     net.add_layer ( type = "dot_product",
                     origin = "G1",
                     id = "G2",
                     num_neurons = 384,
-                    activation = 'tanh',
+                    activation = 'relu',
                     verbose = verbose
                     )