bea3a486261d3f6238316224605b684db1e5bca7,Tars/models/vae.py,VAE,train,#VAE#Any#,40

Before Change

        self.other_distributions.train()

        self.optimizer.zero_grad()
        lower_bound, loss = self._elbo(train_x)

        // backprop
        loss.backward()

        // update params
        self.optimizer.step()

        return lower_bound, loss

    def test(self, test_x):
        self.decoder.eval()
        self.encoder.eval()

After Change

        self.encoder.train()
        self.other_distributions.train()

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

    def test(self, test_x, **kwargs):
        self.decoder.eval()
        self.encoder.eval()

Instances