def __init__(self, z_dim, batch_size):
self.z_dim = z_dim
self.batch_size = batch_size
self.lr = 0.0001
self.gamma = 0.5
# -- encoder -------
self.enc = Encoder([3, 64, 128, 256], 2048, z_dim)
# -- decoder -------
self.dec = Decoder(z_dim, [256, 128, 32, 3])
# -- discriminator --
self.disc = Discriminator([3, 32, 128, 256, 256], 512)
评论列表
文章目录