Method forward

diff2flow/ddpm.py:264–265

(self, *args, **kwargs)

        return loss, loss_dict

    def forward(self, *args, **kwargs):
        return self.training_losses(*args, **kwargs)

    """ Sampling """