bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/vi.py,VI,test,#VI#Any#,53

Before Change

        self.q.eval()
        self.other_distributions.eval()

        with torch.no_grad():
            lower_bound, loss = self._elbo(test_x, **kwargs)

        return lower_bound, loss

    def _elbo(self, x, **kwargs):
        

After Change

        self.q.eval()
        self.other_distributions.eval()

        return super().test(test_x, **kwargs)

Instances