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

Before Change

        self.opt = opt
        self.isTrain = opt.isTrain
        // define tensors
        self.input_A = self.Tensor(opt.batchSize, opt.input_nc,
                                   opt.fineSize, opt.fineSize)
        self.input_B = self.Tensor(opt.batchSize, opt.output_nc,
                                   opt.fineSize, opt.fineSize)

        // batchsize should be 1 for mask_global

After Change

        self.visual_names = ["real_A", "fake_B", "real_B"]
        // specify the models you want to save to the disk. The program will call base_model.save_networks and base_model.load_networks
        if self.isTrain:
            self.model_names = ["G", "D"]
        else:  // during test time, only load Gs
            self.model_names = ["G"]

Instances