Model.py 文件源码

def __init__(self,  z_dim, batch_size):
        self.batch_size = batch_size
        self.z_dim = z_dim

        # -- generator -----
        self.gen = Generator([64, 128, 256, 512, 512, 512, 512, 512],
                             [512, 512, 512, 512, 256, 128, 64],
                             256, 256, 3)

        # -- discriminator --
        self.disc = Discriminator([64, 128, 256, 512])

        # -- learning parms ---
        self.lr = 0.0002
        self.Lambda = 100.0