Update image_wgan.py

image_wgan.py

@@ -106,14 +106,12 @@ class ImageWgan:
 
     def generate(
         self,
-        sample_folder: str = 'samples',
-        seed: int = 'seed'
+        sample_folder: str = 'samples'
     ):
         if not exists(sample_folder):
             mkdir(sample_folder)
 
         Tensor = torch.cuda.FloatTensor if self.use_cuda else torch.FloatTensor
-        np.random.seed(seed)
         z = Variable(Tensor(np.random.normal(0, 1, (self.image_shape[0], self.latent_space_dimension))))
         gen_imgs = self.generator(z)
         generator_loss = -torch.mean(self.discriminator(gen_imgs))