bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/vi.py,VI,train,#VI#Any#,37

Before Change

        lower_bound, loss = self._elbo(train_x, **kwargs)

        // backprop
        loss.backward()

        // update params
        self.optimizer.step()

        return lower_bound, loss

    def test(self, test_x=None, **kwargs):
        self.p.eval()
        self.q.eval()

After Change

        self.q.train()
        self.other_distributions.train()

        return super().train(train_x, **kwargs)

    def test(self, test_x, **kwargs):
        self.p.eval()
        self.q.eval()

Instances