Explorar o código

Fixed constant noise size in training.

Kristian Schultz %!s(int64=4) %!d(string=hai) anos
pai
achega
85ffe717b6
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      library/generators/SimpleGan.py

+ 1 - 1
library/generators/SimpleGan.py

@@ -117,7 +117,7 @@ class SimpleGan(GanBaseClass):
                 self.discriminator.train_on_batch(X, y_dis)
 
                 #Tricking the noised input of the Generator as real data
-                noise = np.random.normal(0, 1, [self.batchSize, 100])
+                noise = np.random.normal(0, 1, [self.batchSize, self.noiseSize])
                 y_gen = np.ones(self.batchSize)
 
                 # During the training of gan,