bd155f2d58357de72c4a80699fde34e2c515c540,codes/models/SFTGAN_ACD_model.py,SFTGAN_ACD_Model,feed_data,#SFTGAN_ACD_Model#Any#Any#Any#,131

Before Change

        self.var_L = Variable(self.input_L, volatile=volatile)
        // seg
        input_seg = data["seg"]
        self.input_seg.resize_(input_seg.size()).copy_(input_seg)
        self.var_seg = Variable(self.input_seg, volatile=volatile)
        // category
        input_cat = data["category"]
        self.input_cat.resize_(input_cat.size()).copy_(input_cat)
        self.var_cat = Variable(self.input_cat, volatile=volatile)

        if need_HR:  // train or val
            input_H = data["HR"]
            self.input_H.resize_(input_H.size()).copy_(input_H)
            self.var_H = Variable(self.input_H, volatile=volatile)

    def optimize_parameters(self, step):
        // G
        self.optimizer_G_SFT.zero_grad()

After Change

        // seg
        self.var_seg = data["seg"].to(self.device)
        // category
        self.var_cat = data["category"].long().to(self.device)

        if need_HR:  // train or val
            self.var_H = data["HR"].to(self.device)

    def optimize_parameters(self, step):
        // G
        self.optimizer_G_SFT.zero_grad()

Instances