def forward(self, image, hint):
    data = {'A': image[:, 0:1, :, :], 'B': image[:, 1:3, :, :], 'hint_B': hint[:, 0:2, :, :], 'mask_B': hint[:, 2:3, :, :]}
    self.model.set_input(data)
    self.model.forward()
    fake_reg = torch.cat((self.model.real_A, self.model.fake_B_reg), dim=1)
    return fake_reg
def forward(self, image, hint):
    data = {'A': image[:, 0:1, :, :], 'B': image[:, 1:3, :, :], 'hint_B': hint[:, 0:2, :, :], 'mask_B': hint[:, 2:3, :, :]}
    self.model.set_input(data)
    self.model.forward()
    fake_reg = torch.cat((self.model.real_A, self.model.fake_B_reg), dim=1)
    return fake_reg