6c17a1019665e1bbaee6826b3204bb7d14996a9f,Tars/models/vi.py,VI,__init__,#VI#Any#Any#Any#Any#Any#Any#,9

Before Change

        self.p = p
        self.q = approximate_dist
        self.other_distributions = nn.ModuleList(tolist(other_distributions))

        // set losses
        elbo = ELBO(self.p, self.q)
        other_losses = other_losses
        loss_cls = (-elbo + other_losses).mean()
        self.loss_cls = loss_cls
        self.test_loss_cls = loss_cls
        self.loss_text = str(loss_cls)

        // set params and optim
        q_params = list(self.q.parameters())
        p_params = list(self.p.parameters())
        other_params = list(self.other_distributions.parameters())
        params = q_params + p_params + other_params
        self.optimizer = optimizer(params, **optimizer_params)

    def train(self, train_x, **kwargs):
        self.p.train()

After Change

                 optimizer=optim.Adam,
                 optimizer_params={}):

        distributions = nn.ModuleList([p, approximate_dist] + tolist(other_distributions))
        super().__init__(distributions)

        // set losses
        elbo = ELBO(p, approximate_dist)

Instances