1c5c2f50da5ae101077c27fdac2a12fb1619ec86,models/pix2pix_model.py,Pix2PixModel,set_input,#Pix2PixModel#Any#,63

Before Change

        AtoB = self.opt.which_direction == "AtoB"
        input_A = input["A" if AtoB else "B"]
        input_B = input["B" if AtoB else "A"]
        self.input_A.resize_(input_A.size()).copy_(input_A)
        self.input_B.resize_(input_B.size()).copy_(input_B)
        self.image_paths = input["A_paths" if AtoB else "B_paths"]

    def forward(self):

After Change

    def set_input(self, input):
        AtoB = self.opt.which_direction == "AtoB"
        input_A = input["A" if AtoB else "B"]
        input_B = input["B" if AtoB else "A"]
        if len(self.gpu_ids) > 0:
            input_A = input_A.cuda(self.gpu_ids[0], async=True)
            input_B = input_B.cuda(self.gpu_ids[0], async=True)
        self.input_A = input_A
        self.input_B = input_B
        self.image_paths = input["A_paths" if AtoB else "B_paths"]

    def forward(self):
        self.real_A = Variable(self.input_A)

Instances