vae_m1.py 文件源码

def __init__(self, conf, name="vae"):
        conf.check()
        self.encoder, self.decoder = self.build(conf)
        self.name = name

        self.optimizer_encoder = optimizers.Adam(alpha=conf.learning_rate, beta1=conf.gradient_momentum)
        self.optimizer_encoder.setup(self.encoder)
        # self.optimizer_encoder.add_hook(optimizer.WeightDecay(0.001))
        self.optimizer_encoder.add_hook(GradientClipping(conf.gradient_clipping))

        self.optimizer_decoder = optimizers.Adam(alpha=conf.learning_rate, beta1=conf.gradient_momentum)
        self.optimizer_decoder.setup(self.decoder)
        # self.optimizer_decoder.add_hook(optimizer.WeightDecay(0.001))
        self.optimizer_decoder.add_hook(GradientClipping(conf.gradient_clipping))

        self.type_pz = conf.type_pz
        self.type_qz = conf.type_qz