bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/vi.py,VI,__init__,#VI#Any#Any#Any#Any#Any#Any#,11

Before Change

        // set input_var
        self.input_var = copy(self.q.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=None, **kwargs):
        self.p.train()

After Change

        loss_cls = (-elbo + other_losses).mean()
        self.loss_cls = loss_cls
        self.test_loss_cls = loss_cls
        self.loss_text = str(loss_cls)

        // set params and optim
        q_params = list(self.q.parameters())
        p_params = list(self.p.parameters())

Instances