Model.py 文件源码

python
阅读 21 收藏 0 点赞 0 评论 0

项目:VAE_GAN 作者: takat0m0 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号