bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/ml.py,ML,train,#ML#Any#,19

Before Change

        self.optimizer.zero_grad()
        log_like = self.p.log_likelihood(train_x)
        loss = -torch.mean(log_like)

        // backprop
        loss.backward()

        // update params
        self.optimizer.step()

        return log_like, loss

    def test(self, test_x):
        self.p.eval()

After Change

        self.p.train()
        self.other_distributions.train()

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

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

Instances