def elbo(self) -> torch.Tensor: """Returns the Evidence Lower Bound associated to the object.""" elbo = compute_elbo(self.model, self) logger.debug("ELBO : %.4f" % elbo) return elbo
def elbo(self) -> float: elbo = compute_elbo(self.model, self, mode=self.mode) logger.debug("ELBO : %.4f" % elbo) return elbo