Ver código fonte

Fixed constant noise size in training.

Kristian Schultz 4 anos atrás
pai
commit
85ffe717b6
1 arquivos alterados com 1 adições e 1 exclusões
  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,