bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/vae.py,VAE,__init__,#VAE#Any#Any#Any#Any#Any#Any#,12

Before Change

        self.optimizer = optimizer(params, **optimizer_params)

        // set input_var
        self.input_var = copy(self.encoder.cond_var)
        for distribution in self.other_distributions:
            self.input_var += copy(distribution.cond_var)
        self.input_var = list(set(self.input_var))

    def train(self, train_x):
        self.decoder.train()

After Change

                 regularizer=[],
                 optimizer=optim.Adam,
                 optimizer_params={}):
        super().__init__()

        self.encoder = encoder
        self.decoder = decoder
        self.other_distributions = nn.ModuleList(tolist(other_distributions))

Instances