Prechádzať zdrojové kódy

Fixed constant noise size in training.

Kristian Schultz 4 rokov pred
rodič
commit
85ffe717b6
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  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,