bd155f2d58357de72c4a80699fde34e2c515c540,codes/models/SFTGAN_ACD_model.py,SFTGAN_ACD_Model,__init__,#SFTGAN_ACD_Model#Any#,18

Before Change

        self.input_L = self.Tensor()
        self.input_H = self.Tensor()
        self.input_seg = self.Tensor()
        self.input_cat = self.Tensor().long()  // category

        // define networks and load pretrained models
        self.netG = networks.define_G(opt)  // G

After Change

            self.D_init_iters = train_opt["D_init_iters"] if train_opt["D_init_iters"] else 0

            if train_opt["gan_type"] == "wgan-gp":
                self.random_pt = torch.Tensor(1, 1, 1, 1).to(self.device)
                // gradient penalty loss
                self.cri_gp = GradientPenaltyLoss(tensor=self.Tensor)
                self.l_gp_w = train_opt["gp_weigth"]

Instances