0c81011805c9ab4d6f7f314f674d39e51f5ba8eb,Tars/losses/gan_loss.py,GANLoss,estimate,#GANLoss#Any#Any#,26

Before Change

            sample_data = get_dict_values(sample_data_dict, self.d.var)[0]
            return self.d_criterion(sample_data, sample)  // TODO: detach

        return self.g_criterion(sample)

    @staticmethod
    def d_criterion(sample_data, sample):
        return - torch.log(sample_data) - torch.log(1 - sample)

After Change

        // sample y from x
        y_dict = self.d.sample(x_dict)
        y = get_dict_values(y_dict, self.d.var)[0]

        return self.bce_loss(y, t_data)

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

Instances