def __init__(self, params_G, params_D): super(GAN, self).__init__() self.fake = None self.G = CNN.Unet(params_G) self.D = CNN.Encoder(params_D)