8f6a6f153781d0908fb0904349aae844494026ea,models/shiftnet_model.py,ShiftNetModel,initialize,#ShiftNetModel#Any#,17

Before Change

            networks.print_network(self.netG)
            if self.isTrain:
                networks.print_network(self.netD)
            print("-----------------------------------------------")

    def set_input(self, input):
        input_A = input["A"]
        input_B = input["B"]

After Change

        if not self.isTrain or opt.continue_train:
            self.load_networks(opt.which_epoch)

        self.print_networks(opt.verbose)

    def set_input(self, input):
        real_A = input["A"].to(self.device)
        real_B = input["B"].to(self.device)

Instances