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

Before Change

        // 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