bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/ml.py,ML,test,#ML#Any#,34

Before Change

        self.p.eval()

        with torch.no_grad():
            log_like = self.p.log_likelihood(test_x)
            loss = -torch.mean(log_like)

        return log_like, loss

After Change

        self.p.eval()
        self.other_distributions.eval()

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

Instances