8c0e9b7a20e0bf25216de5121a475c77b18c02ae,pixyz/models/model.py,Model,__str__,#Model#,27

Before Change

        self.optimizer = optimizer(params, **optimizer_params)

    def __str__(self):
        prob_text = [prob.prob_text for prob in self.distributions._modules.values()]

        text = "Distributions (for training): \n  {} \n".format(", ".join(prob_text))
        text += "Loss function: \n  {} \n".format(str(self.loss_cls))
        optimizer_text = re.sub("^", " " * 2, str(self.optimizer), flags=re.MULTILINE)

After Change

                func_text.append(prob.__str__())

        text = "Distributions (for training): \n  {} \n".format(", ".join(prob_text))
        if len(func_text) > 0:
            text += "Deterministic functions (for training): \n  {} \n".format(", ".join(func_text))

        text += "Loss function: \n  {} \n".format(str(self.loss_cls))
        optimizer_text = re.sub("^", " " * 2, str(self.optimizer), flags=re.MULTILINE)
        text += "Optimizer: \n{}".format(optimizer_text)
        return text

Instances